setools/tests/dta.py
2014-07-08 14:28:55 -04:00

471 lines
17 KiB
Python

# Copyright 2014, Tresys Technology, LLC
#
# This file is part of SETools.
#
# SETools is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# (at your option) any later version.
#
# SETools is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with SETools. If not, see <http://www.gnu.org/licenses/>.
#
import unittest
import networkx as nx
from libapol import SELinuxPolicy
from libapol.dta import DomainTransitionAnalysis
from libapol.policyrep.rule import RuleNotConditional
class InfoFlowAnalysisTest(unittest.TestCase):
def setUp(self):
self.p = SELinuxPolicy("tests/dta.conf")
self.a = DomainTransitionAnalysis(self.p)
self.a._build_graph()
def test_000_graph_structure(self):
"""DTA: verify graph structure."""
# don't check node list since the disconnected nodes are not
# removed after removing invalid domain transitions
edges = sorted(list(self.a.G.out_edges_iter()))
self.assertListEqual(edges, [("dyntrans100", "bothtrans200"),
("start", "dyntrans100"),
("start", "trans1"),
("trans1", "trans2"),
("trans2", "trans3"),
("trans3", "trans5")])
def test_001_bothtrans(self):
"""DTA: type_transition, setexeccon(), and setcon() transitions."""
s = "dyntrans100"
t = "bothtrans200"
e = "bothtrans200_exec"
# regular transition
r = self.a.G.edge[s][t]["transition"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, t)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["transition", "dyntransition"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# setexec perms
r = self.a.G.edge[s][t]["setexec"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, s)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["setexec", "setcurrent"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# exec perms
k = self.a.G.edge[s][t]["execute"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["execute"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["execute"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# entrypoint perms
k = self.a.G.edge[s][t]["entrypoint"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["entrypoint"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, t)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["entrypoint"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# type_transition
k = self.a.G.edge[s][t]["type_transition"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["type_transition"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "type_transition")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "process")
self.assertEqual(r[0].default, t)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# dynamic transition
r = self.a.G.edge[s][t]["dyntransition"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, t)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["transition", "dyntransition"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# setcurrent
r = self.a.G.edge[s][t]["setcurrent"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, s)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["setexec", "setcurrent"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
def test_010_dyntrans(self):
"""DTA: setcon() transition."""
s = "start"
t = "dyntrans100"
# regular transition
r = self.a.G.edge[s][t]["transition"]
self.assertEqual(len(r), 0)
# setexec perms
r = self.a.G.edge[s][t]["setexec"]
self.assertEqual(len(r), 0)
# exec perms
k = self.a.G.edge[s][t]["execute"].keys()
self.assertEqual(len(k), 0)
# entrypoint perms
k = self.a.G.edge[s][t]["entrypoint"].keys()
self.assertEqual(len(k), 0)
# type_transition
k = self.a.G.edge[s][t]["type_transition"].keys()
self.assertEqual(len(k), 0)
# dynamic transition
r = self.a.G.edge[s][t]["dyntransition"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, t)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["dyntransition"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# setcurrent
r = self.a.G.edge[s][t]["setcurrent"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, s)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["setcurrent"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
def test_020_trans(self):
"""DTA: type_transition transition."""
s = "start"
t = "trans1"
e = "trans1_exec"
# regular transition
r = self.a.G.edge[s][t]["transition"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, t)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["transition"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# setexec perms
r = self.a.G.edge[s][t]["setexec"]
self.assertEqual(len(r), 0)
# exec perms
k = self.a.G.edge[s][t]["execute"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["execute"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["execute"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# entrypoint perms
k = self.a.G.edge[s][t]["entrypoint"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["entrypoint"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, t)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["entrypoint"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# type_transition
k = self.a.G.edge[s][t]["type_transition"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["type_transition"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "type_transition")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "process")
self.assertEqual(r[0].default, t)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# dynamic transition
r = self.a.G.edge[s][t]["dyntransition"]
self.assertEqual(len(r), 0)
# setcurrent
r = self.a.G.edge[s][t]["setcurrent"]
self.assertEqual(len(r), 0)
def test_030_setexec(self):
"""DTA: setexec() transition."""
s = "trans1"
t = "trans2"
e = "trans2_exec"
# regular transition
r = self.a.G.edge[s][t]["transition"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, t)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["transition"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# setexec perms
r = self.a.G.edge[s][t]["setexec"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, s)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["setexec"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# exec perms
k = self.a.G.edge[s][t]["execute"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["execute"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["execute"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# entrypoint perms
k = self.a.G.edge[s][t]["entrypoint"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["entrypoint"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, t)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["entrypoint"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# type_transition
k = self.a.G.edge[s][t]["type_transition"].keys()
self.assertEqual(len(k), 0)
# dynamic transition
r = self.a.G.edge[s][t]["dyntransition"]
self.assertEqual(len(r), 0)
# setcurrent
r = self.a.G.edge[s][t]["setcurrent"]
self.assertEqual(len(r), 0)
def test_040_two_entrypoint(self):
"""DTA: 2 entrypoints, only one by type_transition."""
s = "trans2"
t = "trans3"
e = ["trans3_exec1", "trans3_exec2"]
# regular transition
r = self.a.G.edge[s][t]["transition"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, t)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["transition"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# setexec perms
r = self.a.G.edge[s][t]["setexec"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, s)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["setexec"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# exec perms
k = self.a.G.edge[s][t]["execute"].keys()
self.assertEqual(k, e)
r = self.a.G.edge[s][t]["execute"][e[0]]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e[0])
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["execute"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
r = self.a.G.edge[s][t]["execute"][e[1]]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e[1])
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["execute"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# entrypoint perms
k = self.a.G.edge[s][t]["entrypoint"].keys()
self.assertEqual(k, e)
r = self.a.G.edge[s][t]["entrypoint"][e[0]]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, t)
self.assertEqual(r[0].target, e[0])
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["entrypoint"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
r = self.a.G.edge[s][t]["entrypoint"][e[1]]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, t)
self.assertEqual(r[0].target, e[1])
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["entrypoint"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# type_transition
k = self.a.G.edge[s][t]["type_transition"].keys()
self.assertEqual(k, [e[0]])
r = self.a.G.edge[s][t]["type_transition"][e[0]]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "type_transition")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e[0])
self.assertEqual(r[0].tclass, "process")
self.assertEqual(r[0].default, t)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# dynamic transition
r = self.a.G.edge[s][t]["dyntransition"]
self.assertEqual(len(r), 0)
# setcurrent
r = self.a.G.edge[s][t]["setcurrent"]
self.assertEqual(len(r), 0)
def test_050_cond_type_trans(self):
"""DTA: conditional type_transition."""
s = "trans3"
t = "trans5"
e = "trans5_exec"
# regular transition
r = self.a.G.edge[s][t]["transition"]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, t)
self.assertEqual(r[0].tclass, "process")
self.assertSetEqual(set(["transition"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# setexec perms
r = self.a.G.edge[s][t]["setexec"]
self.assertEqual(len(r), 0)
# exec perms
k = self.a.G.edge[s][t]["execute"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["execute"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["execute"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# entrypoint perms
k = self.a.G.edge[s][t]["entrypoint"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["entrypoint"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "allow")
self.assertEqual(r[0].source, t)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "file")
self.assertSetEqual(set(["entrypoint"]), r[0].perms)
self.assertRaises(RuleNotConditional, getattr, r[0], "conditional")
# type_transition
k = self.a.G.edge[s][t]["type_transition"].keys()
self.assertEqual(k, [e])
r = self.a.G.edge[s][t]["type_transition"][e]
self.assertEqual(len(r), 1)
self.assertEqual(r[0].ruletype, "type_transition")
self.assertEqual(r[0].source, s)
self.assertEqual(r[0].target, e)
self.assertEqual(r[0].tclass, "process")
self.assertEqual(r[0].default, t)
self.assertEqual(r[0].conditional, "trans5")
# dynamic transition
r = self.a.G.edge[s][t]["dyntransition"]
self.assertEqual(len(r), 0)
# setcurrent
r = self.a.G.edge[s][t]["setcurrent"]
self.assertEqual(len(r), 0)