mirror of
https://github.com/lunarmodules/luasocket.git
synced 2024-11-08 14:28:21 +01:00
73 lines
2.0 KiB
Lua
73 lines
2.0 KiB
Lua
function set_add(set, sock)
|
|
table.insert(set, sock)
|
|
end
|
|
|
|
function set_remove(set, sock)
|
|
for i = 1, table.getn(set) do
|
|
if set[i] == sock then
|
|
table.remove(set, i)
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
host = host or "*"
|
|
port1 = port1 or 8080
|
|
port2 = port2 or 8081
|
|
if arg then
|
|
host = arg[1] or host
|
|
port1 = arg[2] or port1
|
|
port2 = arg[3] or port2
|
|
end
|
|
|
|
server1, error = socket.bind(host, port1)
|
|
if not server1 then print(error) exit() end
|
|
server1:timeout(1)
|
|
server2, error = socket.bind(host, port2)
|
|
if not server2 then print(error) exit() end
|
|
server2:timeout(1)
|
|
|
|
sock_set = {server1, server2}
|
|
|
|
sock_id = {}
|
|
sock_id[server1] = 1
|
|
sock_id[server2] = 2
|
|
next_id = 3
|
|
|
|
while 1 do
|
|
local readable, _, error = socket.select(sock_set, nil)
|
|
for _, sock in readable do
|
|
-- is it a server socket
|
|
if sock_id[sock] < 3 then
|
|
local incomming = sock:accept()
|
|
if incomming then
|
|
incomming:timeout(1)
|
|
sock_id[incomming] = next_id
|
|
set_add(sock_set, incomming)
|
|
io.write("Added client id ", next_id, ". ",
|
|
table.getn(sock_set)-2, " total.\n")
|
|
next_id = next_id + 1
|
|
end
|
|
-- it is a client socket
|
|
else
|
|
local line, error = sock:receive()
|
|
local id = sock_id[sock]
|
|
if error then
|
|
sock:close()
|
|
set_remove(sock_set, sock)
|
|
io.write("Removed client number ", id, ". ",
|
|
getn(sock_set)-2, " total.\n")
|
|
else
|
|
io.write("Broadcasting line '", id, "> ", line, "'.\n")
|
|
__, writable, error = socket.select(nil, sock_set, 1)
|
|
if not error then
|
|
for ___, outgoing in writable do
|
|
io.write("Sending to client ", sock_id[outgoing], "\n")
|
|
outgoing:send(id, "> ", line, "\r\n")
|
|
end
|
|
else io.write("No one ready to listen!!!\n") end
|
|
end
|
|
end
|
|
end
|
|
end
|