Andrew's git - gitweb.git/blob

cc05794ec0b69aa2a05f7d02a412e18c87687047
   1/*
   2 * test-delta.c: test code to exercise diff-delta.c and patch-delta.c
   3 *
   4 * (C) 2005 Nicolas Pitre <nico@cam.org>
   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 <stdio.h>
  12#include <unistd.h>
  13#include <string.h>
  14#include <fcntl.h>
  15#include <sys/types.h>
  16#include <sys/stat.h>
  17#include <sys/mman.h>
  18#include "delta.h"
  19
  20static const char usage[] =
  21        "test-delta (-d|-p) <from_file> <data_file> <out_file>";
  22
  23int main(int argc, char *argv[])
  24{
  25        int fd;
  26        struct stat st;
  27        void *from_buf, *data_buf, *out_buf;
  28        unsigned long from_size, data_size, out_size;
  29
  30        if (argc != 5 || (strcmp(argv[1], "-d") && strcmp(argv[1], "-p"))) {
  31                fprintf(stderr, "Usage: %s\n", usage);
  32                return 1;
  33        }
  34
  35        fd = open(argv[2], O_RDONLY);
  36        if (fd < 0 || fstat(fd, &st)) {
  37                perror(argv[2]);
  38                return 1;
  39        }
  40        from_size = st.st_size;
  41        if (from_size)
  42                from_buf = mmap(NULL, from_size, PROT_READ, MAP_PRIVATE, fd, 0);
  43        else
  44                from_buf = "";
  45        if (from_buf == MAP_FAILED) {
  46                perror(argv[2]);
  47                close(fd);
  48                return 1;
  49        }
  50        close(fd);
  51
  52        fd = open(argv[3], O_RDONLY);
  53        if (fd < 0 || fstat(fd, &st)) {
  54                perror(argv[3]);
  55                return 1;
  56        }
  57        data_size = st.st_size;
  58
  59        if (data_size)
  60                data_buf = mmap(NULL, data_size, PROT_READ, MAP_PRIVATE, fd, 0);
  61        else
  62                data_buf = "";
  63        if (data_buf == MAP_FAILED) {
  64                perror(argv[3]);
  65                close(fd);
  66                return 1;
  67        }
  68        close(fd);
  69
  70        if (argv[1][1] == 'd')
  71                out_buf = diff_delta(from_buf, from_size,
  72                                     data_buf, data_size,
  73                                     &out_size, 0);
  74        else
  75                out_buf = patch_delta(from_buf, from_size,
  76                                      data_buf, data_size,
  77                                      &out_size);
  78        if (!out_buf) {
  79                fprintf(stderr, "delta operation failed (returned NULL)\n");
  80                return 1;
  81        }
  82
  83        fd = open (argv[4], O_WRONLY|O_CREAT|O_TRUNC, 0666);
  84        if (fd < 0 || write(fd, out_buf, out_size) != out_size) {
  85                perror(argv[4]);
  86                return 1;
  87        }
  88
  89        return 0;
  90}