src/Sockets/tests/httpd.cpp

   1 #include <HttpdSocket.h>
   2 #include <SocketHandler.h>
   3 #include <ListenSocket.h>
   4 #include <StdoutLog.h>
   5
   6
   7 class sSocket : public HttpdSocket
   8 {
   9 public:
  10         sSocket(ISocketHandler& h) : HttpdSocket(h) {
  11         }
  12
  13         void Init()
  14         {
  15                 if (GetParent() -> GetPort() == 443 || GetParent() -> GetPort() == 8443)
  16                 {
  17 #ifdef HAVE_OPENSSL
  18                         EnableSSL();
  19 #else
  20                         fprintf(stderr, "SSL not available\n");
  21 #endif
  22                 }
  23         }
  24
  25         void Exec()
  26         {
  27                 CreateHeader();
  28                 GenerateDocument();
  29         }
  30
  31         void CreateHeader()
  32         {
  33                 SetStatus("200");
  34                 SetStatusText("OK");
  35 fprintf(stderr, "Uri: '%s'\n", GetUri().c_str());
  36                 {
  37                         size_t x = 0;
  38                         for (size_t i = 0; i < GetUri().size(); i++)
  39                                 if (GetUri()[i] == '.')
  40                                         x = i;
  41                         std::string ext = GetUri().substr(x + 1);
  42                         if (ext == "gif" || ext == "jpg" || ext == "png")
  43                                 AddResponseHeader("Content-type", "image/" + ext);
  44                         else
  45                                 AddResponseHeader("Content-type", "text/" + ext);
  46                 }
  47                 AddResponseHeader("Connection", "close");
  48                 SendResponse();
  49         }
  50
  51         void GenerateDocument()
  52         {
  53                 std::string fn = GetUri().substr(1);
  54                 FILE *fil = fopen(fn.c_str(), "rb");
  55                 if (fil)
  56                 {
  57                         char slask[1000];
  58                         int n = fread(slask,1,1000,fil);
  59                         while (n > 0)
  60                         {
  61                                 SendBuf(slask, n);
  62                                 n = fread(slask,1,1000,fil);
  63                         }
  64                         fclose(fil);
  65                 }
  66                 else
  67                 {
  68                         SetStatus("404");
  69                         SetStatusText("Not Found");
  70                 }
  71                 SetCloseAndDelete();
  72         }
  73
  74 #ifdef HAVE_OPENSSL
  75         void InitSSLServer()
  76         {
  77                 InitializeContext("httpd", "comb.pem", "", SSLv23_method());
  78         }
  79 #endif
  80
  81 };
  82
  83
  84 int main(int argc, char *argv[])
  85 {
  86         std::string host = argc > 1 ? argv[1] : "www.alhem.net";
  87         StdoutLog log;
  88         SocketHandler h(&log);
  89         ListenSocket<sSocket> l(h);
  90         if (l.Bind(1028))
  91         {
  92                 printf("Bind port 1028 failed\n");
  93                 return -1;
  94         }
  95         h.Add(&l);
  96         ListenSocket<sSocket> l2(h);
  97         if (l2.Bind(8443))
  98         {
  99                 printf("Bind port 8443 failed\n");
 100                 return -1;
 101         }
 102         h.Add(&l2);
 103         while (h.GetCount())
 104         {
 105                 h.Select(1, 0);
 106         }
 107 }
 108
 109