Move buffer size into util_common.h

Signed-off-by: Jonathan Dieter <jdieter@gmail.com>

diff --git a/src/unzck.c b/src/unzck.c
index d73ab00a445e33633871dfb798d39d91746539cc..9f30e3bb59efa30871c686b8946d1fb3bef7eb6c 100644 (file)
--- a/src/unzck.c
+++ b/src/unzck.c
@@ -37,8 +37,6 @@
 
 #include "util_common.h"
 
-#define BLK_SIZE 32768
-
 static char doc[] = "unzck - Decompress a zchunk file";
 
 static char args_doc[] = "<file>";
@@ -130,10 +128,10 @@ int main (int argc, char *argv[]) {
     if(zck == NULL)
         goto error1;
 
-    char *data = malloc(BLK_SIZE);
+    char *data = malloc(BUF_SIZE);
     size_t total = 0;
     while(True) {
-        ssize_t read = zck_read(zck, data, BLK_SIZE);
+        ssize_t read = zck_read(zck, data, BUF_SIZE);
         if(read < 0)
             goto error2;
         if(read == 0)

diff --git a/src/util_common.h b/src/util_common.h
index dd0103a3c615ce40d16b0de2072d8d1a52ead936..2f74cfcccaba51e30fa74c8d20804a2d8848c579 100644 (file)
--- a/src/util_common.h
+++ b/src/util_common.h
@@ -4,6 +4,10 @@
 #define ZCK_NAME "zchunk"
 #define ZCK_COPYRIGHT_YEAR "2018"
 
+#ifndef BUF_SIZE
+#define BUF_SIZE 32768
+#endif
+
 void version();
 
 #endif