aboutsummaryrefslogtreecommitdiffstats
path: root/websocketpp/http/impl/request.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'websocketpp/http/impl/request.hpp')
-rw-r--r--websocketpp/http/impl/request.hpp191
1 files changed, 191 insertions, 0 deletions
diff --git a/websocketpp/http/impl/request.hpp b/websocketpp/http/impl/request.hpp
new file mode 100644
index 00000000..311a620f
--- /dev/null
+++ b/websocketpp/http/impl/request.hpp
@@ -0,0 +1,191 @@
+/*
+ * Copyright (c) 2014, Peter Thorson. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ * * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * * Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * * Neither the name of the WebSocket++ Project nor the
+ * names of its contributors may be used to endorse or promote products
+ * derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL PETER THORSON BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ */
+
+#ifndef HTTP_PARSER_REQUEST_IMPL_HPP
+#define HTTP_PARSER_REQUEST_IMPL_HPP
+
+#include <algorithm>
+#include <sstream>
+#include <string>
+
+#include <websocketpp/http/parser.hpp>
+
+namespace websocketpp {
+namespace http {
+namespace parser {
+
+inline size_t request::consume(char const * buf, size_t len) {
+ size_t bytes_processed;
+
+ if (m_ready) {return 0;}
+
+ if (m_body_bytes_needed > 0) {
+ bytes_processed = process_body(buf,len);
+ if (body_ready()) {
+ m_ready = true;
+ }
+ return bytes_processed;
+ }
+
+ // copy new header bytes into buffer
+ m_buf->append(buf,len);
+
+ // Search for delimiter in buf. If found read until then. If not read all
+ std::string::iterator begin = m_buf->begin();
+ std::string::iterator end;
+
+ for (;;) {
+ // search for line delimiter
+ end = std::search(
+ begin,
+ m_buf->end(),
+ header_delimiter,
+ header_delimiter+sizeof(header_delimiter)-1
+ );
+
+ m_header_bytes += (end-begin+sizeof(header_delimiter));
+
+ if (m_header_bytes > max_header_size) {
+ // exceeded max header size
+ throw exception("Maximum header size exceeded.",
+ status_code::request_header_fields_too_large);
+ }
+
+ if (end == m_buf->end()) {
+ // we are out of bytes. Discard the processed bytes and copy the
+ // remaining unprecessed bytes to the beginning of the buffer
+ std::copy(begin,end,m_buf->begin());
+ m_buf->resize(static_cast<std::string::size_type>(end-begin));
+ m_header_bytes -= m_buf->size();
+
+ return len;
+ }
+
+ //the range [begin,end) now represents a line to be processed.
+ if (end-begin == 0) {
+ // we got a blank line
+ if (m_method.empty() || get_header("Host").empty()) {
+ throw exception("Incomplete Request",status_code::bad_request);
+ }
+
+ bytes_processed = (
+ len - static_cast<std::string::size_type>(m_buf->end()-end)
+ + sizeof(header_delimiter) - 1
+ );
+
+ // frees memory used temporarily during request parsing
+ m_buf.reset();
+
+ // if this was not an upgrade request and has a content length
+ // continue capturing content-length bytes and expose them as a
+ // request body.
+
+ if (prepare_body()) {
+ bytes_processed += process_body(buf+bytes_processed,len-bytes_processed);
+ if (body_ready()) {
+ m_ready = true;
+ }
+ return bytes_processed;
+ } else {
+ m_ready = true;
+
+ // return number of bytes processed (starting bytes - bytes left)
+ return bytes_processed;
+ }
+ } else {
+ if (m_method.empty()) {
+ this->process(begin,end);
+ } else {
+ this->process_header(begin,end);
+ }
+ }
+
+ begin = end+(sizeof(header_delimiter)-1);
+ }
+}
+
+inline std::string request::raw() const {
+ // TODO: validation. Make sure all required fields have been set?
+ std::stringstream ret;
+
+ ret << m_method << " " << m_uri << " " << get_version() << "\r\n";
+ ret << raw_headers() << "\r\n" << m_body;
+
+ return ret.str();
+}
+
+inline std::string request::raw_head() const {
+ // TODO: validation. Make sure all required fields have been set?
+ std::stringstream ret;
+
+ ret << m_method << " " << m_uri << " " << get_version() << "\r\n";
+ ret << raw_headers() << "\r\n";
+
+ return ret.str();
+}
+
+inline void request::set_method(std::string const & method) {
+ if (std::find_if(method.begin(),method.end(),is_not_token_char) != method.end()) {
+ throw exception("Invalid method token.",status_code::bad_request);
+ }
+
+ m_method = method;
+}
+
+inline void request::set_uri(std::string const & uri) {
+ // TODO: validation?
+ m_uri = uri;
+}
+
+inline void request::process(std::string::iterator begin, std::string::iterator
+ end)
+{
+ std::string::iterator cursor_start = begin;
+ std::string::iterator cursor_end = std::find(begin,end,' ');
+
+ if (cursor_end == end) {
+ throw exception("Invalid request line1",status_code::bad_request);
+ }
+
+ set_method(std::string(cursor_start,cursor_end));
+
+ cursor_start = cursor_end+1;
+ cursor_end = std::find(cursor_start,end,' ');
+
+ if (cursor_end == end) {
+ throw exception("Invalid request line2",status_code::bad_request);
+ }
+
+ set_uri(std::string(cursor_start,cursor_end));
+ set_version(std::string(cursor_end+1,end));
+}
+
+} // namespace parser
+} // namespace http
+} // namespace websocketpp
+
+#endif // HTTP_PARSER_REQUEST_IMPL_HPP