354 lines
7.3 KiB
C++
354 lines
7.3 KiB
C++
#include "webcc/request_parser.h"
|
|
|
|
#include <vector>
|
|
|
|
#include "boost/algorithm/string.hpp"
|
|
|
|
#include "webcc/logger.h"
|
|
#include "webcc/request.h"
|
|
#include "webcc/string.h"
|
|
#include "webcc/utility.h"
|
|
|
|
namespace webcc
|
|
{
|
|
|
|
RequestParser::RequestParser() : request_(nullptr)
|
|
{
|
|
}
|
|
|
|
void RequestParser::Init(Request* request, ViewMatcher view_matcher)
|
|
{
|
|
assert(view_matcher);
|
|
|
|
Parser::Init(request);
|
|
|
|
request_ = request;
|
|
view_matcher_ = view_matcher;
|
|
header_parsed_ = false; // 【归位】：在 Init 时重置状态，防止对象复用残留
|
|
}
|
|
|
|
bool RequestParser::OnHeadersEnd()
|
|
{
|
|
header_parsed_ = true; // 【核心位置】：请求头解析完毕，触发标记！
|
|
|
|
bool matched = view_matcher_(request_->method(), request_->url().path(), &stream_, &matched_view_);
|
|
|
|
if (!matched)
|
|
{
|
|
LOG_WARN("No view matches the request: %s %s", request_->method().c_str(), request_->url().path().c_str());
|
|
}
|
|
|
|
return matched;
|
|
}
|
|
|
|
bool RequestParser::ParseStartLine(const std::string& line)
|
|
{
|
|
std::vector<string_view> parts;
|
|
Split(line, ' ', true, &parts);
|
|
|
|
if (parts.size() != 3)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
request_->set_method(parts[0]);
|
|
request_->set_url(Url{ parts[1] });
|
|
|
|
// HTTP version is ignored.
|
|
|
|
return true;
|
|
}
|
|
|
|
bool RequestParser::ParseContent(const char* data, std::size_t length)
|
|
{
|
|
if (content_type_.multipart())
|
|
{
|
|
return ParseMultipartContent(data, length);
|
|
}
|
|
else
|
|
{
|
|
return Parser::ParseContent(data, length);
|
|
}
|
|
}
|
|
|
|
bool RequestParser::ParseMultipartContent(const char* data, std::size_t length)
|
|
{
|
|
pending_data_.append(data, length);
|
|
|
|
if (!content_length_parsed_ || content_length_ == kInvalidLength)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
const std::size_t kMaxAllowedMultipartSize = g_max_multipart_size; // 1 GB
|
|
|
|
while (true)
|
|
{
|
|
if (pending_data_.empty())
|
|
{
|
|
break;
|
|
}
|
|
|
|
if (step_ == Step::kStart)
|
|
{
|
|
std::string line;
|
|
if (!GetNextLine(0, &line, true))
|
|
{
|
|
break;
|
|
}
|
|
if (!IsBoundary(line, 0, line.size()))
|
|
{
|
|
LOG_ERRO("Invalid boundary: %s", line.c_str());
|
|
return false;
|
|
}
|
|
step_ = Step::kBoundaryParsed;
|
|
continue;
|
|
}
|
|
|
|
if (step_ == Step::kBoundaryParsed)
|
|
{
|
|
if (!part_)
|
|
{
|
|
part_.reset(new FormPart{});
|
|
if (content_length_ > 0)
|
|
{
|
|
try
|
|
{
|
|
part_->ReserveData(content_length_);
|
|
}
|
|
catch (const std::bad_alloc&)
|
|
{
|
|
LOG_ERRO("Failed to allocate memory for size: %zu", content_length_);
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
bool need_more_data = false;
|
|
if (ParsePartHeaders(&need_more_data))
|
|
{
|
|
step_ = Step::kHeadersParsed;
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
if (need_more_data)
|
|
{
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (step_ == Step::kHeadersParsed)
|
|
{
|
|
std::string strict_boundary = "\r\n--" + content_type_.boundary();
|
|
std::size_t pos = pending_data_.find(strict_boundary);
|
|
|
|
if (pos == std::string::npos)
|
|
{
|
|
std::size_t reserve_len = strict_boundary.size() + 8;
|
|
if (pending_data_.size() > reserve_len)
|
|
{
|
|
std::size_t safe_len = pending_data_.size() - reserve_len;
|
|
|
|
// 【修正点】：使用 part_->data().size() 替代 part_->GetDataSize()，避免触发异常！
|
|
if (part_->data().size() + safe_len > kMaxAllowedMultipartSize)
|
|
{
|
|
LOG_ERRO("Multipart payload size exceeded max limit without boundary!");
|
|
return false;
|
|
}
|
|
|
|
part_->AppendData(pending_data_.data(), safe_len);
|
|
pending_data_.erase(0, safe_len);
|
|
}
|
|
break;
|
|
}
|
|
|
|
std::size_t boundary_start = pos;
|
|
std::size_t line_end = pending_data_.find(kCRLF, boundary_start + 2);
|
|
if (line_end == std::string::npos)
|
|
{
|
|
break;
|
|
}
|
|
|
|
std::size_t count = line_end - (boundary_start + 2);
|
|
bool ended = false;
|
|
|
|
if (!IsBoundary(pending_data_, boundary_start + 2, count, &ended))
|
|
{
|
|
std::size_t safe_len = boundary_start + 2;
|
|
|
|
// 【修正点】：同理替换为 part_->data().size()
|
|
if (part_->data().size() + safe_len > kMaxAllowedMultipartSize)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
part_->AppendData(pending_data_.data(), safe_len);
|
|
pending_data_.erase(0, safe_len);
|
|
continue;
|
|
}
|
|
|
|
if (boundary_start > 0)
|
|
{
|
|
if (part_->data().size() + boundary_start > kMaxAllowedMultipartSize)
|
|
{
|
|
return false;
|
|
}
|
|
part_->AppendData(pending_data_.data(), boundary_start);
|
|
}
|
|
|
|
pending_data_.erase(0, line_end + 2);
|
|
|
|
form_parts_.push_back(part_);
|
|
part_.reset();
|
|
|
|
if (ended)
|
|
{
|
|
step_ = Step::kEnded;
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
step_ = Step::kBoundaryParsed;
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (step_ == Step::kEnded)
|
|
{
|
|
LOG_INFO("Multipart data has ended");
|
|
auto body = std::make_shared<FormBody>(form_parts_, content_type_.boundary());
|
|
request_->SetBody(body, false);
|
|
Finish();
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool RequestParser::ParsePartHeaders(bool* need_more_data)
|
|
{
|
|
std::size_t off = 0;
|
|
|
|
while (true)
|
|
{
|
|
std::string line;
|
|
if (!GetNextLine(off, &line, false))
|
|
{
|
|
// Need more data from next read.
|
|
*need_more_data = true;
|
|
return false;
|
|
}
|
|
|
|
off = off + line.size() + 2; // +2 for CRLF
|
|
|
|
if (line.empty())
|
|
{
|
|
// Headers finished.
|
|
break;
|
|
}
|
|
|
|
Header header;
|
|
if (!SplitKV(line, ':', true, &header.first, &header.second))
|
|
{
|
|
LOG_ERRO("Invalid part header line: %s", line.c_str());
|
|
return false;
|
|
}
|
|
|
|
LOG_INFO("Part header (%s: %s)", header.first.c_str(),
|
|
header.second.c_str());
|
|
|
|
// Parse Content-Disposition.
|
|
if (boost::iequals(header.first, headers::kContentDisposition))
|
|
{
|
|
ContentDisposition content_disposition{ header.second };
|
|
if (!content_disposition.valid())
|
|
{
|
|
LOG_ERRO("Invalid content-disposition header: %s",
|
|
header.second.c_str());
|
|
return false;
|
|
}
|
|
part_->set_name(content_disposition.name());
|
|
part_->set_file_name(content_disposition.file_name());
|
|
LOG_INFO("Content-Disposition (name=%s; filename=%s)",
|
|
part_->name().c_str(), part_->file_name().c_str());
|
|
}
|
|
|
|
// TODO: Parse other headers.
|
|
}
|
|
|
|
// Remove the data which has just been parsed.
|
|
pending_data_.erase(0, off);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool RequestParser::GetNextBoundaryLine(std::size_t* b_off, std::size_t* b_len,
|
|
bool* ended)
|
|
{
|
|
std::size_t off = 0;
|
|
|
|
while (true)
|
|
{
|
|
std::size_t pos = pending_data_.find(kCRLF, off);
|
|
if (pos == std::string::npos)
|
|
{
|
|
break;
|
|
}
|
|
|
|
std::size_t len = pos - off;
|
|
if (len == 0)
|
|
{
|
|
off = pos + 2;
|
|
continue; // Empty line
|
|
}
|
|
|
|
if (IsBoundary(pending_data_, off, len, ended))
|
|
{
|
|
*b_off = off;
|
|
*b_len = len;
|
|
return true;
|
|
}
|
|
|
|
off = pos + 2;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool RequestParser::IsBoundary(const std::string& str, std::size_t off,
|
|
std::size_t count, bool* end) const
|
|
{
|
|
const std::string& boundary = content_type_.boundary();
|
|
|
|
if (count != boundary.size() + 2 && count != boundary.size() + 4)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (str[off] != '-' || str[off + 1] != '-')
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (count == boundary.size() + 4)
|
|
{
|
|
if (str[off + count - 1] != '-' || str[off + count - 2] != '-')
|
|
{
|
|
return false;
|
|
}
|
|
if (end != nullptr)
|
|
{
|
|
*end = true;
|
|
}
|
|
}
|
|
|
|
return strncmp(boundary.c_str(), &str[off + 2], boundary.size()) == 0;
|
|
}
|
|
|
|
} // namespace webcc
|