Fiemap: more input arguments validation

Validate the input argument of 'block_is_mapped()'.

Change-Id: I31ec0e4b5ff6250ed80cf4b2ea1a095ecf517927
Signed-off-by: Artem Bityutskiy <artem.bityutskiy@intel.com>

diff --git a/bmaptools/Fiemap.py b/bmaptools/Fiemap.py
index 09821e0..f9671d5 100644 (file)
--- a/bmaptools/Fiemap.py
+++ b/bmaptools/Fiemap.py
@@ -94,6 +94,10 @@ class Fiemap:
         """ This function returns 'True' if block number 'block' of the image
         file is mapped and 'False' otherwise. """
 
+        if block < 0 or block >= self.blocks_cnt:
+            raise Error("bad block number %d, should be within [0, %d]" \
+                        % (block, self.blocks_cnt))
+
         # Prepare a 'struct fiemap' buffer which contains a single
         # 'struct fiemap_extent' element.
         buf = struct.pack(_FIEMAP_FORMAT, block * self.block_size,