summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--mode-tcplistener.lua160
-rw-r--r--openwrt/rhctl/files/rhctl.config7
2 files changed, 167 insertions, 0 deletions
diff --git a/mode-tcplistener.lua b/mode-tcplistener.lua
new file mode 100644
index 0000000..f1f3e92
--- /dev/null
+++ b/mode-tcplistener.lua
@@ -0,0 +1,160 @@
+--
+-- rhctl
+--
+-- Copyright (C) 2009 Christian Pointner <equinox@spreadspace.org>
+--
+-- This file is part of rhctl.
+--
+-- rhctl is free software: you can redistribute it and/or modify
+-- it under the terms of the GNU General Public License as published by
+-- the Free Software Foundation, either version 3 of the License, or
+-- any later version.
+--
+-- rhctl is distributed in the hope that it will be useful,
+-- but WITHOUT ANY WARRANTY; without even the implied warranty of
+-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+-- GNU General Public License for more details.
+--
+-- You should have received a copy of the GNU General Public License
+-- along with rhctl. If not, see <http://www.gnu.org/licenses/>.
+--
+
+socket = require("socket")
+
+current_mode = nil
+
+function init_server(host, port)
+ local server = assert(socket.tcp())
+
+ assert(server:setoption('reuseaddr', true))
+ assert(server:bind(host, port))
+ assert(server:listen(5))
+
+ return server
+end
+
+local clients = {}
+
+function add_client(hdl)
+ -- print("new client(" .. hdl:getfd() .. ") from " .. hdl:getpeername())
+ local client = {}
+ if current_mode then
+ client.buffer = current_mode .. "\n"
+ else
+ client.buffer = ""
+ end
+ client.hdl = hdl
+ client.getfd = function() return hdl:getfd() end
+ client.dirty = function() return hdl:dirty() end
+ table.insert(clients, client)
+end
+
+function remove_client(c)
+ local idx = 0
+ for idx, client in ipairs(clients) do
+ if client == c then
+ break
+ end
+ end
+
+ -- print("removing client(" .. c.hdl:getfd() .. ")")
+ c.hdl:close()
+ table.remove(clients, idx)
+end
+
+function cleanup_clients()
+ for _, client in ipairs(clients) do
+ client:close()
+ end
+end
+
+function clients_get_writeables()
+ local fds = {}
+
+ for _, client in ipairs(clients) do
+ if client.buffer ~= "" then
+ table.insert(fds, client)
+ end
+ end
+
+ return fds
+end
+
+function clients_senddata(data)
+ for _, client in ipairs(clients) do
+ client.buffer = client.buffer .. data .. "\n"
+ end
+end
+
+function process_cmd(message)
+ log.printf(log.DEBUG, "received message: '%s'", message)
+
+ local new_mode = nil
+ local exps = { "Current Mode: (%a+)", "new Mode: (%a+)" }
+ for _, exp in ipairs(exps) do
+ new_mode = string.match(message, exp)
+ if(new_mode) then
+ new_mode = string.lower(new_mode)
+ break
+ end
+ end
+
+ if(new_mode and new_mode ~= current_mode) then
+ clients_senddata(new_mode)
+ current_mode = new_mode
+ end
+
+ return 0
+end
+
+function main_loop(opt)
+ log.printf(log.NOTICE, "main_loop started")
+ local sig = signal.init()
+ local cmdfd = cmd.init()
+
+ local server = init_server("*", "2345")
+
+ cmd.send_string("listen mode");
+ cmd.send_string("status");
+
+ local return_value = 0
+ while return_value == 0 do
+ local readable, _, err = socket.select({ sig , cmdfd , server , unpack(clients) }, clients_get_writeables())
+ if(err) then
+ log.printf(log.ERROR, "select returned with error: %s", err)
+ return_value = -1
+ else
+ for _, input in ipairs(readable) do
+ if input == sig then
+ return_value = signal.handle()
+ if(return_value == 1) then break end
+ elseif input == cmdfd then
+ return_value = cmd.recv_data(process_cmd)
+ if(return_value ~= 0) then break end
+ elseif input == server then
+ local client = server:accept()
+ if(client == nil) then
+ return_value =-3
+ break
+ else
+ add_client(client)
+ end
+ else
+ if input.hdl then
+ -- receive is insanely stupid, therefore we must always read one byte only
+ local _, err = input.hdl:receive(1)
+ if err then
+ remove_client(input)
+ end
+ end
+ end
+ end
+ end
+ end
+
+ server:close()
+ cleanup_clients()
+
+ signal.stop()
+ return return_value
+end
diff --git a/openwrt/rhctl/files/rhctl.config b/openwrt/rhctl/files/rhctl.config
index cf25990..6dbb9c0 100644
--- a/openwrt/rhctl/files/rhctl.config
+++ b/openwrt/rhctl/files/rhctl.config
@@ -46,3 +46,10 @@ config 'luaclient'
option log 'syslog:3,luaclient-silence-watch,daemon'
option command_sock '/var/run/rhctl/switchctl.sock'
option lua_file '/usr/share/rhctl/silence-watch.lua'
+
+config 'luaclient'
+# option disabled 1
+ option name 'tcpserver'
+ option log 'syslog:3,luaclient-mode-tcpserver,daemon'
+ option command_sock '/var/run/rhctl/switchctl.sock'
+ option lua_file '/usr/share/rhctl/mode-tcpserver.lua'