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 8704c7ce 2021-03-10 stsp 
 44 8704c7ce 2021-03-10 stsp 	base_file = got_opentemp();
 45 8704c7ce 2021-03-10 stsp 	if (base_file == NULL)
 46 8704c7ce 2021-03-10 stsp 		return 1;
 47 8704c7ce 2021-03-10 stsp 
 48 8704c7ce 2021-03-10 stsp 	derived_file = got_opentemp();
 49 8704c7ce 2021-03-10 stsp 	if (derived_file == NULL)
 50 8704c7ce 2021-03-10 stsp 		return 1;
 51 8704c7ce 2021-03-10 stsp 
 52 8704c7ce 2021-03-10 stsp 	result_file = got_opentemp();
 53 8704c7ce 2021-03-10 stsp 	if (result_file == NULL)
 54 8704c7ce 2021-03-10 stsp 		return 1;
 55 8704c7ce 2021-03-10 stsp 
 56 8704c7ce 2021-03-10 stsp 	for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
 57 8704c7ce 2021-03-10 stsp 		fputc('a', base_file);
 58 8704c7ce 2021-03-10 stsp 		fputc('a', derived_file);
 59 8704c7ce 2021-03-10 stsp 	}
 60 8704c7ce 2021-03-10 stsp 	for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
 61 8704c7ce 2021-03-10 stsp 		fputc('b', base_file);
 62 8704c7ce 2021-03-10 stsp 		fputc('x', derived_file);
 63 8704c7ce 2021-03-10 stsp 	}
 64 8704c7ce 2021-03-10 stsp 	for (i = 0; i < GOT_DELTIFY_MAXCHUNK; i++) {
 65 8704c7ce 2021-03-10 stsp 		fputc('c', base_file);
 66 8704c7ce 2021-03-10 stsp 		fputc('c', derived_file);
 67 8704c7ce 2021-03-10 stsp 	}
 68 8704c7ce 2021-03-10 stsp 
 69 8704c7ce 2021-03-10 stsp 	rewind(base_file);
 70 8704c7ce 2021-03-10 stsp 	rewind(derived_file);
 71 8704c7ce 2021-03-10 stsp 
 72 8704c7ce 2021-03-10 stsp 	err = got_deltify_init(&dt, base_file, 0, 3 * GOT_DELTIFY_MAXCHUNK);
 73 8704c7ce 2021-03-10 stsp 	if (err)
 74 8704c7ce 2021-03-10 stsp 		goto done;
 75 8704c7ce 2021-03-10 stsp 
 76 8704c7ce 2021-03-10 stsp 	for (i = 0; i < dt->nalloc; i++) {
 77 8704c7ce 2021-03-10 stsp 		if (dt->blocks[i].len > 0)
 78 8704c7ce 2021-03-10 stsp 			have_nblocks++;
 79 8704c7ce 2021-03-10 stsp 	}
 80 8704c7ce 2021-03-10 stsp 	if (have_nblocks != dt->nblocks) {
 81 8704c7ce 2021-03-10 stsp 		err = got_error(GOT_ERR_BAD_DELTA);
 82 8704c7ce 2021-03-10 stsp 		goto done;
 83 8704c7ce 2021-03-10 stsp 	}
 84 8704c7ce 2021-03-10 stsp 
 85 8704c7ce 2021-03-10 stsp 	err = got_deltify(&deltas, &ndeltas, derived_file, 0,
 86 f34b169e 2021-06-18 stsp 	    3 * GOT_DELTIFY_MAXCHUNK, dt, base_file, 0,
 87 f34b169e 2021-06-18 stsp 	    3 * GOT_DELTIFY_MAXCHUNK);
 88 8704c7ce 2021-03-10 stsp 	if (err)
 89 8704c7ce 2021-03-10 stsp 		goto done;
 90 8704c7ce 2021-03-10 stsp 
 91 8704c7ce 2021-03-10 stsp 	if (ndeltas != 3) {
 92 8704c7ce 2021-03-10 stsp 		err = got_error(GOT_ERR_BAD_DELTA);
 93 8704c7ce 2021-03-10 stsp 		goto done;
 94 8704c7ce 2021-03-10 stsp 	}
 95 8704c7ce 2021-03-10 stsp 	/* Copy 'aaaa...' from base file. */
 96 8704c7ce 2021-03-10 stsp 	if (!(deltas[0].copy == 1 && deltas[0].offset == 0 &&
 97 8704c7ce 2021-03-10 stsp 	    deltas[0].len == GOT_DELTIFY_MAXCHUNK)) {
 98 8704c7ce 2021-03-10 stsp 		err = got_error(GOT_ERR_BAD_DELTA);
 99 8704c7ce 2021-03-10 stsp 		goto done;
100 8704c7ce 2021-03-10 stsp 	}
101 8704c7ce 2021-03-10 stsp 	/* Copy 'xxxx...' from derived file. */
102 8704c7ce 2021-03-10 stsp 	if (!(deltas[1].copy == 0 && deltas[1].offset == GOT_DELTIFY_MAXCHUNK &&
103 8704c7ce 2021-03-10 stsp 	    deltas[1].len == GOT_DELTIFY_MAXCHUNK)) {
104 8704c7ce 2021-03-10 stsp 		err = got_error(GOT_ERR_BAD_DELTA);
105 8704c7ce 2021-03-10 stsp 		goto done;
106 8704c7ce 2021-03-10 stsp 	}
107 8704c7ce 2021-03-10 stsp 	/* Copy 'ccccc...' from base file. */
108 8704c7ce 2021-03-10 stsp 	if (!(deltas[2].copy == 1 &&
109 8704c7ce 2021-03-10 stsp 	    deltas[2].offset == 2 * GOT_DELTIFY_MAXCHUNK &&
110 8704c7ce 2021-03-10 stsp 	    deltas[2].len == GOT_DELTIFY_MAXCHUNK)) {
111 8704c7ce 2021-03-10 stsp 		err = got_error(GOT_ERR_BAD_DELTA);
112 8704c7ce 2021-03-10 stsp 		goto done;
113 8704c7ce 2021-03-10 stsp 	}
114 8704c7ce 2021-03-10 stsp 
115 8704c7ce 2021-03-10 stsp done:
116 8704c7ce 2021-03-10 stsp 	got_deltify_free(dt);
117 8704c7ce 2021-03-10 stsp 	fclose(base_file);
118 8704c7ce 2021-03-10 stsp 	fclose(derived_file);
119 8704c7ce 2021-03-10 stsp 	fclose(result_file);
120 8704c7ce 2021-03-10 stsp 	return (err == NULL);
121 8704c7ce 2021-03-10 stsp }
122 8704c7ce 2021-03-10 stsp 
123 8704c7ce 2021-03-10 stsp static int quiet;
124 8704c7ce 2021-03-10 stsp 
125 8704c7ce 2021-03-10 stsp #define RUN_TEST(expr, name) \
126 8704c7ce 2021-03-10 stsp 	{ test_ok = (expr);  \
127 8704c7ce 2021-03-10 stsp 	if (!quiet) printf("test_%s %s\n", (name), test_ok ? "ok" : "failed"); \
128 8704c7ce 2021-03-10 stsp 	failure = (failure || !test_ok); }
129 8704c7ce 2021-03-10 stsp 
130 8704c7ce 2021-03-10 stsp static void
131 8704c7ce 2021-03-10 stsp usage(void)
132 8704c7ce 2021-03-10 stsp {
133 8704c7ce 2021-03-10 stsp 	fprintf(stderr, "usage: delta_test [-q]\n");
134 8704c7ce 2021-03-10 stsp }
135 8704c7ce 2021-03-10 stsp 
136 8704c7ce 2021-03-10 stsp int
137 8704c7ce 2021-03-10 stsp main(int argc, char *argv[])
138 8704c7ce 2021-03-10 stsp {
139 8704c7ce 2021-03-10 stsp 	int test_ok;
140 8704c7ce 2021-03-10 stsp 	int failure = 0;
141 8704c7ce 2021-03-10 stsp 	int ch;
142 8704c7ce 2021-03-10 stsp 
143 8704c7ce 2021-03-10 stsp 	while ((ch = getopt(argc, argv, "q")) != -1) {
144 8704c7ce 2021-03-10 stsp 		switch (ch) {
145 8704c7ce 2021-03-10 stsp 		case 'q':
146 8704c7ce 2021-03-10 stsp 			quiet = 1;
147 8704c7ce 2021-03-10 stsp 			break;
148 8704c7ce 2021-03-10 stsp 		default:
149 8704c7ce 2021-03-10 stsp 			usage();
150 8704c7ce 2021-03-10 stsp 			return 1;
151 8704c7ce 2021-03-10 stsp 		}
152 8704c7ce 2021-03-10 stsp 	}
153 8704c7ce 2021-03-10 stsp 
154 8704c7ce 2021-03-10 stsp 	argc -= optind;
155 8704c7ce 2021-03-10 stsp 	argv += optind;
156 8704c7ce 2021-03-10 stsp 
157 8704c7ce 2021-03-10 stsp 	if (argc != 0) {
158 8704c7ce 2021-03-10 stsp 		usage();
159 8704c7ce 2021-03-10 stsp 		return 1;
160 8704c7ce 2021-03-10 stsp 	}
161 8704c7ce 2021-03-10 stsp 
162 8704c7ce 2021-03-10 stsp #ifndef PROFILE
163 8704c7ce 2021-03-10 stsp 	if (pledge("stdio rpath wpath cpath unveil", NULL) == -1)
164 8704c7ce 2021-03-10 stsp 		err(1, "pledge");
165 8704c7ce 2021-03-10 stsp #endif
166 8704c7ce 2021-03-10 stsp 	if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
167 8704c7ce 2021-03-10 stsp 		err(1, "unveil");
168 8704c7ce 2021-03-10 stsp 
169 8704c7ce 2021-03-10 stsp 	if (unveil(NULL, NULL) != 0)
170 8704c7ce 2021-03-10 stsp 		err(1, "unveil");
171 8704c7ce 2021-03-10 stsp 
172 8704c7ce 2021-03-10 stsp 	RUN_TEST(deltify_abc_axc(), "deltify_abc_axc");
173 8704c7ce 2021-03-10 stsp 
174 8704c7ce 2021-03-10 stsp 	return failure ? 1 : 0;
175 8704c7ce 2021-03-10 stsp }