Start using classes

This commit is contained in:
Gregory Soutade 2014-11-21 14:46:12 +01:00
parent 7dada493ab
commit c3c201fda1
1 changed files with 319 additions and 326 deletions

645
iwla.py
View File

@ -10,100 +10,95 @@ import gzip
from display import * from display import *
# Default configuration from default_conf import *
DB_ROOT = './output/'
DISPLAY_ROOT = './output/'
log_format = '$server_name:$server_port $remote_addr - $remote_user [$time_local] ' +\
'"$request" $status $body_bytes_sent ' +\
'"$http_referer" "$http_user_agent"'
time_format = '%d/%b/%Y:%H:%M:%S +0100'
pre_analysis_hooks = []
post_analysis_hooks = []
display_hooks = []
from conf import * from conf import *
print '==> Start' class IWLA(object):
meta_visit = {} ANALYSIS_CLASS = 'HTTP'
analyse_started = False API_VERSION = 1
current_visits = {}
cache_plugins = {}
display = {}
log_format_extracted = re.sub(r'([^\$\w])', r'\\\g<1>', log_format) def __init__(self):
log_format_extracted = re.sub(r'\$(\w+)', '(?P<\g<1>>.+)', log_format_extracted) print '==> Start'
http_request_extracted = re.compile(r'(?P<http_method>\S+) (?P<http_uri>\S+) (?P<http_version>\S+)')
log_re = re.compile(log_format_extracted) self.meta_infos = {}
uri_re = re.compile(r'(?P<extract_uri>[^\?]*)[\?(?P<extract_parameters>.*)]?') self.analyse_started = False
pages_extensions = ['/', 'html', 'xhtml', 'py', 'pl', 'rb', 'php'] self.current_analysis = {}
viewed_http_codes = [200] self.cache_plugins = {}
self.display = {}
self.valid_visitors = None
HOOKS_ROOT = './plugins/' self.log_format_extracted = re.sub(r'([^\$\w])', r'\\\g<1>', log_format)
PRE_HOOK_DIRECTORY = HOOKS_ROOT + 'pre_analysis/' self.log_format_extracted = re.sub(r'\$(\w+)', '(?P<\g<1>>.+)', self.log_format_extracted)
POST_HOOK_DIRECTORY = HOOKS_ROOT + 'post_analysis/' self.http_request_extracted = re.compile(r'(?P<http_method>\S+) (?P<http_uri>\S+) (?P<http_version>\S+)')
DISPLAY_HOOK_DIRECTORY = HOOKS_ROOT + 'display/' self.log_re = re.compile(self.log_format_extracted)
META_PATH = DB_ROOT + 'meta.db' self.uri_re = re.compile(r'(?P<extract_uri>[^\?]*)[\?(?P<extract_parameters>.*)]?')
DB_FILENAME = 'iwla.db' self.plugins = {PRE_HOOK_DIRECTORY : pre_analysis_hooks,
POST_HOOK_DIRECTORY : post_analysis_hooks,
DISPLAY_HOOK_DIRECTORY : display_hooks}
plugins = {PRE_HOOK_DIRECTORY : pre_analysis_hooks, POST_HOOK_DIRECTORY : post_analysis_hooks, DISPLAY_HOOK_DIRECTORY : display_hooks} def _preloadPlugins(self):
ret = True
for root in self.plugins.keys():
for plugin_name in self.plugins[root]:
p = root + '/' + plugin_name
try:
fp, pathname, description = imp.find_module(plugin_name, [root])
self.cache_plugins[p] = imp.load_module(plugin_name, fp, pathname, description)
mod = self.cache_plugins[p]
infos = mod.get_plugins_infos()
if infos['class'] != IWLA.ANALYSIS_CLASS or \
IWLA.API_VERSION < infos['min_version'] or\
(infos['max_version'] != -1 and (IWLA.API_VERSION > infos['max_version'])):
del self.cache_plugins[p]
elif not mod.load():
del self.cache_plugins[p]
except Exception as e:
print 'Error loading \'%s\' => %s' % (p, e)
ret = False
return ret
ANALYSIS_CLASS = 'HTTP' def _clearVisits(self):
API_VERSION = 1 self.current_analysis = {
'days_stats' : {},
'month_stats' : {},
'visits' : {}
}
self.valid_visitors = None
return self.current_analysis
def preloadPlugins(): def getDaysStats(self):
ret = True return self.current_analysis['days_stats']
for root in plugins.keys():
for plugin_name in plugins[root]:
p = root + '/' + plugin_name
try:
fp, pathname, description = imp.find_module(plugin_name, [root])
cache_plugins[p] = imp.load_module(plugin_name, fp, pathname, description)
#cache_plugins[p] = imp.load_module(p,None,p,("py","r",imp.PKG_DIRECTORY))
#cache_plugins[p] = imp.load_source(p, p)
mod = cache_plugins[p]
#print dir(mod)
#print "Register %s -> %s" % (p, mod)
infos = mod.get_plugins_infos()
if infos['class'] != ANALYSIS_CLASS or \
API_VERSION < infos['min_version'] or\
(infos['max_version'] != -1 and (API_VERSION > infos['max_version'])):
del cache_plugins[p]
elif not mod.load():
del cache_plugins[p]
except Exception as e:
print 'Error loading \'%s\' => %s' % (p, e)
ret = False
return ret
def createEmptyVisits(): def getMonthStatsStats(self):
visits = {'days_stats' : {}, 'month_stats' : {}, 'visits' : {}} return self.current_analysis['month_stats']
return visits
def createEmptyMeta(): def getCurrentVisists(self):
meta = {'last_time' : None} return self.current_analysis['visits']
return meta
def createEmptyDisplay(): def getValidVisitors(self):
display = {} return self.current_analysis['visits']
return display
def getDBFilename(time): def _clearMeta(self):
return (DB_ROOT + '%d/%d_%s') % (time.tm_year, time.tm_mon, DB_FILENAME) self.meta_infos = {
'last_time' : None
}
return self.meta_infos
def serialize(obj, filename): def _clearDisplay(self):
base = os.path.dirname(filename) self.display = {}
if not os.path.exists(base): return self.display
os.makedirs(base)
# TODO : remove return def getDBFilename(self, time):
return return (DB_ROOT + '%d/%d_%s') % (time.tm_year, time.tm_mon, DB_FILENAME)
def _serialize(self, obj, filename):
base = os.path.dirname(filename)
if not os.path.exists(base):
os.makedirs(base)
# TODO : remove return
return
with open(filename + '.tmp', 'wb+') as f: with open(filename + '.tmp', 'wb+') as f:
pickle.dump(obj, f) pickle.dump(obj, f)
@ -112,300 +107,298 @@ def serialize(obj, filename):
fzip.write(f.read()) fzip.write(f.read())
os.remove(filename + '.tmp') os.remove(filename + '.tmp')
def deserialize(filename): def _deserialize(self, filename):
if not os.path.exists(filename): if not os.path.exists(filename):
return None
with gzip.open(filename, 'r') as f:
return pickle.load(f)
return None return None
with gzip.open(filename, 'r') as f: def _callPlugins(self, root, *args):
return pickle.load(f) print '==> Call plugins (%s)' % root
return None for p in self.plugins[root]:
print '\t%s' % (p)
mod = self.cache_plugins[root + '/' + p]
mod.hook(*args)
def callPlugins(root, *args): def isPage(self, request):
print '==> Call plugins (%s)' % root for e in pages_extensions:
for p in plugins[root]: if request.endswith(e):
print '\t%s' % (p) return True
mod = cache_plugins[root + '/' + p]
mod.hook(*args)
def isPage(request):
for e in pages_extensions:
if request.endswith(e):
return True
return False
def appendHit(hit):
remote_addr = hit['remote_addr']
if not remote_addr in current_visits['visits'].keys():
createUser(hit)
return
super_hit = current_visits['visits'][remote_addr]
super_hit['requests'].append(hit)
super_hit['bandwidth'] += int(hit['body_bytes_sent'])
super_hit['last_access'] = meta_visit['last_time']
request = hit['extract_request']
if 'extract_uri' in request.keys():
uri = request['extract_uri']
else:
uri = request['http_uri']
hit['is_page'] = isPage(uri)
# Don't count 3xx status
status = int(hit['status'])
if status >= 300 and status < 400: return
if super_hit['robot'] or\
not status in viewed_http_codes:
page_key = 'not_viewed_pages'
hit_key = 'not_viewed_hits'
else:
page_key = 'viewed_pages'
hit_key = 'viewed_hits'
if hit['is_page']:
super_hit[page_key] += 1
else:
super_hit[hit_key] += 1
def createUser(hit):
super_hit = current_visits['visits'][hit['remote_addr']] = {}
super_hit['remote_addr'] = hit['remote_addr']
super_hit['viewed_pages'] = 0
super_hit['viewed_hits'] = 0
super_hit['not_viewed_pages'] = 0
super_hit['not_viewed_hits'] = 0
super_hit['bandwidth'] = 0
super_hit['last_access'] = meta_visit['last_time']
super_hit['requests'] = []
super_hit['robot'] = False
super_hit['hit_only'] = 0
appendHit(hit)
def decodeHTTPRequest(hit):
if not 'request' in hit.keys(): return False
groups = http_request_extracted.match(hit['request'])
if groups:
hit['extract_request'] = groups.groupdict()
uri_groups = uri_re.match(hit['extract_request']['http_uri'])
if uri_groups:
d = uri_groups.groupdict()
hit['extract_request']['extract_uri'] = d['extract_uri']
if 'extract_parameters' in d.keys():
hit['extract_request']['extract_parameters'] = d['extract_parameters']
else:
print "Bad request extraction " + hit['request']
return False return False
referer_groups = uri_re.match(hit['http_referer']) def _appendHit(self, hit):
if referer_groups: remote_addr = hit['remote_addr']
referer = hit['extract_referer'] = referer_groups.groupdict()
return True if not remote_addr in self.current_analysis['visits'].keys():
self._createUser(hit)
return
super_hit = self.current_analysis['visits'][remote_addr]
super_hit['requests'].append(hit)
super_hit['bandwidth'] += int(hit['body_bytes_sent'])
super_hit['last_access'] = self.meta_infos['last_time']
def decodeTime(hit): request = hit['extract_request']
t = hit['time_local']
hit['time_decoded'] = time.strptime(t, time_format) if 'extract_uri' in request.keys():
uri = request['extract_uri']
else:
uri = request['http_uri']
def getDisplayIndex(): hit['is_page'] = self.isPage(uri)
cur_time = meta_visit['last_time']
filename = '%d/index_%d.html' % (cur_time.tm_year, cur_time.tm_mon)
return display.get(filename, None) # Don't count 3xx status
status = int(hit['status'])
if status >= 300 and status < 400: return
def generateDisplayDaysStat(): if super_hit['robot'] or\
cur_time = meta_visit['last_time'] not status in viewed_http_codes:
title = 'Stats %d/%d' % (cur_time.tm_mon, cur_time.tm_year) page_key = 'not_viewed_pages'
filename = '%d/index_%d.html' % (cur_time.tm_year, cur_time.tm_mon) hit_key = 'not_viewed_hits'
page = createPage(display, filename, title) else:
page_key = 'viewed_pages'
days = createTable('By day', ['Day', 'Visits', 'Pages', 'Hits', 'Bandwidth', 'Robot Bandwidth']) hit_key = 'viewed_hits'
keys = current_visits['days_stats'].keys() if hit['is_page']:
keys.sort() super_hit[page_key] += 1
nb_visits = 0 else:
for k in keys: super_hit[hit_key] += 1
stats = current_visits['days_stats'][k]
row = [k, stats['nb_visitors'], stats['viewed_pages'], stats['viewed_hits'], stats['viewed_bandwidth'], stats['not_viewed_bandwidth']] def _createUser(self, hit):
super_hit = self.current_analysis['visits'][hit['remote_addr']] = {}
super_hit['remote_addr'] = hit['remote_addr']
super_hit['viewed_pages'] = 0
super_hit['viewed_hits'] = 0
super_hit['not_viewed_pages'] = 0
super_hit['not_viewed_hits'] = 0
super_hit['bandwidth'] = 0
super_hit['last_access'] = self.meta_infos['last_time']
super_hit['requests'] = []
super_hit['robot'] = False
super_hit['hit_only'] = 0
self._appendHit(hit)
def _decodeHTTPRequest(self, hit):
if not 'request' in hit.keys(): return False
groups = self.http_request_extracted.match(hit['request'])
if groups:
hit['extract_request'] = groups.groupdict()
uri_groups = self.uri_re.match(hit['extract_request']['http_uri'])
if uri_groups:
d = uri_groups.groupdict()
hit['extract_request']['extract_uri'] = d['extract_uri']
if 'extract_parameters' in d.keys():
hit['extract_request']['extract_parameters'] = d['extract_parameters']
else:
print "Bad request extraction " + hit['request']
return False
referer_groups = self.uri_re.match(hit['http_referer'])
if referer_groups:
referer = hit['extract_referer'] = referer_groups.groupdict()
return True
def _decodeTime(self, hit):
hit['time_decoded'] = time.strptime(hit['time_local'], time_format)
def getDisplayIndex(self):
cur_time = self.meta_infos['last_time']
filename = '%d/index_%d.html' % (cur_time.tm_year, cur_time.tm_mon)
return self.display.get(filename, None)
def _generateDisplayDaysStat(self):
cur_time = self.meta_infos['last_time']
title = 'Stats %d/%d' % (cur_time.tm_mon, cur_time.tm_year)
filename = '%d/index_%d.html' % (cur_time.tm_year, cur_time.tm_mon)
page = createPage(self.display, filename, title)
days = createTable('By day', ['Day', 'Visits', 'Pages', 'Hits', 'Bandwidth', 'Robot Bandwidth'])
keys = self.current_analysis['days_stats'].keys()
keys.sort()
nb_visits = 0
for k in keys:
stats = self.current_analysis['days_stats'][k]
row = [k, stats['nb_visitors'], stats['viewed_pages'], stats['viewed_hits'], stats['viewed_bandwidth'], stats['not_viewed_bandwidth']]
row = map(lambda(v): str(v), row)
appendRowToTable(days, row)
nb_visits += stats['nb_visitors']
stats = self.current_analysis['month_stats']
nb_days = len(keys)
row = [0, nb_visits, stats['viewed_pages'], stats['viewed_hits'], stats['viewed_bandwidth'], stats['not_viewed_bandwidth']]
if nb_days:
row = map(lambda(v): str(int(v/nb_days)), row)
else:
row = map(lambda(v): '0', row)
row[0] = 'Average'
appendRowToTable(days, row)
row = ['Total', nb_visits, stats['viewed_pages'], stats['viewed_hits'], stats['viewed_bandwidth'], stats['not_viewed_bandwidth']]
row = map(lambda(v): str(v), row) row = map(lambda(v): str(v), row)
appendRowToTable(days, row) appendRowToTable(days, row)
nb_visits += stats['nb_visitors'] appendBlockToPage(page, days)
stats = current_visits['month_stats'] def _generateDisplay(self):
self._generateDisplayDaysStat()
self._callPlugins(DISPLAY_HOOK_DIRECTORY, self.current_analysis, self.display)
buildPages(DISPLAY_ROOT, self.display)
nb_days = len(keys) def _generateStats(self, visits):
row = [0, nb_visits, stats['viewed_pages'], stats['viewed_hits'], stats['viewed_bandwidth'], stats['not_viewed_bandwidth']] stats = {}
if nb_days: stats['viewed_bandwidth'] = 0
row = map(lambda(v): str(int(v/nb_days)), row) stats['not_viewed_bandwidth'] = 0
else: stats['viewed_pages'] = 0
row = map(lambda(v): '0', row) stats['viewed_hits'] = 0
#stats['requests'] = set()
stats['nb_visitors'] = 0
row[0] = 'Average' for k in visits.keys():
appendRowToTable(days, row) super_hit = visits[k]
if super_hit['robot']:
stats['not_viewed_bandwidth'] += super_hit['bandwidth']
continue
row = ['Total', nb_visits, stats['viewed_pages'], stats['viewed_hits'], stats['viewed_bandwidth'], stats['not_viewed_bandwidth']] #print "[%s] =>\t%d/%d" % (k, super_hit['viewed_pages'], super_hit['viewed_hits'])
row = map(lambda(v): str(v), row)
appendRowToTable(days, row)
appendBlockToPage(page, days)
def generateDisplay(): if not super_hit['hit_only']:
generateDisplayDaysStat() stats['nb_visitors'] += 1
callPlugins(DISPLAY_HOOK_DIRECTORY, current_visits, display) stats['viewed_bandwidth'] += super_hit['bandwidth']
buildPages(DISPLAY_ROOT, display) stats['viewed_pages'] += super_hit['viewed_pages']
stats['viewed_hits'] += super_hit['viewed_hits']
def generateStats(visits): # for p in super_hit['requests']:
stats = {} # if not p['is_page']: continue
stats['viewed_bandwidth'] = 0 # req = p['extract_request']
stats['not_viewed_bandwidth'] = 0 # stats['requests'].add(req['extract_uri'])
stats['viewed_pages'] = 0
stats['viewed_hits'] = 0
#stats['requests'] = set()
stats['nb_visitors'] = 0
for k in visits.keys(): return stats
super_hit = visits[k]
if super_hit['robot']:
stats['not_viewed_bandwidth'] += super_hit['bandwidth']
continue
#print "[%s] =>\t%d/%d" % (k, super_hit['viewed_pages'], super_hit['viewed_hits']) def _generateMonthStats(self):
self._clearDisplay()
if not super_hit['hit_only']:
stats['nb_visitors'] += 1
stats['viewed_bandwidth'] += super_hit['bandwidth']
stats['viewed_pages'] += super_hit['viewed_pages']
stats['viewed_hits'] += super_hit['viewed_hits']
# for p in super_hit['requests']: visits = self.current_analysis['visits']
# if not p['is_page']: continue
# req = p['extract_request']
# stats['requests'].add(req['extract_uri'])
return stats stats = self._generateStats(visits)
def generateMonthStats():
display = createEmptyDisplay()
visits = current_visits['visits'] cur_time = self.meta_infos['last_time']
print "== Stats for %d/%d ==" % (cur_time.tm_year, cur_time.tm_mon)
stats = generateStats(visits) print stats
cur_time = meta_visit['last_time']
print "== Stats for %d/%d ==" % (cur_time.tm_year, cur_time.tm_mon)
print stats
valid_visitors = {k: v for (k,v) in visits.items() if not visits[k]['robot']} self.valid_visitors = {k: v for (k,v) in visits.items() if not visits[k]['robot']}
callPlugins(POST_HOOK_DIRECTORY, valid_visitors, stats) self._callPlugins(POST_HOOK_DIRECTORY, valid_visitors, stats)
current_visits['month_stats'] = stats self.current_analysis['month_stats'] = stats
path = getDBFilename(cur_time) path = self.getDBFilename(cur_time)
if os.path.exists(path): if os.path.exists(path):
os.remove(path) os.remove(path)
print "==> Serialize to %s" % path print "==> Serialize to %s" % path
serialize(current_visits, path) self._serialize(self.current_analysis, path)
generateDisplay() self._generateDisplay()
def generateDayStats(): def _generateDayStats(self):
visits = current_visits['visits'] visits = self.current_analysis['visits']
callPlugins(PRE_HOOK_DIRECTORY, visits)
stats = generateStats(visits) self._callPlugins(PRE_HOOK_DIRECTORY, visits)
cur_time = meta_visit['last_time']
print "== Stats for %d/%d/%d ==" % (cur_time.tm_year, cur_time.tm_mon, cur_time.tm_mday)
if cur_time.tm_mday > 1: stats = self._generateStats(visits)
last_day = cur_time.tm_mday - 1
while last_day:
if last_day in current_visits['days_stats'].keys():
break
last_day -= 1
if last_day:
for k in stats.keys():
stats[k] -= current_visits['days_stats'][last_day][k]
stats['nb_visitors'] = 0
for k in visits.keys():
if visits[k]['robot']: continue
if visits[k]['last_access'].tm_mday == cur_time.tm_mday:
stats['nb_visitors'] += 1
print stats
current_visits['days_stats'][cur_time.tm_mday] = stats cur_time = self.meta_infos['last_time']
print "== Stats for %d/%d/%d ==" % (cur_time.tm_year, cur_time.tm_mon, cur_time.tm_mday)
def newHit(hit): if cur_time.tm_mday > 1:
global current_visits last_day = cur_time.tm_mday - 1
global analyse_started while last_day:
if last_day in self.current_analysis['days_stats'].keys():
break
last_day -= 1
if last_day:
for k in stats.keys():
stats[k] -= self.current_analysis['days_stats'][last_day][k]
stats['nb_visitors'] = 0
for k in visits.keys():
if visits[k]['robot']: continue
if visits[k]['last_access'].tm_mday == cur_time.tm_mday:
stats['nb_visitors'] += 1
print stats
decodeTime(hit) self.current_analysis['days_stats'][cur_time.tm_mday] = stats
t = hit['time_decoded'] def _newHit(self, hit):
self._decodeTime(hit)
cur_time = meta_visit['last_time'] t = hit['time_decoded']
if cur_time == None: cur_time = self.meta_infos['last_time']
current_visits = deserialize(getDBFilename(t)) or createEmptyVisits()
analyse_started = True
else:
if not analyse_started:
if time.mktime(cur_time) >= time.mktime(t):
return
else:
analyse_started = True
if cur_time.tm_mon != t.tm_mon:
generateMonthStats()
current_visits = deserialize(getDBFilename(t)) or createEmptyVisits()
elif cur_time.tm_mday != t.tm_mday:
generateDayStats()
meta_visit['last_time'] = t if cur_time == None:
self.current_analysis = self._deserialize(self.getDBFilename(t)) or self._clearVisits()
self.analyse_started = True
else:
if not self.analyse_started:
if time.mktime(cur_time) >= time.mktime(t):
return
else:
self.analyse_started = True
if cur_time.tm_mon != t.tm_mon:
self._generateMonthStats()
self.current_analysis = self._deserialize(self.getDBFilename(t)) or self._clearVisits()
elif cur_time.tm_mday != t.tm_mday:
self._generateDayStats()
if not decodeHTTPRequest(hit): return False self.meta_infos['last_time'] = t
for k in hit.keys(): if not self._decodeHTTPRequest(hit): return False
if hit[k] == '-': hit[k] = ''
appendHit(hit) for k in hit.keys():
if hit[k] == '-': hit[k] = ''
return True self._appendHit(hit)
preloadPlugins() return True
print '==> Analysing log' def start(self):
self._preloadPlugins()
meta_visit = deserialize(META_PATH) or createEmptyMeta() print '==> Analysing log'
if meta_visit['last_time']:
current_visits = deserialize(getDBFilename(meta_visit['last_time'])) or createEmptyVisits()
else:
current_visits = createEmptyVisits()
f = open(analyzed_filename) self.meta_infos = self._deserialize(META_PATH) or self._clearMeta()
for l in f: if self.meta_infos['last_time']:
# print "line " + l self.current_analysis = self._deserialize(self.getDBFilename(self.meta_infos['last_time'])) or self._clearVisits()
else:
groups = log_re.match(l) self._clearVisits()
if groups:
if not newHit(groups.groupdict()):
break
else:
print "No match " + l
f.close()
if analyse_started: with open(analyzed_filename) as f:
generateDayStats() for l in f:
generateMonthStats() # print "line " + l
serialize(meta_visit, META_PATH)
else: groups = self.log_re.match(l)
print '==> Analyse not started : nothing to do'
generateMonthStats() if groups:
if not self._newHit(groups.groupdict()):
break
else:
print "No match for " + l
if self.analyse_started:
self._generateDayStats()
self._generateMonthStats()
self._serialize(meta_infos, META_PATH)
else:
print '==> Analyse not started : nothing to do'
self._generateMonthStats()
iwla = IWLA()
iwla.start()