forked from OSchip/llvm-project
91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# The LLVM Compiler Infrastructure
|
|
#
|
|
# This file is distributed under the University of Illinois Open Source
|
|
# License. See LICENSE.TXT for details.
|
|
|
|
import libear
|
|
import libscanbuild.intercept as sut
|
|
import unittest
|
|
import os.path
|
|
|
|
|
|
class InterceptUtilTest(unittest.TestCase):
|
|
|
|
def test_format_entry_filters_action(self):
|
|
def test(command):
|
|
trace = {'command': command, 'directory': '/opt/src/project'}
|
|
return list(sut.format_entry(trace))
|
|
|
|
self.assertTrue(test(['cc', '-c', 'file.c', '-o', 'file.o']))
|
|
self.assertFalse(test(['cc', '-E', 'file.c']))
|
|
self.assertFalse(test(['cc', '-MM', 'file.c']))
|
|
self.assertFalse(test(['cc', 'this.o', 'that.o', '-o', 'a.out']))
|
|
|
|
def test_format_entry_normalize_filename(self):
|
|
parent = os.path.join(os.sep, 'home', 'me')
|
|
current = os.path.join(parent, 'project')
|
|
|
|
def test(filename):
|
|
trace = {'directory': current, 'command': ['cc', '-c', filename]}
|
|
return list(sut.format_entry(trace))[0]['file']
|
|
|
|
self.assertEqual(os.path.join(current, 'file.c'), test('file.c'))
|
|
self.assertEqual(os.path.join(current, 'file.c'), test('./file.c'))
|
|
self.assertEqual(os.path.join(parent, 'file.c'), test('../file.c'))
|
|
self.assertEqual(os.path.join(current, 'file.c'),
|
|
test(os.path.join(current, 'file.c')))
|
|
|
|
def test_sip(self):
|
|
def create_status_report(filename, message):
|
|
content = """#!/usr/bin/env sh
|
|
echo 'sa-la-la-la'
|
|
echo 'la-la-la'
|
|
echo '{0}'
|
|
echo 'sa-la-la-la'
|
|
echo 'la-la-la'
|
|
""".format(message)
|
|
lines = [line.strip() for line in content.split('\n')]
|
|
with open(filename, 'w') as handle:
|
|
handle.write('\n'.join(lines))
|
|
handle.close()
|
|
os.chmod(filename, 0x1ff)
|
|
|
|
def create_csrutil(dest_dir, status):
|
|
filename = os.path.join(dest_dir, 'csrutil')
|
|
message = 'System Integrity Protection status: {0}'.format(status)
|
|
return create_status_report(filename, message)
|
|
|
|
def create_sestatus(dest_dir, status):
|
|
filename = os.path.join(dest_dir, 'sestatus')
|
|
message = 'SELinux status:\t{0}'.format(status)
|
|
return create_status_report(filename, message)
|
|
|
|
ENABLED = 'enabled'
|
|
DISABLED = 'disabled'
|
|
|
|
OSX = 'darwin'
|
|
|
|
with libear.TemporaryDirectory() as tmpdir:
|
|
saved = os.environ['PATH']
|
|
try:
|
|
os.environ['PATH'] = tmpdir + ':' + saved
|
|
|
|
create_csrutil(tmpdir, ENABLED)
|
|
self.assertTrue(sut.is_preload_disabled(OSX))
|
|
|
|
create_csrutil(tmpdir, DISABLED)
|
|
self.assertFalse(sut.is_preload_disabled(OSX))
|
|
finally:
|
|
os.environ['PATH'] = saved
|
|
|
|
saved = os.environ['PATH']
|
|
try:
|
|
os.environ['PATH'] = ''
|
|
# shall be false when it's not in the path
|
|
self.assertFalse(sut.is_preload_disabled(OSX))
|
|
|
|
self.assertFalse(sut.is_preload_disabled('unix'))
|
|
finally:
|
|
os.environ['PATH'] = saved
|