Merge branch 'bw/protocol-v2' into maint
[git] / t / helper / test-sha1.c
1 #include "test-tool.h"
2 #include "cache.h"
3
4 int cmd__sha1(int ac, const char **av)
5 {
6         git_SHA_CTX ctx;
7         unsigned char sha1[20];
8         unsigned bufsz = 8192;
9         int binary = 0;
10         char *buffer;
11
12         if (ac == 2) {
13                 if (!strcmp(av[1], "-b"))
14                         binary = 1;
15                 else
16                         bufsz = strtoul(av[1], NULL, 10) * 1024 * 1024;
17         }
18
19         if (!bufsz)
20                 bufsz = 8192;
21
22         while ((buffer = malloc(bufsz)) == NULL) {
23                 fprintf(stderr, "bufsz %u is too big, halving...\n", bufsz);
24                 bufsz /= 2;
25                 if (bufsz < 1024)
26                         die("OOPS");
27         }
28
29         git_SHA1_Init(&ctx);
30
31         while (1) {
32                 ssize_t sz, this_sz;
33                 char *cp = buffer;
34                 unsigned room = bufsz;
35                 this_sz = 0;
36                 while (room) {
37                         sz = xread(0, cp, room);
38                         if (sz == 0)
39                                 break;
40                         if (sz < 0)
41                                 die_errno("test-sha1");
42                         this_sz += sz;
43                         cp += sz;
44                         room -= sz;
45                 }
46                 if (this_sz == 0)
47                         break;
48                 git_SHA1_Update(&ctx, buffer, this_sz);
49         }
50         git_SHA1_Final(sha1, &ctx);
51
52         if (binary)
53                 fwrite(sha1, 1, 20, stdout);
54         else
55                 puts(sha1_to_hex(sha1));
56         exit(0);
57 }