# Copyright (c) 2017 Linaro Limited. # # SPDX-License-Identifier: Apache-2.0 '''Runner for pyOCD .''' from functools import partial import os from os import path from runners.core import ZephyrBinaryRunner, RunnerCaps, \ BuildConfiguration, depr_action DEFAULT_PYOCD_GDB_PORT = 3333 DEFAULT_PYOCD_TELNET_PORT = 4444 class PyOcdBinaryRunner(ZephyrBinaryRunner): '''Runner front-end for pyOCD.''' def __init__(self, cfg, target, pyocd='pyocd', dev_id=None, flash_addr=0x0, erase=False, flash_opts=None, gdb_port=DEFAULT_PYOCD_GDB_PORT, telnet_port=DEFAULT_PYOCD_TELNET_PORT, tui=False, pyocd_config=None, daparg=None, frequency=None, tool_opt=None): super().__init__(cfg) default = path.join(cfg.board_dir, 'support', 'pyocd.yaml') if path.exists(default): self.pyocd_config = default else: self.pyocd_config = None self.target_args = ['-t', target] self.pyocd = pyocd self.flash_addr_args = ['-a', hex(flash_addr)] if flash_addr else [] self.erase = erase self.gdb_cmd = [cfg.gdb] if cfg.gdb is not None else None self.gdb_port = gdb_port self.telnet_port = telnet_port self.tui_args = ['-tui'] if tui else [] self.hex_name = cfg.hex_file self.bin_name = cfg.bin_file self.elf_name = cfg.elf_file pyocd_config_args = [] if self.pyocd_config is not None: pyocd_config_args = ['--config', self.pyocd_config] self.pyocd_config_args = pyocd_config_args board_args = [] if dev_id is not None: board_args = ['-u', dev_id] self.board_args = board_args daparg_args = [] if daparg is not None: daparg_args = ['-da', daparg] self.daparg_args = daparg_args frequency_args = [] if frequency is not None: frequency_args = ['-f', frequency] self.frequency_args = frequency_args self.tool_opt_args = tool_opt or [] self.flash_extra = flash_opts if flash_opts else [] @classmethod def name(cls): return 'pyocd' @classmethod def capabilities(cls): return RunnerCaps(commands={'flash', 'debug', 'debugserver', 'attach'}, dev_id=True, flash_addr=True, erase=True, tool_opt=True) @classmethod def dev_id_help(cls) -> str: return '''Device identifier. Use it to select the probe's unique ID or substring thereof.''' @classmethod def do_add_parser(cls, parser): parser.add_argument('--target', required=True, help='target override') parser.add_argument('--daparg', help='Additional -da arguments to pyocd tool') parser.add_argument('--pyocd', default='pyocd', help='path to pyocd tool, default is pyocd') parser.add_argument('--flash-opt', default=[], action='append', help='''Additional options for pyocd flash, e.g. --flash-opt="-e=chip" to chip erase''') parser.add_argument('--frequency', help='SWD clock frequency in Hz') parser.add_argument('--gdb-port', default=DEFAULT_PYOCD_GDB_PORT, help='pyocd gdb port, defaults to {}'.format( DEFAULT_PYOCD_GDB_PORT)) parser.add_argument('--telnet-port', default=DEFAULT_PYOCD_TELNET_PORT, help='pyocd telnet port, defaults to {}'.format( DEFAULT_PYOCD_TELNET_PORT)) parser.add_argument('--tui', default=False, action='store_true', help='if given, GDB uses -tui') parser.add_argument('--board-id', dest='dev_id', action=partial(depr_action, replacement='-i/--dev-id'), help='Deprecated: use -i/--dev-id instead') @classmethod def tool_opt_help(cls) -> str: return """Additional options for pyocd commander, e.g. '--script=user.py'""" @classmethod def do_create(cls, cfg, args): build_conf = BuildConfiguration(cfg.build_dir) flash_addr = cls.get_flash_address(args, build_conf) ret = PyOcdBinaryRunner( cfg, args.target, pyocd=args.pyocd, flash_addr=flash_addr, erase=args.erase, flash_opts=args.flash_opt, gdb_port=args.gdb_port, telnet_port=args.telnet_port, tui=args.tui, dev_id=args.dev_id, daparg=args.daparg, frequency=args.frequency, tool_opt=args.tool_opt) daparg = os.environ.get('PYOCD_DAPARG') if not ret.daparg_args and daparg: ret.logger.warning('PYOCD_DAPARG is deprecated; use --daparg') ret.logger.debug('--daparg={} via PYOCD_DAPARG'.format(daparg)) ret.daparg_args = ['-da', daparg] return ret def port_args(self): return ['-p', str(self.gdb_port), '-T', str(self.telnet_port)] def do_run(self, command, **kwargs): self.require(self.pyocd) if command == 'flash': self.flash(**kwargs) else: self.debug_debugserver(command, **kwargs) def flash(self, **kwargs): if self.hex_name is not None and os.path.isfile(self.hex_name): fname = self.hex_name elif self.bin_name is not None and os.path.isfile(self.bin_name): self.logger.warning( 'hex file ({}) does not exist; falling back on .bin ({}). '. format(self.hex_name, self.bin_name) + 'Consider enabling CONFIG_BUILD_OUTPUT_HEX.') fname = self.bin_name else: raise ValueError( 'Cannot flash; no hex ({}) or bin ({}) files found. '.format( self.hex_name, self.bin_name)) erase_method = 'chip' if self.erase else 'sector' cmd = ([self.pyocd] + ['flash'] + self.pyocd_config_args + ['-e', erase_method] + self.flash_addr_args + self.daparg_args + self.target_args + self.board_args + self.frequency_args + self.tool_opt_args + self.flash_extra + [fname]) self.logger.info('Flashing file: {}'.format(fname)) self.check_call(cmd) def log_gdbserver_message(self): self.logger.info('pyOCD GDB server running on port {}'. format(self.gdb_port)) def debug_debugserver(self, command, **kwargs): server_cmd = ([self.pyocd] + ['gdbserver'] + self.daparg_args + self.port_args() + self.target_args + self.board_args + self.frequency_args + self.tool_opt_args) if command == 'debugserver': self.log_gdbserver_message() self.check_call(server_cmd) else: if self.gdb_cmd is None: raise ValueError('Cannot debug; gdb is missing') if self.elf_name is None: raise ValueError('Cannot debug; elf is missing') client_cmd = (self.gdb_cmd + self.tui_args + [self.elf_name] + ['-ex', 'target remote :{}'.format(self.gdb_port)]) if command == 'debug': client_cmd += ['-ex', 'monitor halt', '-ex', 'monitor reset', '-ex', 'load'] self.require(client_cmd[0]) self.log_gdbserver_message() self.run_server_and_client(server_cmd, client_cmd)