git.rot13.org Git - goodfet/blob - client/goodfet.bslv2

   1 #!/usr/bin/env python
   2
   3 import hidapi
   4 import struct
   5 import time
   6
   7 BSL_VID = 0x2047
   8 BSL_PID = 0x0200
   9
  10 def _fmt_addr(addr, count=3):
  11     return "".join([chr((addr >> (n*8)) & 0xff) for n in range(count)])
  12
  13 RAM_BSL = """
  14 @2400
  15 E2 43 04 02 E2 43 02 02 80 00 04 25
  16 @2500
  17 00 05 04 34 31 40 90 33 B0 13 5E 2E 0C 93 00 24
  18 B0 13 F4 2D FF 3F 12 01 00 02 00 00 00 08 47 20
  19 00 02 04 01 00 00 00 01 06 00 FF 09 01 A1 01 85
  20 3F 95 3F 75 08 25 01 15 01 09 01 81 02 85 3F 95
  21 3F 75 08 25 01 15 01 09 01 91 02 C0 09 02 29 00
  22 01 01 00 80 32 09 04 00 00 02 03 00 00 00 09 21
  23 01 01 00 01 22 24 00 07 05 81 03 40 00 01 07 05
  24 01 03 40 00 01
  25 @2576
  26 F2 D2 20 09 F2 D2 22 09 10 01 5E 42 02 24 7E 93
  27 25 24 7E 90 09 00 04 28 7D 42 7E 82 5F 43 0C 3C
  28 7E 92 02 2C 4D 4E 06 3C 7D 42 D2 93 0E 24 02 20
  29 4E 43 F4 3F 7E 43 4F 43 C2 4F 10 24 C2 4E 02 24
  30 4F 43 07 3C 1E 42 06 24 EF 4E 78 23 92 53 06 24
  31 5F 53 4F 9D F7 2B C2 4D 21 09 10 01 C2 43 10 24
  32 10 01 82 4C 06 24 5E 42 86 23 C2 9E 02 24 04 28
  33 C2 4E 02 24 4E 43 01 3C 5E 43 C2 4E 0E 24 80 00
  34 80 25 F2 B0 0F 00 84 23 14 20 C2 93 84 23 03 34
  35 5E 42 20 09 02 3C 5E 42 22 09 7E F2 C2 4E 60 24
  36 5E 42 60 24 42 19 4E 10 C2 4E 60 24 B0 13 C4 27
  37 09 3C C2 93 84 23 03 34 5E 42 C8 23 EE 3F 5E 42
  38 88 23 EB 3F 3C 40 60 24 80 00 D8 25 F2 43 02 24
  39 C2 43 10 24 C2 43 21 09 10 01 C2 93 82 23 12 20
  40 5E 42 84 23 7E F0 0F 00 02 20 80 00 42 26 5E 93
  41 0B 20 C2 93 84 23 03 34 F2 D2 C8 23 F6 3F F2 D2
  42 88 23 F3 3F B0 13 76 25 10 01 C2 93 80 23 04 34
  43 1F 43 D2 D3 3C 09 03 3C 0F 43 D2 C3 3C 09 5E 42
  44 80 23 7E B0 60 00 90 20 5D 42 81 23 4D 83 81 24
  45 5D 83 6B 24 6D 83 67 24 6D 83 45 24 5D 83 09 24
  46 6D 83 52 24 5D 83 46 24 5D 83 33 24 5D 83 54 24
  47 7B 3C 0F 93 79 24 5E 42 83 23 5E 83 08 24 5E 83
  48 0F 24 7E 80 1F 00 1C 24 5E 83 13 24 6D 3C C2 43
  49 23 09 F2 40 12 00 02 24 3C 40 16 25 80 00 D8 25
  50 C2 43 23 09 F2 40 29 00 02 24 3C 40 4C 25 80 00
  51 D8 25 F2 40 24 00 02 24 3C 40 28 25 80 00 D8 25
  52 C2 43 23 09 F2 40 09 00 02 24 3C 40 5E 25 80 00
  53 D8 25 0F 93 49 24 B0 13 C4 27 C2 43 60 24 D2 42
  54 01 24 61 24 3B 3C B0 13 CE 27 D2 42 82 23 3F 09
  55 80 00 42 26 B0 13 CE 27 D2 42 82 23 00 24 B0 13
  56 42 26 D2 43 12 24 10 01 C2 43 23 09 D2 43 02 24
  57 3C 40 00 24 80 00 D8 25 B0 13 CE 27 D2 42 84 23
  58 01 24 80 00 42 26 80 00 50 26 5E 42 84 23 7E F0
  59 0F 00 02 20 80 00 42 26 5E 93 18 20 C2 93 84 23
  60 04 34 F2 F0 D7 00 C8 23 F5 3F F2 F0 D7 00 88 23
  61 F1 3F 7E 90 80 00 03 20 B0 13 C4 27 43 3F 7E 90
  62 82 00 02 20 80 00 F8 25 B0 13 76 25 10 01 C2 43
  63 23 09 E2 43 02 24 10 01 D5 3E 1B 15 1F 42 5A 24
  64 5B 4F 03 00 5E 4F 01 00 5C 4F 02 00 8C 10 0C DE
  65 0D 4B 0E 4F 2E 52 6A 4F 7A 80 10 00 29 24 5A 83
  66 14 24 5A 83 2A 24 5A 83 2E 24 6A 83 23 24 5A 83
  67 3A 24 5A 83 15 24 5A 83 3B 24 5A 83 3E 24 6A 83
  68 41 20 5F 43 B0 13 E2 2B 41 3C 1F 53 0C 4F B0 13
  69 38 2C 4C 93 02 20 4C 43 37 3C 7C 40 05 00 34 3C
  70 B0 13 66 2E 03 20 B0 13 6E 2E F5 3F 6C 42 2C 3C
  71 4F 43 E8 3F B0 13 D4 2D 27 3C 0E 4C 0F 4B 4C 43
  72 B0 13 D4 2C 21 3C B0 13 66 2E F0 23 4C 43 1F 42
  73 58 24 3F 50 40 00 1B 42 44 01 3B F0 10 00 0F 5B
  74 82 4F 44 01 11 3C B0 13 3C 2E B0 13 12 2B 0E 3C
  75 B0 13 3C 2E B0 13 66 29 09 3C 2E 42 3C 40 00 25
  76 0D 43 F8 3F 7C 40 07 00 B0 13 12 2E 1A 17 10 01
  77 E2 B2 3E 09 14 28 F2 40 80 00 23 09 03 3C F2 F0
  78 FA 00 3E 09 C2 43 10 24 C2 43 60 24 C2 43 61 24
  79 B0 13 80 26 D2 B3 3E 09 F2 2F E2 C2 3E 09 1F 42
  80 32 09 7F 90 0A 00 0C 20 B0 13 4A 2E B0 13 86 2C
  81 B0 13 08 2A B2 F0 F9 FF 08 09 A2 D3 02 09 10 01
  82 7F 90 0C 00 06 20 B0 13 4A 2E B2 40 04 A5 20 01
  83 10 01 7F 90 12 00 0A 20 C2 43 23 09 D2 93 10 24
  84 02 20 80 00 80 25 F2 D2 20 09 10 01 7F 90 16 00
  85 02 20 80 00 08 2A 7F 90 18 00 0C 20 D2 43 11 24
  86 F2 C0 40 00 3E 09 B2 40 80 00 10 09 F2 40 20 00
  87 3D 09 10 01 7F 90 1A 00 0A 20 B0 13 86 2C F2 F0
  88 9F 00 3E 09 F2 40 C0 00 3D 09 C2 43 11 24 10 01
  89 7B 15 0A 4C 0B 4D 0F 4E 3F E3 0F 5F 0F 7F 08 4C
  90 09 4D 08 5E 09 6F 47 43 0B 3C 1F 42 5C 24 FF 40
  91 3A 00 00 00 0C 46 1C 53 B0 13 2A 2E 0A 56 0B 63
  92 0B 99 03 28 34 20 0A 98 32 2C 47 93 30 20 0E 48
  93 0F 49 0E 8A 0F 7B 03 20 3E 90 3E 00 03 28 36 40
  94 3D 00 02 3C 06 48 06 8A 14 42 5C 24 14 53 0E 46
  95 0F 46 3F E3 0F 5F 0F 7F 0E 5A 0F 6B 09 3C 1F 15
  96 0D 16 6C 4D 0D 4E 0D 8A 05 44 05 5D C5 4C 00 00
  97 3E 53 3F 63 0F 9B C9 2B 02 20 0E 9A C6 2B B0 13
  98 66 2E ED 27 67 42 6C 42 B0 13 12 2E C7 3F 74 17
  99 10 01 F2 40 10 00 3C 09 C2 43 12 24 C2 43 11 24
 100 C2 43 00 24 C2 43 01 24 C2 43 3C 09 F2 43 02 24
 101 F2 43 04 24 C2 43 10 24 7E 40 80 00 C2 4E 21 09
 102 C2 4E 23 09 F2 40 8C 00 20 09 F2 40 8C 00 22 09
 103 F2 40 03 00 2F 09 F2 40 03 00 2E 09 C2 4E C8 23
 104 F2 40 10 00 C9 23 C2 4E CA 23 C2 4E CE 23 F2 40
 105 40 00 CF 23 C2 4E 88 23 C2 43 89 23 C2 43 8A 23
 106 F2 40 40 00 8F 23 F2 40 40 00 3C 09 C2 43 3E 09
 107 F2 40 C0 00 3D 09 10 01 7B 15 08 4C 07 4D 04 4F
 108 4C 43 0A 48 0B 4D 0F 4E 3F E3 0F 5F 0F 7F 06 48
 109 06 5E 07 6F 02 3C 1A 53 0B 63 0B 97 03 28 2C 20
 110 0A 96 2A 2C 18 B3 08 2C 0E 46 0F 47 3E 53 3F 63
 111 0A 9E 19 20 0B 9F 17 20 6E 44 B0 13 66 2E 10 20
 112 4C 43 B0 13 56 2E 1B 15 0F 16 CF 4E 00 00 B0 13
 113 56 2E 1B 15 0F 16 6D 4F 4E 9D 03 24 5C 43 01 3C
 114 6C 42 14 53 07 3C 3E 44 0C 4A 0D 4B B0 13 64 2D
 115 1A 53 0B 63 4C 93 CF 27 74 17 10 01 3B 15 0A 4E
 116 B2 43 54 01 08 4C 09 4D 07 3C 19 15 0E 16 6F 4E
 117 C2 4F 52 01 18 53 09 63 0E 4C 0F 4D 0E 5A 0F 63
 118 09 9F 03 28 09 20 08 9E 07 2C B0 13 66 2E ED 27
 119 6C 42 B0 13 12 2E 15 3C 1E 42 54 01 1F 42 5C 24
 120 FF 40 3A 00 00 00 1B 42 5C 24 CB 4E 01 00 47 18
 121 0E 11 1F 42 5C 24 CF 4E 02 00 3C 40 03 00 B0 13
 122 2A 2E 38 17 10 01 32 C2 03 43 B2 40 02 1C 5A 24
 123 B2 40 17 24 5C 24 B2 40 28 96 00 09 82 43 02 09
 124 82 43 60 01 B2 40 F3 10 64 01 B2 40 40 00 62 01
 125 B2 40 44 02 68 01 C2 43 0E 24 C2 43 11 24 B2 40
 126 28 96 00 09 B2 40 40 1E 08 09 B2 40 80 00 04 09
 127 B0 13 4A 2E C2 43 12 24 B2 B2 08 09 06 28 B0 13
 128 86 2C B0 13 08 2A A2 D3 02 09 10 01 3B 15 4A 4F
 129 6F 42 3B 40 58 24 B0 13 66 2E 08 20 4F 43 A2 4B
 130 44 01 28 4B 38 50 40 00 82 48 40 01 4F 93 0B 20
 131 B2 90 05 00 5E 24 07 38 0F 4E 1E 42 5E 24 2E 82
 132 B0 13 8E 2A 4F 4C 4A 93 03 20 4C 4F B0 13 12 2E
 133 A2 4B 40 01 2F 4B 3F 50 10 00 82 4F 44 01 38 17
 134 10 01 1B 15 21 83 0D 43 3A 40 E0 FF 0B 43 7E 4A
 135 0F 4C 0F 5B 6F 4F 0E EF 0D DE 1B 53 3B 90 20 00
 136 F6 2B 0D 93 0E 20 B1 40 FF 7F 00 00 02 3C B1 53
 137 00 00 91 93 00 00 FB 37 B2 40 A5 A5 56 24 4C 43
 138 04 3C B0 13 D4 2D 7C 40 05 00 21 53 1A 17 10 01
 139 21 82 81 43 02 00 B2 40 28 96 00 09 92 D3 02 09
 140 92 42 14 24 12 09 B2 40 00 13 10 09 82 43 14 09
 141 81 43 00 00 02 3C 91 53 00 00 B1 90 64 00 00 00
 142 FA 2B 1F 41 02 00 0E 4F 1E 53 81 4E 02 00 3F 90
 143 E9 03 03 2C 82 93 14 09 E9 23 21 52 10 01 B0 13
 144 66 2E 0E 20 4C 43 B0 13 FA 2C 1D 42 58 24 2D 53
 145 82 4D 40 01 1F 15 0D 16 CD 43 00 00 80 00 08 2D
 146 6C 42 10 01 92 B3 44 01 FD 2F 92 42 58 24 44 01
 147 10 01 92 B3 44 01 FD 2F 1F 42 58 24 3F 50 10 00
 148 82 4F 44 01 10 01 82 43 5E 24 C2 43 8A 23 B0 13
 149 A6 28 D2 93 12 24 0D 20 C2 93 11 24 0A 20 4F 43
 150 C2 93 8A 23 04 34 5F 42 8A 23 7F F0 7F 00 82 4F
 151 5E 24 82 93 5E 24 EB 27 92 93 5E 24 06 38 5F 42
 152 01 1C 82 4F 5E 24 5C 43 10 01 4C 43 10 01 1B 15
 153 B0 13 66 2E 15 20 4F 43 B0 13 56 2E 1D 15 0A 16
 154 8A 4E 00 00 B0 13 56 2E 1D 15 0A 16 2B 4A 0E 9B
 155 01 24 5F 43 92 B3 46 01 04 28 7F 40 03 00 01 3C
 156 6F 42 4C 4F 1A 17 10 01 0A 12 7E 40 3F 00 C2 93
 157 CA 23 11 34 C2 4E 80 1C 3D 40 81 1C 4F 43 0A 4C
 158 0A 5F ED 4A 00 00 1D 53 5F 53 4F 9E F8 2B F2 40
 159 40 00 CA 23 01 3C 4E 43 4C 4E 3A 41 10 01 B0 13
 160 FA 2C B0 13 56 2E 1F 42 58 24 3F 50 06 00 82 4F
 161 40 01 C2 43 E0 FF B0 13 08 2D 4C 43 10 01 B2 40
 162 A5 A5 56 24 B2 40 00 A5 58 24 B0 13 7C 2B B0 13
 163 1C 2D 5C B3 FC 2B B0 13 D0 27 F9 3F 1F 42 5C 24
 164 FF 40 3B 00 00 00 1F 42 5C 24 CF 4C 01 00 2C 43
 165 80 00 2A 2E C2 4C 16 24 3C 40 16 24 B0 13 9E 2D
 166 4C 93 FA 27 10 01 6E 4E 5F 4F 05 00 47 18 0F 5F
 167 0E DF 10 01 03 43 3F 40 DE 2E 3F 53 FE 2F 10 01
 168 92 B3 44 01 FD 2F 10 01 B2 40 80 5A 5C 01 10 01
 169 B2 90 A5 A5 56 24 10 01 1D 15 10 01
 170 @FFFE
 171 04 25
 172 q"""
 173
 174 BLINK_BSL = """@8000
 175 31 40 00 34 B0 13 0C 80 B0 13 32 80 21 83 D2 43
 176 04 02 D2 43 02 02 B2 40 80 5A 5C 01 07 3C 91 53
 177 00 00 B1 93 00 00 FB 23 D2 E3 02 02 81 43 00 00
 178 F8 3F 80 00 36 80 80 00 3A 80 FF 3F
 179 @FFE0
 180 12 34 56 78 99 10 11 12 13 14 15 16 17 18 19 20
 181 12 34 56 78 99 10 11 12 13 14 15 16 17 18 00 80
 182 q"""
 183
 184 BLINK_BSL2 = """@2000
 185 31 40 00 34 B0 13 0C 20 B0 13 32 20 21 83 D2 43
 186 04 02 D2 43 02 02 B2 40 80 5A 5C 01 07 3C 91 53
 187 00 00 B1 93 00 00 FB 23 D2 E3 02 02 81 43 00 00
 188 F8 3F 80 00 36 20 80 00 3A 20 FF 3F
 189 @FFE0
 190 12 34 56 78 99 10 11 12 13 14 15 16 17 18 19 20
 191 12 34 56 78 99 10 11 12 13 14 15 16 17 18 00 20
 192 q"""
 193 def hexline(pkt):
 194     return " ".join(c.encode("hex") for c in pkt)
 195
 196 class TiHex(object):
 197     def __init__(self, string):
 198         self.content = content = []
 199         addr = 0
 200         for line in string.split("\n"):
 201             if line.startswith("@"):
 202                 addr = int(line[1:],16)
 203             elif line.startswith("q"):
 204                 return
 205             elif len(line) == 0:
 206                 continue
 207             else:
 208                 data = line.replace(" ", "").decode("hex")
 209                 if len(data) > 0:
 210                     content.append((addr, data))
 211                     addr += len(data)
 212     def __iter__(self):
 213         for x in self.content:
 214             yield x
 215
 216
 217
 218 class BSL(object):
 219     MSGS = ["SUCCESS",
 220             "Flash write check failed",
 221             "Flash Fail Bit Set",
 222             "Voltage Change During Program",
 223             "BSL Locked",
 224             "BSL Password Error",
 225             "Byte Write Forbidden",
 226             "Unknown Command",
 227             "Packet Length Exceeds Buffer Size"]
 228
 229     def __init__(self, vid = BSL_VID, pid = BSL_PID):
 230         self.vid = vid
 231         self.pid = pid
 232         self.device = hidapi.HidDevice(vid, pid)
 233
 234     def _send_command(self, num, data, expect_response = True):
 235         if len(data) >= 255:
 236             raise Exception("Data too long")
 237         packet = '\x3f' + chr(len(data) + 1) + chr(num) + data
 238         #print " ".join(c.encode("hex") for c in packet)
 239         r = self.device.write(packet)
 240         #print r
 241         if expect_response:
 242             rdata = self.device.read(64)
 243             if len(rdata) < 2:
 244                 raise Exception("Short response")
 245             if rdata[0] != '\x3f':
 246                 raise Exception("Malformed packet")
 247             print repr(rdata)
 248             resp = rdata[2: 2+ord(rdata[1])]
 249             if resp[0] == '\x3a':
 250                 return resp[1:]
 251             elif resp[0] == '\x3b' and len(resp) == 2:
 252                 if resp[1] == '\0':
 253                     return True
 254                 else:
 255                     raise Exception(self.MSGS[ord(resp[1])])
 256             else:
 257                 raise Exception("Slightly malformed response")
 258
 259     def RxPassword(self, passwd):
 260         return self._send_command(0x11, passwd)
 261
 262
 263     def RxDataBlock(self, addr, data):
 264         return self._send_command(0x10, _fmt_addr(addr) + data)
 265
 266     def RxDataBlockFast(self, addr, data):
 267         return self._send_command(0x1b, _fmt_addr(addr) + data, False)
 268
 269     def EraseSegment(self, addr):
 270         return self._send_command(0x12, _fmt_addr(addr))
 271
 272     def ToggleInfoLock(self):
 273         return self._send_command(0x13, "")
 274     def MassErase(self):
 275         return self._send_command(0x15, "")
 276     def CrcCheck(self, addr, length):
 277         return self._send_command(0x16, _fmt_addr(addr) + _fmt_addr(length, 2))
 278     def LoadPc(self, addr):
 279         return self._send_command(0x17, _fmt_addr(addr), False)
 280     def TxDataBlock(self, addr, length):
 281         return self._send_command(0x18, _fmt_addr(addr) + _fmt_addr(length, 2))
 282     def TxBslVersion(self):
 283         return self._send_command(0x19, "")
 284     def TxBufferSize(self):
 285         return self._send_command(0x1a, "")
 286
 287     # Helpers
 288     def RxTIHexFast(self, hexstring):
 289         tihex = TiHex(hexstring)
 290         for addr, value in tihex:
 291             self.RxDataBlockFast(addr, value)
 292
 293     def bounce_hid(self):
 294         "Reconnect to the target"
 295         self.device.close()
 296         time.sleep(0.250)
 297         self.device = hidapi.HidDevice(self.vid, self.pid)
 298         print self.device._device
 299
 300 def main():
 301     bsl = BSL()
 302     #print BLINK_BSL
 303     #return
 304     try:
 305         print repr(bsl.RxPassword('\xff' * 32))
 306         print repr(bsl.RxTIHexFast(RAM_BSL))
 307         print repr(bsl.LoadPc(0x2400))
 308         #print repr(bsl.RxTIHexFast(file("blinky.tihex","r").read()))
 309         #print repr(bsl.LoadPc(0x2000))
 310         #bsl.device.close()
 311         print repr("Bouncing...")
 312         time.sleep(3)
 313         print repr(bsl.bounce_hid())
 314         print repr("waiting")
 315         print repr("Next")
 316         print repr(bsl.TxBufferSize())
 317         print repr(bsl.TxBslVersion())
 318     finally:
 319         bsl.device.close()
 320
 321 main()