+def check_boot_flag(device):
+ if device[-1:].isdigit():
+ boot_dev = re.match(r'(.*?)\d*$', device).group(1)
+ else:
+ boot_dev = device
+
+ with open(boot_dev, 'r') as image:
+ data = image.read(512)
+ bootcode = data[440:]
+ if bootcode[6] == '\x80':
+ logging.debug("bootflag is enabled")
+ else:
+ logging.debug("bootflag is NOT enabled")
+ raise VerifyException("Device %s does not have the bootflag set. "
+ "Please enable it to be able to boot." % boot_dev)
+
+