1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
/*
* Copyright (C) 2011-2018 Team Kodi
* This file is part of Kodi - https://kodi.tv
*
* SPDX-License-Identifier: GPL-2.0-or-later
* See LICENSES/README.md for more information.
*/
#include "IHTTPRequestHandler.h"
#include "network/WebServer.h"
#include "network/httprequesthandler/HTTPRequestHandlerUtils.h"
#include "utils/StringUtils.h"
#include <limits>
#include <utility>
static const std::string HTTPMethodHead = "HEAD";
static const std::string HTTPMethodGet = "GET";
static const std::string HTTPMethodPost = "POST";
HTTPMethod GetHTTPMethod(const char *method)
{
if (HTTPMethodGet.compare(method) == 0)
return GET;
if (HTTPMethodPost.compare(method) == 0)
return POST;
if (HTTPMethodHead.compare(method) == 0)
return HEAD;
return UNKNOWN;
}
std::string GetHTTPMethod(HTTPMethod method)
{
switch (method)
{
case HEAD:
return HTTPMethodHead;
case GET:
return HTTPMethodGet;
case POST:
return HTTPMethodPost;
case UNKNOWN:
break;
}
return "";
}
IHTTPRequestHandler::IHTTPRequestHandler()
: m_request(),
m_response(),
m_postFields()
{ }
IHTTPRequestHandler::IHTTPRequestHandler(const HTTPRequest &request)
: m_request(request),
m_response(),
m_postFields()
{
m_response.type = HTTPError;
m_response.status = MHD_HTTP_INTERNAL_SERVER_ERROR;
m_response.totalLength = 0;
}
bool IHTTPRequestHandler::HasResponseHeader(const std::string &field) const
{
if (field.empty())
return false;
return m_response.headers.find(field) != m_response.headers.end();
}
bool IHTTPRequestHandler::AddResponseHeader(const std::string &field, const std::string &value, bool allowMultiple /* = false */)
{
if (field.empty() || value.empty())
return false;
if (!allowMultiple && HasResponseHeader(field))
return false;
m_response.headers.insert(std::make_pair(field, value));
return true;
}
void IHTTPRequestHandler::AddPostField(const std::string &key, const std::string &value)
{
if (key.empty())
return;
std::map<std::string, std::string>::iterator field = m_postFields.find(key);
if (field == m_postFields.end())
m_postFields[key] = value;
else
m_postFields[key].append(value);
}
bool IHTTPRequestHandler::AddPostData(const char *data, size_t size)
{
if (size > 0)
return appendPostData(data, size);
return true;
}
bool IHTTPRequestHandler::GetRequestedRanges(uint64_t totalLength)
{
if (!m_ranged || m_request.webserver == NULL || m_request.connection == NULL)
return false;
m_request.ranges.Clear();
if (totalLength == 0)
return true;
return HTTPRequestHandlerUtils::GetRequestedRanges(m_request.connection, totalLength, m_request.ranges);
}
bool IHTTPRequestHandler::GetHostnameAndPort(std::string& hostname, uint16_t &port)
{
if (m_request.webserver == NULL || m_request.connection == NULL)
return false;
std::string hostnameAndPort = HTTPRequestHandlerUtils::GetRequestHeaderValue(m_request.connection, MHD_HEADER_KIND, MHD_HTTP_HEADER_HOST);
if (hostnameAndPort.empty())
return false;
size_t pos = hostnameAndPort.find(':');
hostname = hostnameAndPort.substr(0, pos);
if (hostname.empty())
return false;
if (pos != std::string::npos)
{
std::string strPort = hostnameAndPort.substr(pos + 1);
if (!StringUtils::IsNaturalNumber(strPort))
return false;
unsigned long portL = strtoul(strPort.c_str(), NULL, 0);
if (portL > std::numeric_limits<uint16_t>::max())
return false;
port = static_cast<uint16_t>(portL);
}
else
port = 80;
return true;
}
|
