Added HTTP

This commit is contained in:
John W Kerns 2017-09-02 00:41:24 -07:00
parent 171e2669a0
commit f3adef81e3

View file

@ -11,8 +11,6 @@
version = "v1.0.0"
import os
import sys
import time
@ -184,6 +182,72 @@ def ssh_talker(client, valdict):
#quit()
def http_talker(client, valdict):
def recv_all(sock):
prev_timeout = sock.gettimeout()
try:
sock.settimeout(0.01)
rdata = []
while True:
try:
rdata.append(sock.recv(MAX_PACKET))
except socket.timeout:
return ''.join(rdata)
finally:
sock.settimeout(prev_timeout)
def normalize_line_endings(s):
return ''.join((line + '\n') for line in s.splitlines())
while True:
# headers and body are divided with \n\n (or \r\n\r\n - that's why we
# normalize endings). In real application usage, you should handle
# all variations of line endings not to screw request body
request = normalize_line_endings(recv_all(client)) # hack again
request_head, request_body = request.split('\n\n', 1)
# first line is request headline, and others are headers
request_head = request_head.splitlines()
request_headline = request_head[0]
# headers have their name up to first ': '. In real world uses, they
# could duplicate, and dict drops duplicates by default, so
# be aware of this.
request_headers = dict(x.split(': ', 1) for x in request_head[1:])
# headline has form of "POST /can/i/haz/requests HTTP/1.0"
request_method, request_uri, request_proto = request_headline.split(' ', 3)
response_body = [
'<html><body><h1>Hello, world!</h1>',
'<p>This page is in location %(request_uri)r, was requested ' % locals(),
'using %(request_method)r, and with %(request_proto)r.</p>' % locals(),
'<p>Request body is %(request_body)r</p>' % locals(),
'<p>Actual set of headers received:</p>',
'<ul>',
]
for request_header_name, request_header_value in request_headers.iteritems():
response_body.append('<li><b>%r</b> == %r</li>' % (request_header_name, \
request_header_value))
response_body.append('</ul></body></html>')
response_body_raw = ''.join(response_body)
# Clearly state that connection will be closed after this response,
# and specify length of response body
response_headers = {
'Content-Type': 'text/html; encoding=utf8',
'Content-Length': len(response_body_raw),
'Connection': 'close',
}
response_headers_raw = ''.join('%s: %s\n' % (k, v) for k, v in \
response_headers.iteritems())
# Reply as HTTP/1.1 server, saying "HTTP OK" (code 200).
response_proto = 'HTTP/1.1'
response_status = '200'
response_status_text = 'OK' # this can be random
# sending all this stuff
client.send('%s %s %s' % (response_proto, response_status, \
response_status_text))
client.send(response_headers_raw)
client.send('\n') # to separate headers from body
client.send(response_body_raw)
# and closing connection, as we stated before
client.close()
def start():
talkers = {22: ssh_talker, 23: telnet_talker, 80: telnet_talker}
for talker in talkers: