#!/usr/bin/env python

from __future__ import print_function

import sys

import getopt

import semanage

usage = "\

Choose one of the following tests:\n\

-m for modules\n\

-u for users\n\

-U for add user (warning this will write!)\n\

-s for seusers\n\

-S for add seuser (warning this will write!)\n\

-p for ports\n\

-P for add port (warning this will write!)\n\

-f for file contexts \n\

-F for add file context (warning this will write!)\n\

-i for network interfaces \n\

-I for add network interface (warning this will write!)\n\

-b for booleans \n\

-B for add boolean (warning this will write!)\n\

-c for aCtive booleans\n\

-C for set aCtive boolean (warning this will write!)\n\n\

-n for network nodes\n\

-N for add node (warning this will write!)\n\n\

Other options:\n\

-h for this help\n\

-v for verbose output\

"

class Usage(Exception):

    def __init__(self, msg):

        Exception.__init__(self)

        self.msg = msg

class Status(Exception):

    def __init__(self, msg):

        Exception.__init__(self)

        self.msg = msg

class Error(Exception):

    def __init__(self, msg):

        Exception.__init__(self)

        self.msg = msg

class Tests:

    def __init__(self):

        self.all = False

        self.users = False

        self.writeuser = False

        self.seusers = False

        self.writeseuser = False

        self.ports = False

        self.writeport = False

        self.fcontexts = False

        self.writefcontext = False

        self.interfaces = False

        self.writeinterface = False

        self.booleans = False

        self.writeboolean = False

        self.abooleans = False

        self.writeaboolean = False

        self.nodes = False

        self.writenode = False

        self.modules = False

        self.verbose = False

    def selected(self):

        return (

            self.all or

            self.users or

            self.modules or

            self.seusers or

            self.ports or

            self.fcontexts or

            self.interfaces or

            self.booleans or

            self.abooleans or

            self.writeuser or

            self.writeseuser or

            self.writeport or

            self.writefcontext or

            self.writeinterface or

            self.writeboolean or

            self.writeaboolean or

            self.nodes or

            self.writenode)

    def run(self, handle):

        if self.users or self.all:

            self.test_users(handle)

            print("")

        if self.seusers or self.all:

            self.test_seusers(handle)

            print("")

        if self.ports or self.all:

            self.test_ports(handle)

            print("")

        if self.modules or self.all:

            self.test_modules(handle)

            print("")

        if self.fcontexts or self.all:

            self.test_fcontexts(handle)

            print("")

        if self.interfaces or self.all:

            self.test_interfaces(handle)

            print("")

        if self.booleans or self.all:

            self.test_booleans(handle)

            print("")

        if self.abooleans or self.all:

            self.test_abooleans(handle)

            print("")

        if self.nodes or self.all:

            self.test_nodes(handle)

            print("")

        if self.writeuser or self.all:

            self.test_writeuser(handle)

            print("")

        if self.writeseuser or self.all:

            self.test_writeseuser(handle)

            print("")

        if self.writeport or self.all:

            self.test_writeport(handle)

            print("")

        if self.writefcontext or self.all:

            self.test_writefcontext(handle)

            print("")

        if self.writeinterface or self.all:

            self.test_writeinterface(handle)

            print("")

        if self.writeboolean or self.all:

            self.test_writeboolean(handle)

            print("")

        if self.writeaboolean or self.all:

            self.test_writeaboolean(handle)

            print("")

        if self.writenode or self.all:

            self.test_writenode(handle)

            print("")

    def test_modules(self, sh):

        print("Testing modules...")

        (trans_cnt, mlist, mlist_size) = semanage.semanage_module_list(sh)

        print("Transaction number: %s" % trans_cnt)

        print("Module list size: %s" % mlist_size)

        if self.verbose:

            print("List reference: %s" % mlist)

        if mlist_size == 0:

            print("No modules installed!")

            print("This is not necessarily a test failure.")

            return

        for idx in range(mlist_size):

            module = semanage.semanage_module_list_nth(mlist, idx)

            if self.verbose:

                print("Module reference: %s" % module)

            print("Module name: %s" % semanage.semanage_module_get_name(module))

    def test_seusers(self, sh):

        print("Testing seusers...")

        (status, slist) = semanage.semanage_seuser_list(sh)

        if status < 0:

            raise Error("Could not list seusers")

        print("Query status (commit number): %s" % status)

        if len(slist) == 0:

            print("No seusers found!")

            print("This is not necessarily a test failure.")

            return

        for seuser in slist:

            if self.verbose:

                print("seseuser reference: %s" % seuser)

            print("seuser name: %s" % semanage.semanage_seuser_get_name(seuser))

            print("   seuser mls range: %s" % semanage.semanage_seuser_get_mlsrange(seuser))

            print("   seuser sename: %s" % semanage.semanage_seuser_get_sename(seuser))

            semanage.semanage_seuser_free(seuser)

    def test_users(self, sh):

        print("Testing users...")

        (status, ulist) = semanage.semanage_user_list(sh)

        if status < 0:

            raise Error("Could not list users")

        print("Query status (commit number): %s" % status)

        if len(ulist) == 0:

            print("No users found!")

            print("This is not necessarily a test failure.")

            return

        for user in ulist:

            if self.verbose:

                print("User reference: %s" % user)

            print("User name: %s" % semanage.semanage_user_get_name(user))

            print("   User labeling prefix: %s" % semanage.semanage_user_get_prefix(user))

            print("   User mls level: %s" % semanage.semanage_user_get_mlslevel(user))

            print("   User mls range: %s" % semanage.semanage_user_get_mlsrange(user))

            print("   User number of roles: %s" % semanage.semanage_user_get_num_roles(user))

            print("   User roles: ")

            (status, rlist) = semanage.semanage_user_get_roles(sh, user)

            if status < 0:

                raise Error("Could not get user roles")

            for role in rlist:

                print("      %s" % role)

            semanage.semanage_user_free(user)

    def test_ports(self, sh):

        print("Testing ports...")

        (status, plist) = semanage.semanage_port_list(sh)

        if status < 0:

            raise Error("Could not list ports")

        print("Query status (commit number): %s" % status)

        if len(plist) == 0:

            print("No ports found!")

            print("This is not necessarily a test failure.")

            return

        for port in plist:

            if self.verbose:

                print("Port reference: %s" % port)

            low = semanage.semanage_port_get_low(port)

            high = semanage.semanage_port_get_high(port)

            con = semanage.semanage_port_get_con(port)

            proto = semanage.semanage_port_get_proto(port)

            proto_str = semanage.semanage_port_get_proto_str(proto)

            if low == high:

                range_str = str(low)

            else:

                range_str = str(low) + "-" + str(high)

            (rc, con_str) = semanage.semanage_context_to_string(sh, con)

            if rc < 0:

                con_str = ""

            print("Port: %s %s Context: %s" % (range_str, proto_str, con_str))

            semanage.semanage_port_free(port)

    def test_fcontexts(self, sh):

        print("Testing file contexts...")

        (status, flist) = semanage.semanage_fcontext_list(sh)

        if status < 0:

            raise Error("Could not list file contexts")

        print("Query status (commit number): %s" % status)

        if len(flist) == 0:

            print("No file contexts found!")

            print("This is not necessarily a test failure.")

            return

        for fcon in flist:

            if self.verbose:

                print("File Context reference: %s" % fcon)

            expr = semanage.semanage_fcontext_get_expr(fcon)

            type = semanage.semanage_fcontext_get_type(fcon)

            type_str = semanage.semanage_fcontext_get_type_str(type)

            con = semanage.semanage_fcontext_get_con(fcon)

            if not con:

                con_str = "<<none>>"

            else:

                (rc, con_str) = semanage.semanage_context_to_string(sh, con)

                if rc < 0:

                    con_str = ""

            print("File Expr: %s [%s] Context: %s" % (expr, type_str, con_str))

            semanage.semanage_fcontext_free(fcon)

    def test_interfaces(self, sh):

        print("Testing network interfaces...")

        (status, ilist) = semanage.semanage_iface_list(sh)

        if status < 0:

            raise Error("Could not list interfaces")

        print("Query status (commit number): %s" % status)

        if len(ilist) == 0:

            print("No network interfaces found!")

            print("This is not necessarily a test failure.")

            return

        for iface in ilist:

            if self.verbose:

                print("Interface reference: %s" % iface)

            name = semanage.semanage_iface_get_name(iface)

            msg_con = semanage.semanage_iface_get_msgcon(iface)

            if_con = semanage.semanage_iface_get_ifcon(iface)

            (rc, msg_con_str) = semanage.semanage_context_to_string(sh, msg_con)

            if rc < 0:

                msg_con_str = ""

            (rc, if_con_str) = semanage.semanage_context_to_string(sh, if_con)

            if rc < 0:

                if_con_str = ""

            print("Interface: %s Context: %s Message Context: %s" % (name, if_con_str, msg_con_str))

            semanage.semanage_iface_free(iface)

    def test_booleans(self, sh):

        print("Testing booleans...")

        (status, blist) = semanage.semanage_bool_list(sh)

        if status < 0:

            raise Error("Could not list booleans")

        print("Query status (commit number): %s" % status)

        if len(blist) == 0:

            print("No booleans found!")

            print("This is not necessarily a test failure.")

            return

        for pbool in blist:

            if self.verbose:

                print("Boolean reference: %s" % pbool)

            name = semanage.semanage_bool_get_name(pbool)

            value = semanage.semanage_bool_get_value(pbool)

            print("Boolean: %s Value: %s" % (name, value))

            semanage.semanage_bool_free(pbool)

    def test_abooleans(self, sh):

        print("Testing active booleans...")

        (status, ablist) = semanage.semanage_bool_list_active(sh)

        if status < 0:

            raise Error("Could not list active booleans")

        print("Query status (commit number): %s" % status)

        if len(ablist) == 0:

            print("No active booleans found!")

            print("This is not necessarily a test failure.")

            return

        for abool in ablist:

            if self.verbose:

                print("Active boolean reference: %s" % abool)

            name = semanage.semanage_bool_get_name(abool)

            value = semanage.semanage_bool_get_value(abool)

            print("Active Boolean: %s Value: %s" % (name, value))

            semanage.semanage_bool_free(abool)

    def test_nodes(self, sh):

        print("Testing network nodes...")

        (status, nlist) = semanage.semanage_node_list(sh)

        if status < 0:

            raise Error("Could not list network nodes")

        print("Query status (commit number): %s" % status)

        if len(nlist) == 0:

            print("No network nodes found!")

            print("This is not necessarily a test failure.")

            return

        for node in nlist:

            if self.verbose:

                print("Network node reference: %s" % node)

            (status, addr) = semanage.semanage_node_get_addr(sh, node)

            if status < 0:

                addr = ""

            (status, mask) = semanage.semanage_node_get_mask(sh, node)

            if status < 0:

                mask = ""

            proto = semanage.semanage_node_get_proto(node)

            proto_str = semanage.semanage_node_get_proto_str(proto)

            con = semanage.semanage_node_get_con(node)

            (status, con_str) = semanage.semanage_context_to_string(sh, con)

            if status < 0:

                con_str = ""

            print("Network Node: %s/%s (%s) Context: %s" % (addr, mask, proto_str, con_str))

            semanage.semanage_node_free(node)

    def test_writeuser(self, sh):

        print("Testing user write...")

        (status, user) = semanage.semanage_user_create(sh)

        if status < 0:

            raise Error("Could not create user object")

        if self.verbose:

            print("User object created")

        status = semanage.semanage_user_set_name(sh, user, "testPyUser")

        if status < 0:

            raise Error("Could not set user name")

        if self.verbose:

            print("User name set: %s" % semanage.semanage_user_get_name(user))

        status = semanage.semanage_user_add_role(sh, user, "user_r")

        if status < 0:

            raise Error("Could not add role")

        status = semanage.semanage_user_set_prefix(sh, user, "user")

        if status < 0:

            raise Error("Could not set labeling prefix")

        if self.verbose:

            print("User prefix set: %s" % semanage.semanage_user_get_prefix(user))

        status = semanage.semanage_user_set_mlsrange(sh, user, "s0")

        if status < 0:

            raise Error("Could not set MLS range")

        if self.verbose:

            print("User mlsrange: %s" % semanage.semanage_user_get_mlsrange(user))

        status = semanage.semanage_user_set_mlslevel(sh, user, "s0")

        if status < 0:

            raise Error("Could not set MLS level")

        if self.verbose:

            print("User mlslevel: %s" % semanage.semanage_user_get_mlslevel(user))

        (status, key) = semanage.semanage_user_key_extract(sh, user)

        if status < 0:

            raise Error("Could not extract user key")

        if self.verbose:

            print("User key extracted: %s" % key)

        (status, exists) = semanage.semanage_user_exists_local(sh, key)

        if status < 0:

            raise Error("Could not check if user exists")

        if self.verbose:

            print("Exists status (commit number): %s" % status)

        if exists:

            (status, old_user) = semanage.semanage_user_query_local(sh, key)

            if status < 0:

                raise Error("Could not query old user")

            if self.verbose:

                print("Query status (commit number): %s" % status)

        print("Starting transaction..")

        status = semanage.semanage_begin_transaction(sh)

        if status < 0:

            raise Error("Could not start semanage transaction")

        status = semanage.semanage_user_modify_local(sh, key, user)

        if status < 0:

            raise Error("Could not modify user")

        status = semanage.semanage_commit(sh)

        if status < 0:

            raise Error("Could not commit test transaction")

        print("Commit status (transaction number): %s" % status)

        status = semanage.semanage_begin_transaction(sh)

        if status < 0:

            raise Error("Could not start semanage transaction")

        if not exists:

            print("Removing user...")

            status = semanage.semanage_user_del_local(sh, key)

            if status < 0:

                raise Error("Could not delete test user")

            if self.verbose:

                print("User delete: %s" % status)

        else:

            print("Resetting user...")

            status = semanage.semanage_user_modify_local(sh, key, old_user)

            if status < 0:

                raise Error("Could not reset test user")

            if self.verbose:

                print("User modify: %s" % status)

        status = semanage.semanage_commit(sh)

        if status < 0:

            raise Error("Could not commit reset transaction")

        print("Commit status (transaction number): %s" % status)

        semanage.semanage_user_key_free(key)

        semanage.semanage_user_free(user)

        if exists:

            semanage.semanage_user_free(old_user)

    def test_writeseuser(self, sh):

        print("Testing seuser write...")

        (status, seuser) = semanage.semanage_seuser_create(sh)

        if status < 0:

            raise Error("Could not create SEUser object")

        if self.verbose:

            print("SEUser object created.")

        status = semanage.semanage_seuser_set_name(sh, seuser, "testPySEUser")

        if status < 0:

            raise Error("Could not set name")

        if self.verbose:

            print("SEUser name set: %s" % semanage.semanage_seuser_get_name(seuser))

        status = semanage.semanage_seuser_set_sename(sh, seuser, "root")

        if status < 0:

            raise Error("Could not set sename")

        if self.verbose:

            print("SEUser seuser: %s" % semanage.semanage_seuser_get_sename(seuser))

        status = semanage.semanage_seuser_set_mlsrange(sh, seuser, "s0:c0.c255")

        if status < 0:

            raise Error("Could not set MLS range")

        if self.verbose:

            print("SEUser mlsrange: %s" % semanage.semanage_seuser_get_mlsrange(seuser))

        (status, key) = semanage.semanage_seuser_key_extract(sh, seuser)

        if status < 0:

            raise Error("Could not extract SEUser key")

        if self.verbose:

            print("SEUser key extracted: %s" % key)

        (status, exists) = semanage.semanage_seuser_exists_local(sh, key)

        if status < 0:

            raise Error("Could not check if SEUser exists")

        if self.verbose:

            print("Exists status (commit number): %s" % status)

        if exists:

            (status, old_seuser) = semanage.semanage_seuser_query_local(sh, key)

            if status < 0:

                raise Error("Could not query old SEUser")

            if self.verbose:

                print("Query status (commit number): %s" % status)

        print("Starting transaction...")

        status = semanage.semanage_begin_transaction(sh)

        if status < 0:

            raise Error("Could not start semanage transaction")

        status = semanage.semanage_seuser_modify_local(sh, key, seuser)

        if status < 0:

            raise Error("Could not modify SEUser")

        status = semanage.semanage_commit(sh)

        if status < 0:

            raise Error("Could not commit test transaction")

        print("Commit status (transaction number): %s" % status)

        status = semanage.semanage_begin_transaction(sh)

        if status < 0:

            raise Error("Could not start semanage transaction")

        if not exists:

            print("Removing seuser...")

            status = semanage.semanage_seuser_del_local(sh, key)

            if status < 0:

                raise Error("Could not delete test SEUser")

            if self.verbose:

                print("Seuser delete: %s" % status)

        else:

            print("Resetting seuser...")

            status = semanage.semanage_seuser_modify_local(sh, key, old_seuser)

            if status < 0:

                raise Error("Could not reset test SEUser")

            if self.verbose:

                print("Seuser modify: %s" % status)

        status = semanage.semanage_commit(sh)

        if status < 0:

            raise Error("Could not commit reset transaction")

        print("Commit status (transaction number): %s" % status)

        semanage.semanage_seuser_key_free(key)

        semanage.semanage_seuser_free(seuser)

        if exists:

            semanage.semanage_seuser_free(old_seuser)

    def test_writeport(self, sh):

        print("Testing port write...")

        (status, port) = semanage.semanage_port_create(sh)

        if status < 0:

            raise Error("Could not create SEPort object")

        if self.verbose:

            print("SEPort object created.")

        semanage.semanage_port_set_range(port, 150, 200)

        low = semanage.semanage_port_get_low(port)

        high = semanage.semanage_port_get_high(port)

        if self.verbose:

            print("SEPort range set: %s-%s" % (low, high))

        semanage.semanage_port_set_proto(port, semanage.SEMANAGE_PROTO_TCP)

        if self.verbose:

            print("SEPort protocol set: %s" % semanage.semanage_port_get_proto_str(semanage.SEMANAGE_PROTO_TCP))

        (status, con) = semanage.semanage_context_create(sh)

        if status < 0:

            raise Error("Could not create SEContext object")

        if self.verbose:

            print("SEContext object created (for port).")

        status = semanage.semanage_context_set_user(sh, con, "system_u")

        if status < 0:

            raise Error("Could not set context user")

        if self.verbose:

            print("SEContext user: %s" % semanage.semanage_context_get_user(con))

        status = semanage.semanage_context_set_role(sh, con, "object_r")

        if status < 0:

            raise Error("Could not set context role")

        if self.verbose:

            print("SEContext role: %s" % semanage.semanage_context_get_role(con))

        status = semanage.semanage_context_set_type(sh, con, "http_port_t")

        if status < 0:

            raise Error("Could not set context type")

        if self.verbose:

            print("SEContext type: %s" % semanage.semanage_context_get_type(con))

        status = semanage.semanage_context_set_mls(sh, con, "s0:c0.c255")

        if status < 0:

            raise Error("Could not set context MLS fields")

        if self.verbose:

            print("SEContext mls: %s" % semanage.semanage_context_get_mls(con))

        status = semanage.semanage_port_set_con(sh, port, con)

        if status < 0:

            raise Error("Could not set SEPort context")

        if self.verbose:

            print("SEPort context set: %s" % con)

        (status, key) = semanage.semanage_port_key_extract(sh, port)

        if status < 0:

            raise Error("Could not extract SEPort key")

        if self.verbose:

            print("SEPort key extracted: %s" % key)

        (status, exists) = semanage.semanage_port_exists_local(sh, key)

        if status < 0:

            raise Error("Could not check if SEPort exists")

        if self.verbose:

            print("Exists status (commit number): %s" % status)

        if exists:

            (status, old_port) = semanage.semanage_port_query_local(sh, key)

            if status < 0:

                raise Error("Could not query old SEPort")

            if self.verbose:

                print("Query status (commit number): %s" % status)

        print("Starting transaction...")

        status = semanage.semanage_begin_transaction(sh)

        if status < 0:

            raise Error("Could not start semanage transaction")

        status = semanage.semanage_port_modify_local(sh, key, port)

        if status < 0:

            raise Error("Could not modify SEPort")

        status = semanage.semanage_commit(sh)

        if status < 0:

            raise Error("Could not commit test transaction")

        print("Commit status (transaction number): %s" % status)

        status = semanage.semanage_begin_transaction(sh)

        if status < 0:

            raise Error("Could not start semanage transaction")

        if not exists:

            print("Removing port range...")

            status = semanage.semanage_port_del_local(sh, key)

            if status < 0:

                raise Error("Could not delete test SEPort")

            if self.verbose:

                print("Port range delete: %s" % status)

        else:

            print("Resetting port range...")

            status = semanage.semanage_port_modify_local(sh, key, old_port)

            if status < 0:

                raise Error("Could not reset test SEPort")

            if self.verbose:

                print("Port range modify: %s" % status)

        status = semanage.semanage_commit(sh)

        if status < 0:

            raise Error("Could not commit reset transaction")

        print("Commit status (transaction number): %s" % status)

        semanage.semanage_context_free(con)

        semanage.semanage_port_key_free(key)

        semanage.semanage_port_free(port)

        if exists:

            semanage.semanage_port_free(old_port)

    def test_writefcontext(self, sh):

        print("Testing file context write...")

        (status, fcon) = semanage.semanage_fcontext_create(sh)

        if status < 0:

            raise Error("Could not create SEFcontext object")

        if self.verbose:

            print("SEFcontext object created.")

        status = semanage.semanage_fcontext_set_expr(sh, fcon, "/test/fcontext(/.*)?")

        if status < 0:

            raise Error("Could not set expression")

        if self.verbose:

            print("SEFContext expr set: %s" % semanage.semanage_fcontext_get_expr(fcon))

        semanage.semanage_fcontext_set_type(fcon, semanage.SEMANAGE_FCONTEXT_REG)

        if self.verbose:

            ftype = semanage.semanage_fcontext_get_type(fcon)

            print("SEFContext type set: %s" % semanage.semanage_fcontext_get_type_str(ftype))

        (status, con) = semanage.semanage_context_create(sh)

        if status < 0:

            raise Error("Could not create SEContext object")

        if self.verbose:

            print("SEContext object created (for file context).")

        status = semanage.semanage_context_set_user(sh, con, "system_u")

        if status < 0:

            raise Error("Could not set context user")

        if self.verbose:

            print("SEContext user: %s" % semanage.semanage_context_get_user(con))

        status = semanage.semanage_context_set_role(sh, con, "object_r")

        if status < 0:

            raise Error("Could not set context role")

        if self.verbose:

            print("SEContext role: %s" % semanage.semanage_context_get_role(con))

        status = semanage.semanage_context_set_type(sh, con, "default_t")

        if status < 0:

            raise Error("Could not set context type")

        if self.verbose:

            print("SEContext type: %s" % semanage.semanage_context_get_type(con))

        status = semanage.semanage_context_set_mls(sh, con, "s0:c0.c255")

        if status < 0:

            raise Error("Could not set context MLS fields")

        if self.verbose:

            print("SEContext mls: %s" % semanage.semanage_context_get_mls(con))

        status = semanage.semanage_fcontext_set_con(sh, fcon, con)

        if status < 0:

            raise Error("Could not set SEFcontext context")

        if self.verbose:

            print("SEFcontext context set: %s" % con)

        (status, key) = semanage.semanage_fcontext_key_extract(sh, fcon)

        if status < 0:

            raise Error("Could not extract SEFcontext key")

        if self.verbose:

            print("SEFcontext key extracted: %s" % key)

        (status, exists) = semanage.semanage_fcontext_exists_local(sh, key)

        if status < 0:

            raise Error("Could not check if SEFcontext exists")

        if self.verbose:

            print("Exists status (commit number): %s" % status)

        if exists:

            (status, old_fcontext) = semanage.semanage_fcontext_query_local(sh, key)

            if status < 0:

                raise Error("Could not query old SEFcontext")

            if self.verbose:

                print("Query status (commit number): %s" % status)

        print("Starting transaction...")

        status = semanage.semanage_begin_transaction(sh)

        if status < 0:

            raise Error("Could not start semanage transaction")

        status = semanage.semanage_fcontext_modify_local(sh, key, fcon)

        if status < 0:

            raise Error("Could not modify SEFcontext")

        status = semanage.semanage_commit(sh)

        if status < 0:

            raise Error("Could not commit test transaction")

        print("Commit status (transaction number): %s" % status)

        status = semanage.semanage_begin_transaction(sh)

        if status < 0:

            raise Error("Could not start semanage transaction")

        if not exists:

            print("Removing file context...")

            status = semanage.semanage_fcontext_del_local(sh, key)

            if status < 0:

                raise Error("Could not delete test SEFcontext")

            if self.verbose:

                print("File context delete: %s" % status)

        else:

            print("Resetting file context...")

            status = semanage.semanage_fcontext_modify_local(sh, key, old_fcontext)

            if status < 0:

                raise Error("Could not reset test FContext")

            if self.verbose:

                print("File context modify: %s" % status)

        status = semanage.semanage_commit(sh)

        if status < 0:

            raise Error("Could not commit reset transaction")

        print("Commit status (transaction number): %s" % status)

        semanage.semanage_context_free(con)

        semanage.semanage_fcontext_key_free(key)

        semanage.semanage_fcontext_free(fcon)

        if exists:

            semanage.semanage_fcontext_free(old_fcontext)

    def test_writeinterface(self, sh):

        print("Testing network interface write...")

        (status, iface) = semanage.semanage_iface_create(sh)

        if status < 0:

            raise Error("Could not create SEIface object")

        if self.verbose:

            print("SEIface object created.")

        status = semanage.semanage_iface_set_name(sh, iface, "test_iface")

        if status < 0:

            raise Error("Could not set SEIface name")

        if self.verbose:

            print("SEIface name set: %s" % semanage.semanage_iface_get_name(iface))

        (status, con) = semanage.semanage_context_create(sh)

        if status < 0:

            raise Error("Could not create SEContext object")

        if self.verbose:

            print("SEContext object created (for network interface)")

        status = semanage.semanage_context_set_user(sh, con, "system_u")

        if status < 0:

            raise Error("Could not set interface context user")

        if self.verbose:

            print("SEContext user: %s" % semanage.semanage_context_get_user(con))

        status = semanage.semanage_context_set_role(sh, con, "object_r")

        if status < 0:

            raise Error("Could not set interface context role")

        if self.verbose:

            print("SEContext role: %s" % semanage.semanage_context_get_role(con))

        status = semanage.semanage_context_set_type(sh, con, "default_t")

        if status < 0:

            raise Error("Could not set interface context type")

        if self.verbose:

            print("SEContext type: %s" % semanage.semanage_context_get_type(con))

        status = semanage.semanage_context_set_mls(sh, con, "s0:c0.c255")

        if status < 0:

            raise Error("Could not set interface context MLS fields")

        if self.verbose:

            print("SEContext mls: %s" % semanage.semanage_context_get_mls(con))

        status = semanage.semanage_iface_set_ifcon(sh, iface, con)

        if status < 0:

            raise Error("Could not set SEIface interface context")

        if self.verbose:

            print("SEIface interface context set: %s" % con)

        status = semanage.semanage_iface_set_msgcon(sh, iface, con)

        if status < 0:

            raise Error("Could not set SEIface message context")

        if self.verbose:

            print("SEIface message context set: %s" % con)

        (status, key) = semanage.semanage_iface_key_extract(sh, iface)

        if status < 0:

            raise Error("Could not extract SEIface key")

        if self.verbose:

            print("SEIface key extracted: %s" % key)

        (status, exists) = semanage.semanage_iface_exists_local(sh, key)

        if status < 0:

            raise Error("Could not check if SEIface exists")

        if self.verbose:

            print("Exists status (commit number): %s" % status)

        if exists:

            (status, old_iface) = semanage.semanage_iface_query_local(sh, key)

            if status < 0:

                raise Error("Could not query old SEIface")

            if self.verbose:

                print("Query status (commit number): %s" % status)

        print("Starting transaction...")

        status = semanage.semanage_begin_transaction(sh)