md: Fix hashing for data >= 256 GB
[libgcrypt.git] / tests / genhashdata.c
1 /* genhashdata.c - Create data for hash tests
2  * Copyright (C) 2013 g10 Code GmbH
3  *
4  * This file is part of Libgcrypt.
5  *
6  * Libgcrypt is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU Lesser General Public License as
8  * published by the Free Software Foundation; either version 2.1 of
9  * the License, or (at your option) any later version.
10  *
11  * Libgcrypt is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this program; if not, see <http://www.gnu.org/licenses/>.
18  */
19
20 /* Results:
21
22 $  for i in -64 -1 0 1 64; do ./genhashdata --gigs 256 --bytes $i|sha1sum;done
23 92fc51850c7b750e6e774b75f294f6979d4059f0  -
24 4bddeeb4c08683f02d4944d93dbcb02ebab50134  -
25 71b923afde1c8c040884c723a2e3335b333e64c6  -
26 2d99f9b5b86e9c9c937104f4242bd6b8bc0927ef  -
27 a60dabe8d749f798b7ec3a684cc3eab487451482  -
28
29 $ for i in -64 -1 0 1 64; do ./genhashdata --gigs 256 --bytes $i|sha224sum;done
30 b5672b54d2480a5688a2dc727a1ad4db7a81ef31ce8999e0bbaeffdc  -
31 814ea7159473e6ffc1c64b90026a542e13ac6980f7f3ca3c4582a9b8  -
32 9ec0e1829455db8650ec7a8b06912196f97a7358bc3a73c79911cd4e  -
33 e578d5d523320876565bbbc892511a485427caee6dd754d57e3e58c2  -
34 ff0464df248cd298b63765bc4f87f21e25c93c657fdf3656d3c878e5  -
35
36 $ for i in -64 -1 0 1 64; do ./genhashdata --gigs 256 --bytes $i|sha256sum;done
37 87a9828d3de78d55d252341db2a622908c4e0ceaee9961ecf9768700fc799ec8  -
38 823bf95f64ef04a4a77579c38760b1d401b56bf3a8e664bdf56ca15afb468a03  -
39 2d0723878cb2c3d5c59dfad910cdb857f4430a6ba2a7d687938d7a20e63dde47  -
40 5a2e21b1e79cd866acf53a2a18ca76bd4e02c4b01bf4627354171824c812d95f  -
41 34444808af8e9d995e67f9e155ed94bf55f195a51dc1d8a989e6bcf95511c8a2  -
42
43 */
44
45 #include <stdarg.h>
46 #include <stdio.h>
47 #include <stdlib.h>
48 #include <string.h>
49 #include <errno.h>
50
51 #define PGM "genhashdata"
52
53 static void
54 die (const char *format, ...)
55 {
56   va_list arg_ptr ;
57
58   fflush (stdout);
59   fprintf (stderr, "%s: ", PGM);
60   va_start (arg_ptr, format ) ;
61   vfprintf (stderr, format, arg_ptr );
62   va_end(arg_ptr);
63   if (*format && format[strlen(format)-1] != '\n')
64     putc ('\n', stderr);
65   exit (1);
66 }
67
68 int
69 main (int argc, char **argv)
70 {
71   int last_argc = -1;
72   int gigs = 0;
73   int bytes = 0;
74   char pattern[1024];
75   int i, g;
76
77   if (argc)
78     { argc--; argv++; }
79
80   while (argc && last_argc != argc )
81     {
82       last_argc = argc;
83       if (!strcmp (*argv, "--"))
84         {
85           argc--; argv++;
86           break;
87         }
88       else if (!strcmp (*argv, "--help"))
89         {
90           fputs ("usage: " PGM " [options]\n"
91                  "Options:\n"
92                  "  --gigs  N     Emit N GiB of test bytes\n"
93                  "  --bytes DIFF  Stop DIFF bytes earlier or later\n",
94                  stdout);
95           exit (0);
96         }
97       else if (!strcmp (*argv, "--gigs"))
98         {
99           argc--; argv++;
100           if (argc)
101             {
102               gigs = atoi (*argv);
103               argc--; argv++;
104             }
105         }
106       else if (!strcmp (*argv, "--bytes"))
107         {
108           argc--; argv++;
109           if (argc)
110             {
111               bytes = atoi (*argv);
112               argc--; argv++;
113             }
114         }
115       else if (!strncmp (*argv, "--", 2))
116         die ("unknown option '%s'", *argv);
117     }
118
119   if (gigs < 0 || gigs > 1024*1024)
120     die ("value for --gigs must be in the range 0 to %d", 1024*1024);
121   if (bytes < -1024 || bytes > 1024)
122       die ("value for --bytes must be in the range -1024 to 1024");
123   if (sizeof pattern != 1024)
124     die ("internal error");
125
126   if (argc > 1)
127     die ("arguments are not expected");
128
129   memset (pattern, 'a', sizeof pattern);
130
131   for (g=0; g < gigs; g++)
132     {
133       if (g + 1 == gigs && bytes < 0)
134         {
135           for (i = 0; i < 1024*1023; i++)
136             if (fwrite (pattern, sizeof pattern, 1, stdout) != 1)
137               die ("writing to stdout failed: %s", strerror (errno));
138           for (i = 0; i < 1023; i++)
139             if (fwrite (pattern, sizeof pattern, 1, stdout) != 1)
140               die ("writing to stdout failed: %s", strerror (errno));
141           if (fwrite (pattern, sizeof pattern + bytes, 1, stdout) != 1)
142             die ("writing to stdout failed: %s", strerror (errno));
143         }
144       else
145         {
146           for (i = 0; i < 1024*1024; i++)
147             if (fwrite (pattern, sizeof pattern, 1, stdout) != 1)
148               die ("writing to stdout failed: %s", strerror (errno));
149         }
150     }
151   if (bytes > 0)
152     if (fwrite (pattern, bytes, 1, stdout) != 1)
153       die ("writing to stdout failed: %s", strerror (errno));
154   if (fflush (stdout))
155     die ("writing to stdout failed: %s", strerror (errno));
156
157   return 0;
158 }