summaryrefslogtreecommitdiff
path: root/src/main_loop.lua
blob: ec68b70ef0cd3104d8b5014bfdc3fb6ee14e1931 (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
--
--  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.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(readables, 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 = module_list:init(opt)
   if(return_value == defines.KILL_DAEMON) then
      return_value = -1
   else
      return_value = 0
   end

   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 _, 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_CLIENT) then
               client_list:unregister(writer.client_instance)
            end
         end
         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
   end

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

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