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