1 | #include <stdio.h>
|
---|
2 | #include <stdlib.h>
|
---|
3 | #include <math.h>
|
---|
4 |
|
---|
5 | #define BUFSZ (1 << 22)
|
---|
6 |
|
---|
7 | #ifdef WIN32
|
---|
8 | // whatever
|
---|
9 | static
|
---|
10 | double drand48() {
|
---|
11 | double r = rand() / (double)RAND_MAX;
|
---|
12 | return r;
|
---|
13 | }
|
---|
14 | long lrand48() {
|
---|
15 | long l = 0;
|
---|
16 | int i;
|
---|
17 | for (i = 0; i < 32; i++) {
|
---|
18 | l = l ^ (l << 2) ^ (l << 1) ^ rand();
|
---|
19 | }
|
---|
20 | return l;
|
---|
21 | }
|
---|
22 | #endif
|
---|
23 |
|
---|
24 | #ifdef _WIN32
|
---|
25 | #define XD3_WIN32 1
|
---|
26 | #else
|
---|
27 | #define XD3_POSIX 1
|
---|
28 | #endif
|
---|
29 | #define XD3_MAIN 1
|
---|
30 | #define main notmain
|
---|
31 | #define EXTERNAL_COMPRESSION 0
|
---|
32 | #define XD3_USE_LARGEFILE64 1
|
---|
33 | #include "xdelta3.c"
|
---|
34 | #undef main
|
---|
35 |
|
---|
36 |
|
---|
37 | double error_prob = 0.0001;
|
---|
38 | usize_t mean_change = 100;
|
---|
39 | xoff_t total_change = 0;
|
---|
40 | xoff_t total_size = 0;
|
---|
41 | usize_t max_change = 0;
|
---|
42 | usize_t num_change = 0;
|
---|
43 |
|
---|
44 |
|
---|
45 | static usize_t
|
---|
46 | edist (usize_t mean, usize_t max)
|
---|
47 | {
|
---|
48 | double mean_d = mean;
|
---|
49 | double erand = log (1.0 / drand48 ());
|
---|
50 | usize_t x = (usize_t) (mean_d * erand + 0.5);
|
---|
51 |
|
---|
52 | return (x < max) ? (x > 0 ? x : 1) : max;
|
---|
53 | }
|
---|
54 |
|
---|
55 | void modify (char *buf, usize_t size)
|
---|
56 | {
|
---|
57 | usize_t bufpos = 0, j;
|
---|
58 | usize_t last_end = 0;
|
---|
59 |
|
---|
60 | for (;; /* bufpos and j are incremented in the inner loop */)
|
---|
61 | {
|
---|
62 | /* The size of the next modification. */
|
---|
63 | usize_t next_size = edist (mean_change, 1 << 31);
|
---|
64 | /* The expected interval of such a change. */
|
---|
65 | double expect_interval = ((double) next_size * (1.0 - error_prob)) / error_prob;
|
---|
66 | /* The number of bytes until the next modification. */
|
---|
67 | usize_t next_mod = edist ((usize_t)expect_interval, 1 << 31);
|
---|
68 |
|
---|
69 | if (next_size + next_mod + bufpos > size) { break; }
|
---|
70 |
|
---|
71 | if (max_change < next_size) { max_change = next_size; }
|
---|
72 |
|
---|
73 | bufpos += next_mod;
|
---|
74 |
|
---|
75 | fprintf (stderr, "COPY: %I64u-%I64u (%u)\n",
|
---|
76 | total_size + (xoff_t)last_end,
|
---|
77 | total_size + (xoff_t)bufpos,
|
---|
78 | bufpos - last_end);
|
---|
79 | fprintf (stderr, "ADD: %I64u-%I64u (%u) is change %u\n",
|
---|
80 | total_size + (xoff_t)bufpos,
|
---|
81 | total_size + (xoff_t)(bufpos + next_size),
|
---|
82 | next_size, num_change);
|
---|
83 |
|
---|
84 | total_change += next_size;
|
---|
85 | num_change += 1;
|
---|
86 |
|
---|
87 | for (j = 0; j < next_size; j += 1, bufpos += 1)
|
---|
88 | {
|
---|
89 | buf[bufpos] = (char)(lrand48 () >> 3);
|
---|
90 | }
|
---|
91 |
|
---|
92 | last_end = bufpos;
|
---|
93 | }
|
---|
94 |
|
---|
95 | fprintf (stderr, "COPY: %I64u-%I64u (%u)\n",
|
---|
96 | total_size + last_end,
|
---|
97 | total_size + size, size - last_end);
|
---|
98 |
|
---|
99 | total_size += size;
|
---|
100 | }
|
---|
101 |
|
---|
102 | int main(int argc, char **argv)
|
---|
103 | {
|
---|
104 | main_file inp, out;
|
---|
105 | char *buf = malloc(BUFSZ);
|
---|
106 | int c, ret;
|
---|
107 | main_file_init(&inp);
|
---|
108 | main_file_init(&out);
|
---|
109 | option_force = 1;
|
---|
110 | if (argc > 5)
|
---|
111 | {
|
---|
112 | fprintf (stderr, "usage: badcopy [byte_error_prob [mean_error_size]]\n");
|
---|
113 | return 1;
|
---|
114 | }
|
---|
115 |
|
---|
116 | if (argc > 4) { mean_change = atoi (argv[4]); }
|
---|
117 | if (argc > 3) { error_prob = atof (argv[3]); }
|
---|
118 | fprintf (stderr, "mean change = %u; error_prob = %0.10f\n", mean_change, error_prob);
|
---|
119 |
|
---|
120 | if ((ret = main_file_open (&inp, argv[1], XO_READ)) != 0) {
|
---|
121 | return 1;
|
---|
122 | }
|
---|
123 | if ((ret = main_file_open (&out, argv[2], XO_WRITE)) != 0) {
|
---|
124 | return 1;
|
---|
125 | }
|
---|
126 |
|
---|
127 | if (error_prob < 0.0 || error_prob > 1.0)
|
---|
128 | {
|
---|
129 | fprintf (stderr, "warning: error probability out of range\n");
|
---|
130 | return 1;
|
---|
131 | }
|
---|
132 |
|
---|
133 | do
|
---|
134 | {
|
---|
135 | if ((ret = main_file_read (&inp, buf, BUFSZ, &c, "read failed")) != 0) {
|
---|
136 | return 1;
|
---|
137 | }
|
---|
138 |
|
---|
139 | if (c == 0) { break; }
|
---|
140 |
|
---|
141 | modify (buf, c);
|
---|
142 |
|
---|
143 | if ((ret = main_file_write (&out, buf, c, "write failed")) != 0) {
|
---|
144 | return 1;
|
---|
145 | }
|
---|
146 | }
|
---|
147 | while (c == BUFSZ);
|
---|
148 |
|
---|
149 | if ((ret = main_file_close (&out)))
|
---|
150 | {
|
---|
151 | return 1;
|
---|
152 | }
|
---|
153 |
|
---|
154 | fprintf (stderr, "add_prob %f; %u adds; total_change %u of %u bytes; add percentage %f; max add size %u\n",
|
---|
155 | error_prob, num_change, total_change, total_size, (double) total_change / (double) total_size, max_change);
|
---|
156 |
|
---|
157 | return 0;
|
---|
158 | }
|
---|