mirror of
https://github.com/lunarmodules/luasocket.git
synced 2024-12-26 12:28:21 +01:00
105 lines
3.4 KiB
Lua
105 lines
3.4 KiB
Lua
-----------------------------------------------------------------------------
|
|
-- LuaSocket automated test module
|
|
-- server.lua
|
|
-- This is the server module. It's completely controled by the client module
|
|
-- by the use of a control connection.
|
|
-----------------------------------------------------------------------------
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- Read command definitions
|
|
-----------------------------------------------------------------------------
|
|
assert(dofile("testcmd.lua"))
|
|
test_debug_mode()
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- Get host and port from command line
|
|
-----------------------------------------------------------------------------
|
|
HOST = "localhost"
|
|
PORT = 2020
|
|
if arg then
|
|
HOST = arg[1] or HOST
|
|
PORT = arg[2] or PORT
|
|
end
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- Start control connection
|
|
-----------------------------------------------------------------------------
|
|
server, err = bind(HOST, PORT)
|
|
if not server then
|
|
fail(err)
|
|
exit(1)
|
|
end
|
|
print("server: waiting for control connection...")
|
|
control = accept(server)
|
|
print("server: control connection stablished!")
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- Executes a command, detecting any possible failures
|
|
-- Input
|
|
-- cmd: command to be executed
|
|
-- par: command parameters, if needed
|
|
-----------------------------------------------------------------------------
|
|
function execute_command(cmd, par)
|
|
if cmd == CONNECT then
|
|
print("server: waiting for data connection...")
|
|
data = accept(server)
|
|
if not data then
|
|
fail("server: unable to start data connection!")
|
|
else
|
|
print("server: data connection stablished!")
|
|
end
|
|
elseif cmd == CLOSE then
|
|
print("server: closing connection with client...")
|
|
if data then
|
|
close(data)
|
|
data = nil
|
|
end
|
|
elseif cmd == ECHO_LINE then
|
|
str, err = receive(data)
|
|
if err then fail("server: " .. err) end
|
|
err = send(data, str, "\n")
|
|
if err then fail("server: " .. err) end
|
|
elseif cmd == ECHO_BLOCK then
|
|
str, err = receive(data, par)
|
|
print(format("server: received %d bytes", strlen(str)))
|
|
if err then fail("server: " .. err) end
|
|
print(format("server: sending %d bytes", strlen(str)))
|
|
err = send(data, str)
|
|
if err then fail("server: " .. err) end
|
|
elseif cmd == RECEIVE_BLOCK then
|
|
str, err = receive(data, par)
|
|
print(format("server: received %d bytes", strlen(str)))
|
|
elseif cmd == SEND_BLOCK then
|
|
print(format("server: sending %d bytes", strlen(str)))
|
|
err = send(data, str)
|
|
elseif cmd == ECHO_TIMEOUT then
|
|
str, err = receive(data, par)
|
|
if err then fail("server: " .. err) end
|
|
err = send(data, str)
|
|
if err then fail("server: " .. err) end
|
|
elseif cmd == COMMAND then
|
|
cmd, par = get_command()
|
|
send_command(cmd, par)
|
|
elseif cmd == EXIT then
|
|
print("server: exiting...")
|
|
exit(0)
|
|
elseif cmd == SYNC then
|
|
print("server: synchronizing...")
|
|
send_command(SYNC)
|
|
elseif cmd == SLEEP then
|
|
print("server: sleeping for " .. par .. " seconds...")
|
|
sleep(par)
|
|
print("server: woke up!")
|
|
end
|
|
end
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- Loop forever, accepting and executing commands
|
|
-----------------------------------------------------------------------------
|
|
while 1 do
|
|
cmd, par = get_command()
|
|
if not cmd then fail("server: " .. par) end
|
|
print_command(cmd, par)
|
|
execute_command(cmd, par)
|
|
end
|