From 8eab43e5ee1a48bd3a5a391b9546faa4469bad91 Mon Sep 17 00:00:00 2001 From: kris Date: Thu, 4 May 2017 21:31:55 +0100 Subject: [PATCH] Add check for disk length not aligned to sector boundary --- insert_disks.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/insert_disks.py b/insert_disks.py index 341a8d3..d88b17e 100644 --- a/insert_disks.py +++ b/insert_disks.py @@ -38,8 +38,15 @@ def main(): print "(%d/%d:%d%%) %s" % (idx+1, num_disks, (idx+1)*100/num_disks, f) disk = bytearray(open(f, 'r').read()) - if len(disk) < 140*1024: - print "Disk %s truncated (%d bytes)" % (len(disk)) + length = len(disk) + if length < 140*1024: + print "Disk %s truncated (%d bytes)" % length + continue + + if length % 256 != 0: + print "Disk length %d does not align to sector boundary" % length + continue + boot1 = disk[:256] sha1 = hashlib.sha1(disk).hexdigest()