trezor-firmware/core/site_scons/tools.py

import os.path

import subprocess
from boards import trezor_1, trezor_r_v3, trezor_r_v4, trezor_t, discovery



def add_font(font_name, font, defines, sources):
    if font is not None:
        defines += [
            'TREZOR_FONT_' + font_name + '_ENABLE=' + font,
            'TREZOR_FONT_' + font_name + '_INCLUDE=\\"' + font.lower() + '.h\\"',
        ]
        sourcefile = 'embed/lib/fonts/' + font.lower() + '.c'
        if sourcefile not in sources:
            sources.append(sourcefile)


def configure_board(model, features_wanted, env, defines, sources):
    model_r_version = 4

    disc = True
    if disc:
        return discovery.configure(env, features_wanted, defines, sources)
    else:
        if model in ('1',):
            return trezor_1.configure(env, features_wanted, defines, sources)
        elif model in ('T',):
            return trezor_t.configure(env, features_wanted, defines, sources)
        elif model in ('R',):
            if model_r_version == 3:
                return trezor_r_v3.configure(env, features_wanted, defines, sources)
            else:
                return trezor_r_v4.configure(env, features_wanted, defines, sources)
        else:
            raise Exception("Unknown model")



def get_model_identifier(model):
    if model == '1':
        return "T1B1"
    elif model == 'T':
        return "T2T1"
    elif model == 'R':
        return "T2B1"
    else:
        raise Exception("Unknown model")


def get_version(file):
    major = 0
    minor = 0
    patch = 0

    if not os.path.exists(file):
        file = os.path.join("..", "..", file)

    with open(file, 'r') as f:
        for line in f:
            if line.startswith('#define VERSION_MAJOR '):
                major = line.split('VERSION_MAJOR')[1].strip()
            if line.startswith('#define VERSION_MINOR '):
                minor = line.split('VERSION_MINOR')[1].strip()
            if line.startswith('#define VERSION_PATCH '):
                patch = line.split('VERSION_PATCH')[1].strip()
        return f'{major}.{minor}.{patch}'


def get_version_int(file):
    major = 0
    minor = 0
    patch = 0

    if not os.path.exists(file):
        file = os.path.join("..", "..", file)

    with open(file, 'r') as f:
        for line in f:
            if line.startswith('#define VERSION_MAJOR '):
                major = int(line.split('VERSION_MAJOR')[1].strip())
            if line.startswith('#define VERSION_MINOR '):
                minor = int(line.split('VERSION_MINOR')[1].strip())
            if line.startswith('#define VERSION_PATCH '):
                patch = int(line.split('VERSION_PATCH')[1].strip())
        if major > 99 or minor > 99 or patch > 99:
            raise Exception("Version number too large")
        return major * 10000 + minor * 100 + patch


def get_git_revision_hash() -> str:
    return subprocess.check_output(['git', 'rev-parse', 'HEAD']).decode('ascii').strip()


def get_git_revision_short_hash() -> str:
    return subprocess.check_output(['git', 'rev-parse', '--short', 'HEAD']).decode('ascii').strip()


def get_git_modified() -> bool:
    return subprocess.check_output(['git', 'diff', '--name-status']).decode('ascii').strip() != ''


def get_defs_for_cmake(defs):
    result = []
    for d in defs:
        if type(d) is tuple:
            result.append(d[0] + "=" + d[1])
        else:
            result.append(d)
    return result