增加分块校验：pyChecknum.exe COAB21232D.bin -t crc32 -e little -s 1024，不分块用 -s 0

pyChecknum.py

@@ -15,49 +15,102 @@ def intToBytes(value, length):
     return result
 
 #查看文件后的16进制数据
-def ReadBinFile(in_file, type="sum32", en='little'):
-    get_checknum = None
+def ReadBinFile(in_file, type="sum32", en='little', block_size=1024):
     filename, file_type = os.path.splitext(in_file)  
     out_name = None
     checknum_bytes = None
 
     # 方式一：累加和
     if type == "sum32":
-        with open(in_file, "rb") as f:      #二进制打开文件
-            size =os.path.getsize(in_file)  #获取文件大小
-            checknum = 0
-            for i in range(size):
-                datas = f.read(1)
-                checknum += datas[0]
+        out_name0 = filename+file_type+".sum32"
+        f = open(in_file, "rb")
+        o = open(out_name0, "wb+")
+        if f and o:      #二进制打开文件
+            checknum_all = 0
+            if block_size:  
+                while True:
+                    # 读一块
+                    datas = f.read(block_size)
+                    if len(datas) == 0:
+                        break
+                    # 这一块的checknum
+                    checknum = 0
+                    for j in range(len(datas)):
+                        checknum += datas[0]
+                        checknum_all += datas[0]  # 总文件的checknum
+                    checknum_bytes = checknum.to_bytes(4, en)    #"little", "big"  # 大端序
+                    # 写入数据和checknum
+                    o.seek(0,2)  #位移到最后    SEEK_END(值为2)     SEEK_CUR(值为1)  SEEK_SET(值为0)
+                    o.write(datas) 
+                    o.write(bytes(checknum_bytes))  
+
+            else:
+                while True:    
+                    datas = f.read(1)
+                    if len(datas) == 0:
+                        break                    
+                    checknum_all += datas[0]    # 总文件的checknum
+                    o.seek(0,2)  #位移到最后    SEEK_END(值为2)     SEEK_CUR(值为1)  SEEK_SET(值为0)
+                    o.write(datas) 
+
             f.close()
-            get_checknum = True
-            # checknum_str = '{:08X}'.format(checknum)
-            checknum_bytes = checknum.to_bytes(4, en)    #"little", "big"  # 大端序
-            checknum_str = checknum_bytes.hex()
-            print(in_file, size, "checknum", checknum_str)
+            o.close()
+
+            checknum_all_bytes = checknum_all.to_bytes(4, en)    #"little", "big"  # 大端序
+            checknum_all_str = checknum_all_bytes.hex()
+            print(in_file, os.path.getsize(in_file), "checknum", checknum_all_str)
             # 输出文件名
-            out_name = filename+file_type+".sum32["+checknum_str+"]"
-            shutil.copyfile(in_file, out_name)
+            out_name1 = filename+file_type+".sum32["+checknum_all_str+"]"
+            shutil.move(out_name0, out_name1)
+            # 追加到文件最后
+            with open(out_name1, "rb+") as f:
+                f.seek(0,2)  #位移到最后    SEEK_END(值为2)     SEEK_CUR(值为1)  SEEK_SET(值为0)
+                f.write(bytes(checknum_all_bytes)) 
+                f.close()
+
     elif type == "crc32":
     # 方式2：crc32
-        with open(in_file, "rb") as f:      #二进制打开文件
-            size =os.path.getsize(in_file)  #获取文件大小
-            filedata = f.read()
-            checknum = binascii.crc32(filedata)
-            checknum_bytes = checknum.to_bytes(4, en)    #"little", "big"  # 大端序
-            # checknum_str = '{:08X}'.format(checknum)
-            checknum_str = checknum_bytes.hex()
-            print(in_file, size, "checknum", checknum_str)
-            # 输出文件名
-            out_name = filename+file_type+".crc32["+checknum_str+"]"
-            shutil.copyfile(in_file, out_name)
+        out_name0 = filename+file_type+".crc32"
+        f = open(in_file, "rb")
+        o = open(out_name0, "wb+")
+        if f and o:      #二进制打开文件
+            checknum_all = 0
+            if block_size:  
+                while True:
+                    # 读一块
+                    datas = f.read(block_size)
+                    if len(datas) == 0:
+                        break
+                    # 这一块的checknum
+                    checknum = binascii.crc32(datas)
+                    checknum_bytes = checknum.to_bytes(4, en)    #"little", "big"  # 大端序
+                    # 写入数据和checknum
+                    o.seek(0,2)  #位移到最后    SEEK_END(值为2)     SEEK_CUR(值为1)  SEEK_SET(值为0)
+                    o.write(datas) 
+                    o.write(bytes(checknum_bytes))  
+            else:
+                datas = f.read()
+                o.seek(0,2)  #位移到最后    SEEK_END(值为2)     SEEK_CUR(值为1)  SEEK_SET(值为0)
+                o.write(datas) 
 
-    if out_name:
-        # 追加到文件最后
-        with open(out_name, "rb+") as f:
-            f.seek(0,2)  #位移到最后
-            f.write(bytes(checknum_bytes)) 
+            f.seek(0,0)
+            datas = f.read()            
+            checknum_all = binascii.crc32(datas)
             f.close()
+            o.close()
+
+            checknum_all_bytes = checknum_all.to_bytes(4, en)    #"little", "big"  # 大端序
+            checknum_all_str = checknum_all_bytes.hex()
+            print(in_file, os.path.getsize(in_file), "checknum", checknum_all_str)
+            # 输出文件名
+            out_name1 = filename+file_type+".crc32["+checknum_all_str+"]"
+            shutil.move(out_name0, out_name1)
+            # 追加到文件最后
+            with open(out_name1, "rb+") as f:
+                f.seek(0,2)  #位移到最后    SEEK_END(值为2)     SEEK_CUR(值为1)  SEEK_SET(值为0)
+                f.write(bytes(checknum_all_bytes)) 
+                f.close()
+
 
 
 if __name__ == "__main__":
@@ -65,15 +118,19 @@ if __name__ == "__main__":
     parser.add_argument("n", help="bin文件名，譬如 COAB21232D.bin")
     parser.add_argument("-t", help="校验方式：sum32,crc32, 默认sum32")
     parser.add_argument("-e", help="校验码大小端：little,big, 默认little")
+    parser.add_argument("-s", help="分块大小：0,1024,2048, 默认0不分块")
 
     args = parser.parse_args()
     project = str(args.n)
     type = "sum32"      #默认 累加和
     en = "little"       #默认 checknum 小端
+    block_size = 0      #默认 0
     if args.t:
         type = str(args.t)
     if args.e:
         en = str(args.e)
+    if args.s:
+        block_size = int(args.s)    
     # print(project, type, en)
-    ReadBinFile(project, type, en)
+    ReadBinFile(project, type, en, block_size)