initial source of sdat2img

initial source of sdat2img wrote by luxi78

sdat2img.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+#encoding:utf8
+#===============================================================================
+#          FILE: sdat2img.py
+#        AUTHOR: luxi78@gmail.com 
+#       CREATED: 2014年12月21日 15时25分15秒 CST
+#      REVISION:  ---
+#===============================================================================
+import sys
+
+TRANSFER_LIST_FILE = 'system.transfer.list'
+NEW_DATA_FILE = 'system.new.dat'
+OUTPUT_IMAGE_FILE = 'system.img'
+BLOCK_SIZE = 4096
+
+def rangeset(src):
+    src_set = src.split(',')
+    num_set =  [int(item) for item in src_set]
+    if len(num_set) != num_set[0]+1:
+        print 'Error for parsiing following data to rangeset:\n%s' % src
+        sys.exit(1)
+
+    return tuple ([ (num_set[i], num_set[i+1]) for i in range(1, len(num_set), 2) ])
+
+def parse_transfer_list_file(path):
+    trans_list = open(TRANSFER_LIST_FILE, 'r')
+    version = int(trans_list.readline())
+    new_blocks = int(trans_list.readline())
+    
+    for line in trans_list:
+        line = line.split(' ')
+        cmd = line[0]
+        if 'erase' == cmd:
+            erase_block_set = rangeset(line[1])
+        elif 'new' == cmd:
+            new_block_set = rangeset(line[1])
+        else:
+            print 'Error command %s' % cmd
+            trans_list.close()
+            sys.exit(1)
+
+    trans_list.close()
+    return version, new_blocks, erase_block_set, new_block_set 
+
+def init_output_file_size(output_file_obj, erase_block_set):
+    max_block_num = max(pair[1] for pair in erase_block_set)
+    output_file_obj.seek(max_block_num*BLOCK_SIZE - 1)
+    output_file_obj.write('\0')
+    output_file_obj.flush()
+
+def main(argv):
+    version, new_blocks, erase_block_set, new_block_set =  parse_transfer_list_file(TRANSFER_LIST_FILE)
+    output_img = file( OUTPUT_IMAGE_FILE, 'wb')
+    init_output_file_size(output_img, erase_block_set)
+    new_data_file = file(NEW_DATA_FILE, 'rb')
+   
+    for block in new_block_set:
+        begin = block[0]
+        end = block[1]
+        block_count = end - begin
+        print "Reading %d blocks..." % block_count,
+        data = new_data_file.read(block_count*BLOCK_SIZE)
+        print "writing to %d..." % begin,
+        output_img.seek(begin*BLOCK_SIZE)
+        output_img.write(data)
+        print "finish!"
+
+    output_img.close()
+    new_data_file.close()
+
+if __name__ == "__main__":
+    main(sys.argv)