skynet實踐（8）-接入websocket

    我從開源項目（https://github.com/lipp/lua-websockets，這里我們簡稱LWS）中抽出了websocket的部分處理，步驟如下：

    1）首先是解決LWS的幾個依賴問題。LWS在握手階段的base64編解碼使用了luasocket中的mime，因此在3rd文件夾中建立mime文件夾，將mime.h和mime.c文件放進去，修改skynet的Makefile文件，作為一個單獨的動態庫編譯：

LUA_CLIB = skynet \
  client \
  bson md5 sproto lpeg mime

$(LUA_CLIB_PATH)/mime.so : 3rd/mime/mime.c | $(LUA_CLIB_PATH)
    $(CC) $(CFLAGS) $(SHARED) -I3rd/mime $^ -o $@ 

    在tools.lua中引用了mime，由於我們只用了C文件，因此修改require "mime" 為： require "mime.core" 。

對於luabitop，skynet自帶的lua5.3源碼已經支持（在lbitlib.c文件中），但是默認是關閉的，修改lua5.3的Makefile，增加LUA_COMPAT_BITLIB宏並重新編譯：

MYCFLAGS=-I../../skynet-src -g -DLUA_COMPAT_BITLIB

    這樣在bit.lua中對bit32就能正確引用了。

    2）下一步則是在skynet端偵聽tcp連接：

local skynet = require "skynet"
local socket = require "skynet.socket"

skynet.start(function()
    local id = socket.listen("0.0.0.0", 8001)
    skynet.error("web server listen on web port 8001")

    socket.start(id, function(id, addr)
        local agent = skynet.newservice("wsagent")
        skynet.error(string.format("%s connected, pass it to agent :%08x", addr, agent))
        skynet.send(agent, "lua", id, "start")
    end)
end)

    這里每次偵聽到一個新連接后就建立wsagent，后續的握手及數據傳輸一並交給其處理。

    3）在lualib下建立websocket文件夾，放入handshake.lua（握手協議）、frame.lua（幀數據格式解析）、bit.lua、tools.lua。在wsagent.lua中實現握手、數據傳輸、關閉連接如下：

local skynet = require "skynet"
local socket = require "skynet.socket"
local frame  = require "websocket.frame"
local handshake = require "websocket.handshake"
local sockethelper = require "http.sockethelper"

local agent = { }
local REQUEST = { }
local FD, read, write

local function _send(message)
    local encoded = frame.encode(message, frame.TEXT)
    write(encoded)
end

local function _handshake(fd)
    FD = fd
    read  = sockethelper.readfunc(fd)
    write = sockethelper.writefunc(fd)

    local header = ""
    while true do
        local bytes = read()
        header = header .. bytes
        if #header > 8192 then
            skynet.error("<websocket.handshake>error: header size > 8192")
            return
        end

        local _, to = header:find("\r\n\r\n", -#bytes-3, true)
        if to then
            header = header:sub(1, to)
            break
        end
    end

    print("accept handshake http request:" .. header)

    local protocols = { } -- todo: how to set protocols?
    local response, protocol = handshake.accept_upgrade(header, protocols)
    if not response then
        skynet.error("<websocket.handshake>error: handshake parse header fault")
        return
    end

    print("send handshake http response:" .. response)

    write(response)
    skynet.error(string.format("<websocket.handshake>web socket %q connection established", fd))

    return true
end

local function _close()
    local encoded = frame.encode_close(1000, 'force close')
    encoded = frame.encode(encoded, frame.CLOSE)

    print("force close:" .. encoded)

    write(encoded)
    socket.close(FD)
end

local function _dispatch(text, opcode)
    print(string.format("<websocket>opcode:%q message:%q", opcode, text))

    local TEXT  = assert(frame.TEXT)
    local CLOSE = assert(frame.CLOSE)
    assert(opcode == TEXT or opcode == CLOSE, opcode)

    if opcode == TEXT then
        -- todo: logic
        return true
    end

    if opcode == CLOSE then
        local code, reason = frame.decode_close(message)
        print(string.format("<websocket>CLOSE code:%q reason:%q", code, reason))
        local encoded = frame.encode_close(code)
        encoded = frame.encode(encoded, frame.CLOSE)

        local ok, err = pcall(write, encoded)
        if not ok then
            -- remote endpoint may has closed tcp-connection already
            skynet.error("write close protocol failure:" .. tostring(err))
        end
        socket.close(assert(FD))
    end
end

local function _recv()
    local last
    local frames = {}
    local first_opcode

    while true do
        -- skynet will report error and close socket if socket error (see socket.lua)
        local encoded = read()
        if last then
            encoded = last .. encoded
            last = nil
        end

        repeat
            local decoded, fin, opcode, rest = frame.decode(encoded)
            if decoded then

                if not first_opcode then
                    first_opcode = opcode
                end
                table.insert(frames, decoded)
                encoded = rest
                if fin == true then
                    if not _dispatch(table.concat(frames), first_opcode) then
                        -- socket closed in [_dispatch]
                        return
                    end
                    frames = { }
                    first_opcode = nil
                end
            end
        until (not decoded)
        
        if #encoded > 0 then
            last = encoded
        end
    end
end

function agent.start(fd)
    socket.start(fd)

    skynet.error("<websocket>start handshake")
    if not _handshake(fd) then
        socket.close(fd)
        skynet.exit()
        return
    end

    skynet.error("<websocket>receive and dispatch")
    _recv()

    skynet.error("<websocket>exit")
    skynet.exit()
end

skynet.start(function()
    skynet.dispatch("lua", function (_, _, fd, method, ...)
        local f = assert(agent[method])
        skynet.retpack(f(fd, ...))
    end)
end)

    代碼我已放到git上：https://github.com/zhoujijian/skynet-websocket