]> git.stg.codes - stg.git/blob - projects/sgauth/web.cpp
Отключено переподключение пользователя при изменении его IP-адреса в
[stg.git] / projects / sgauth / web.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.7 $
23  $Date: 2010/03/15 12:58:17 $
24  */
25
26 #include <stdio.h>
27 #include <string.h>
28 #include <time.h>
29 #include <libintl.h>
30
31 #include "web.h"
32 #include "common.h"
33 #include "ia_auth_c.h"
34
35 extern WEB * web;
36 extern IA_CLIENT_PROT * clnp;
37
38 #define LISTEN_PORT (5580)
39
40 #include "css.h"
41
42 //---------------------------------------------------------------------------
43 #ifndef WIN32
44 void * RunWeb(void *)
45 #else
46 unsigned long WINAPI RunWeb(void *)
47 #endif
48 {
49 while (1)
50     web->Run();
51 return NULL;
52 }
53 //---------------------------------------------------------------------------
54 WEB::WEB()
55     : res(0),
56       listenSocket(0),
57       outerSocket(0),
58       refreshPeriod(0),
59       listenWebAddr(0),
60       outerAddrLen(0)
61 {
62 #ifdef WIN32
63 res = WSAStartup(MAKEWORD(2,0), &wsaData);
64 #endif
65
66 for (int i = 0; i < DIR_NUM; i++)
67     dirName[i] = "-";
68
69 refreshPeriod = 5;
70 }
71 //---------------------------------------------------------------------------
72 void WEB::Start()
73 {
74 #ifdef WIN32
75 unsigned long pt;
76 CreateThread(
77     NULL,   // pointer to thread security attributes
78     16384,  // initial thread stack size, in bytes
79     RunWeb, // pointer to thread function
80     NULL,   // argument for new thread
81     0,      // CREATE_SUSPENDED, // creation flags
82     &pt     // pointer to returned thread identifier
83    );
84 #else
85 pthread_create(&thread, NULL, RunWeb, NULL);
86 #endif
87 }
88 //---------------------------------------------------------------------------
89 void WEB::PrepareNet()
90 {
91 listenSocket = socket(PF_INET, SOCK_STREAM, 0);
92 listenAddr.sin_family = AF_INET;
93 listenAddr.sin_port = htons(LISTEN_PORT);
94 listenAddr.sin_addr.s_addr = listenWebAddr;
95
96 int lng = 1;
97
98 #ifndef WIN32
99 if (0 != setsockopt(listenSocket, SOL_SOCKET, SO_REUSEADDR, &lng, 4))
100     {
101     printf("Setsockopt Fail\n");
102     printf(">>> Error %s\n", strerror(errno));
103     }
104 #else
105 //??? TODO
106 #endif
107
108
109 res = bind(listenSocket, (struct sockaddr*)&listenAddr, sizeof(listenAddr));
110
111 if (res == -1)
112     {
113     printf("Bind failed.\n");
114     exit(0);
115     }
116
117 res = listen(listenSocket, 0);
118 if (res == -1)
119     {
120     printf("Listen failed.\n");
121     exit(0);
122     }
123
124 outerAddrLen = sizeof(outerAddr);
125 }
126 //---------------------------------------------------------------------------
127 void WEB::SetRefreshPagePeriod(int p)
128 {
129 refreshPeriod = p;
130 if (refreshPeriod <= 0 || refreshPeriod > 24*3600)
131     refreshPeriod = 5;
132 }
133 //---------------------------------------------------------------------------
134 void WEB::SetListenAddr(uint32_t ip)
135 {
136 listenWebAddr = ip;
137 }
138 //---------------------------------------------------------------------------
139 void WEB::Run()
140 {
141 PrepareNet();
142 char recvBuffer[4096];
143 while (1)
144     {
145     outerSocket = accept(listenSocket, (struct sockaddr*)&outerAddr, &outerAddrLen);
146     if (outerSocket == -1)
147         {
148         printf(">>> Error %s\n", strerror(errno));
149         continue;
150         }
151     recv(outerSocket, recvBuffer, sizeof(recvBuffer), 0);
152
153     if (strncmp(recvBuffer, "GET /sgauth.css", strlen("GET /sgauth.css")) == 0)
154         {
155         SendCSS();
156         //printf("(1) recvBuffer=%s\n", recvBuffer);
157         }
158     else if (strncmp(recvBuffer, "GET /disconnect", strlen("GET /disconnect")) == 0)
159         {
160         clnp->Disconnect();
161         Redirect("/");
162         //printf("(2) recvBuffer=%s\n", recvBuffer);
163         }
164     else if (strncmp(recvBuffer, "GET /connect", strlen("GET /connect")) == 0)
165         {
166         clnp->Connect();
167         Redirect("/");
168         //printf("(3) recvBuffer=%s\n", recvBuffer);
169         }
170     else if (strncmp(recvBuffer, "GET /exit", strlen("GET /exit")) == 0)
171         {
172         Redirect("/");
173         clnp->Disconnect();
174         #ifdef WIN32
175         Sleep(1000);
176         #else
177         usleep(1000000);
178         #endif
179         exit(0);
180         }
181     else
182        {
183        SendReply();
184        //printf("(4) recvBuffer=%s\n", recvBuffer);
185        }
186
187     #ifdef WIN32
188     closesocket(outerSocket);
189     #else
190     close(outerSocket);
191     #endif
192     }
193 }
194 //---------------------------------------------------------------------------
195 int WEB::Redirect(const char * url)
196 {
197 const char * redirect =
198     "HTTP/1.0 200 OK\n"
199     "Content-Type: text/html\n"
200     "Connection: close"
201     "\n\n"
202     "<html>\n"
203     "<head>\n"
204     "<META HTTP-EQUIV=\"Refresh\" CONTENT=\"0;%s\">\n"
205     "</head>\n"
206     "<body>\n"
207     "</body></html>\n\n";
208
209 char buff[2000];
210 sprintf(buff, redirect, url);
211 send(outerSocket, buff, strlen(buff), 0);
212
213 return 0;
214 }
215 //---------------------------------------------------------------------------
216 int WEB::SendReply()
217 {
218 int j, rowNum;
219
220 const char * replyHeader =
221     "HTTP/1.0 200 OK\n"
222     "Content-Type: text/html\n"
223     "Connection: close"
224     "\n\n"
225     "<html>\n"
226     "<head>\n"
227     "<META HTTP-EQUIV=\"Refresh\" CONTENT=\"%d\">\n"
228     "<meta http-equiv=\"Content-Type\" content=\"text/html; charset=windows-1251\">\n"
229     "<title>sgauth</title>\n"
230     "<link rel=\"Stylesheet\" href=\"sgauth.css\">"
231     "</head>\n"
232     "<body>\n"
233     "<H3>Stargazer</H3><p>\n";
234
235 const char * replyFooter = "</body></html>\n\n";
236
237 char replyHeaderBuffer[2000];
238 sprintf(replyHeaderBuffer, replyHeader, refreshPeriod);
239
240 send(outerSocket, replyHeaderBuffer, strlen(replyHeaderBuffer), 0);
241
242 char str[512];
243
244 int st = clnp->GetAuthorized();
245
246 sprintf(str, "<a href=\"connect\">%s</a><p>\n", gettext("Connect"));
247 res = send(outerSocket, str, strlen(str), 0);
248
249 sprintf(str, "<a href=\"disconnect\">%s</a><p>\n", gettext("Disconnect"));
250 res = send(outerSocket, str, strlen(str), 0);
251
252 sprintf(str, "<a href=\"/\">%s</a><p>\n", gettext("Refresh"));
253 res = send(outerSocket, str, strlen(str), 0);
254
255 sprintf(str, "<a href=\"exit\">%s</a><p>\n", gettext("Exit"));
256 res = send(outerSocket, str, strlen(str), 0);
257
258 sprintf(str, "<div id=\"%s\">%s</div><p>\n" , st ? "ConnectionStateOnline":"ConnectionStateOffline", st ? "Online":"Offline");
259 res = send(outerSocket, str, strlen(str), 0);
260
261 sprintf(str, "<div id=\"Cash\">%s: %.3f</div><p>\n" , gettext("Cash"), ls.cash / 1000.0);
262 res = send(outerSocket, str, strlen(str), 0);
263
264 sprintf(str, "<div id=\"Prepaid Traffic\">%s: %s</div><p>\n" ,
265         gettext("PrepaidTraffic"),
266         ls.freeMb[0] == 'C' ? ls.freeMb + 1 : ls.freeMb);
267 res = send(outerSocket, str, strlen(str), 0);
268
269 sprintf(str, "<TABLE id=\"TraffTable\">\n");
270 res = send(outerSocket, str, strlen(str), 0);
271 sprintf(str, "    <TR id=\"TraffTableCaptionRow\">\n");
272 res = send(outerSocket, str, strlen(str), 0);
273 sprintf(str, "       <TD id=\"TraffTableCaptionCellC\">&nbsp;</TD>\n");
274 res = send(outerSocket, str, strlen(str), 0);
275
276 rowNum = 0;
277 for (j = 0; j < DIR_NUM; j++)
278     {
279     if (dirName[j][0] == 0)
280         continue;
281     string s;
282     KOIToWin(dirName[j], &s);// +++++++++ sigsegv ==========   TODO too long dir name crashes sgauth
283     sprintf(str, "       <TD id=\"TraffTableCaptionCell%d\">%s</TD>\n", rowNum++, s.c_str());
284     send(outerSocket, str, strlen(str), 0);
285     }
286
287 sprintf(str,"    </TR>\n");
288 send(outerSocket, str, strlen(str), 0);
289
290 sprintf(str,"    <TR id=\"TraffTableUMRow\">\n");
291 send(outerSocket, str, strlen(str), 0);
292
293 sprintf(str,"        <TD id=\"TraffTableUMCellC\">%s</TD>\n", gettext("Month Upload"));
294 send(outerSocket, str, strlen(str), 0);
295
296 rowNum = 0;
297 for (j = 0; j < DIR_NUM; j++)
298     {
299     if (dirName[j][0] == 0)
300         continue;
301     sprintf(str,"        <TD id=\"TraffTableUMCell%d\">%s</TD>\n", rowNum++, IntToKMG(ls.mu[j], ST_F));
302     res = send(outerSocket, str, strlen(str), 0);
303     }
304
305 sprintf(str,"    </TR>\n");
306 res = send(outerSocket, str, strlen(str), 0);
307 sprintf(str,"    <TR id=\"TraffTableDMRow\">\n");
308 res = send(outerSocket, str, strlen(str), 0);
309 sprintf(str,"        <TD id=\"TraffTableDMCellC\">%s</TD>\n", gettext("Month Download"));
310 res = send(outerSocket, str, strlen(str), 0);
311
312 rowNum = 0;
313 for (j = 0; j < DIR_NUM; j++)
314     {
315     if (dirName[j][0] == 0)
316         continue;
317     sprintf(str,"        <TD id=\"TraffTableDMCell%d\">%s</TD>\n", rowNum++, IntToKMG(ls.md[j], ST_F));
318     res = send(outerSocket, str, strlen(str), 0);
319     }
320 sprintf(str,"    </TR>\n");
321 res = send(outerSocket, str, strlen(str), 0);
322
323
324 sprintf(str,"    <TR id=\"TraffTableUSRow\">\n");
325 res = send(outerSocket, str, strlen(str), 0);
326 sprintf(str,"        <TD id=\"TraffTableUSCellC\">%s</TD>\n", gettext("Session Upload"));
327 res = send(outerSocket, str, strlen(str), 0);
328
329 rowNum = 0;
330 for (j = 0; j < DIR_NUM; j++)
331     {
332     if (dirName[j][0] == 0)
333         continue;
334     sprintf(str,"        <TD id=\"TraffTableUSCell%d\">%s</TD>\n", rowNum++, IntToKMG(ls.su[j], ST_F));
335     res = send(outerSocket, str, strlen(str), 0);
336     }
337
338 sprintf(str,"    </TR>\n");
339 res = send(outerSocket, str, strlen(str), 0);
340 sprintf(str,"    <TR id=\"TraffTableDSRow\">\n");
341 res = send(outerSocket, str, strlen(str), 0);
342 sprintf(str,"        <TD id=\"TraffTableDSCellC\">%s</TD>\n", gettext("Session Download"));
343 res = send(outerSocket, str, strlen(str), 0);
344
345 rowNum = 0;
346 for (j = 0; j < DIR_NUM; j++)
347     {
348     if (dirName[j][0] == 0)
349         continue;
350     sprintf(str,"        <TD id=\"TraffTableDSCell%d\">%s</TD>\n", j, IntToKMG(ls.sd[j], ST_F));
351     res = send(outerSocket, str, strlen(str), 0);
352     }
353
354 sprintf(str,"    </TR>\n");
355 res = send(outerSocket, str, strlen(str), 0);
356
357 sprintf(str,"</TABLE>\n");
358 res = send(outerSocket, str, strlen(str), 0);
359
360 rowNum = 0;
361 if (!messages.empty())
362     {
363     sprintf(str,"    <TABLE id=\"MessagesTable\">\n");
364     res = send(outerSocket, str, strlen(str), 0);
365
366     sprintf(str,"        <TR id=\"MessagesTableRowC\">\n");
367     send(outerSocket, str, strlen(str), 0);
368     sprintf(str,"            <TD>Date</TD>\n");
369     send(outerSocket, str, strlen(str), 0);
370     sprintf(str,"            <TD>Text</TD>\n");
371     send(outerSocket, str, strlen(str), 0);
372     sprintf(str,"        </TR>\n");
373     send(outerSocket, str, strlen(str), 0);
374
375     list<STG_MESSAGE>::reverse_iterator it;
376     it = messages.rbegin();
377     while (it != messages.rend())
378         {
379         sprintf(str,"        <TR id=\"MessagesTableRow%d\">\n", rowNum);
380         send(outerSocket, str, strlen(str), 0);
381         sprintf(str,"            <TD>%s</TD>\n", it->recvTime.c_str());
382         send(outerSocket, str, strlen(str), 0);
383         sprintf(str,"            <TD>%s</TD>\n", it->msg.c_str());
384         send(outerSocket, str, strlen(str), 0);
385         sprintf(str,"        </TR>\n");
386         send(outerSocket, str, strlen(str), 0);
387         ++it;
388         ++rowNum;
389         }
390
391     sprintf(str,"   </TABLE>\n");
392     res = send(outerSocket, str, strlen(str), 0);
393     }
394
395 time_t t = time(NULL);
396 sprintf(str,"Îáíîâëåíî: %s</b>" , ctime(&t));
397 res = send(outerSocket, str, strlen(str), 0);
398
399 send(outerSocket, replyFooter, strlen(replyFooter), 0);
400
401 return 0;
402 }
403 //---------------------------------------------------------------------------
404 int WEB::SendCSS()
405 {
406 const char * replyHeader =
407     "HTTP/1.0 200 OK\n"
408     "Content-Type: text/css\n"
409     "Connection: close\n\n";
410
411 const char * replyFooter= "\n\n";
412
413 send(outerSocket, replyHeader, strlen(replyHeader), 0);
414 send(outerSocket, css, strlen(css), 0);
415 send(outerSocket, replyFooter, strlen(replyFooter), 0);
416
417 return 0;
418 }
419 //---------------------------------------------------------------------------
420 void WEB::SetDirName(const string & dn, int n)
421 {
422 web->dirName[n] =  dn;
423 }
424 //---------------------------------------------------------------------------
425 void WEB::AddMessage(const string & message, int type)
426 {
427 time_t t = time(NULL);
428 STG_MESSAGE m;
429
430 m.msg = message;
431 m.type = type;
432 m.recvTime = ctime(&t);
433
434 messages.push_back(m);
435
436 if (messages.size() > MAX_MESSAGES)
437     messages.pop_front();
438
439 }
440 //---------------------------------------------------------------------------
441 void WEB::UpdateStat(const LOADSTAT & ls)
442 {
443 memcpy((void*)&(WEB::ls), &ls, sizeof(LOADSTAT));
444 }
445 //---------------------------------------------------------------------------
446