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