scturtle/index.html

## index.html
<!DOCTYPE html>
<html lang="en">
  <head>
    <meta charset=utf-8>
    <title>WebSocket test</title>
  </head>
  <body>
    <script>
      if (window.MozWebSocket) {
        window.WebSocket = window.MozWebSocket;
      }

      function openConnection() {
          conn = new WebSocket('ws://127.0.0.1:9999');
          conn.onopen = function () {
            alert("open");
            conn.send("hello");
          };

          conn.onmessage = function (event) {
            var message = event.data; //JSON.parse(event.data);
            alert(message);
            conn.send("hello");
          };

          conn.onclose = function (event) {
          };
        }

      setTimeout(function () {
        openConnection();
      }, 1000);
    </script>
  </body>
</html>

## main.py
import socket
import threading
from ws import *


def clientJob(conn):
    while 1:
        data = RecvData(8196, conn)
        if not data:
            break
        print data
        SendData("Hello, too", conn)


if __name__ == '__main__':
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        sock.bind(("", 9999))
        sock.listen(1)
    except:
        print("Port unavailable. Exit.")
        exit()
    while True:
        conn, addr = sock.accept()
        if handshake(conn):
            t = threading.Thread(target=clientJob,
                                 args=(conn,))
            t.start()

## ws.py
# modified from: http://0x14.iteye.com/blog/1396511
import base64
import hashlib
import json
import struct


# shake hands after connection
def handshake(conn):
    headers = {}
    shake = conn.recv(1024)

    if not len(shake):
        return False

    header, data = shake.split('\r\n\r\n', 1)
    for line in header.split("\r\n")[1:]:
        key, value = line.split(": ", 1)
        headers[key] = value

    if "Sec-WebSocket-Key" not in headers:
        print("this socket is not websocket,close")
        conn.close()
        return False

    key = base64.b64encode(hashlib.sha1(
        headers["Sec-WebSocket-Key"] +
        '258EAFA5-E914-47DA-95CA-C5AB0DC85B11').digest())
    host = headers["Host"]

    response = "HTTP/1.1 101 Switching Protocols\r\n"\
               "Upgrade:websocket\r\n"\
               "Connection: Upgrade\r\n"\
               "Sec-WebSocket-Accept:" + key + "\r\n"\
               "WebSocket-Location: ws://" + host + "/WebManagerSocket\r\n"\
               "WebSocket-Protocol:WebManagerSocket\r\n\r\n"

    conn.send(response)
    return True


def RecvData(num, conn):
    try:
        recv_data = conn.recv(num)
        if not len(recv_data):
            return False
    except:
        return False
    else:
        length = ord(recv_data[1]) & 127
        if length == 126:
            masks = recv_data[4:8]
            data = recv_data[8:]
        elif length == 127:
            masks = recv_data[10:14]
            data = recv_data[14:]
        else:
            masks = recv_data[2:6]
            data = recv_data[6:]

        raw_str = ""
        i = 0
        for d in data:
            raw_str += chr(ord(d) ^ ord(masks[i % 4]))
            i += 1

        return raw_str


def SendData(data, conn):
    try:
        if not isinstance(data, str):
            data = json.dumps(data)
    except:
        return False

    token = "\x81"
    length = len(data)
    if length < 126:
        token += struct.pack("B", length)
    elif length <= 0xFFFF:
        token += struct.pack("!BH", 126, length)
    else:
        token += struct.pack("!BQ", 127, length)
    data = '%s%s' % (token, data)

    conn.send(data)
    return True
	<!DOCTYPE html>
	<html lang="en">
	<head>
	<meta charset=utf-8>
	<title>WebSocket test</title>
	</head>
	<body>
	<script>
	if (window.MozWebSocket) {
	window.WebSocket = window.MozWebSocket;
	}

	function openConnection() {
	conn = new WebSocket('ws://127.0.0.1:9999');
	conn.onopen = function () {
	alert("open");
	conn.send("hello");
	};

	conn.onmessage = function (event) {
	var message = event.data; //JSON.parse(event.data);
	alert(message);
	conn.send("hello");
	};

	conn.onclose = function (event) {
	};
	}

	setTimeout(function () {
	openConnection();
	}, 1000);
	</script>
	</body>
	</html>
	import socket
	import threading
	from ws import *


	def clientJob(conn):
	while 1:
	data = RecvData(8196, conn)
	if not data:
	break
	print data
	SendData("Hello, too", conn)


	if __name__ == '__main__':
	sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
	try:
	sock.bind(("", 9999))
	sock.listen(1)
	except:
	print("Port unavailable. Exit.")
	exit()
	while True:
	conn, addr = sock.accept()
	if handshake(conn):
	t = threading.Thread(target=clientJob,
	args=(conn,))
	t.start()
	# modified from: http://0x14.iteye.com/blog/1396511
	import base64
	import hashlib
	import json
	import struct


	# shake hands after connection
	def handshake(conn):
	headers = {}
	shake = conn.recv(1024)

	if not len(shake):
	return False

	header, data = shake.split('\r\n\r\n', 1)
	for line in header.split("\r\n")[1:]:
	key, value = line.split(": ", 1)
	headers[key] = value

	if "Sec-WebSocket-Key" not in headers:
	print("this socket is not websocket,close")
	conn.close()
	return False

	key = base64.b64encode(hashlib.sha1(
	headers["Sec-WebSocket-Key"] +
	'258EAFA5-E914-47DA-95CA-C5AB0DC85B11').digest())
	host = headers["Host"]

	response = "HTTP/1.1 101 Switching Protocols\r\n"\
	"Upgrade:websocket\r\n"\
	"Connection: Upgrade\r\n"\
	"Sec-WebSocket-Accept:" + key + "\r\n"\
	"WebSocket-Location: ws://" + host + "/WebManagerSocket\r\n"\
	"WebSocket-Protocol:WebManagerSocket\r\n\r\n"

	conn.send(response)
	return True


	def RecvData(num, conn):
	try:
	recv_data = conn.recv(num)
	if not len(recv_data):
	return False
	except:
	return False
	else:
	length = ord(recv_data[1]) & 127
	if length == 126:
	masks = recv_data[4:8]
	data = recv_data[8:]
	elif length == 127:
	masks = recv_data[10:14]
	data = recv_data[14:]
	else:
	masks = recv_data[2:6]
	data = recv_data[6:]

	raw_str = ""
	i = 0
	for d in data:
	raw_str += chr(ord(d) ^ ord(masks[i % 4]))
	i += 1

	return raw_str


	def SendData(data, conn):
	try:
	if not isinstance(data, str):
	data = json.dumps(data)
	except:
	return False

	token = "\x81"
	length = len(data)
	if length < 126:
	token += struct.pack("B", length)
	elif length <= 0xFFFF:
	token += struct.pack("!BH", 126, length)
	else:
	token += struct.pack("!BQ", 127, length)
	data = '%s%s' % (token, data)

	conn.send(data)
	return True