]> git.stg.codes - stg.git/blob - projects/sgauth/main.cpp
Merge branch 'fix-gts-auth-errors' into full-month-stats
[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 #include <unistd.h>
28 #include <sys/socket.h>
29 #include <netinet/in.h>
30 #include <arpa/inet.h>
31
32 #include <csignal>
33 #include <cstdio>
34 #include <cstring>
35 #include <iostream>
36 #include <vector>
37
38 #include "stg/ia.h"
39 #include "stg/common.h"
40 #include "web.h"
41 #include "settings_impl.h"
42
43 int mes;
44 char infoText[256];
45 char messageText[256];
46
47 const int winKOI = 0;
48
49 IA_CLIENT_PROT * clnp;
50 WEB * web = NULL;
51
52 using namespace std;
53
54 time_t stgTime;
55
56 //-----------------------------------------------------------------------------
57 //-----------------------------------------------------------------------------
58 //-----------------------------------------------------------------------------
59 void Usage()
60 {
61 printf("sgauth <path_to_config>\n");
62 }
63 //-----------------------------------------------------------------------------
64 void SetDirName(const vector<string> & dn, void *)
65 {
66 for (int j = 0; j < DIR_NUM; j++)
67     {
68     if (winKOI)
69         {
70         string dir;
71         KOIToWin(dn[j], &dir);
72         if (web)
73             web->SetDirName(dir, j);
74         }
75     else
76         {
77         if (web)
78             web->SetDirName(dn[j], j);
79         }
80     }
81 }
82 //-----------------------------------------------------------------------------
83 void StatUpdate(const LOADSTAT & ls, void *)
84 {
85 if (web)
86     web->UpdateStat(ls);
87 }
88 //-----------------------------------------------------------------------------
89 void StatusChanged(int, void *)
90 {
91 }
92 //-----------------------------------------------------------------------------
93 void ShowMessage(const string & message, int i, int, int, void *)
94 {
95 if (web)
96     web->AddMessage(message, i);
97 }
98 //-----------------------------------------------------------------------------
99 void ShowError(const string & message, int, void *)
100 {
101 if (web)
102      web->AddMessage(message, 0);
103 }
104 //-----------------------------------------------------------------------------
105 void CatchUSR1(int)
106 {
107 if (clnp->GetAuthorized())
108     {
109     cout << "Connect" << endl;
110     clnp->Connect();
111     }
112 }
113 //-----------------------------------------------------------------------------
114 void CatchUSR2(int)
115 {
116 cout << "Disconnect" << endl;
117 clnp->Disconnect();
118 }
119 //-----------------------------------------------------------------------------
120 void CatchTERM(int)
121 {
122 cout << "Terminated" << endl;
123 clnp->Disconnect();
124 sleep(2);
125 exit(0);
126 }
127 //-----------------------------------------------------------------------------
128 static void SetSignalHandlers()
129 {
130 struct sigaction newsa, oldsa;
131 sigset_t sigmask;
132
133 sigemptyset(&sigmask);
134 sigaddset(&sigmask, SIGTERM);
135 newsa.sa_handler = CatchTERM;
136 newsa.sa_mask = sigmask;
137 newsa.sa_flags = 0;
138 sigaction(SIGTERM, &newsa, &oldsa);
139
140 sigemptyset(&sigmask);
141 sigaddset(&sigmask, SIGINT);
142 newsa.sa_handler = CatchTERM;
143 newsa.sa_mask = sigmask;
144 newsa.sa_flags = 0;
145 sigaction(SIGINT, &newsa, &oldsa);
146
147 sigemptyset(&sigmask);
148 sigaddset(&sigmask, SIGUSR1);
149 newsa.sa_handler = CatchUSR1;
150 newsa.sa_mask = sigmask;
151 newsa.sa_flags = 0;
152 sigaction(SIGUSR1, &newsa, &oldsa);
153
154 sigemptyset(&sigmask);
155 sigaddset(&sigmask, SIGUSR2);
156 newsa.sa_handler = CatchUSR2;
157 newsa.sa_mask = sigmask;
158 newsa.sa_flags = 0;
159 sigaction(SIGUSR2, &newsa, &oldsa);
160
161 return;
162 }
163 //-----------------------------------------------------------------------------
164 int main(int argc, char *argv[])
165 {
166 SETTINGS_IMPL settings;
167
168 if (argc == 2)
169     {
170     settings.SetConfFile(argv[1]);
171     }
172 else
173     {
174     // Usage
175     }
176
177 if (settings.ReadSettings())
178     {
179     printf("ReadSettingsError\n");
180     printf("%s\n", settings.GetStrError().c_str());
181     exit(-1);
182     }
183 settings.Print();
184
185 if (settings.GetDaemon())
186     {
187     switch (fork())
188         {
189         case -1:
190             exit(1);
191             break;
192
193         case 0:
194             setsid();
195             break;
196
197         default:
198             exit(0);
199             break;
200         }
201     }
202
203 clnp = new IA_CLIENT_PROT(settings.GetServerName(), settings.GetServerPort(), settings.GetLocalName(), settings.GetLocalPort());
204
205 if (!settings.GetNoWeb())
206     {
207     web = new WEB();
208     web->SetRefreshPagePeriod(settings.GetRefreshPeriod());
209     web->SetListenAddr(settings.GetListenWebIP());
210     web->Start();
211     }
212
213 clnp->SetLogin(settings.GetLogin());
214 clnp->SetPassword(settings.GetPassword());
215
216 clnp->SetStatusChangedCb(StatusChanged, NULL);
217 clnp->SetInfoCb(ShowMessage, NULL);
218 clnp->SetErrorCb(ShowError, NULL);
219 clnp->SetDirNameCb(SetDirName, NULL);
220 clnp->SetStatChangedCb(StatUpdate, NULL);
221 clnp->SetReconnect(settings.GetReconnect());
222
223 clnp->Start();
224
225 SetSignalHandlers();
226
227 #ifdef LINUX
228 for (int i = 1; i < argc; i++)
229     memset(argv[i], 0, strlen(argv[i]));
230
231 if(argc > 1)
232     strcpy(argv[1], "Connecting...");
233 #endif
234
235 #ifdef FREEBSD
236 setproctitle("Connecting...");
237 #endif
238 clnp->Connect();
239
240 while (1)
241     {
242     struct timespec ts = {0, 200000000};
243     nanosleep(&ts, NULL);
244
245     char state[20];
246
247     if (clnp->GetAuthorized())
248         {
249         if (settings.GetShowPid())
250             sprintf(state, "On %d", getpid());
251         else
252             strcpy(state, "Online");
253         }
254     else
255         {
256         if (settings.GetShowPid())
257             sprintf(state, "Off %d", getpid());
258         else
259             strcpy(state, "Offline");
260         }
261
262     #ifdef LINUX
263     for (int i = 1; i < argc; i++)
264         memset(argv[i], 0, strlen(argv[i]));
265     if(argc > 1)
266         strcpy(argv[1], state);
267     #endif
268
269     #ifdef FREEBSD
270     setproctitle(state);
271     #endif
272
273     #ifdef FREEBSD_5
274     setproctitle(state);
275     #endif
276     }
277
278 return 0;
279 }
280 //-----------------------------------------------------------------------------