xref: /aosp_15_r20/external/selinux/dbus/selinux_client.py (revision 2d543d20722ada2425b5bdab9d0d1d29470e7bba)
1*2d543d20SAndroid Build Coastguard Workerimport dbus
2*2d543d20SAndroid Build Coastguard Workerimport dbus.service
3*2d543d20SAndroid Build Coastguard Workerfrom sepolicy.sedbus import SELinuxDBus
4*2d543d20SAndroid Build Coastguard Worker
5*2d543d20SAndroid Build Coastguard Worker
6*2d543d20SAndroid Build Coastguard Workerdef convert_customization(buf):
7*2d543d20SAndroid Build Coastguard Worker    cust_dict = {}
8*2d543d20SAndroid Build Coastguard Worker    cust_dict["fcontext-equiv"] = {}
9*2d543d20SAndroid Build Coastguard Worker    for i in buf.split("\n"):
10*2d543d20SAndroid Build Coastguard Worker        rec = i.split()
11*2d543d20SAndroid Build Coastguard Worker        if len(rec) == 0:
12*2d543d20SAndroid Build Coastguard Worker            continue
13*2d543d20SAndroid Build Coastguard Worker        if rec[1] == "-D":
14*2d543d20SAndroid Build Coastguard Worker            continue
15*2d543d20SAndroid Build Coastguard Worker        if rec[0] not in cust_dict:
16*2d543d20SAndroid Build Coastguard Worker            cust_dict[rec[0]] = {}
17*2d543d20SAndroid Build Coastguard Worker        if rec[0] == "boolean":
18*2d543d20SAndroid Build Coastguard Worker            cust_dict["boolean"][rec[-1]] = {"active": rec[2] == "-1"}
19*2d543d20SAndroid Build Coastguard Worker        if rec[0] == "login":
20*2d543d20SAndroid Build Coastguard Worker            cust_dict["login"][rec[-1]] = {"seuser": rec[3], "range": rec[5]}
21*2d543d20SAndroid Build Coastguard Worker        if rec[0] == "interface":
22*2d543d20SAndroid Build Coastguard Worker            cust_dict["login"][rec[-1]] = {"type": rec[3]}
23*2d543d20SAndroid Build Coastguard Worker        if rec[0] == "user":
24*2d543d20SAndroid Build Coastguard Worker            cust_dict["user"][rec[-1]] = {"level": rec[3], "range": rec[5], "role": rec[7]}
25*2d543d20SAndroid Build Coastguard Worker        if rec[0] == "port":
26*2d543d20SAndroid Build Coastguard Worker            cust_dict["port"][(rec[-1], rec[-2])] = {"type": rec[3]}
27*2d543d20SAndroid Build Coastguard Worker        if rec[0] == "node":
28*2d543d20SAndroid Build Coastguard Worker            cust_dict["node"][rec[-1]] = {"mask": rec[3], "protocol": rec[5], "type": rec[7]}
29*2d543d20SAndroid Build Coastguard Worker        if rec[0] == "fcontext":
30*2d543d20SAndroid Build Coastguard Worker            if rec[2] == "-e":
31*2d543d20SAndroid Build Coastguard Worker                cust_dict["fcontext-equiv"][(rec[-1])] = {"equiv": rec[3]}
32*2d543d20SAndroid Build Coastguard Worker            else:
33*2d543d20SAndroid Build Coastguard Worker                cust_dict["fcontext"][(rec[-1], rec[3])] = {"type": rec[5]}
34*2d543d20SAndroid Build Coastguard Worker        if rec[0] == "module":
35*2d543d20SAndroid Build Coastguard Worker            cust_dict["module"][rec[-1]] = {"enabled": rec[2] != "-d"}
36*2d543d20SAndroid Build Coastguard Worker
37*2d543d20SAndroid Build Coastguard Worker    return cust_dict
38*2d543d20SAndroid Build Coastguard Workerif __name__ == "__main__":
39*2d543d20SAndroid Build Coastguard Worker    try:
40*2d543d20SAndroid Build Coastguard Worker        dbus_proxy = SELinuxDBus()
41*2d543d20SAndroid Build Coastguard Worker        resp = dbus_proxy.customized()
42*2d543d20SAndroid Build Coastguard Worker        print(convert_customization(resp))
43*2d543d20SAndroid Build Coastguard Worker    except dbus.DBusException as e:
44*2d543d20SAndroid Build Coastguard Worker        print(e)
45