#!/usr/bin/python from __future__ import print_function import sys import json import os import re import fnmatch FIRMWARE_TYPES = ["AntennaTracker", "Copter", "Plane", "Rover", "Sub"] RELEASE_TYPES = ["beta", "latest", "stable", "stable-*"] class Firmware(): def __init__(self, date=None, platform=None, vehicletype=None, filepath=None, git_sha=None, frame=None): self.atts = dict() self.atts["date"] = date self.atts["platform"] = platform self.atts["vehicletype"] = vehicletype self.atts["filepath"] = filepath self.atts["git_sha"] = git_sha self.atts["frame"] = frame self.atts["release-type"] = None self.atts["firmware-version"] = None def __getitem__(self, what): return self.atts[what] def __setitem__(self, name, value): self.atts[name] = value class ManifestGenerator(): '''Return a JSON string describing "binary" directory contents under basedir''' def __init__(self, basedir, baseurl): self.basedir = basedir self.baseurl = baseurl def frame_map(self, frame): '''translate from ArduPilot frame type terminology into mavlink terminology''' frame_to_mavlink_dict = { "quad": "QUADROTOR", "hexa": "HEXAROTOR", "y6": "ARDUPILOT_Y6", "tri": "TRICOPTER", "octa": "OCTOROTOR", "octa-quad": "ARDUPILOT_OCTAQUAD", "heli": "HELICOPTER", "Plane": "FIXED_WING", "AntennaTracker": "ANTENNA_TRACKER", "Rover": "GROUND_ROVER", "Sub": "SUBMARINE" } if frame in frame_to_mavlink_dict: return frame_to_mavlink_dict[frame] return frame def releasetype_map(self, releasetype): '''translate from ArduPilot release type terminology into mavlink terminology''' if releasetype == 'stable': return 'OFFICIAL' return releasetype.upper() def looks_like_binaries_directory(self, dir): '''returns True if dir looks like it is a build_binaries.py output directory''' for entry in os.listdir(dir): if entry in FIRMWARE_TYPES: return True return False def git_sha_from_git_version(self, filepath): '''parses get-version.txt (as emitted by build_binaries.py, returns git sha from it''' content = open(filepath).read() sha_regex = re.compile("commit (?P[0-9a-f]+)") m = sha_regex.search(content) if m is None: raise Exception( "filepath (%s) does not contain a git sha" % (filepath,)) return m.group("sha") def add_USB_IDs_PX4(self, firmware): '''add USB IDs to a .px4 firmware''' url = firmware['url'] suffix = url.split('-')[-1] if suffix == "v1.px4": firmware['USBID'] = ['0x26AC/0x0010'] firmware['board_id'] = 5 firmware['bootloader_str'] = ['PX4 BL FMU v1.x'] elif suffix in ["v2.px4", "v3.px4"]: firmware['USBID'] = ['0x26AC/0x0011'] firmware['board_id'] = 9 firmware['bootloader_str'] = ['PX4 BL FMU v2.x'] elif suffix == "v4.px4": firmware['USBID'] = ['0x26AC/0x0012'] firmware['board_id'] = 11 firmware['bootloader_str'] = ['PX4 BL FMU v4.x'] elif suffix == "v4pro.px4": firmware['USBID'] = ['0x26AC/0x0013'] firmware['board_id'] = 13 firmware['bootloader_str'] = ['PX4 BL FMU v4.x PRO'] def add_USB_IDs_ChibiOS(self, firmware): '''add USB IDs to a ChbiOS apj firmware''' url = firmware['url'][len(self.baseurl)+1:] apj_path = os.path.join(self.basedir, url) if not os.path.exists(apj_path): print("bad apj path %s" % apj_path, file=sys.stderr) return apj_json = json.load(open(apj_path, 'r')) if 'board_id' not in apj_json: print("no board_id in %s" % apj_path, file=sys.stderr) return if 'platform' not in firmware: print("no platform for %s" % apj_path, file=sys.stderr) return board_id = apj_json['board_id'] platform = firmware['platform'] # all ChibiOS builds can have platform as bootloader_str and board_id from # hwdef.dat firmware['board_id'] = board_id firmware['bootloader_str'] = [platform+"-BL"] # map of vendor specific USB IDs USBID_MAP = { 'CubeBlack': ['0x2DAE/0x1011'], 'CubeOrange': ['0x2DAE/0x1016'], 'CubePurple': ['0x2DAE/0x1005'], 'CubeYellow': ['0x2DAE/0x1002'], 'Pixhawk4': ['0x3162/0x0047'], 'PH4-mini': ['0x3162/0x0049'], 'Pixhawk6': ['0x3162/0x004B'], 'VRBrain-v51': ['0x27AC/0x1151'], 'VRBrain-v52': ['0x27AC/0x1152'], 'VRBrain-v54': ['0x27AC/0x1154'], 'VRCore-v10': ['0x27AC/0x1910'], 'VRUBrain-v51': ['0x27AC/0x1351'] } if platform in USBID_MAP: firmware['USBID'] = USBID_MAP[platform] else: # all others use a single USB VID/PID firmware['USBID'] = ['0x0483/0x5740'] if board_id == 50: # special case for FMUv5, they always get the px4 bootloader IDs as an option firmware['bootloader_str'].append('PX4 BL FMU v5.x') firmware['USBID'].append('0x26AC/0x0032') if board_id == 9: # special case for FMUv3, they always get the px4 bootloader IDs as an option firmware['bootloader_str'].append('PX4 BL FMU v2.x') firmware['USBID'].append('0x26AC/0x0011') if board_id == 11: # special case for FMUv4, they always get the px4 bootloader IDs as an option firmware['bootloader_str'].append('PX4 BL FMU v4.x') firmware['USBID'].append('0x26AC/0x0012') if board_id == 13: # special case for FMUv4pro, they always get the px4 bootloader IDs as an option firmware['bootloader_str'].append('PX4 BL FMU v4.x PRO') firmware['USBID'].append('0x26AC/0x0013') if board_id == 88: # special case for MindPX-v2 boards firmware['bootloader_str'].append('MindPX BL FMU v2.x') firmware['USBID'].append('0x26AC/0x0030') def add_USB_IDs(self, firmware): '''add USB IDs to a firmware''' fmt = firmware['format'] if fmt == "px4": self.add_USB_IDs_PX4(firmware) return if fmt == "apj": self.add_USB_IDs_ChibiOS(firmware) return def add_firmware_data_from_dir(self, dir, firmware_data, vehicletype, releasetype="dev"): '''accumulate additional information about firmwares from directory''' platform_frame_regex = re.compile( "(?PPX4|navio|pxf)(-(?P.+))?") variant_firmware_regex = re.compile("[^-]+-(?Pv\d+)[.px4]") if not os.path.isdir(dir): return try: dlist = os.listdir(dir) except Exception: print("Error listing '%s'" % dir) return for platformdir in dlist: if platformdir.startswith("."): continue some_dir = os.path.join(dir, platformdir) if not os.path.isdir(some_dir): continue git_version_txt = os.path.join(some_dir, "git-version.txt") if not os.path.exists(git_version_txt): print("No file %s" % git_version_txt, file=sys.stderr) continue try: git_sha = self.git_sha_from_git_version(git_version_txt) except Exception as ex: print("Failed to parse %s" % git_version_txt, ex, file=sys.stderr) continue # we require a firmware-version.txt. These files have been added to # old builds that didn't have them firmware_version_file = os.path.join(some_dir, "firmware-version.txt") if not os.path.exists(firmware_version_file): print("Missing %s" % firmware_version_file, file=sys.stderr) continue try: firmware_version = open(firmware_version_file).read() firmware_version = firmware_version.strip() (version_numbers, release_type) = firmware_version.split("-") except ValueError: print("malformed firmware-version.txt at (%s)" % (firmware_version_file,), file=sys.stderr) continue except Exception as ex: print("bad file %s" % firmware_version_file, file=sys.stderr) # this exception is swallowed.... the current archive # is incomplete. continue m = platform_frame_regex.match(platformdir) if m is not None: # the model type (quad/tri) is # encoded in the platform name # (e.g. navio-octa) platform = m.group("board") # e.g. navio frame = m.group("frame") # e.g. octa if frame is None: frame = vehicletype else: frame = vehicletype # e.g. Plane platform = platformdir # e.g. apm2 for file in os.listdir(some_dir): if file in ["git-version.txt", "firmware-version.txt", "files.html"]: continue if file.startswith("."): continue m = variant_firmware_regex.match(file) if m: # the platform variant is # encoded in the firmware filename # (e.g. the "v1" in # ArduCopter-v1.px4) variant = m.group("variant") file_platform = "-".join([platform, variant]) else: file_platform = platform firmware_format = "".join(file.split(".")[-1:]) firmware = Firmware() # translate from supplied "release type" into both a # "latest" flag and an actual release type. Also sort # out which filepath we should use: firmware["latest"] = 0 if releasetype == "dev": if firmware["filepath"] is None: firmware["filepath"] = os.path.join(some_dir, file) if firmware["release-type"] is None: firmware["release-type"] = "dev" elif releasetype == "latest": firmware["latest"] = 1 firmware["filepath"] = os.path.join(some_dir, file) if firmware["release-type"] is None: firmware["release-type"] = "dev" else: if (not firmware["latest"]): firmware["filepath"] = os.path.join(some_dir, file) firmware["release-type"] = releasetype firmware["platform"] = file_platform firmware["vehicletype"] = vehicletype firmware["git_sha"] = git_sha firmware["frame"] = frame firmware["timestamp"] = os.path.getctime(firmware["filepath"]) firmware["format"] = firmware_format firmware["firmware-version"] = firmware_version firmware_data.append(firmware) def valid_release_type(self, tag): '''check for valid release type''' for r in RELEASE_TYPES: if fnmatch.fnmatch(tag, r): return True return False def parse_fw_version(self, version): (version_numbers, release_type) = version.split("-") (major, minor, patch) = version_numbers.split(".") return (major, minor, patch, version) def walk_directory(self, basedir): '''walks directory structure created by build_binaries, returns Python structure representing releases in that structure''' year_month_regex = re.compile("(?P\d{4})-(?P\d{2})") firmwares = [] # used to listdir basedir here, but since this is also a web # document root, there's a lot of other stuff accumulated... vehicletypes = FIRMWARE_TYPES for vehicletype in vehicletypes: try: # the sort means we prefer 'stable' to 'stable-x.y.z' when they # both contain the same contents vdir = sorted(os.listdir(os.path.join(basedir, vehicletype)), reverse=True) except OSError as e: if e.errno == 2: continue for firstlevel in vdir: if firstlevel == "files.html" or firstlevel.startswith("."): # generated file which should be ignored continue # skip any non-directories (e.g. "files.html"): if year_month_regex.match(firstlevel): # this is a dated directory e.g. binaries/Copter/2016-02 # we do not include dated directories in the manifest ATM: continue # assume this is a release directory such as # "beta", or the "latest" directory (treated as a # release and handled specially later) tag = firstlevel if not self.valid_release_type(tag): print("Unknown tag (%s) in directory (%s)" % (tag, os.path.join(vdir)), file=sys.stderr) continue tag_path = os.path.join(basedir, vehicletype, tag) if not os.path.isdir(tag_path): continue self.add_firmware_data_from_dir(tag_path, firmwares, vehicletype, releasetype=tag) # convert from ardupilot-naming conventions to common JSON format: firmware_json = [] for firmware in firmwares: filepath = firmware["filepath"] # replace the base directory with the base URL urlifier = re.compile("^" + re.escape(basedir)) url = re.sub(urlifier, self.baseurl, filepath) version_type = self.releasetype_map(firmware["release-type"]) some_json = dict({ "mav-autopilot": "ARDUPILOTMEGA", # "vehicletype": firmware["vehicletype"], "platform": firmware["platform"], "git-sha": firmware["git_sha"], "url": url, "mav-type": self.frame_map(firmware["frame"]), "mav-firmware-version-type": version_type, "latest": firmware["latest"], "format": firmware["format"], }) if firmware["firmware-version"]: try: (major, minor, patch, release_type) = self.parse_fw_version( firmware["firmware-version"]) except Exception: print("Badly formed firmware-version.txt %s" % firmware["firmware-version"], file=sys.stderr) continue some_json["mav-firmware-version"] = ".".join([major, minor, patch]) some_json["mav-firmware-version-major"] = major some_json["mav-firmware-version-minor"] = minor some_json["mav-firmware-version-patch"] = patch self.add_USB_IDs(some_json) #print(some_json['url']) firmware_json.append(some_json) ret = { "format-version": "1.0.0", # semantic versioning "firmware": firmware_json } return ret def json(self): '''walk directory supplied in constructor, return json string''' if not self.looks_like_binaries_directory(self.basedir): print("Warning: this does not look like a binaries directory", file=sys.stderr) structure = self.walk_directory(self.basedir) return json.dumps(structure, indent=4) def usage(): return '''Usage: generate-manifest.py basedir''' if __name__ == "__main__": import argparse parser = argparse.ArgumentParser(description='generate manifest.json') parser.add_argument('--outfile', type=str, default=None, help='output file, default stdout') parser.add_argument('--baseurl', type=str, default="http://firmware.ardupilot.org", help='base binaries directory') parser.add_argument('basedir', type=str, default="-", help='base binaries directory') args = parser.parse_args() generator = ManifestGenerator(args.basedir, args.baseurl) if args.outfile is None: print(generator.json()) else: f = open(args.outfile, "w") f.write(generator.json()) f.close()