1 efd2a263 2018-01-19 stsp /*
  2 efd2a263 2018-01-19 stsp  * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
  3 efd2a263 2018-01-19 stsp  *
  4 efd2a263 2018-01-19 stsp  * Permission to use, copy, modify, and distribute this software for any
  5 efd2a263 2018-01-19 stsp  * purpose with or without fee is hereby granted, provided that the above
  6 efd2a263 2018-01-19 stsp  * copyright notice and this permission notice appear in all copies.
  7 efd2a263 2018-01-19 stsp  *
  8 efd2a263 2018-01-19 stsp  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 efd2a263 2018-01-19 stsp  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 efd2a263 2018-01-19 stsp  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 efd2a263 2018-01-19 stsp  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 efd2a263 2018-01-19 stsp  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 efd2a263 2018-01-19 stsp  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 efd2a263 2018-01-19 stsp  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 efd2a263 2018-01-19 stsp  */
 16 efd2a263 2018-01-19 stsp 
 17 efd2a263 2018-01-19 stsp #include <sys/queue.h>
 18 efd2a263 2018-01-19 stsp 
 19 efd2a263 2018-01-19 stsp #include <stdio.h>
 20 96f5e8b3 2018-01-23 stsp #include <stdlib.h>
 21 b2f7af54 2018-11-11 stsp #include <stdint.h>
 22 96f5e8b3 2018-01-23 stsp #include <string.h>
 23 efd2a263 2018-01-19 stsp #include <zlib.h>
 24 efd2a263 2018-01-19 stsp #include <sha1.h>
 25 788c352e 2018-06-16 stsp #include <time.h>
 26 efd2a263 2018-01-19 stsp 
 27 efd2a263 2018-01-19 stsp #include "got_error.h"
 28 efd2a263 2018-01-19 stsp #include "got_repository.h"
 29 efd2a263 2018-01-19 stsp #include "got_object.h"
 30 efd2a263 2018-01-19 stsp 
 31 718b3ab0 2018-03-17 stsp #include "got_lib_delta.h"
 32 718b3ab0 2018-03-17 stsp #include "got_lib_path.h"
 33 63581804 2018-07-09 stsp #include "got_lib_inflate.h"
 34 efd2a263 2018-01-19 stsp 
 35 885d3e02 2018-01-27 stsp #ifndef MIN
 36 885d3e02 2018-01-27 stsp #define	MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
 37 885d3e02 2018-01-27 stsp #endif
 38 885d3e02 2018-01-27 stsp 
 39 c3703302 2018-01-23 stsp struct got_delta *
 40 c336f889 2018-07-23 stsp got_delta_open(off_t offset, size_t tslen, int type, size_t size,
 41 c336f889 2018-07-23 stsp     off_t data_offset, uint8_t *delta_buf, size_t delta_len)
 42 96f5e8b3 2018-01-23 stsp {
 43 c3703302 2018-01-23 stsp 	struct got_delta *delta;
 44 96f5e8b3 2018-01-23 stsp 
 45 14118581 2018-11-05 stsp 	delta = malloc(sizeof(*delta));
 46 c3703302 2018-01-23 stsp 	if (delta == NULL)
 47 96f5e8b3 2018-01-23 stsp 		return NULL;
 48 96f5e8b3 2018-01-23 stsp 
 49 c3703302 2018-01-23 stsp 	delta->type = type;
 50 c3703302 2018-01-23 stsp 	delta->offset = offset;
 51 0e22967e 2018-02-11 stsp 	delta->tslen = tslen;
 52 c3703302 2018-01-23 stsp 	delta->size = size;
 53 bdd2fbb3 2018-02-11 stsp 	delta->data_offset = data_offset;
 54 a53d2f13 2018-03-17 stsp 	delta->delta_buf = delta_buf;
 55 a53d2f13 2018-03-17 stsp 	delta->delta_len = delta_len;
 56 c3703302 2018-01-23 stsp 	return delta;
 57 96f5e8b3 2018-01-23 stsp }
 58 96f5e8b3 2018-01-23 stsp 
 59 96f5e8b3 2018-01-23 stsp void
 60 c3703302 2018-01-23 stsp got_delta_close(struct got_delta *delta)
 61 96f5e8b3 2018-01-23 stsp {
 62 a53d2f13 2018-03-17 stsp 	free(delta->delta_buf);
 63 c3703302 2018-01-23 stsp 	free(delta);
 64 96f5e8b3 2018-01-23 stsp }
 65 96f5e8b3 2018-01-23 stsp 
 66 efd2a263 2018-01-19 stsp const struct got_error *
 67 96f5e8b3 2018-01-23 stsp got_delta_chain_get_base_type(int *type, struct got_delta_chain *deltas)
 68 96f5e8b3 2018-01-23 stsp {
 69 c3703302 2018-01-23 stsp 	struct got_delta *delta;
 70 96f5e8b3 2018-01-23 stsp 
 71 6691714a 2018-01-23 stsp 	/* The first delta in the chain should represent the base object. */
 72 6691714a 2018-01-23 stsp 	delta = SIMPLEQ_FIRST(&deltas->entries);
 73 6691714a 2018-01-23 stsp 	if (delta->type == GOT_OBJ_TYPE_COMMIT ||
 74 6691714a 2018-01-23 stsp 	    delta->type == GOT_OBJ_TYPE_TREE ||
 75 6691714a 2018-01-23 stsp 	    delta->type == GOT_OBJ_TYPE_BLOB ||
 76 6691714a 2018-01-23 stsp 	    delta->type == GOT_OBJ_TYPE_TAG) {
 77 6691714a 2018-01-23 stsp 		*type = delta->type;
 78 6691714a 2018-01-23 stsp 		return NULL;
 79 96f5e8b3 2018-01-23 stsp 	}
 80 96f5e8b3 2018-01-23 stsp 
 81 96f5e8b3 2018-01-23 stsp 	return got_error(GOT_ERR_BAD_DELTA_CHAIN);
 82 96f5e8b3 2018-01-23 stsp }
 83 96f5e8b3 2018-01-23 stsp 
 84 885d3e02 2018-01-27 stsp /* Fetch another (required) byte from the delta stream. */
 85 885d3e02 2018-01-27 stsp static const struct got_error *
 86 885d3e02 2018-01-27 stsp next_delta_byte(const uint8_t **p, size_t *remain)
 87 efd2a263 2018-01-19 stsp {
 88 885d3e02 2018-01-27 stsp 	if (--(*remain) == 0)
 89 885d3e02 2018-01-27 stsp 		return got_error(GOT_ERR_BAD_DELTA);
 90 885d3e02 2018-01-27 stsp 	(*p)++;
 91 885d3e02 2018-01-27 stsp 	return NULL;
 92 efd2a263 2018-01-19 stsp }
 93 885d3e02 2018-01-27 stsp 
 94 885d3e02 2018-01-27 stsp static const struct got_error *
 95 885d3e02 2018-01-27 stsp parse_size(uint64_t *size, const uint8_t **p, size_t *remain)
 96 885d3e02 2018-01-27 stsp {
 97 885d3e02 2018-01-27 stsp 	const struct got_error *err = NULL;
 98 885d3e02 2018-01-27 stsp 	int i = 0;
 99 885d3e02 2018-01-27 stsp 
100 885d3e02 2018-01-27 stsp 	*size = 0;
101 885d3e02 2018-01-27 stsp 	do {
102 885d3e02 2018-01-27 stsp 		/* We do not support size values which don't fit in 64 bit. */
103 885d3e02 2018-01-27 stsp 		if (i > 9)
104 885d3e02 2018-01-27 stsp 			return got_error(GOT_ERR_NO_SPACE);
105 885d3e02 2018-01-27 stsp 
106 885d3e02 2018-01-27 stsp 		if (i == 0)
107 885d3e02 2018-01-27 stsp 			*size = ((**p) & GOT_DELTA_SIZE_VAL_MASK);
108 885d3e02 2018-01-27 stsp 		else {
109 885d3e02 2018-01-27 stsp 			size_t shift = GOT_DELTA_SIZE_SHIFT * i;
110 885d3e02 2018-01-27 stsp 			*size |= (((**p) & GOT_DELTA_SIZE_VAL_MASK) << shift);
111 885d3e02 2018-01-27 stsp 		}
112 885d3e02 2018-01-27 stsp 
113 885d3e02 2018-01-27 stsp 		if (((**p) & GOT_DELTA_SIZE_MORE) == 0)
114 885d3e02 2018-01-27 stsp 			break;
115 885d3e02 2018-01-27 stsp 		i++;
116 885d3e02 2018-01-27 stsp 		err = next_delta_byte(p, remain);
117 885d3e02 2018-01-27 stsp 	} while (err == NULL);
118 885d3e02 2018-01-27 stsp 
119 885d3e02 2018-01-27 stsp 	return err;
120 885d3e02 2018-01-27 stsp }
121 885d3e02 2018-01-27 stsp 
122 885d3e02 2018-01-27 stsp static const struct got_error *
123 885d3e02 2018-01-27 stsp parse_opcode(off_t *offset, size_t *len, const uint8_t **p, size_t *remain)
124 885d3e02 2018-01-27 stsp {
125 885d3e02 2018-01-27 stsp 	const struct got_error *err = NULL;
126 885d3e02 2018-01-27 stsp 	off_t o = 0;
127 885d3e02 2018-01-27 stsp 	size_t l = 0;
128 885d3e02 2018-01-27 stsp 	uint8_t opcode = **p;
129 885d3e02 2018-01-27 stsp 
130 885d3e02 2018-01-27 stsp 	if (opcode & GOT_DELTA_COPY_OFF1) {
131 885d3e02 2018-01-27 stsp 		err = next_delta_byte(p, remain);
132 885d3e02 2018-01-27 stsp 		if (err)
133 885d3e02 2018-01-27 stsp 			return err;
134 885d3e02 2018-01-27 stsp 		o = (off_t)(**p);
135 885d3e02 2018-01-27 stsp 	}
136 885d3e02 2018-01-27 stsp 	if (opcode & GOT_DELTA_COPY_OFF2) {
137 885d3e02 2018-01-27 stsp 		err = next_delta_byte(p, remain);
138 885d3e02 2018-01-27 stsp 		if (err)
139 885d3e02 2018-01-27 stsp 			return err;
140 885d3e02 2018-01-27 stsp 		o |= ((off_t)(**p)) << 8;
141 885d3e02 2018-01-27 stsp 	}
142 885d3e02 2018-01-27 stsp 	if (opcode & GOT_DELTA_COPY_OFF3) {
143 885d3e02 2018-01-27 stsp 		err = next_delta_byte(p, remain);
144 885d3e02 2018-01-27 stsp 		if (err)
145 885d3e02 2018-01-27 stsp 			return err;
146 885d3e02 2018-01-27 stsp 		o |= ((off_t)(**p)) << 16;
147 885d3e02 2018-01-27 stsp 	}
148 885d3e02 2018-01-27 stsp 	if (opcode & GOT_DELTA_COPY_OFF4) {
149 885d3e02 2018-01-27 stsp 		err = next_delta_byte(p, remain);
150 885d3e02 2018-01-27 stsp 		if (err)
151 885d3e02 2018-01-27 stsp 			return err;
152 885d3e02 2018-01-27 stsp 		o |= ((off_t)(**p)) << 24;
153 885d3e02 2018-01-27 stsp 	}
154 885d3e02 2018-01-27 stsp 
155 885d3e02 2018-01-27 stsp 	if (opcode & GOT_DELTA_COPY_LEN1) {
156 885d3e02 2018-01-27 stsp 		err = next_delta_byte(p, remain);
157 885d3e02 2018-01-27 stsp 		if (err)
158 885d3e02 2018-01-27 stsp 			return err;
159 885d3e02 2018-01-27 stsp 		l = (off_t)(**p);
160 885d3e02 2018-01-27 stsp 	}
161 885d3e02 2018-01-27 stsp 	if (opcode & GOT_DELTA_COPY_LEN2) {
162 885d3e02 2018-01-27 stsp 		err = next_delta_byte(p, remain);
163 885d3e02 2018-01-27 stsp 		if (err)
164 885d3e02 2018-01-27 stsp 			return err;
165 885d3e02 2018-01-27 stsp 		l |= ((off_t)(**p)) << 8;
166 885d3e02 2018-01-27 stsp 	}
167 885d3e02 2018-01-27 stsp 	if (opcode & GOT_DELTA_COPY_LEN3) {
168 885d3e02 2018-01-27 stsp 		err = next_delta_byte(p, remain);
169 885d3e02 2018-01-27 stsp 		if (err)
170 885d3e02 2018-01-27 stsp 			return err;
171 885d3e02 2018-01-27 stsp 		l |= ((off_t)(**p)) << 16;
172 885d3e02 2018-01-27 stsp 	}
173 885d3e02 2018-01-27 stsp 
174 885d3e02 2018-01-27 stsp 	if (o == 0)
175 885d3e02 2018-01-27 stsp 		o = GOT_DELTA_COPY_DEFAULT_OFF;
176 885d3e02 2018-01-27 stsp 	if (l == 0)
177 885d3e02 2018-01-27 stsp 		l = GOT_DELTA_COPY_DEFAULT_LEN;
178 885d3e02 2018-01-27 stsp 
179 885d3e02 2018-01-27 stsp 	*offset = o;
180 885d3e02 2018-01-27 stsp 	*len = l;
181 885d3e02 2018-01-27 stsp 	return NULL;
182 885d3e02 2018-01-27 stsp }
183 885d3e02 2018-01-27 stsp 
184 885d3e02 2018-01-27 stsp static const struct got_error *
185 885d3e02 2018-01-27 stsp copy_from_base(FILE *base_file, off_t offset, size_t size, FILE *outfile)
186 885d3e02 2018-01-27 stsp {
187 885d3e02 2018-01-27 stsp 	if (fseeko(base_file, offset, SEEK_SET) != 0)
188 885d3e02 2018-01-27 stsp 		return got_error_from_errno();
189 885d3e02 2018-01-27 stsp 
190 885d3e02 2018-01-27 stsp 	while (size > 0) {
191 885d3e02 2018-01-27 stsp 		uint8_t data[2048];
192 885d3e02 2018-01-27 stsp 		size_t len = MIN(size, sizeof(data));
193 885d3e02 2018-01-27 stsp 		size_t n;
194 885d3e02 2018-01-27 stsp 
195 885d3e02 2018-01-27 stsp 		n = fread(data, len, 1, base_file);
196 885d3e02 2018-01-27 stsp 		if (n != 1)
197 885d3e02 2018-01-27 stsp 			return got_ferror(base_file, GOT_ERR_IO);
198 885d3e02 2018-01-27 stsp 
199 885d3e02 2018-01-27 stsp 		n = fwrite(data, len, 1, outfile);
200 885d3e02 2018-01-27 stsp 		if (n != 1)
201 885d3e02 2018-01-27 stsp 			return got_ferror(outfile, GOT_ERR_IO);
202 885d3e02 2018-01-27 stsp 
203 885d3e02 2018-01-27 stsp 		size -= len;
204 885d3e02 2018-01-27 stsp 	}
205 885d3e02 2018-01-27 stsp 
206 885d3e02 2018-01-27 stsp 	return NULL;
207 885d3e02 2018-01-27 stsp }
208 885d3e02 2018-01-27 stsp 
209 885d3e02 2018-01-27 stsp static const struct got_error *
210 885d3e02 2018-01-27 stsp copy_from_delta(const uint8_t **p, size_t *remain, size_t len, FILE *outfile)
211 885d3e02 2018-01-27 stsp {
212 885d3e02 2018-01-27 stsp 	size_t n;
213 885d3e02 2018-01-27 stsp 
214 885d3e02 2018-01-27 stsp 	if (*remain < len)
215 885d3e02 2018-01-27 stsp 		return got_error(GOT_ERR_BAD_DELTA);
216 885d3e02 2018-01-27 stsp 
217 885d3e02 2018-01-27 stsp 	n = fwrite(*p, len, 1, outfile);
218 885d3e02 2018-01-27 stsp 	if (n != 1)
219 885d3e02 2018-01-27 stsp 		return got_ferror(outfile, GOT_ERR_IO);
220 885d3e02 2018-01-27 stsp 
221 885d3e02 2018-01-27 stsp 	*p += len;
222 885d3e02 2018-01-27 stsp 	*remain -= len;
223 885d3e02 2018-01-27 stsp 	return NULL;
224 885d3e02 2018-01-27 stsp }
225 885d3e02 2018-01-27 stsp 
226 22484865 2018-03-13 stsp static const struct got_error *
227 22484865 2018-03-13 stsp parse_delta_sizes(uint64_t *base_size, uint64_t *result_size,
228 22484865 2018-03-13 stsp     const uint8_t **p, size_t *remain)
229 22484865 2018-03-13 stsp {
230 22484865 2018-03-13 stsp 	const struct got_error *err;
231 22484865 2018-03-13 stsp 
232 22484865 2018-03-13 stsp 	/* Read the two size fields at the beginning of the stream. */
233 22484865 2018-03-13 stsp 	err = parse_size(base_size, p, remain);
234 22484865 2018-03-13 stsp 	if (err)
235 22484865 2018-03-13 stsp 		return err;
236 22484865 2018-03-13 stsp 	err = next_delta_byte(p, remain);
237 22484865 2018-03-13 stsp 	if (err)
238 22484865 2018-03-13 stsp 		return err;
239 22484865 2018-03-13 stsp 	err = parse_size(result_size, p, remain);
240 22484865 2018-03-13 stsp 	if (err)
241 22484865 2018-03-13 stsp 		return err;
242 22484865 2018-03-13 stsp 
243 22484865 2018-03-13 stsp 	return NULL;
244 22484865 2018-03-13 stsp }
245 22484865 2018-03-13 stsp 
246 885d3e02 2018-01-27 stsp const struct got_error *
247 22484865 2018-03-13 stsp got_delta_get_sizes(uint64_t *base_size, uint64_t *result_size,
248 22484865 2018-03-13 stsp     const uint8_t *delta_buf, size_t delta_len)
249 22484865 2018-03-13 stsp {
250 22484865 2018-03-13 stsp 	size_t remain;
251 22484865 2018-03-13 stsp 	const uint8_t *p;
252 22484865 2018-03-13 stsp 
253 22484865 2018-03-13 stsp 	if (delta_len < GOT_DELTA_STREAM_LENGTH_MIN)
254 22484865 2018-03-13 stsp 		return got_error(GOT_ERR_BAD_DELTA);
255 22484865 2018-03-13 stsp 
256 22484865 2018-03-13 stsp 	p = delta_buf;
257 22484865 2018-03-13 stsp 	remain = delta_len;
258 22484865 2018-03-13 stsp 	return parse_delta_sizes(base_size, result_size, &p, &remain);
259 8628c62d 2018-03-15 stsp }
260 8628c62d 2018-03-15 stsp 
261 8628c62d 2018-03-15 stsp const struct got_error *
262 34fca9c3 2018-11-11 stsp got_delta_apply_in_mem(uint8_t *base_buf, size_t base_bufsz,
263 34fca9c3 2018-11-11 stsp     const uint8_t *delta_buf, size_t delta_len, uint8_t *outbuf,
264 34fca9c3 2018-11-11 stsp     size_t *outsize, size_t maxoutsize)
265 8628c62d 2018-03-15 stsp {
266 8628c62d 2018-03-15 stsp 	const struct got_error *err = NULL;
267 8628c62d 2018-03-15 stsp 	uint64_t base_size, result_size;
268 8628c62d 2018-03-15 stsp 	size_t remain;
269 8628c62d 2018-03-15 stsp 	const uint8_t *p;
270 8628c62d 2018-03-15 stsp 
271 8628c62d 2018-03-15 stsp 	*outsize= 0;
272 8628c62d 2018-03-15 stsp 
273 8628c62d 2018-03-15 stsp 	if (delta_len < GOT_DELTA_STREAM_LENGTH_MIN)
274 8628c62d 2018-03-15 stsp 		return got_error(GOT_ERR_BAD_DELTA);
275 8628c62d 2018-03-15 stsp 
276 8628c62d 2018-03-15 stsp 	p = delta_buf;
277 8628c62d 2018-03-15 stsp 	remain = delta_len;
278 8628c62d 2018-03-15 stsp 	err = parse_delta_sizes(&base_size, &result_size, &p, &remain);
279 8628c62d 2018-03-15 stsp 	if (err)
280 8628c62d 2018-03-15 stsp 		return err;
281 8628c62d 2018-03-15 stsp 
282 8628c62d 2018-03-15 stsp 	/* Decode and execute copy instructions from the delta stream. */
283 8628c62d 2018-03-15 stsp 	err = next_delta_byte(&p, &remain);
284 8628c62d 2018-03-15 stsp 	while (err == NULL && remain > 0) {
285 8628c62d 2018-03-15 stsp 		if (*p & GOT_DELTA_BASE_COPY) {
286 8628c62d 2018-03-15 stsp 			off_t offset = 0;
287 8628c62d 2018-03-15 stsp 			size_t len = 0;
288 8628c62d 2018-03-15 stsp 			err = parse_opcode(&offset, &len, &p, &remain);
289 8628c62d 2018-03-15 stsp 			if (err)
290 8628c62d 2018-03-15 stsp 				break;
291 b2f7af54 2018-11-11 stsp 			if (SIZE_MAX - offset < len || offset + len < 0 ||
292 b2f7af54 2018-11-11 stsp 			    base_bufsz < offset + len ||
293 34fca9c3 2018-11-11 stsp 			    *outsize + len > maxoutsize)
294 34fca9c3 2018-11-11 stsp 				return got_error(GOT_ERR_BAD_DELTA);
295 8628c62d 2018-03-15 stsp 			memcpy(outbuf + *outsize, base_buf + offset, len);
296 8628c62d 2018-03-15 stsp 			if (err == NULL) {
297 8628c62d 2018-03-15 stsp 				*outsize += len;
298 8628c62d 2018-03-15 stsp 				if (remain > 0) {
299 8628c62d 2018-03-15 stsp 					p++;
300 8628c62d 2018-03-15 stsp 					remain--;
301 8628c62d 2018-03-15 stsp 				}
302 8628c62d 2018-03-15 stsp 			}
303 8628c62d 2018-03-15 stsp 		} else {
304 8628c62d 2018-03-15 stsp 			size_t len = (size_t)*p;
305 8628c62d 2018-03-15 stsp 			if (len == 0) {
306 8628c62d 2018-03-15 stsp 				err = got_error(GOT_ERR_BAD_DELTA);
307 8628c62d 2018-03-15 stsp 				break;
308 8628c62d 2018-03-15 stsp 			}
309 8628c62d 2018-03-15 stsp 			err = next_delta_byte(&p, &remain);
310 8628c62d 2018-03-15 stsp 			if (err)
311 8628c62d 2018-03-15 stsp 				break;
312 b2f7af54 2018-11-11 stsp 			if (remain < len || SIZE_MAX - *outsize < len ||
313 b2f7af54 2018-11-11 stsp 			    *outsize + len > maxoutsize)
314 8628c62d 2018-03-15 stsp 				return got_error(GOT_ERR_BAD_DELTA);
315 8628c62d 2018-03-15 stsp 			memcpy(outbuf + *outsize, p, len);
316 8628c62d 2018-03-15 stsp 			p += len;
317 8628c62d 2018-03-15 stsp 			remain -= len;
318 8628c62d 2018-03-15 stsp 			*outsize += len;
319 8628c62d 2018-03-15 stsp 		}
320 8628c62d 2018-03-15 stsp 	}
321 8628c62d 2018-03-15 stsp 
322 8628c62d 2018-03-15 stsp 	if (*outsize != result_size)
323 8628c62d 2018-03-15 stsp 		err = got_error(GOT_ERR_BAD_DELTA);
324 8628c62d 2018-03-15 stsp 	return err;
325 22484865 2018-03-13 stsp }
326 22484865 2018-03-13 stsp 
327 22484865 2018-03-13 stsp const struct got_error *
328 0e22967e 2018-02-11 stsp got_delta_apply(FILE *base_file, const uint8_t *delta_buf,
329 b29656e2 2018-03-16 stsp     size_t delta_len, FILE *outfile, size_t *outsize)
330 885d3e02 2018-01-27 stsp {
331 885d3e02 2018-01-27 stsp 	const struct got_error *err = NULL;
332 885d3e02 2018-01-27 stsp 	uint64_t base_size, result_size;
333 b29656e2 2018-03-16 stsp 	size_t remain = 0;
334 885d3e02 2018-01-27 stsp 	const uint8_t *p;
335 39ff877f 2018-03-13 stsp 	FILE *memstream = NULL;
336 39ff877f 2018-03-13 stsp 	char *memstream_buf = NULL;
337 39ff877f 2018-03-13 stsp 	size_t memstream_size = 0;
338 b29656e2 2018-03-16 stsp 
339 b29656e2 2018-03-16 stsp 	*outsize = 0;
340 885d3e02 2018-01-27 stsp 
341 885d3e02 2018-01-27 stsp 	if (delta_len < GOT_DELTA_STREAM_LENGTH_MIN)
342 885d3e02 2018-01-27 stsp 		return got_error(GOT_ERR_BAD_DELTA);
343 885d3e02 2018-01-27 stsp 
344 885d3e02 2018-01-27 stsp 	p = delta_buf;
345 885d3e02 2018-01-27 stsp 	remain = delta_len;
346 22484865 2018-03-13 stsp 	err = parse_delta_sizes(&base_size, &result_size, &p, &remain);
347 885d3e02 2018-01-27 stsp 	if (err)
348 885d3e02 2018-01-27 stsp 		return err;
349 885d3e02 2018-01-27 stsp 
350 39ff877f 2018-03-13 stsp 	if (result_size < GOT_DELTA_RESULT_SIZE_CACHED_MAX)
351 39ff877f 2018-03-13 stsp 		memstream = open_memstream(&memstream_buf, &memstream_size);
352 39ff877f 2018-03-13 stsp 
353 885d3e02 2018-01-27 stsp 	/* Decode and execute copy instructions from the delta stream. */
354 885d3e02 2018-01-27 stsp 	err = next_delta_byte(&p, &remain);
355 06e5fc98 2018-02-11 stsp 	while (err == NULL && remain > 0) {
356 824801e7 2018-01-27 stsp 		if (*p & GOT_DELTA_BASE_COPY) {
357 885d3e02 2018-01-27 stsp 			off_t offset = 0;
358 885d3e02 2018-01-27 stsp 			size_t len = 0;
359 885d3e02 2018-01-27 stsp 			err = parse_opcode(&offset, &len, &p, &remain);
360 885d3e02 2018-01-27 stsp 			if (err)
361 885d3e02 2018-01-27 stsp 				break;
362 39ff877f 2018-03-13 stsp 			err = copy_from_base(base_file, offset, len,
363 39ff877f 2018-03-13 stsp 			    memstream ? memstream : outfile);
364 06e5fc98 2018-02-11 stsp 			if (err == NULL) {
365 b29656e2 2018-03-16 stsp 				*outsize += len;
366 06e5fc98 2018-02-11 stsp 				if (remain > 0) {
367 06e5fc98 2018-02-11 stsp 					p++;
368 06e5fc98 2018-02-11 stsp 					remain--;
369 06e5fc98 2018-02-11 stsp 				}
370 06e5fc98 2018-02-11 stsp 			}
371 885d3e02 2018-01-27 stsp 		} else {
372 885d3e02 2018-01-27 stsp 			size_t len = (size_t)*p;
373 885d3e02 2018-01-27 stsp 			if (len == 0) {
374 885d3e02 2018-01-27 stsp 				err = got_error(GOT_ERR_BAD_DELTA);
375 885d3e02 2018-01-27 stsp 				break;
376 885d3e02 2018-01-27 stsp 			}
377 885d3e02 2018-01-27 stsp 			err = next_delta_byte(&p, &remain);
378 885d3e02 2018-01-27 stsp 			if (err)
379 06e5fc98 2018-02-11 stsp 				break;
380 39ff877f 2018-03-13 stsp 			err = copy_from_delta(&p, &remain, len,
381 39ff877f 2018-03-13 stsp 			    memstream ? memstream : outfile);
382 885d3e02 2018-01-27 stsp 			if (err == NULL)
383 b29656e2 2018-03-16 stsp 				*outsize += len;
384 885d3e02 2018-01-27 stsp 		}
385 885d3e02 2018-01-27 stsp 	}
386 885d3e02 2018-01-27 stsp 
387 b29656e2 2018-03-16 stsp 	if (*outsize != result_size)
388 885d3e02 2018-01-27 stsp 		err = got_error(GOT_ERR_BAD_DELTA);
389 885d3e02 2018-01-27 stsp 
390 39ff877f 2018-03-13 stsp 	if (memstream != NULL) {
391 39ff877f 2018-03-13 stsp 		fclose(memstream);
392 39ff877f 2018-03-13 stsp 		if (err == NULL) {
393 39ff877f 2018-03-13 stsp 			size_t n;
394 39ff877f 2018-03-13 stsp 			n = fwrite(memstream_buf, 1, memstream_size, outfile);
395 39ff877f 2018-03-13 stsp 			if (n != memstream_size)
396 39ff877f 2018-03-13 stsp 				err = got_ferror(outfile, GOT_ERR_IO);
397 39ff877f 2018-03-13 stsp 		}
398 39ff877f 2018-03-13 stsp 		free(memstream_buf);
399 39ff877f 2018-03-13 stsp 	}
400 885d3e02 2018-01-27 stsp 	if (err == NULL)
401 885d3e02 2018-01-27 stsp 		rewind(outfile);
402 885d3e02 2018-01-27 stsp 	return err;
403 885d3e02 2018-01-27 stsp }