def _config_honeypot(self, drone, db_session, config):
if drone.discriminator != 'honeypot':
# meh, better way do do this?
drone_id = drone.id
ip_address = drone.ip_address
db_session.delete(drone)
db_session.commit()
drone = Honeypot(id=drone_id)
drone.ip_address = ip_address
db_session.add(drone)
db_session.commit()
# common properties
drone.name = config['name']
# certificate information
drone.cert_common_name = config['certificate']['common_name']
drone.cert_country = config['certificate']['country']
drone.cert_state = config['certificate']['state']
drone.cert_locality = config['certificate']['locality']
drone.cert_organization = config['certificate']['organization']
drone.cert_organization_unit = config['certificate']['organization_unit']
# add capabilities
drone.capabilities = []
for protocol_name, protocol_config in config['capabilities'].items():
if 'protocol_specific_data' in protocol_config:
protocol_specific_data = protocol_config['protocol_specific_data']
else:
protocol_specific_data = {}
drone.add_capability(protocol_name, protocol_config['port'], protocol_specific_data)
db_session.add(drone)
db_session.commit()
self._handle_command_drone_config_changed(drone.id)
def _config_honeypot(self, drone, db_session, config):
if drone.discriminator != 'honeypot':
# meh, better way do do this?
drone_id = drone.id
ip_address = drone.ip_address
db_session.delete(drone)
db_session.commit()
drone = Honeypot(id=drone_id)
drone.ip_address = ip_address
db_session.add(drone)
db_session.commit()
# common properties
drone.name = config['name']
# certificate information
drone.cert_common_name = config['certificate']['common_name']
drone.cert_country = config['certificate']['country']
drone.cert_state = config['certificate']['state']
drone.cert_locality = config['certificate']['locality']
drone.cert_organization = config['certificate']['organization']
drone.cert_organization_unit = config['certificate']['organization_unit']
# add capabilities
drone.capabilities = []
for protocol_name, protocol_config in config['capabilities'].items():
if 'protocol_specific_data' in protocol_config:
protocol_specific_data = protocol_config['protocol_specific_data']
else:
protocol_specific_data = {}
drone.add_capability(protocol_name, protocol_config['port'], protocol_specific_data)
db_session.add(drone)
db_session.commit()
self._handle_command_drone_config_changed(drone.id)
