]> git.stg.codes - stg.git/blob - projects/sgauth/main.cpp
b19ff6ed27eafe5c74f754085a44c2351d6b2fe7
[stg.git] / projects / sgauth / 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  */
20
21  /*
22  $Revision: 1.13 $
23  $Date: 2010/04/14 09:01:29 $
24  $Author: faust $
25  */
26
27
28 #include <stdio.h>
29 #include <stdlib.h>
30
31 #ifndef WIN32
32 #include <unistd.h>
33 #include <sys/socket.h>
34 #include <netinet/in.h>
35 #include <arpa/inet.h>
36 #include <signal.h>
37 #include "conffiles.h"
38 #endif
39
40 #include <string.h>
41 #include <vector>
42 #include <iostream>
43
44 #include "ia_auth_c.h"
45 #include "common.h"
46 #include "common_settings.h"
47 #include "web.h"
48
49 int mes;
50 char infoText[256];
51 char messageText[256];
52
53 const int winKOI = 0;
54
55 IA_CLIENT_PROT * clnp;
56 WEB * web = NULL;
57
58 using namespace std;
59
60 time_t stgTime;
61
62 //-----------------------------------------------------------------------------
63 class SETTINGS: public COMMON_SETTINGS
64 {
65 public:
66                     SETTINGS();
67     virtual         ~SETTINGS(){};
68     virtual int     Reload(){ return 0; };
69     void            SetConfFile(string confFile);
70     virtual int     ReadSettings();
71
72     virtual string  GetStrError() const;
73
74     string          GetServerName() const;
75     uint16_t        GetServerPort() const;
76     uint16_t        GetLocalPort() const;
77
78     string          GetLogin() const;
79     string          GetPassword() const;
80
81     bool            GetDaemon() const;
82     bool            GetShowPid() const;
83     bool            GetNoWeb() const;
84     bool            GetReconnect() const;
85     int             GetRefreshPeriod() const;
86     uint32_t        GetListenWebIP() const;
87
88     void            Print() const;
89
90 private:
91     string          login;
92     string          password;
93     string          serverName;
94     int             port;
95     int             localPort;
96     uint32_t        listenWebIP;
97     int             refreshPeriod;
98
99     bool            daemon;
100     bool            noWeb;
101     bool            reconnect;
102     bool            showPid;
103
104     string          confFile;
105 };
106 //-----------------------------------------------------------------------------
107 SETTINGS::SETTINGS()
108 {
109 confFile = "/etc/sgauth.conf";
110 }
111 //-----------------------------------------------------------------------------
112 void SETTINGS::SetConfFile(string confFile)
113 {
114 SETTINGS::confFile = confFile;
115 }
116 //-----------------------------------------------------------------------------
117 int SETTINGS::ReadSettings()
118 {
119 CONFIGFILE * cf;
120
121 cf = new CONFIGFILE(confFile);
122 string tmp;
123 int e = cf->Error();
124
125 if (e)
126     {
127     printf("Cannot read file.\n");
128     delete cf;
129     return -1;
130     }
131
132 cf->ReadString("Login", &login, "/?--?--?*");
133 if (login == "/?--?--?*")
134     {
135     strError = "Parameter \'Login\' not found.";
136     delete cf;
137     return -1;
138     }
139
140 cf->ReadString("Password", &password, "/?--?--?*");
141 if (login == "/?--?--?*")
142     {
143     strError = "Parameter \'Password\' not found.";
144     delete cf;
145     return -1;
146     }
147
148 cf->ReadString("ServerName", &serverName, "?*?*?");
149 if (serverName == "?*?*?")
150     {
151     strError = "Parameter \'ServerName\' not found.";
152     delete cf;
153     return -1;
154     }
155
156 cf->ReadString("ListenWebIP", &tmp, "127.0.0.1");
157 listenWebIP = inet_addr(tmp.c_str());
158 if (listenWebIP == INADDR_NONE)
159     {
160     strError = "Parameter \'ListenWebIP\' is not valid.";
161     delete cf;
162     return -1;
163     }
164
165 cf->ReadString("ServerPort", &tmp, "5555");
166 if (ParseIntInRange(tmp, 1, 65535, &port))
167     {
168     strError = "Parameter \'ServerPort\' is not valid.";
169     delete cf;
170     return -1;
171     }
172
173 cf->ReadString("LocalPort", &tmp, "0");
174 if (ParseIntInRange(tmp, 0, 65535, &localPort))
175     {
176     strError = "Parameter \'LocalPort\' is not valid.";
177     delete cf;
178     return -1;
179     }
180
181 printf("LocalPort=%d\n", localPort);
182
183 cf->ReadString("RefreshPeriod", &tmp, "5");
184 if (ParseIntInRange(tmp, 1, 24*3600, &refreshPeriod))
185     {
186     strError = "Parameter \'RefreshPeriod\' is not valid.";
187     delete cf;
188     return -1;
189     }
190
191 cf->ReadString("Reconnect", &tmp, "yes");
192 if (ParseYesNo(tmp, &reconnect))
193     {
194     strError = "Parameter \'Reconnect\' is not valid.";
195     delete cf;
196     return -1;
197     }
198
199 cf->ReadString("Daemon", &tmp, "yes");
200 if (ParseYesNo(tmp, &daemon))
201     {
202     strError = "Parameter \'Daemon\' is not valid.";
203     delete cf;
204     return -1;
205     }
206
207 cf->ReadString("ShowPid", &tmp, "no");
208 if (ParseYesNo(tmp, &showPid))
209     {
210     strError = "Parameter \'ShowPid\' is not valid.";
211     delete cf;
212     return -1;
213     }
214
215 cf->ReadString("DisableWeb", &tmp, "no");
216 if (ParseYesNo(tmp, &noWeb))
217     {
218     strError = "Parameter \'DisableWeb\' is not valid.";
219     delete cf;
220     return -1;
221     }
222
223 delete cf;
224 return 0;
225 }
226 //-----------------------------------------------------------------------------
227 string SETTINGS::GetStrError() const
228 {
229 return strError;
230 }
231 //-----------------------------------------------------------------------------
232 string SETTINGS::GetLogin() const
233 {
234 return login;
235 }
236 //-----------------------------------------------------------------------------
237 string SETTINGS::GetPassword() const
238 {
239 return password;
240 }
241 //-----------------------------------------------------------------------------
242 string SETTINGS::GetServerName() const
243 {
244 return serverName;
245 }
246 //-----------------------------------------------------------------------------
247 uint16_t SETTINGS::GetServerPort() const
248 {
249 return port;
250 }
251 //-----------------------------------------------------------------------------
252 uint16_t SETTINGS::GetLocalPort() const
253 {
254 return localPort;
255 }
256 //-----------------------------------------------------------------------------
257 int SETTINGS::GetRefreshPeriod() const
258 {
259 return refreshPeriod;
260 }
261 //-----------------------------------------------------------------------------
262 bool SETTINGS::GetDaemon() const
263 {
264 return daemon;
265 }
266 //-----------------------------------------------------------------------------
267 bool SETTINGS::GetNoWeb() const
268 {
269 return noWeb;
270 }
271 //-----------------------------------------------------------------------------
272 bool SETTINGS::GetShowPid() const
273 {
274 return showPid;
275 }
276 //-----------------------------------------------------------------------------
277 bool SETTINGS::GetReconnect() const
278 {
279 return reconnect;
280 }
281 //-----------------------------------------------------------------------------
282 uint32_t SETTINGS::GetListenWebIP() const
283 {
284 return listenWebIP;
285 }
286 //-----------------------------------------------------------------------------
287 void SETTINGS::Print() const
288 {
289 cout << "login = " << login << endl;
290 cout << "password = " << password << endl;
291 cout << "ip = " << serverName << endl;
292 cout << "port = " << port << endl;
293 cout << "localPort = " << localPort << endl;
294 cout << "listenWebIP = " << inet_ntostring(listenWebIP) << endl;
295 cout << "DisableWeb = " << noWeb << endl;
296 cout << "refreshPeriod = " << refreshPeriod << endl;
297 cout << "daemon = " << daemon << endl;
298 cout << "reconnect = " << reconnect << endl;
299 }
300 //-----------------------------------------------------------------------------
301 void Usage()
302 {
303 printf("sgauth <server> <port> <login> <password>\n"); //TODO change to correct
304 }
305 //-----------------------------------------------------------------------------
306 void SetDirName(const vector<string> & dn, void *)
307 {
308 for (int j = 0; j < DIR_NUM; j++)
309     {
310     if (winKOI)
311         {
312         string dir;
313         KOIToWin(dn[j], &dir);
314         if (web)
315             web->SetDirName(dir, j);
316         }
317     else
318         {
319         if (web)
320             web->SetDirName(dn[j], j);
321         }
322     }
323 }
324 //-----------------------------------------------------------------------------
325 void StatUpdate(const LOADSTAT & ls, void *)
326 {
327 if (web)
328     web->UpdateStat(ls);
329 }
330 //-----------------------------------------------------------------------------
331 void StatusChanged(int, void *)
332 {
333
334 }
335 //-----------------------------------------------------------------------------
336 void ShowMessage(const string & message, int i, int, int, void *)
337 {
338 if (web)
339     web->AddMessage(message, i);
340 }
341 //-----------------------------------------------------------------------------
342 void ShowError(const string & message, int, void *)
343 {
344 if (web)
345      web->AddMessage(message, 0);
346 }
347 //-----------------------------------------------------------------------------
348 #ifndef WIN32
349 void CatchUSR1(int)
350 {
351 if (clnp->GetAuthorized())
352     {
353     cout << "Connect" << endl;
354     clnp->Connect();
355         }
356 }
357 //-----------------------------------------------------------------------------
358 void CatchUSR2(int)
359 {
360 cout << "Disconnect" << endl;
361 clnp->Disconnect();
362 }
363 //-----------------------------------------------------------------------------
364 void CatchTERM(int)
365 {
366 cout << "Terminated" << endl;
367 clnp->Disconnect();
368 sleep(2);
369 exit(0);
370 }
371 //-----------------------------------------------------------------------------
372 static void SetSignalHandlers()
373 {
374 struct sigaction newsa, oldsa;
375 sigset_t sigmask;
376
377 sigemptyset(&sigmask);
378 sigaddset(&sigmask, SIGTERM);
379 newsa.sa_handler = CatchTERM;
380 newsa.sa_mask = sigmask;
381 newsa.sa_flags = 0;
382 sigaction(SIGTERM, &newsa, &oldsa);
383
384 sigemptyset(&sigmask);
385 sigaddset(&sigmask, SIGINT);
386 newsa.sa_handler = CatchTERM;
387 newsa.sa_mask = sigmask;
388 newsa.sa_flags = 0;
389 sigaction(SIGINT, &newsa, &oldsa);
390
391 sigemptyset(&sigmask);
392 sigaddset(&sigmask, SIGUSR1);
393 newsa.sa_handler = CatchUSR1;
394 newsa.sa_mask = sigmask;
395 newsa.sa_flags = 0;
396 sigaction(SIGUSR1, &newsa, &oldsa);
397
398 sigemptyset(&sigmask);
399 sigaddset(&sigmask, SIGUSR2);
400 newsa.sa_handler = CatchUSR2;
401 newsa.sa_mask = sigmask;
402 newsa.sa_flags = 0;
403 sigaction(SIGUSR2, &newsa, &oldsa);
404
405 return;
406 }
407 #endif
408 //-----------------------------------------------------------------------------
409 int main(int argc, char *argv[])
410 {
411 //int port;
412 //char *endptr;
413
414 SETTINGS settings;
415
416 #ifndef WIN32
417 if (argc == 2)
418 #else
419 if(0)
420 #endif
421     {
422     settings.SetConfFile(argv[1]);
423     if (settings.ReadSettings())
424         {
425         printf("ReadSettingsError\n");
426         printf("%s\n", settings.GetStrError().c_str());
427         exit(-1);
428         }
429     settings.Print();
430     }
431 else
432     {
433     /*if (argc != 5)
434         {
435         Usage();
436         exit(1);
437         }
438     else
439         {
440         string serverName(argv[1]);
441         port = strtol(argv[2], &endptr, 10);
442         if (*endptr != 0)
443             {
444             printf("Invalid port!\n");
445             exit(1);
446             }
447         login = argv[3];
448         passwd = argv[4];
449         }*/
450     }
451
452 //settings.Print();
453
454 #ifndef WIN32
455 if (settings.GetDaemon())
456     {
457     /*close(0);
458     close(1);
459     close(2);*/
460
461     switch (fork())
462         {
463         case -1:    // ìÁÖÁ
464             exit(1);
465             break;
466
467         case 0:    // ðÏÔÏÍÏË
468             setsid();
469             break;
470
471         default:    // ïÓÎÏ×ÎÏÊ ÐÒÏÃÅÓÓ
472             exit(0);
473             break;
474         }
475     }
476
477
478
479 #endif
480
481 clnp = new IA_CLIENT_PROT(settings.GetServerName(), settings.GetServerPort(), settings.GetLocalPort());
482
483 if (!settings.GetNoWeb())
484     {
485     web = new WEB();
486     web->SetRefreshPagePeriod(settings.GetRefreshPeriod());
487     web->SetListenAddr(settings.GetListenWebIP());
488     web->Start();
489     }
490
491 clnp->SetLogin(settings.GetLogin());
492 clnp->SetPassword(settings.GetPassword());
493
494 clnp->SetStatusChangedCb(StatusChanged, NULL);
495 clnp->SetInfoCb(ShowMessage, NULL);
496 clnp->SetErrorCb(ShowError, NULL);
497 clnp->SetDirNameCb(SetDirName, NULL);
498 clnp->SetStatChangedCb(StatUpdate, NULL);
499 clnp->SetReconnect(settings.GetReconnect());
500
501
502 clnp->Start();
503
504 SetSignalHandlers();
505
506 #ifdef LINUX
507 for (int i = 1; i < argc; i++)
508     memset(argv[i], 0, strlen(argv[i]));
509
510 if(argc > 1)
511     strcpy(argv[1], "Connecting...");
512 #endif
513
514 #ifdef FREEBSD
515 setproctitle("Connecting...");
516 #endif
517 clnp->Connect();
518
519 while (1)
520     {
521     #ifdef WIN32
522     Sleep(200);
523     #else
524     usleep(200000);
525     #endif
526
527     char state[20];
528
529     if (clnp->GetAuthorized())
530         {
531         if (settings.GetShowPid())
532             sprintf(state, "On %d", getpid());
533         else
534             strcpy(state, "Online");
535         }
536     else
537         {
538         if (settings.GetShowPid())
539             sprintf(state, "Off %d", getpid());
540         else
541             strcpy(state, "Offline");
542         }
543
544     #ifdef LINUX
545     for (int i = 1; i < argc; i++)
546         memset(argv[i], 0, strlen(argv[i]));
547     if(argc > 1)
548         strcpy(argv[1], state);
549     #endif
550
551     #ifdef FREEBSD
552     setproctitle(state);
553     #endif
554
555     #ifdef FREEBSD_5
556     setproctitle(state);
557     #endif
558     }
559
560 return 0;
561 }
562 //-----------------------------------------------------------------------------
563
564