Refine cross protection code

jondy · jondy · commit 34409dd733cf · 2020-06-27T11:02:20.000+08:00
diff --git a/src/protect_code.pt b/src/protect_code.pt
@@ -41,13 +41,13 @@ def protect_pytransform():
         else:
             filename = pytransform.os.path.join({rpath}, {spath}, libname)
 
-        size = pytransform.os.path.getsize(filename) & 0xFFFFFFF0
-        n = size >> 2
         with open(filename, 'rb') as f:
-            buf = f.read(size)
-        fmt = 'I' * n
-        checksum = sum(pytransform.struct.unpack(fmt, buf)) & 0xFFFFFFFF
-        if checksum not in {checksum}:
+            buf = bytearray(f.read())
+        value = sum(buf)
+        if getattr(pytransform.sys, 'frozen', False) and sys.platform == 'darwin':
+            value += 1217
+
+        if value not in {checksum}:
             raise RuntimeError('unexpected %s' % filename)
 
     assert_builtin(sum)
diff --git a/src/utils.py b/src/utils.py
@@ -528,8 +528,7 @@ def copy3(src, dst, onlycopy=False):
         data = _patch_extension(target, keylist, suffix)
         with open(target, 'wb') as f:
             f.write(data)
-
-        checklist.append(_get_checksum(target))
+        checklist.append(sum(bytearray(data)))
 
     if not platforms:
         libfile = pytransform._pytransform._name
