RAConnect.py: minor cleanup debug and print statements
[renesas-ra-flasher] / src / RAFlasher.py
index f450162..f9eab39 100644 (file)
 import os
 import math
+import time
 import argparse
+from tqdm import tqdm
 from RAConnect import *
 from RAPacker import *
 
 SECTOR_SIZE = 2048
 
+def int_to_hex_list(num):
+    hex_string = hex(num)[2:].upper()  # convert to hex string
+    hex_string = hex_string.zfill(8) # pad for 8 char's long
+    hex_list = [f'0x{hex_string[c:c+2]}' for c in range(0, 8, 2)]
+    return hex_list
+
+def inquire_connection(dev):
+    packed = pack_pkt(INQ_CMD, "")
+    dev.send_data(packed)
+    info = dev.recv_data(7)
+    if info == bytearray(b'\x00') or info == bytearray(b''):
+        return False
+    msg = unpack_pkt(info)
+    #print("Connection already established")
+    return True
+
+def get_area_info(dev):
+    for i in [0,1,2]:
+        print("===================")
+        packed = pack_pkt(ARE_CMD, [str(i)])
+        dev.send_data(packed)
+        info = dev.recv_data(23)
+        msg = unpack_pkt(info)
+        fmt = '>BIIII'
+        KOA, SAD, EAD, EAU, WAU = struct.unpack(fmt, bytes(int(x, 16) for x in msg))
+        print(f'Area {KOA} - {hex(SAD)}:{hex(EAD)}')
+        print(f'Erase {hex(EAU)} bytes - write {hex(WAU)} bytes')
+
 def get_dev_info(dev):
 
     packed = pack_pkt(SIG_CMD, "")
     dev.send_data(packed)
-    #info_ret = dev.recv_data(17)
-    info = b'\x81\x00\x0D\x3A\x01\x31\x2d\x00\x00\x1e\x84\x80\x04\x02\x0a\x08' # test
-    fmt = '>IIIBBH'
-    _HEADER, SCI, RMB, NOA, TYP, BFV = struct.unpack(fmt, info)
-    print(f'Ver{BFV >> 8}.{BFV & 0xFF}')
+    info = dev.recv_data(18)
+    fmt = '>IIIBBHH'
+    _HEADER, SCI, RMB, NOA, TYP, BFV, _FOOTER = struct.unpack(fmt, info)
+    print('Chip info:')
+    print('====================')
+    print(f'Serial interface speed: {SCI} Hz')
+    print(f'Recommend max UART baud rate {RMB} bps')
+    print(f'User area in Code flash [{NOA & 0x1}|{NOA & 0x02 >> 1}]')
+    print(f'User area in Data flash [{NOA & 0x03 >> 2}]')
+    print(f'Config area [{NOA & 0x04 >> 3}]')
+    if TYP == 0x02:
+        print('RA MCU + RA2/RA4 Series')
+    elif TYP == 0x03:
+        print('RA MCU + RA6 Series')
+    else:
+        print('Unknown MCU type')
+    print(f'Boot firmware version {BFV >> 8}.{BFV & 0xFF}')
+    print('====================')
+
 
 def verify_img(dev, img, start_addr, end_addr):
     raise Exception("Not implemented")
 
+def read_img(dev, img, start_addr, end_addr):
+    
+    # calculate / check start and end address 
+    if start_addr == None or end_addr == None:
+        if start_addr == None:
+            start_addr = 0
+        # align start addr
+        if start_addr % SECTOR_SIZE:
+            raise ValueError(f"start addr not aligned on sector size {SECTOR_SIZE}")
+        blocks = (file_size + SECTOR_SIZE - 1) // SECTOR_SIZE
+        end_addr = blocks * SECTOR_SIZE + start_addr
+    
+    if (start_addr > 0xFF800): # for RA4 series
+        raise ValueError("start address value error")
+    if (end_addr <= start_addr or end_addr > 0xFF800):
+        raise ValueError("end address value error")
+    
+    # setup initial communication
+    SAD = int_to_hex_list(start_addr)
+    EAD = int_to_hex_list(end_addr)
+    packed = pack_pkt(REA_CMD, SAD + EAD)
+    dev.send_data(packed)
+
+    # calculate how many packets are have to be received
+    nr_packet = (end_addr - start_addr) // 1024 # TODO: set other than just 1024
+
+    with open('output.bin', 'wb') as f:
+        for i in tqdm(range(0, nr_packet+1), desc="Reading progess"):
+            ret = dev.recv_data(1024 + 6)
+            chunk = unpack_pkt(ret)
+            chunky = bytes(int(x, 16) for x in chunk)
+            f.write(chunky)
+            packed = pack_pkt(REA_CMD, ['0x00'], ack=True)
+            dev.send_data(packed)
+
+
 def write_img(dev, img, start_addr, end_addr, verify=False):
 
     if os.path.exists(img):
@@ -34,30 +114,39 @@ def write_img(dev, img, start_addr, end_addr, verify=False):
         if start_addr % SECTOR_SIZE:
             raise ValueError(f"start addr not aligned on sector size {SECTOR_SIZE}")
         blocks = (file_size + SECTOR_SIZE - 1) // SECTOR_SIZE
-        end_addr = blocks * SECTOR_SIZE + start_addr
-        print(end_addr)
+        end_addr = blocks * SECTOR_SIZE + start_addr - 1
     
-    chunk_size = 64 # max is 1024
-    if (start_addr > 0xFF800): # for RA4 series
-        raise ValueError("start address value error")
-    if (end_addr <= start_addr or end_addr > 0xFF800):
-        raise ValueError("end address value error")
-
-    # @TODO setup initial communication
+    chunk_size = 1024 # max is 1024
+    #if (start_addr > 0xFF800): # for RA4 series
+    #    raise ValueError("start address value error")
+    #if (end_addr <= start_addr or end_addr > 0xFF800):
+    #    raise ValueError("end address value error")
+
+    # setup initial communication
+    SAD = int_to_hex_list(start_addr)
+    EAD = int_to_hex_list(end_addr)
+    #packed = pack_pkt(ERA_CMD, SAD + EAD) # actually works
+    packed = pack_pkt(WRI_CMD, SAD + EAD)
+    dev.send_data(packed)
+    ret = dev.recv_data(7)
+    unpack_pkt(ret) 
 
     with open(img, 'rb') as f:
-        chunk = f.read(chunk_size)
-        while chunk:
-            packed = pack_pkt(WRI_CMD, chunk)
-            print(f'Sending {len(chunk)} bytes')
-            dev.send_data(packed)
-            reply_len = 7
-            reply = dev.recv_data(reply_len)
-            #reply = b'\x81\x00\x02\x00\x00\xFE\x03' # test reply
-            if not reply == False:
-                msg = unpack_pkt(reply)
-                print(msg)
+        with tqdm(total=file_size, desc="Sending chunks") as pbar:
             chunk = f.read(chunk_size)
+            pbar.update(len(chunk))
+            while chunk:
+                if len(chunk) != chunk_size:
+                    padding_length = chunk_size - len(chunk)
+                    chunk += b'\0' * padding_length
+                packed = pack_pkt(WRI_CMD, chunk, ack=True)
+                dev.send_data(packed)
+                reply_len = 7
+                reply = dev.recv_data(reply_len)
+                msg = unpack_pkt(reply)
+                chunk = f.read(chunk_size)
+                pbar.update(len(chunk))
+
 
 def main():
     parser = argparse.ArgumentParser(description="RA Flasher Tool")
@@ -83,14 +172,24 @@ def main():
     args = parser.parse_args()
 
     if args.command == "write":
-        dev = RAConnect(vendor_id=0x1a86, product_id=0x7523)
-        print(args)
-        write_img(dev, args.file_name, args.start_address, args.end_address, args.verify)
+        dev = RAConnect(vendor_id=0x045B, product_id=0x0261)
+        status_con = inquire_connection(dev)
+        if not status_con:
+            dev.confirm_connection()
+        write_img(dev, "src/sample.bin", 0x8000, None)
     elif args.command == "read":
-        print('read command')
+        dev = RAConnect(vendor_id=0x045B, product_id=0x0261)
+        status_con = inquire_connection(dev)
+        if not status_con:
+            dev.confirm_connection()
+        read_img(dev, "save.bin", 0x0000, 0x3FFFF)
     elif args.command == "info":
-        dev = RAConnect(vendor_id=0x1a86, product_id=0x7523)
+        dev = RAConnect(vendor_id=0x045B, product_id=0x0261)
+        status_con = inquire_connection(dev)
+        if not status_con:
+            dev.confirm_connection()
         get_dev_info(dev)
+        get_area_info(dev)
     else:
         parser.print_help()