mirror of
https://github.com/lunarmodules/luasocket.git
synced 2024-11-08 14:28:21 +01:00
91 lines
3.0 KiB
Lua
91 lines
3.0 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
|
||
|
|
-----------------------------------------------------------------------------
|
||
|
|
dofile("command.lua")
|
||
|
|
test_debug_mode()
|
||
|
|
|
||
|
|
-----------------------------------------------------------------------------
|
||
|
|
-- Bind to address and wait for control connection
|
||
|
|
-----------------------------------------------------------------------------
|
||
|
|
server, err = bind(HOST, PORT)
|
||
|
|
if not server then
|
||
|
|
print(err)
|
||
|
|
exit(1)
|
||
|
|
end
|
||
|
|
print("server: waiting for control connection...")
|
||
|
|
control = server:accept()
|
||
|
|
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 = server:accept()
|
||
|
|
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
|
||
|
|
data:close()
|
||
|
|
data = nil
|
||
|
|
end
|
||
|
|
elseif cmd == ECHO_LINE then
|
||
|
|
str, err = data:receive()
|
||
|
|
if err then fail("server: " .. err) end
|
||
|
|
err = data:send(str, "\n")
|
||
|
|
if err then fail("server: " .. err) end
|
||
|
|
elseif cmd == ECHO_BLOCK then
|
||
|
|
str, err = data:receive(par)
|
||
|
|
if err then fail("server: " .. err) end
|
||
|
|
err = data:send(str)
|
||
|
|
if err then fail("server: " .. err) end
|
||
|
|
elseif cmd == RECEIVE_BLOCK then
|
||
|
|
str, err = data:receive(par)
|
||
|
|
elseif cmd == SEND_BLOCK then
|
||
|
|
err = data:send(str)
|
||
|
|
elseif cmd == ECHO_TIMEOUT then
|
||
|
|
str, err = data:receive(par)
|
||
|
|
if err then fail("server: " .. err) end
|
||
|
|
err = data:send(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
|