]> git.stg.codes - stg.git/blob - projects/rscriptd/main.cpp
Fixed Split.
[stg.git] / projects / rscriptd / main.cpp
1 /*
2  *    This program is free software; you can redistribute it and/or modify
3  *    it under the terms of the GNU General Public License as published by
4  *    the Free Software Foundation; either version 2 of the License, or
5  *    (at your option) any later version.
6  *
7  *    This program is distributed in the hope that it will be useful,
8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  *    GNU General Public License for more details.
11  *
12  *    You should have received a copy of the GNU General Public License
13  *    along with this program; if not, write to the Free Software
14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
15  */
16
17 /*
18  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
19  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
20  */
21
22  /*
23  $Revision: 1.19 $
24  $Author: faust $
25  $Date: 2010/09/10 06:37:45 $
26  */
27
28 #include <sys/types.h>
29 #include <sys/ipc.h>
30 #include <sys/msg.h>
31 #include <sys/stat.h>
32 #include <sys/wait.h>
33 #include <fcntl.h> // creat
34 #include <unistd.h>
35
36 #include <cstdlib>
37 #include <cstdio>
38 #include <csignal>
39 #include <cerrno>
40 #include <cstring> // strerror
41 #include <set>
42
43 #include "stg/common.h"
44 #include "stg/logger.h"
45 #include "stg/scriptexecuter.h"
46 #include "stg/conffiles.h"
47 #include "stg/version.h"
48 #include "listener.h"
49 #include "pidfile.h"
50
51 using namespace std;
52
53 #ifdef DEBUG
54 # define MAIN_DEBUG  1
55 # define NO_DAEMON    1
56 #endif
57
58 #define START_FILE "/._ST_ART_ED_"
59
60 set<pid_t> executersPid;
61 volatile time_t stgTime = time(NULL);
62
63 //-----------------------------------------------------------------------------
64 void KillExecuters()
65 {
66 set<pid_t>::iterator pid;
67 pid = executersPid.begin();
68 while (pid != executersPid.end())
69     {
70     printfd(__FILE__, "KillExecuters pid=%d\n", *pid);
71     kill(*pid, SIGUSR1);
72     ++pid;
73     }
74 }
75 //-----------------------------------------------------------------------------
76 #ifdef LINUX
77 int StartScriptExecuter(char * procName, int msgKey, int * msgID)
78 #else
79 int StartScriptExecuter(char *, int msgKey, int * msgID)
80 #endif
81 {
82 STG_LOGGER & WriteServLog = GetStgLogger();
83
84 if (*msgID == -11)   // If msgID == -11 - first call. Create queue
85     {
86     for (int i = 0; i < 2; i++)
87         {
88         *msgID = msgget(msgKey, IPC_CREAT | IPC_EXCL | 0600);
89
90         if (*msgID == -1)
91             {
92             *msgID = msgget(msgKey, 0);
93             if (*msgID == -1)
94                 {
95                 WriteServLog("Message queue not created.");
96                 return -1;
97                 }
98             else
99                 {
100                 msgctl(*msgID, IPC_RMID, NULL);
101                 }
102             }
103         else
104             {
105             WriteServLog("Message queue created successfully. msgKey=%d msgID=%d", msgKey, *msgID);
106             break;
107             }
108         }
109     }
110
111 pid_t executerPid = fork();
112
113 switch (executerPid)
114     {
115     case -1:    // Failure
116         WriteServLog("Fork error!");
117         return -1;
118
119     case 0:     // Child
120         //close(0);
121         //close(1);
122         //close(2);
123         //setsid();
124 #ifdef LINUX
125         Executer(*msgID, executerPid, procName);
126 #else
127         Executer(*msgID, executerPid);
128 #endif
129         return 1;
130
131     default:    // Parent
132         if (executersPid.empty())
133 #ifdef LINUX
134             Executer(*msgID, executerPid, NULL);
135 #else
136             Executer(*msgID, executerPid);
137 #endif
138         executersPid.insert(executerPid);
139     }
140 return 0;
141 }
142 //-----------------------------------------------------------------------------
143 void StopScriptExecuter(int msgID)
144 {
145 STG_LOGGER & WriteServLog = GetStgLogger();
146
147 for (int i = 0; i < 5; ++i)
148     {
149     struct msqid_ds data;
150     if (msgctl(msgID, IPC_STAT, &data))
151         {
152         int e = errno;
153         printfd(__FILE__, "StopScriptExecuter() - msgctl for IPC_STAT failed: '%s'\n", strerror(e));
154         WriteServLog( "Failed to check queue emptiness: '%s'", strerror(e));
155         break;
156         }
157
158     WriteServLog("Messages in queue: %d", data.msg_qnum);
159
160     if (data.msg_qnum == 0)
161         break;
162
163     struct timespec ts = {1, 0};
164     nanosleep(&ts, NULL);
165     }
166
167 if (msgctl(msgID, IPC_RMID, NULL))
168     {
169     int e = errno;
170     printfd(__FILE__, "StopScriptExecuter() - msgctl for IPC_STAT failed: '%s'\n", strerror(e));
171     WriteServLog("Failed to remove queue: '%s'", strerror(e));
172     }
173 else
174     {
175     WriteServLog("Queue removed successfully.");
176     }
177
178 KillExecuters();
179 }
180 //-----------------------------------------------------------------------------
181 #ifdef NO_DAEMON
182 int ForkAndWait(const string &)
183 #else
184 int ForkAndWait(const string & confDir)
185 #endif
186 {
187 #ifndef NO_DAEMON
188 pid_t childPid = fork();
189
190 switch (childPid)
191     {
192     case -1:    // Failure
193         return -1;
194         break;
195
196     case 0:     // Child
197         //close(0);
198         close(1);
199         close(2);
200         setsid();
201         break;
202
203     default:    // Parent
204         exit(1);
205         break;
206     }
207 #endif
208 return 0;
209 }
210 //-----------------------------------------------------------------------------
211 int main(int argc, char * argv[])
212 {
213 CONFIGFILE * cfg = NULL;
214 LISTENER * listener = NULL;
215 int msgID = -11;
216 int execNum = 0;
217 int execMsgKey = 0;
218
219 string logFileName;
220 string confDir;
221 string password;
222 string onConnect;
223 string onDisconnect;
224 int port;
225 int userTimeout;
226
227 if (getuid())
228     {
229     printf("You must be root. Exit.\n");
230     exit(1);
231     }
232
233 if (argc == 2)
234     cfg = new CONFIGFILE(argv[1]);
235 else
236     cfg = new CONFIGFILE("/etc/rscriptd/rscriptd.conf");
237
238 if (cfg->Error())
239     {
240     STG_LOGGER & WriteServLog = GetStgLogger();
241     WriteServLog.SetLogFileName("/var/log/rscriptd.log");
242     WriteServLog("Error reading config file!");
243     delete cfg;
244     return EXIT_FAILURE;
245     }
246
247 cfg->ReadString("LogFileName", &logFileName, "/var/log/rscriptd.log");
248 cfg->ReadInt("ExecutersNum", &execNum, 1);
249 cfg->ReadInt("ExecMsgKey", &execMsgKey, 5555);
250 cfg->ReadString("ConfigDir", &confDir, "/etc/rscriptd");
251 cfg->ReadString("Password", &password, "");
252 cfg->ReadInt("Port", &port, 5555);
253 cfg->ReadInt("UserTimeout", &userTimeout, 60);
254 cfg->ReadString("ScriptOnConnect", &onConnect, "/etc/rscriptd/OnConnect");
255 cfg->ReadString("ScriptOnDisconnect", &onDisconnect, "/etc/rscriptd/OnDisconnect");
256
257 if (ForkAndWait(confDir) < 0)
258     {
259     STG_LOGGER & WriteServLog = GetStgLogger();
260     WriteServLog("Fork error!");
261     delete cfg;
262     return EXIT_FAILURE;
263     }
264
265 STG_LOGGER & WriteServLog = GetStgLogger();
266 PIDFile pidFile("/var/run/rscriptd.pid");
267 WriteServLog.SetLogFileName(logFileName);
268 WriteServLog("rscriptd v. %s", SERVER_VERSION);
269
270 for (int i = 0; i < execNum; i++)
271     {
272     int ret = StartScriptExecuter(argv[0], execMsgKey, &msgID);
273     if (ret < 0)
274         {
275         STG_LOGGER & WriteServLog = GetStgLogger();
276         WriteServLog("Start Script Executer error!");
277         delete cfg;
278         return EXIT_FAILURE;
279         }
280     if (ret == 1)
281         {
282         delete cfg;
283         return EXIT_SUCCESS;
284         }
285     }
286
287 listener = new LISTENER();
288 listener->SetPort(port);
289 listener->SetPassword(password);
290 listener->SetUserTimeout(userTimeout);
291 listener->SetScriptOnConnect(onConnect);
292 listener->SetScriptOnDisconnect(onDisconnect);
293
294 listener->Start();
295
296 WriteServLog("rscriptd started successfully.");
297 WriteServLog("+++++++++++++++++++++++++++++++++++++++++++++");
298
299 sigset_t signalSet;
300 sigfillset(&signalSet);
301 pthread_sigmask(SIG_BLOCK, &signalSet, NULL);
302
303 while (true)
304     {
305     sigfillset(&signalSet);
306     int sig = 0;
307     printfd(__FILE__, "Before sigwait\n");
308     sigwait(&signalSet, &sig);
309     printfd(__FILE__, "After sigwait. Signal: %d\n", sig);
310     bool stop = false;
311     switch (sig)
312         {
313         case SIGTERM:
314             stop = true;
315             break;
316         case SIGINT:
317             stop = true;
318             break;
319         default:
320             WriteServLog("Ignore signel %d", sig);
321             break;
322         }
323     if (stop)
324         break;
325     }
326
327 listener->Stop();
328
329 WriteServLog("+++++++++++++++++++++++++++++++++++++++++++++");
330
331 StopScriptExecuter(msgID);
332
333 WriteServLog("rscriptd stopped successfully.");
334 WriteServLog("---------------------------------------------");
335
336 delete listener;
337 delete cfg;
338 return EXIT_SUCCESS;
339 }
340 //-----------------------------------------------------------------------------