1*e5dd7070Spatrick# -*- coding: utf-8 -*- 2*e5dd7070Spatrick# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 3*e5dd7070Spatrick# See https://llvm.org/LICENSE.txt for license information. 4*e5dd7070Spatrick# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 5*e5dd7070Spatrick 6*e5dd7070Spatrickimport libear 7*e5dd7070Spatrickimport libscanbuild.intercept as sut 8*e5dd7070Spatrickimport unittest 9*e5dd7070Spatrickimport os.path 10*e5dd7070Spatrick 11*e5dd7070Spatrick 12*e5dd7070Spatrickclass InterceptUtilTest(unittest.TestCase): 13*e5dd7070Spatrick 14*e5dd7070Spatrick def test_format_entry_filters_action(self): 15*e5dd7070Spatrick def test(command): 16*e5dd7070Spatrick trace = {'command': command, 'directory': '/opt/src/project'} 17*e5dd7070Spatrick return list(sut.format_entry(trace)) 18*e5dd7070Spatrick 19*e5dd7070Spatrick self.assertTrue(test(['cc', '-c', 'file.c', '-o', 'file.o'])) 20*e5dd7070Spatrick self.assertFalse(test(['cc', '-E', 'file.c'])) 21*e5dd7070Spatrick self.assertFalse(test(['cc', '-MM', 'file.c'])) 22*e5dd7070Spatrick self.assertFalse(test(['cc', 'this.o', 'that.o', '-o', 'a.out'])) 23*e5dd7070Spatrick 24*e5dd7070Spatrick def test_format_entry_normalize_filename(self): 25*e5dd7070Spatrick parent = os.path.join(os.sep, 'home', 'me') 26*e5dd7070Spatrick current = os.path.join(parent, 'project') 27*e5dd7070Spatrick 28*e5dd7070Spatrick def test(filename): 29*e5dd7070Spatrick trace = {'directory': current, 'command': ['cc', '-c', filename]} 30*e5dd7070Spatrick return list(sut.format_entry(trace))[0]['file'] 31*e5dd7070Spatrick 32*e5dd7070Spatrick self.assertEqual(os.path.join(current, 'file.c'), test('file.c')) 33*e5dd7070Spatrick self.assertEqual(os.path.join(current, 'file.c'), test('./file.c')) 34*e5dd7070Spatrick self.assertEqual(os.path.join(parent, 'file.c'), test('../file.c')) 35*e5dd7070Spatrick self.assertEqual(os.path.join(current, 'file.c'), 36*e5dd7070Spatrick test(os.path.join(current, 'file.c'))) 37*e5dd7070Spatrick 38*e5dd7070Spatrick def test_sip(self): 39*e5dd7070Spatrick def create_status_report(filename, message): 40*e5dd7070Spatrick content = """#!/usr/bin/env sh 41*e5dd7070Spatrick echo 'sa-la-la-la' 42*e5dd7070Spatrick echo 'la-la-la' 43*e5dd7070Spatrick echo '{0}' 44*e5dd7070Spatrick echo 'sa-la-la-la' 45*e5dd7070Spatrick echo 'la-la-la' 46*e5dd7070Spatrick """.format(message) 47*e5dd7070Spatrick lines = [line.strip() for line in content.split('\n')] 48*e5dd7070Spatrick with open(filename, 'w') as handle: 49*e5dd7070Spatrick handle.write('\n'.join(lines)) 50*e5dd7070Spatrick handle.close() 51*e5dd7070Spatrick os.chmod(filename, 0x1ff) 52*e5dd7070Spatrick 53*e5dd7070Spatrick def create_csrutil(dest_dir, status): 54*e5dd7070Spatrick filename = os.path.join(dest_dir, 'csrutil') 55*e5dd7070Spatrick message = 'System Integrity Protection status: {0}'.format(status) 56*e5dd7070Spatrick return create_status_report(filename, message) 57*e5dd7070Spatrick 58*e5dd7070Spatrick def create_sestatus(dest_dir, status): 59*e5dd7070Spatrick filename = os.path.join(dest_dir, 'sestatus') 60*e5dd7070Spatrick message = 'SELinux status:\t{0}'.format(status) 61*e5dd7070Spatrick return create_status_report(filename, message) 62*e5dd7070Spatrick 63*e5dd7070Spatrick ENABLED = 'enabled' 64*e5dd7070Spatrick DISABLED = 'disabled' 65*e5dd7070Spatrick 66*e5dd7070Spatrick OSX = 'darwin' 67*e5dd7070Spatrick 68*e5dd7070Spatrick with libear.TemporaryDirectory() as tmpdir: 69*e5dd7070Spatrick saved = os.environ['PATH'] 70*e5dd7070Spatrick try: 71*e5dd7070Spatrick os.environ['PATH'] = tmpdir + ':' + saved 72*e5dd7070Spatrick 73*e5dd7070Spatrick create_csrutil(tmpdir, ENABLED) 74*e5dd7070Spatrick self.assertTrue(sut.is_preload_disabled(OSX)) 75*e5dd7070Spatrick 76*e5dd7070Spatrick create_csrutil(tmpdir, DISABLED) 77*e5dd7070Spatrick self.assertFalse(sut.is_preload_disabled(OSX)) 78*e5dd7070Spatrick finally: 79*e5dd7070Spatrick os.environ['PATH'] = saved 80*e5dd7070Spatrick 81*e5dd7070Spatrick saved = os.environ['PATH'] 82*e5dd7070Spatrick try: 83*e5dd7070Spatrick os.environ['PATH'] = '' 84*e5dd7070Spatrick # shall be false when it's not in the path 85*e5dd7070Spatrick self.assertFalse(sut.is_preload_disabled(OSX)) 86*e5dd7070Spatrick 87*e5dd7070Spatrick self.assertFalse(sut.is_preload_disabled('unix')) 88*e5dd7070Spatrick finally: 89*e5dd7070Spatrick os.environ['PATH'] = saved 90