1 | /* Copyright (C) 2007 Josh MacDonald */ |
---|
2 | |
---|
3 | #include <stdio.h> |
---|
4 | |
---|
5 | #define PAGE_SIZE 4096 |
---|
6 | |
---|
7 | #define SPACE_MAX 131072 // how much memory per process |
---|
8 | #define OUTPUT_MAX 1024 // max size for output |
---|
9 | #define XD3_ALLOCSIZE 256 // internal size for various buffers |
---|
10 | #define IOPT_SIZE 128 // instruction buffer |
---|
11 | |
---|
12 | // SPACE_MAX of 32K is sufficient for most inputs with XD3_COMPLEVEL_1 |
---|
13 | // XD3_COMPLEVEL_9 requires about 4x more space than XD3_COMPLEVEL_1 |
---|
14 | |
---|
15 | #include "xdelta3.h" |
---|
16 | #include "xdelta3.c" |
---|
17 | |
---|
18 | typedef struct _context { |
---|
19 | uint8_t *buffer; |
---|
20 | int allocated; |
---|
21 | } context_t; |
---|
22 | |
---|
23 | static int max_allocated = 0; |
---|
24 | |
---|
25 | void* |
---|
26 | process_alloc (void* opaque, usize_t items, usize_t size) |
---|
27 | { |
---|
28 | context_t *ctx = (context_t*) opaque; |
---|
29 | usize_t t = items * size; |
---|
30 | void *ret; |
---|
31 | |
---|
32 | if (ctx->allocated + t > SPACE_MAX) |
---|
33 | { |
---|
34 | return NULL; |
---|
35 | } |
---|
36 | |
---|
37 | ret = ctx->buffer + ctx->allocated; |
---|
38 | ctx->allocated += t; |
---|
39 | return ret; |
---|
40 | } |
---|
41 | |
---|
42 | void |
---|
43 | process_free (void* opaque, void *ptr) |
---|
44 | { |
---|
45 | } |
---|
46 | |
---|
47 | int |
---|
48 | process_page (int is_encode, |
---|
49 | int (*func) (xd3_stream *), |
---|
50 | const uint8_t *input, |
---|
51 | usize_t input_size, |
---|
52 | const uint8_t *source, |
---|
53 | uint8_t *output, |
---|
54 | usize_t *output_size, |
---|
55 | usize_t output_size_max, |
---|
56 | int flags) { |
---|
57 | |
---|
58 | /* On my x86 this is 1072 of objects on the stack */ |
---|
59 | xd3_stream stream; |
---|
60 | xd3_config config; |
---|
61 | xd3_source src; |
---|
62 | context_t *ctx = calloc(SPACE_MAX, 1); |
---|
63 | int ret; |
---|
64 | |
---|
65 | if (ctx == NULL) |
---|
66 | { |
---|
67 | printf("calloc failed\n"); |
---|
68 | return -1; |
---|
69 | } |
---|
70 | |
---|
71 | ctx->buffer = (uint8_t*)ctx; |
---|
72 | ctx->allocated = sizeof(*ctx); |
---|
73 | |
---|
74 | config.flags = flags; |
---|
75 | config.winsize = PAGE_SIZE; |
---|
76 | config.sprevsz = PAGE_SIZE; |
---|
77 | config.srcwin_maxsz = PAGE_SIZE; |
---|
78 | config.iopt_size = IOPT_SIZE; |
---|
79 | config.alloc = &process_alloc; |
---|
80 | config.freef = &process_free; |
---|
81 | config.opaque = (void*) ctx; |
---|
82 | |
---|
83 | src.size = PAGE_SIZE; |
---|
84 | src.blksize = PAGE_SIZE; |
---|
85 | src.onblk = PAGE_SIZE; |
---|
86 | src.curblk = source; |
---|
87 | src.curblkno = 0; |
---|
88 | |
---|
89 | if ((ret = xd3_config_stream (&stream, &config)) != 0 || |
---|
90 | (ret = xd3_set_source (&stream, &src)) != 0 || |
---|
91 | (ret = xd3_process_stream (is_encode, |
---|
92 | &stream, |
---|
93 | func, 1, |
---|
94 | input, input_size, |
---|
95 | output, output_size, |
---|
96 | output_size_max)) != 0) |
---|
97 | { |
---|
98 | if (stream.msg != NULL) |
---|
99 | { |
---|
100 | fprintf(stderr, "stream message: %s\n", stream.msg); |
---|
101 | } |
---|
102 | } |
---|
103 | |
---|
104 | xd3_free_stream (&stream); |
---|
105 | if (max_allocated < ctx->allocated) |
---|
106 | { |
---|
107 | max_allocated = ctx->allocated; |
---|
108 | fprintf(stderr, "max allocated %d\n", max_allocated); |
---|
109 | } |
---|
110 | |
---|
111 | free(ctx); |
---|
112 | return ret; |
---|
113 | } |
---|
114 | |
---|
115 | int test(int stride, int encode_flags) |
---|
116 | { |
---|
117 | uint8_t frompg[PAGE_SIZE]; |
---|
118 | uint8_t topg[PAGE_SIZE]; |
---|
119 | uint8_t output[OUTPUT_MAX]; |
---|
120 | uint8_t reout[PAGE_SIZE]; |
---|
121 | usize_t output_size; |
---|
122 | usize_t re_size; |
---|
123 | int i, j, ret; |
---|
124 | |
---|
125 | for (i = 0; i < PAGE_SIZE; i++) |
---|
126 | { |
---|
127 | topg[i] = frompg[i] = (rand() >> 3 ^ rand() >> 6 ^ rand() >> 9); |
---|
128 | } |
---|
129 | |
---|
130 | // change 1 byte every stride |
---|
131 | if (stride > 0) |
---|
132 | { |
---|
133 | for (j = stride; j <= PAGE_SIZE; j += stride) |
---|
134 | { |
---|
135 | topg[j - 1] ^= 0xff; |
---|
136 | } |
---|
137 | } |
---|
138 | |
---|
139 | if ((ret = process_page (1, xd3_encode_input, |
---|
140 | topg, PAGE_SIZE, |
---|
141 | frompg, output, |
---|
142 | &output_size, OUTPUT_MAX, |
---|
143 | encode_flags)) != 0) |
---|
144 | { |
---|
145 | fprintf (stderr, "encode failed: stride %u flags 0x%x\n", stride, encode_flags); |
---|
146 | return ret; |
---|
147 | } |
---|
148 | |
---|
149 | if ((ret = process_page (0, xd3_decode_input, |
---|
150 | output, output_size, |
---|
151 | frompg, reout, |
---|
152 | &re_size, PAGE_SIZE, |
---|
153 | 0)) != 0) |
---|
154 | { |
---|
155 | fprintf (stderr, "decode failed: stride %u output_size %u flags 0x%x\n", |
---|
156 | stride, output_size, encode_flags); |
---|
157 | return ret; |
---|
158 | } |
---|
159 | |
---|
160 | if (output_size > OUTPUT_MAX || re_size != PAGE_SIZE) |
---|
161 | { |
---|
162 | fprintf (stderr, "internal error: %u != %u\n", output_size, re_size); |
---|
163 | return -1; |
---|
164 | } |
---|
165 | |
---|
166 | for (i = 0; i < PAGE_SIZE; i++) |
---|
167 | { |
---|
168 | if (reout[i] != topg[i]) |
---|
169 | { |
---|
170 | fprintf (stderr, "encode-decode error: position %d\n", i); |
---|
171 | return -1; |
---|
172 | } |
---|
173 | } |
---|
174 | |
---|
175 | fprintf(stderr, "stride %d flags 0x%x size %u ", stride, encode_flags, output_size); |
---|
176 | fprintf(stderr, "%s\n", (ret == 0) ? "OK" : "FAIL"); |
---|
177 | |
---|
178 | return 0; |
---|
179 | } |
---|
180 | |
---|
181 | int main() |
---|
182 | { |
---|
183 | int stride; |
---|
184 | int level; |
---|
185 | |
---|
186 | for (level = 1; level < 10; level = (level == 1 ? 3 : level + 3)) |
---|
187 | { |
---|
188 | int lflag = level << XD3_COMPLEVEL_SHIFT; |
---|
189 | |
---|
190 | for (stride = 2; stride <= PAGE_SIZE; stride += 2) |
---|
191 | { |
---|
192 | test(stride, lflag); |
---|
193 | test(stride, lflag | XD3_SEC_DJW); |
---|
194 | } |
---|
195 | } |
---|
196 | |
---|
197 | return 0; |
---|
198 | } |
---|