Merge branch 'ma/pkt-line-leakfix'
[git] / 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 "git-compat-util.h"
12 #include "delta.h"
13 #include "cache.h"
14
15 static const char usage_str[] =
16         "test-delta (-d|-p) <from_file> <data_file> <out_file>";
17
18 int cmd_main(int argc, const char **argv)
19 {
20         int fd;
21         struct stat st;
22         void *from_buf, *data_buf, *out_buf;
23         unsigned long from_size, data_size, out_size;
24
25         if (argc != 5 || (strcmp(argv[1], "-d") && strcmp(argv[1], "-p"))) {
26                 fprintf(stderr, "usage: %s\n", usage_str);
27                 return 1;
28         }
29
30         fd = open(argv[2], O_RDONLY);
31         if (fd < 0 || fstat(fd, &st)) {
32                 perror(argv[2]);
33                 return 1;
34         }
35         from_size = st.st_size;
36         from_buf = mmap(NULL, from_size, PROT_READ, MAP_PRIVATE, fd, 0);
37         if (from_buf == MAP_FAILED) {
38                 perror(argv[2]);
39                 close(fd);
40                 return 1;
41         }
42         close(fd);
43
44         fd = open(argv[3], O_RDONLY);
45         if (fd < 0 || fstat(fd, &st)) {
46                 perror(argv[3]);
47                 return 1;
48         }
49         data_size = st.st_size;
50         data_buf = mmap(NULL, data_size, PROT_READ, MAP_PRIVATE, fd, 0);
51         if (data_buf == MAP_FAILED) {
52                 perror(argv[3]);
53                 close(fd);
54                 return 1;
55         }
56         close(fd);
57
58         if (argv[1][1] == 'd')
59                 out_buf = diff_delta(from_buf, from_size,
60                                      data_buf, data_size,
61                                      &out_size, 0);
62         else
63                 out_buf = patch_delta(from_buf, from_size,
64                                       data_buf, data_size,
65                                       &out_size);
66         if (!out_buf) {
67                 fprintf(stderr, "delta operation failed (returned NULL)\n");
68                 return 1;
69         }
70
71         fd = open (argv[4], O_WRONLY|O_CREAT|O_TRUNC, 0666);
72         if (fd < 0 || write_in_full(fd, out_buf, out_size) != out_size) {
73                 perror(argv[4]);
74                 return 1;
75         }
76
77         return 0;
78 }