mirror of
https://bitbucket.org/librepilot/librepilot.git
synced 2024-11-29 07:24:13 +01:00
100 lines
3.1 KiB
Python
Executable File
100 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# Generate a version blob for
|
|
# the OpenPilot firmware.
|
|
#
|
|
# By E. Lafargue (c) 2011 E. Lafargue & the OpenPilot team
|
|
# Licence: GPLv3
|
|
#
|
|
###
|
|
# Usage:
|
|
# versionblob.py --append --firmware=<Firmware file name> --boardid=<Board code>
|
|
#
|
|
# append: if present, then append blob to firmware file directly, otherwise create "blob.bin"
|
|
# firmware: the filename of the firmware binary
|
|
# boardid: as a string, the board code, for example "0401" for CC board version 1.
|
|
# should match the codes in firmware description files.
|
|
#
|
|
# We have 100 bytes for the whole description.
|
|
#
|
|
# Only the first 40 are visible on the FirmwareIAP uavobject, the remaining
|
|
# 60 are ok to use for packaging and will be saved in the flash
|
|
#
|
|
# Structure is:
|
|
# 4 bytes: header: "OpFw"
|
|
# 4 bytes: GIT commit tag (short version of SHA1)
|
|
# 4 bytes: Unix timestamp of compile time
|
|
# 2 bytes: target platform. Should follow same rule as BOARD_TYPE and BOARD_REVISION in board define files.
|
|
# 26 bytes: commit tag if it is there, otherwise branch name. Zero-padded
|
|
# ---- 40 bytes limit ---
|
|
# 20 bytes: SHA1 sum of the firmware.
|
|
# 40 bytes: free for now.
|
|
|
|
import binascii
|
|
import os
|
|
from time import time
|
|
import argparse
|
|
|
|
# Do the argument parsing:
|
|
parser = argparse.ArgumentParser(description='Generate firmware desciption blob')
|
|
parser.add_argument('--append', action='store_true')
|
|
parser.add_argument('--firmware', help='name of firmware binary file' , required=True)
|
|
parser.add_argument('--boardid', help='ID of board model, for example 0401 for CopterControl', required=True)
|
|
args = parser.parse_args()
|
|
print args
|
|
|
|
if args.append == True:
|
|
print 'Appending description blob directly to ' + args.firmware
|
|
filename = args.firmware
|
|
file = open(filename,"ab")
|
|
else:
|
|
filename = 'blob.bin'
|
|
file = open(filename,"wb")
|
|
|
|
|
|
# Write the magic value:
|
|
file.write("OpFw")
|
|
# Get the short commit tag of the current git repository.
|
|
# Strip it to 8 characters for a 4 byte (int32) value.
|
|
# We have no full guarantee of unicity, but it is good enough
|
|
# with the rest of the information in the structure.
|
|
hs= os.popen('git rev-parse --short=8 HEAD').read().strip()
|
|
print "Version: " + hs
|
|
hb=binascii.a2b_hex(hs)
|
|
file.write(hb)
|
|
# Then the Unix time into a 32 bit integer:
|
|
print "Date: " + hex(int(time())).lstrip('0x')
|
|
hb = binascii.a2b_hex(hex(int(time())).lstrip('0x'))
|
|
file.write(hb)
|
|
|
|
# Then write board type and board revision
|
|
hb = binascii.a2b_hex(args.boardid)
|
|
file.write(hb)
|
|
|
|
# Last: a user-friendly description if it exists in GIT, otherwise
|
|
# just "unreleased"
|
|
hs = os.popen('git describe --exact-match').read()
|
|
if len(hs) == 0 :
|
|
print "Unreleased: get branch name instead"
|
|
hs = os.popen('git branch --contains HEAD').read()
|
|
|
|
file.write(hs[0:26])
|
|
file.write("\0"*(26-len(hs)))
|
|
|
|
## Now we are at the 40 bytes mark.
|
|
|
|
## Add the 20 byte SHA1 hash of the firmware:
|
|
import hashlib
|
|
sha1 = hashlib.sha1()
|
|
with open('build/coptercontrol/CopterControl.bin','rb') as f:
|
|
for chunk in iter(lambda: f.read(8192), ''):
|
|
sha1.update(chunk)
|
|
file.write(sha1.digest())
|
|
|
|
# Pad will null bytes:
|
|
file.write('\0'*40)
|
|
|
|
|
|
file.close()
|
|
|