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