micropython-stubber 1.23.1.post1__py3-none-any.whl → 1.23.2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (152) hide show
  1. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.2.dist-info}/LICENSE +30 -30
  2. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.2.dist-info}/METADATA +4 -4
  3. micropython_stubber-1.23.2.dist-info/RECORD +158 -0
  4. mpflash/README.md +220 -220
  5. mpflash/libusb_flash.ipynb +203 -203
  6. mpflash/mpflash/add_firmware.py +98 -98
  7. mpflash/mpflash/ask_input.py +236 -236
  8. mpflash/mpflash/basicgit.py +284 -284
  9. mpflash/mpflash/bootloader/__init__.py +2 -2
  10. mpflash/mpflash/bootloader/activate.py +60 -60
  11. mpflash/mpflash/bootloader/detect.py +82 -82
  12. mpflash/mpflash/bootloader/manual.py +101 -101
  13. mpflash/mpflash/bootloader/micropython.py +12 -12
  14. mpflash/mpflash/bootloader/touch1200.py +36 -36
  15. mpflash/mpflash/cli_download.py +129 -129
  16. mpflash/mpflash/cli_flash.py +224 -216
  17. mpflash/mpflash/cli_group.py +111 -111
  18. mpflash/mpflash/cli_list.py +87 -87
  19. mpflash/mpflash/cli_main.py +39 -39
  20. mpflash/mpflash/common.py +210 -166
  21. mpflash/mpflash/config.py +44 -44
  22. mpflash/mpflash/connected.py +96 -77
  23. mpflash/mpflash/download.py +364 -364
  24. mpflash/mpflash/downloaded.py +130 -130
  25. mpflash/mpflash/errors.py +9 -9
  26. mpflash/mpflash/flash/__init__.py +55 -55
  27. mpflash/mpflash/flash/esp.py +59 -59
  28. mpflash/mpflash/flash/stm32.py +19 -19
  29. mpflash/mpflash/flash/stm32_dfu.py +104 -104
  30. mpflash/mpflash/flash/uf2/__init__.py +88 -88
  31. mpflash/mpflash/flash/uf2/boardid.py +15 -15
  32. mpflash/mpflash/flash/uf2/linux.py +136 -130
  33. mpflash/mpflash/flash/uf2/macos.py +42 -42
  34. mpflash/mpflash/flash/uf2/uf2disk.py +12 -12
  35. mpflash/mpflash/flash/uf2/windows.py +43 -43
  36. mpflash/mpflash/flash/worklist.py +170 -170
  37. mpflash/mpflash/list.py +106 -106
  38. mpflash/mpflash/logger.py +41 -41
  39. mpflash/mpflash/mpboard_id/__init__.py +93 -93
  40. mpflash/mpflash/mpboard_id/add_boards.py +251 -251
  41. mpflash/mpflash/mpboard_id/board.py +37 -37
  42. mpflash/mpflash/mpboard_id/board_id.py +86 -86
  43. mpflash/mpflash/mpboard_id/store.py +43 -43
  44. mpflash/mpflash/mpremoteboard/__init__.py +266 -266
  45. mpflash/mpflash/mpremoteboard/mpy_fw_info.py +141 -141
  46. mpflash/mpflash/mpremoteboard/runner.py +140 -140
  47. mpflash/mpflash/vendor/click_aliases.py +91 -91
  48. mpflash/mpflash/vendor/dfu.py +165 -165
  49. mpflash/mpflash/vendor/pydfu.py +605 -605
  50. mpflash/mpflash/vendor/readme.md +2 -2
  51. mpflash/mpflash/versions.py +135 -135
  52. mpflash/poetry.lock +1599 -1599
  53. mpflash/pyproject.toml +65 -65
  54. mpflash/stm32_udev_rules.md +62 -62
  55. stubber/__init__.py +3 -3
  56. stubber/board/board_info.csv +193 -193
  57. stubber/board/boot.py +34 -34
  58. stubber/board/createstubs.py +1004 -986
  59. stubber/board/createstubs_db.py +826 -825
  60. stubber/board/createstubs_db_min.py +332 -331
  61. stubber/board/createstubs_db_mpy.mpy +0 -0
  62. stubber/board/createstubs_lvgl.py +741 -741
  63. stubber/board/createstubs_lvgl_min.py +741 -741
  64. stubber/board/createstubs_mem.py +767 -766
  65. stubber/board/createstubs_mem_min.py +307 -306
  66. stubber/board/createstubs_mem_mpy.mpy +0 -0
  67. stubber/board/createstubs_min.py +295 -294
  68. stubber/board/createstubs_mpy.mpy +0 -0
  69. stubber/board/fw_info.py +141 -141
  70. stubber/board/info.py +183 -183
  71. stubber/board/main.py +19 -19
  72. stubber/board/modulelist.txt +247 -247
  73. stubber/board/pyrightconfig.json +34 -34
  74. stubber/bulk/mcu_stubber.py +437 -437
  75. stubber/codemod/_partials/__init__.py +48 -48
  76. stubber/codemod/_partials/db_main.py +147 -147
  77. stubber/codemod/_partials/lvgl_main.py +77 -77
  78. stubber/codemod/_partials/modules_reader.py +80 -80
  79. stubber/codemod/add_comment.py +53 -53
  80. stubber/codemod/add_method.py +65 -65
  81. stubber/codemod/board.py +317 -317
  82. stubber/codemod/enrich.py +151 -145
  83. stubber/codemod/merge_docstub.py +284 -284
  84. stubber/codemod/modify_list.py +54 -54
  85. stubber/codemod/utils.py +56 -56
  86. stubber/commands/build_cmd.py +94 -94
  87. stubber/commands/cli.py +49 -49
  88. stubber/commands/clone_cmd.py +78 -78
  89. stubber/commands/config_cmd.py +29 -29
  90. stubber/commands/enrich_folder_cmd.py +71 -71
  91. stubber/commands/get_core_cmd.py +71 -71
  92. stubber/commands/get_docstubs_cmd.py +92 -92
  93. stubber/commands/get_frozen_cmd.py +117 -117
  94. stubber/commands/get_mcu_cmd.py +102 -102
  95. stubber/commands/merge_cmd.py +66 -66
  96. stubber/commands/publish_cmd.py +118 -118
  97. stubber/commands/stub_cmd.py +31 -31
  98. stubber/commands/switch_cmd.py +62 -62
  99. stubber/commands/variants_cmd.py +48 -48
  100. stubber/cst_transformer.py +178 -178
  101. stubber/data/board_info.csv +193 -193
  102. stubber/data/board_info.json +1729 -1729
  103. stubber/data/micropython_tags.csv +15 -15
  104. stubber/data/requirements-core-micropython.txt +38 -38
  105. stubber/data/requirements-core-pycopy.txt +39 -39
  106. stubber/downloader.py +37 -37
  107. stubber/freeze/common.py +72 -72
  108. stubber/freeze/freeze_folder.py +69 -69
  109. stubber/freeze/freeze_manifest_2.py +126 -126
  110. stubber/freeze/get_frozen.py +131 -131
  111. stubber/get_cpython.py +112 -112
  112. stubber/get_lobo.py +59 -59
  113. stubber/minify.py +423 -423
  114. stubber/publish/bump.py +86 -86
  115. stubber/publish/candidates.py +275 -275
  116. stubber/publish/database.py +18 -18
  117. stubber/publish/defaults.py +40 -40
  118. stubber/publish/enums.py +24 -24
  119. stubber/publish/helpers.py +29 -29
  120. stubber/publish/merge_docstubs.py +136 -132
  121. stubber/publish/missing_class_methods.py +51 -51
  122. stubber/publish/package.py +150 -150
  123. stubber/publish/pathnames.py +51 -51
  124. stubber/publish/publish.py +120 -120
  125. stubber/publish/pypi.py +42 -42
  126. stubber/publish/stubpackage.py +1055 -1051
  127. stubber/rst/__init__.py +9 -9
  128. stubber/rst/classsort.py +78 -78
  129. stubber/rst/lookup.py +533 -531
  130. stubber/rst/output_dict.py +401 -401
  131. stubber/rst/reader.py +814 -814
  132. stubber/rst/report_return.py +77 -77
  133. stubber/rst/rst_utils.py +541 -541
  134. stubber/stubber.py +38 -38
  135. stubber/stubs_from_docs.py +90 -90
  136. stubber/tools/manifestfile.py +654 -654
  137. stubber/tools/readme.md +6 -6
  138. stubber/update_fallback.py +117 -117
  139. stubber/update_module_list.py +123 -123
  140. stubber/utils/__init__.py +6 -6
  141. stubber/utils/config.py +137 -137
  142. stubber/utils/makeversionhdr.py +54 -54
  143. stubber/utils/manifest.py +90 -90
  144. stubber/utils/post.py +80 -80
  145. stubber/utils/repos.py +156 -156
  146. stubber/utils/stubmaker.py +139 -139
  147. stubber/utils/typed_config_toml.py +80 -80
  148. stubber/variants.py +106 -106
  149. micropython_stubber-1.23.1.post1.dist-info/RECORD +0 -159
  150. mpflash/basicgit.py +0 -288
  151. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.2.dist-info}/WHEEL +0 -0
  152. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.2.dist-info}/entry_points.txt +0 -0
@@ -1,91 +1,91 @@
1
- # Copyright (c) 2016 Robbin Bonthond
2
-
3
- # Permission is hereby granted, free of charge, to any person obtaining a copy
4
- # of this software and associated documentation files (the "Software"), to deal
5
- # in the Software without restriction, including without limitation the rights
6
- # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7
- # copies of the Software, and to permit persons to whom the Software is
8
- # furnished to do so, subject to the following conditions:
9
-
10
- # The above copyright notice and this permission notice shall be included in all
11
- # copies or substantial portions of the Software.
12
- # ------------------------------------------------------------------------------------
13
- # modified to avoid conflcts with rich_click
14
-
15
- # sourcery skip: assign-if-exp, use-named-expression
16
-
17
- import rich_click as click
18
-
19
- _click7 = click.__version__[0] >= "7"
20
-
21
-
22
- class ClickAliasedGroup(click.RichGroup):
23
- def __init__(self, *args, **kwargs):
24
- super().__init__(*args, **kwargs)
25
- self._commands = {}
26
- self._aliases = {}
27
-
28
- def add_command(self, *args, **kwargs):
29
- aliases = kwargs.pop("aliases", [])
30
- super().add_command(*args, **kwargs)
31
- if aliases:
32
- cmd = args[0]
33
- name = args[1] if len(args) > 1 else None
34
- name = name or cmd.name
35
- if name is None:
36
- raise TypeError("Command has no name.")
37
-
38
- self._commands[name] = aliases
39
- for alias in aliases:
40
- self._aliases[alias] = cmd.name
41
-
42
- def command(self, *args, **kwargs):
43
- aliases = kwargs.pop("aliases", [])
44
- decorator = super().command(*args, **kwargs)
45
- if not aliases:
46
- return decorator
47
-
48
- def _decorator(f):
49
- cmd = decorator(f)
50
- if aliases:
51
- self._commands[cmd.name] = aliases
52
- for alias in aliases:
53
- self._aliases[alias] = cmd.name
54
- return cmd
55
-
56
- return _decorator
57
-
58
- def group(self, *args, **kwargs):
59
- aliases = kwargs.pop("aliases", [])
60
- decorator = super().group(*args, **kwargs)
61
- if not aliases:
62
- return decorator
63
-
64
- def _decorator(f):
65
- cmd = decorator(f)
66
- if aliases:
67
- self._commands[cmd.name] = aliases
68
- for alias in aliases:
69
- self._aliases[alias] = cmd.name
70
- return cmd
71
-
72
- return _decorator
73
-
74
- def resolve_alias(self, cmd_name):
75
- if cmd_name in self._aliases:
76
- return self._aliases[cmd_name]
77
- return cmd_name
78
-
79
- def get_command(self, ctx, cmd_name):
80
- cmd_name = self.resolve_alias(cmd_name)
81
- command = super().get_command(ctx, cmd_name)
82
- if command:
83
- return command
84
- return None
85
-
86
- # def format_commands(self, ctx, formatter):
87
- # TODO: output alias with commands - but that is a significant re-write
88
- # for now add alias to help text
89
-
90
-
91
- # ------------------------------------------------------------------------------------
1
+ # Copyright (c) 2016 Robbin Bonthond
2
+
3
+ # Permission is hereby granted, free of charge, to any person obtaining a copy
4
+ # of this software and associated documentation files (the "Software"), to deal
5
+ # in the Software without restriction, including without limitation the rights
6
+ # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7
+ # copies of the Software, and to permit persons to whom the Software is
8
+ # furnished to do so, subject to the following conditions:
9
+
10
+ # The above copyright notice and this permission notice shall be included in all
11
+ # copies or substantial portions of the Software.
12
+ # ------------------------------------------------------------------------------------
13
+ # modified to avoid conflcts with rich_click
14
+
15
+ # sourcery skip: assign-if-exp, use-named-expression
16
+
17
+ import rich_click as click
18
+
19
+ _click7 = click.__version__[0] >= "7"
20
+
21
+
22
+ class ClickAliasedGroup(click.RichGroup):
23
+ def __init__(self, *args, **kwargs):
24
+ super().__init__(*args, **kwargs)
25
+ self._commands = {}
26
+ self._aliases = {}
27
+
28
+ def add_command(self, *args, **kwargs):
29
+ aliases = kwargs.pop("aliases", [])
30
+ super().add_command(*args, **kwargs)
31
+ if aliases:
32
+ cmd = args[0]
33
+ name = args[1] if len(args) > 1 else None
34
+ name = name or cmd.name
35
+ if name is None:
36
+ raise TypeError("Command has no name.")
37
+
38
+ self._commands[name] = aliases
39
+ for alias in aliases:
40
+ self._aliases[alias] = cmd.name
41
+
42
+ def command(self, *args, **kwargs):
43
+ aliases = kwargs.pop("aliases", [])
44
+ decorator = super().command(*args, **kwargs)
45
+ if not aliases:
46
+ return decorator
47
+
48
+ def _decorator(f):
49
+ cmd = decorator(f)
50
+ if aliases:
51
+ self._commands[cmd.name] = aliases
52
+ for alias in aliases:
53
+ self._aliases[alias] = cmd.name
54
+ return cmd
55
+
56
+ return _decorator
57
+
58
+ def group(self, *args, **kwargs):
59
+ aliases = kwargs.pop("aliases", [])
60
+ decorator = super().group(*args, **kwargs)
61
+ if not aliases:
62
+ return decorator
63
+
64
+ def _decorator(f):
65
+ cmd = decorator(f)
66
+ if aliases:
67
+ self._commands[cmd.name] = aliases
68
+ for alias in aliases:
69
+ self._aliases[alias] = cmd.name
70
+ return cmd
71
+
72
+ return _decorator
73
+
74
+ def resolve_alias(self, cmd_name):
75
+ if cmd_name in self._aliases:
76
+ return self._aliases[cmd_name]
77
+ return cmd_name
78
+
79
+ def get_command(self, ctx, cmd_name):
80
+ cmd_name = self.resolve_alias(cmd_name)
81
+ command = super().get_command(ctx, cmd_name)
82
+ if command:
83
+ return command
84
+ return None
85
+
86
+ # def format_commands(self, ctx, formatter):
87
+ # TODO: output alias with commands - but that is a significant re-write
88
+ # for now add alias to help text
89
+
90
+
91
+ # ------------------------------------------------------------------------------------
@@ -1,165 +1,165 @@
1
- # sourcery skip: require-parameter-annotation
2
- # sourcery skip: replace-interpolation-with-fstring
3
- # type: ignore
4
- #!/usr/bin/python
5
-
6
- # Written by Antonio Galea - 2010/11/18
7
- # Distributed under Gnu LGPL 3.0
8
- # see http://www.gnu.org/licenses/lgpl-3.0.txt
9
-
10
- import os
11
- import struct
12
- import sys
13
- import zlib
14
- from optparse import OptionParser
15
-
16
- DEFAULT_DEVICE = "0x0483:0xdf11"
17
-
18
-
19
- def named(tuple, names):
20
- return dict(zip(names.split(), tuple))
21
-
22
-
23
- def consume(fmt, data, names):
24
- n = struct.calcsize(fmt)
25
- return named(struct.unpack(fmt, data[:n]), names), data[n:]
26
-
27
-
28
- def cstring(string):
29
- return string.split(b"\0", 1)[0]
30
-
31
-
32
- def compute_crc(data):
33
- return 0xFFFFFFFF & -zlib.crc32(data) - 1
34
-
35
-
36
- def parse(file, dump_images=False):
37
- print('File: "%s"' % file)
38
- data = open(file, "rb").read()
39
- crc = compute_crc(data[:-4])
40
- prefix, data = consume("<5sBIB", data, "signature version size targets")
41
- print("%(signature)s v%(version)d, image size: %(size)d, targets: %(targets)d" % prefix)
42
- for t in range(prefix["targets"]):
43
- tprefix, data = consume("<6sBI255s2I", data, "signature altsetting named name size elements")
44
- tprefix["num"] = t
45
- if tprefix["named"]:
46
- tprefix["name"] = cstring(tprefix["name"])
47
- else:
48
- tprefix["name"] = ""
49
- print(
50
- '%(signature)s %(num)d, alt setting: %(altsetting)s, name: "%(name)s", size: %(size)d, elements: %(elements)d'
51
- % tprefix
52
- )
53
- tsize = tprefix["size"]
54
- target, data = data[:tsize], data[tsize:]
55
- for e in range(tprefix["elements"]):
56
- eprefix, target = consume("<2I", target, "address size")
57
- eprefix["num"] = e
58
- print(" %(num)d, address: 0x%(address)08x, size: %(size)d" % eprefix)
59
- esize = eprefix["size"]
60
- image, target = target[:esize], target[esize:]
61
- if dump_images:
62
- out = "%s.target%d.image%d.bin" % (file, t, e)
63
- open(out, "wb").write(image)
64
- print(' DUMPED IMAGE TO "%s"' % out)
65
- if len(target):
66
- print("target %d: PARSE ERROR" % t)
67
- suffix = named(struct.unpack("<4H3sBI", data[:16]), "device product vendor dfu ufd len crc")
68
- print(
69
- "usb: %(vendor)04x:%(product)04x, device: 0x%(device)04x, dfu: 0x%(dfu)04x, %(ufd)s, %(len)d, 0x%(crc)08x"
70
- % suffix
71
- )
72
- if crc != suffix["crc"]:
73
- print("CRC ERROR: computed crc32 is 0x%08x" % crc)
74
- data = data[16:]
75
- if data:
76
- print("PARSE ERROR")
77
-
78
-
79
- def build(file, targets, device=DEFAULT_DEVICE):
80
- data = b""
81
- for _, target in enumerate(targets):
82
- tdata = b""
83
- for image in target:
84
- # pad image to 8 bytes (needed at least for L476)
85
- pad = (8 - len(image["data"]) % 8) % 8
86
- image["data"] = image["data"] + bytes(bytearray(8)[0:pad])
87
- #
88
- tdata += struct.pack("<2I", image["address"], len(image["data"])) + image["data"]
89
- tdata = struct.pack("<6sBI255s2I", b"Target", 0, 1, b"ST...", len(tdata), len(target)) + tdata
90
- data += tdata
91
- data = struct.pack("<5sBIB", b"DfuSe", 1, len(data) + 11, len(targets)) + data
92
- v, d = map(lambda x: int(x, 0) & 0xFFFF, device.split(":", 1))
93
- data += struct.pack("<4H3sB", 0, d, v, 0x011A, b"UFD", 16)
94
- crc = compute_crc(data)
95
- data += struct.pack("<I", crc)
96
- open(file, "wb").write(data)
97
-
98
-
99
- if __name__ == "__main__":
100
- usage = """
101
- %prog [-d|--dump] infile.dfu
102
- %prog {-b|--build} address:file.bin [-b address:file.bin ...] [{-D|--device}=vendor:device] outfile.dfu"""
103
- parser = OptionParser(usage=usage)
104
- parser.add_option(
105
- "-b",
106
- "--build",
107
- action="append",
108
- dest="binfiles",
109
- help="build a DFU file from given BINFILES",
110
- metavar="BINFILES",
111
- )
112
- parser.add_option(
113
- "-D",
114
- "--device",
115
- action="store",
116
- dest="device",
117
- help="build for DEVICE, defaults to %s" % DEFAULT_DEVICE,
118
- metavar="DEVICE",
119
- )
120
- parser.add_option(
121
- "-d",
122
- "--dump",
123
- action="store_true",
124
- dest="dump_images",
125
- default=False,
126
- help="dump contained images to current directory",
127
- )
128
- (options, args) = parser.parse_args()
129
-
130
- if options.binfiles and len(args) == 1:
131
- target = []
132
- for arg in options.binfiles:
133
- try:
134
- address, binfile = arg.split(":", 1)
135
- except ValueError:
136
- print("Address:file couple '%s' invalid." % arg)
137
- sys.exit(1)
138
- try:
139
- address = int(address, 0) & 0xFFFFFFFF
140
- except ValueError:
141
- print("Address %s invalid." % address)
142
- sys.exit(1)
143
- if not os.path.isfile(binfile):
144
- print("Unreadable file '%s'." % binfile)
145
- sys.exit(1)
146
- target.append({"address": address, "data": open(binfile, "rb").read()})
147
- outfile = args[0]
148
- device = DEFAULT_DEVICE
149
- if options.device:
150
- device = options.device
151
- try:
152
- v, d = map(lambda x: int(x, 0) & 0xFFFF, device.split(":", 1))
153
- except:
154
- print("Invalid device '%s'." % device)
155
- sys.exit(1)
156
- build(outfile, [target], device)
157
- elif len(args) == 1:
158
- infile = args[0]
159
- if not os.path.isfile(infile):
160
- print("Unreadable file '%s'." % infile)
161
- sys.exit(1)
162
- parse(infile, dump_images=options.dump_images)
163
- else:
164
- parser.print_help()
165
- sys.exit(1)
1
+ # sourcery skip: require-parameter-annotation
2
+ # sourcery skip: replace-interpolation-with-fstring
3
+ # type: ignore
4
+ #!/usr/bin/python
5
+
6
+ # Written by Antonio Galea - 2010/11/18
7
+ # Distributed under Gnu LGPL 3.0
8
+ # see http://www.gnu.org/licenses/lgpl-3.0.txt
9
+
10
+ import os
11
+ import struct
12
+ import sys
13
+ import zlib
14
+ from optparse import OptionParser
15
+
16
+ DEFAULT_DEVICE = "0x0483:0xdf11"
17
+
18
+
19
+ def named(tuple, names):
20
+ return dict(zip(names.split(), tuple))
21
+
22
+
23
+ def consume(fmt, data, names):
24
+ n = struct.calcsize(fmt)
25
+ return named(struct.unpack(fmt, data[:n]), names), data[n:]
26
+
27
+
28
+ def cstring(string):
29
+ return string.split(b"\0", 1)[0]
30
+
31
+
32
+ def compute_crc(data):
33
+ return 0xFFFFFFFF & -zlib.crc32(data) - 1
34
+
35
+
36
+ def parse(file, dump_images=False):
37
+ print('File: "%s"' % file)
38
+ data = open(file, "rb").read()
39
+ crc = compute_crc(data[:-4])
40
+ prefix, data = consume("<5sBIB", data, "signature version size targets")
41
+ print("%(signature)s v%(version)d, image size: %(size)d, targets: %(targets)d" % prefix)
42
+ for t in range(prefix["targets"]):
43
+ tprefix, data = consume("<6sBI255s2I", data, "signature altsetting named name size elements")
44
+ tprefix["num"] = t
45
+ if tprefix["named"]:
46
+ tprefix["name"] = cstring(tprefix["name"])
47
+ else:
48
+ tprefix["name"] = ""
49
+ print(
50
+ '%(signature)s %(num)d, alt setting: %(altsetting)s, name: "%(name)s", size: %(size)d, elements: %(elements)d'
51
+ % tprefix
52
+ )
53
+ tsize = tprefix["size"]
54
+ target, data = data[:tsize], data[tsize:]
55
+ for e in range(tprefix["elements"]):
56
+ eprefix, target = consume("<2I", target, "address size")
57
+ eprefix["num"] = e
58
+ print(" %(num)d, address: 0x%(address)08x, size: %(size)d" % eprefix)
59
+ esize = eprefix["size"]
60
+ image, target = target[:esize], target[esize:]
61
+ if dump_images:
62
+ out = "%s.target%d.image%d.bin" % (file, t, e)
63
+ open(out, "wb").write(image)
64
+ print(' DUMPED IMAGE TO "%s"' % out)
65
+ if len(target):
66
+ print("target %d: PARSE ERROR" % t)
67
+ suffix = named(struct.unpack("<4H3sBI", data[:16]), "device product vendor dfu ufd len crc")
68
+ print(
69
+ "usb: %(vendor)04x:%(product)04x, device: 0x%(device)04x, dfu: 0x%(dfu)04x, %(ufd)s, %(len)d, 0x%(crc)08x"
70
+ % suffix
71
+ )
72
+ if crc != suffix["crc"]:
73
+ print("CRC ERROR: computed crc32 is 0x%08x" % crc)
74
+ data = data[16:]
75
+ if data:
76
+ print("PARSE ERROR")
77
+
78
+
79
+ def build(file, targets, device=DEFAULT_DEVICE):
80
+ data = b""
81
+ for _, target in enumerate(targets):
82
+ tdata = b""
83
+ for image in target:
84
+ # pad image to 8 bytes (needed at least for L476)
85
+ pad = (8 - len(image["data"]) % 8) % 8
86
+ image["data"] = image["data"] + bytes(bytearray(8)[0:pad])
87
+ #
88
+ tdata += struct.pack("<2I", image["address"], len(image["data"])) + image["data"]
89
+ tdata = struct.pack("<6sBI255s2I", b"Target", 0, 1, b"ST...", len(tdata), len(target)) + tdata
90
+ data += tdata
91
+ data = struct.pack("<5sBIB", b"DfuSe", 1, len(data) + 11, len(targets)) + data
92
+ v, d = map(lambda x: int(x, 0) & 0xFFFF, device.split(":", 1))
93
+ data += struct.pack("<4H3sB", 0, d, v, 0x011A, b"UFD", 16)
94
+ crc = compute_crc(data)
95
+ data += struct.pack("<I", crc)
96
+ open(file, "wb").write(data)
97
+
98
+
99
+ if __name__ == "__main__":
100
+ usage = """
101
+ %prog [-d|--dump] infile.dfu
102
+ %prog {-b|--build} address:file.bin [-b address:file.bin ...] [{-D|--device}=vendor:device] outfile.dfu"""
103
+ parser = OptionParser(usage=usage)
104
+ parser.add_option(
105
+ "-b",
106
+ "--build",
107
+ action="append",
108
+ dest="binfiles",
109
+ help="build a DFU file from given BINFILES",
110
+ metavar="BINFILES",
111
+ )
112
+ parser.add_option(
113
+ "-D",
114
+ "--device",
115
+ action="store",
116
+ dest="device",
117
+ help="build for DEVICE, defaults to %s" % DEFAULT_DEVICE,
118
+ metavar="DEVICE",
119
+ )
120
+ parser.add_option(
121
+ "-d",
122
+ "--dump",
123
+ action="store_true",
124
+ dest="dump_images",
125
+ default=False,
126
+ help="dump contained images to current directory",
127
+ )
128
+ (options, args) = parser.parse_args()
129
+
130
+ if options.binfiles and len(args) == 1:
131
+ target = []
132
+ for arg in options.binfiles:
133
+ try:
134
+ address, binfile = arg.split(":", 1)
135
+ except ValueError:
136
+ print("Address:file couple '%s' invalid." % arg)
137
+ sys.exit(1)
138
+ try:
139
+ address = int(address, 0) & 0xFFFFFFFF
140
+ except ValueError:
141
+ print("Address %s invalid." % address)
142
+ sys.exit(1)
143
+ if not os.path.isfile(binfile):
144
+ print("Unreadable file '%s'." % binfile)
145
+ sys.exit(1)
146
+ target.append({"address": address, "data": open(binfile, "rb").read()})
147
+ outfile = args[0]
148
+ device = DEFAULT_DEVICE
149
+ if options.device:
150
+ device = options.device
151
+ try:
152
+ v, d = map(lambda x: int(x, 0) & 0xFFFF, device.split(":", 1))
153
+ except:
154
+ print("Invalid device '%s'." % device)
155
+ sys.exit(1)
156
+ build(outfile, [target], device)
157
+ elif len(args) == 1:
158
+ infile = args[0]
159
+ if not os.path.isfile(infile):
160
+ print("Unreadable file '%s'." % infile)
161
+ sys.exit(1)
162
+ parse(infile, dump_images=options.dump_images)
163
+ else:
164
+ parser.print_help()
165
+ sys.exit(1)