selinux/gui/portsPage.py
Stephen Smalley 2e4d0bc862 Move policycoreutils/gui to gui.
Signed-off-by: Stephen Smalley <sds@tycho.nsa.gov>
2016-11-16 11:19:50 -05:00

268 lines
10 KiB
Python

## portsPage.py - show selinux mappings
## Copyright (C) 2006 Red Hat, Inc.
## This program 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.
## This program 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 this program; if not, write to the Free Software
## Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
## Author: Dan Walsh
import string
import gtk
import gtk.glade
import os
import gobject
import sys
import seobject
TYPE_COL = 0
PROTOCOL_COL = 1
MLS_COL = 2
PORT_COL = 3
try:
from subprocess import getstatusoutput
except ImportError:
from commands import getstatusoutput
from semanagePage import *
##
## I18N
##
PROGNAME = "policycoreutils"
try:
import gettext
kwargs = {}
if sys.version_info < (3,):
kwargs['unicode'] = True
gettext.install(PROGNAME,
localedir="/usr/share/locale",
codeset='utf-8',
**kwargs)
except:
try:
import builtins
builtins.__dict__['_'] = str
except ImportError:
import __builtin__
__builtin__.__dict__['_'] = unicode
class portsPage(semanagePage):
def __init__(self, xml):
semanagePage.__init__(self, xml, "ports", _("Network Port"))
xml.signal_connect("on_group_clicked", self.on_group_clicked)
self.group = False
self.ports_filter = xml.get_widget("portsFilterEntry")
self.ports_filter.connect("focus_out_event", self.filter_changed)
self.ports_filter.connect("activate", self.filter_changed)
self.ports_name_entry = xml.get_widget("portsNameEntry")
self.ports_protocol_combo = xml.get_widget("portsProtocolCombo")
self.ports_number_entry = xml.get_widget("portsNumberEntry")
self.ports_mls_entry = xml.get_widget("portsMLSEntry")
self.ports_add_button = xml.get_widget("portsAddButton")
self.ports_properties_button = xml.get_widget("portsPropertiesButton")
self.ports_delete_button = xml.get_widget("portsDeleteButton")
liststore = self.ports_protocol_combo.get_model()
iter = liststore.get_iter_first()
self.ports_protocol_combo.set_active_iter(iter)
self.init_store()
self.edit = True
self.load()
def filter_changed(self, *arg):
filter = arg[0].get_text()
if filter != self.filter:
if self.edit:
self.load(filter)
else:
self.group_load(filter)
def init_store(self):
self.store = gtk.ListStore(gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING, gobject.TYPE_STRING)
self.view.set_model(self.store)
self.store.set_sort_column_id(0, gtk.SORT_ASCENDING)
self.view.set_search_equal_func(self.search)
col = gtk.TreeViewColumn(_("SELinux Port\nType"), gtk.CellRendererText(), text=TYPE_COL)
col.set_sort_column_id(TYPE_COL)
col.set_resizable(True)
self.view.append_column(col)
self.store.set_sort_column_id(TYPE_COL, gtk.SORT_ASCENDING)
col = gtk.TreeViewColumn(_("Protocol"), gtk.CellRendererText(), text=PROTOCOL_COL)
col.set_sort_column_id(PROTOCOL_COL)
col.set_resizable(True)
self.view.append_column(col)
self.mls_col = gtk.TreeViewColumn(_("MLS/MCS\nLevel"), gtk.CellRendererText(), text=MLS_COL)
self.mls_col.set_resizable(True)
self.mls_col.set_sort_column_id(MLS_COL)
self.view.append_column(self.mls_col)
col = gtk.TreeViewColumn(_("Port"), gtk.CellRendererText(), text=PORT_COL)
col.set_sort_column_id(PORT_COL)
col.set_resizable(True)
self.view.append_column(col)
self.store.set_sort_func(PORT_COL, self.sort_int, "")
def sort_int(self, treemodel, iter1, iter2, user_data):
try:
p1 = int(treemodel.get_value(iter1, PORT_COL).split('-')[0])
p2 = int(treemodel.get_value(iter2, PORT_COL).split('-')[0])
if p1 > p2:
return 1
if p1 == p2:
return 0
return -1
except:
return 0
def load(self, filter=""):
self.filter = filter
self.port = seobject.portRecords()
dict = self.port.get_all(self.local)
self.store.clear()
for k in sorted(dict.keys()):
if not (self.match(str(k[0]), filter) or self.match(dict[k][0], filter) or self.match(k[2], filter) or self.match(dict[k][1], filter) or self.match(dict[k][1], filter)):
continue
iter = self.store.append()
if k[0] == k[1]:
self.store.set_value(iter, PORT_COL, k[0])
else:
rec = "%s-%s" % k[:2]
self.store.set_value(iter, PORT_COL, rec)
self.store.set_value(iter, TYPE_COL, dict[k][0])
self.store.set_value(iter, PROTOCOL_COL, k[2])
self.store.set_value(iter, MLS_COL, dict[k][1])
self.view.get_selection().select_path((0,))
def group_load(self, filter=""):
self.filter = filter
self.port = seobject.portRecords()
dict = self.port.get_all_by_type(self.local)
self.store.clear()
for k in sorted(dict.keys()):
ports_string = ", ".join(dict[k])
if not (self.match(ports_string, filter) or self.match(k[0], filter) or self.match(k[1], filter)):
continue
iter = self.store.append()
self.store.set_value(iter, TYPE_COL, k[0])
self.store.set_value(iter, PROTOCOL_COL, k[1])
self.store.set_value(iter, PORT_COL, ports_string)
self.store.set_value(iter, MLS_COL, "")
self.view.get_selection().select_path((0,))
def propertiesDialog(self):
if self.edit:
semanagePage.propertiesDialog(self)
def dialogInit(self):
store, iter = self.view.get_selection().get_selected()
self.ports_number_entry.set_text(store.get_value(iter, PORT_COL))
self.ports_number_entry.set_sensitive(False)
self.ports_protocol_combo.set_sensitive(False)
self.ports_name_entry.set_text(store.get_value(iter, TYPE_COL))
self.ports_mls_entry.set_text(store.get_value(iter, MLS_COL))
protocol = store.get_value(iter, PROTOCOL_COL)
liststore = self.ports_protocol_combo.get_model()
iter = liststore.get_iter_first()
while iter != None and liststore.get_value(iter, 0) != protocol:
iter = liststore.iter_next(iter)
if iter != None:
self.ports_protocol_combo.set_active_iter(iter)
def dialogClear(self):
self.ports_number_entry.set_text("")
self.ports_number_entry.set_sensitive(True)
self.ports_protocol_combo.set_sensitive(True)
self.ports_name_entry.set_text("")
self.ports_mls_entry.set_text("s0")
def delete(self):
store, iter = self.view.get_selection().get_selected()
port = store.get_value(iter, PORT_COL)
protocol = store.get_value(iter, 1)
try:
self.wait()
(rc, out) = getstatusoutput("semanage port -d -p %s %s" % (protocol, port))
self.ready()
if rc != 0:
return self.error(out)
store.remove(iter)
self.view.get_selection().select_path((0,))
except ValueError as e:
self.error(e.args[0])
def add(self):
target = self.ports_name_entry.get_text().strip()
mls = self.ports_mls_entry.get_text().strip()
port_number = self.ports_number_entry.get_text().strip()
if port_number == "":
port_number = "1"
for i in port_number.split("-"):
if not i.isdigit():
self.error(_("Port number \"%s\" is not valid. 0 < PORT_NUMBER < 65536 ") % port_number)
return False
list_model = self.ports_protocol_combo.get_model()
iter = self.ports_protocol_combo.get_active_iter()
protocol = list_model.get_value(iter, 0)
self.wait()
(rc, out) = getstatusoutput("semanage port -a -p %s -r %s -t %s %s" % (protocol, mls, target, port_number))
self.ready()
if rc != 0:
self.error(out)
return False
iter = self.store.append()
self.store.set_value(iter, TYPE_COL, target)
self.store.set_value(iter, PORT_COL, port_number)
self.store.set_value(iter, PROTOCOL_COL, protocol)
self.store.set_value(iter, MLS_COL, mls)
def modify(self):
target = self.ports_name_entry.get_text().strip()
mls = self.ports_mls_entry.get_text().strip()
port_number = self.ports_number_entry.get_text().strip()
list_model = self.ports_protocol_combo.get_model()
iter = self.ports_protocol_combo.get_active_iter()
protocol = list_model.get_value(iter, 0)
self.wait()
(rc, out) = getstatusoutput("semanage port -m -p %s -r %s -t %s %s" % (protocol, mls, target, port_number))
self.ready()
if rc != 0:
self.error(out)
return False
store, iter = self.view.get_selection().get_selected()
self.store.set_value(iter, TYPE_COL, target)
self.store.set_value(iter, PORT_COL, port_number)
self.store.set_value(iter, PROTOCOL_COL, protocol)
self.store.set_value(iter, MLS_COL, mls)
def on_group_clicked(self, button):
self.ports_add_button.set_sensitive(self.group)
self.ports_properties_button.set_sensitive(self.group)
self.ports_delete_button.set_sensitive(self.group)
self.mls_col.set_visible(self.group)
self.group = not self.group
if self.group:
button.set_label(_("List View"))
self.group_load(self.filter)
else:
button.set_label(_("Group View"))
self.load(self.filter)
return True