src/Sockets/tests/https.cpp

   1 #include <HTTPSocket.h>
   2 #include <SocketHandler.h>
   3 #include <ListenSocket.h>
   4 #include <Exception.h>
   5
   6
   7 class tSocket : public HTTPSocket
   8 {
   9 public:
  10         tSocket(ISocketHandler& h) : HTTPSocket(h)
  11         , m_first(false)
  12         , m_sz(0) {
  13                 EnableSSL();
  14         }
  15
  16         tSocket(ISocketHandler& h, const std::string& host) : HTTPSocket(h), m_host(host)
  17         , m_first(false)
  18         , m_sz(0) {
  19                 EnableSSL();
  20                 Open(host, 4443);
  21         }
  22
  23         void InitSSLServer() {
  24                 InitializeContext("", "comb.pem", "", SSLv23_method());
  25         }
  26
  27         void OnSSLAccept() {
  28                 printf("OnSSLAccept()\n");
  29                 HTTPSocket::OnSSLAccept();
  30         }
  31
  32         void OnSSLConnect() {
  33                 printf("OnSSLConnect()\n");
  34                 HTTPSocket::OnSSLConnect();
  35         }
  36
  37         void OnAccept() {
  38 printf("OnAccept\n");
  39         }
  40
  41         void OnConnect() {
  42 printf("OnConnect\n");
  43                 Send("GET /index.html HTTP/1.0\r\n");
  44                 Send("Host: " + m_host + "\r\n");
  45                 Send("\r\n");
  46         }
  47
  48         void OnFirst() {
  49         }
  50
  51         void OnHeader(const std::string& key, const std::string& value) {
  52                 fprintf(stderr, "%s: %s\n", key.c_str(), value.c_str());
  53         }
  54
  55         void OnHeaderComplete() {
  56                 fprintf(stderr, "\n");
  57         }
  58
  59         void OnData(const char *buf, size_t sz) {
  60                 if (1||!m_first)
  61                 {
  62                         std::string str = buf;
  63                         str.resize( sz );
  64                         printf("%s", str.c_str());
  65                         m_first = true;
  66                 }
  67                 m_sz += sz;
  68         }
  69
  70         void OnDelete() {
  71                 fprintf(stderr, "Content length: %d\n", m_sz);
  72         }
  73
  74 private:
  75         std::string m_host;
  76         bool m_first;
  77         size_t m_sz;
  78 };
  79
  80
  81 int main(int argc, char *argv[])
  82 {
  83         try
  84         {
  85                 SocketHandler h;
  86                 if (argc > 1 && !strcmp(argv[1], "-server"))
  87                 {
  88                         ListenSocket<tSocket> l(h);
  89                         l.Bind(4443);
  90                         h.Add(&l);
  91                         while (h.GetCount())
  92                         {
  93                                 h.Select(1, 0);
  94                         }
  95                 }
  96                 else
  97                 {
  98                         std::string host = argc > 1 ? argv[1] : "www.alhem.net";
  99                         tSocket sock(h, host);
 100                         h.Add(&sock);
 101                         while (h.GetCount())
 102                         {
 103                                 h.Select(1, 0);
 104                         }
 105                 }
 106         }
 107         catch (const Exception& e)
 108         {
 109                 printf("%s\n", e.ToString().c_str());
 110         }
 111 }
 112
 113