request.hpp 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  1. /*
  2. * Copyright (c) 2014, Peter Thorson. All rights reserved.
  3. *
  4. * Redistribution and use in source and binary forms, with or without
  5. * modification, are permitted provided that the following conditions are met:
  6. * * Redistributions of source code must retain the above copyright
  7. * notice, this list of conditions and the following disclaimer.
  8. * * Redistributions in binary form must reproduce the above copyright
  9. * notice, this list of conditions and the following disclaimer in the
  10. * documentation and/or other materials provided with the distribution.
  11. * * Neither the name of the WebSocket++ Project nor the
  12. * names of its contributors may be used to endorse or promote products
  13. * derived from this software without specific prior written permission.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  16. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18. * ARE DISCLAIMED. IN NO EVENT SHALL PETER THORSON BE LIABLE FOR ANY
  19. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  20. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  21. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  22. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  24. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25. *
  26. */
  27. #ifndef HTTP_PARSER_REQUEST_HPP
  28. #define HTTP_PARSER_REQUEST_HPP
  29. #include <string>
  30. #include <websocketpp/common/memory.hpp>
  31. #include <websocketpp/http/parser.hpp>
  32. namespace websocketpp {
  33. namespace http {
  34. namespace parser {
  35. /// Stores, parses, and manipulates HTTP requests
  36. /**
  37. * http::request provides the following functionality for working with HTTP
  38. * requests.
  39. *
  40. * - Initialize request via manually setting each element
  41. * - Initialize request via reading raw bytes and parsing
  42. * - Once initialized, access individual parsed elements
  43. * - Once initialized, read entire request as raw bytes
  44. */
  45. class request : public parser {
  46. public:
  47. typedef request type;
  48. typedef lib::shared_ptr<type> ptr;
  49. request()
  50. : m_buf(lib::make_shared<std::string>())
  51. , m_ready(false) {}
  52. /// Process bytes in the input buffer
  53. /**
  54. * Process up to len bytes from input buffer buf. Returns the number of
  55. * bytes processed. Bytes left unprocessed means bytes left over after the
  56. * final header delimiters.
  57. *
  58. * Consume is a streaming processor. It may be called multiple times on one
  59. * request and the full headers need not be available before processing can
  60. * begin. If the end of the request was reached during this call to consume
  61. * the ready flag will be set. Further calls to consume once ready will be
  62. * ignored.
  63. *
  64. * Consume will throw an http::exception in the case of an error. Typical
  65. * error reasons include malformed requests, incomplete requests, and max
  66. * header size being reached.
  67. *
  68. * @param buf Pointer to byte buffer
  69. * @param len Size of byte buffer
  70. * @return Number of bytes processed.
  71. */
  72. size_t consume(char const * buf, size_t len);
  73. /// Returns whether or not the request is ready for reading.
  74. bool ready() const {
  75. return m_ready;
  76. }
  77. /// Returns the full raw request (including the body)
  78. std::string raw() const;
  79. /// Returns the raw request headers only (similar to an HTTP HEAD request)
  80. std::string raw_head() const;
  81. /// Set the HTTP method. Must be a valid HTTP token
  82. void set_method(std::string const & method);
  83. /// Return the request method
  84. std::string const & get_method() const {
  85. return m_method;
  86. }
  87. /// Set the HTTP uri. Must be a valid HTTP uri
  88. void set_uri(std::string const & uri);
  89. /// Return the requested URI
  90. std::string const & get_uri() const {
  91. return m_uri;
  92. }
  93. private:
  94. /// Helper function for message::consume. Process request line
  95. void process(std::string::iterator begin, std::string::iterator end);
  96. lib::shared_ptr<std::string> m_buf;
  97. std::string m_method;
  98. std::string m_uri;
  99. bool m_ready;
  100. };
  101. } // namespace parser
  102. } // namespace http
  103. } // namespace websocketpp
  104. #include <websocketpp/http/impl/request.hpp>
  105. #endif // HTTP_PARSER_REQUEST_HPP