def remove(self):
if get_guid_bin(self.guid) == '':
print ('*** Error *** Invalid GUID: {}'.format(self.guid))
return
if not os.path.isfile(self.rom_file):
print ('*** Error *** File doesn\'t exist: {}'.format(self.rom_file))
return
rom_image = read_file(self.rom_file)
new_image = modify_uefi_region(rom_image, CMD_UEFI_FILE_REMOVE, self.guid)
write_file(self.new_file, new_image)
def insert_after(self):
if get_guid_bin(self.guid) == '':
print ('*** Error *** Invalid GUID: {}'.format(self.guid))
return
if not os.path.isfile(self.rom_file):
print ('*** Error *** File doesn\'t exist: {}'.format(self.rom_file))
return
if not os.path.isfile(self.efi_file):
print ('*** Error *** File doesn\'t exist: {}'.format(self.efi_file))
return
rom_image = read_file(self.rom_file)
efi_image = read_file(self.efi_file)
new_image = modify_uefi_region(rom_image, CMD_UEFI_FILE_INSERT_AFTER, self.guid, efi_image)
write_file(self.new_file, new_image)