1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
# -*- 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 libscanbuild.intercept as sut
from . import fixtures
import os.path
class InterceptUtilTest(fixtures.TestCase):
def test_is_compiler_call_filter(self):
def test(command):
return sut.is_compiler_call({'command': [command]})
self.assertTrue(test('clang'))
self.assertTrue(test('clang-3.6'))
self.assertTrue(test('clang++'))
self.assertTrue(test('clang++-3.5.1'))
self.assertTrue(test('cc'))
self.assertTrue(test('c++'))
self.assertTrue(test('gcc'))
self.assertTrue(test('g++'))
self.assertTrue(test('/usr/local/bin/gcc'))
self.assertTrue(test('/usr/local/bin/g++'))
self.assertTrue(test('/usr/local/bin/clang'))
self.assertTrue(test('armv7_neno-linux-gnueabi-g++'))
self.assertFalse(test(''))
self.assertFalse(test('ld'))
self.assertFalse(test('as'))
self.assertFalse(test('/usr/local/bin/compiler'))
def test_format_entry_filters_action(self):
def test(command):
return list(sut.format_entry(
{'command': command, 'directory': '/opt/src/project'}))
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']))
self.assertFalse(test(['cc', '-print-prog-name']))
def test_format_entry_normalize_filename(self):
directory = os.path.join(os.sep, 'home', 'me', 'project')
def test(command):
result = list(sut.format_entry(
{'command': command, 'directory': directory}))
return result[0]['file']
self.assertEqual(test(['cc', '-c', 'file.c']),
os.path.join(directory, 'file.c'))
self.assertEqual(test(['cc', '-c', './file.c']),
os.path.join(directory, 'file.c'))
self.assertEqual(test(['cc', '-c', '../file.c']),
os.path.join(os.path.dirname(directory), 'file.c'))
self.assertEqual(test(['cc', '-c', '/opt/file.c']),
'/opt/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'
LINUX = 'linux'
with fixtures.TempDir() as tmpdir:
try:
saved = os.environ['PATH']
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))
create_sestatus(tmpdir, ENABLED)
self.assertTrue(sut.is_preload_disabled(LINUX))
create_sestatus(tmpdir, DISABLED)
self.assertFalse(sut.is_preload_disabled(LINUX))
finally:
os.environ['PATH'] = saved
try:
saved = os.environ['PATH']
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(LINUX))
self.assertFalse(sut.is_preload_disabled('unix'))
finally:
os.environ['PATH'] = saved
|
