forked from Archive/PX4-Autopilot
124 lines
4.8 KiB
Python
Executable File
124 lines
4.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
############################################################################
|
|
#
|
|
# Copyright (C) 2012, 2013 PX4 Development Team. All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions
|
|
# are met:
|
|
#
|
|
# 1. Redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer.
|
|
# 2. Redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in
|
|
# the documentation and/or other materials provided with the
|
|
# distribution.
|
|
# 3. Neither the name PX4 nor the names of its contributors may be
|
|
# used to endorse or promote products derived from this software
|
|
# without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
|
# FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
|
|
# COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
|
|
# OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
|
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
|
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
|
#
|
|
############################################################################
|
|
|
|
#
|
|
# PX4 firmware image generator
|
|
#
|
|
# The PX4 firmware file is a JSON-encoded Python object, containing
|
|
# metadata fields and a zlib-compressed base64-encoded firmware image.
|
|
#
|
|
|
|
import sys
|
|
import argparse
|
|
import json
|
|
import base64
|
|
import zlib
|
|
import time
|
|
import subprocess
|
|
|
|
#
|
|
# Construct a basic firmware description
|
|
#
|
|
def mkdesc():
|
|
proto = {}
|
|
proto['magic'] = "PX4FWv1"
|
|
proto['board_id'] = 0
|
|
proto['board_revision'] = 0
|
|
proto['version'] = ""
|
|
proto['summary'] = ""
|
|
proto['description'] = ""
|
|
proto['git_identity'] = ""
|
|
proto['build_time'] = 0
|
|
proto['image'] = bytes()
|
|
proto['image_size'] = 0
|
|
return proto
|
|
|
|
# Parse commandline
|
|
parser = argparse.ArgumentParser(description="Firmware generator for the PX autopilot system.")
|
|
parser.add_argument("--prototype", action="store", help="read a prototype description from a file")
|
|
parser.add_argument("--board_id", action="store", help="set the board ID required")
|
|
parser.add_argument("--board_revision", action="store", help="set the board revision required")
|
|
parser.add_argument("--version", action="store", help="set a version string")
|
|
parser.add_argument("--summary", action="store", help="set a brief description")
|
|
parser.add_argument("--description", action="store", help="set a longer description")
|
|
parser.add_argument("--git_identity", action="store", help="the working directory to check for git identity")
|
|
parser.add_argument("--parameter_xml", action="store", help="the parameters.xml file")
|
|
parser.add_argument("--airframe_xml", action="store", help="the airframes.xml file")
|
|
parser.add_argument("--image", action="store", help="the firmware image")
|
|
args = parser.parse_args()
|
|
|
|
# Fetch the firmware descriptor prototype if specified
|
|
if args.prototype != None:
|
|
f = open(args.prototype,"r")
|
|
desc = json.load(f)
|
|
f.close()
|
|
else:
|
|
desc = mkdesc()
|
|
|
|
desc['build_time'] = int(time.time())
|
|
|
|
if args.board_id != None:
|
|
desc['board_id'] = int(args.board_id)
|
|
if args.board_revision != None:
|
|
desc['board_revision'] = int(args.board_revision)
|
|
if args.version != None:
|
|
desc['version'] = str(args.version)
|
|
if args.summary != None:
|
|
desc['summary'] = str(args.summary)
|
|
if args.description != None:
|
|
desc['description'] = str(args.description)
|
|
if args.git_identity != None:
|
|
cmd = " ".join(["git", "--git-dir", args.git_identity + "/.git", "describe", "--always", "--dirty"])
|
|
p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE).stdout
|
|
desc['git_identity'] = str(p.read().strip())
|
|
p.close()
|
|
if args.parameter_xml != None:
|
|
f = open(args.parameter_xml, "rb")
|
|
bytes = f.read()
|
|
desc['parameter_xml_size'] = len(bytes)
|
|
desc['parameter_xml'] = base64.b64encode(zlib.compress(bytes,9)).decode('utf-8')
|
|
desc['mav_autopilot'] = 12 # 12 = MAV_AUTOPILOT_PX4
|
|
if args.airframe_xml != None:
|
|
f = open(args.airframe_xml, "rb")
|
|
bytes = f.read()
|
|
desc['airframe_xml_size'] = len(bytes)
|
|
desc['airframe_xml'] = base64.b64encode(zlib.compress(bytes,9)).decode('utf-8')
|
|
if args.image != None:
|
|
f = open(args.image, "rb")
|
|
bytes = f.read()
|
|
desc['image_size'] = len(bytes)
|
|
desc['image'] = base64.b64encode(zlib.compress(bytes,9)).decode('utf-8')
|
|
|
|
print(json.dumps(desc, indent=4))
|