summaryrefslogtreecommitdiff
path: root/demo/cursorcon.py
blob: 98ec08df20fc15f8391765a4d8300772273e4f69 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#!/usr/bin/env python

# example helloworld.py

import xcb
import xcb.xproto
import xcb.xselinux

import pygtk
pygtk.require('2.0')
import gtk

class CursorControl:

    def delete_event(self, widget, event, data=None):
        gtk.main_quit()
        return False

    def destroy(self, widget, data=None):
        gtk.main_quit()

    def error_dialog(self, msg):
        dialog = gtk.MessageDialog(self.window,
                                   gtk.DIALOG_MODAL | gtk.DIALOG_DESTROY_WITH_PARENT,
                                   gtk.MESSAGE_ERROR,
                                   gtk.BUTTONS_OK,
                                   msg)
        dialog.show()
        dialog.run()
        dialog.destroy()

    def relabel(self, devid, ctx):
        cookie = self.conn.selinux.SetDeviceContextChecked(devid, len(ctx), ctx)
        try:
            cookie.check()
        except xcb.ProtocolException, error:
            self.error_dialog("Failed to relabel device to %s:\n%s" % (ctx, error.__class__.__name__))
        except:
            self.error_dialog("Failed to relabel device to %s:\nUnexpected error" % ctx)

    def clicked(self, widget, label):
        if widget.get_active():
            if label == "Master":
                self.relabel(2, "staff_u:staff_r:xserver_t:s0-s0:c0.c1023")
                self.relabel(3, "staff_u:staff_r:xserver_t:s0-s0:c0.c1023")
            elif label == "High":
                self.relabel(2, "staff_u:staff_r:high_xdevice_t:s0-s0:c0.c1023")
                self.relabel(3, "staff_u:staff_r:high_xdevice_t:s0-s0:c0.c1023")
            elif label == "Medium":
                self.relabel(2, "staff_u:staff_r:med_xdevice_t:s0-s0:c0.c1023")
                self.relabel(3, "staff_u:staff_r:med_xdevice_t:s0-s0:c0.c1023")
            elif label == "Low":
                self.relabel(2, "staff_u:staff_r:low_xdevice_t:s0-s0:c0.c1023")
                self.relabel(3, "staff_u:staff_r:low_xdevice_t:s0-s0:c0.c1023")

    def __init__(self):
        self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
        self.vbox = gtk.VBox()
        self.window.add(self.vbox)

        self.window.connect("delete_event", self.delete_event)
        self.window.connect("destroy", self.destroy)
        self.window.set_border_width(10)

        # Set up XCB
        self.conn = xcb.connect()
        self.conn.selinux = self.conn(xcb.xselinux.key)

        # Set up radio buttons
        radio = gtk.RadioButton(label="Master")
        group = radio
        radio.connect("clicked", self.clicked, "Master")
        self.vbox.pack_start(radio)

        radio = gtk.RadioButton(group=group, label="High")
        radio.connect("clicked", self.clicked, "High")
        self.vbox.pack_start(radio)

        radio = gtk.RadioButton(group=group, label="Medium")
        radio.connect("clicked", self.clicked, "Medium")
        self.vbox.pack_start(radio)

        radio = gtk.RadioButton(group=group, label="Low")
        radio.connect("clicked", self.clicked, "Low")
        self.vbox.pack_start(radio)
    
        self.window.show_all()

    def main(self):
        gtk.main()

if __name__ == "__main__":
    control = CursorControl()
    control.main()