src/Sockets/tests/http.cpp

   1 #include <HTTPSocket.h>
   2 #include <SocketHandler.h>
   3
   4
   5 class tSocket : public HTTPSocket
   6 {
   7 public:
   8         tSocket(ISocketHandler& h, const std::string& host) : HTTPSocket(h), m_host(host)
   9         , m_first(false)
  10         , m_sz(0) {
  11                 Open(host, 80);
  12         }
  13
  14         void OnConnect() {
  15                 Send("GET /index.html HTTP/1.0\r\n");
  16                 Send("Host: " + m_host + "\r\n");
  17                 Send("\r\n");
  18         }
  19
  20         void OnFirst() {
  21         }
  22
  23         void OnHeader(const std::string& key, const std::string& value) {
  24                 fprintf(stderr, "%s: %s\n", key.c_str(), value.c_str());
  25         }
  26
  27         void OnHeaderComplete() {
  28                 fprintf(stderr, "\n");
  29         }
  30
  31         void OnData(const char *buf, size_t sz) {
  32                 if (1||!m_first)
  33                 {
  34                         std::string str = buf;
  35                         str.resize( sz );
  36                         printf("%s", str.c_str());
  37                         m_first = true;
  38                 }
  39                 m_sz += sz;
  40         }
  41
  42         void OnDelete() {
  43                 fprintf(stderr, "Content length: %d\n", m_sz);
  44         }
  45
  46 private:
  47         std::string m_host;
  48         bool m_first;
  49         size_t m_sz;
  50 };
  51
  52
  53 int main(int argc, char *argv[])
  54 {
  55         std::string host = argc > 1 ? argv[1] : "www.alhem.net";
  56         SocketHandler h;
  57         tSocket sock(h, host);
  58         h.Add(&sock);
  59         while (h.GetCount())
  60         {
  61                 h.Select(1, 0);
  62         }
  63 }
  64
  65