mGBA-lua-socket-loadtest/SocketScript.lua at main · nikouu/mGBA-lua-socket-loadtest · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
local ENABLE_LOGGING = true
local TERMINATION_MARKER = "<|END|>"

local function log(msg)
	if ENABLE_LOGGING then
        console:log(msg)
    end
end

lastkeys = nil
server = nil
ST_sockets = {}
nextID = 1

function ST_stop(id)
    local sock = ST_sockets[id]
    ST_sockets[id] = nil
    sock:close()
end

function ST_format(id, msg, isError)
    local prefix = "Socket " .. id
    if isError then
        prefix = prefix .. " Error: "
    else
        prefix = prefix .. " Received: "
    end
    return prefix .. (msg and tostring(msg) or "Probably exceeding limit")
end

function ST_error(id, err)
    console:error(ST_format(id, err, true))
    ST_stop(id)
end

function ST_received(id)
    log("ST_received 1")
    local sock = ST_sockets[id]
    if not sock then return end
    sock._buffer = sock._buffer or ""
    while true do
        local chunk, err = sock:receive(1024)
        log("ST_received 2")
        if chunk then
            sock._buffer = sock._buffer .. chunk
            while true do
                local marker_start, marker_end = sock._buffer:find(TERMINATION_MARKER, 1, true)
                if not marker_start then break end
                local message = sock._buffer:sub(1, marker_start - 1)
                sock._buffer = sock._buffer:sub(marker_end + 1)
                log("ST_received 3")
                log(ST_format(id, message:match("^(.-)%s*$")))
                -- Echo back the message with the marker
                sock:send(message .. TERMINATION_MARKER)
            end
        else
            log("ST_received 4")
            if err ~= socket.ERRORS.AGAIN then
                log("ST_received 5")
                if err == "disconnected" then
                    log("ST_received 6")
                    log(ST_format(id, err, true))
                elseif err == socket.ERRORS.UNKNOWN_ERROR then
                    log("ST_received 7")
                    log(ST_format(id, err, true))
                else
                    log("ST_received 8")
                    console:error(ST_format(id, err, true))
                end
                ST_stop(id)
            end
            return
        end
    end
end

function ST_accept()
    local sock, err = server:accept()
    if err then
        console:error(ST_format("Accept", err, true))
        return
    end
    local id = nextID
    nextID = id + 1
    ST_sockets[id] = sock
    sock:add("received", function() ST_received(id) end)
    sock:add("error", function() ST_error(id) end)
    log(ST_format(id, "Connected"))
end

local port = 8888
server = nil
while not server do
    server, err = socket.bind(nil, port)
    if err then
        if err == socket.ERRORS.ADDRESS_IN_USE then
            port = port + 1
        else
            console:error(ST_format("Bind", err, true))
            break
        end
    else
        local ok
        ok, err = server:listen()
        if err then
            server:close()
            console:error(ST_format("Listen", err, true))
        else
            console:log("Socket Server Test: Listening on port " .. port)
            console:log("Logging set to: " .. tostring(ENABLE_LOGGING))
            server:add("received", ST_accept)
        end
    end
end