def init(self, data):
self.emit(
RequestHandler('authentication', {
'username': "******",
'password': "******"
}))
self.emit(
RequestHandler("pi:login", {
"name": "rasp1",
"ip": self.get_ip_address('wlan0')
}))
self.emit(
RequestHandler("pi:ip:set", {'ip': self.get_ip_address('wlan0')}))
self.emit(RequestHandler("alarm:get", None))
self.emit(RequestHandler("music:playlist:get", None))
class QueryResolver:
def __init__(self, protocol, support_resolving=True):
self.response_handler = ResponseHandler()
self.request_handler = RequestHandler()
self.network_handler = NetworkManager(protocol)
self.support_resolving = support_resolving
def get_next_address(self, address, dns_server_ip, req_type=PackageType.A, port=53):
request = self.request_handler.create_request(address, self.support_resolving, req_type=req_type)
resp = self.network_handler.send(request, dns_server_ip, port)
self.response_handler.parse_response(resp)
if self.response_handler.answers:
yield self.response_handler.answers[0]
if self.response_handler.additional:
yield self.response_handler.additional[0]
if self.response_handler.authority:
yield self.response_handler.authority[0]
def get_ip(self, address, root_ip, req_type=PackageType.A, port=53):
if address is None:
raise ValueError
for answer in self.get_next_address(address, root_ip, req_type, port):
if answer.internal_type == 'answer':
return answer.address
if answer.internal_type == 'additional':
if answer.type == PackageType.A.value or answer.type == PackageType.AAAA.value:
return self.get_ip(address, answer.address, req_type=req_type)
return self.get_ip(address, self.get_ip(answer.address, root_ip, port=port), req_type=req_type)
if answer.internal_type == 'authority':
return self.get_ip(address, self.get_ip(answer.address, root_ip, port=port), req_type=req_type)
def initBasicConfig(self):
self.flask = Flask(__name__,
template_folder=self.template_folder,
static_folder=self.static_folder)
self.config = self.flask.config
self.run = self.flask.run
RequestHandler(self)
def runRequest(self):
if(not self.isLoggedIn):
print("writeOutputHTMLFile:not logged in. Can not execute request to challenge. Please check your credentials")
return
'''
runs the http request
'''
requestHandle=RequestHandler( header=self.header,
targetUrl=self.targetUrl,
filterTag=self.filterTags,
httpHandler=self.httpHandler,
httpRequestType=self.httpMethod,
debugPrint=self.debugOutput
)
requestHandle.run()
self.requestResult=requestHandle.getHttpContent(False)
def setkeyused(key):
return Handler.set_key_used(key)
def keyinfo(key):
return Handler.get_key_info(key)
def getkey():
return Handler.get_key()
def index():
return Handler.get_keys_left_count()
HOST = arg1 if arg1 else '127.0.0.1'
PRNT_BLUE = '\033[94m'
PRNT_ENDC = '\033[0m'
print('Starting server for neural nets')
prepareSegTrainCmd = 'echo'
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server:
server.bind((HOST, PORT))
print(PRNT_BLUE + 'Server listening on: ' + str(PORT) + PRNT_ENDC)
server.listen()
requestHandler = RequestHandler()
while True:
try:
conn, addr = server.accept()
data = conn.recv(10240)
message = data.decode("utf-8")
print(PRNT_BLUE + 'Connected by' + PRNT_ENDC, addr)
print(PRNT_BLUE + 'Received' + PRNT_ENDC, message)
requestHandler.handle_request(message)
conn.send(bytes("ok", encoding='utf-8'))
conn.close()
except Exception as e:
tb = traceback.format_exc()
def __init__(self, protocol, support_resolving=True):
self.response_handler = ResponseHandler()
self.request_handler = RequestHandler()
self.network_handler = NetworkManager(protocol)
self.support_resolving = support_resolving