support running periodic checks interactively

1 files changed, 99 insertions, 2 deletions

diff --git a/src/msec/msecgui.py b/src/msec/msecgui.py
index fefc3a9..99167a3 100755
--- a/src/msec/msecgui.py
+++ b/src/msec/msecgui.py
@@ -11,6 +11,8 @@ import signal
 import traceback
 import Queue
 from textwrap import wrap
+from threading import Thread
+import time
 
 # PyGTK
 import warnings
@@ -104,6 +106,24 @@ SAVE_SETTINGS_TEXT=_("""Save and apply new configuration?""")
 DEFAULT_SPACING=5
 BANNER="msec.png"
 
+class BackgroundRunner(Thread):
+    # background task runner
+    def __init__(self, finish, program):
+        Thread.__init__(self)
+        """Runs a program in background."""
+        self.program = program
+        self.finish = finish
+
+    def run(self):
+        """Installs tomoyo policy"""
+        print "Running %s" % self.program
+        try:
+            res = os.system(self.program)
+            self.finish.put(res)
+        except:
+            print "Aborted: %s" % sys.exc_value
+            self.finish.put(-1)
+
 class MsecGui:
     """Msec GUI"""
     # common columns
@@ -639,15 +659,92 @@ class MsecGui:
             label.set_property("xalign", 0.0)
             table.attach(label, 2, 3, row, row + 1, gtk.EXPAND | gtk.FILL, 0, 0, 0)
 
+            #button = gtk.Button(_("Show results"))
+            #button.connect('clicked', self.show_test_results, logfile)
+            #table.attach(button, 3, 4, row, row + 1, gtk.EXPAND | gtk.FILL, 0, 0, 0)
             button = gtk.Button(_("Run now"))
-            button.connect('clicked', self.show_test_results, logfile)
+            button.connect('clicked', self.run_periodic_check, check)
             table.attach(button, 3, 4, row, row + 1, gtk.EXPAND | gtk.FILL, 0, 0, 0)
             row += 1
 
         return vbox
 
-    def show_test_results(self, widget, logfile):
+    def process_events(self):
+        """Process pending gtk events"""
+        while gtk.events_pending():
+            gtk.main_iteration(False)
+
+    def run_periodic_check(self, widget, check):
         """Shows results for the test"""
+        dialog = gtk.MessageDialog(
+                parent=self.window,
+                flags=0,
+                type=gtk.MESSAGE_INFO,
+                message_format = _("Do you want to run periodic check <b>%s</b>? Please note that it could take a considerable time to finish.") % check,
+                buttons=gtk.BUTTONS_YES_NO)
+        dialog.show_all()
+        ret = dialog.run()
+        dialog.destroy()
+        if ret != gtk.RESPONSE_YES:
+            return
+        # progress bar
+        progress = gtk.Window()
+        progress.set_title(_("Please wait, running checks..."))
+        progress.set_transient_for(self.window)
+        progress.set_modal(True)
+        progress.connect('delete-event', lambda *w: None)
+
+        vbox = gtk.VBox(spacing=10)
+        progress.add(vbox)
+        progressbar = gtk.ProgressBar()
+        progressbar.set_text(_("Please wait, running checks..."))
+        vbox.pack_start(progressbar)
+
+        label = gtk.Label(_("Please wait, this might take a few minutes."))
+        vbox.pack_start(label)
+
+        # show window
+        progress.show_all()
+
+        self.process_events()
+        # queue to signal that job is finished
+        q = Queue.Queue()
+
+        if check == "manual":
+            program = "/usr/share/msec/security.sh"
+        else:
+            program = "/etc/cron.%s/msec" % check
+        installer = BackgroundRunner(finish=q, program=program)
+        installer.start()
+
+        while 1:
+            self.process_events()
+            if not q.empty():
+                result = q.get()
+                break
+            else:
+                progressbar.pulse()
+                time.sleep(0.5)
+
+        progress.destroy()
+
+        if result == 0:
+            text = _("Periodic check was executed successfully!")
+            type = gtk.MESSAGE_INFO
+        else:
+            text = _("An error occurred while running periodic check.")
+            type = gtk.MESSAGE_ERROR
+        # policy was initialized
+        dialog = gtk.MessageDialog(
+                parent=self.window,
+                flags=0,
+                type=type,
+                message_format=text,
+                buttons=gtk.BUTTONS_OK
+                )
+        dialog.show_all()
+        dialog.run()
+        dialog.destroy()
 
     def run_configure_app(self, widget, cmd):
         """Runs application-specific configuration"""