xref: /openbmc/bmcweb/http/http_request.hpp (revision 6fb96ce33d221a4cfa20d156eff5a3166ecb2661)
1 #pragma once
2 
3 #include "common.hpp"
4 #include "sessions.hpp"
5 
6 #include <boost/asio/io_context.hpp>
7 #include <boost/asio/ip/address.hpp>
8 #include <boost/beast/http/message.hpp>
9 #include <boost/beast/http/string_body.hpp>
10 #include <boost/beast/websocket.hpp>
11 #include <boost/url/url.hpp>
12 
13 #include <string>
14 #include <string_view>
15 #include <system_error>
16 
17 namespace crow
18 {
19 
20 struct Request
21 {
22     boost::beast::http::request<boost::beast::http::string_body> req;
23 
24   private:
25     boost::urls::url urlBase{};
26 
27   public:
28     bool isSecure{false};
29 
30     boost::asio::io_context* ioService{};
31     boost::asio::ip::address ipAddress{};
32 
33     std::shared_ptr<persistent_data::UserSession> session;
34 
35     std::string userRole{};
36     Request(boost::beast::http::request<boost::beast::http::string_body> reqIn,
37             std::error_code& ec) :
38         req(std::move(reqIn))
39     {
40         if (!setUrlInfo())
41         {
42             ec = std::make_error_code(std::errc::invalid_argument);
43         }
44     }
45 
46     Request(std::string_view bodyIn, std::error_code& /*ec*/) : req({}, bodyIn)
47     {}
48 
49     Request() = default;
50 
51     Request(const Request& other) = default;
52     Request(Request&& other) = default;
53 
54     Request& operator=(const Request&) = default;
55     Request& operator=(Request&&) = default;
56     ~Request() = default;
57 
58     void addHeader(std::string_view key, std::string_view value)
59     {
60         req.insert(key, value);
61     }
62 
63     void addHeader(boost::beast::http::field key, std::string_view value)
64     {
65         req.insert(key, value);
66     }
67 
68     boost::beast::http::verb method() const
69     {
70         return req.method();
71     }
72 
73     std::string_view getHeaderValue(std::string_view key) const
74     {
75         return req[key];
76     }
77 
78     std::string_view getHeaderValue(boost::beast::http::field key) const
79     {
80         return req[key];
81     }
82 
83     std::string_view methodString() const
84     {
85         return req.method_string();
86     }
87 
88     std::string_view target() const
89     {
90         return req.target();
91     }
92 
93     boost::urls::url_view url() const
94     {
95         return {urlBase};
96     }
97 
98     const boost::beast::http::fields& fields() const
99     {
100         return req.base();
101     }
102 
103     const std::string& body() const
104     {
105         return req.body();
106     }
107 
108     bool target(std::string_view target)
109     {
110         req.target(target);
111         return setUrlInfo();
112     }
113 
114     unsigned version() const
115     {
116         return req.version();
117     }
118 
119     bool isUpgrade() const
120     {
121         return boost::beast::websocket::is_upgrade(req);
122     }
123 
124     bool keepAlive() const
125     {
126         return req.keep_alive();
127     }
128 
129   private:
130     bool setUrlInfo()
131     {
132         auto result = boost::urls::parse_relative_ref(target());
133 
134         if (!result)
135         {
136             return false;
137         }
138         urlBase = *result;
139         return true;
140     }
141 };
142 
143 } // namespace crow
144