mirror of
https://gitlab.com/apparmor/apparmor.git
synced 2025-03-04 16:35:02 +01:00
8d348b328b
Add a check to parse_profile_data() to detect if a file contains two profiles with the same name. Note: Two profiles with the same name, but in different files, won't be detected by this check. Also add basic tests to ensure that a valid profile gets parsed, and two profiles with the same name inside the same file raise an exception. (Sidenote: these simple tests improve aa.py coverage from 9% to 12%, which also confirms the function is too long ;-) Acked-by: Steve Beattie <steve@nxnw.org>
589 lines
30 KiB
Python
589 lines
30 KiB
Python
#! /usr/bin/env python
|
|
# ------------------------------------------------------------------
|
|
#
|
|
# Copyright (C) 2014-2015 Christian Boltz
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of version 2 of the GNU General Public
|
|
# License published by the Free Software Foundation.
|
|
#
|
|
# ------------------------------------------------------------------
|
|
|
|
import unittest
|
|
from common_test import AATest, setup_all_loops
|
|
from common_test import read_file, write_file
|
|
|
|
from apparmor.aa import (check_for_apparmor, get_profile_flags, set_profile_flags, is_skippable_file, is_skippable_dir,
|
|
parse_profile_start, parse_profile_data, separate_vars, store_list_var, write_header, serialize_parse_profile_start)
|
|
from apparmor.common import AppArmorException, AppArmorBug
|
|
|
|
class AaTestWithTempdir(AATest):
|
|
def AASetup(self):
|
|
self.createTmpdir()
|
|
|
|
|
|
class AaTest_check_for_apparmor(AaTestWithTempdir):
|
|
FILESYSTEMS_WITH_SECURITYFS = 'nodev\tdevtmpfs\nnodev\tsecurityfs\nnodev\tsockfs\n\text3\n\text2\n\text4'
|
|
FILESYSTEMS_WITHOUT_SECURITYFS = 'nodev\tdevtmpfs\nnodev\tsockfs\n\text3\n\text2\n\text4'
|
|
|
|
MOUNTS_WITH_SECURITYFS = ( 'proc /proc proc rw,relatime 0 0\n'
|
|
'securityfs %s/security securityfs rw,nosuid,nodev,noexec,relatime 0 0\n'
|
|
'/dev/sda1 / ext3 rw,noatime,data=ordered 0 0' )
|
|
|
|
MOUNTS_WITHOUT_SECURITYFS = ( 'proc /proc proc rw,relatime 0 0\n'
|
|
'/dev/sda1 / ext3 rw,noatime,data=ordered 0 0' )
|
|
|
|
def test_check_for_apparmor_None_1(self):
|
|
filesystems = write_file(self.tmpdir, 'filesystems', self.FILESYSTEMS_WITHOUT_SECURITYFS)
|
|
mounts = write_file(self.tmpdir, 'mounts', self.MOUNTS_WITH_SECURITYFS)
|
|
self.assertEqual(None, check_for_apparmor(filesystems, mounts))
|
|
|
|
def test_check_for_apparmor_None_2(self):
|
|
filesystems = write_file(self.tmpdir, 'filesystems', self.FILESYSTEMS_WITHOUT_SECURITYFS)
|
|
mounts = write_file(self.tmpdir, 'mounts', self.MOUNTS_WITHOUT_SECURITYFS)
|
|
self.assertEqual(None, check_for_apparmor(filesystems, mounts))
|
|
|
|
def test_check_for_apparmor_None_3(self):
|
|
filesystems = write_file(self.tmpdir, 'filesystems', self.FILESYSTEMS_WITH_SECURITYFS)
|
|
mounts = write_file(self.tmpdir, 'mounts', self.MOUNTS_WITHOUT_SECURITYFS)
|
|
self.assertEqual(None, check_for_apparmor(filesystems, mounts))
|
|
|
|
def test_check_for_apparmor_securityfs_invalid_filesystems(self):
|
|
filesystems = ''
|
|
mounts = write_file(self.tmpdir, 'mounts', self.MOUNTS_WITH_SECURITYFS % self.tmpdir)
|
|
self.assertEqual(None, check_for_apparmor(filesystems, mounts))
|
|
|
|
def test_check_for_apparmor_securityfs_invalid_mounts(self):
|
|
filesystems = write_file(self.tmpdir, 'filesystems', self.FILESYSTEMS_WITH_SECURITYFS)
|
|
mounts = ''
|
|
self.assertEqual(None, check_for_apparmor(filesystems, mounts))
|
|
|
|
def test_check_for_apparmor_invalid_securityfs_path(self):
|
|
filesystems = write_file(self.tmpdir, 'filesystems', self.FILESYSTEMS_WITH_SECURITYFS)
|
|
mounts = write_file(self.tmpdir, 'mounts', self.MOUNTS_WITH_SECURITYFS % 'xxx')
|
|
self.assertEqual(None, check_for_apparmor(filesystems, mounts))
|
|
|
|
def test_check_for_apparmor_securityfs_mounted(self):
|
|
filesystems = write_file(self.tmpdir, 'filesystems', self.FILESYSTEMS_WITH_SECURITYFS)
|
|
mounts = write_file(self.tmpdir, 'mounts', self.MOUNTS_WITH_SECURITYFS % self.tmpdir)
|
|
self.assertEqual('%s/security/apparmor' % self.tmpdir, check_for_apparmor(filesystems, mounts))
|
|
|
|
class AaTest_get_profile_flags(AaTestWithTempdir):
|
|
def _test_get_flags(self, profile_header, expected_flags):
|
|
file = write_file(self.tmpdir, 'profile', '%s {\n}\n' % profile_header)
|
|
flags = get_profile_flags(file, '/foo')
|
|
self.assertEqual(flags, expected_flags)
|
|
|
|
def test_get_flags_01(self):
|
|
self._test_get_flags('/foo', None)
|
|
def test_get_flags_02(self):
|
|
self._test_get_flags('/foo ( complain )', ' complain ')
|
|
def test_get_flags_04(self):
|
|
self._test_get_flags('/foo (complain)', 'complain')
|
|
def test_get_flags_05(self):
|
|
self._test_get_flags('/foo flags=(complain)', 'complain')
|
|
def test_get_flags_06(self):
|
|
self._test_get_flags('/foo flags=(complain, audit)', 'complain, audit')
|
|
|
|
def test_get_flags_invalid_01(self):
|
|
with self.assertRaises(AppArmorException):
|
|
self._test_get_flags('/foo ()', None)
|
|
def test_get_flags_invalid_02(self):
|
|
with self.assertRaises(AppArmorException):
|
|
self._test_get_flags('/foo flags=()', None)
|
|
def test_get_flags_invalid_03(self):
|
|
with self.assertRaises(AppArmorException):
|
|
self._test_get_flags('/foo ( )', ' ')
|
|
|
|
def test_get_flags_other_profile(self):
|
|
with self.assertRaises(AppArmorException):
|
|
self._test_get_flags('/no-such-profile flags=(complain)', 'complain')
|
|
|
|
class AaTest_set_profile_flags(AaTestWithTempdir):
|
|
def _test_set_flags(self, profile, old_flags, new_flags, whitespace='', comment='',
|
|
more_rules='', expected_more_rules='@-@-@',
|
|
expected_flags='@-@-@', check_new_flags=True, profile_name='/foo'):
|
|
if old_flags:
|
|
old_flags = ' %s' % old_flags
|
|
|
|
if expected_flags == '@-@-@':
|
|
expected_flags = new_flags
|
|
|
|
if expected_flags:
|
|
expected_flags = ' flags=(%s)' % (expected_flags)
|
|
else:
|
|
expected_flags = ''
|
|
|
|
if expected_more_rules == '@-@-@':
|
|
expected_more_rules = more_rules
|
|
|
|
if comment:
|
|
comment = ' %s' % comment
|
|
|
|
dummy_profile_content = ' #include <abstractions/base>\n capability chown,\n /bar r,'
|
|
prof_template = '%s%s%s {%s\n%s\n%s\n}\n'
|
|
old_prof = prof_template % (whitespace, profile, old_flags, comment, more_rules, dummy_profile_content)
|
|
new_prof = prof_template % (whitespace, profile, expected_flags, comment, expected_more_rules, dummy_profile_content)
|
|
|
|
self.file = write_file(self.tmpdir, 'profile', old_prof)
|
|
set_profile_flags(self.file, profile_name, new_flags)
|
|
if check_new_flags:
|
|
real_new_prof = read_file(self.file)
|
|
self.assertEqual(new_prof, real_new_prof)
|
|
|
|
# tests that actually don't change the flags
|
|
def test_set_flags_nochange_01(self):
|
|
self._test_set_flags('/foo', '', '')
|
|
def test_set_flags_nochange_02(self):
|
|
self._test_set_flags('/foo', '( complain )', ' complain ', whitespace=' ')
|
|
def test_set_flags_nochange_03(self):
|
|
self._test_set_flags('/foo', '(complain)', 'complain')
|
|
def test_set_flags_nochange_04(self):
|
|
self._test_set_flags('/foo', 'flags=(complain)', 'complain')
|
|
def test_set_flags_nochange_05(self):
|
|
self._test_set_flags('/foo', 'flags=(complain, audit)', 'complain, audit', whitespace=' ')
|
|
def test_set_flags_nochange_06(self):
|
|
self._test_set_flags('/foo', 'flags=(complain, audit)', 'complain, audit', whitespace=' ', comment='# a comment')
|
|
def test_set_flags_nochange_07(self):
|
|
self._test_set_flags('/foo', 'flags=(complain, audit)', 'complain, audit', whitespace=' ', more_rules=' # a comment\n#another comment')
|
|
def test_set_flags_nochange_08(self):
|
|
self._test_set_flags('profile /foo', 'flags=(complain)', 'complain')
|
|
def test_set_flags_nochange_09(self):
|
|
self._test_set_flags('profile xy /foo', 'flags=(complain)', 'complain', profile_name='xy')
|
|
def test_set_flags_nochange_10(self):
|
|
self._test_set_flags('profile "/foo bar"', 'flags=(complain)', 'complain', profile_name='/foo bar')
|
|
def test_set_flags_nochange_11(self):
|
|
self._test_set_flags('/foo', '(complain)', 'complain', profile_name=None)
|
|
#def test_set_flags_nochange_12(self):
|
|
# XXX changes the flags for the child profile (which happens to have the same profile name) to 'complain'
|
|
# self._test_set_flags('/foo', 'flags=(complain)', 'complain', more_rules=' profile /foo {\n}')
|
|
|
|
# tests that change the flags
|
|
def test_set_flags_01(self):
|
|
self._test_set_flags('/foo', '', 'audit')
|
|
def test_set_flags_02(self):
|
|
self._test_set_flags('/foo', '( complain )', 'audit ', whitespace=' ')
|
|
def test_set_flags_04(self):
|
|
self._test_set_flags('/foo', '(complain)', 'audit')
|
|
def test_set_flags_05(self):
|
|
self._test_set_flags('/foo', 'flags=(complain)', 'audit')
|
|
def test_set_flags_06(self):
|
|
self._test_set_flags('/foo', 'flags=(complain, audit)', None, whitespace=' ')
|
|
def test_set_flags_07(self):
|
|
self._test_set_flags('/foo', 'flags=(complain, audit)', '', expected_flags=None)
|
|
def test_set_flags_08(self):
|
|
self._test_set_flags('/foo', '( complain )', 'audit ', whitespace=' ', profile_name=None)
|
|
def test_set_flags_09(self):
|
|
self._test_set_flags('profile /foo', 'flags=(complain)', 'audit')
|
|
def test_set_flags_10(self):
|
|
self._test_set_flags('profile xy /foo', 'flags=(complain)', 'audit', profile_name='xy')
|
|
def test_set_flags_11(self):
|
|
self._test_set_flags('profile "/foo bar"', 'flags=(complain)', 'audit', profile_name='/foo bar')
|
|
def test_set_flags_12(self):
|
|
self._test_set_flags('profile xy "/foo bar"', 'flags=(complain)', 'audit', profile_name='xy')
|
|
|
|
# test handling of hat flags
|
|
def test_set_flags_with_hat_01(self):
|
|
self._test_set_flags('/foo', 'flags=(complain)', 'audit',
|
|
more_rules='\n ^foobar {\n}\n',
|
|
expected_more_rules='\n ^foobar flags=(audit) {\n}\n'
|
|
)
|
|
|
|
def test_set_flags_with_hat_02(self):
|
|
self._test_set_flags('/foo', 'flags=(complain)', 'audit',
|
|
profile_name=None,
|
|
more_rules='\n ^foobar {\n}\n',
|
|
expected_more_rules='\n ^foobar flags=(audit) {\n}\n'
|
|
)
|
|
|
|
def test_set_flags_with_hat_03(self):
|
|
self._test_set_flags('/foo', 'flags=(complain)', 'audit',
|
|
more_rules='\n^foobar (attach_disconnected) { # comment\n}\n', # XXX attach_disconnected will be lost!
|
|
expected_more_rules='\n^foobar flags=(audit) { # comment\n}\n'
|
|
)
|
|
|
|
def test_set_flags_with_hat_04(self):
|
|
self._test_set_flags('/foo', '', 'audit',
|
|
more_rules='\n hat foobar (attach_disconnected) { # comment\n}\n', # XXX attach_disconnected will be lost!
|
|
expected_more_rules='\n hat foobar flags=(audit) { # comment\n}\n'
|
|
)
|
|
|
|
# test handling of child profiles
|
|
def test_set_flags_with_child_01(self):
|
|
self._test_set_flags('/foo', 'flags=(complain)', 'audit',
|
|
profile_name=None,
|
|
more_rules='\n profile /bin/bar {\n}\n',
|
|
expected_more_rules='\n profile /bin/bar flags=(audit) {\n}\n'
|
|
)
|
|
|
|
#def test_set_flags_with_child_02(self):
|
|
# XXX child profile flags aren't changed if profile parameter is not None
|
|
#self._test_set_flags('/foo', 'flags=(complain)', 'audit',
|
|
# more_rules='\n profile /bin/bar {\n}\n',
|
|
# expected_more_rules='\n profile /bin/bar flags=(audit) {\n}\n'
|
|
#)
|
|
|
|
|
|
def test_set_flags_invalid_01(self):
|
|
with self.assertRaises(AppArmorBug):
|
|
self._test_set_flags('/foo', '()', None, check_new_flags=False)
|
|
def test_set_flags_invalid_02(self):
|
|
with self.assertRaises(AppArmorBug):
|
|
self._test_set_flags('/foo', 'flags=()', None, check_new_flags=False)
|
|
def test_set_flags_invalid_03(self):
|
|
with self.assertRaises(AppArmorException):
|
|
self._test_set_flags('/foo', '( )', '', check_new_flags=False)
|
|
def test_set_flags_invalid_04(self):
|
|
with self.assertRaises(AppArmorBug):
|
|
self._test_set_flags('/foo', 'flags=(complain, audit)', ' ', check_new_flags=False) # whitespace-only newflags
|
|
|
|
def test_set_flags_other_profile(self):
|
|
# test behaviour if the file doesn't contain the specified /foo profile
|
|
orig_prof = '/no-such-profile flags=(complain) {\n}'
|
|
self.file = write_file(self.tmpdir, 'profile', orig_prof)
|
|
|
|
with self.assertRaises(AppArmorBug):
|
|
set_profile_flags(self.file, '/foo', 'audit')
|
|
|
|
# the file should not be changed
|
|
real_new_prof = read_file(self.file)
|
|
self.assertEqual(orig_prof, real_new_prof)
|
|
|
|
def test_set_flags_no_profile_found(self):
|
|
# test behaviour if the file doesn't contain any profile
|
|
orig_prof = '# /comment flags=(complain) {\n# }'
|
|
self.file = write_file(self.tmpdir, 'profile', orig_prof)
|
|
|
|
with self.assertRaises(AppArmorBug):
|
|
set_profile_flags(self.file, None, 'audit')
|
|
|
|
# the file should not be changed
|
|
real_new_prof = read_file(self.file)
|
|
self.assertEqual(orig_prof, real_new_prof)
|
|
|
|
def test_set_flags_file_not_found(self):
|
|
with self.assertRaises(IOError):
|
|
set_profile_flags('%s/file-not-found' % self.tmpdir, '/foo', 'audit')
|
|
|
|
|
|
class AaTest_is_skippable_file(AATest):
|
|
def test_not_skippable_01(self):
|
|
self.assertFalse(is_skippable_file('bin.ping'))
|
|
def test_not_skippable_02(self):
|
|
self.assertFalse(is_skippable_file('usr.lib.dovecot.anvil'))
|
|
def test_not_skippable_03(self):
|
|
self.assertFalse(is_skippable_file('bin.~ping'))
|
|
def test_not_skippable_04(self):
|
|
self.assertFalse(is_skippable_file('bin.rpmsave.ping'))
|
|
def test_not_skippable_05(self):
|
|
# normally is_skippable_file should be called without directory, but it shouldn't hurt too much
|
|
self.assertFalse(is_skippable_file('/etc/apparmor.d/bin.ping'))
|
|
def test_not_skippable_06(self):
|
|
self.assertFalse(is_skippable_file('bin.pingrej'))
|
|
|
|
def test_skippable_01(self):
|
|
self.assertTrue(is_skippable_file('bin.ping.dpkg-new'))
|
|
def test_skippable_02(self):
|
|
self.assertTrue(is_skippable_file('bin.ping.dpkg-old'))
|
|
def test_skippable_03(self):
|
|
self.assertTrue(is_skippable_file('bin.ping..dpkg-dist'))
|
|
def test_skippable_04(self):
|
|
self.assertTrue(is_skippable_file('bin.ping..dpkg-bak'))
|
|
def test_skippable_05(self):
|
|
self.assertTrue(is_skippable_file('bin.ping.rpmnew'))
|
|
def test_skippable_06(self):
|
|
self.assertTrue(is_skippable_file('bin.ping.rpmsave'))
|
|
def test_skippable_07(self):
|
|
self.assertTrue(is_skippable_file('bin.ping.orig'))
|
|
def test_skippable_08(self):
|
|
self.assertTrue(is_skippable_file('bin.ping.rej'))
|
|
def test_skippable_09(self):
|
|
self.assertTrue(is_skippable_file('bin.ping~'))
|
|
def test_skippable_10(self):
|
|
self.assertTrue(is_skippable_file('.bin.ping'))
|
|
def test_skippable_11(self):
|
|
self.assertTrue(is_skippable_file('')) # empty filename
|
|
def test_skippable_12(self):
|
|
self.assertTrue(is_skippable_file('/etc/apparmor.d/')) # directory without filename
|
|
def test_skippable_13(self):
|
|
self.assertTrue(is_skippable_file('README'))
|
|
|
|
|
|
class AaTest_is_skippable_dir(AATest):
|
|
tests = [
|
|
('disable', True),
|
|
('cache', True),
|
|
('lxc', True),
|
|
('force-complain', True),
|
|
('/etc/apparmor.d/cache', True),
|
|
('/etc/apparmor.d/lxc/', True),
|
|
|
|
('dont_disable', False),
|
|
('/etc/apparmor.d/cache_foo', False),
|
|
('abstractions', False),
|
|
('apache2.d', False),
|
|
('/etc/apparmor.d/apache2.d', False),
|
|
('local', False),
|
|
('/etc/apparmor.d/local/', False),
|
|
('tunables', False),
|
|
('/etc/apparmor.d/tunables', False),
|
|
('/etc/apparmor.d/tunables/multiarch.d', False),
|
|
('/etc/apparmor.d/tunables/xdg-user-dirs.d', False),
|
|
('/etc/apparmor.d/tunables/home.d', False),
|
|
('/etc/apparmor.d/abstractions', False),
|
|
('/etc/apparmor.d/abstractions/ubuntu-browsers.d', False),
|
|
('/etc/apparmor.d/abstractions/apparmor_api', False),
|
|
]
|
|
|
|
def _run_test(self, params, expected):
|
|
self.assertEqual(is_skippable_dir(params), expected)
|
|
|
|
class AaTest_parse_profile_start(AATest):
|
|
def _parse(self, line, profile, hat):
|
|
return parse_profile_start(line, 'somefile', 1, profile, hat)
|
|
# (profile, hat, flags, in_contained_hat, pps_set_profile, pps_set_hat_external)
|
|
|
|
def test_parse_profile_start_01(self):
|
|
result = self._parse('/foo {', None, None)
|
|
expected = ('/foo', '/foo', None, None, False, False, False)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_parse_profile_start_02(self):
|
|
result = self._parse('/foo (complain) {', None, None)
|
|
expected = ('/foo', '/foo', None, 'complain', False, False, False)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_parse_profile_start_03(self):
|
|
result = self._parse('profile foo /foo {', None, None) # named profile
|
|
expected = ('foo', 'foo', '/foo', None, False, False, False)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_parse_profile_start_04(self):
|
|
result = self._parse('profile /foo {', '/bar', '/bar') # child profile
|
|
expected = ('/bar', '/foo', None, None, True, True, False)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_parse_profile_start_05(self):
|
|
result = self._parse('/foo//bar {', None, None) # external hat
|
|
expected = ('/foo', 'bar', None, None, False, False, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_parse_profile_start_06(self):
|
|
result = self._parse('profile "/foo" (complain) {', None, None)
|
|
expected = ('/foo', '/foo', None, 'complain', False, False, False)
|
|
self.assertEqual(result, expected)
|
|
|
|
|
|
def test_parse_profile_start_invalid_01(self):
|
|
with self.assertRaises(AppArmorException):
|
|
self._parse('/foo {', '/bar', '/bar') # child profile without profile keyword
|
|
|
|
def test_parse_profile_start_invalid_02(self):
|
|
with self.assertRaises(AppArmorBug):
|
|
self._parse('xy', '/bar', '/bar') # not a profile start
|
|
|
|
class AaTest_parse_profile_data(AATest):
|
|
def test_parse_empty_profile_01(self):
|
|
prof = parse_profile_data('/foo {\n}\n'.split(), 'somefile', False)
|
|
|
|
self.assertEqual(list(prof.keys()), ['/foo'])
|
|
self.assertEqual(list(prof['/foo'].keys()), ['/foo'])
|
|
self.assertEqual(prof['/foo']['/foo']['name'], '/foo')
|
|
self.assertEqual(prof['/foo']['/foo']['filename'], 'somefile')
|
|
self.assertEqual(prof['/foo']['/foo']['flags'], None)
|
|
|
|
def test_parse_empty_profile_02(self):
|
|
with self.assertRaises(AppArmorException):
|
|
# file contains two profiles with the same name
|
|
parse_profile_data('profile /foo {\n}\nprofile /foo {\n}\n'.split(), 'somefile', False)
|
|
|
|
class AaTest_separate_vars(AATest):
|
|
tests = [
|
|
('' , set() ),
|
|
(' ' , set() ),
|
|
(' foo bar' , {'foo', 'bar' }),
|
|
('foo " ' , {'foo' }), # XXX " is ignored
|
|
(' " foo ' , {' "', 'foo' }), # XXX really?
|
|
(' foo bar ' , {'foo', 'bar' }),
|
|
(' foo bar # comment' , {'foo', 'bar', 'comment' }), # XXX should comments be stripped?
|
|
('foo' , {'foo' }),
|
|
('"foo" "bar baz"' , {'foo', 'bar baz' }),
|
|
('foo "bar baz" xy' , {'foo', 'bar baz', 'xy' }),
|
|
('foo "bar baz ' , {'foo', 'bar', 'baz' }), # half-quoted
|
|
(' " foo" bar' , {' foo', 'bar' }),
|
|
]
|
|
|
|
def _run_test(self, params, expected):
|
|
result = separate_vars(params)
|
|
self.assertEqual(result, expected)
|
|
|
|
|
|
class AaTest_store_list_var(AATest):
|
|
tests = [
|
|
# old var value operation expected (False for exception)
|
|
([ {} , 'foo' , '=' ], {'foo'} ), # set
|
|
([ {} , 'foo bar' , '=' ], {'foo', 'bar'} ), # set multi
|
|
([ {'@{var}': {'foo'}} , 'bar' , '=' ], False ), # redefine var
|
|
([ {} , 'bar' , '+=' ], False ), # add to undefined var
|
|
([ {'@{var}': {'foo'}} , 'bar' , '+=' ], {'foo', 'bar'} ), # add
|
|
([ {'@{var}': {'foo'}} , 'bar baz' , '+=' ], {'foo', 'bar', 'baz'} ), # add multi
|
|
([ {'@{var}': {'foo', 'xy'}} , 'bar baz' , '+=' ], {'foo', 'xy', 'bar', 'baz'} ), # add multi to multi
|
|
([ {} , 'foo' , '-=' ], False ), # unknown operation
|
|
]
|
|
|
|
def _run_test(self, params, expected):
|
|
var = params[0]
|
|
value = params[1]
|
|
operation = params[2]
|
|
|
|
if not expected:
|
|
with self.assertRaises(AppArmorException):
|
|
store_list_var(var, '@{var}', value, operation, 'somefile')
|
|
return
|
|
|
|
# dumy value that must not be changed
|
|
var['@{foo}'] = {'one', 'two'}
|
|
|
|
exp_var = {
|
|
'@{foo}': {'one', 'two'},
|
|
'@{var}': expected,
|
|
}
|
|
|
|
store_list_var(var, '@{var}', value, operation, 'somefile')
|
|
|
|
self.assertEqual(var.keys(), exp_var.keys())
|
|
|
|
for key in exp_var:
|
|
self.assertEqual(var[key], exp_var[key])
|
|
|
|
|
|
class AaTest_write_header(AATest):
|
|
tests = [
|
|
# name embedded_hat write_flags depth flags attachment prof.keyw. comment expected
|
|
(['/foo', False, True, 1, 'complain', None, None, None ], ' /foo flags=(complain) {'),
|
|
(['/foo', True, True, 1, 'complain', None, None, None ], ' profile /foo flags=(complain) {'),
|
|
(['/foo sp', False, False, 2, 'complain', None, None, None ], ' "/foo sp" {'),
|
|
(['/foo' ,False, False, 2, 'complain', None, None, None ], ' /foo {'),
|
|
(['/foo', True, False, 2, 'complain', None, None, None ], ' profile /foo {'),
|
|
(['/foo', False, True, 0, None, None, None, None ], '/foo {'),
|
|
(['/foo', True, True, 0, None, None, None, None ], 'profile /foo {'),
|
|
(['/foo', False, False, 0, None, None, None, None ], '/foo {'),
|
|
(['/foo', True, False, 0, None, None, None, None ], 'profile /foo {'),
|
|
(['bar', False, True, 1, 'complain', None, None, None ], ' profile bar flags=(complain) {'),
|
|
(['bar', False, True, 1, 'complain', '/foo', None, None ], ' profile bar /foo flags=(complain) {'),
|
|
(['bar', True, True, 1, 'complain', '/foo', None, None ], ' profile bar /foo flags=(complain) {'),
|
|
(['bar baz', False, True, 1, None, '/foo', None, None ], ' profile "bar baz" /foo {'),
|
|
(['bar', True, True, 1, None, '/foo', None, None ], ' profile bar /foo {'),
|
|
(['bar baz', False, True, 1, 'complain', '/foo sp', None, None ], ' profile "bar baz" "/foo sp" flags=(complain) {'),
|
|
(['^foo', False, True, 1, 'complain', None, None, None ], ' profile ^foo flags=(complain) {'),
|
|
(['^foo', True, True, 1, 'complain', None, None, None ], ' ^foo flags=(complain) {'),
|
|
(['^foo', True, True, 1.5, 'complain', None, None, None ], ' ^foo flags=(complain) {'),
|
|
(['^foo', True, True, 1.3, 'complain', None, None, None ], ' ^foo flags=(complain) {'),
|
|
(['/foo', False, True, 1, 'complain', None, 'profile', None ], ' profile /foo flags=(complain) {'),
|
|
(['/foo', True, True, 1, 'complain', None, 'profile', None ], ' profile /foo flags=(complain) {'),
|
|
(['/foo', False, True, 1, 'complain', None, None, '# x' ], ' /foo flags=(complain) { # x'),
|
|
(['/foo', True, True, 1, None, None, None, '# x' ], ' profile /foo { # x'),
|
|
(['/foo', False, True, 1, None, None, 'profile', '# x' ], ' profile /foo { # x'),
|
|
(['/foo', True, True, 1, 'complain', None, 'profile', '# x' ], ' profile /foo flags=(complain) { # x'),
|
|
]
|
|
|
|
def _run_test(self, params, expected):
|
|
name = params[0]
|
|
embedded_hat = params[1]
|
|
write_flags = params[2]
|
|
depth = params[3]
|
|
prof_data = { 'flags': params[4], 'attachment': params[5], 'profile_keyword': params[6], 'header_comment': params[7] }
|
|
|
|
result = write_header(prof_data, depth, name, embedded_hat, write_flags)
|
|
self.assertEqual(result, [expected])
|
|
|
|
class AaTest_serialize_parse_profile_start(AATest):
|
|
def _parse(self, line, profile, hat, prof_data_profile, prof_data_external):
|
|
# 'correct' is always True in the code that uses serialize_parse_profile_start() (set some lines above the function call)
|
|
return serialize_parse_profile_start(line, 'somefile', 1, profile, hat, prof_data_profile, prof_data_external, True)
|
|
|
|
def test_serialize_parse_profile_start_01(self):
|
|
result = self._parse('/foo {', None, None, False, False)
|
|
expected = ('/foo', '/foo', None, None, False, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_02(self):
|
|
result = self._parse('/foo (complain) {', None, None, False, False)
|
|
expected = ('/foo', '/foo', None, 'complain', False, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_03(self):
|
|
result = self._parse('profile foo /foo {', None, None, False, False) # named profile
|
|
expected = ('foo', 'foo', '/foo', None, False, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_04(self):
|
|
result = self._parse('profile /foo {', '/bar', '/bar', False, False) # child profile
|
|
expected = ('/bar', '/foo', None, None, True, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_05(self):
|
|
result = self._parse('/foo//bar {', None, None, False, False) # external hat
|
|
expected = ('/foo', 'bar', None, None, False, False) # note correct == False here
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_06(self):
|
|
result = self._parse('profile "/foo" (complain) {', None, None, False, False)
|
|
expected = ('/foo', '/foo', None, 'complain', False, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_07(self):
|
|
result = self._parse('/foo {', None, None, True, False)
|
|
expected = ('/foo', '/foo', None, None, False, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_08(self):
|
|
result = self._parse('/foo {', None, None, False, True)
|
|
expected = ('/foo', '/foo', None, None, False, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_09(self):
|
|
result = self._parse('/foo {', None, None, True, True)
|
|
expected = ('/foo', '/foo', None, None, False, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_10(self):
|
|
result = self._parse('profile /foo {', '/bar', '/bar', True, False) # child profile
|
|
expected = ('/bar', '/foo', None, None, True, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_11(self):
|
|
result = self._parse('profile /foo {', '/bar', '/bar', False, True) # child profile
|
|
expected = ('/bar', '/foo', None, None, True, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_serialize_parse_profile_start_12(self):
|
|
result = self._parse('profile /foo {', '/bar', '/bar', True, True) # child profile
|
|
expected = ('/bar', '/foo', None, None, True, True)
|
|
self.assertEqual(result, expected)
|
|
|
|
class AaTestInvalid_serialize_parse_profile_start(AATest):
|
|
tests = [
|
|
# line profile hat p_d_profile p_d_external expected
|
|
(['/foo {', '/bar', '/bar', False, False ], AppArmorException), # child profile without 'profile' keyword
|
|
(['profile /foo {', '/bar', '/xy', False, False ], AppArmorException), # already inside a child profile - nesting level reached
|
|
(['/ext//hat {', '/bar', '/bar', True, True ], AppArmorException), # external hat inside a profile
|
|
(['/ext//hat {', '/bar', '/bar', True, False ], AppArmorException), # external hat inside a profile
|
|
(['xy', '/bar', '/bar', False, False ], AppArmorBug ), # not a profile start
|
|
]
|
|
|
|
def _run_test(self, params, expected):
|
|
line = params[0]
|
|
profile = params[1]
|
|
hat = params[2]
|
|
prof_data_profile = params[3]
|
|
prof_data_external = params[4]
|
|
|
|
with self.assertRaises(expected):
|
|
# 'correct' is always True in the code that uses serialize_parse_profile_start() (set some lines above the function call)
|
|
serialize_parse_profile_start(line, 'somefile', 1, profile, hat, prof_data_profile, prof_data_external, True)
|
|
|
|
|
|
setup_all_loops(__name__)
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=2)
|