src/asio_server_response_impl.h

   1 /*
   2  * nghttp2 - HTTP/2 C Library
   3  *
   4  * Copyright (c) 2015 Tatsuhiro Tsujikawa
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining
   7  * a copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sublicense, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice shall be
  15  * included in all copies or substantial portions of the Software.
  16  *
  17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  18  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  19  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  20  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  21  * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  22  * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  23  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  24  */
  25 #ifndef ASIO_SERVER_RESPONSE_IMPL_H
  26 #define ASIO_SERVER_RESPONSE_IMPL_H
  27
  28 #include "nghttp2_config.h"
  29
  30 #include <nghttp2/asio_http2_server.h>
  31
  32 namespace nghttp2 {
  33 namespace asio_http2 {
  34 namespace server {
  35
  36 class stream;
  37
  38 enum class response_state {
  39   INITIAL,
  40   // response_impl::write_head() was called
  41   HEADER_DONE,
  42   // response_impl::end() was called
  43   BODY_STARTED,
  44 };
  45
  46 class response_impl {
  47 public:
  48   response_impl();
  49   void write_head(unsigned int status_code, header_map h = header_map{});
  50   void end(std::string data = "");
  51   void end(generator_cb cb);
  52   void write_trailer(header_map h);
  53   void on_close(close_cb cb);
  54   void resume();
  55
  56   void cancel(uint32_t error_code);
  57
  58   response *push(boost::system::error_code &ec, std::string method,
  59                  std::string raw_path_query, header_map) const;
  60
  61   boost::asio::io_service &io_service();
  62
  63   void start_response();
  64
  65   unsigned int status_code() const;
  66   const header_map &header() const;
  67   void pushed(bool f);
  68   void push_promise_sent();
  69   void stream(class stream *s);
  70   generator_cb::result_type call_read(uint8_t *data, std::size_t len,
  71                                       uint32_t *data_flags);
  72   void call_on_close(uint32_t error_code);
  73
  74 private:
  75   class stream *strm_;
  76   header_map header_;
  77   generator_cb generator_cb_;
  78   close_cb close_cb_;
  79   unsigned int status_code_;
  80   response_state state_;
  81   // true if this is pushed stream's response
  82   bool pushed_;
  83   // true if PUSH_PROMISE is sent if this is response of a pushed
  84   // stream
  85   bool push_promise_sent_;
  86 };
  87
  88 } // namespace server
  89 } // namespace asio_http2
  90 } // namespace nghttp2
  91
  92 #endif // ASIO_SERVER_RESPONSE_IMPL_H