aha.py gets parameters from config file

aha/aha.py

@@ -2,14 +2,10 @@
 #Core of the adaptive honeypot alternative
 # (c) Gerard Wagener
 #License GPL
-import os,sys,random
+import os,sys,random,getopt,ConfigParser
 from pyinotify import *
 from ctypes import *
 from ahalib import *
-KERNEL_OUT="/home/gerard/kernel/linux-2.6/out"
-KERNEL_IN="/home/gerard/kernel/linux-2.6/in"
-insultmaxidx = 3
-
 
 class KernelEvents(ProcessEvent):
     def __init__(self,inqueue,outqueue,insultmaxidx):
@@ -48,18 +44,54 @@ class KernelEvents(ProcessEvent):
             sys.stderr.write("Kernel message (%s) could not be loaded or \
                              decison failed\n"%event.name)
 
-if __name__ == '__main__':
-    print "Setting up listeners..."
+def usage(exitcode):
+    print """
+Setup listener for kernel events of the user mode linux
+    -h Shows this screen
+    -c Specifies the config file
 
+AUTHOR
+    Gerard Wagener
+
+LICENSE
+    GPL
+"""
+    sys.exit(exitcode)
+def shutdown(notifier):
+    if notifier != None:
+        print "Stop listening..."
+        notifier.stop()
+
+if __name__ == '__main__':
+    notifier = None
+    configfile = None
+    try:
+        opts,args = getopt.getopt(sys.argv[1:],"hc:",["help","config="])
+        for o,a in opts:
+            if o  in ('--help','-h'):
+                usage(0)
+            if o in ('--config','-c'):
+                configfile = a
+
+        if configfile == None:
+            sys.stderr.write('A configuration file needs to be specified\n')
+            sys.exit(1)
+        #Load config file and get opts
+        c=ConfigParser.ConfigParser()
+        c.read(configfile)
+        inqueue = c.get('common','inqueue')
+        outqueue  = c.get('common','outqueue')
+        insultmaxidx = int(c.get('insults','maxidx'))
+
+        print "Setting up listeners..."
         wm = WatchManager()
         mask = IN_CLOSE_WRITE  # watched events
 
-    notifier = Notifier(wm, KernelEvents(KERNEL_IN,KERNEL_OUT,insultmaxidx))
-    wdd = wm.add_watch(KERNEL_OUT, mask, rec=True)
+        notifier = Notifier(wm, KernelEvents(inqueue,outqueue,insultmaxidx))
+        wdd = wm.add_watch(outqueue, mask, rec=True)
 
         print "Waiting for events..."
         while True:
-        try:
             # process the queue of events as explained above
             notifier.process_events()
             if notifier.check_events():
@@ -67,7 +99,10 @@ if __name__ == '__main__':
                 notifier.read_events()
     except KeyboardInterrupt:
     # destroy the inotify's instance on this interrupt (stop monitoring)
-            print "Stop listening..."
-            notifier.stop()
-            break
-sys.exit(0)
+        shutdown(notifier)
+    except getopt.GetoptError,e:
+        usage(1)
+    except ConfigParser.NoOptionError,e:
+        sys.stderr.write('Configuration error. (%s)\n'%(str(e)))
+        sys.exit(1)
+    sys.exit(0)