#!/usr/bin/env python3
#
# Test the bFLT loader format
#
# Copyright (C) 2019 Philippe Mathieu-Daudé <f4bug@amsat.org>
#
# SPDX-License-Identifier: GPL-2.0-or-later

import os
import bz2

from qemu_test import QemuUserTest, Asset
from qemu_test import has_cmd
from qemu_test.utils import cpio_extract
from unittest import skipUnless


class LoadBFLT(QemuUserTest):

    ASSET_ROOTFS = Asset(
        ('https://elinux.org/images/5/51/Stm32_mini_rootfs.cpio.bz2'),
         'eefb788e4980c9e8d6c9d60ce7d15d4da6bf4fbc6a80f487673824600d5ba9cc')

    @skipUnless(*has_cmd('cpio'))
    @skipUnless(os.getenv('QEMU_TEST_ALLOW_UNTRUSTED_CODE'), 'untrusted code')
    def test_stm32(self):
        # See https://elinux.org/STM32#User_Space
        rootfs_path_bz2 = self.ASSET_ROOTFS.fetch()
        busybox_path = os.path.join(self.workdir, "bin/busybox")

        with bz2.open(rootfs_path_bz2, 'rb') as cpio_handle:
            cpio_extract(cpio_handle, self.workdir)

        res = self.run_cmd(busybox_path)
        ver = 'BusyBox v1.24.0.git (2015-02-03 22:17:13 CET) multi-call binary.'
        self.assertIn(ver, res.stdout)

        res = self.run_cmd(busybox_path, ['uname', '-a'])
        unm = 'armv7l GNU/Linux'
        self.assertIn(unm, res.stdout)


if __name__ == '__main__':
    QemuUserTest.main()