test-tool genzeros: generate large amounts of data more efficiently

In this developer's tests, producing one gigabyte worth of NULs in a busy loop that writes out individual bytes, unbuffered, took ~27sec. Writing chunked 256kB buffers instead only took ~0.6sec This matters because we are about to introduce a pair of test cases that want to be able to produce 5GB of NULs, and we cannot use `/dev/zero` because of the HP NonStop platform's lack of support for that device. Signed-off-by: Johannes Schindelin <johannes.schindelin@gmx.de> Signed-off-by: Junio C Hamano <gitster@pobox.com>

Johannes Schindelin committed Nov 2, 2021 at 15:46 UTC df7000cd910a5f6991f73991b87d838a5d75e2fc
1 file changed +15 -2
t/helper/test-genzeros.c
+15 -2
@@ -3,7 +3,10 @@
3
4 int cmd__genzeros(int argc, const char **argv)
5 {
6 + /* static, so that it is NUL-initialized */
7 + static const char zeros[256 * 1024];
8 intmax_t count;
9 + ssize_t n;
10
11 if (argc > 2) {
12 fprintf(stderr, "usage: %s [<count>]\n", argv[0]);
@@ -12,9 +15,19 @@ int cmd__genzeros(int argc, const char **argv)
15
16 count = argc > 1 ? strtoimax(argv[1], NULL, 0) : -1;
17
15 - while (count < 0 || count--) {
16 - if (putchar(0) == EOF)
18 + /* Writing out individual NUL bytes is slow... */
19 + while (count < 0)
20 + if (write(1, zeros, ARRAY_SIZE(zeros)) < 0)
21 return -1;
22 +
23 + while (count > 0) {
24 + n = write(1, zeros, count < ARRAY_SIZE(zeros) ?
25 + count : ARRAY_SIZE(zeros));
26 +
27 + if (n < 0)
28 + return -1;
29 +
30 + count -= n;
31 }
32
33 return 0;