From 52793b733de2ce5176b61d332b31e47ecfb844f0 Mon Sep 17 00:00:00 2001 From: Pali Rohár Date: Mon, 30 Sep 2019 17:42:16 +0200 Subject: disk: Check if reading from block device failed --- src/disk.c | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/disk.c b/src/disk.c index f58b1dd..99fb7c8 100644 --- a/src/disk.c +++ b/src/disk.c @@ -270,6 +270,11 @@ int disk_dump_dev(int fd, const char * file) { size = read(fd, global_buf, need); if ( size == 0 ) break; + if ( size < 0 ) { + PRINTF_ERROR("Reading from block device failed"); + close(fd2); + return -1; + } if ( write(fd2, global_buf, size) != size ) { PRINTF_ERROR("Dumping image failed"); close(fd2); -- cgit v1.2.3