|
 |
003ee8 |
From 0e70861c9f72fb2e06f9a121e6dac0e542ee1230 Mon Sep 17 00:00:00 2001
|
|
|
003ee8 |
From: "Brian C. Lane" <bcl@redhat.com>
|
|
|
003ee8 |
Date: Wed, 27 Jun 2018 13:45:09 -0700
|
|
|
003ee8 |
Subject: [PATCH 82/86] Modify gpt-header-move and msdos-overlap to work with
|
|
|
003ee8 |
py2 or py3
|
|
|
003ee8 |
|
|
|
003ee8 |
Distributions are starting to remove python2 and only use python3.
|
|
|
003ee8 |
Modify these test scripts so that they will work with either python 2.7
|
|
|
003ee8 |
or python 3.X
|
|
|
003ee8 |
---
|
|
|
003ee8 |
tests/gpt-header-move | 15 ++++++++-------
|
|
|
003ee8 |
tests/msdos-overlap | 5 ++---
|
|
|
003ee8 |
2 files changed, 10 insertions(+), 10 deletions(-)
|
|
|
003ee8 |
|
|
|
003ee8 |
diff --git a/tests/gpt-header-move b/tests/gpt-header-move
|
|
|
003ee8 |
index 05cdc65..3dda5cb 100755
|
|
|
003ee8 |
--- a/tests/gpt-header-move
|
|
|
003ee8 |
+++ b/tests/gpt-header-move
|
|
|
003ee8 |
@@ -3,20 +3,21 @@
|
|
|
003ee8 |
# open img file, subtract 33 from altlba address, and move the last 33 sectors
|
|
|
003ee8 |
# back by 33 sectors
|
|
|
003ee8 |
|
|
|
003ee8 |
-from struct import *
|
|
|
003ee8 |
+from struct import unpack_from, pack_into
|
|
|
003ee8 |
from zipfile import crc32
|
|
|
003ee8 |
import array
|
|
|
003ee8 |
import sys
|
|
|
003ee8 |
+
|
|
|
003ee8 |
file = open(sys.argv[1],'rb+')
|
|
|
003ee8 |
file.seek(512)
|
|
|
003ee8 |
gptheader = file.read(512)
|
|
|
003ee8 |
-altlba = unpack_from('
|
|
|
003ee8 |
-gptheader = array.array('c',gptheader)
|
|
|
003ee8 |
+altlba = unpack_from('
|
|
|
003ee8 |
+gptheader = array.array('B', gptheader)
|
|
|
003ee8 |
pack_into('
|
|
|
003ee8 |
#zero header crc
|
|
|
003ee8 |
pack_into('
|
|
|
003ee8 |
#compute new crc
|
|
|
003ee8 |
-newcrc = ((crc32(buffer(gptheader,0,92))) & 0xFFFFFFFF)
|
|
|
003ee8 |
+newcrc = ((crc32(gptheader[:92])) & 0xFFFFFFFF)
|
|
|
003ee8 |
pack_into('
|
|
|
003ee8 |
file.seek(512)
|
|
|
003ee8 |
file.write(gptheader)
|
|
|
003ee8 |
@@ -25,7 +26,7 @@ gptheader = file.read(512)
|
|
|
003ee8 |
file.seek(512*(altlba-32))
|
|
|
003ee8 |
backup = file.read(512*32)
|
|
|
003ee8 |
altlba -= 33
|
|
|
003ee8 |
-gptheader = array.array('c',gptheader)
|
|
|
003ee8 |
+gptheader = array.array('B',gptheader)
|
|
|
003ee8 |
#update mylba
|
|
|
003ee8 |
pack_into('
|
|
|
003ee8 |
#update table lba
|
|
|
003ee8 |
@@ -33,9 +34,9 @@ pack_into('
|
|
|
003ee8 |
#zero header crc
|
|
|
003ee8 |
pack_into('
|
|
|
003ee8 |
#compute new crc
|
|
|
003ee8 |
-newcrc = ((crc32(buffer(gptheader,0,92))) & 0xFFFFFFFF)
|
|
|
003ee8 |
+newcrc = ((crc32(gptheader[:92])) & 0xFFFFFFFF)
|
|
|
003ee8 |
pack_into('
|
|
|
003ee8 |
file.seek(512*(altlba-32))
|
|
|
003ee8 |
file.write(backup)
|
|
|
003ee8 |
file.write(gptheader)
|
|
|
003ee8 |
-file.write("\0" * (512 * 33))
|
|
|
003ee8 |
+file.write(b"\0" * (512 * 33))
|
|
|
003ee8 |
diff --git a/tests/msdos-overlap b/tests/msdos-overlap
|
|
|
003ee8 |
index 5bddfb0..d6ae8d6 100755
|
|
|
003ee8 |
--- a/tests/msdos-overlap
|
|
|
003ee8 |
+++ b/tests/msdos-overlap
|
|
|
003ee8 |
@@ -14,12 +14,11 @@ BAD_ENTRY = (0x72, 0xf5, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
|
003ee8 |
OFFSET = 0x1b8
|
|
|
003ee8 |
|
|
|
003ee8 |
if len(sys.argv) < 2:
|
|
|
003ee8 |
- print "%s: <image or device>"
|
|
|
003ee8 |
+ print("%s: <image or device>" % sys.argv[0])
|
|
|
003ee8 |
sys.exit(1)
|
|
|
003ee8 |
|
|
|
003ee8 |
-data = "".join(chr(c) for c in BAD_ENTRY)
|
|
|
003ee8 |
with open(sys.argv[1], "rb+") as f:
|
|
|
003ee8 |
f.seek(OFFSET, 0)
|
|
|
003ee8 |
- f.write(data)
|
|
|
003ee8 |
+ f.write(bytes(bytearray(BAD_ENTRY)))
|
|
|
003ee8 |
|
|
|
003ee8 |
sys.exit(0)
|
|
|
003ee8 |
--
|
|
|
003ee8 |
2.17.1
|
|
|
003ee8 |
|