Make sure we hand invalid reads in zck tool

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

diff --git a/src/zck.c b/src/zck.c
index bf58eedf678a178f557566a292dbf8847992f502..4edecb165f298776df3b445355db664095b6ce46 100644 (file)
--- a/src/zck.c
+++ b/src/zck.c
@@ -127,7 +127,10 @@ int main (int argc, char *argv[]) {
     }
     if(in_size > 0) {
         data = malloc(in_size);
-        read(in_fd, data, in_size);
+        if(read(in_fd, data, in_size) < in_size) {
+            printf("Unable to read from input file\n");
+            exit(1);
+        }
         close(in_fd);
 
         /* Chunk based on string in argv[2] (Currently with ugly hack to group srpms together) */