jcarr
/
noVNC
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Handle socket policy response in proxy itself.

This commit is contained in:
Joel Martin 2010-04-17 17:53:28 -05:00
parent 5d8e7ec068
commit 10ae1427f8
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
wsproxy.py
View File

@ -16,7 +16,7 @@ WebSocket-Protocol: sample\r
\r \r
""" """
policy_response = """<cross-domain-policy><allow-access-from domain="*" to-ports="*" /></cross-domain-policy>""" policy_response = """<cross-domain-policy><allow-access-from domain="*" to-ports="*" /></cross-domain-policy>\n"""
traffic_legend = """ traffic_legend = """
Traffic Legend: Traffic Legend:
@ -31,19 +31,24 @@ Traffic Legend:
""" """
def handshake(client): def do_handshake(client):
handshake = client.recv(1024) handshake = client.recv(1024)
print "Handshake [%s]" % handshake print "Handshake [%s]" % handshake
if handshake.startswith("<policy-file-request/>"): if handshake.startswith("<policy-file-request/>"):
print "Sending:", policy_response print "Sending:", policy_response
client.send(policy_response) client.send(policy_response)
handshake = client.recv(1024) client.close()
print "Handshake [%s]" % handshake return False
#handshake = client.recv(1024)
#if len(handshake) == 0:
# raise Exception("Policy exchange failed")
#print "Handshake [%s]" % handshake
req_lines = handshake.split("\r\n") req_lines = handshake.split("\r\n")
_, path, _ = req_lines[0].split(" ") _, path, _ = req_lines[0].split(" ")
_, origin = req_lines[4].split(" ") _, origin = req_lines[4].split(" ")
_, host = req_lines[3].split(" ") _, host = req_lines[3].split(" ")
client.send(server_handshake % (origin, host, path)) client.send(server_handshake % (origin, host, path))
return True
def traffic(token="."): def traffic(token="."):
sys.stdout.write(token) sys.stdout.write(token)
@ -137,7 +142,7 @@ def start_server(listen_port, target_host, target_port):
print 'waiting for connection on port %s' % listen_port print 'waiting for connection on port %s' % listen_port
csock, address = lsock.accept() csock, address = lsock.accept()
print 'Got client connection from %s' % address[0] print 'Got client connection from %s' % address[0]
handshake(csock) if not do_handshake(csock): continue
print "Connecting to: %s:%s" % (target_host, target_port) print "Connecting to: %s:%s" % (target_host, target_port)
tsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) tsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
tsock.connect((target_host, target_port)) tsock.connect((target_host, target_port))