adulau/aha
1
0
Fork 0
mirror of https://github.com/adulau/aha.git synced 2024-12-27 19:26:25 +00:00
Code Issues Projects Releases Packages Wiki Activity

Focus only on decisions and retrieval of messages; rest went to ahalib

Browse source
This commit is contained in:
Gerard Wagener 2010-01-09 18:36:36 +01:00
parent 317b01bcf4
commit 3859a6d83b
1 changed files with 35 additions and 76 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

89
aha/aha.py
View file

@ -5,110 +5,69 @@
import os,sys,random
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 KERNEL_ERRORS():
EPERM = -1
ENOENT = -2
EIO = -5
ENOMEM = -12
EACESS = -13
EFAULT = -14
EPIPE = -32
ETXTBSY = -26
def __init__(self):
self.evec = (EPERM,ENOENT,EIO,ENOMEM,EACESS,EFAULT,EPIPE,ETXTBSY)
class ReplyMessage(Structure):
_fields_ = [ ("block" , c_int), ("exitcode" , c_int),
("substitue" ,c_int),("insult" , c_int) ]
class KernelEvents(ProcessEvent):
def silent_clean(self,filename):
try:
os.unlink(filename)
except OSError,e:
pass
def create_message(self,filename,block,exitcode,substitue,insult):
reply = ReplyMessage(block=block,exitcode=exitcode,substitue=substitue,
insult = insult)
fn = KERNEL_IN + os.sep + filename
f = open (fn,'wb')
f.write(reply)
f.close()
reply="(key=%s, block=%d,exitcode=%d,substitue=%d,insult=%d)"\
%(filename,block,exitcode, substitue,insult)
print reply
def load_file(self,filename):
msg = {}
fp = open(filename,'r')
for i in fp.read().split('\n'):
try:
(key,value) = i.split('=')
except ValueError,e:
pass
if msg.has_key(key) == False:
msg[key]=[]
msg[key].append(value)
fp.close()
return msg
def __init__(self,inqueue,outqueue,insultmaxidx):
self.ahaa = AHAActions(inqueue,outqueue)
def decision(self,filekey,msg):
insultmaxidx = 3
print msg
try:
command = msg['file'][0]
print "Got command: ",command
if msg['file'][0] == '/usr/bin/bvi':
self.create_message(filekey, block=1,
self.ahaa.create_message(filekey, block=1,
exitcode=KERNEL_ERRORS.ENOMEM,
insult = 0, substitue=0)
return
if msg['file'][0] == '/usr/bin/vi':
# The index 0 is reserved
idx = random.randint(1,insultmaxidx)
self.create_message(filekey, block=0, exitcode=0, insult=idx, substitue=0)
self.ahaa.create_message(filekey, block=0, exitcode=0,
insult=idx, substitue=0)
return
except KeyError,e:
pass
except IndexError,w:
pass
#Default action; allow-> out of memory
self.create_message(filekey,block=0,exitcode=0,insult=0,substitue=0)
self.ahaa.create_message(filekey,block=0,exitcode=0,insult=0,
substitue=0)
def process_IN_CLOSE_WRITE(self, event):
try:
filename = os.path.join(event.path,event.name)
msg = self.load_file(filename)
msg = self.ahaa.load_file(filename)
#Send back a message
self.decision(event.name,msg)
#Cleanup the file
self.silent_clean(filename)
except IOError,e:
sys.stderr.write("Kernel message (%s) could not be loaded or \
decison failed\n"%event.name)
if __name__ == '__main__':
print "Setting up listeners..."
wm = WatchManager()
wm = WatchManager()
mask = IN_CLOSE_WRITE # watched events
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())
wdd = wm.add_watch(KERNEL_OUT, mask, rec=True)
while True:
print "Waiting for events..."
while True:
try:
# process the queue of events as explained above
notifier.process_events()
if notifier.check_events():
# read notified events and enqeue them
notifier.read_events()
#TODO manage a global queue of unfinished events
#If inotify on close works this should not be necessary
except KeyboardInterrupt:
# destroy the inotify's instance on this interrupt (stop monitoring)
print "Stop listening..."
notifier.stop()
break
sys.exit(0)