lib/cpp/src/transport/THttpClient.h

   1 // Copyright (c) 2006- Facebook
   2 // Distributed under the Thrift Software License
   3 //
   4 // See accompanying file LICENSE or visit the Thrift site at:
   5 // http://developers.facebook.com/thrift/
   6
   7 #ifndef _THRIFT_TRANSPORT_THTTPCLIENT_H_
   8 #define _THRIFT_TRANSPORT_THTTPCLIENT_H_ 1
   9
  10 #include <transport/TTransportUtils.h>
  11
  12 namespace facebook { namespace thrift { namespace transport {
  13
  14 /**
  15  * HTTP client implementation of the thrift transport. This was irritating
  16  * to write, but the alternatives in C++ land are daunting. Linking CURL
  17  * requires 23 dynamic libraries last time I checked (WTF?!?). All we have
  18  * here is a VERY basic HTTP/1.1 client which supports HTTP 100 Continue,
  19  * chunked transfer encoding, keepalive, etc. Tested against Apache.
  20  *
  21  * @author Mark Slee <mcslee@facebook.com>
  22  */
  23 class THttpClient : public TTransport {
  24  public:
  25   THttpClient(boost::shared_ptr<TTransport> transport, std::string host, std::string path="");
  26
  27   THttpClient(std::string host, int port, std::string path="");
  28
  29   virtual ~THttpClient();
  30
  31   void open() {
  32     transport_->open();
  33   }
  34
  35   bool isOpen() {
  36     return transport_->isOpen();
  37   }
  38
  39   bool peek() {
  40     return transport_->peek();
  41   }
  42
  43   void close() {
  44     transport_->close();
  45   }
  46
  47   uint32_t read(uint8_t* buf, uint32_t len);
  48
  49   void readEnd();
  50
  51   void write(const uint8_t* buf, uint32_t len);
  52
  53   void flush();
  54
  55  private:
  56   void init();
  57
  58  protected:
  59
  60   boost::shared_ptr<TTransport> transport_;
  61
  62   TMemoryBuffer writeBuffer_;
  63   TMemoryBuffer readBuffer_;
  64
  65   std::string host_;
  66   std::string path_;
  67
  68   bool readHeaders_;
  69   bool chunked_;
  70   bool chunkedDone_;
  71   uint32_t chunkSize_;
  72   uint32_t contentLength_;
  73
  74   char* httpBuf_;
  75   uint32_t httpPos_;
  76   uint32_t httpBufLen_;
  77   uint32_t httpBufSize_;
  78
  79   uint32_t readMoreData();
  80   char* readLine();
  81
  82   void readHeaders();
  83   void parseHeader(char* header);
  84   bool parseStatusLine(char* status);
  85
  86   uint32_t readChunked();
  87   void readChunkedFooters();
  88   uint32_t parseChunkSize(char* line);
  89
  90   uint32_t readContent(uint32_t size);
  91
  92   void refill();
  93   void shift();
  94
  95 };
  96
  97 }}} // facebook::thrift::transport
  98
  99 #endif // #ifndef _THRIFT_TRANSPORT_THTTPCLIENT_H_