def connect_to_database(name_of_database):
c = config.loadConfig()
connection = c['TESTBED']['CONNECTION_TO_DATABASE']
client = MongoClient(connection)
db = client.get_database(name_of_database)
print('Database found...')
return [client, db]
def connectToGateway(moduleName):
'''optional
called when the system connects to the cloud.'''
global _cloud, _pinLayouts
_device = device.Device(moduleName, 'beagle')
configs = config.loadConfig('beaglePins', True)
_pinLayouts = configs['pinLayouts']
setupGPIO()
def connectToGateway(moduleName):
"""optional
called when the system connects to the cloud."""
global _cloud, _pinLayouts
_device = device.Device(moduleName, "beagle")
configs = config.loadConfig("beaglePins", True)
_pinLayouts = configs["pinLayouts"]
setupGPIO()
def setUp(self):
c = config.loadConfig()
self.comPortNRF = c['TESTBED']['COM_PORT_NRF']
self.advertiser_address = c['TESTBED']['ADVERTISER_ADDRESS']
self.master_address = '5d:36:ac:90:0b:22'
self.access_address = 0x9a328370
log.info('Advertiser Address: ' + self.advertiser_address.upper())
def connectToGateway(moduleName):
'''optional
called when the system connects to the cloud.
Always called first (before updateAssets) '''
global _cloud, _pinLayouts
_device = device.Device(moduleName, 'beagle')
configs = config.loadConfig('beaglePins', True)
_pinLayouts = configs['pinLayouts']
ADC.setup() # need to start the ADC driver as well
setupGPIO()
def connectToGateway(moduleName):
'''optional
called when the system connects to the cloud.
Always called first (before updateAssets) '''
global _cloud, _pinLayouts
_device = device.Device(moduleName, 'beagle')
configs = config.loadConfig('beaglePins', True)
_pinLayouts = configs['pinLayouts']
ADC.setup() # need to start the ADC driver as well
setupGPIO()
def insert_data_to_collection_info_tests(name_of_attack):
c = config.loadConfig()
info = {
'Attack':
name_of_attack,
'Day_of_Attack':
str(datetime.datetime.today()).split()[0].encode('ascii', 'ignore'),
'Advertiser_Address':
str(c['TESTBED']['ADVERTISER_ADDRESS']),
'Scanner_Address':
str(c['TESTBED']['MASTER_ADDRESS']),
'Access_Adress':
str(int(c['TESTBED']['ACCESS_ADDRESS'], 16)),
'Port':
str(c['TESTBED']['COM_PORT_NRF'])
}
client, records = connect_to_collection_info_tests()
records.insert_one(info)
return client
def setUpClass(cls):
c = config.loadConfig()
attack = 'Invalid Sequence'
cls.comPortNRF = c['TESTBED']['COM_PORT_NRF']
cls.advertiser_address = c['TESTBED']['ADVERTISER_ADDRESS']
cls.master_address = c['TESTBED']['MASTER_ADDRESS']
cls.access_address = int(c['TESTBED']['ACCESS_ADDRESS'], 16)
cls.connection = c['TESTBED']['CONNECTION_TO_DATABASE']
log.info('Advertiser Address: ' + cls.advertiser_address.upper())
# Open serial port of NRF52 Dongle
cls.driver = NRF52Dongle(cls.comPortNRF, '115200')
date = str(datetime.datetime.today()).split()[0].encode(
'ascii', 'ignore')
# Choose from test.conf file type of database to insert data
if str(c['TESTBED']['DATABASE_VERSION']) == 'SQL':
insert_data_to_link_layer_tests(cls.advertiser_address,
cls.master_address,
cls.access_address, date, attack,
cls.comPortNRF)
elif str(c['TESTBED']['DATABASE_VERSION']) == 'MONGODB':
client = insert_data_to_collection_info_tests(attack)
close_connection_to_database(client)
elif str(c['TESTBED']['DATABASE_VERSION']) == 'BOTH':
insert_data_to_link_layer_tests(cls.advertiser_address,
cls.master_address,
cls.access_address, date, attack,
cls.comPortNRF)
client = insert_data_to_collection_info_tests(attack)
close_connection_to_database(client)
else:
print('Continue with no database')
pass
self.dialog = Ui(self.state)
self.dialog.setupUi(self.main)
self.state["dialog"] = self.dialog
self.state["mainwindow"] = self.main
def run(self):
self.main.show()
f = ChatFactory(self.state)
self.state["factory"] = f
f.addhook("*", self.state["dialog"].hook)
self.state["dialog"].addLog("Connecting to the server...")
general = self.state["config"]["general"]
reactor.connectTCP(general["server"], int(general["port"]), f)
reactor.run()
if __name__ == '__main__':
config = loadConfig("client.conf", "conf")
if not config:
exit(1)
state = {
"reactor": reactor,
"app": app,
"config": config,
}
m = Main(state)
m.run()
from twisted.application import service, internet
from twisted.internet import reactor
from core.config import loadConfig, errors
from core.core import ChatFactory
config = loadConfig("server.conf", "conf")
if not config:
print "Errors encountered in config:"
for error in errors:
print "\t" + error
state = {
"config": config,
}
application = service.Application("supportserver")
factory = ChatFactory(state)
port = config["general"]["port"]
serv = internet.TCPServer(int(port), factory)
serv.setServiceParent(application)
serv.startService()
reactor.run()
