badcopy.c

   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 }