7 from RAConnect import *
14 def int_to_hex_list(num):
15 hex_string = hex(num)[2:].upper() # convert to hex string
16 hex_string = hex_string.zfill(8) # pad for 8 char's long
17 hex_list = [f'0x{hex_string[c:c+2]}' for c in range(0, 8, 2)]
22 value = int(string, 16)
25 raise argparse.ArgumentTypeError(f"'{string}' is not a valid hexadecimal value.")
27 def set_size_boundaries(start_addr, size):
28 if start_addr % SECTOR_SIZE:
29 raise ValueError(f"start addr not aligned on sector size {SECTOR_SIZE}")
31 if size < SECTOR_SIZE:
32 print("Warning: you are trying to write something that is less than one sector size: padding with zeroes")
34 blocks = (size + SECTOR_SIZE - 1) // SECTOR_SIZE
35 end_addr = blocks * SECTOR_SIZE + start_addr - 1
37 if (end_addr <= start_addr):
38 raise ValueError(f"End address smaller or equal than start_address")
40 if (end_addr > 0x3FFFF):
41 raise ValueError(f"Binary file is bigger than available ROM space")
43 return (start_addr, end_addr)
45 def get_area_info(dev):
47 packed = pack_pkt(ARE_CMD, [str(i)])
49 info = dev.recv_data(23)
50 msg = unpack_pkt(info)
52 KOA, SAD, EAD, EAU, WAU = struct.unpack(fmt, bytes(int(x, 16) for x in msg))
53 print(f'Area {KOA}: {hex(SAD)}:{hex(EAD)} (erase {hex(EAU)} - write {hex(WAU)})')
55 def get_dev_info(dev):
56 packed = pack_pkt(SIG_CMD, "")
58 info = dev.recv_data(18)
60 _HEADER, SCI, RMB, NOA, TYP, BFV, _FOOTER = struct.unpack(fmt, info)
61 print('====================')
63 print('Chip: RA MCU + RA2/RA4 Series')
65 print('Chip: RA MCU + RA6 Series')
67 rint('Unknown MCU type')
68 print(f'Serial interface speed: {SCI} Hz')
69 print(f'Recommend max UART baud rate: {RMB} bps')
70 print(f'User area in Code flash [{NOA & 0x1}|{NOA & 0x02 >> 1}]')
71 print(f'User area in Data flash [{NOA & 0x03 >> 2}]')
72 print(f'Config area [{NOA & 0x04 >> 3}]')
73 print(f'Boot firmware: version {BFV >> 8}.{BFV & 0xFF}')
75 def erase_chip(dev, start_addr, size):
78 size = 0x3FFFF - start_addr # erase all
80 (start_addr, end_addr) = set_size_boundaries(start_addr, size)
81 print(f'Erasing {hex(start_addr)}:{hex(end_addr)}')
83 # setup initial communication
84 SAD = int_to_hex_list(start_addr)
85 EAD = int_to_hex_list(end_addr)
86 packed = pack_pkt(ERA_CMD, SAD + EAD)
89 ret = dev.recv_data(7, timeout=1000) # erase takes usually a bit longer
91 print("Erase complete")
93 def read_img(dev, img, start_addr, size):
96 size = 0x3FFFF - start_addr # read maximum possible
98 (start_addr, end_addr) = set_size_boundaries(start_addr, size)
100 # setup initial communication
101 SAD = int_to_hex_list(start_addr)
102 EAD = int_to_hex_list(end_addr)
103 packed = pack_pkt(REA_CMD, SAD + EAD)
104 dev.send_data(packed)
106 # calculate how many packets are have to be received
107 nr_packet = (end_addr - start_addr) // 1024 # TODO: set other than just 1024
109 with open(img, 'wb') as f:
110 for i in tqdm(range(0, nr_packet+1), desc="Reading progress"):
111 ret = dev.recv_data(1024 + 6)
112 chunk = unpack_pkt(ret)
113 chunky = bytes(int(x, 16) for x in chunk)
115 packed = pack_pkt(REA_CMD, ['0x00'], ack=True)
116 dev.send_data(packed)
119 def write_img(dev, img, start_addr, size, verify=False):
121 if os.path.exists(img):
122 file_size = os.path.getsize(img)
124 raise Exception(f'file {img} does not exist')
130 raise ValueError("Write size > file size")
132 (start_addr, end_addr) = set_size_boundaries(start_addr, size)
134 chunk_size = 1024 # max is 1024 according to protocol
136 # setup initial communication
137 SAD = int_to_hex_list(start_addr)
138 EAD = int_to_hex_list(end_addr)
139 packed = pack_pkt(WRI_CMD, SAD + EAD)
140 dev.send_data(packed)
141 ret = dev.recv_data(7)
145 with open(img, 'rb') as f:
146 with tqdm(total=size, desc="Writing progress") as pbar:
147 chunk = f.read(chunk_size)
148 pbar.update(len(chunk))
149 while chunk and totalread < size:
150 if len(chunk) != chunk_size:
151 padding_length = chunk_size - len(chunk)
152 chunk += b'\0' * padding_length
153 packed = pack_pkt(WRI_CMD, chunk, ack=True)
154 dev.send_data(packed)
156 reply = dev.recv_data(reply_len)
157 msg = unpack_pkt(reply)
158 chunk = f.read(chunk_size)
159 totalread += chunk_size
160 pbar.update(len(chunk))
164 with tempfile.NamedTemporaryFile(prefix='.hidden_', delete=False) as tmp_file, open(img, 'rb') as cmp_file:
165 read_img(dev, tmp_file.name, start_addr, size)
166 c1 = tmp_file.read(file_size) # due to byte alignment read file is longer
169 print("Verify complete")
171 print("Verify failed")
174 parser = argparse.ArgumentParser(description="RA Flasher Tool")
176 subparsers = parser.add_subparsers(dest="command", title="Commands")
178 # Subparser for the write command
179 write_parser = subparsers.add_parser("write", help="Write data to flash")
180 write_parser.add_argument("--start_address", type=hex_type, default='0x0000', help="Start address")
181 write_parser.add_argument("--size", type=hex_type, default=None, help="Size in bytes")
182 write_parser.add_argument("--verify", action="store_true", help="Verify after writing")
183 write_parser.add_argument("file_name", type=str, help="File name")
185 # Subparser for the read command
186 read_parser = subparsers.add_parser("read", help="Read data from flash")
187 read_parser.add_argument("--start_address", type=hex_type, default='0x0000', help="Start address")
188 read_parser.add_argument("--size", type=hex_type, default=None, help="Size in bytes")
189 read_parser.add_argument("file_name", type=str, help="File name")
191 erase_parser = subparsers.add_parser("erase", help="Erase sectors")
192 erase_parser.add_argument("--start_address", default='0x0000', type=hex_type, help="Start address")
193 erase_parser.add_argument("--size", type=hex_type, help="Size")
195 # Subparser for the info command
196 subparsers.add_parser("info", help="Show flasher information")
198 args = parser.parse_args()
200 if args.command == "write":
201 dev = RAConnect(VENDOR_ID, PRODUCT_ID)
202 write_img(dev, args.file_name, args.start_address, args.size, args.verify)
203 elif args.command == "read":
204 dev = RAConnect(VENDOR_ID, PRODUCT_ID)
205 read_img(dev, args.file_name, args.start_address, args.size)
206 elif args.command == "erase":
207 dev = RAConnect(VENDOR_ID, PRODUCT_ID)
208 erase_chip(dev, args.start_address, args.size)
209 elif args.command == "info":
210 dev = RAConnect(VENDOR_ID, PRODUCT_ID)
216 if __name__ == "__main__":