class DriDaemon(DaemonBase):
"""
Manages all functions that need to take place on the router
on a regular basis
"""
def __init__(self, options):
DaemonBase.__init__(self, options)
self.kill_switch = False
self.loops = 0
self.downloader = Downloader(options)
self.uploader = Uploader(options)
self.allowed_traffic = self.downloader.get_allowed_traffic()
self.policy_mgr = PolicyMgr("/tmp/dnsmasq.log", self.allowed_traffic,
self.options)
self.policy_mgr.prep_system()
self.policy_mgr.initial_load()
self.policy_mgr.rotate_log()
def main_loop(self):
"""
Runs forever. We're a daemon
"""
self.log('Starting dri...')
while not self.kill_switch:
start_time = time.time()
while time.time() < (start_time + INTER_LOOP_SLEEP):
try:
has_more = self.policy_mgr.check_for_new_stuff()
if not has_more:
time.sleep(1)
except (PolicyMgrException, CommandException):
self.log('Help! Policy Manager')
if MAX_LOOPS:
self.loops += 1
if self.loops > MAX_LOOPS:
sys.exit(0)
try:
self.downloader.get_addresses()
allowed_traffic = self.downloader.get_allowed_traffic()
self.policy_mgr.process_new_allowed(allowed_traffic)
log_open_files("downloader")
except (DownloadException, CommandException):
self.log('Help! Downloading')
try:
self.uploader.upload_arp_table()
log_open_files("uploader")
except (UploadException, CommandException):
self.log('Help! Uploading')
#print "I LIVE"
def terminate(self):
self.kill_switch = True
print "dying"
def __init__(self, options):
DaemonBase.__init__(self, options)
self.kill_switch = False
self.loops = 0
self.downloader = Downloader(options)
self.uploader = Uploader(options)
self.allowed_traffic = self.downloader.get_allowed_traffic()
self.policy_mgr = PolicyMgr("/tmp/dnsmasq.log", self.allowed_traffic,
self.options)
self.policy_mgr.prep_system()
self.policy_mgr.initial_load()
self.policy_mgr.rotate_log()
