2 56e7a15f 2021-03-10 stsp * Copyright (c) 2020 Ori Bernstein
3 69aa0e90 2021-03-10 stsp * Copyright (c) 2021 Stefan Sperling <stsp@openbsd.org>
5 2a0e67f7 2021-03-10 stsp * Permission to use, copy, modify, and distribute this software for any
6 2a0e67f7 2021-03-10 stsp * purpose with or without fee is hereby granted, provided that the above
7 2a0e67f7 2021-03-10 stsp * copyright notice and this permission notice appear in all copies.
9 2a0e67f7 2021-03-10 stsp * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 2a0e67f7 2021-03-10 stsp * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 2a0e67f7 2021-03-10 stsp * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 2a0e67f7 2021-03-10 stsp * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 2a0e67f7 2021-03-10 stsp * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 2a0e67f7 2021-03-10 stsp * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 2a0e67f7 2021-03-10 stsp * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 f364801d 2021-03-10 stsp #include <sys/types.h>
20 f364801d 2021-03-10 stsp #include <assert.h>
21 69aa0e90 2021-03-10 stsp #include <errno.h>
22 f364801d 2021-03-10 stsp #include <limits.h>
23 f364801d 2021-03-10 stsp #include <stdint.h>
24 f364801d 2021-03-10 stsp #include <stdio.h>
25 f364801d 2021-03-10 stsp #include <stdlib.h>
26 f364801d 2021-03-10 stsp #include <string.h>
28 f364801d 2021-03-10 stsp #include "got_error.h"
30 69aa0e90 2021-03-10 stsp #include "got_lib_deltify.h"
33 69aa0e90 2021-03-10 stsp #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
37 2539ccf2 2021-10-08 thomas * The algorihm used here is FastCDC (Fast Content-Defined Chunking)
38 2539ccf2 2021-10-08 thomas * https://www.usenix.org/conference/atc16/technical-sessions/presentation/xia
41 82e16997 2021-10-08 thomas static uint32_t geartab[256] = {
42 69aa0e90 2021-03-10 stsp 0x67ed26b7, 0x32da500c, 0x53d0fee0, 0xce387dc7, 0xcd406d90, 0x2e83a4d4,
43 69aa0e90 2021-03-10 stsp 0x9fc9a38d, 0xb67259dc, 0xca6b1722, 0x6d2ea08c, 0x235cea2e, 0x3149bb5f,
44 69aa0e90 2021-03-10 stsp 0x1beda787, 0x2a6b77d5, 0x2f22d9ac, 0x91fc0544, 0xe413acfa, 0x5a30ff7a,
45 69aa0e90 2021-03-10 stsp 0xad6fdde0, 0x444fd0f5, 0x7ad87864, 0x58c5ff05, 0x8d2ec336, 0x2371f853,
46 69aa0e90 2021-03-10 stsp 0x550f8572, 0x6aa448dd, 0x7c9ddbcf, 0x95221e14, 0x2a82ec33, 0xcbec5a78,
47 69aa0e90 2021-03-10 stsp 0xc6795a0d, 0x243995b7, 0x1c909a2f, 0x4fded51c, 0x635d334b, 0x0e2b9999,
48 69aa0e90 2021-03-10 stsp 0x2702968d, 0x856de1d5, 0x3325d60e, 0xeb6a7502, 0xec2a9844, 0x0905835a,
49 69aa0e90 2021-03-10 stsp 0xa1820375, 0xa4be5cab, 0x96a6c058, 0x2c2ccd70, 0xba40fce3, 0xd794c46b,
50 69aa0e90 2021-03-10 stsp 0x8fbae83e, 0xc3aa7899, 0x3d3ff8ed, 0xa0d42b5b, 0x571c0c97, 0xd2811516,
51 69aa0e90 2021-03-10 stsp 0xf7e7b96c, 0x4fd2fcbd, 0xe2fdec94, 0x282cc436, 0x78e8e95c, 0x80a3b613,
52 69aa0e90 2021-03-10 stsp 0xcfbee20c, 0xd4a32d1c, 0x2a12ff13, 0x6af82936, 0xe5630258, 0x8efa6a98,
53 69aa0e90 2021-03-10 stsp 0x294fb2d1, 0xdeb57086, 0x5f0fddb3, 0xeceda7ce, 0x4c87305f, 0x3a6d3307,
54 69aa0e90 2021-03-10 stsp 0xe22d2942, 0x9d060217, 0x1e42ed02, 0xb6f63b52, 0x4367f39f, 0x055cf262,
55 69aa0e90 2021-03-10 stsp 0x03a461b2, 0x5ef9e382, 0x386bc03a, 0x2a1e79c7, 0xf1a0058b, 0xd4d2dea9,
56 69aa0e90 2021-03-10 stsp 0x56baf37d, 0x5daff6cc, 0xf03a951d, 0xaef7de45, 0xa8f4581e, 0x3960b555,
57 69aa0e90 2021-03-10 stsp 0xffbfff6d, 0xbe702a23, 0x8f5b6d6f, 0x061739fb, 0x98696f47, 0x3fd596d4,
58 69aa0e90 2021-03-10 stsp 0x151eac6b, 0xa9fcc4f5, 0x69181a12, 0x3ac5a107, 0xb5198fe7, 0x96bcb1da,
59 69aa0e90 2021-03-10 stsp 0x1b5ddf8e, 0xc757d650, 0x65865c3a, 0x8fc0a41a, 0x87435536, 0x99eda6f2,
60 69aa0e90 2021-03-10 stsp 0x41874794, 0x29cff4e8, 0xb70efd9a, 0x3103f6e7, 0x84d2453b, 0x15a450bd,
61 69aa0e90 2021-03-10 stsp 0x74f49af1, 0x60f664b1, 0xa1c86935, 0xfdafbce1, 0xe36353e3, 0x5d9ba739,
62 69aa0e90 2021-03-10 stsp 0xbc0559ba, 0x708b0054, 0xd41d808c, 0xb2f31723, 0x9027c41f, 0xf136d165,
63 69aa0e90 2021-03-10 stsp 0xb5374b12, 0x9420a6ac, 0x273958b6, 0xe6c2fad0, 0xebdc1f21, 0xfb33af8b,
64 69aa0e90 2021-03-10 stsp 0xc71c25cd, 0xe9a2d8e5, 0xbeb38a50, 0xbceb7cc2, 0x4e4e73f0, 0xcd6c251d,
65 69aa0e90 2021-03-10 stsp 0xde4c032c, 0x4b04ac30, 0x725b8b21, 0x4eb8c33b, 0x20d07b75, 0x0567aa63,
66 69aa0e90 2021-03-10 stsp 0xb56b2bb7, 0xc1f5fd3a, 0xcafd35ca, 0x470dd4da, 0xfe4f94cd, 0xfb8de424,
67 69aa0e90 2021-03-10 stsp 0xe8dbcf40, 0xfe50a37a, 0x62db5b5d, 0xf32f4ab6, 0x2c4a8a51, 0x18473dc0,
68 69aa0e90 2021-03-10 stsp 0xfe0cbb6e, 0xfe399efd, 0xdf34ecc9, 0x6ccd5055, 0x46097073, 0x139135c2,
69 69aa0e90 2021-03-10 stsp 0x721c76f6, 0x1c6a94b4, 0x6eee014d, 0x8a508e02, 0x3da538f5, 0x280d394f,
70 69aa0e90 2021-03-10 stsp 0x5248a0c4, 0x3ce94c6c, 0x9a71ad3a, 0x8493dd05, 0xe43f0ab6, 0x18e4ed42,
71 69aa0e90 2021-03-10 stsp 0x6c5c0e09, 0x42b06ec9, 0x8d330343, 0xa45b6f59, 0x2a573c0c, 0xd7fd3de6,
72 69aa0e90 2021-03-10 stsp 0xeedeab68, 0x5c84dafc, 0xbbd1b1a8, 0xa3ce1ad1, 0x85b70bed, 0xb6add07f,
73 69aa0e90 2021-03-10 stsp 0xa531309c, 0x8f8ab852, 0x564de332, 0xeac9ed0c, 0x73da402c, 0x3ec52761,
74 69aa0e90 2021-03-10 stsp 0x43af2f4d, 0xd6ff45c8, 0x4c367462, 0xd553bd6a, 0x44724855, 0x3b2aa728,
75 69aa0e90 2021-03-10 stsp 0x56e5eb65, 0xeaf16173, 0x33fa42ff, 0xd714bb5d, 0xfbd0a3b9, 0xaf517134,
76 69aa0e90 2021-03-10 stsp 0x9416c8cd, 0x534cf94f, 0x548947c2, 0x34193569, 0x32f4389a, 0xfe7028bc,
77 69aa0e90 2021-03-10 stsp 0xed73b1ed, 0x9db95770, 0x468e3922, 0x0440c3cd, 0x60059a62, 0x33504562,
78 69aa0e90 2021-03-10 stsp 0x2b229fbd, 0x5174dca5, 0xf7028752, 0xd63c6aa8, 0x31276f38, 0x0646721c,
79 69aa0e90 2021-03-10 stsp 0xb0191da8, 0xe00e6de0, 0x9eac1a6e, 0x9f7628a5, 0xed6c06ea, 0x0bb8af15,
80 69aa0e90 2021-03-10 stsp 0xf119fb12, 0x38693c1c, 0x732bc0fe, 0x84953275, 0xb82ec888, 0x33a4f1b3,
81 69aa0e90 2021-03-10 stsp 0x3099835e, 0x028a8782, 0x5fdd51d7, 0xc6c717b3, 0xb06caf71, 0x17c8c111,
82 69aa0e90 2021-03-10 stsp 0x61bad754, 0x9fd03061, 0xe09df1af, 0x3bc9eb73, 0x85878413, 0x9889aaf2,
83 69aa0e90 2021-03-10 stsp 0x3f5a9e46, 0x42c9f01f, 0x9984a4f4, 0xd5de43cc, 0xd294daed, 0xbecba2d2,
84 69aa0e90 2021-03-10 stsp 0xf1f6e72c, 0x5551128a, 0x83af87e2, 0x6f0342ba,
87 f364801d 2021-03-10 stsp static uint64_t
88 69aa0e90 2021-03-10 stsp hashblk(const unsigned char *p, off_t n)
90 f364801d 2021-03-10 stsp unsigned char buf[SHA1_DIGEST_LENGTH];
92 69aa0e90 2021-03-10 stsp SHA1_CTX ctx;
93 69aa0e90 2021-03-10 stsp SHA1Init(&ctx);
94 69aa0e90 2021-03-10 stsp SHA1Update(&ctx, p, n);
95 69aa0e90 2021-03-10 stsp SHA1Final(buf, &ctx);
96 f364801d 2021-03-10 stsp memcpy(&h, buf, sizeof(h));
97 f364801d 2021-03-10 stsp return be64toh(h);
100 f364801d 2021-03-10 stsp static const struct got_error *
101 740bba1c 2021-06-18 stsp addblk(struct got_delta_table *dt, FILE *f, off_t file_offset0, off_t len,
102 740bba1c 2021-06-18 stsp off_t offset, uint64_t h)
104 69aa0e90 2021-03-10 stsp const struct got_error *err = NULL;
106 69aa0e90 2021-03-10 stsp uint8_t buf[GOT_DELTIFY_MAXCHUNK];
107 0d15f6dc 2021-06-13 stsp uint8_t buf2[GOT_DELTIFY_MAXCHUNK];
108 69aa0e90 2021-03-10 stsp size_t r = 0;
110 69aa0e90 2021-03-10 stsp if (len == 0)
111 69aa0e90 2021-03-10 stsp return NULL;
113 69aa0e90 2021-03-10 stsp i = h % dt->nalloc;
114 69aa0e90 2021-03-10 stsp while (dt->blocks[i].len != 0) {
116 69aa0e90 2021-03-10 stsp * Avoid adding duplicate blocks.
117 69aa0e90 2021-03-10 stsp * NB: A matching hash is insufficient for detecting equality.
118 51a494da 2021-06-11 stsp * The hash can only detect inequality.
120 51a494da 2021-06-11 stsp if (len == dt->blocks[i].len && h == dt->blocks[i].hash) {
121 0d15f6dc 2021-06-13 stsp if (r == 0) {
122 740bba1c 2021-06-18 stsp if (fseeko(f, file_offset0 + offset, SEEK_SET) == -1)
123 0d15f6dc 2021-06-13 stsp return got_error_from_errno("fseeko");
124 0d15f6dc 2021-06-13 stsp r = fread(buf, 1, len, f);
125 0d15f6dc 2021-06-13 stsp if (r != len) {
126 0d15f6dc 2021-06-13 stsp if (!ferror(f))
127 0d15f6dc 2021-06-13 stsp return NULL;
128 0d15f6dc 2021-06-13 stsp return got_ferror(f, GOT_ERR_IO);
131 740bba1c 2021-06-18 stsp if (fseeko(f, file_offset0 + dt->blocks[i].offset,
132 740bba1c 2021-06-18 stsp SEEK_SET) == -1)
133 69aa0e90 2021-03-10 stsp return got_error_from_errno("fseeko");
134 69aa0e90 2021-03-10 stsp if (fread(buf2, 1, len, f) != len)
135 69aa0e90 2021-03-10 stsp return got_ferror(f, GOT_ERR_IO);
136 69aa0e90 2021-03-10 stsp if (memcmp(buf, buf2, len) == 0)
137 69aa0e90 2021-03-10 stsp return NULL;
140 69aa0e90 2021-03-10 stsp i = (i + 1) % dt->nalloc;
142 69aa0e90 2021-03-10 stsp assert(dt->blocks[i].len == 0);
143 69aa0e90 2021-03-10 stsp dt->blocks[i].len = len;
144 69aa0e90 2021-03-10 stsp dt->blocks[i].offset = offset;
145 69aa0e90 2021-03-10 stsp dt->blocks[i].hash = h;
146 69aa0e90 2021-03-10 stsp dt->nblocks++;
147 e89540a9 2021-06-11 stsp if (dt->nalloc < dt->nblocks + 64) {
148 69aa0e90 2021-03-10 stsp struct got_delta_block *db;
149 e89540a9 2021-06-11 stsp size_t old_size = dt->nalloc;
150 69aa0e90 2021-03-10 stsp db = dt->blocks;
151 e89540a9 2021-06-11 stsp dt->blocks = calloc(dt->nalloc + 64,
152 e89540a9 2021-06-11 stsp sizeof(struct got_delta_block));
153 69aa0e90 2021-03-10 stsp if (dt->blocks == NULL) {
154 69aa0e90 2021-03-10 stsp err = got_error_from_errno("calloc");
155 69aa0e90 2021-03-10 stsp dt->blocks = db;
156 69aa0e90 2021-03-10 stsp return err;
158 e89540a9 2021-06-11 stsp dt->nalloc += 64;
160 69aa0e90 2021-03-10 stsp * Recompute all block positions. Hash-based indices of blocks
161 69aa0e90 2021-03-10 stsp * in the array depend on the allocated length of the array.
163 69aa0e90 2021-03-10 stsp dt->nblocks = 0;
164 a893025f 2021-06-11 stsp for (i = 0; i < old_size; i++) {
165 69aa0e90 2021-03-10 stsp if (db[i].len == 0)
167 740bba1c 2021-06-18 stsp err = addblk(dt, f, file_offset0, db[i].len,
168 2b0ae357 2022-01-10 thomas db[i].offset, db[i].hash);
172 2b0ae357 2022-01-10 thomas free(db);
175 2b0ae357 2022-01-10 thomas return err;
178 2b0ae357 2022-01-10 thomas static const struct got_error *
179 2b0ae357 2022-01-10 thomas addblk_mem(struct got_delta_table *dt, uint8_t *data, off_t file_offset0,
180 2b0ae357 2022-01-10 thomas off_t len, off_t offset, uint64_t h)
182 2b0ae357 2022-01-10 thomas const struct got_error *err = NULL;
184 2b0ae357 2022-01-10 thomas uint8_t *block1;
185 2b0ae357 2022-01-10 thomas uint8_t *block2;
187 2b0ae357 2022-01-10 thomas if (len == 0)
188 2b0ae357 2022-01-10 thomas return NULL;
190 2b0ae357 2022-01-10 thomas i = h % dt->nalloc;
191 2b0ae357 2022-01-10 thomas while (dt->blocks[i].len != 0) {
193 2b0ae357 2022-01-10 thomas * Avoid adding duplicate blocks.
194 2b0ae357 2022-01-10 thomas * NB: A matching hash is insufficient for detecting equality.
195 2b0ae357 2022-01-10 thomas * The hash can only detect inequality.
197 2b0ae357 2022-01-10 thomas if (len == dt->blocks[i].len && h == dt->blocks[i].hash) {
198 2b0ae357 2022-01-10 thomas block1 = data + file_offset0 + dt->blocks[i].offset;
199 2b0ae357 2022-01-10 thomas block2 = data + file_offset0 + offset;
200 2b0ae357 2022-01-10 thomas if (memcmp(block1, block2, len) == 0)
201 2b0ae357 2022-01-10 thomas return NULL;
204 2b0ae357 2022-01-10 thomas i = (i + 1) % dt->nalloc;
206 2b0ae357 2022-01-10 thomas assert(dt->blocks[i].len == 0);
207 2b0ae357 2022-01-10 thomas dt->blocks[i].len = len;
208 2b0ae357 2022-01-10 thomas dt->blocks[i].offset = offset;
209 2b0ae357 2022-01-10 thomas dt->blocks[i].hash = h;
210 2b0ae357 2022-01-10 thomas dt->nblocks++;
211 2b0ae357 2022-01-10 thomas if (dt->nalloc < dt->nblocks + 64) {
212 2b0ae357 2022-01-10 thomas struct got_delta_block *db;
213 2b0ae357 2022-01-10 thomas size_t old_size = dt->nalloc;
214 2b0ae357 2022-01-10 thomas db = dt->blocks;
215 2b0ae357 2022-01-10 thomas dt->blocks = calloc(dt->nalloc + 64,
216 2b0ae357 2022-01-10 thomas sizeof(struct got_delta_block));
217 2b0ae357 2022-01-10 thomas if (dt->blocks == NULL) {
218 2b0ae357 2022-01-10 thomas err = got_error_from_errno("calloc");
219 2b0ae357 2022-01-10 thomas dt->blocks = db;
220 2b0ae357 2022-01-10 thomas return err;
222 2b0ae357 2022-01-10 thomas dt->nalloc += 64;
224 2b0ae357 2022-01-10 thomas * Recompute all block positions. Hash-based indices of blocks
225 2b0ae357 2022-01-10 thomas * in the array depend on the allocated length of the array.
227 2b0ae357 2022-01-10 thomas dt->nblocks = 0;
228 2b0ae357 2022-01-10 thomas for (i = 0; i < old_size; i++) {
229 2b0ae357 2022-01-10 thomas if (db[i].len == 0)
230 2b0ae357 2022-01-10 thomas continue;
231 2b0ae357 2022-01-10 thomas err = addblk_mem(dt, data, file_offset0, db[i].len,
232 740bba1c 2021-06-18 stsp db[i].offset, db[i].hash);
239 69aa0e90 2021-03-10 stsp return err;
242 69aa0e90 2021-03-10 stsp static const struct got_error *
243 69aa0e90 2021-03-10 stsp lookupblk(struct got_delta_block **block, struct got_delta_table *dt,
244 740bba1c 2021-06-18 stsp unsigned char *p, off_t len, FILE *basefile, off_t basefile_offset0)
247 f364801d 2021-03-10 stsp uint64_t h;
248 69aa0e90 2021-03-10 stsp uint8_t buf[GOT_DELTIFY_MAXCHUNK];
251 69aa0e90 2021-03-10 stsp *block = NULL;
253 69aa0e90 2021-03-10 stsp h = hashblk(p, len);
254 69aa0e90 2021-03-10 stsp for (i = h % dt->nalloc; dt->blocks[i].len != 0;
255 69aa0e90 2021-03-10 stsp i = (i + 1) % dt->nalloc) {
256 69aa0e90 2021-03-10 stsp if (dt->blocks[i].hash != h ||
257 69aa0e90 2021-03-10 stsp dt->blocks[i].len != len)
259 740bba1c 2021-06-18 stsp if (fseeko(basefile, basefile_offset0 + dt->blocks[i].offset,
260 740bba1c 2021-06-18 stsp SEEK_SET) == -1)
261 69aa0e90 2021-03-10 stsp return got_error_from_errno("fseeko");
262 69aa0e90 2021-03-10 stsp r = fread(buf, 1, len, basefile);
263 69aa0e90 2021-03-10 stsp if (r != len)
264 69aa0e90 2021-03-10 stsp return got_ferror(basefile, GOT_ERR_IO);
265 69aa0e90 2021-03-10 stsp if (memcmp(p, buf, len) == 0) {
266 69aa0e90 2021-03-10 stsp *block = &dt->blocks[i];
270 f364801d 2021-03-10 stsp return NULL;
273 69aa0e90 2021-03-10 stsp static const struct got_error *
274 2b0ae357 2022-01-10 thomas lookupblk_mem(struct got_delta_block **block, struct got_delta_table *dt,
275 2b0ae357 2022-01-10 thomas unsigned char *p, off_t len, uint8_t *basedata, off_t basefile_offset0)
278 2b0ae357 2022-01-10 thomas uint64_t h;
279 2b0ae357 2022-01-10 thomas uint8_t *b;
281 2b0ae357 2022-01-10 thomas *block = NULL;
283 2b0ae357 2022-01-10 thomas h = hashblk(p, len);
284 2b0ae357 2022-01-10 thomas for (i = h % dt->nalloc; dt->blocks[i].len != 0;
285 2b0ae357 2022-01-10 thomas i = (i + 1) % dt->nalloc) {
286 2b0ae357 2022-01-10 thomas if (dt->blocks[i].hash != h ||
287 2b0ae357 2022-01-10 thomas dt->blocks[i].len != len)
288 2b0ae357 2022-01-10 thomas continue;
289 2b0ae357 2022-01-10 thomas b = basedata + basefile_offset0 + dt->blocks[i].offset;
290 2b0ae357 2022-01-10 thomas if (memcmp(p, b, len) == 0) {
291 2b0ae357 2022-01-10 thomas *block = &dt->blocks[i];
295 2b0ae357 2022-01-10 thomas return NULL;
298 2b0ae357 2022-01-10 thomas static const struct got_error *
299 69aa0e90 2021-03-10 stsp nextblk(uint8_t *buf, off_t *blocklen, FILE *f)
301 f364801d 2021-03-10 stsp uint32_t gh;
302 69aa0e90 2021-03-10 stsp const unsigned char *p;
304 69aa0e90 2021-03-10 stsp off_t pos = ftello(f);
306 69aa0e90 2021-03-10 stsp *blocklen = 0;
308 69aa0e90 2021-03-10 stsp r = fread(buf, 1, GOT_DELTIFY_MAXCHUNK, f);
309 69aa0e90 2021-03-10 stsp if (r == 0 && ferror(f))
310 69aa0e90 2021-03-10 stsp return got_ferror(f, GOT_ERR_IO);
311 69aa0e90 2021-03-10 stsp if (r < GOT_DELTIFY_MINCHUNK)
312 69aa0e90 2021-03-10 stsp return NULL; /* no more delta-worthy blocks left */
314 69aa0e90 2021-03-10 stsp /* Got a deltifiable block. Find the split-point where it ends. */
315 69aa0e90 2021-03-10 stsp p = buf + GOT_DELTIFY_MINCHUNK;
317 69aa0e90 2021-03-10 stsp while (p != buf + r) {
318 f364801d 2021-03-10 stsp gh = (gh << 1) + geartab[*p++];
319 69aa0e90 2021-03-10 stsp if ((gh & GOT_DELTIFY_SPLITMASK) == 0)
323 69aa0e90 2021-03-10 stsp *blocklen = (p - buf);
324 69aa0e90 2021-03-10 stsp if (fseeko(f, pos + *blocklen, SEEK_SET) == -1)
325 69aa0e90 2021-03-10 stsp return got_error_from_errno("fseeko");
327 69aa0e90 2021-03-10 stsp return NULL;
330 2b0ae357 2022-01-10 thomas static const struct got_error *
331 2b0ae357 2022-01-10 thomas nextblk_mem(off_t *blocklen, uint8_t *data, off_t fileoffset, off_t filesize)
333 2b0ae357 2022-01-10 thomas uint32_t gh;
334 2b0ae357 2022-01-10 thomas const unsigned char *p;
336 2b0ae357 2022-01-10 thomas *blocklen = 0;
338 2b0ae357 2022-01-10 thomas if (fileoffset >= filesize ||
339 2b0ae357 2022-01-10 thomas filesize - fileoffset < GOT_DELTIFY_MINCHUNK)
340 2b0ae357 2022-01-10 thomas return NULL; /* no more delta-worthy blocks left */
342 2b0ae357 2022-01-10 thomas /* Got a deltifiable block. Find the split-point where it ends. */
343 2b0ae357 2022-01-10 thomas p = data + fileoffset + GOT_DELTIFY_MINCHUNK;
345 2b0ae357 2022-01-10 thomas while (p != data + MIN(fileoffset + GOT_DELTIFY_MAXCHUNK, filesize)) {
346 2b0ae357 2022-01-10 thomas gh = (gh << 1) + geartab[*p++];
347 2b0ae357 2022-01-10 thomas if ((gh & GOT_DELTIFY_SPLITMASK) == 0)
351 2b0ae357 2022-01-10 thomas *blocklen = (p - (data + fileoffset));
352 2b0ae357 2022-01-10 thomas return NULL;
355 f364801d 2021-03-10 stsp const struct got_error *
356 69aa0e90 2021-03-10 stsp got_deltify_init(struct got_delta_table **dt, FILE *f, off_t fileoffset,
357 69aa0e90 2021-03-10 stsp off_t filesize)
359 69aa0e90 2021-03-10 stsp const struct got_error *err = NULL;
360 f364801d 2021-03-10 stsp uint64_t h;
361 740bba1c 2021-06-18 stsp const off_t offset0 = fileoffset;
363 69aa0e90 2021-03-10 stsp *dt = calloc(1, sizeof(**dt));
364 69aa0e90 2021-03-10 stsp if (*dt == NULL)
365 f364801d 2021-03-10 stsp return got_error_from_errno("calloc");
367 69aa0e90 2021-03-10 stsp (*dt)->nblocks = 0;
368 69aa0e90 2021-03-10 stsp (*dt)->nalloc = 128;
369 69aa0e90 2021-03-10 stsp (*dt)->blocks = calloc((*dt)->nalloc, sizeof(struct got_delta_block));
370 69aa0e90 2021-03-10 stsp if ((*dt)->blocks == NULL) {
371 69aa0e90 2021-03-10 stsp err = got_error_from_errno("calloc");
375 69aa0e90 2021-03-10 stsp if (fseeko(f, fileoffset, SEEK_SET) == -1)
376 69aa0e90 2021-03-10 stsp return got_error_from_errno("fseeko");
378 69aa0e90 2021-03-10 stsp while (fileoffset < filesize) {
379 69aa0e90 2021-03-10 stsp uint8_t buf[GOT_DELTIFY_MAXCHUNK];
380 69aa0e90 2021-03-10 stsp off_t blocklen;
381 69aa0e90 2021-03-10 stsp err = nextblk(buf, &blocklen, f);
384 69aa0e90 2021-03-10 stsp if (blocklen == 0)
386 69aa0e90 2021-03-10 stsp h = hashblk(buf, blocklen);
387 740bba1c 2021-06-18 stsp err = addblk(*dt, f, offset0, blocklen,
388 740bba1c 2021-06-18 stsp fileoffset - offset0, h);
391 69aa0e90 2021-03-10 stsp fileoffset += blocklen;
392 68bdcdc2 2021-06-11 stsp if (fseeko(f, fileoffset, SEEK_SET) == -1)
393 68bdcdc2 2021-06-11 stsp return got_error_from_errno("fseeko");
396 2b0ae357 2022-01-10 thomas if (err) {
397 2b0ae357 2022-01-10 thomas free((*dt)->blocks);
398 2b0ae357 2022-01-10 thomas free(*dt);
399 2b0ae357 2022-01-10 thomas *dt = NULL;
402 2b0ae357 2022-01-10 thomas return err;
405 2b0ae357 2022-01-10 thomas const struct got_error *
406 2b0ae357 2022-01-10 thomas got_deltify_init_mem(struct got_delta_table **dt, uint8_t *data,
407 2b0ae357 2022-01-10 thomas off_t fileoffset, off_t filesize)
409 2b0ae357 2022-01-10 thomas const struct got_error *err = NULL;
410 2b0ae357 2022-01-10 thomas uint64_t h;
411 2b0ae357 2022-01-10 thomas const off_t offset0 = fileoffset;
413 2b0ae357 2022-01-10 thomas *dt = calloc(1, sizeof(**dt));
414 2b0ae357 2022-01-10 thomas if (*dt == NULL)
415 2b0ae357 2022-01-10 thomas return got_error_from_errno("calloc");
417 2b0ae357 2022-01-10 thomas (*dt)->nblocks = 0;
418 2b0ae357 2022-01-10 thomas (*dt)->nalloc = 128;
419 2b0ae357 2022-01-10 thomas (*dt)->blocks = calloc((*dt)->nalloc, sizeof(struct got_delta_block));
420 2b0ae357 2022-01-10 thomas if ((*dt)->blocks == NULL) {
421 2b0ae357 2022-01-10 thomas err = got_error_from_errno("calloc");
422 2b0ae357 2022-01-10 thomas goto done;
425 2b0ae357 2022-01-10 thomas while (fileoffset < filesize) {
426 2b0ae357 2022-01-10 thomas off_t blocklen;
427 2b0ae357 2022-01-10 thomas err = nextblk_mem(&blocklen, data, fileoffset, filesize);
429 2b0ae357 2022-01-10 thomas goto done;
430 2b0ae357 2022-01-10 thomas if (blocklen == 0)
432 2b0ae357 2022-01-10 thomas h = hashblk(data + fileoffset, blocklen);
433 2b0ae357 2022-01-10 thomas err = addblk_mem(*dt, data, offset0, blocklen,
434 2b0ae357 2022-01-10 thomas fileoffset - offset0, h);
436 2b0ae357 2022-01-10 thomas goto done;
437 2b0ae357 2022-01-10 thomas fileoffset += blocklen;
441 69aa0e90 2021-03-10 stsp free((*dt)->blocks);
443 69aa0e90 2021-03-10 stsp *dt = NULL;
446 69aa0e90 2021-03-10 stsp return err;
450 69aa0e90 2021-03-10 stsp got_deltify_free(struct got_delta_table *dt)
452 dbbf4a5f 2021-05-20 stsp if (dt == NULL)
454 69aa0e90 2021-03-10 stsp free(dt->blocks);
458 f364801d 2021-03-10 stsp static const struct got_error *
459 dd29967c 2021-08-22 stsp emitdelta(struct got_delta_instruction **deltas, size_t *nalloc, int *ndeltas,
460 dd29967c 2021-08-22 stsp const size_t alloc_chunk_size, int copy, off_t offset, off_t len)
462 69aa0e90 2021-03-10 stsp struct got_delta_instruction *d, *p;
464 dd29967c 2021-08-22 stsp if (*nalloc < *ndeltas + alloc_chunk_size) {
465 dd29967c 2021-08-22 stsp p = reallocarray(*deltas, *nalloc + alloc_chunk_size,
466 dd29967c 2021-08-22 stsp sizeof(struct got_delta_instruction));
467 dd29967c 2021-08-22 stsp if (p == NULL)
468 7d76c4e7 2021-10-15 thomas return got_error_from_errno("reallocarray");
469 dd29967c 2021-08-22 stsp *deltas = p;
470 dd29967c 2021-08-22 stsp *nalloc += alloc_chunk_size;
472 69aa0e90 2021-03-10 stsp *ndeltas += 1;
473 69aa0e90 2021-03-10 stsp d = &(*deltas)[*ndeltas - 1];
474 69aa0e90 2021-03-10 stsp d->copy = copy;
475 69aa0e90 2021-03-10 stsp d->offset = offset;
476 56e7a15f 2021-03-10 stsp d->len = len;
477 f364801d 2021-03-10 stsp return NULL;
480 69aa0e90 2021-03-10 stsp static const struct got_error *
481 740bba1c 2021-06-18 stsp stretchblk(FILE *basefile, off_t base_offset0, struct got_delta_block *block,
482 740bba1c 2021-06-18 stsp FILE *f, off_t filesize, off_t *blocklen)
484 69aa0e90 2021-03-10 stsp uint8_t basebuf[GOT_DELTIFY_MAXCHUNK], buf[GOT_DELTIFY_MAXCHUNK];
485 69aa0e90 2021-03-10 stsp size_t base_r, r, i;
486 69aa0e90 2021-03-10 stsp int buf_equal = 1;
488 5de743f8 2021-08-29 stsp if (fseeko(basefile, base_offset0 + block->offset + *blocklen,
489 5de743f8 2021-08-29 stsp SEEK_SET) == -1)
490 69aa0e90 2021-03-10 stsp return got_error_from_errno("fseeko");
492 69aa0e90 2021-03-10 stsp while (buf_equal && *blocklen < (1 << 24) - 1) {
493 69aa0e90 2021-03-10 stsp base_r = fread(basebuf, 1, sizeof(basebuf), basefile);
494 69aa0e90 2021-03-10 stsp if (base_r == 0) {
495 69aa0e90 2021-03-10 stsp if (ferror(basefile))
496 69aa0e90 2021-03-10 stsp return got_ferror(basefile, GOT_ERR_IO);
499 69aa0e90 2021-03-10 stsp r = fread(buf, 1, sizeof(buf), f);
500 69aa0e90 2021-03-10 stsp if (r == 0) {
501 69aa0e90 2021-03-10 stsp if (ferror(f))
502 69aa0e90 2021-03-10 stsp return got_ferror(f, GOT_ERR_IO);
505 69aa0e90 2021-03-10 stsp for (i = 0; i < MIN(base_r, r); i++) {
506 69aa0e90 2021-03-10 stsp if (buf[i] != basebuf[i]) {
507 69aa0e90 2021-03-10 stsp buf_equal = 0;
510 69aa0e90 2021-03-10 stsp (*blocklen)++;
514 2b0ae357 2022-01-10 thomas return NULL;
517 2b0ae357 2022-01-10 thomas static const struct got_error *
518 2b0ae357 2022-01-10 thomas stretchblk_file_mem(uint8_t *basedata, off_t base_offset0, off_t basefile_size,
519 2b0ae357 2022-01-10 thomas struct got_delta_block *block, FILE *f, off_t filesize, off_t *blocklen)
521 2b0ae357 2022-01-10 thomas uint8_t buf[GOT_DELTIFY_MAXCHUNK];
522 2b0ae357 2022-01-10 thomas size_t r, i;
523 2b0ae357 2022-01-10 thomas int buf_equal = 1;
524 2b0ae357 2022-01-10 thomas off_t base_offset = base_offset0 + block->offset + *blocklen;
526 2b0ae357 2022-01-10 thomas if (base_offset > basefile_size) {
527 2b0ae357 2022-01-10 thomas return got_error_fmt(GOT_ERR_RANGE,
528 2b0ae357 2022-01-10 thomas "read beyond the size of delta base at offset %llu",
529 2b0ae357 2022-01-10 thomas base_offset);
532 2b0ae357 2022-01-10 thomas while (buf_equal && *blocklen < (1 << 24) - 1) {
533 2b0ae357 2022-01-10 thomas if (base_offset + *blocklen >= basefile_size)
535 2b0ae357 2022-01-10 thomas r = fread(buf, 1, sizeof(buf), f);
536 2b0ae357 2022-01-10 thomas if (r == 0) {
537 2b0ae357 2022-01-10 thomas if (ferror(f))
538 2b0ae357 2022-01-10 thomas return got_ferror(f, GOT_ERR_IO);
541 2b0ae357 2022-01-10 thomas for (i = 0; i < MIN(basefile_size - base_offset, r); i++) {
542 2b0ae357 2022-01-10 thomas if (buf[i] != *(basedata + base_offset + i)) {
543 2b0ae357 2022-01-10 thomas buf_equal = 0;
546 2b0ae357 2022-01-10 thomas (*blocklen)++;
550 2b0ae357 2022-01-10 thomas return NULL;
553 2b0ae357 2022-01-10 thomas static const struct got_error *
554 2b0ae357 2022-01-10 thomas stretchblk_mem_file(FILE *basefile, off_t base_offset0,
555 2b0ae357 2022-01-10 thomas struct got_delta_block *block, uint8_t *data, off_t fileoffset,
556 2b0ae357 2022-01-10 thomas off_t filesize, off_t *blocklen)
558 2b0ae357 2022-01-10 thomas uint8_t basebuf[GOT_DELTIFY_MAXCHUNK];
559 2b0ae357 2022-01-10 thomas size_t base_r, i;
560 2b0ae357 2022-01-10 thomas int buf_equal = 1;
562 2b0ae357 2022-01-10 thomas if (fileoffset > filesize) {
563 2b0ae357 2022-01-10 thomas return got_error_fmt(GOT_ERR_RANGE,
564 2b0ae357 2022-01-10 thomas "read beyond the size of deltify file at offset %llu",
565 2b0ae357 2022-01-10 thomas fileoffset);
568 2b0ae357 2022-01-10 thomas if (fseeko(basefile, base_offset0 + block->offset + *blocklen,
569 2b0ae357 2022-01-10 thomas SEEK_SET) == -1)
570 2b0ae357 2022-01-10 thomas return got_error_from_errno("fseeko");
572 2b0ae357 2022-01-10 thomas while (buf_equal && *blocklen < (1 << 24) - 1) {
573 2b0ae357 2022-01-10 thomas if (fileoffset + *blocklen >= filesize)
575 2b0ae357 2022-01-10 thomas base_r = fread(basebuf, 1, sizeof(basebuf), basefile);
576 2b0ae357 2022-01-10 thomas if (base_r == 0) {
577 2b0ae357 2022-01-10 thomas if (ferror(basefile))
578 2b0ae357 2022-01-10 thomas return got_ferror(basefile, GOT_ERR_IO);
581 2b0ae357 2022-01-10 thomas for (i = 0; i < MIN(base_r, filesize - fileoffset); i++) {
582 2b0ae357 2022-01-10 thomas if (*(data + fileoffset + i) != basebuf[i]) {
583 2b0ae357 2022-01-10 thomas buf_equal = 0;
586 2b0ae357 2022-01-10 thomas (*blocklen)++;
590 2b0ae357 2022-01-10 thomas return NULL;
593 2b0ae357 2022-01-10 thomas static const struct got_error *
594 2b0ae357 2022-01-10 thomas stretchblk_mem_mem(uint8_t *basedata, off_t base_offset0, off_t basefile_size,
595 2b0ae357 2022-01-10 thomas struct got_delta_block *block, uint8_t *data, off_t fileoffset,
596 2b0ae357 2022-01-10 thomas off_t filesize, off_t *blocklen)
598 2b0ae357 2022-01-10 thomas off_t i, maxlen;
599 2b0ae357 2022-01-10 thomas off_t base_offset = base_offset0 + block->offset + *blocklen;
600 2b0ae357 2022-01-10 thomas uint8_t *p, *q;
602 2b0ae357 2022-01-10 thomas if (base_offset > basefile_size) {
603 2b0ae357 2022-01-10 thomas return got_error_fmt(GOT_ERR_RANGE,
604 2b0ae357 2022-01-10 thomas "read beyond the size of delta base at offset %llu",
605 2b0ae357 2022-01-10 thomas base_offset);
608 2b0ae357 2022-01-10 thomas if (fileoffset > filesize) {
609 2b0ae357 2022-01-10 thomas return got_error_fmt(GOT_ERR_RANGE,
610 2b0ae357 2022-01-10 thomas "read beyond the size of deltify file at offset %llu",
611 2b0ae357 2022-01-10 thomas fileoffset);
614 2b0ae357 2022-01-10 thomas p = data + fileoffset;
615 2b0ae357 2022-01-10 thomas q = basedata + base_offset;
616 2b0ae357 2022-01-10 thomas maxlen = MIN(basefile_size - base_offset, filesize - fileoffset);
617 2b0ae357 2022-01-10 thomas for (i = 0; i < maxlen && *blocklen < (1 << 24) - 1; i++) {
618 2b0ae357 2022-01-10 thomas if (p[i] != q[i])
620 2b0ae357 2022-01-10 thomas (*blocklen)++;
623 69aa0e90 2021-03-10 stsp return NULL;
626 69aa0e90 2021-03-10 stsp const struct got_error *
627 69aa0e90 2021-03-10 stsp got_deltify(struct got_delta_instruction **deltas, int *ndeltas,
628 69aa0e90 2021-03-10 stsp FILE *f, off_t fileoffset, off_t filesize,
629 69aa0e90 2021-03-10 stsp struct got_delta_table *dt, FILE *basefile,
630 f34b169e 2021-06-18 stsp off_t basefile_offset0, off_t basefile_size)
632 69aa0e90 2021-03-10 stsp const struct got_error *err = NULL;
633 69aa0e90 2021-03-10 stsp const off_t offset0 = fileoffset;
634 dd29967c 2021-08-22 stsp size_t nalloc = 0;
635 dd29967c 2021-08-22 stsp const size_t alloc_chunk_size = 64;
637 69aa0e90 2021-03-10 stsp *deltas = NULL;
638 69aa0e90 2021-03-10 stsp *ndeltas = 0;
641 69aa0e90 2021-03-10 stsp * offset0 indicates where data to be deltified begins.
642 69aa0e90 2021-03-10 stsp * For example, we want to avoid deltifying a Git object header at
643 69aa0e90 2021-03-10 stsp * the beginning of the file.
645 69aa0e90 2021-03-10 stsp if (fseeko(f, offset0, SEEK_SET) == -1)
646 69aa0e90 2021-03-10 stsp return got_error_from_errno("fseeko");
648 dd29967c 2021-08-22 stsp *deltas = reallocarray(NULL, alloc_chunk_size,
649 dd29967c 2021-08-22 stsp sizeof(struct got_delta_instruction));
650 dd29967c 2021-08-22 stsp if (*deltas == NULL)
651 dd29967c 2021-08-22 stsp return got_error_from_errno("reallocarray");
652 dd29967c 2021-08-22 stsp nalloc = alloc_chunk_size;
654 69aa0e90 2021-03-10 stsp while (fileoffset < filesize) {
655 69aa0e90 2021-03-10 stsp uint8_t buf[GOT_DELTIFY_MAXCHUNK];
656 69aa0e90 2021-03-10 stsp off_t blocklen;
657 69aa0e90 2021-03-10 stsp struct got_delta_block *block;
658 69aa0e90 2021-03-10 stsp err = nextblk(buf, &blocklen, f);
661 9a8dc2b3 2021-06-18 stsp if (blocklen == 0) {
662 9a8dc2b3 2021-06-18 stsp /* Source remainder from the file itself. */
663 9a8dc2b3 2021-06-18 stsp if (fileoffset < filesize) {
664 dd29967c 2021-08-22 stsp err = emitdelta(deltas, &nalloc, ndeltas,
665 dd29967c 2021-08-22 stsp alloc_chunk_size, 0, fileoffset - offset0,
666 9a8dc2b3 2021-06-18 stsp filesize - fileoffset);
670 740bba1c 2021-06-18 stsp err = lookupblk(&block, dt, buf, blocklen, basefile,
671 740bba1c 2021-06-18 stsp basefile_offset0);
674 69aa0e90 2021-03-10 stsp if (block != NULL) {
676 69aa0e90 2021-03-10 stsp * We have found a matching block in the delta base.
677 69aa0e90 2021-03-10 stsp * Attempt to stretch the block as far as possible and
678 69aa0e90 2021-03-10 stsp * generate a copy instruction.
680 740bba1c 2021-06-18 stsp err = stretchblk(basefile, basefile_offset0, block,
681 740bba1c 2021-06-18 stsp f, filesize, &blocklen);
684 dd29967c 2021-08-22 stsp err = emitdelta(deltas, &nalloc, ndeltas,
685 dd29967c 2021-08-22 stsp alloc_chunk_size, 1, block->offset, blocklen);
690 69aa0e90 2021-03-10 stsp * No match.
691 69aa0e90 2021-03-10 stsp * This block needs to be sourced from the file itself.
693 dd29967c 2021-08-22 stsp err = emitdelta(deltas, &nalloc, ndeltas,
694 dd29967c 2021-08-22 stsp alloc_chunk_size, 0, fileoffset - offset0, blocklen);
698 69aa0e90 2021-03-10 stsp fileoffset += blocklen;
699 0af64e86 2021-08-22 stsp if (fseeko(f, fileoffset, SEEK_SET) == -1) {
700 0af64e86 2021-08-22 stsp err = got_error_from_errno("fseeko");
706 69aa0e90 2021-03-10 stsp free(*deltas);
707 69aa0e90 2021-03-10 stsp *deltas = NULL;
708 69aa0e90 2021-03-10 stsp *ndeltas = 0;
710 69aa0e90 2021-03-10 stsp return err;
713 2b0ae357 2022-01-10 thomas const struct got_error *
714 2b0ae357 2022-01-10 thomas got_deltify_file_mem(struct got_delta_instruction **deltas, int *ndeltas,
715 2b0ae357 2022-01-10 thomas FILE *f, off_t fileoffset, off_t filesize,
716 2b0ae357 2022-01-10 thomas struct got_delta_table *dt, uint8_t *basedata,
717 2b0ae357 2022-01-10 thomas off_t basefile_offset0, off_t basefile_size)
719 2b0ae357 2022-01-10 thomas const struct got_error *err = NULL;
720 2b0ae357 2022-01-10 thomas const off_t offset0 = fileoffset;
721 2b0ae357 2022-01-10 thomas size_t nalloc = 0;
722 2b0ae357 2022-01-10 thomas const size_t alloc_chunk_size = 64;
724 2b0ae357 2022-01-10 thomas *deltas = NULL;
725 2b0ae357 2022-01-10 thomas *ndeltas = 0;
728 2b0ae357 2022-01-10 thomas * offset0 indicates where data to be deltified begins.
729 2b0ae357 2022-01-10 thomas * For example, we want to avoid deltifying a Git object header at
730 2b0ae357 2022-01-10 thomas * the beginning of the file.
732 2b0ae357 2022-01-10 thomas if (fseeko(f, offset0, SEEK_SET) == -1)
733 2b0ae357 2022-01-10 thomas return got_error_from_errno("fseeko");
735 2b0ae357 2022-01-10 thomas *deltas = reallocarray(NULL, alloc_chunk_size,
736 2b0ae357 2022-01-10 thomas sizeof(struct got_delta_instruction));
737 2b0ae357 2022-01-10 thomas if (*deltas == NULL)
738 2b0ae357 2022-01-10 thomas return got_error_from_errno("reallocarray");
739 2b0ae357 2022-01-10 thomas nalloc = alloc_chunk_size;
741 2b0ae357 2022-01-10 thomas while (fileoffset < filesize) {
742 2b0ae357 2022-01-10 thomas uint8_t buf[GOT_DELTIFY_MAXCHUNK];
743 2b0ae357 2022-01-10 thomas off_t blocklen;
744 2b0ae357 2022-01-10 thomas struct got_delta_block *block;
745 2b0ae357 2022-01-10 thomas err = nextblk(buf, &blocklen, f);
748 2b0ae357 2022-01-10 thomas if (blocklen == 0) {
749 2b0ae357 2022-01-10 thomas /* Source remainder from the file itself. */
750 2b0ae357 2022-01-10 thomas if (fileoffset < filesize) {
751 2b0ae357 2022-01-10 thomas err = emitdelta(deltas, &nalloc, ndeltas,
752 2b0ae357 2022-01-10 thomas alloc_chunk_size, 0, fileoffset - offset0,
753 2b0ae357 2022-01-10 thomas filesize - fileoffset);
757 2b0ae357 2022-01-10 thomas err = lookupblk_mem(&block, dt, buf, blocklen, basedata,
758 2b0ae357 2022-01-10 thomas basefile_offset0);
761 2b0ae357 2022-01-10 thomas if (block != NULL) {
763 2b0ae357 2022-01-10 thomas * We have found a matching block in the delta base.
764 2b0ae357 2022-01-10 thomas * Attempt to stretch the block as far as possible and
765 2b0ae357 2022-01-10 thomas * generate a copy instruction.
767 2b0ae357 2022-01-10 thomas err = stretchblk_file_mem(basedata, basefile_offset0,
768 2b0ae357 2022-01-10 thomas basefile_size, block, f, filesize, &blocklen);
771 2b0ae357 2022-01-10 thomas err = emitdelta(deltas, &nalloc, ndeltas,
772 2b0ae357 2022-01-10 thomas alloc_chunk_size, 1, block->offset, blocklen);
777 2b0ae357 2022-01-10 thomas * No match.
778 2b0ae357 2022-01-10 thomas * This block needs to be sourced from the file itself.
780 2b0ae357 2022-01-10 thomas err = emitdelta(deltas, &nalloc, ndeltas,
781 2b0ae357 2022-01-10 thomas alloc_chunk_size, 0, fileoffset - offset0, blocklen);
785 2b0ae357 2022-01-10 thomas fileoffset += blocklen;
786 2b0ae357 2022-01-10 thomas if (fseeko(f, fileoffset, SEEK_SET) == -1) {
787 2b0ae357 2022-01-10 thomas err = got_error_from_errno("fseeko");
792 2b0ae357 2022-01-10 thomas if (err) {
793 2b0ae357 2022-01-10 thomas free(*deltas);
794 2b0ae357 2022-01-10 thomas *deltas = NULL;
795 2b0ae357 2022-01-10 thomas *ndeltas = 0;
797 2b0ae357 2022-01-10 thomas return err;
800 2b0ae357 2022-01-10 thomas const struct got_error *
801 2b0ae357 2022-01-10 thomas got_deltify_mem_file(struct got_delta_instruction **deltas, int *ndeltas,
802 2b0ae357 2022-01-10 thomas uint8_t *data, off_t fileoffset, off_t filesize,
803 2b0ae357 2022-01-10 thomas struct got_delta_table *dt, FILE *basefile,
804 2b0ae357 2022-01-10 thomas off_t basefile_offset0, off_t basefile_size)
806 2b0ae357 2022-01-10 thomas const struct got_error *err = NULL;
807 2b0ae357 2022-01-10 thomas const off_t offset0 = fileoffset;
808 2b0ae357 2022-01-10 thomas size_t nalloc = 0;
809 2b0ae357 2022-01-10 thomas const size_t alloc_chunk_size = 64;
811 2b0ae357 2022-01-10 thomas *deltas = NULL;
812 2b0ae357 2022-01-10 thomas *ndeltas = 0;
814 2b0ae357 2022-01-10 thomas *deltas = reallocarray(NULL, alloc_chunk_size,
815 2b0ae357 2022-01-10 thomas sizeof(struct got_delta_instruction));
816 2b0ae357 2022-01-10 thomas if (*deltas == NULL)
817 2b0ae357 2022-01-10 thomas return got_error_from_errno("reallocarray");
818 2b0ae357 2022-01-10 thomas nalloc = alloc_chunk_size;
820 2b0ae357 2022-01-10 thomas while (fileoffset < filesize) {
821 2b0ae357 2022-01-10 thomas off_t blocklen;
822 2b0ae357 2022-01-10 thomas struct got_delta_block *block;
823 2b0ae357 2022-01-10 thomas err = nextblk_mem(&blocklen, data, fileoffset, filesize);
826 2b0ae357 2022-01-10 thomas if (blocklen == 0) {
827 2b0ae357 2022-01-10 thomas /* Source remainder from the file itself. */
828 2b0ae357 2022-01-10 thomas if (fileoffset < filesize) {
829 2b0ae357 2022-01-10 thomas err = emitdelta(deltas, &nalloc, ndeltas,
830 2b0ae357 2022-01-10 thomas alloc_chunk_size, 0, fileoffset - offset0,
831 2b0ae357 2022-01-10 thomas filesize - fileoffset);
835 2b0ae357 2022-01-10 thomas err = lookupblk(&block, dt, data + fileoffset, blocklen,
836 2b0ae357 2022-01-10 thomas basefile, basefile_offset0);
839 2b0ae357 2022-01-10 thomas if (block != NULL) {
841 2b0ae357 2022-01-10 thomas * We have found a matching block in the delta base.
842 2b0ae357 2022-01-10 thomas * Attempt to stretch the block as far as possible and
843 2b0ae357 2022-01-10 thomas * generate a copy instruction.
845 2b0ae357 2022-01-10 thomas err = stretchblk_mem_file(basefile, basefile_offset0,
846 2b0ae357 2022-01-10 thomas block, data, fileoffset + blocklen, filesize,
847 2b0ae357 2022-01-10 thomas &blocklen);
850 2b0ae357 2022-01-10 thomas err = emitdelta(deltas, &nalloc, ndeltas,
851 2b0ae357 2022-01-10 thomas alloc_chunk_size, 1, block->offset, blocklen);
856 2b0ae357 2022-01-10 thomas * No match.
857 2b0ae357 2022-01-10 thomas * This block needs to be sourced from the file itself.
859 2b0ae357 2022-01-10 thomas err = emitdelta(deltas, &nalloc, ndeltas,
860 2b0ae357 2022-01-10 thomas alloc_chunk_size, 0, fileoffset - offset0, blocklen);
864 2b0ae357 2022-01-10 thomas fileoffset += blocklen;
867 2b0ae357 2022-01-10 thomas if (err) {
868 2b0ae357 2022-01-10 thomas free(*deltas);
869 2b0ae357 2022-01-10 thomas *deltas = NULL;
870 2b0ae357 2022-01-10 thomas *ndeltas = 0;
872 2b0ae357 2022-01-10 thomas return err;
875 2b0ae357 2022-01-10 thomas const struct got_error *
876 2b0ae357 2022-01-10 thomas got_deltify_mem_mem(struct got_delta_instruction **deltas, int *ndeltas,
877 2b0ae357 2022-01-10 thomas uint8_t *data, off_t fileoffset, off_t filesize,
878 2b0ae357 2022-01-10 thomas struct got_delta_table *dt, uint8_t *basedata,
879 2b0ae357 2022-01-10 thomas off_t basefile_offset0, off_t basefile_size)
881 2b0ae357 2022-01-10 thomas const struct got_error *err = NULL;
882 2b0ae357 2022-01-10 thomas const off_t offset0 = fileoffset;
883 2b0ae357 2022-01-10 thomas size_t nalloc = 0;
884 2b0ae357 2022-01-10 thomas const size_t alloc_chunk_size = 64;
886 2b0ae357 2022-01-10 thomas *deltas = NULL;
887 2b0ae357 2022-01-10 thomas *ndeltas = 0;
889 2b0ae357 2022-01-10 thomas *deltas = reallocarray(NULL, alloc_chunk_size,
890 2b0ae357 2022-01-10 thomas sizeof(struct got_delta_instruction));
891 2b0ae357 2022-01-10 thomas if (*deltas == NULL)
892 2b0ae357 2022-01-10 thomas return got_error_from_errno("reallocarray");
893 2b0ae357 2022-01-10 thomas nalloc = alloc_chunk_size;
895 2b0ae357 2022-01-10 thomas while (fileoffset < filesize) {
896 2b0ae357 2022-01-10 thomas off_t blocklen;
897 2b0ae357 2022-01-10 thomas struct got_delta_block *block;
898 2b0ae357 2022-01-10 thomas err = nextblk_mem(&blocklen, data, fileoffset, filesize);
901 2b0ae357 2022-01-10 thomas if (blocklen == 0) {
902 2b0ae357 2022-01-10 thomas /* Source remainder from the file itself. */
903 2b0ae357 2022-01-10 thomas if (fileoffset < filesize) {
904 2b0ae357 2022-01-10 thomas err = emitdelta(deltas, &nalloc, ndeltas,
905 2b0ae357 2022-01-10 thomas alloc_chunk_size, 0, fileoffset - offset0,
906 2b0ae357 2022-01-10 thomas filesize - fileoffset);
910 2b0ae357 2022-01-10 thomas err = lookupblk_mem(&block, dt, data + fileoffset, blocklen,
911 2b0ae357 2022-01-10 thomas basedata, basefile_offset0);
914 2b0ae357 2022-01-10 thomas if (block != NULL) {
916 2b0ae357 2022-01-10 thomas * We have found a matching block in the delta base.
917 2b0ae357 2022-01-10 thomas * Attempt to stretch the block as far as possible and
918 2b0ae357 2022-01-10 thomas * generate a copy instruction.
920 2b0ae357 2022-01-10 thomas err = stretchblk_mem_mem(basedata, basefile_offset0,
921 2b0ae357 2022-01-10 thomas basefile_size, block, data, fileoffset + blocklen,
922 2b0ae357 2022-01-10 thomas filesize, &blocklen);
925 2b0ae357 2022-01-10 thomas err = emitdelta(deltas, &nalloc, ndeltas,
926 2b0ae357 2022-01-10 thomas alloc_chunk_size, 1, block->offset, blocklen);
931 2b0ae357 2022-01-10 thomas * No match.
932 2b0ae357 2022-01-10 thomas * This block needs to be sourced from the file itself.
934 2b0ae357 2022-01-10 thomas err = emitdelta(deltas, &nalloc, ndeltas,
935 2b0ae357 2022-01-10 thomas alloc_chunk_size, 0, fileoffset - offset0, blocklen);
939 2b0ae357 2022-01-10 thomas fileoffset += blocklen;
942 2b0ae357 2022-01-10 thomas if (err) {
943 2b0ae357 2022-01-10 thomas free(*deltas);
944 2b0ae357 2022-01-10 thomas *deltas = NULL;
945 2b0ae357 2022-01-10 thomas *ndeltas = 0;
947 2b0ae357 2022-01-10 thomas return err;
