From f676ce5eb2eda78ed00d4f792963a679b16834ae Mon Sep 17 00:00:00 2001 From: Ayoub Zaki Date: Thu, 19 Dec 2024 18:50:14 +0100 Subject: [PATCH] check bmap ranges not empty --- bmap-writer.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/bmap-writer.cpp b/bmap-writer.cpp index cbf8ec9..2912272 100644 --- a/bmap-writer.cpp +++ b/bmap-writer.cpp @@ -55,7 +55,6 @@ bmap_t parseBMap(const std::string &filename) { xmlDocPtr doc = xmlReadFile(filename.c_str(), NULL, 0); if (doc == NULL) { - std::cerr << "Failed to parse " << filename << std::endl; return bmapData; } @@ -283,8 +282,8 @@ int main(int argc, const char *argv[]) { } auto start = std::chrono::high_resolution_clock::now(); bmap_t bmap = parseBMap(bmapFile); - if (bmap.blockSize == 0) { - std::cerr << "BlockSize not found in BMAP file" << std::endl; + if (bmap.blockSize == 0 || bmap.ranges.empty()) { + std::cerr << "Failed to parse file: " << bmapFile << std::endl; return 1; } int ret = BmapWriteImage(imageFile, bmap, device);