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