luasocket/samples/tinyirc.lua
2001-03-12 19:42:46 +00:00

62 lines
1.4 KiB
Lua

function set_add(set, sock)
tinsert(set, sock)
end
function set_remove(set, sock)
for i = 1, getn(set) do
if set[i] == sock then
tremove(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 = bind(host, port1)
server1:timeout(1)
server1.is_server = 1
server2 = bind(host, port2)
server2:timeout(1)
server2.is_server = 1
set = {server1, server2}
while 1 do
local r, s, e, l
r, _, e = select(set, nil)
for i, v in r do
if v.is_server then
s = v:accept()
if s then
s:timeout(1)
set_add(set, s)
write("Added new client. ", getn(set)-2, " total.\n")
end
else
l, e = v:receive()
if e then
v:close()
set_remove(set, v)
write("Removed client. ", getn(set)-2, " total.\n")
end
write("Broadcasting line '", tostring(l), "'.\n")
_, s, e = select(nil, set, 1)
if not e then
for i,v in s do
v:send(l, "\r\n")
end
else
write("No one ready to listen!!!\n")
end
end
end
end