def __init__(self, request, client_address, server): self.buffersize = 12 self.vnc_clients = [] self.migraId = None self.serverInit = None RequestHandler.__init__(self, request, client_address, server)
def __init__(self, request, client_address, server): self.buffersize = 12 self.vnc_server = None self.migraId = None # Minimum intrusion grid remote access identity self.password = '******' RequestHandler.__init__(self, request, client_address, server)
def process_request(self):
"""Attempt to generate an HttpRequest object from request_queue"""
while self.queue_contains_request():
request = self.pop_next_request()
request_obj = HttpRequest(request)
response = RequestHandler(self.config, request_obj).return_response()
self.client.send(response.generate_status_code())
self.client.send(response.generate_response_headers())
self.client.send(response.generate_response_body())
self.client.close()
def testRequestHandler():
rh = RequestHandler.Instance()
# Light query testing.
print('LIGHT QUERY TESTING')
for i in range(len(lightQueries)):
q = lightQueries[i]
e = lightQueryExpectedLengths[i]
c = lightQueryComments[i]
l = len(rh.lightQuery(q)['lights'])
print(str(l == e) + c + ' (result: ' + str(l) + ')')
# Make all the test state queries.
print('STATE QUERY TESTING')
for i in range(len(stateQueries)):
q = stateQueries[i]
e = stateQueryExpectedMessages[i]
c = stateQueryComments[i]
print(str(rh.stateQuery(q)['message'] == e) + c)
class ChatHandler(asyncore.dispatcher):
'''
Handler for the ChatServer module -- 1 loop instance per client
'''
def __init__(self, sock, addr):
asyncore.dispatcher.__init__(self, sock=sock)
self.reqHandler = RequestHandler(addr, sock)
def handle_read(self):
if not hasattr(self, 'reqHandler'):
return
try:
self.send(self.reqHandler.handle_request(self.recv(BUFSIZE)))
except socket.error:
pass
def handle_close(self):
self.close()
def close(self):
if hasattr(self, 'reqHandler'):
del self.reqHandler
asyncore.dispatcher.close(self)
def __init__(self, sock, addr):
asyncore.dispatcher.__init__(self, sock=sock)
self.reqHandler = RequestHandler(addr, sock)
"""Handle requests in a separate thread."""
if __name__ == '__main__':
if args.gen_cert is not None:
cert_folder = Path(args.gen_cert)
if not cert_folder.exists():
os.mkdir(cert_folder)
sp.check_call(['openssl', 'req', '-x509', '-nodes', '-days', '365', '-newkey', 'rsa:2048', '-keyout',
str(cert_folder/'key.pem'), '-out', str(cert_folder/'cert.pem')])
exit(0)
if args.cert is not None:
print('starting https')
th_redir = None
httpd = None
if not args.no_redirect:
httpd = ThreadedHTTPServer(('', args.http_port), HTTPSRedirectHandler(args.https_port))
th_redir = threading.Thread(target=httpd.serve_forever)
th_redir.start()
httpsd = ThreadedHTTPServer(('', args.https_port), RequestHandler(WebCFG(args.cfg, args.www), args.buf_size, args.debug))
httpsd.socket = ssl.wrap_socket(httpsd.socket, certfile=args.cert/'cert.pem', keyfile=args.cert/'key.pem',
server_side=True)
try:
httpsd.serve_forever()
finally:
if httpd:
httpd.shutdown()
else:
print('starting http')
httpd = ThreadedHTTPServer(('', args.http_port), RequestHandler(WebCFG(args.cfg, args.www), args.buf_size, args.debug))
httpd.serve_forever()
Returns:
None.
Preconditions:
None.
Postconditions:
The program is running.
"""
# Disable normal logging so that we don't clutter up things.
log = logging.getLogger('werkzeug')
log.setLevel(logging.WARNING)
# Create the request handler.
rh = RequestHandler.Instance()
# Try to load the components of the request handler.
am, lm = rh.load()
# If either fail we have to act.
if not (am and lm):
printInitialSetupHeader()
if not am:
try:
createAliasConfigFile()
except KeyboardInterrupt:
print('\n\n Keyboard interrupt. Cancelling.')
print(
'-------------------------------------------------------------------------------'
)
