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