]>
Commit | Line | Data |
---|---|---|
683fe750 BV |
1 | #!/usr/bin/python3 |
2 | ## | |
d0b7f2e8 | 3 | ## This file is part of the sigrok-util project. |
683fe750 BV |
4 | ## |
5 | ## Copyright (C) 2012 Bert Vermeulen <bert@biot.com> | |
6 | ## | |
7 | ## This program is free software; you can redistribute it and/or modify | |
8 | ## it under the terms of the GNU General Public License as published by | |
9 | ## the Free Software Foundation; either version 3 of the License, or | |
10 | ## (at your option) any later version. | |
11 | ## | |
12 | ## This program is distributed in the hope that it will be useful, | |
13 | ## but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 | ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
15 | ## GNU General Public License for more details. | |
16 | ## | |
17 | ## You should have received a copy of the GNU General Public License | |
18 | ## along with this program; if not, see <http://www.gnu.org/licenses/>. | |
19 | ## | |
20 | ||
21 | import sys | |
22 | import os | |
23 | import re | |
24 | import struct | |
25 | from array import array | |
26 | ||
27 | import parsepe | |
28 | ||
29 | ||
30 | def find_model(filename): | |
31 | filename = os.path.split(filename)[-1] | |
58c70ebb | 32 | m = re.search('^dso([a-z0-9]+)1.sys$', filename, re.I) |
683fe750 | 33 | if m: |
58c70ebb BV |
34 | model = m.group(1).upper() |
35 | model = model.replace('X86', '').replace('AMD64', '').replace('IA64', '') | |
36 | if model == '520A': | |
37 | model = '5200A' | |
683fe750 BV |
38 | else: |
39 | model = 'unknown' | |
40 | ||
41 | return model | |
42 | ||
43 | ||
44 | def unsparse(data): | |
45 | p = 0 | |
46 | maxaddr = 0 | |
47 | blob = array('B', [0] * 0x4000) | |
48 | while p <= len(data) and data[p+4] == 0: | |
49 | num_bytes = struct.unpack("<H", data[p:p+2])[0] | |
50 | address = struct.unpack("<H", data[p+2:p+4])[0] | |
51 | chunk = array('B') | |
52 | chunk.frombytes(data[p+5:p+5+num_bytes]) | |
53 | p += 22 | |
54 | ||
55 | if address > 0x4000: | |
56 | # the FX2 only has 16K RAM. other writes are to registers | |
57 | # in the 0xe000 region, skip those | |
58 | continue | |
59 | ||
60 | blob[address:address+num_bytes] = chunk | |
61 | ||
62 | if address + num_bytes > maxaddr: | |
63 | maxaddr = address + num_bytes | |
64 | ||
65 | return blob[:maxaddr].tostring() | |
66 | ||
67 | ||
68 | def usage(): | |
231faffa | 69 | print("sigrok-fwextract-hantek-dso <driverfile>") |
683fe750 BV |
70 | sys.exit() |
71 | ||
72 | ||
73 | # | |
74 | # main | |
75 | # | |
76 | ||
77 | if len(sys.argv) != 2: | |
78 | usage() | |
79 | ||
80 | try: | |
81 | filename = sys.argv[1] | |
82 | binihx = parsepe.extract_symbol(filename, '_firmware') | |
83 | if binihx is None: | |
84 | raise Exception("no firmware found") | |
85 | blob = unsparse(binihx) | |
86 | outfile = 'hantek-dso-' + find_model(filename) + '.fw' | |
87 | open(outfile, 'wb').write(blob) | |
88 | print("saved %d bytes to %s" % (len(blob), outfile)) | |
89 | except Exception as e: | |
90 | print("Error: %s" % str(e)) |