read_write_all: use calloc instead of huge stack var in tests.

[ccan] / ccan / read_write_all / test / run-write_all.c

diff --git a/ccan/read_write_all/test/run-write_all.c b/ccan/read_write_all/test/run-write_all.c
index ab21cd31575f72c56d0ae7df658f5d04acdf7fdc..5b310688aea6283ec178515c92ed81d2e4474a18 100644 (file)
--- a/ccan/read_write_all/test/run-write_all.c
+++ b/ccan/read_write_all/test/run-write_all.c
@@ -1,8 +1,8 @@
 /* FIXME: Do something tricky to ensure we really do loop in write_all. */
 
-#include "read_write_all/read_write_all.h"
-#include "read_write_all/read_write_all.c"
-#include "tap/tap.h"
+#include <ccan/read_write_all/read_write_all.h>
+#include <ccan/read_write_all/read_write_all.c>
+#include <ccan/tap/tap.h>
 #include <unistd.h>
 #include <sys/types.h>
 #include <signal.h>
@@ -24,11 +24,12 @@ static void got_signal(int sig)
 
 int main(int argc, char *argv[])
 {
-       char buffer[BUFSZ*2] = { 0 };
+       char *buffer;
        int p2c[2];
        int status;
        pid_t child;
 
+       buffer = calloc(BUFSZ, 1);
        plan_tests(4);
 
        /* We fork and torture parent. */
@@ -43,11 +44,11 @@ int main(int argc, char *argv[])
                        exit(1);
                if (kill(getppid(), SIGUSR1) != 0)
                        exit(2);
-               if (!read_all(p2c[0], buffer+1, sizeof(buffer)-1))
+               if (!read_all(p2c[0], buffer+1, BUFSZ-1))
                        exit(3);
-               if (memchr(buffer, 0, sizeof(buffer))) {
+               if (memchr(buffer, 0, BUFSZ)) {
                        fprintf(stderr, "buffer has 0 at offset %ti\n",
-                               memchr(buffer, 0, sizeof(buffer)) - (void *)buffer);
+                               memchr(buffer, 0, BUFSZ) - (void *)buffer);
                        exit(4);
                }
                exit(0);
@@ -56,9 +57,9 @@ int main(int argc, char *argv[])
                err(1, "forking");
 
        close(p2c[0]);
-       memset(buffer, 0xff, sizeof(buffer));
+       memset(buffer, 0xff, BUFSZ);
        signal(SIGUSR1, got_signal);
-       ok1(write_all(p2c[1], buffer, sizeof(buffer)));
+       ok1(write_all(p2c[1], buffer, BUFSZ));
        ok1(sigcount == 1);
        ok1(wait(&status) == child);
        ok(WIFEXITED(status) && WEXITSTATUS(status) == 0,