t/helper/test-delta.c

   1 /*
   2  * test-delta.c: test code to exercise diff-delta.c and patch-delta.c
   3  *
   4  * (C) 2005 Nicolas Pitre <nico@fluxnic.net>
   5  *
   6  * This code is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 as
   8  * published by the Free Software Foundation.
   9  */
  10
  11 #include "test-tool.h"
  12 #include "git-compat-util.h"
  13 #include "delta.h"
  14 #include "cache.h"
  15
  16 static const char usage_str[] =
  17         "test-tool delta (-d|-p) <from_file> <data_file> <out_file>";
  18
  19 int cmd__delta(int argc, const char **argv)
  20 {
  21         int fd;
  22         struct stat st;
  23         void *from_buf, *data_buf, *out_buf;
  24         unsigned long from_size, data_size, out_size;
  25
  26         if (argc != 5 || (strcmp(argv[1], "-d") && strcmp(argv[1], "-p"))) {
  27                 fprintf(stderr, "usage: %s\n", usage_str);
  28                 return 1;
  29         }
  30
  31         fd = open(argv[2], O_RDONLY);
  32         if (fd < 0 || fstat(fd, &st)) {
  33                 perror(argv[2]);
  34                 return 1;
  35         }
  36         from_size = st.st_size;
  37         from_buf = mmap(NULL, from_size, PROT_READ, MAP_PRIVATE, fd, 0);
  38         if (from_buf == MAP_FAILED) {
  39                 perror(argv[2]);
  40                 close(fd);
  41                 return 1;
  42         }
  43         close(fd);
  44
  45         fd = open(argv[3], O_RDONLY);
  46         if (fd < 0 || fstat(fd, &st)) {
  47                 perror(argv[3]);
  48                 return 1;
  49         }
  50         data_size = st.st_size;
  51         data_buf = mmap(NULL, data_size, PROT_READ, MAP_PRIVATE, fd, 0);
  52         if (data_buf == MAP_FAILED) {
  53                 perror(argv[3]);
  54                 close(fd);
  55                 return 1;
  56         }
  57         close(fd);
  58
  59         if (argv[1][1] == 'd')
  60                 out_buf = diff_delta(from_buf, from_size,
  61                                      data_buf, data_size,
  62                                      &out_size, 0);
  63         else
  64                 out_buf = patch_delta(from_buf, from_size,
  65                                       data_buf, data_size,
  66                                       &out_size);
  67         if (!out_buf) {
  68                 fprintf(stderr, "delta operation failed (returned NULL)\n");
  69                 return 1;
  70         }
  71
  72         fd = open (argv[4], O_WRONLY|O_CREAT|O_TRUNC, 0666);
  73         if (fd < 0 || write_in_full(fd, out_buf, out_size) < 0) {
  74                 perror(argv[4]);
  75                 return 1;
  76         }
  77
  78         return 0;
  79 }