projects/sgauth/web.h

   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.3 $
  23  $Date: 2007/12/17 08:39:08 $
  24  */
  25
  26 #include "stg/const.h"
  27 #include "stg/ia_packets.h"
  28
  29 #include <string>
  30 #include <list>
  31 #pragma GCC diagnostic push
  32 #pragma GCC diagnostic ignored "-Wshadow"
  33 #include <jthread.hpp>
  34 #pragma GCC diagnostic pop
  35
  36 #ifndef WIN32
  37 #include <sys/types.h>
  38 #include <sys/socket.h>
  39 #include <netinet/in.h>
  40 #include <arpa/inet.h>
  41 #include <unistd.h>
  42 #include <errno.h>
  43 #include <stdlib.h>
  44 #else
  45 #include <winsock2.h>
  46 #endif
  47
  48 #define MAX_MESSAGES    (10)
  49 //-----------------------------------------------------------------------------
  50 struct STG_MESSAGE
  51 {
  52     std::string  msg;
  53     std::string  recvTime;
  54     int     type;
  55 };
  56 //-----------------------------------------------------------------------------
  57 class WEB
  58 {
  59     public:
  60         WEB();
  61         void SetDirName(const std::string & dn, int n);
  62         void SetRefreshPagePeriod(int p);
  63         void SetListenAddr(uint32_t ip);
  64         void AddMessage(const std::string & message, int type);
  65         void UpdateStat(const LOADSTAT & ls);
  66         void Start();
  67     private:
  68         void Run(std::stop_token token) noexcept;
  69         void PrepareNet();
  70         int SendReply();
  71         int SendCSS();
  72         int Redirect(const char * url);
  73
  74         #ifdef WIN32
  75         WSADATA m_wsaData;
  76         #endif
  77
  78         std::jthread m_thread;
  79
  80         std::string m_dirName[DIR_NUM];
  81         int m_res;
  82         int m_listenSocket;
  83         int m_outerSocket;
  84         int m_refreshPeriod;
  85
  86         uint32_t m_listenWebAddr;
  87         LOADSTAT m_ls;
  88
  89         std::list<STG_MESSAGE> m_messages;
  90 };
  91 //-----------------------------------------------------------------------------