def test_logging_not_done_bee(self):
"""
Tests that the consumer does not process bait sessions that are not marked as done.
"""
sessions = {}
BaitSession.client_id = 'dummy_client_id'
BaitSession.honeypot_id = 'dummy_hive_id'
beesession = BaitSession('telnet', '123', '1234', '4321')
beesession.alldone = False
sessions[beesession.id] = beesession
# mock a dummy logger
dummy_logger = DummyLogger()
log_mock = Mock()
dummy_logger.log = log_mock
consumer = Consumer(sessions, {}, '')
consumer.logger = dummy_logger
gevent.spawn(consumer.start_handling)
# forcing cooperative yield.
gevent.sleep(0)
#assert that the log method was not called
self.assertFalse(log_mock.called)
#assert that we still has a single item in the queue
self.assertEquals(len(sessions), 1)
consumer.stop_handling()
def start(self):
"""
Starts sending client bees to the configured Honeypot.
"""
logger.info('Starting client.')
sessions = {}
# greenlet to consume and maintain data in sessions list
self.sessions_consumer = Consumer(sessions, self.config, self.my_ip)
gevent.spawn(self.sessions_consumer.start_handling)
self.dispatcher_greenlets = []
for honeypot_id, entry in self.config['baits'].items():
for b in clientbase.ClientBase.__subclasses__():
bait_name = b.__name__.lower()
# if the bait has a entry in the config we consider the bait enabled
if bait_name in entry:
bait_options = entry[bait_name]
#bait_session = b(sessions, bait_options)
dispatcher = BaitDispatcher(sessions, b, bait_options)
dispatcher.start()
self.dispatcher_greenlets.append(dispatcher)
logger.info('Adding {0} bait'.format(bait_name))
logger.debug(
'Bait added with options: {0}'.format(bait_options))
gevent.joinall(self.dispatcher_greenlets)
def test_logging_done_bee(self):
"""
Tests that the consumer calls a logger class and that the beesession is removed
from the queue afterwards.
"""
sessions = {}
BaitSession.client_id = 'dummy_client_id'
BaitSession.honeypot_id = 'dummy_hive_id'
beesession = BaitSession('telnet', '1234', '4321', '123')
beesession.alldone = True
sessions[beesession.id] = beesession
# mock a dummy logger
dummy_logger = DummyLogger()
log_mock = Mock()
dummy_logger.log = log_mock
consumer = Consumer(sessions, {}, '')
# inject the dummy logger into the consumer
consumer.logger = dummy_logger
gevent.spawn(consumer.start_handling)
#forcing cooperative yield.
gevent.sleep(0)
#assert that the log method of the logger object was called with beesession as parameter.
dummy_logger.log.assert_called_once_with(beesession)
#assert that the beesession was removed from the queue
self.assertEquals(len(sessions), 0)
consumer.stop_handling()
