def __init__(self, session_id, profiles, post_test_action, res_queue, file_bag):
self.session_id = session_id
self.result = SessionEndResult(self.session_id)
self.res_queue = res_queue
self.file_bag = file_bag
self.profiles = profiles
self.post_test_action = post_test_action
self.nused_profiles = 0
self.lock = threading.Lock() # this lock has to be acquired before using nused_profiles and result attributes
self.res_queue.put(SessionStartEvent(self.session_id, self.profiles))
class ClientServerSessionHandler(object):
'''
Instances of this class hold information about the progress and the results of an audit of a client connecting to
a server.
Normally the clients are distinguished by IP address and the servers are distinguished by address/port tuple.
Objects of this class get instantiated on the very first connection from a client connecting to a server, and same
instance is supposed to handle all subsequent connections from the same client connecting to the same server.
For each connection it fetches a next server profile object from the list of profiles and treats the client with it.
It sends ClientAuditStartEvent event on first client connection.
After each connection is handled, it pushes the result returned by the handler, which normally is
ConnectionAuditResult or another subclass of ConnectionAuditEvent.
After the last auditor has finished its work it pushes ClientAuditEndEvent and ClientAuditResult into the queue.
'''
logger = logging.getLogger('ClientServerSessionHandler')
def __init__(self, session_id, profiles, post_test_action, res_queue, file_bag):
self.session_id = session_id
self.result = SessionEndResult(self.session_id)
self.res_queue = res_queue
self.file_bag = file_bag
self.profiles = profiles
self.post_test_action = post_test_action
self.nused_profiles = 0
self.lock = threading.Lock() # this lock has to be acquired before using nused_profiles and result attributes
self.res_queue.put(SessionStartEvent(self.session_id, self.profiles))
def handle(self, conn):
'''
This method is invoked when a new connection arrives. Can be invoked more then once in parallel,
from different threads. It takes the next unused profile from the list (in a thread-safe way),
uses it to handle this connection, and submits the result of handling this specific connection
to the results queue. It detects when the very last handler quits and issues audit-end-res event.
'''
# get the index of the profile to use to handle this connection
# in PTA_REPEAT mode, 'excess' flag will be set if the number of handled connections exceeds
# the number of available profiles
with self.lock:
if self.nused_profiles < len(self.profiles):
profile_index = self.nused_profiles
self.nused_profiles += 1
excess = False
else:
if (self.post_test_action == CFG_PTA_DROP) or (self.post_test_action == CFG_PTA_EXIT):
# no more profiles to apply, just let the connection drop
self.logger.debug('no unused profiles for connection %s', conn)
return
if self.post_test_action != CFG_PTA_REPEAT:
raise ValueError('unexpected post-test-action value')
profile_index = self.nused_profiles%len(self.profiles)
self.nused_profiles += 1
excess = True
if True:
# handle this connection with this profile
self.logger.debug('will use profile %d to handle connection %s', profile_index, conn)
profile = self.profiles[profile_index]
handler = profile.get_handler()
res = handler.handle(conn, profile, self.file_bag)
# log the results of the test
self.logger.debug('handling connection %s (excess=%s) using %s (%d/%d) resulted in %s',
conn, str(excess), profile, profile_index, len(self.profiles), res)
#if excess:
# return
# record the results of the test
self.res_queue.put(res)
# see if this thread is the very last handler out there
with self.lock:
self.result.add(res)
if len(self.result.results) >= len(self.profiles):
# the result object seems to contains enough results, this must be the very last handler out there
# submit the final result to the queue
self.logger.debug('last profile for connection %s', conn)
self.res_queue.put(self.result)