Blame


1 8704c7ce 2021-03-10 stsp /*
2 8704c7ce 2021-03-10 stsp * Copyright (c) 2021 Stefan Sperling <stsp@openbsd.org>
3 8704c7ce 2021-03-10 stsp *
4 8704c7ce 2021-03-10 stsp * Permission to use, copy, modify, and distribute this software for any
5 8704c7ce 2021-03-10 stsp * purpose with or without fee is hereby granted, provided that the above
6 8704c7ce 2021-03-10 stsp * copyright notice and this permission notice appear in all copies.
7 8704c7ce 2021-03-10 stsp *
8 8704c7ce 2021-03-10 stsp * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 8704c7ce 2021-03-10 stsp * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 8704c7ce 2021-03-10 stsp * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 8704c7ce 2021-03-10 stsp * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 8704c7ce 2021-03-10 stsp * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 8704c7ce 2021-03-10 stsp * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 8704c7ce 2021-03-10 stsp * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 8704c7ce 2021-03-10 stsp */
16 8704c7ce 2021-03-10 stsp
17 8704c7ce 2021-03-10 stsp #include <stdio.h>
18 8704c7ce 2021-03-10 stsp #include <stdlib.h>
19 8704c7ce 2021-03-10 stsp #include <string.h>
20 8704c7ce 2021-03-10 stsp #include <err.h>
21 8704c7ce 2021-03-10 stsp #include <unistd.h>
22 8704c7ce 2021-03-10 stsp #include <getopt.h>
23 8704c7ce 2021-03-10 stsp
24 8704c7ce 2021-03-10 stsp #include "got_error.h"
25 8704c7ce 2021-03-10 stsp #include "got_opentemp.h"
26 8704c7ce 2021-03-10 stsp
27 8704c7ce 2021-03-10 stsp #include "got_lib_deltify.h"
28 8704c7ce 2021-03-10 stsp
29 8704c7ce 2021-03-10 stsp #ifndef nitems
30 8704c7ce 2021-03-10 stsp #define nitems(_a) (sizeof(_a) / sizeof((_a)[0]))
31 8704c7ce 2021-03-10 stsp #endif
32 8704c7ce 2021-03-10 stsp
33 8704c7ce 2021-03-10 stsp static int
34 8704c7ce 2021-03-10 stsp deltify_abc_axc(void)
35 8704c7ce 2021-03-10 stsp {
36 8704c7ce 2021-03-10 stsp const struct got_error *err = NULL;
37 8704c7ce 2021-03-10 stsp size_t i;
38 8704c7ce 2021-03-10 stsp FILE *base_file, *derived_file, *result_file;
39 8704c7ce 2021-03-10 stsp struct got_delta_table *dt;
40 8704c7ce 2021-03-10 stsp struct got_delta_instruction *deltas;
41 8704c7ce 2021-03-10 stsp int ndeltas;
42 8704c7ce 2021-03-10 stsp int have_nblocks = 0;
43 cc524d36 2022-05-31 thomas uint32_t seed;
44 8704c7ce 2021-03-10 stsp
45 cc524d36 2022-05-31 thomas seed = arc4random();
46 cc524d36 2022-05-31 thomas
47 8704c7ce 2021-03-10 stsp base_file = got_opentemp();
48 8704c7ce 2021-03-10 stsp if (base_file == NULL)
49 8704c7ce 2021-03-10 stsp return 1;
50 8704c7ce 2021-03-10 stsp
51 8704c7ce 2021-03-10 stsp derived_file = got_opentemp();
52 8704c7ce 2021-03-10 stsp if (derived_file == NULL)
53 8704c7ce 2021-03-10 stsp return 1;
54 8704c7ce 2021-03-10 stsp
55 8704c7ce 2021-03-10 stsp result_file = got_opentemp();
56 8704c7ce 2021-03-10 stsp if (result_file == NULL)
57 8704c7ce 2021-03-10 stsp return 1;
58 8704c7ce 2021-03-10 stsp
59 8704c7ce 2021-03-10 stsp for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
60 8704c7ce 2021-03-10 stsp fputc('a', base_file);
61 8704c7ce 2021-03-10 stsp fputc('a', derived_file);
62 8704c7ce 2021-03-10 stsp }
63 8704c7ce 2021-03-10 stsp for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
64 8704c7ce 2021-03-10 stsp fputc('b', base_file);
65 8704c7ce 2021-03-10 stsp fputc('x', derived_file);
66 8704c7ce 2021-03-10 stsp }
67 8704c7ce 2021-03-10 stsp for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
68 8704c7ce 2021-03-10 stsp fputc('c', base_file);
69 8704c7ce 2021-03-10 stsp fputc('c', derived_file);
70 8704c7ce 2021-03-10 stsp }
71 8704c7ce 2021-03-10 stsp
72 8704c7ce 2021-03-10 stsp rewind(base_file);
73 8704c7ce 2021-03-10 stsp rewind(derived_file);
74 8704c7ce 2021-03-10 stsp
75 cc524d36 2022-05-31 thomas err = got_deltify_init(&dt, base_file, 0, 3 * GOT_DELTIFY_MAXCHUNK,
76 cc524d36 2022-05-31 thomas seed);
77 8704c7ce 2021-03-10 stsp if (err)
78 8704c7ce 2021-03-10 stsp goto done;
79 8704c7ce 2021-03-10 stsp
80 8704c7ce 2021-03-10 stsp for (i = 0; i < dt->nalloc; i++) {
81 8704c7ce 2021-03-10 stsp if (dt->blocks[i].len > 0)
82 8704c7ce 2021-03-10 stsp have_nblocks++;
83 8704c7ce 2021-03-10 stsp }
84 8704c7ce 2021-03-10 stsp if (have_nblocks != dt->nblocks) {
85 8704c7ce 2021-03-10 stsp err = got_error(GOT_ERR_BAD_DELTA);
86 8704c7ce 2021-03-10 stsp goto done;
87 8704c7ce 2021-03-10 stsp }
88 8704c7ce 2021-03-10 stsp
89 8704c7ce 2021-03-10 stsp err = got_deltify(&deltas, &ndeltas, derived_file, 0,
90 cc524d36 2022-05-31 thomas 3 * GOT_DELTIFY_MAXCHUNK, seed, dt, base_file, 0,
91 f34b169e 2021-06-18 stsp 3 * GOT_DELTIFY_MAXCHUNK);
92 8704c7ce 2021-03-10 stsp if (err)
93 8704c7ce 2021-03-10 stsp goto done;
94 8704c7ce 2021-03-10 stsp
95 8704c7ce 2021-03-10 stsp if (ndeltas != 3) {
96 8704c7ce 2021-03-10 stsp err = got_error(GOT_ERR_BAD_DELTA);
97 8704c7ce 2021-03-10 stsp goto done;
98 8704c7ce 2021-03-10 stsp }
99 8704c7ce 2021-03-10 stsp /* Copy 'aaaa...' from base file. */
100 8704c7ce 2021-03-10 stsp if (!(deltas[0].copy == 1 && deltas[0].offset == 0 &&
101 8704c7ce 2021-03-10 stsp deltas[0].len == GOT_DELTIFY_MAXCHUNK)) {
102 8704c7ce 2021-03-10 stsp err = got_error(GOT_ERR_BAD_DELTA);
103 8704c7ce 2021-03-10 stsp goto done;
104 8704c7ce 2021-03-10 stsp }
105 8704c7ce 2021-03-10 stsp /* Copy 'xxxx...' from derived file. */
106 8704c7ce 2021-03-10 stsp if (!(deltas[1].copy == 0 && deltas[1].offset == GOT_DELTIFY_MAXCHUNK &&
107 8704c7ce 2021-03-10 stsp deltas[1].len == GOT_DELTIFY_MAXCHUNK)) {
108 8704c7ce 2021-03-10 stsp err = got_error(GOT_ERR_BAD_DELTA);
109 8704c7ce 2021-03-10 stsp goto done;
110 8704c7ce 2021-03-10 stsp }
111 8704c7ce 2021-03-10 stsp /* Copy 'ccccc...' from base file. */
112 8704c7ce 2021-03-10 stsp if (!(deltas[2].copy == 1 &&
113 8704c7ce 2021-03-10 stsp deltas[2].offset == 2 * GOT_DELTIFY_MAXCHUNK &&
114 8704c7ce 2021-03-10 stsp deltas[2].len == GOT_DELTIFY_MAXCHUNK)) {
115 8704c7ce 2021-03-10 stsp err = got_error(GOT_ERR_BAD_DELTA);
116 8704c7ce 2021-03-10 stsp goto done;
117 8704c7ce 2021-03-10 stsp }
118 8704c7ce 2021-03-10 stsp
119 8704c7ce 2021-03-10 stsp done:
120 8704c7ce 2021-03-10 stsp got_deltify_free(dt);
121 8704c7ce 2021-03-10 stsp fclose(base_file);
122 8704c7ce 2021-03-10 stsp fclose(derived_file);
123 2b0ae357 2022-01-10 thomas fclose(result_file);
124 2b0ae357 2022-01-10 thomas return (err == NULL);
125 2b0ae357 2022-01-10 thomas }
126 2b0ae357 2022-01-10 thomas
127 2b0ae357 2022-01-10 thomas static int
128 2b0ae357 2022-01-10 thomas deltify_abc_axc_file_mem(void)
129 2b0ae357 2022-01-10 thomas {
130 2b0ae357 2022-01-10 thomas const struct got_error *err = NULL;
131 2b0ae357 2022-01-10 thomas size_t i;
132 2b0ae357 2022-01-10 thomas uint8_t base_data[3 * GOT_DELTIFY_MAXCHUNK];
133 2b0ae357 2022-01-10 thomas FILE *derived_file, *result_file;
134 2b0ae357 2022-01-10 thomas struct got_delta_table *dt;
135 2b0ae357 2022-01-10 thomas struct got_delta_instruction *deltas;
136 2b0ae357 2022-01-10 thomas int ndeltas;
137 2b0ae357 2022-01-10 thomas int have_nblocks = 0;
138 cc524d36 2022-05-31 thomas uint32_t seed;
139 2b0ae357 2022-01-10 thomas
140 cc524d36 2022-05-31 thomas seed = arc4random();
141 cc524d36 2022-05-31 thomas
142 2b0ae357 2022-01-10 thomas derived_file = got_opentemp();
143 2b0ae357 2022-01-10 thomas if (derived_file == NULL)
144 2b0ae357 2022-01-10 thomas return 1;
145 2b0ae357 2022-01-10 thomas
146 2b0ae357 2022-01-10 thomas result_file = got_opentemp();
147 2b0ae357 2022-01-10 thomas if (result_file == NULL)
148 2b0ae357 2022-01-10 thomas return 1;
149 2b0ae357 2022-01-10 thomas
150 2b0ae357 2022-01-10 thomas for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
151 2b0ae357 2022-01-10 thomas base_data[i] = 'a';
152 2b0ae357 2022-01-10 thomas fputc('a', derived_file);
153 2b0ae357 2022-01-10 thomas }
154 2b0ae357 2022-01-10 thomas for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
155 2b0ae357 2022-01-10 thomas base_data[GOT_DELTIFY_MAXCHUNK + i] = 'b';
156 2b0ae357 2022-01-10 thomas fputc('x', derived_file);
157 2b0ae357 2022-01-10 thomas }
158 2b0ae357 2022-01-10 thomas for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
159 2b0ae357 2022-01-10 thomas base_data[2 * GOT_DELTIFY_MAXCHUNK + i] = 'c';
160 2b0ae357 2022-01-10 thomas fputc('c', derived_file);
161 2b0ae357 2022-01-10 thomas }
162 2b0ae357 2022-01-10 thomas
163 2b0ae357 2022-01-10 thomas rewind(derived_file);
164 2b0ae357 2022-01-10 thomas
165 cc524d36 2022-05-31 thomas err = got_deltify_init_mem(&dt, base_data, 0, 3 * GOT_DELTIFY_MAXCHUNK,
166 cc524d36 2022-05-31 thomas seed);
167 2b0ae357 2022-01-10 thomas if (err)
168 2b0ae357 2022-01-10 thomas goto done;
169 2b0ae357 2022-01-10 thomas
170 2b0ae357 2022-01-10 thomas for (i = 0; i < dt->nalloc; i++) {
171 2b0ae357 2022-01-10 thomas if (dt->blocks[i].len > 0)
172 2b0ae357 2022-01-10 thomas have_nblocks++;
173 2b0ae357 2022-01-10 thomas }
174 2b0ae357 2022-01-10 thomas if (have_nblocks != dt->nblocks) {
175 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
176 2b0ae357 2022-01-10 thomas goto done;
177 2b0ae357 2022-01-10 thomas }
178 2b0ae357 2022-01-10 thomas
179 2b0ae357 2022-01-10 thomas err = got_deltify_file_mem(&deltas, &ndeltas, derived_file, 0,
180 cc524d36 2022-05-31 thomas 3 * GOT_DELTIFY_MAXCHUNK, seed, dt, base_data, 0,
181 2b0ae357 2022-01-10 thomas 3 * GOT_DELTIFY_MAXCHUNK);
182 2b0ae357 2022-01-10 thomas if (err)
183 2b0ae357 2022-01-10 thomas goto done;
184 2b0ae357 2022-01-10 thomas
185 2b0ae357 2022-01-10 thomas if (ndeltas != 3) {
186 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
187 2b0ae357 2022-01-10 thomas goto done;
188 2b0ae357 2022-01-10 thomas }
189 2b0ae357 2022-01-10 thomas /* Copy 'aaaa...' from base file. */
190 2b0ae357 2022-01-10 thomas if (!(deltas[0].copy == 1 && deltas[0].offset == 0 &&
191 2b0ae357 2022-01-10 thomas deltas[0].len == GOT_DELTIFY_MAXCHUNK)) {
192 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
193 2b0ae357 2022-01-10 thomas goto done;
194 2b0ae357 2022-01-10 thomas }
195 2b0ae357 2022-01-10 thomas /* Copy 'xxxx...' from derived file. */
196 2b0ae357 2022-01-10 thomas if (!(deltas[1].copy == 0 && deltas[1].offset == GOT_DELTIFY_MAXCHUNK &&
197 2b0ae357 2022-01-10 thomas deltas[1].len == GOT_DELTIFY_MAXCHUNK)) {
198 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
199 2b0ae357 2022-01-10 thomas goto done;
200 2b0ae357 2022-01-10 thomas }
201 2b0ae357 2022-01-10 thomas /* Copy 'ccccc...' from base file. */
202 2b0ae357 2022-01-10 thomas if (!(deltas[2].copy == 1 &&
203 2b0ae357 2022-01-10 thomas deltas[2].offset == 2 * GOT_DELTIFY_MAXCHUNK &&
204 2b0ae357 2022-01-10 thomas deltas[2].len == GOT_DELTIFY_MAXCHUNK)) {
205 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
206 2b0ae357 2022-01-10 thomas goto done;
207 2b0ae357 2022-01-10 thomas }
208 2b0ae357 2022-01-10 thomas
209 2b0ae357 2022-01-10 thomas done:
210 2b0ae357 2022-01-10 thomas got_deltify_free(dt);
211 2b0ae357 2022-01-10 thomas fclose(derived_file);
212 2b0ae357 2022-01-10 thomas fclose(result_file);
213 2b0ae357 2022-01-10 thomas return (err == NULL);
214 2b0ae357 2022-01-10 thomas }
215 2b0ae357 2022-01-10 thomas
216 2b0ae357 2022-01-10 thomas static int
217 2b0ae357 2022-01-10 thomas deltify_abc_axc_mem_file(void)
218 2b0ae357 2022-01-10 thomas {
219 2b0ae357 2022-01-10 thomas const struct got_error *err = NULL;
220 2b0ae357 2022-01-10 thomas size_t i;
221 2b0ae357 2022-01-10 thomas FILE *base_file, *result_file;
222 2b0ae357 2022-01-10 thomas uint8_t derived_file[3 * GOT_DELTIFY_MAXCHUNK];
223 2b0ae357 2022-01-10 thomas struct got_delta_table *dt;
224 2b0ae357 2022-01-10 thomas struct got_delta_instruction *deltas;
225 2b0ae357 2022-01-10 thomas int ndeltas;
226 2b0ae357 2022-01-10 thomas int have_nblocks = 0;
227 cc524d36 2022-05-31 thomas uint32_t seed;
228 2b0ae357 2022-01-10 thomas
229 cc524d36 2022-05-31 thomas seed = arc4random();
230 cc524d36 2022-05-31 thomas
231 2b0ae357 2022-01-10 thomas base_file = got_opentemp();
232 2b0ae357 2022-01-10 thomas if (base_file == NULL)
233 2b0ae357 2022-01-10 thomas return 1;
234 2b0ae357 2022-01-10 thomas
235 2b0ae357 2022-01-10 thomas result_file = got_opentemp();
236 2b0ae357 2022-01-10 thomas if (result_file == NULL)
237 2b0ae357 2022-01-10 thomas return 1;
238 2b0ae357 2022-01-10 thomas
239 2b0ae357 2022-01-10 thomas for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
240 2b0ae357 2022-01-10 thomas fputc('a', base_file);
241 2b0ae357 2022-01-10 thomas derived_file[i] = 'a';
242 2b0ae357 2022-01-10 thomas }
243 2b0ae357 2022-01-10 thomas for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
244 2b0ae357 2022-01-10 thomas fputc('b', base_file);
245 2b0ae357 2022-01-10 thomas derived_file[GOT_DELTIFY_MAXCHUNK + i] = 'x';
246 2b0ae357 2022-01-10 thomas }
247 2b0ae357 2022-01-10 thomas for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
248 2b0ae357 2022-01-10 thomas fputc('c', base_file);
249 2b0ae357 2022-01-10 thomas derived_file[2 * GOT_DELTIFY_MAXCHUNK + i] = 'c';
250 2b0ae357 2022-01-10 thomas }
251 2b0ae357 2022-01-10 thomas
252 2b0ae357 2022-01-10 thomas rewind(base_file);
253 2b0ae357 2022-01-10 thomas
254 cc524d36 2022-05-31 thomas err = got_deltify_init(&dt, base_file, 0, 3 * GOT_DELTIFY_MAXCHUNK,
255 cc524d36 2022-05-31 thomas seed);
256 2b0ae357 2022-01-10 thomas if (err)
257 2b0ae357 2022-01-10 thomas goto done;
258 2b0ae357 2022-01-10 thomas
259 2b0ae357 2022-01-10 thomas for (i = 0; i < dt->nalloc; i++) {
260 2b0ae357 2022-01-10 thomas if (dt->blocks[i].len > 0)
261 2b0ae357 2022-01-10 thomas have_nblocks++;
262 2b0ae357 2022-01-10 thomas }
263 2b0ae357 2022-01-10 thomas if (have_nblocks != dt->nblocks) {
264 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
265 2b0ae357 2022-01-10 thomas goto done;
266 2b0ae357 2022-01-10 thomas }
267 2b0ae357 2022-01-10 thomas
268 2b0ae357 2022-01-10 thomas err = got_deltify_mem_file(&deltas, &ndeltas, derived_file, 0,
269 cc524d36 2022-05-31 thomas 3 * GOT_DELTIFY_MAXCHUNK, seed, dt, base_file, 0,
270 2b0ae357 2022-01-10 thomas 3 * GOT_DELTIFY_MAXCHUNK);
271 2b0ae357 2022-01-10 thomas if (err)
272 2b0ae357 2022-01-10 thomas goto done;
273 2b0ae357 2022-01-10 thomas
274 2b0ae357 2022-01-10 thomas if (ndeltas != 3) {
275 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
276 2b0ae357 2022-01-10 thomas goto done;
277 2b0ae357 2022-01-10 thomas }
278 2b0ae357 2022-01-10 thomas /* Copy 'aaaa...' from base file. */
279 2b0ae357 2022-01-10 thomas if (!(deltas[0].copy == 1 && deltas[0].offset == 0 &&
280 2b0ae357 2022-01-10 thomas deltas[0].len == GOT_DELTIFY_MAXCHUNK)) {
281 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
282 2b0ae357 2022-01-10 thomas goto done;
283 2b0ae357 2022-01-10 thomas }
284 2b0ae357 2022-01-10 thomas /* Copy 'xxxx...' from derived file. */
285 2b0ae357 2022-01-10 thomas if (!(deltas[1].copy == 0 && deltas[1].offset == GOT_DELTIFY_MAXCHUNK &&
286 2b0ae357 2022-01-10 thomas deltas[1].len == GOT_DELTIFY_MAXCHUNK)) {
287 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
288 2b0ae357 2022-01-10 thomas goto done;
289 2b0ae357 2022-01-10 thomas }
290 2b0ae357 2022-01-10 thomas /* Copy 'ccccc...' from base file. */
291 2b0ae357 2022-01-10 thomas if (!(deltas[2].copy == 1 &&
292 2b0ae357 2022-01-10 thomas deltas[2].offset == 2 * GOT_DELTIFY_MAXCHUNK &&
293 2b0ae357 2022-01-10 thomas deltas[2].len == GOT_DELTIFY_MAXCHUNK)) {
294 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
295 2b0ae357 2022-01-10 thomas goto done;
296 2b0ae357 2022-01-10 thomas }
297 2b0ae357 2022-01-10 thomas
298 2b0ae357 2022-01-10 thomas done:
299 2b0ae357 2022-01-10 thomas got_deltify_free(dt);
300 2b0ae357 2022-01-10 thomas fclose(base_file);
301 8704c7ce 2021-03-10 stsp fclose(result_file);
302 8704c7ce 2021-03-10 stsp return (err == NULL);
303 8704c7ce 2021-03-10 stsp }
304 8704c7ce 2021-03-10 stsp
305 2b0ae357 2022-01-10 thomas static int
306 2b0ae357 2022-01-10 thomas deltify_abc_axc_mem_mem(void)
307 2b0ae357 2022-01-10 thomas {
308 2b0ae357 2022-01-10 thomas const struct got_error *err = NULL;
309 2b0ae357 2022-01-10 thomas size_t i;
310 2b0ae357 2022-01-10 thomas FILE *result_file;
311 2b0ae357 2022-01-10 thomas uint8_t base_file[3 * GOT_DELTIFY_MAXCHUNK];
312 2b0ae357 2022-01-10 thomas uint8_t derived_file[3 * GOT_DELTIFY_MAXCHUNK];
313 2b0ae357 2022-01-10 thomas struct got_delta_table *dt;
314 2b0ae357 2022-01-10 thomas struct got_delta_instruction *deltas;
315 2b0ae357 2022-01-10 thomas int ndeltas;
316 2b0ae357 2022-01-10 thomas int have_nblocks = 0;
317 cc524d36 2022-05-31 thomas uint32_t seed;
318 2b0ae357 2022-01-10 thomas
319 cc524d36 2022-05-31 thomas seed = arc4random();
320 cc524d36 2022-05-31 thomas
321 2b0ae357 2022-01-10 thomas result_file = got_opentemp();
322 2b0ae357 2022-01-10 thomas if (result_file == NULL)
323 2b0ae357 2022-01-10 thomas return 1;
324 2b0ae357 2022-01-10 thomas
325 2b0ae357 2022-01-10 thomas for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
326 2b0ae357 2022-01-10 thomas base_file[i] = 'a';
327 2b0ae357 2022-01-10 thomas derived_file[i] = 'a';
328 2b0ae357 2022-01-10 thomas }
329 2b0ae357 2022-01-10 thomas for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
330 2b0ae357 2022-01-10 thomas base_file[GOT_DELTIFY_MAXCHUNK + i] = 'b';
331 2b0ae357 2022-01-10 thomas derived_file[GOT_DELTIFY_MAXCHUNK + i] = 'x';
332 2b0ae357 2022-01-10 thomas }
333 2b0ae357 2022-01-10 thomas for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
334 2b0ae357 2022-01-10 thomas base_file[2 * GOT_DELTIFY_MAXCHUNK + i] = 'c';
335 2b0ae357 2022-01-10 thomas derived_file[2 * GOT_DELTIFY_MAXCHUNK + i] = 'c';
336 2b0ae357 2022-01-10 thomas }
337 2b0ae357 2022-01-10 thomas
338 cc524d36 2022-05-31 thomas err = got_deltify_init_mem(&dt, base_file, 0, 3 * GOT_DELTIFY_MAXCHUNK,
339 cc524d36 2022-05-31 thomas seed);
340 2b0ae357 2022-01-10 thomas if (err)
341 2b0ae357 2022-01-10 thomas goto done;
342 2b0ae357 2022-01-10 thomas
343 2b0ae357 2022-01-10 thomas for (i = 0; i < dt->nalloc; i++) {
344 2b0ae357 2022-01-10 thomas if (dt->blocks[i].len > 0)
345 2b0ae357 2022-01-10 thomas have_nblocks++;
346 2b0ae357 2022-01-10 thomas }
347 2b0ae357 2022-01-10 thomas if (have_nblocks != dt->nblocks) {
348 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
349 2b0ae357 2022-01-10 thomas goto done;
350 2b0ae357 2022-01-10 thomas }
351 2b0ae357 2022-01-10 thomas
352 2b0ae357 2022-01-10 thomas err = got_deltify_mem_mem(&deltas, &ndeltas, derived_file, 0,
353 cc524d36 2022-05-31 thomas 3 * GOT_DELTIFY_MAXCHUNK, seed, dt, base_file, 0,
354 2b0ae357 2022-01-10 thomas 3 * GOT_DELTIFY_MAXCHUNK);
355 2b0ae357 2022-01-10 thomas if (err)
356 2b0ae357 2022-01-10 thomas goto done;
357 2b0ae357 2022-01-10 thomas
358 2b0ae357 2022-01-10 thomas if (ndeltas != 3) {
359 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
360 2b0ae357 2022-01-10 thomas goto done;
361 2b0ae357 2022-01-10 thomas }
362 2b0ae357 2022-01-10 thomas /* Copy 'aaaa...' from base file. */
363 2b0ae357 2022-01-10 thomas if (!(deltas[0].copy == 1 && deltas[0].offset == 0 &&
364 2b0ae357 2022-01-10 thomas deltas[0].len == GOT_DELTIFY_MAXCHUNK)) {
365 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
366 2b0ae357 2022-01-10 thomas goto done;
367 2b0ae357 2022-01-10 thomas }
368 2b0ae357 2022-01-10 thomas /* Copy 'xxxx...' from derived file. */
369 2b0ae357 2022-01-10 thomas if (!(deltas[1].copy == 0 && deltas[1].offset == GOT_DELTIFY_MAXCHUNK &&
370 2b0ae357 2022-01-10 thomas deltas[1].len == GOT_DELTIFY_MAXCHUNK)) {
371 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
372 2b0ae357 2022-01-10 thomas goto done;
373 2b0ae357 2022-01-10 thomas }
374 2b0ae357 2022-01-10 thomas /* Copy 'ccccc...' from base file. */
375 2b0ae357 2022-01-10 thomas if (!(deltas[2].copy == 1 &&
376 2b0ae357 2022-01-10 thomas deltas[2].offset == 2 * GOT_DELTIFY_MAXCHUNK &&
377 2b0ae357 2022-01-10 thomas deltas[2].len == GOT_DELTIFY_MAXCHUNK)) {
378 2b0ae357 2022-01-10 thomas err = got_error(GOT_ERR_BAD_DELTA);
379 2b0ae357 2022-01-10 thomas goto done;
380 2b0ae357 2022-01-10 thomas }
381 2b0ae357 2022-01-10 thomas
382 2b0ae357 2022-01-10 thomas done:
383 2b0ae357 2022-01-10 thomas got_deltify_free(dt);
384 2b0ae357 2022-01-10 thomas fclose(result_file);
385 2b0ae357 2022-01-10 thomas return (err == NULL);
386 2b0ae357 2022-01-10 thomas }
387 2b0ae357 2022-01-10 thomas
388 8704c7ce 2021-03-10 stsp static int quiet;
389 8704c7ce 2021-03-10 stsp
390 8704c7ce 2021-03-10 stsp #define RUN_TEST(expr, name) \
391 8704c7ce 2021-03-10 stsp { test_ok = (expr); \
392 8704c7ce 2021-03-10 stsp if (!quiet) printf("test_%s %s\n", (name), test_ok ? "ok" : "failed"); \
393 8704c7ce 2021-03-10 stsp failure = (failure || !test_ok); }
394 8704c7ce 2021-03-10 stsp
395 8704c7ce 2021-03-10 stsp static void
396 8704c7ce 2021-03-10 stsp usage(void)
397 8704c7ce 2021-03-10 stsp {
398 8704c7ce 2021-03-10 stsp fprintf(stderr, "usage: delta_test [-q]\n");
399 8704c7ce 2021-03-10 stsp }
400 8704c7ce 2021-03-10 stsp
401 8704c7ce 2021-03-10 stsp int
402 8704c7ce 2021-03-10 stsp main(int argc, char *argv[])
403 8704c7ce 2021-03-10 stsp {
404 8704c7ce 2021-03-10 stsp int test_ok;
405 8704c7ce 2021-03-10 stsp int failure = 0;
406 8704c7ce 2021-03-10 stsp int ch;
407 8704c7ce 2021-03-10 stsp
408 8704c7ce 2021-03-10 stsp while ((ch = getopt(argc, argv, "q")) != -1) {
409 8704c7ce 2021-03-10 stsp switch (ch) {
410 8704c7ce 2021-03-10 stsp case 'q':
411 8704c7ce 2021-03-10 stsp quiet = 1;
412 8704c7ce 2021-03-10 stsp break;
413 8704c7ce 2021-03-10 stsp default:
414 8704c7ce 2021-03-10 stsp usage();
415 8704c7ce 2021-03-10 stsp return 1;
416 8704c7ce 2021-03-10 stsp }
417 8704c7ce 2021-03-10 stsp }
418 8704c7ce 2021-03-10 stsp
419 8704c7ce 2021-03-10 stsp argc -= optind;
420 8704c7ce 2021-03-10 stsp argv += optind;
421 8704c7ce 2021-03-10 stsp
422 8704c7ce 2021-03-10 stsp if (argc != 0) {
423 8704c7ce 2021-03-10 stsp usage();
424 8704c7ce 2021-03-10 stsp return 1;
425 8704c7ce 2021-03-10 stsp }
426 8704c7ce 2021-03-10 stsp
427 8704c7ce 2021-03-10 stsp #ifndef PROFILE
428 8704c7ce 2021-03-10 stsp if (pledge("stdio rpath wpath cpath unveil", NULL) == -1)
429 8704c7ce 2021-03-10 stsp err(1, "pledge");
430 8704c7ce 2021-03-10 stsp #endif
431 8704c7ce 2021-03-10 stsp if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
432 8704c7ce 2021-03-10 stsp err(1, "unveil");
433 8704c7ce 2021-03-10 stsp
434 8704c7ce 2021-03-10 stsp if (unveil(NULL, NULL) != 0)
435 8704c7ce 2021-03-10 stsp err(1, "unveil");
436 8704c7ce 2021-03-10 stsp
437 8704c7ce 2021-03-10 stsp RUN_TEST(deltify_abc_axc(), "deltify_abc_axc");
438 2b0ae357 2022-01-10 thomas RUN_TEST(deltify_abc_axc_file_mem(), "deltify_abc_axc_file_mem");
439 2b0ae357 2022-01-10 thomas RUN_TEST(deltify_abc_axc_mem_file(), "deltify_abc_axc_mem_file");
440 2b0ae357 2022-01-10 thomas RUN_TEST(deltify_abc_axc_mem_mem(), "deltify_abc_axc_mem_mem");
441 8704c7ce 2021-03-10 stsp
442 8704c7ce 2021-03-10 stsp return failure ? 1 : 0;
443 8704c7ce 2021-03-10 stsp }