msptool.py

#!/usr/bin/env python
#
#  Copyright (c) 2021 Upside Down Labs <contact@upsidedownlabs.tech, myupsidedownlab@gmail.com>
#  Author: Deepak Khatri (lorforlinux) <deepak@upsidedownlabs.tech, deepaklorkhatri7@gmail.com>
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:

# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.

# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

import os
import platform
import argparse

try:
    import serial
except ImportError:
    print("Pyserial is not installed!")
    raise

__version__ = "0.3"

# Argument parsing
help = {"port": "USB-UART bridge port, Windows: COM#, Linux: /dev/tty#",
        "firmware": ".txt(TI-TXT)/.elf/.hex/.bin format firmware image from msp430-gcc/CCS/Energia",
        "directory": "relative msptool directory (if required!)"}
parser = argparse.ArgumentParser()
parser.add_argument('-p', '--port', required=True, type = str, help=help["port"])
parser.add_argument('-f', '--firmware', required=True, type=argparse.FileType('r'), help=help["firmware"])
parser.add_argument('-d', '--directory', required=False, type= str, default=".", help=help["directory"])
args = parser.parse_args()

# Extract info from parsed arguments
comPort = args.port
firmwareImage = args.firmware.name
mspdebugDirectory = args.directory

# Method to Reset connected MSP430
def reset():
    '''Generate reset signal on DTR pin of USB-UART bridge'''
    bridge = serial.Serial(comPort, 9600)
    print("Resetting...")
    bridge.dtr = 0
    bridge.dtr = 1
    bridge.close()

# Get the OS specific command to execute
if platform.system() == 'Windows':
    '''Command for Windows'''
    #command = "rom-bsl.exe -c" + str.upper(comPort) + " -m1 -ievpr " + firmwareImage
    command = mspdebugDirectory + "\mspdebug\mspdebug.exe rom-bsl -d " + str.upper(comPort) + " \"prog " + firmwareImage + "\""
elif platform.system() == 'Linux':
    '''Command for Linux'''
    command = mspdebugDirectory + "/mspdebug/mspdebug rom-bsl -d " + comPort + " \"prog " + firmwareImage + "\""
elif platform.system() == 'Darwin':
    '''Command for OS X'''
    command = mspdebugDirectory + "/mspdebug/mspdebug_osx rom-bsl -d " + comPort + " \"prog " + firmwareImage + "\""
else:
    '''Confused!'''
    print("OS detection ERROR!")
    print("Only Linux/Windows/OSx supported..")

# Print the command
print("msptool v" + __version__)
print("Executing: " + command)
# Execute command
os.system(command)
# Reset chip
reset()