1*4882a593Smuzhiyun# 2*4882a593Smuzhiyun# Copyright (C) 2016 Intel Corporation 3*4882a593Smuzhiyun# 4*4882a593Smuzhiyun# SPDX-License-Identifier: MIT 5*4882a593Smuzhiyun# 6*4882a593Smuzhiyun 7*4882a593Smuzhiyunimport os 8*4882a593Smuzhiyunimport subprocess 9*4882a593Smuzhiyun 10*4882a593Smuzhiyunfrom oeqa.core.case import OETestCase 11*4882a593Smuzhiyun 12*4882a593Smuzhiyunclass OESDKTestCase(OETestCase): 13*4882a593Smuzhiyun def _run(self, cmd): 14*4882a593Smuzhiyun return subprocess.check_output(". %s > /dev/null; %s;" % \ 15*4882a593Smuzhiyun (self.tc.sdk_env, cmd), shell=True, executable="/bin/bash", 16*4882a593Smuzhiyun stderr=subprocess.STDOUT, universal_newlines=True) 17*4882a593Smuzhiyun 18*4882a593Smuzhiyun def fetch(self, workdir, dl_dir, url, archive=None): 19*4882a593Smuzhiyun if not archive: 20*4882a593Smuzhiyun from urllib.parse import urlparse 21*4882a593Smuzhiyun archive = os.path.basename(urlparse(url).path) 22*4882a593Smuzhiyun 23*4882a593Smuzhiyun if dl_dir: 24*4882a593Smuzhiyun tarball = os.path.join(dl_dir, archive) 25*4882a593Smuzhiyun if os.path.exists(tarball): 26*4882a593Smuzhiyun return tarball 27*4882a593Smuzhiyun 28*4882a593Smuzhiyun tarball = os.path.join(workdir, archive) 29*4882a593Smuzhiyun subprocess.check_output(["wget", "-O", tarball, url], stderr=subprocess.STDOUT) 30*4882a593Smuzhiyun return tarball 31*4882a593Smuzhiyun 32*4882a593Smuzhiyun def check_elf(self, path, target_os=None, target_arch=None): 33*4882a593Smuzhiyun """ 34*4882a593Smuzhiyun Verify that the ELF binary $path matches the specified target 35*4882a593Smuzhiyun OS/architecture, or if not specified the currently configured MACHINE's 36*4882a593Smuzhiyun OS/architecture. 37*4882a593Smuzhiyun """ 38*4882a593Smuzhiyun import oe.qa, oe.elf 39*4882a593Smuzhiyun 40*4882a593Smuzhiyun if not target_os or not target_arch: 41*4882a593Smuzhiyun output = self._run("echo $OECORE_TARGET_OS:$OECORE_TARGET_ARCH") 42*4882a593Smuzhiyun target_os, target_arch = output.strip().split(":") 43*4882a593Smuzhiyun 44*4882a593Smuzhiyun machine_data = oe.elf.machine_dict(None)[target_os][target_arch] 45*4882a593Smuzhiyun (machine, osabi, abiversion, endian, bits) = machine_data 46*4882a593Smuzhiyun 47*4882a593Smuzhiyun elf = oe.qa.ELFFile(path) 48*4882a593Smuzhiyun elf.open() 49*4882a593Smuzhiyun 50*4882a593Smuzhiyun self.assertEqual(machine, elf.machine(), 51*4882a593Smuzhiyun "Binary was %s but expected %s" % 52*4882a593Smuzhiyun (oe.qa.elf_machine_to_string(elf.machine()), oe.qa.elf_machine_to_string(machine))) 53*4882a593Smuzhiyun self.assertEqual(bits, elf.abiSize()) 54*4882a593Smuzhiyun self.assertEqual(endian, elf.isLittleEndian()) 55