summaryrefslogtreecommitdiff
path: root/src/main_loop.lua
blob: 84afa019ca13ca1c346934b525361608325ffd0c (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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
--
--  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/>.
--

function get_readables()
   local readables = {}
   for _, module in ipairs(module_list.inputs) do
      for _, fd  in ipairs(module:get_read_handles()) do
         table.insert(readables, fd)
      end
   end
   for _, fd  in ipairs(module_list.output.get_read_handles()) do
      table.insert(readables, fd)
   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.inputs) do
      for _, fd  in ipairs(module:get_write_handles()) do
         table.insert(writeables, fd)
      end
   end
   for _, fd  in ipairs(module_list.output.get_write_handles()) do
      table.insert(writeables, fd)
   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()

   local return_value = dispatch_tables:load_handler_tables_file(opt.lua_code)
   if (return_value == defines.KILL_DAEMON) then
      return_value = -1
   else
      return_value = module_list:init(opt)
      if(return_value == defines.KILL_DAEMON) then
         return_value = -1
      else
         return_value = 0
      end
   end

   while return_value == 0 do
      local readable, writeable, err = util.select({ sig, unpack(get_readables()) }, get_writeables(), 10)
      if(err) then
         if(err == "timeout") then
            local child, reason, status = util.waitpid()
            if(child == nil) then
               if(reason ~= nil) then
                  log.printf(log.DEBUG, "waitpid failed: " .. reason)
               end
            else
               log.printf(log.DEBUG, "child with pid %d stopped, reason: %s, status: %s", child.pid, reason, status)
               if(child.cleanup ~= nil) then child:cleanup(reason, status) end
            end
         elseif(err == "signal") then
            -- ignore this
         else
            log.printf(log.ERROR, "select returned with error: %s", err)
            return_value = -1
         end
      else
         for _, reader in ipairs(readable) do
            if(reader == sig) then
               return_value = signal.handle()
               if(return_value == 1) then break end
            else
               local ret = reader:read()
               if(ret == defines.KILL_DAEMON) then
                  return_value = 2
                  break 
               elseif(ret == defines.KILL_MODULE_CLASS) then
                  module_list:unregister_by_class(reader.client_instance.module_instance.class)
               elseif(ret == defines.KILL_MODULE) then
                  module_list:unregister(reader.client_instance.module_instance)
               elseif(ret == defines.KILL_CLIENT) then
                  client_list:unregister(reader.client_instance)
               end
            end
         end
         for _, writer in ipairs(writeable) do
            ret = writer:write()
            if(ret == defines.KILL_DAEMON) then
               return_value = 2
               break 
            elseif(ret == defines.KILL_MODULE_CLASS) then
               module_list:unregister_by_class(writer.client_instance.module_instance.class)
            elseif(ret == defines.KILL_MODULE) then
               module_list:unregister(writer.client_instance.module_instance)
            elseif(ret == defines.KILL_CLIENT) then
               client_list:unregister(writer.client_instance)
            end
         end
      end
      command_queue:check_timeout()
      if(command_queue:command_pending()) then
         local command = command_queue:get_next_command()
         log.printf(log.DEBUG, "sending pending command: %s", command)
         module_list.output:start_command(command)
      end
   end

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

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