7

我正在使用 python 来实现一个简单的 websocket 服务器。我使用的握手来自http://en.wikipedia.org/w/index.php?title=WebSockets&oldid=372387414。

握手本身似乎有效，但是当我点击发送时，我收到一个 javascript 错误：

未捕获的错误：INVALID_STATE_ERR：DOM 异常 11

这是html：

<!doctype html>
<html>
    <head>
        <title>ws_json</title>

    </head>
    <body onload="handleLoad();" onunload="handleUnload();">
        <input type="text" id='input' />
        <input type="button" value="submit" onclick="handleSubmit()" />
        <div id="display"></div>

        <script type="text/javascript">
            function showmsg(str){
                display = document.getElementById("display");
                display.innerHTML += "<p>" + str + "</p>";
            }

            function send(str){
                ws.send(str.length);
                ws.send(str);
            }

            function handleSubmit(){
                input = document.getElementById('input');
                send(input.value);
                input.focus();
                input.value = '';
            }

            function handleLoad(){
                ws = new WebSocket("ws://localhost:8888/");
                ws.onopen = function(){
                    showmsg("websocket opened.");
                }

                ws.onclose = function(){
                    showmsg("websocket closed.");
                }
            }

            function handleUnload(){
                ws.close();
            }
        </script>
    </body>
</html>

这是python代码：

import socket
import threading
import json

PORT = 8888
LOCATION = "localhost:8888"

def handler(s):

    print " in handler "

    ip, _ = s.getpeername()
    print "New connection from %s" % ip
    request = s.recv(1024)

    print "\n%s\n" % request
    print s.getpeername()

    # send response
    response = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n"
    response += "Upgrade: WebSocket\r\n"
    response += "Connection: Upgrade\r\n"
    try:
        peername = s.getpeername()
        response += "Sec-WebSocket-Origin: http://%s\r\n" % peername[0] # % request[request.index("Origin: ")+8:-4]
    except ValueError:
        print "Bad Request"
        raise socket.error
    response += "Sec-WebSocket-Location: ws://%s\r\n" % LOCATION
    response += "Sec-WebSocket-Protocol: sample"
    response = response.strip() + "\r\n\r\n"

    print response
    s.send(response)

    while True:
        length = s.recv(1)
        print length
        if not length:
            break
        length = int(length)
        print "Length: %i" % length
        data = s.recv(length)
        print "Received: %s" % data
        print ""

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind(('localhost', PORT))
s.listen(5)

print "server is running..."
while True:
    sock, addr = s.accept()
    threading.Thread(target=handler, args=(sock, )).start()

有谁知道我在这里做错了什么？

4

2 回答 2

6

我在 Firefox 4 上测试了您的代码，并在点击发送时遇到了同样的错误，但在此之前我得到了

Firefox 无法与位于 ws://localhost:8888/ 的服务器建立连接。

这可能是 WebSocket 对象被破坏的原因。我怀疑您的握手响应缺少某些内容，因此 Firefox 正在关闭套接字。

来自关于 Websockets 的维基百科文章：

Sec-WebSocket-Key1 和 Sec-WebSocket-Key2 字段以及字段后面的 8 个字节是随机令牌，服务器在握手结束时使用这些令牌构造一个 16 字节的令牌，以证明它已读取客户端的握手。

你的服务器的响应底部没有这个特殊的数字，所以我认为我们需要弄清楚如何生成它，并包含它。

编辑：如何生成该号码

让我们从 key1、key2 和握手结束时的 8 个字节开始

key1 = "18x 6]8vM;54 *(5:  {   U1]8  z [  8"
key2 = "1_ tx7X d  <  nw  334J702) 7]o}` 0"
end8 = "Tm[K T2u"

我们通过忽略不是数字 0-9 的每个字符来为每个键创建一个数字。在 Python 中：

def numFromKey(key):
    return int(filter(lambda c: c in map(str,range(10)),key))

接下来我们将该数字除以原始键字符串中的空格数，所以这里有一个函数，它计算字符串中的空格数。

def spacesIn(key):
    return len(filter(lambda c: c==' ',key))

由键产生的两个数字是：

pkey1 = numFromKey(key1)/spacesIn(key1)
pkey2 = numFromKey(key2)/spacesIn(key2)

现在我们需要连接 pkey1、pkey2 和 end8 的字节。处理后的密钥需要表示为 32 位 Big-Endian 数字。

from struct import pack
catstring = pack('>L',pkey1) + pack('>L',pkey2) + end8

然后我们取这些字节的 md5 哈希来获得我们在握手结束时附加的幻数

import md5
magic = md5.new(catstring).digest()

这就是我认为它至少工作的方式

于 2010-07-13T13:01:02.103 回答
0

从版本 8 开始，此协议已弃用，请参阅：

http://tools.ietf.org/id/draft-ietf-hybi-thewebsocketprotocol-12.txt

对于新版本的协议。

于 2011-12-07T23:04:33.817 回答