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