6 from RAConnect import *
11 def int_to_hex_list(num):
12 hex_string = hex(num)[2:].upper() # convert to hex string
13 hex_string = hex_string.zfill(8) # pad for 8 char's long
14 hex_list = [f'0x{hex_string[c:c+2]}' for c in range(0, 8, 2)]
17 def inquire_connection(dev):
18 packed = pack_pkt(INQ_CMD, "")
20 info = dev.recv_data(7)
22 if info == bytearray(b'\x00') or info == bytearray(b''):
24 msg = unpack_pkt(info)
25 print("Connection already established")
29 def get_area_info(dev):
31 print("===================")
32 packed = pack_pkt(ARE_CMD, [str(i)])
34 info = dev.recv_data(23)
35 msg = unpack_pkt(info)
37 KOA, SAD, EAD, EAU, WAU = struct.unpack(fmt, bytes(int(x, 16) for x in msg))
38 print(f'Area {KOA} - {hex(SAD)}:{hex(EAD)}')
39 print(f'Erase {hex(EAU)} bytes - write {hex(WAU)} bytes')
41 def get_dev_info(dev):
43 packed = pack_pkt(SIG_CMD, "")
46 info = dev.recv_data(18)
47 print(info, len(info))
48 #info = b'\x81\x00\x0D\x3A\x01\x31\x2d\x00\x00\x1e\x84\x80\x04\x02\x0a\x08' # test
50 _HEADER, SCI, RMB, NOA, TYP, BFV, _FOOTER = struct.unpack(fmt, info)
52 print('====================')
53 print(f'Serial interface speed: {SCI} Hz')
54 print(f'Recommend max UART baud rate {RMB} bps')
55 print(f'User area in Code flash [{NOA & 0x1}|{NOA & 0x02 >> 1}]')
56 print(f'User area in Data flash [{NOA & 0x03 >> 2}]')
57 print(f'Config area [{NOA & 0x04 >> 3}]')
59 print('RA MCU + RA2/RA4 Series')
61 print('RA MCU + RA6 Series')
63 print('Unknown MCU type')
64 print(f'Boot firmware version {BFV >> 8}.{BFV & 0xFF}')
65 print('====================')
68 def verify_img(dev, img, start_addr, end_addr):
69 raise Exception("Not implemented")
71 def read_img(dev, img, start_addr, end_addr):
73 # calculate / check start and end address
74 if start_addr == None or end_addr == None:
75 if start_addr == None:
78 if start_addr % SECTOR_SIZE:
79 raise ValueError(f"start addr not aligned on sector size {SECTOR_SIZE}")
80 blocks = (file_size + SECTOR_SIZE - 1) // SECTOR_SIZE
81 end_addr = blocks * SECTOR_SIZE + start_addr
84 if (start_addr > 0xFF800): # for RA4 series
85 raise ValueError("start address value error")
86 if (end_addr <= start_addr or end_addr > 0xFF800):
87 raise ValueError("end address value error")
89 # setup initial communication
90 SAD = int_to_hex_list(start_addr)
91 EAD = int_to_hex_list(end_addr)
92 packed = pack_pkt(REA_CMD, SAD + EAD)
95 # calculate how many packets are have to be received
96 nr_packet = (end_addr - start_addr) // 1024 # TODO: set other than just 1024
97 print(f'Sending {nr_packet} packets to MCU')
99 with open('output.bin', 'wb') as f:
100 for i in tqdm(range(0, nr_packet+1), desc="Reading progess"):
101 ret = dev.recv_data(1024 + 6)
102 #print(f'Packet nr: {i}')
103 chunk = unpack_pkt(ret)
104 chunky = bytes(int(x, 16) for x in chunk)
107 packed = pack_pkt(REA_CMD, ['0x00'], ack=True)
108 dev.send_data(packed)
111 def write_img(dev, img, start_addr, end_addr, verify=False):
113 if os.path.exists(img):
114 file_size = os.path.getsize(img)
116 raise Exception(f'file {img} does not exist')
118 # calculate / check start and end address
119 if start_addr == None or end_addr == None:
120 if start_addr == None:
123 if start_addr % SECTOR_SIZE:
124 raise ValueError(f"start addr not aligned on sector size {SECTOR_SIZE}")
125 blocks = (file_size + SECTOR_SIZE - 1) // SECTOR_SIZE
126 end_addr = blocks * SECTOR_SIZE + start_addr
129 chunk_size = 64 # max is 1024
130 if (start_addr > 0xFF800): # for RA4 series
131 raise ValueError("start address value error")
132 if (end_addr <= start_addr or end_addr > 0xFF800):
133 raise ValueError("end address value error")
135 # setup initial communication
136 SAD = int_to_hex_list(start_addr)
137 EAD = int_to_hex_list(end_addr)
138 packed = pack_pkt(WRI_CMD, SAD + EAD)
139 dev.send_data(packed)
140 ret = dev.recv_data(7)
142 with open(img, 'rb') as f:
143 chunk = f.read(chunk_size)
145 packed = pack_pkt(WRI_CMD, chunk)
146 print(f'Sending {len(chunk)} bytes')
147 dev.send_data(packed)
149 reply = dev.recv_data(reply_len)
150 #reply = b'\x81\x00\x02\x00\x00\xFE\x03' # test reply
151 if not reply == False:
152 msg = unpack_pkt(reply)
154 chunk = f.read(chunk_size)
157 parser = argparse.ArgumentParser(description="RA Flasher Tool")
159 subparsers = parser.add_subparsers(dest="command", title="Commands")
161 # Subparser for the write command
162 write_parser = subparsers.add_parser("write", help="Write data to flash")
163 write_parser.add_argument("--start_address", type=int, default=0x0000, help="Start address")
164 write_parser.add_argument("--end_address", type=int, help="End address")
165 write_parser.add_argument("--verify", action="store_true", help="Verify after writing")
166 write_parser.add_argument("file_name", type=str, help="File name")
168 # Subparser for the read command
169 read_parser = subparsers.add_parser("read", help="Read data from flash")
170 read_parser.add_argument("--start_address", type=int, default=0x000, help="Start address")
171 read_parser.add_argument("--size", type=int, default=1024, help="Size in bytes")
172 read_parser.add_argument("file_name", type=str, help="File name")
174 # Subparser for the info command
175 subparsers.add_parser("info", help="Show flasher information")
177 args = parser.parse_args()
179 if args.command == "write":
180 dev = RAConnect(vendor_id=0x045B, product_id=0x0261)
182 write_img(dev, args.file_name, args.start_address, args.end_address, args.verify)
183 elif args.command == "read":
184 print('read command')
185 elif args.command == "info":
186 dev = RAConnect(vendor_id=0x045B, product_id=0x0261)
187 status_con = inquire_connection(dev)
189 #dev.establish_connection()
190 dev.confirm_connection()
193 read_img(dev, "save.bin", 0x0000, 0x3FFFF)
197 if __name__ == "__main__":