summaryrefslogtreecommitdiff
path: root/src/main_loop.lua
blob: 7c745f26a28972e4fa94a43481b3ea44e9fd7fa6 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
--
--  gcsd
--
--  gcsd the generic command sequencer daemon can be used to serialize 
--  commands sent over various paralell communication channels to a
--  single command output. It can be seen as a multiplexer for any
--  kind of communication between a single resource and various clients
--  which want to submit commands to it or query information from it.
--  gcsd is written in C and Lua. The goal is to provide an easy to 
--  understand high level API based on Lua which can be used to
--  implement the business logic of the so formed multiplexer daemon.
--  
--
--  Copyright (C) 2009-2010 Markus Grueneis <gimpf@spreadspace.org>
--                          Christian Pointner <equinox@spreadspace.org>
--
--  This file is part of gcsd.
--
--  gcsd 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.
--
--  gcsd 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 gcsd. If not, see <http://www.gnu.org/licenses/>.
--

local socket = require("socket")
local defines = require("defines")

function get_readables()
   local readables = {}
   for _, module in ipairs(module_list.modules) do
      for _, fd  in ipairs(module:get_read_handles()) do
         table.insert(readables, fd)
      end
   end
   for _, client in ipairs(client_list.clients) do
      for _, fd  in ipairs(client:get_read_handles()) do
         table.insert(readables, fd)
      end
   end
   
   return readables
end

function get_writeables()
   local writeables = { }
   for _, module in ipairs(module_list.modules) do
      for _, fd  in ipairs(module:get_write_handles()) do
         table.insert(writeables, fd)
      end
   end
   for _, client in ipairs(client_list.clients) do
      for _, fd  in ipairs(client:get_write_handles()) do
         table.insert(writeables, fd)
      end
   end
   
   return writeables
end

function main_loop(opt) 
   log.printf(log.NOTICE, "main_loop started")
   local sig = signal.init()

   -- initialize modules
   module_list:init(opt)

   local return_value = 0
   while return_value == 0 do
      local readable, writeable, err = socket.select({ sig, unpack(get_readables()) }, 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
            else
               local ret = input:read()
               if(ret == defines.KILL_DAEMON) then 
                  return_value = 2
                  break 
               elseif(ret == defines.KILL_MODULE_TYPE) then
                  module_list:unregister_by_type(input.client_instance.module_instance.type)
               elseif(ret == defines.KILL_MODULE) then
                  module_list:unregister(input.client_instance.module_instance)
               elseif(ret == defines.KILL_CLIENT) then
                  client_list:unregister(input.client_instance)
               end
            end
         end
         for _, output in ipairs(writeable) do
            ret = output:write()
            if(ret == defines.KILL_CLIENT) then
               client_list:unregister(input.client_instance)
            end
         end
      end
   end

   if(return_value == 2) then return_value = 0 end

   module_list:cleanup()
   signal.stop()
   return return_value
end