Anders Blomdell / mbed-sdk-tools

Clone of official tools

targets.py@7:5af61d55adbe, 2016-06-07 (annotated)

Committer:
screamer
Date:
Tue Jun 07 11:21:44 2016 +0100
Revision:
7:5af61d55adbe
Parent:
0:66f3b5499f7f
Child:
13:ab47a20b66f0
Ported changes and config system from workspace tools

Who changed what in which revision?

UserRevisionLine numberNew contents of line
screamer 0:66f3b5499f7f 1 """
screamer 0:66f3b5499f7f 2 mbed SDK
screamer 0:66f3b5499f7f 3 Copyright (c) 2011-2016 ARM Limited
screamer 0:66f3b5499f7f 4
screamer 0:66f3b5499f7f 5 Licensed under the Apache License, Version 2.0 (the "License");
screamer 0:66f3b5499f7f 6 you may not use this file except in compliance with the License.
screamer 0:66f3b5499f7f 7 You may obtain a copy of the License at
screamer 0:66f3b5499f7f 8
screamer 0:66f3b5499f7f 9 http://www.apache.org/licenses/LICENSE-2.0
screamer 0:66f3b5499f7f 10
screamer 0:66f3b5499f7f 11 Unless required by applicable law or agreed to in writing, software
screamer 0:66f3b5499f7f 12 distributed under the License is distributed on an "AS IS" BASIS,
screamer 0:66f3b5499f7f 13 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
screamer 0:66f3b5499f7f 14 See the License for the specific language governing permissions and
screamer 0:66f3b5499f7f 15 limitations under the License.
screamer 0:66f3b5499f7f 16 """
screamer 0:66f3b5499f7f 17
screamer 0:66f3b5499f7f 18 CORE_LABELS = {
screamer 0:66f3b5499f7f 19 "ARM7TDMI-S": ["ARM7", "LIKE_CORTEX_ARM7"],
screamer 0:66f3b5499f7f 20 "Cortex-M0" : ["M0", "CORTEX_M", "LIKE_CORTEX_M0"],
screamer 0:66f3b5499f7f 21 "Cortex-M0+": ["M0P", "CORTEX_M", "LIKE_CORTEX_M0"],
screamer 0:66f3b5499f7f 22 "Cortex-M1" : ["M1", "CORTEX_M", "LIKE_CORTEX_M1"],
screamer 0:66f3b5499f7f 23 "Cortex-M3" : ["M3", "CORTEX_M", "LIKE_CORTEX_M3"],
screamer 0:66f3b5499f7f 24 "Cortex-M4" : ["M4", "CORTEX_M", "RTOS_M4_M7", "LIKE_CORTEX_M4"],
screamer 0:66f3b5499f7f 25 "Cortex-M4F" : ["M4", "CORTEX_M", "RTOS_M4_M7", "LIKE_CORTEX_M4"],
screamer 0:66f3b5499f7f 26 "Cortex-M7" : ["M7", "CORTEX_M", "RTOS_M4_M7", "LIKE_CORTEX_M7"],
screamer 0:66f3b5499f7f 27 "Cortex-M7F" : ["M7", "CORTEX_M", "RTOS_M4_M7", "LIKE_CORTEX_M7"],
screamer 0:66f3b5499f7f 28 "Cortex-A9" : ["A9", "CORTEX_A", "LIKE_CORTEX_A9"]
screamer 0:66f3b5499f7f 29 }
screamer 0:66f3b5499f7f 30
screamer 0:66f3b5499f7f 31 import os
screamer 0:66f3b5499f7f 32 import binascii
screamer 0:66f3b5499f7f 33 import struct
screamer 0:66f3b5499f7f 34 import shutil
screamer 0:66f3b5499f7f 35 from tools.patch import patch
screamer 0:66f3b5499f7f 36 from paths import TOOLS_BOOTLOADERS
screamer 7:5af61d55adbe 37 import json
screamer 7:5af61d55adbe 38 import inspect
screamer 7:5af61d55adbe 39 import sys
screamer 7:5af61d55adbe 40 from tools.utils import json_file_to_dict
screamer 7:5af61d55adbe 41
screamer 7:5af61d55adbe 42 ########################################################################################################################
screamer 7:5af61d55adbe 43 # Generic Target class that reads and interprets the data in targets.json
screamer 7:5af61d55adbe 44
screamer 7:5af61d55adbe 45 # A simple class that represents all the exceptions associated with hooking
screamer 7:5af61d55adbe 46 class HookError(Exception):
screamer 7:5af61d55adbe 47 pass
screamer 7:5af61d55adbe 48
screamer 7:5af61d55adbe 49 # A simple decorator used for automatically caching data returned by a function
screamer 7:5af61d55adbe 50 caches = {}
screamer 7:5af61d55adbe 51 def cached(func):
screamer 7:5af61d55adbe 52 def wrapper(*args, **kwargs):
screamer 7:5af61d55adbe 53 if not caches.has_key(func):
screamer 7:5af61d55adbe 54 caches[func] = func(*args, **kwargs)
screamer 7:5af61d55adbe 55 return caches[func]
screamer 7:5af61d55adbe 56 return wrapper
screamer 0:66f3b5499f7f 57
screamer 0:66f3b5499f7f 58 class Target:
screamer 7:5af61d55adbe 59 # Cumulative attributes can have values appended to them, so they
screamer 7:5af61d55adbe 60 # need to be computed differently than regular attributes
screamer 7:5af61d55adbe 61 __cumulative_attributes = ['extra_labels', 'macros', 'features']
screamer 7:5af61d55adbe 62
screamer 7:5af61d55adbe 63 # {target_name: target_instance} map for all the targets in the system
screamer 7:5af61d55adbe 64 __target_map = {}
screamer 7:5af61d55adbe 65
screamer 7:5af61d55adbe 66 # List of targets that were added dynamically using "add_py_targets" (see below)
screamer 7:5af61d55adbe 67 __py_targets = set()
screamer 7:5af61d55adbe 68
screamer 7:5af61d55adbe 69 # Load the description of JSON target data
screamer 7:5af61d55adbe 70 @staticmethod
screamer 7:5af61d55adbe 71 @cached
screamer 7:5af61d55adbe 72 def get_json_target_data():
screamer 7:5af61d55adbe 73 return json_file_to_dict(os.path.join(os.path.dirname(os.path.abspath(__file__)), "targets.json"))
screamer 7:5af61d55adbe 74
screamer 7:5af61d55adbe 75 # Get the members of this module using Python's "inspect" module
screamer 7:5af61d55adbe 76 @staticmethod
screamer 7:5af61d55adbe 77 @cached
screamer 7:5af61d55adbe 78 def get_module_data():
screamer 7:5af61d55adbe 79 return dict([(m[0], m[1]) for m in inspect.getmembers(sys.modules[__name__])])
screamer 0:66f3b5499f7f 80
screamer 7:5af61d55adbe 81 # Return the order in which target descriptions are searched for attributes
screamer 7:5af61d55adbe 82 # This mimics the Python 2.2 method resolution order, which is what the old targets.py module used
screamer 7:5af61d55adbe 83 # For more details, check http://makina-corpus.com/blog/metier/2014/python-tutorial-understanding-python-mro-class-search-path
screamer 7:5af61d55adbe 84 # The resolution order contains (name, level) tuples, where "name" is the name of the class and "level"
screamer 7:5af61d55adbe 85 # is the level in the inheritance hierarchy (the target itself is at level 0, its first parent at level 1,
screamer 7:5af61d55adbe 86 # its parent's parent at level 1 and so on)
screamer 7:5af61d55adbe 87 def __get_resolution_order(self, target_name, order, level = 0):
screamer 7:5af61d55adbe 88 if not target_name in [l[0] for l in order]: # the resolution order can't contain duplicate target names
screamer 7:5af61d55adbe 89 order.append((target_name, level))
screamer 7:5af61d55adbe 90 parents = self.get_json_target_data()[target_name].get("inherits", [])
screamer 7:5af61d55adbe 91 for p in parents:
screamer 7:5af61d55adbe 92 order = self.__get_resolution_order(p, order, level + 1)
screamer 7:5af61d55adbe 93 return order
screamer 0:66f3b5499f7f 94
screamer 7:5af61d55adbe 95 # Modify the exporter specification ("progen") by changing all "template" keys to full paths
screamer 7:5af61d55adbe 96 @staticmethod
screamer 7:5af61d55adbe 97 def __add_paths_to_progen(data):
screamer 7:5af61d55adbe 98 out = {}
screamer 7:5af61d55adbe 99 for key, value in data.items():
screamer 7:5af61d55adbe 100 if isinstance(value, dict):
screamer 7:5af61d55adbe 101 out[key] = Target.__add_paths_to_progen(value)
screamer 7:5af61d55adbe 102 elif key == "template":
screamer 7:5af61d55adbe 103 out[key] = [os.path.join(os.path.dirname(__file__), 'export', v) for v in value]
screamer 7:5af61d55adbe 104 else:
screamer 7:5af61d55adbe 105 out[key] = value
screamer 7:5af61d55adbe 106 return out
screamer 0:66f3b5499f7f 107
screamer 7:5af61d55adbe 108 # Comute the value of a given target attribute
screamer 7:5af61d55adbe 109 def __getattr_helper(self, attrname):
screamer 7:5af61d55adbe 110 tdata = self.get_json_target_data()
screamer 7:5af61d55adbe 111 if attrname in self.__cumulative_attributes:
screamer 7:5af61d55adbe 112 # For a cumulative attribute, figure out when it was defined the last time (in attribute
screamer 7:5af61d55adbe 113 # resolution order) then follow the "_add" and "_remove" data fields
screamer 7:5af61d55adbe 114 for idx, t in enumerate(self.resolution_order):
screamer 7:5af61d55adbe 115 if attrname in tdata[t[0]]: # the attribute was defined at this level in the resolution order
screamer 7:5af61d55adbe 116 def_idx = idx
screamer 7:5af61d55adbe 117 break
screamer 7:5af61d55adbe 118 else:
screamer 7:5af61d55adbe 119 raise AttributeError("Attribute '%s' not found in target '%s'" % (attrname, self.name))
screamer 7:5af61d55adbe 120 # Get the starting value of the attribute
screamer 7:5af61d55adbe 121 v = (tdata[self.resolution_order[def_idx][0]][attrname] or [])[:]
screamer 7:5af61d55adbe 122 # Traverse the resolution list in high inheritance to low inheritance level, left to right order
screamer 7:5af61d55adbe 123 # to figure out all the other classes that change the definition by adding or removing elements
screamer 7:5af61d55adbe 124 for idx in xrange(self.resolution_order[def_idx][1] - 1, -1, -1):
screamer 7:5af61d55adbe 125 same_level_targets = [t[0] for t in self.resolution_order if t[1] == idx]
screamer 7:5af61d55adbe 126 for t in same_level_targets:
screamer 7:5af61d55adbe 127 data = tdata[t]
screamer 7:5af61d55adbe 128 # Do we have anything to add ?
screamer 7:5af61d55adbe 129 if data.has_key(attrname + "_add"):
screamer 7:5af61d55adbe 130 v.extend(data[attrname + "_add"])
screamer 7:5af61d55adbe 131 # Do we have anything to remove ?
screamer 7:5af61d55adbe 132 if data.has_key(attrname + "_remove"):
screamer 7:5af61d55adbe 133 # Macros can be defined either without a value (MACRO) or with a value (MACRO=10).
screamer 7:5af61d55adbe 134 # When removing, we specify only the name of the macro, without the value. So we need
screamer 7:5af61d55adbe 135 # to create a mapping between the macro name and its value. This will work for
screamer 7:5af61d55adbe 136 # extra_labels and other type of arrays as well, since they fall into the "macros
screamer 7:5af61d55adbe 137 # without a value" category (simple definitions without a value).
screamer 7:5af61d55adbe 138 name_def_map = {}
screamer 7:5af61d55adbe 139 for crtv in v:
screamer 7:5af61d55adbe 140 if crtv.find('=') != -1:
screamer 7:5af61d55adbe 141 temp = crtv.split('=')
screamer 7:5af61d55adbe 142 if len(temp) != 2:
screamer 7:5af61d55adbe 143 raise ValueError("Invalid macro definition '%s'" % crtv)
screamer 7:5af61d55adbe 144 name_def_map[temp[0]] = crtv
screamer 7:5af61d55adbe 145 else:
screamer 7:5af61d55adbe 146 name_def_map[crtv] = crtv
screamer 7:5af61d55adbe 147 for e in data[attrname + "_remove"]:
screamer 7:5af61d55adbe 148 if not e in name_def_map:
screamer 7:5af61d55adbe 149 raise ValueError("Unable to remove '%s' in '%s.%s' since it doesn't exist" % (e, self.name, attrname))
screamer 7:5af61d55adbe 150 v.remove(name_def_map[e])
screamer 7:5af61d55adbe 151 return v
screamer 7:5af61d55adbe 152 # Look for the attribute in the class and its parents, as defined by the resolution order
screamer 7:5af61d55adbe 153 v = None
screamer 7:5af61d55adbe 154 for t in self.resolution_order:
screamer 7:5af61d55adbe 155 data = tdata[t[0]]
screamer 7:5af61d55adbe 156 if data.has_key(attrname):
screamer 7:5af61d55adbe 157 v = data[attrname]
screamer 7:5af61d55adbe 158 break
screamer 7:5af61d55adbe 159 else: # Attribute not found
screamer 7:5af61d55adbe 160 raise AttributeError("Attribute '%s' not found in target '%s'" % (attrname, self.name))
screamer 7:5af61d55adbe 161 # 'progen' needs the full path to the template (the path in JSON is relative to tools/export)
screamer 7:5af61d55adbe 162 return v if attrname != "progen" else self.__add_paths_to_progen(v)
screamer 0:66f3b5499f7f 163
screamer 7:5af61d55adbe 164 # Return the value of an attribute
screamer 7:5af61d55adbe 165 # This function only computes the attribute's value once, then adds it to the instance attributes
screamer 7:5af61d55adbe 166 # (in __dict__), so the next time it is returned directly
screamer 7:5af61d55adbe 167 def __getattr__(self, attrname):
screamer 7:5af61d55adbe 168 v = self.__getattr_helper(attrname)
screamer 7:5af61d55adbe 169 self.__dict__[attrname] = v
screamer 7:5af61d55adbe 170 return v
screamer 0:66f3b5499f7f 171
screamer 7:5af61d55adbe 172 # Add one or more new target(s) represented as a Python dictionary in 'new_targets'
screamer 7:5af61d55adbe 173 # It it an error to add a target with a name that exists in "targets.json"
screamer 7:5af61d55adbe 174 # However, it is OK to add a target that was previously added via "add_py_targets"
screamer 7:5af61d55adbe 175 # (this makes testing easier without changing the regular semantics)
screamer 7:5af61d55adbe 176 @staticmethod
screamer 7:5af61d55adbe 177 def add_py_targets(new_targets):
screamer 7:5af61d55adbe 178 crt_data = Target.get_json_target_data()
screamer 7:5af61d55adbe 179 # First add all elemnts to the internal dictionary
screamer 7:5af61d55adbe 180 for tk, tv in new_targets.items():
screamer 7:5af61d55adbe 181 if crt_data.has_key(tk) and (not tk in Target.__py_targets):
screamer 7:5af61d55adbe 182 raise Exception("Attempt to add target '%s' that already exists" % tk)
screamer 7:5af61d55adbe 183 crt_data[tk] = tv
screamer 7:5af61d55adbe 184 Target.__py_targets.add(tk)
screamer 7:5af61d55adbe 185 # Then create the new instances and update global variables if needed
screamer 7:5af61d55adbe 186 for tk, tv in new_targets.items():
screamer 7:5af61d55adbe 187 # Is the target already created?
screamer 7:5af61d55adbe 188 old_target = Target.__target_map.get(tk, None)
screamer 7:5af61d55adbe 189 # Instantiate this target. If it is public, update the data in
screamer 7:5af61d55adbe 190 # in TARGETS, TARGET_MAP, TARGET_NAMES
screamer 7:5af61d55adbe 191 new_target = Target(tk)
screamer 7:5af61d55adbe 192 if tv.get("public", True):
screamer 7:5af61d55adbe 193 if old_target: # remove the old target from TARGETS and TARGET_NAMES
screamer 7:5af61d55adbe 194 TARGETS.remove(old_target)
screamer 7:5af61d55adbe 195 TARGET_NAMES.remove(tk)
screamer 7:5af61d55adbe 196 # Add the new target
screamer 7:5af61d55adbe 197 TARGETS.append(new_target)
screamer 7:5af61d55adbe 198 TARGET_MAP[tk] = new_target
screamer 7:5af61d55adbe 199 TARGET_NAMES.append(tk)
screamer 7:5af61d55adbe 200 # Update the target cache
screamer 7:5af61d55adbe 201 Target.__target_map[tk] = new_target
screamer 0:66f3b5499f7f 202
screamer 7:5af61d55adbe 203 # Return the target instance starting from the target name
screamer 7:5af61d55adbe 204 @staticmethod
screamer 7:5af61d55adbe 205 def get_target(name):
screamer 7:5af61d55adbe 206 if not Target.__target_map.has_key(name):
screamer 7:5af61d55adbe 207 Target.__target_map[name] = Target(name)
screamer 7:5af61d55adbe 208 return Target.__target_map[name]
screamer 7:5af61d55adbe 209
screamer 7:5af61d55adbe 210 def __init__(self, name):
screamer 7:5af61d55adbe 211 self.name = name
screamer 7:5af61d55adbe 212
screamer 7:5af61d55adbe 213 # Compute resolution order once (it will be used later in __getattr__)
screamer 7:5af61d55adbe 214 self.resolution_order = self.__get_resolution_order(self.name, [])
screamer 7:5af61d55adbe 215 # Create also a list with only the names of the targets in the resolution order
screamer 7:5af61d55adbe 216 self.resolution_order_names = [t[0] for t in self.resolution_order]
screamer 0:66f3b5499f7f 217
screamer 0:66f3b5499f7f 218 def program_cycle_s(self):
screamer 7:5af61d55adbe 219 try:
screamer 7:5af61d55adbe 220 return self.__getattr__("program_cycle_s")
screamer 7:5af61d55adbe 221 except AttributeError:
screamer 7:5af61d55adbe 222 return 4 if self.is_disk_virtual else 1.5
screamer 0:66f3b5499f7f 223
screamer 0:66f3b5499f7f 224 def get_labels(self):
screamer 0:66f3b5499f7f 225 return [self.name] + CORE_LABELS[self.core] + self.extra_labels
screamer 0:66f3b5499f7f 226
screamer 7:5af61d55adbe 227 # For now, this function only allows "post binary" hooks (hooks that are executed after
screamer 7:5af61d55adbe 228 # the binary image is extracted from the executable file)
screamer 0:66f3b5499f7f 229 def init_hooks(self, hook, toolchain_name):
screamer 7:5af61d55adbe 230 # If there's no hook, simply return
screamer 7:5af61d55adbe 231 try:
screamer 7:5af61d55adbe 232 hook_data = self.post_binary_hook
screamer 7:5af61d55adbe 233 except AttributeError:
screamer 7:5af61d55adbe 234 return
screamer 7:5af61d55adbe 235 # A hook was found. The hook's name is in the format "classname.functionname"
screamer 7:5af61d55adbe 236 temp = hook_data["function"].split(".")
screamer 7:5af61d55adbe 237 if len(temp) != 2:
screamer 7:5af61d55adbe 238 raise HookError("Invalid format for hook '%s' in target '%s' (must be 'class_name.function_name')" % (hook_data["function"], self.name))
screamer 7:5af61d55adbe 239 class_name, function_name = temp[0], temp[1]
screamer 7:5af61d55adbe 240 # "class_name" must refer to a class in this file, so check if the class exists
screamer 7:5af61d55adbe 241 mdata = self.get_module_data()
screamer 7:5af61d55adbe 242 if not mdata.has_key(class_name) or not inspect.isclass(mdata[class_name]):
screamer 7:5af61d55adbe 243 raise HookError("Class '%s' required by '%s' in target '%s' not found in targets.py" % (class_name, hook_data["function"], self.name))
screamer 7:5af61d55adbe 244 # "function_name" must refer to a static function inside class "class_name"
screamer 7:5af61d55adbe 245 cls = mdata[class_name]
screamer 7:5af61d55adbe 246 if (not hasattr(cls, function_name)) or (not inspect.isfunction(getattr(cls, function_name))):
screamer 7:5af61d55adbe 247 raise HookError("Static function '%s' required by '%s' in target '%s' not found in class '%s'" % (function_name, hook_data["function"], self.name, class_name))
screamer 7:5af61d55adbe 248 # Check if the hook specification also has target restrictions
screamer 7:5af61d55adbe 249 toolchain_restrictions = hook_data.get("toolchains", [])
screamer 7:5af61d55adbe 250 if toolchain_restrictions and (toolchain_name not in toolchain_restrictions):
screamer 7:5af61d55adbe 251 return
screamer 7:5af61d55adbe 252 # Finally, hook the requested function
screamer 7:5af61d55adbe 253 hook.hook_add_binary("post", getattr(cls, function_name))
screamer 0:66f3b5499f7f 254
screamer 7:5af61d55adbe 255 ########################################################################################################################
screamer 7:5af61d55adbe 256 # Target specific code goes in this section
screamer 7:5af61d55adbe 257 # This code can be invoked from the target description using the "post_binary_hook" key
screamer 0:66f3b5499f7f 258
screamer 7:5af61d55adbe 259 class LPCTargetCode:
screamer 0:66f3b5499f7f 260 @staticmethod
screamer 0:66f3b5499f7f 261 def lpc_patch(t_self, resources, elf, binf):
screamer 0:66f3b5499f7f 262 t_self.debug("LPC Patch: %s" % os.path.split(binf)[1])
screamer 0:66f3b5499f7f 263 patch(binf)
screamer 0:66f3b5499f7f 264
screamer 7:5af61d55adbe 265 class LPC4088Code:
screamer 0:66f3b5499f7f 266 @staticmethod
screamer 0:66f3b5499f7f 267 def binary_hook(t_self, resources, elf, binf):
screamer 0:66f3b5499f7f 268 if not os.path.isdir(binf):
screamer 0:66f3b5499f7f 269 # Regular binary file, nothing to do
screamer 7:5af61d55adbe 270 LPCTargetCode.lpc_patch(t_self, resources, elf, binf)
screamer 0:66f3b5499f7f 271 return
screamer 0:66f3b5499f7f 272 outbin = open(binf + ".temp", "wb")
screamer 0:66f3b5499f7f 273 partf = open(os.path.join(binf, "ER_IROM1"), "rb")
screamer 0:66f3b5499f7f 274 # Pad the fist part (internal flash) with 0xFF to 512k
screamer 0:66f3b5499f7f 275 data = partf.read()
screamer 0:66f3b5499f7f 276 outbin.write(data)
screamer 0:66f3b5499f7f 277 outbin.write('\xFF' * (512*1024 - len(data)))
screamer 0:66f3b5499f7f 278 partf.close()
screamer 0:66f3b5499f7f 279 # Read and append the second part (external flash) in chunks of fixed size
screamer 0:66f3b5499f7f 280 chunksize = 128 * 1024
screamer 0:66f3b5499f7f 281 partf = open(os.path.join(binf, "ER_IROM2"), "rb")
screamer 0:66f3b5499f7f 282 while True:
screamer 0:66f3b5499f7f 283 data = partf.read(chunksize)
screamer 0:66f3b5499f7f 284 outbin.write(data)
screamer 0:66f3b5499f7f 285 if len(data) < chunksize:
screamer 0:66f3b5499f7f 286 break
screamer 0:66f3b5499f7f 287 partf.close()
screamer 0:66f3b5499f7f 288 outbin.close()
screamer 0:66f3b5499f7f 289 # Remove the directory with the binary parts and rename the temporary
screamer 0:66f3b5499f7f 290 # file to 'binf'
screamer 0:66f3b5499f7f 291 shutil.rmtree(binf, True)
screamer 0:66f3b5499f7f 292 os.rename(binf + '.temp', binf)
screamer 0:66f3b5499f7f 293 t_self.debug("Generated custom binary file (internal flash + SPIFI)")
screamer 7:5af61d55adbe 294 LPCTargetCode.lpc_patch(t_self, resources, elf, binf)
screamer 0:66f3b5499f7f 295
screamer 7:5af61d55adbe 296 class TEENSY3_1Code:
screamer 0:66f3b5499f7f 297 @staticmethod
screamer 0:66f3b5499f7f 298 def binary_hook(t_self, resources, elf, binf):
screamer 0:66f3b5499f7f 299 from intelhex import IntelHex
screamer 0:66f3b5499f7f 300 binh = IntelHex()
screamer 0:66f3b5499f7f 301 binh.loadbin(binf, offset = 0)
screamer 0:66f3b5499f7f 302
screamer 0:66f3b5499f7f 303 with open(binf.replace(".bin", ".hex"), "w") as f:
screamer 0:66f3b5499f7f 304 binh.tofile(f, format='hex')
screamer 0:66f3b5499f7f 305
screamer 7:5af61d55adbe 306 class MTSCode:
screamer 0:66f3b5499f7f 307 @staticmethod
screamer 7:5af61d55adbe 308 def _combine_bins_helper(target_name, t_self, resources, elf, binf):
screamer 7:5af61d55adbe 309 loader = os.path.join(TOOLS_BOOTLOADERS, target_name, "bootloader.bin")
screamer 0:66f3b5499f7f 310 target = binf + ".tmp"
screamer 0:66f3b5499f7f 311 if not os.path.exists(loader):
screamer 0:66f3b5499f7f 312 print "Can't find bootloader binary: " + loader
screamer 0:66f3b5499f7f 313 return
screamer 0:66f3b5499f7f 314 outbin = open(target, 'w+b')
screamer 0:66f3b5499f7f 315 part = open(loader, 'rb')
screamer 0:66f3b5499f7f 316 data = part.read()
screamer 0:66f3b5499f7f 317 outbin.write(data)
screamer 0:66f3b5499f7f 318 outbin.write('\xFF' * (64*1024 - len(data)))
screamer 0:66f3b5499f7f 319 part.close()
screamer 0:66f3b5499f7f 320 part = open(binf, 'rb')
screamer 0:66f3b5499f7f 321 data = part.read()
screamer 0:66f3b5499f7f 322 outbin.write(data)
screamer 0:66f3b5499f7f 323 part.close()
screamer 0:66f3b5499f7f 324 outbin.seek(0, 0)
screamer 0:66f3b5499f7f 325 data = outbin.read()
screamer 0:66f3b5499f7f 326 outbin.seek(0, 1)
screamer 0:66f3b5499f7f 327 crc = struct.pack('<I', binascii.crc32(data) & 0xFFFFFFFF)
screamer 0:66f3b5499f7f 328 outbin.write(crc)
screamer 0:66f3b5499f7f 329 outbin.close()
screamer 0:66f3b5499f7f 330 os.remove(binf)
screamer 0:66f3b5499f7f 331 os.rename(target, binf)
screamer 0:66f3b5499f7f 332
screamer 0:66f3b5499f7f 333 @staticmethod
screamer 7:5af61d55adbe 334 def combine_bins_mts_dot(t_self, resources, elf, binf):
screamer 7:5af61d55adbe 335 MTSCode._combine_bins_helper("MTS_MDOT_F411RE", t_self, resources, elf, binf)
screamer 0:66f3b5499f7f 336
screamer 7:5af61d55adbe 337 @staticmethod
screamer 7:5af61d55adbe 338 def combine_bins_mts_dragonfly(t_self, resources, elf, binf):
screamer 7:5af61d55adbe 339 MTSCode._combine_bins_helper("MTS_DRAGONFLY_F411RE", t_self, resources, elf, binf)
screamer 0:66f3b5499f7f 340
screamer 7:5af61d55adbe 341 class MCU_NRF51Code:
screamer 0:66f3b5499f7f 342 @staticmethod
screamer 0:66f3b5499f7f 343 def binary_hook(t_self, resources, elf, binf):
screamer 0:66f3b5499f7f 344
screamer 0:66f3b5499f7f 345 # Scan to find the actual paths of soft device
screamer 0:66f3b5499f7f 346 sdf = None
screamer 0:66f3b5499f7f 347 for softdeviceAndOffsetEntry in t_self.target.EXPECTED_SOFTDEVICES_WITH_OFFSETS:
screamer 0:66f3b5499f7f 348 for hexf in resources.hex_files:
screamer 0:66f3b5499f7f 349 if hexf.find(softdeviceAndOffsetEntry['name']) != -1:
screamer 0:66f3b5499f7f 350 t_self.debug("SoftDevice file found %s." % softdeviceAndOffsetEntry['name'])
screamer 0:66f3b5499f7f 351 sdf = hexf
screamer 0:66f3b5499f7f 352
screamer 0:66f3b5499f7f 353 if sdf is not None: break
screamer 0:66f3b5499f7f 354 if sdf is not None: break
screamer 0:66f3b5499f7f 355
screamer 0:66f3b5499f7f 356 if sdf is None:
screamer 0:66f3b5499f7f 357 t_self.debug("Hex file not found. Aborting.")
screamer 0:66f3b5499f7f 358 return
screamer 0:66f3b5499f7f 359
screamer 0:66f3b5499f7f 360 # Look for bootloader file that matches this soft device or bootloader override image
screamer 0:66f3b5499f7f 361 blf = None
screamer 0:66f3b5499f7f 362 if t_self.target.MERGE_BOOTLOADER is True:
screamer 0:66f3b5499f7f 363 for hexf in resources.hex_files:
screamer 0:66f3b5499f7f 364 if hexf.find(t_self.target.OVERRIDE_BOOTLOADER_FILENAME) != -1:
screamer 0:66f3b5499f7f 365 t_self.debug("Bootloader file found %s." % t_self.target.OVERRIDE_BOOTLOADER_FILENAME)
screamer 0:66f3b5499f7f 366 blf = hexf
screamer 0:66f3b5499f7f 367 break
screamer 0:66f3b5499f7f 368 elif hexf.find(softdeviceAndOffsetEntry['boot']) != -1:
screamer 0:66f3b5499f7f 369 t_self.debug("Bootloader file found %s." % softdeviceAndOffsetEntry['boot'])
screamer 0:66f3b5499f7f 370 blf = hexf
screamer 0:66f3b5499f7f 371 break
screamer 0:66f3b5499f7f 372
screamer 0:66f3b5499f7f 373 # Merge user code with softdevice
screamer 0:66f3b5499f7f 374 from intelhex import IntelHex
screamer 0:66f3b5499f7f 375 binh = IntelHex()
screamer 0:66f3b5499f7f 376 binh.loadbin(binf, offset=softdeviceAndOffsetEntry['offset'])
screamer 0:66f3b5499f7f 377
screamer 0:66f3b5499f7f 378 if t_self.target.MERGE_SOFT_DEVICE is True:
screamer 0:66f3b5499f7f 379 t_self.debug("Merge SoftDevice file %s" % softdeviceAndOffsetEntry['name'])
screamer 0:66f3b5499f7f 380 sdh = IntelHex(sdf)
screamer 0:66f3b5499f7f 381 binh.merge(sdh)
screamer 0:66f3b5499f7f 382
screamer 0:66f3b5499f7f 383 if t_self.target.MERGE_BOOTLOADER is True and blf is not None:
screamer 0:66f3b5499f7f 384 t_self.debug("Merge BootLoader file %s" % blf)
screamer 0:66f3b5499f7f 385 blh = IntelHex(blf)
screamer 0:66f3b5499f7f 386 binh.merge(blh)
screamer 0:66f3b5499f7f 387
screamer 0:66f3b5499f7f 388 with open(binf.replace(".bin", ".hex"), "w") as f:
screamer 0:66f3b5499f7f 389 binh.tofile(f, format='hex')
screamer 0:66f3b5499f7f 390
screamer 7:5af61d55adbe 391 ########################################################################################################################
screamer 0:66f3b5499f7f 392
screamer 7:5af61d55adbe 393 # Instantiate all public targets
screamer 7:5af61d55adbe 394 TARGETS = [Target.get_target(name) for name, value in Target.get_json_target_data().items() if value.get("public", True)]
screamer 0:66f3b5499f7f 395
screamer 0:66f3b5499f7f 396 # Map each target name to its unique instance
screamer 7:5af61d55adbe 397 TARGET_MAP = dict([(t.name, t) for t in TARGETS])
screamer 0:66f3b5499f7f 398
screamer 0:66f3b5499f7f 399 TARGET_NAMES = TARGET_MAP.keys()
screamer 0:66f3b5499f7f 400
screamer 0:66f3b5499f7f 401 # Some targets with different name have the same exporters
screamer 7:5af61d55adbe 402 EXPORT_MAP = {}
screamer 0:66f3b5499f7f 403
screamer 0:66f3b5499f7f 404 # Detection APIs
screamer 0:66f3b5499f7f 405 def get_target_detect_codes():
screamer 0:66f3b5499f7f 406 """ Returns dictionary mapping detect_code -> platform_name
screamer 0:66f3b5499f7f 407 """
screamer 0:66f3b5499f7f 408 result = {}
screamer 0:66f3b5499f7f 409 for target in TARGETS:
screamer 0:66f3b5499f7f 410 for detect_code in target.detect_code:
screamer 0:66f3b5499f7f 411 result[detect_code] = target.name
screamer 0:66f3b5499f7f 412 return result