1
0
mirror of synced 2025-07-19 05:03:13 +03:00
* Fix #1578

* Update README

* Update

* Update

* Update

* Update

* Update

* Update
This commit is contained in:
yhirose
2025-07-06 21:27:24 -04:00
committed by GitHub
parent a3f5569196
commit af73377611
5 changed files with 620 additions and 465 deletions

103
README.md
View File

@ -356,16 +356,80 @@ svr.set_pre_request_handler([](const auto& req, auto& res) {
}); });
``` ```
### 'multipart/form-data' POST data ### Form data handling
#### URL-encoded form data ('application/x-www-form-urlencoded')
```cpp ```cpp
svr.Post("/multipart", [&](const auto& req, auto& res) { svr.Post("/form", [&](const auto& req, auto& res) {
auto size = req.files.size(); // URL query parameters and form-encoded data are accessible via req.params
auto ret = req.has_file("name1"); std::string username = req.get_param_value("username");
const auto& file = req.get_file_value("name1"); std::string password = req.get_param_value("password");
// file.filename;
// file.content_type; // Handle multiple values with same name
// file.content; auto interests = req.get_param_values("interests");
// Check existence
if (req.has_param("newsletter")) {
// Handle newsletter subscription
}
});
```
#### 'multipart/form-data' POST data
```cpp
svr.Post("/multipart", [&](const Request& req, Response& res) {
// Access text fields (from form inputs without files)
std::string username = req.form.get_field("username");
std::string bio = req.form.get_field("bio");
// Access uploaded files
if (req.form.has_file("avatar")) {
const auto& file = req.form.get_file("avatar");
std::cout << "Uploaded file: " << file.filename
<< " (" << file.content_type << ") - "
<< file.content.size() << " bytes" << std::endl;
// Access additional headers if needed
for (const auto& header : file.headers) {
std::cout << "Header: " << header.first << " = " << header.second << std::endl;
}
// Save to disk
std::ofstream ofs(file.filename, std::ios::binary);
ofs << file.content;
}
// Handle multiple values with same name
auto tags = req.form.get_fields("tags"); // e.g., multiple checkboxes
for (const auto& tag : tags) {
std::cout << "Tag: " << tag << std::endl;
}
auto documents = req.form.get_files("documents"); // multiple file upload
for (const auto& doc : documents) {
std::cout << "Document: " << doc.filename
<< " (" << doc.content.size() << " bytes)" << std::endl;
}
// Check existence before accessing
if (req.form.has_field("newsletter")) {
std::cout << "Newsletter subscription: " << req.form.get_field("newsletter") << std::endl;
}
// Get counts for validation
if (req.form.get_field_count("tags") > 5) {
res.status = StatusCode::BadRequest_400;
res.set_content("Too many tags", "text/plain");
return;
}
// Summary
std::cout << "Received " << req.form.fields.size() << " text fields and "
<< req.form.files.size() << " files" << std::endl;
res.set_content("Upload successful", "text/plain");
}); });
``` ```
@ -376,16 +440,29 @@ svr.Post("/content_receiver",
[&](const Request &req, Response &res, const ContentReader &content_reader) { [&](const Request &req, Response &res, const ContentReader &content_reader) {
if (req.is_multipart_form_data()) { if (req.is_multipart_form_data()) {
// NOTE: `content_reader` is blocking until every form data field is read // NOTE: `content_reader` is blocking until every form data field is read
MultipartFormDataItems files; // This approach allows streaming processing of large files
std::vector<FormData> items;
content_reader( content_reader(
[&](const MultipartFormData &file) { [&](const FormData &item) {
files.push_back(file); items.push_back(item);
return true; return true;
}, },
[&](const char *data, size_t data_length) { [&](const char *data, size_t data_length) {
files.back().content.append(data, data_length); items.back().content.append(data, data_length);
return true; return true;
}); });
// Process the received items
for (const auto& item : items) {
if (item.filename.empty()) {
// Text field
std::cout << "Field: " << item.name << " = " << item.content << std::endl;
} else {
// File
std::cout << "File: " << item.name << " (" << item.filename << ") - "
<< item.content.size() << " bytes" << std::endl;
}
}
} else { } else {
std::string body; std::string body;
content_reader([&](const char *data, size_t data_length) { content_reader([&](const char *data, size_t data_length) {
@ -691,7 +768,7 @@ auto res = cli.Post("/post", params);
### POST with Multipart Form Data ### POST with Multipart Form Data
```c++ ```c++
httplib::MultipartFormDataItems items = { httplib::UploadFormDataItems items = {
{ "text1", "text default", "", "" }, { "text1", "text default", "", "" },
{ "text2", "aωb", "", "" }, { "text2", "aωb", "", "" },
{ "file1", "h\ne\n\nl\nl\no\n", "hello.txt", "text/plain" }, { "file1", "h\ne\n\nl\nl\no\n", "hello.txt", "text/plain" },

View File

@ -27,13 +27,26 @@ string dump_headers(const Headers &headers) {
return s; return s;
} }
string dump_multipart_files(const MultipartFormDataMap &files) { string dump_multipart_formdata(const MultipartFormData &form) {
string s; string s;
char buf[BUFSIZ]; char buf[BUFSIZ];
s += "--------------------------------\n"; s += "--------------------------------\n";
for (const auto &x : files) { for (const auto &x : form.fields) {
const auto &name = x.first;
const auto &field = x.second;
snprintf(buf, sizeof(buf), "name: %s\n", name.c_str());
s += buf;
snprintf(buf, sizeof(buf), "text length: %zu\n", field.content.size());
s += buf;
s += "----------------\n";
}
for (const auto &x : form.files) {
const auto &name = x.first; const auto &name = x.first;
const auto &file = x.second; const auto &file = x.second;
@ -77,7 +90,7 @@ string log(const Request &req, const Response &res) {
s += buf; s += buf;
s += dump_headers(req.headers); s += dump_headers(req.headers);
s += dump_multipart_files(req.files); s += dump_multipart_formdata(req.form);
s += "--------------------------------\n"; s += "--------------------------------\n";
@ -101,7 +114,7 @@ int main(int argc, const char **argv) {
#endif #endif
svr.Post("/multipart", [](const Request &req, Response &res) { svr.Post("/multipart", [](const Request &req, Response &res) {
auto body = dump_headers(req.headers) + dump_multipart_files(req.files); auto body = dump_headers(req.headers) + dump_multipart_formdata(req.form);
res.set_content(body, "text/plain"); res.set_content(body, "text/plain");
}); });

View File

@ -37,8 +37,8 @@ int main(void) {
}); });
svr.Post("/post", [](const Request &req, Response &res) { svr.Post("/post", [](const Request &req, Response &res) {
auto image_file = req.get_file_value("image_file"); const auto &image_file = req.form.get_file("image_file");
auto text_file = req.get_file_value("text_file"); const auto &text_file = req.form.get_file("text_file");
cout << "image file length: " << image_file.content.length() << endl cout << "image file length: " << image_file.content.length() << endl
<< "image file name: " << image_file.filename << endl << "image file name: " << image_file.filename << endl

354
httplib.h
View File

@ -561,24 +561,47 @@ using UploadProgress = std::function<bool(uint64_t current, uint64_t total)>;
struct Response; struct Response;
using ResponseHandler = std::function<bool(const Response &response)>; using ResponseHandler = std::function<bool(const Response &response)>;
struct MultipartFormData { struct FormData {
std::string name; std::string name;
std::string content; std::string content;
std::string filename; std::string filename;
std::string content_type; std::string content_type;
Headers headers; Headers headers;
}; };
using MultipartFormDataItems = std::vector<MultipartFormData>;
using MultipartFormDataMap = std::multimap<std::string, MultipartFormData>;
struct MultipartFormDataForClientInput { struct FormField {
std::string name;
std::string content;
Headers headers;
};
using FormFields = std::multimap<std::string, FormField>;
using FormFiles = std::multimap<std::string, FormData>;
struct MultipartFormData {
FormFields fields; // Text fields from multipart
FormFiles files; // Files from multipart
// Text field access
std::string get_field(const std::string &key, size_t id = 0) const;
std::vector<std::string> get_fields(const std::string &key) const;
bool has_field(const std::string &key) const;
size_t get_field_count(const std::string &key) const;
// File access
FormData get_file(const std::string &key, size_t id = 0) const;
std::vector<FormData> get_files(const std::string &key) const;
bool has_file(const std::string &key) const;
size_t get_file_count(const std::string &key) const;
};
struct UploadFormData {
std::string name; std::string name;
std::string content; std::string content;
std::string filename; std::string filename;
std::string content_type; std::string content_type;
}; };
using MultipartFormDataItemsForClientInput = using UploadFormDataItems = std::vector<UploadFormData>;
std::vector<MultipartFormDataForClientInput>;
class DataSink { class DataSink {
public: public:
@ -621,13 +644,13 @@ using ContentProviderWithoutLength =
using ContentProviderResourceReleaser = std::function<void(bool success)>; using ContentProviderResourceReleaser = std::function<void(bool success)>;
struct MultipartFormDataProvider { struct FormDataProvider {
std::string name; std::string name;
ContentProviderWithoutLength provider; ContentProviderWithoutLength provider;
std::string filename; std::string filename;
std::string content_type; std::string content_type;
}; };
using MultipartFormDataProviderItems = std::vector<MultipartFormDataProvider>; using FormDataProviderItems = std::vector<FormDataProvider>;
using ContentReceiverWithProgress = using ContentReceiverWithProgress =
std::function<bool(const char *data, size_t data_length, uint64_t offset, std::function<bool(const char *data, size_t data_length, uint64_t offset,
@ -636,22 +659,20 @@ using ContentReceiverWithProgress =
using ContentReceiver = using ContentReceiver =
std::function<bool(const char *data, size_t data_length)>; std::function<bool(const char *data, size_t data_length)>;
using MultipartContentHeader = using FormDataHeader = std::function<bool(const FormData &file)>;
std::function<bool(const MultipartFormData &file)>;
class ContentReader { class ContentReader {
public: public:
using Reader = std::function<bool(ContentReceiver receiver)>; using Reader = std::function<bool(ContentReceiver receiver)>;
using MultipartReader = std::function<bool(MultipartContentHeader header, using FormDataReader =
ContentReceiver receiver)>; std::function<bool(FormDataHeader header, ContentReceiver receiver)>;
ContentReader(Reader reader, MultipartReader multipart_reader) ContentReader(Reader reader, FormDataReader multipart_reader)
: reader_(std::move(reader)), : reader_(std::move(reader)),
multipart_reader_(std::move(multipart_reader)) {} formdata_reader_(std::move(multipart_reader)) {}
bool operator()(MultipartContentHeader header, bool operator()(FormDataHeader header, ContentReceiver receiver) const {
ContentReceiver receiver) const { return formdata_reader_(std::move(header), std::move(receiver));
return multipart_reader_(std::move(header), std::move(receiver));
} }
bool operator()(ContentReceiver receiver) const { bool operator()(ContentReceiver receiver) const {
@ -659,7 +680,7 @@ public:
} }
Reader reader_; Reader reader_;
MultipartReader multipart_reader_; FormDataReader formdata_reader_;
}; };
using Range = std::pair<ssize_t, ssize_t>; using Range = std::pair<ssize_t, ssize_t>;
@ -681,7 +702,7 @@ struct Request {
// for server // for server
std::string version; std::string version;
std::string target; std::string target;
MultipartFormDataMap files; MultipartFormData form;
Ranges ranges; Ranges ranges;
Match matches; Match matches;
std::unordered_map<std::string, std::string> path_params; std::unordered_map<std::string, std::string> path_params;
@ -711,10 +732,6 @@ struct Request {
bool is_multipart_form_data() const; bool is_multipart_form_data() const;
bool has_file(const std::string &key) const;
MultipartFormData get_file_value(const std::string &key) const;
std::vector<MultipartFormData> get_file_values(const std::string &key) const;
// private members... // private members...
size_t redirect_count_ = CPPHTTPLIB_REDIRECT_MAX_COUNT; size_t redirect_count_ = CPPHTTPLIB_REDIRECT_MAX_COUNT;
size_t content_length_ = 0; size_t content_length_ = 0;
@ -1159,14 +1176,14 @@ private:
Response &res, const std::string &boundary, Response &res, const std::string &boundary,
const std::string &content_type); const std::string &content_type);
bool read_content(Stream &strm, Request &req, Response &res); bool read_content(Stream &strm, Request &req, Response &res);
bool bool read_content_with_content_receiver(Stream &strm, Request &req,
read_content_with_content_receiver(Stream &strm, Request &req, Response &res, Response &res,
ContentReceiver receiver, ContentReceiver receiver,
MultipartContentHeader multipart_header, FormDataHeader multipart_header,
ContentReceiver multipart_receiver); ContentReceiver multipart_receiver);
bool read_content_core(Stream &strm, Request &req, Response &res, bool read_content_core(Stream &strm, Request &req, Response &res,
ContentReceiver receiver, ContentReceiver receiver,
MultipartContentHeader multipart_header, FormDataHeader multipart_header,
ContentReceiver multipart_receiver) const; ContentReceiver multipart_receiver) const;
virtual bool process_and_close_socket(socket_t sock); virtual bool process_and_close_socket(socket_t sock);
@ -1333,16 +1350,16 @@ public:
Result Post(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Params &params); Result Post(const std::string &path, const Params &params);
Result Post(const std::string &path, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Post(const std::string &path, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers); Result Post(const std::string &path, const Headers &headers);
Result Post(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const Params &params); Result Post(const std::string &path, const Headers &headers, const Params &params);
Result Post(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const std::string &boundary, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const std::string &boundary, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const MultipartFormDataProviderItems &provider_items, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const FormDataProviderItems &provider_items, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr);
Result Put(const std::string &path); Result Put(const std::string &path);
@ -1351,16 +1368,16 @@ public:
Result Put(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Params &params); Result Put(const std::string &path, const Params &params);
Result Put(const std::string &path, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Put(const std::string &path, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers); Result Put(const std::string &path, const Headers &headers);
Result Put(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const Params &params); Result Put(const std::string &path, const Headers &headers, const Params &params);
Result Put(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const std::string &boundary, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const std::string &boundary, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const MultipartFormDataProviderItems &provider_items, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const FormDataProviderItems &provider_items, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr);
Result Patch(const std::string &path); Result Patch(const std::string &path);
@ -1369,16 +1386,16 @@ public:
Result Patch(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Params &params); Result Patch(const std::string &path, const Params &params);
Result Patch(const std::string &path, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Patch(const std::string &path, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const Params &params); Result Patch(const std::string &path, const Headers &headers, const Params &params);
Result Patch(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const std::string &boundary, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const std::string &boundary, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const MultipartFormDataProviderItems &provider_items, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const FormDataProviderItems &provider_items, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr);
Result Delete(const std::string &path, DownloadProgress progress = nullptr); Result Delete(const std::string &path, DownloadProgress progress = nullptr);
@ -1628,9 +1645,8 @@ private:
ContentProviderWithoutLength content_provider_without_length, ContentProviderWithoutLength content_provider_without_length,
const std::string &content_type, UploadProgress progress); const std::string &content_type, UploadProgress progress);
ContentProviderWithoutLength get_multipart_content_provider( ContentProviderWithoutLength get_multipart_content_provider(
const std::string &boundary, const std::string &boundary, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items, const FormDataProviderItems &provider_items) const;
const MultipartFormDataProviderItems &provider_items) const;
std::string adjust_host_string(const std::string &host) const; std::string adjust_host_string(const std::string &host) const;
@ -1684,16 +1700,16 @@ public:
Result Post(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Params &params); Result Post(const std::string &path, const Params &params);
Result Post(const std::string &path, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Post(const std::string &path, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers); Result Post(const std::string &path, const Headers &headers);
Result Post(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const Params &params); Result Post(const std::string &path, const Headers &headers, const Params &params);
Result Post(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const std::string &boundary, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const std::string &boundary, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const MultipartFormDataProviderItems &provider_items, UploadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const FormDataProviderItems &provider_items, UploadProgress progress = nullptr);
Result Post(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr); Result Post(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr);
Result Put(const std::string &path); Result Put(const std::string &path);
@ -1702,16 +1718,16 @@ public:
Result Put(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Params &params); Result Put(const std::string &path, const Params &params);
Result Put(const std::string &path, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Put(const std::string &path, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers); Result Put(const std::string &path, const Headers &headers);
Result Put(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const Params &params); Result Put(const std::string &path, const Headers &headers, const Params &params);
Result Put(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const std::string &boundary, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const std::string &boundary, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const MultipartFormDataProviderItems &provider_items, UploadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const FormDataProviderItems &provider_items, UploadProgress progress = nullptr);
Result Put(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr); Result Put(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr);
Result Patch(const std::string &path); Result Patch(const std::string &path);
@ -1720,16 +1736,16 @@ public:
Result Patch(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Params &params); Result Patch(const std::string &path, const Params &params);
Result Patch(const std::string &path, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Patch(const std::string &path, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers); Result Patch(const std::string &path, const Headers &headers);
Result Patch(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const char *body, size_t content_length, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, size_t content_length, ContentProvider content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, ContentProviderWithoutLength content_provider, const std::string &content_type, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const Params &params); Result Patch(const std::string &path, const Headers &headers, const Params &params);
Result Patch(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const std::string &boundary, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const std::string &boundary, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const MultipartFormDataItemsForClientInput &items, const MultipartFormDataProviderItems &provider_items, UploadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items, const FormDataProviderItems &provider_items, UploadProgress progress = nullptr);
Result Patch(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr); Result Patch(const std::string &path, const Headers &headers, const std::string &body, const std::string &content_type, ContentReceiver content_receiver, DownloadProgress progress = nullptr);
Result Delete(const std::string &path, DownloadProgress progress = nullptr); Result Delete(const std::string &path, DownloadProgress progress = nullptr);
@ -5327,9 +5343,9 @@ inline bool parse_accept_header(const std::string &s,
return true; return true;
} }
class MultipartFormDataParser { class FormDataParser {
public: public:
MultipartFormDataParser() = default; FormDataParser() = default;
void set_boundary(std::string &&boundary) { void set_boundary(std::string &&boundary) {
boundary_ = boundary; boundary_ = boundary;
@ -5339,8 +5355,8 @@ public:
bool is_valid() const { return is_valid_; } bool is_valid() const { return is_valid_; }
bool parse(const char *buf, size_t n, const ContentReceiver &content_callback, bool parse(const char *buf, size_t n, const FormDataHeader &header_callback,
const MultipartContentHeader &header_callback) { const ContentReceiver &content_callback) {
buf_append(buf, n); buf_append(buf, n);
@ -5381,7 +5397,7 @@ public:
return false; return false;
} }
// parse and emplace space trimmed headers into a map // Parse and emplace space trimmed headers into a map
if (!parse_header( if (!parse_header(
header.data(), header.data() + header.size(), header.data(), header.data() + header.size(),
[&](const std::string &key, const std::string &val) { [&](const std::string &key, const std::string &val) {
@ -5512,7 +5528,7 @@ private:
size_t state_ = 0; size_t state_ = 0;
bool is_valid_ = false; bool is_valid_ = false;
MultipartFormData file_; FormData file_;
// Buffer // Buffer
bool start_with(const std::string &a, size_t spos, size_t epos, bool start_with(const std::string &a, size_t spos, size_t epos,
@ -5650,7 +5666,7 @@ serialize_multipart_formdata_get_content_type(const std::string &boundary) {
} }
inline std::string inline std::string
serialize_multipart_formdata(const MultipartFormDataItemsForClientInput &items, serialize_multipart_formdata(const UploadFormDataItems &items,
const std::string &boundary, bool finish = true) { const std::string &boundary, bool finish = true) {
std::string body; std::string body;
@ -6422,19 +6438,47 @@ inline bool Request::is_multipart_form_data() const {
return !content_type.rfind("multipart/form-data", 0); return !content_type.rfind("multipart/form-data", 0);
} }
inline bool Request::has_file(const std::string &key) const { // Multipart FormData implementation
return files.find(key) != files.end(); inline std::string MultipartFormData::get_field(const std::string &key,
size_t id) const {
auto rng = fields.equal_range(key);
auto it = rng.first;
std::advance(it, static_cast<ssize_t>(id));
if (it != rng.second) { return it->second.content; }
return std::string();
} }
inline MultipartFormData Request::get_file_value(const std::string &key) const { inline std::vector<std::string>
auto it = files.find(key); MultipartFormData::get_fields(const std::string &key) const {
if (it != files.end()) { return it->second; } std::vector<std::string> values;
return MultipartFormData(); auto rng = fields.equal_range(key);
for (auto it = rng.first; it != rng.second; it++) {
values.push_back(it->second.content);
}
return values;
} }
inline std::vector<MultipartFormData> inline bool MultipartFormData::has_field(const std::string &key) const {
Request::get_file_values(const std::string &key) const { return fields.find(key) != fields.end();
std::vector<MultipartFormData> values; }
inline size_t MultipartFormData::get_field_count(const std::string &key) const {
auto r = fields.equal_range(key);
return static_cast<size_t>(std::distance(r.first, r.second));
}
inline FormData MultipartFormData::get_file(const std::string &key,
size_t id) const {
auto rng = files.equal_range(key);
auto it = rng.first;
std::advance(it, static_cast<ssize_t>(id));
if (it != rng.second) { return it->second; }
return FormData();
}
inline std::vector<FormData>
MultipartFormData::get_files(const std::string &key) const {
std::vector<FormData> values;
auto rng = files.equal_range(key); auto rng = files.equal_range(key);
for (auto it = rng.first; it != rng.second; it++) { for (auto it = rng.first; it != rng.second; it++) {
values.push_back(it->second); values.push_back(it->second);
@ -6442,6 +6486,15 @@ Request::get_file_values(const std::string &key) const {
return values; return values;
} }
inline bool MultipartFormData::has_file(const std::string &key) const {
return files.find(key) != files.end();
}
inline size_t MultipartFormData::get_file_count(const std::string &key) const {
auto r = files.equal_range(key);
return static_cast<size_t>(std::distance(r.first, r.second));
}
// Response implementation // Response implementation
inline bool Response::has_header(const std::string &key) const { inline bool Response::has_header(const std::string &key) const {
return headers.find(key) != headers.end(); return headers.find(key) != headers.end();
@ -7300,8 +7353,10 @@ Server::write_content_with_provider(Stream &strm, const Request &req,
} }
inline bool Server::read_content(Stream &strm, Request &req, Response &res) { inline bool Server::read_content(Stream &strm, Request &req, Response &res) {
MultipartFormDataMap::iterator cur; FormFields::iterator cur_field;
auto file_count = 0; FormFiles::iterator cur_file;
auto is_text_field = false;
size_t count = 0;
if (read_content_core( if (read_content_core(
strm, req, res, strm, req, res,
// Regular // Regular
@ -7310,18 +7365,32 @@ inline bool Server::read_content(Stream &strm, Request &req, Response &res) {
req.body.append(buf, n); req.body.append(buf, n);
return true; return true;
}, },
// Multipart // Multipart FormData
[&](const MultipartFormData &file) { [&](const FormData &file) {
if (file_count++ == CPPHTTPLIB_MULTIPART_FORM_DATA_FILE_MAX_COUNT) { if (count++ == CPPHTTPLIB_MULTIPART_FORM_DATA_FILE_MAX_COUNT) {
return false; return false;
} }
cur = req.files.emplace(file.name, file);
if (file.filename.empty()) {
cur_field = req.form.fields.emplace(
file.name, FormField{file.name, file.content, file.headers});
is_text_field = true;
} else {
cur_file = req.form.files.emplace(file.name, file);
is_text_field = false;
}
return true; return true;
}, },
[&](const char *buf, size_t n) { [&](const char *buf, size_t n) {
auto &content = cur->second.content; if (is_text_field) {
auto &content = cur_field->second.content;
if (content.size() + n > content.max_size()) { return false; } if (content.size() + n > content.max_size()) { return false; }
content.append(buf, n); content.append(buf, n);
} else {
auto &content = cur_file->second.content;
if (content.size() + n > content.max_size()) { return false; }
content.append(buf, n);
}
return true; return true;
})) { })) {
const auto &content_type = req.get_header_value("Content-Type"); const auto &content_type = req.get_header_value("Content-Type");
@ -7339,19 +7408,16 @@ inline bool Server::read_content(Stream &strm, Request &req, Response &res) {
inline bool Server::read_content_with_content_receiver( inline bool Server::read_content_with_content_receiver(
Stream &strm, Request &req, Response &res, ContentReceiver receiver, Stream &strm, Request &req, Response &res, ContentReceiver receiver,
MultipartContentHeader multipart_header, FormDataHeader multipart_header, ContentReceiver multipart_receiver) {
ContentReceiver multipart_receiver) {
return read_content_core(strm, req, res, std::move(receiver), return read_content_core(strm, req, res, std::move(receiver),
std::move(multipart_header), std::move(multipart_header),
std::move(multipart_receiver)); std::move(multipart_receiver));
} }
inline bool inline bool Server::read_content_core(
Server::read_content_core(Stream &strm, Request &req, Response &res, Stream &strm, Request &req, Response &res, ContentReceiver receiver,
ContentReceiver receiver, FormDataHeader multipart_header, ContentReceiver multipart_receiver) const {
MultipartContentHeader multipart_header, detail::FormDataParser multipart_form_data_parser;
ContentReceiver multipart_receiver) const {
detail::MultipartFormDataParser multipart_form_data_parser;
ContentReceiverWithProgress out; ContentReceiverWithProgress out;
if (req.is_multipart_form_data()) { if (req.is_multipart_form_data()) {
@ -7364,19 +7430,8 @@ Server::read_content_core(Stream &strm, Request &req, Response &res,
multipart_form_data_parser.set_boundary(std::move(boundary)); multipart_form_data_parser.set_boundary(std::move(boundary));
out = [&](const char *buf, size_t n, uint64_t /*off*/, uint64_t /*len*/) { out = [&](const char *buf, size_t n, uint64_t /*off*/, uint64_t /*len*/) {
/* For debug return multipart_form_data_parser.parse(buf, n, multipart_header,
size_t pos = 0; multipart_receiver);
while (pos < n) {
auto read_size = (std::min)<size_t>(1, n - pos);
auto ret = multipart_form_data_parser.parse(
buf + pos, read_size, multipart_receiver, multipart_header);
if (!ret) { return false; }
pos += read_size;
}
return true;
*/
return multipart_form_data_parser.parse(buf, n, multipart_receiver,
multipart_header);
}; };
} else { } else {
out = [receiver](const char *buf, size_t n, uint64_t /*off*/, out = [receiver](const char *buf, size_t n, uint64_t /*off*/,
@ -7582,7 +7637,7 @@ inline bool Server::routing(Request &req, Response &res, Stream &strm) {
return read_content_with_content_receiver( return read_content_with_content_receiver(
strm, req, res, std::move(receiver), nullptr, nullptr); strm, req, res, std::move(receiver), nullptr, nullptr);
}, },
[&](MultipartContentHeader header, ContentReceiver receiver) { [&](FormDataHeader header, ContentReceiver receiver) {
return read_content_with_content_receiver(strm, req, res, nullptr, return read_content_with_content_receiver(strm, req, res, nullptr,
std::move(header), std::move(header),
std::move(receiver)); std::move(receiver));
@ -8949,9 +9004,8 @@ inline bool ClientImpl::process_request(Stream &strm, Request &req,
} }
inline ContentProviderWithoutLength ClientImpl::get_multipart_content_provider( inline ContentProviderWithoutLength ClientImpl::get_multipart_content_provider(
const std::string &boundary, const std::string &boundary, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items, const FormDataProviderItems &provider_items) const {
const MultipartFormDataProviderItems &provider_items) const {
size_t cur_item = 0; size_t cur_item = 0;
size_t cur_start = 0; size_t cur_start = 0;
// cur_item and cur_start are copied to within the std::function and maintain // cur_item and cur_start are copied to within the std::function and maintain
@ -9164,16 +9218,14 @@ inline Result ClientImpl::Post(const std::string &path, const Headers &headers,
return Post(path, headers, query, "application/x-www-form-urlencoded"); return Post(path, headers, query, "application/x-www-form-urlencoded");
} }
inline Result inline Result ClientImpl::Post(const std::string &path,
ClientImpl::Post(const std::string &path, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items,
UploadProgress progress) { UploadProgress progress) {
return Post(path, Headers(), items, progress); return Post(path, Headers(), items, progress);
} }
inline Result inline Result ClientImpl::Post(const std::string &path, const Headers &headers,
ClientImpl::Post(const std::string &path, const Headers &headers, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items,
UploadProgress progress) { UploadProgress progress) {
const auto &boundary = detail::make_multipart_data_boundary(); const auto &boundary = detail::make_multipart_data_boundary();
const auto &content_type = const auto &content_type =
@ -9182,10 +9234,10 @@ ClientImpl::Post(const std::string &path, const Headers &headers,
return Post(path, headers, body, content_type, progress); return Post(path, headers, body, content_type, progress);
} }
inline Result inline Result ClientImpl::Post(const std::string &path, const Headers &headers,
ClientImpl::Post(const std::string &path, const Headers &headers, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items, const std::string &boundary,
const std::string &boundary, UploadProgress progress) { UploadProgress progress) {
if (!detail::is_multipart_boundary_chars_valid(boundary)) { if (!detail::is_multipart_boundary_chars_valid(boundary)) {
return Result{nullptr, Error::UnsupportedMultipartBoundaryChars}; return Result{nullptr, Error::UnsupportedMultipartBoundaryChars};
} }
@ -9232,10 +9284,9 @@ inline Result ClientImpl::Post(const std::string &path, const Headers &headers,
progress); progress);
} }
inline Result inline Result ClientImpl::Post(const std::string &path, const Headers &headers,
ClientImpl::Post(const std::string &path, const Headers &headers, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items, const FormDataProviderItems &provider_items,
const MultipartFormDataProviderItems &provider_items,
UploadProgress progress) { UploadProgress progress) {
const auto &boundary = detail::make_multipart_data_boundary(); const auto &boundary = detail::make_multipart_data_boundary();
const auto &content_type = const auto &content_type =
@ -9320,13 +9371,13 @@ inline Result ClientImpl::Put(const std::string &path, const Headers &headers,
} }
inline Result ClientImpl::Put(const std::string &path, inline Result ClientImpl::Put(const std::string &path,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
UploadProgress progress) { UploadProgress progress) {
return Put(path, Headers(), items, progress); return Put(path, Headers(), items, progress);
} }
inline Result ClientImpl::Put(const std::string &path, const Headers &headers, inline Result ClientImpl::Put(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
UploadProgress progress) { UploadProgress progress) {
const auto &boundary = detail::make_multipart_data_boundary(); const auto &boundary = detail::make_multipart_data_boundary();
const auto &content_type = const auto &content_type =
@ -9336,7 +9387,7 @@ inline Result ClientImpl::Put(const std::string &path, const Headers &headers,
} }
inline Result ClientImpl::Put(const std::string &path, const Headers &headers, inline Result ClientImpl::Put(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
const std::string &boundary, const std::string &boundary,
UploadProgress progress) { UploadProgress progress) {
if (!detail::is_multipart_boundary_chars_valid(boundary)) { if (!detail::is_multipart_boundary_chars_valid(boundary)) {
@ -9385,10 +9436,9 @@ inline Result ClientImpl::Put(const std::string &path, const Headers &headers,
progress); progress);
} }
inline Result inline Result ClientImpl::Put(const std::string &path, const Headers &headers,
ClientImpl::Put(const std::string &path, const Headers &headers, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items, const FormDataProviderItems &provider_items,
const MultipartFormDataProviderItems &provider_items,
UploadProgress progress) { UploadProgress progress) {
const auto &boundary = detail::make_multipart_data_boundary(); const auto &boundary = detail::make_multipart_data_boundary();
const auto &content_type = const auto &content_type =
@ -9474,16 +9524,14 @@ inline Result ClientImpl::Patch(const std::string &path, const Headers &headers,
return Patch(path, headers, query, "application/x-www-form-urlencoded"); return Patch(path, headers, query, "application/x-www-form-urlencoded");
} }
inline Result inline Result ClientImpl::Patch(const std::string &path,
ClientImpl::Patch(const std::string &path, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items,
UploadProgress progress) { UploadProgress progress) {
return Patch(path, Headers(), items, progress); return Patch(path, Headers(), items, progress);
} }
inline Result inline Result ClientImpl::Patch(const std::string &path, const Headers &headers,
ClientImpl::Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items,
UploadProgress progress) { UploadProgress progress) {
const auto &boundary = detail::make_multipart_data_boundary(); const auto &boundary = detail::make_multipart_data_boundary();
const auto &content_type = const auto &content_type =
@ -9492,10 +9540,10 @@ ClientImpl::Patch(const std::string &path, const Headers &headers,
return Patch(path, headers, body, content_type, progress); return Patch(path, headers, body, content_type, progress);
} }
inline Result inline Result ClientImpl::Patch(const std::string &path, const Headers &headers,
ClientImpl::Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items, const std::string &boundary,
const std::string &boundary, UploadProgress progress) { UploadProgress progress) {
if (!detail::is_multipart_boundary_chars_valid(boundary)) { if (!detail::is_multipart_boundary_chars_valid(boundary)) {
return Result{nullptr, Error::UnsupportedMultipartBoundaryChars}; return Result{nullptr, Error::UnsupportedMultipartBoundaryChars};
} }
@ -9543,10 +9591,9 @@ inline Result ClientImpl::Patch(const std::string &path, const Headers &headers,
progress); progress);
} }
inline Result inline Result ClientImpl::Patch(const std::string &path, const Headers &headers,
ClientImpl::Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items, const FormDataProviderItems &provider_items,
const MultipartFormDataProviderItems &provider_items,
UploadProgress progress) { UploadProgress progress) {
const auto &boundary = detail::make_multipart_data_boundary(); const auto &boundary = detail::make_multipart_data_boundary();
const auto &content_type = const auto &content_type =
@ -10880,24 +10927,24 @@ inline Result Client::Post(const std::string &path, const Headers &headers,
return cli_->Post(path, headers, params); return cli_->Post(path, headers, params);
} }
inline Result Client::Post(const std::string &path, inline Result Client::Post(const std::string &path,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
UploadProgress progress) { UploadProgress progress) {
return cli_->Post(path, items, progress); return cli_->Post(path, items, progress);
} }
inline Result Client::Post(const std::string &path, const Headers &headers, inline Result Client::Post(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
UploadProgress progress) { UploadProgress progress) {
return cli_->Post(path, headers, items, progress); return cli_->Post(path, headers, items, progress);
} }
inline Result Client::Post(const std::string &path, const Headers &headers, inline Result Client::Post(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
const std::string &boundary, const std::string &boundary,
UploadProgress progress) { UploadProgress progress) {
return cli_->Post(path, headers, items, boundary, progress); return cli_->Post(path, headers, items, boundary, progress);
} }
inline Result Client::Post(const std::string &path, const Headers &headers, inline Result Client::Post(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
const MultipartFormDataProviderItems &provider_items, const FormDataProviderItems &provider_items,
UploadProgress progress) { UploadProgress progress) {
return cli_->Post(path, headers, items, provider_items, progress); return cli_->Post(path, headers, items, provider_items, progress);
} }
@ -10973,24 +11020,24 @@ inline Result Client::Put(const std::string &path, const Headers &headers,
return cli_->Put(path, headers, params); return cli_->Put(path, headers, params);
} }
inline Result Client::Put(const std::string &path, inline Result Client::Put(const std::string &path,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
UploadProgress progress) { UploadProgress progress) {
return cli_->Put(path, items, progress); return cli_->Put(path, items, progress);
} }
inline Result Client::Put(const std::string &path, const Headers &headers, inline Result Client::Put(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
UploadProgress progress) { UploadProgress progress) {
return cli_->Put(path, headers, items, progress); return cli_->Put(path, headers, items, progress);
} }
inline Result Client::Put(const std::string &path, const Headers &headers, inline Result Client::Put(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
const std::string &boundary, const std::string &boundary,
UploadProgress progress) { UploadProgress progress) {
return cli_->Put(path, headers, items, boundary, progress); return cli_->Put(path, headers, items, boundary, progress);
} }
inline Result Client::Put(const std::string &path, const Headers &headers, inline Result Client::Put(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
const MultipartFormDataProviderItems &provider_items, const FormDataProviderItems &provider_items,
UploadProgress progress) { UploadProgress progress) {
return cli_->Put(path, headers, items, provider_items, progress); return cli_->Put(path, headers, items, provider_items, progress);
} }
@ -11069,25 +11116,24 @@ inline Result Client::Patch(const std::string &path, const Headers &headers,
return cli_->Patch(path, headers, params); return cli_->Patch(path, headers, params);
} }
inline Result Client::Patch(const std::string &path, inline Result Client::Patch(const std::string &path,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
UploadProgress progress) { UploadProgress progress) {
return cli_->Patch(path, items, progress); return cli_->Patch(path, items, progress);
} }
inline Result Client::Patch(const std::string &path, const Headers &headers, inline Result Client::Patch(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
UploadProgress progress) { UploadProgress progress) {
return cli_->Patch(path, headers, items, progress); return cli_->Patch(path, headers, items, progress);
} }
inline Result Client::Patch(const std::string &path, const Headers &headers, inline Result Client::Patch(const std::string &path, const Headers &headers,
const MultipartFormDataItemsForClientInput &items, const UploadFormDataItems &items,
const std::string &boundary, const std::string &boundary,
UploadProgress progress) { UploadProgress progress) {
return cli_->Patch(path, headers, items, boundary, progress); return cli_->Patch(path, headers, items, boundary, progress);
} }
inline Result inline Result Client::Patch(const std::string &path, const Headers &headers,
Client::Patch(const std::string &path, const Headers &headers, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items, const FormDataProviderItems &provider_items,
const MultipartFormDataProviderItems &provider_items,
UploadProgress progress) { UploadProgress progress) {
return cli_->Patch(path, headers, items, provider_items, progress); return cli_->Patch(path, headers, items, provider_items, progress);
} }

View File

@ -55,15 +55,14 @@ const std::string JSON_DATA = "{\"hello\":\"world\"}";
const string LARGE_DATA = string(1024 * 1024 * 100, '@'); // 100MB const string LARGE_DATA = string(1024 * 1024 * 100, '@'); // 100MB
MultipartFormData &get_file_value(MultipartFormDataItems &files, FormData &get_file_value(std::vector<FormData> &items, const char *key) {
const char *key) { auto it = std::find_if(items.begin(), items.end(), [&](const FormData &file) {
auto it = std::find_if( return file.name == key;
files.begin(), files.end(), });
[&](const MultipartFormData &file) { return file.name == key; });
#ifdef CPPHTTPLIB_NO_EXCEPTIONS #ifdef CPPHTTPLIB_NO_EXCEPTIONS
return *it; return *it;
#else #else
if (it != files.end()) { return *it; } if (it != items.end()) { return *it; }
throw std::runtime_error("invalid multipart form data name error"); throw std::runtime_error("invalid multipart form data name error");
#endif #endif
} }
@ -3176,65 +3175,72 @@ protected:
}) })
.Post("/multipart", .Post("/multipart",
[&](const Request &req, Response & /*res*/) { [&](const Request &req, Response & /*res*/) {
EXPECT_EQ(6u, req.files.size()); EXPECT_EQ(4u, req.form.get_field_count("text1") +
ASSERT_TRUE(!req.has_file("???")); req.form.get_field_count("text2") +
req.form.get_field_count("file3") +
req.form.get_field_count("file4"));
EXPECT_EQ(2u, req.form.get_file_count("file1") +
req.form.get_file_count("file2"));
ASSERT_TRUE(!req.form.has_file("???"));
ASSERT_TRUE(!req.form.has_field("???"));
ASSERT_TRUE(req.body.empty()); ASSERT_TRUE(req.body.empty());
{ {
const auto &file = req.get_file_value("text1"); const auto &text = req.form.get_field("text1");
EXPECT_TRUE(file.filename.empty()); EXPECT_EQ("text default", text);
EXPECT_EQ("text default", file.content);
} }
{ {
const auto &file = req.get_file_value("text2"); const auto &text = req.form.get_field("text2");
EXPECT_TRUE(file.filename.empty()); EXPECT_EQ("aωb", text);
EXPECT_EQ("aωb", file.content);
} }
{ {
const auto &file = req.get_file_value("file1"); const auto &file = req.form.get_file("file1");
EXPECT_EQ("hello.txt", file.filename); EXPECT_EQ("hello.txt", file.filename);
EXPECT_EQ("text/plain", file.content_type); EXPECT_EQ("text/plain", file.content_type);
EXPECT_EQ("h\ne\n\nl\nl\no\n", file.content); EXPECT_EQ("h\ne\n\nl\nl\no\n", file.content);
} }
{ {
const auto &file = req.get_file_value("file3"); const auto &file = req.form.get_file("file2");
EXPECT_TRUE(file.filename.empty()); EXPECT_EQ("world.json", file.filename);
EXPECT_EQ("application/octet-stream", file.content_type); EXPECT_EQ("application/json", file.content_type);
EXPECT_EQ(0u, file.content.size()); EXPECT_EQ("{\n \"world\", true\n}\n", file.content);
} }
{ {
const auto &file = req.get_file_value("file4"); const auto &text = req.form.get_field("file3");
EXPECT_TRUE(file.filename.empty()); EXPECT_EQ(0u, text.size());
EXPECT_EQ(0u, file.content.size()); }
EXPECT_EQ("application/json tmp-string", file.content_type);
{
const auto &text = req.form.get_field("file4");
EXPECT_EQ(0u, text.size());
} }
}) })
.Post("/multipart/multi_file_values", .Post("/multipart/multi_file_values",
[&](const Request &req, Response & /*res*/) { [&](const Request &req, Response & /*res*/) {
EXPECT_EQ(5u, req.files.size()); EXPECT_EQ(3u, req.form.get_field_count("text") +
ASSERT_TRUE(!req.has_file("???")); req.form.get_field_count("multi_text1"));
EXPECT_EQ(2u, req.form.get_file_count("multi_file1"));
ASSERT_TRUE(!req.form.has_file("???"));
ASSERT_TRUE(!req.form.has_field("???"));
ASSERT_TRUE(req.body.empty()); ASSERT_TRUE(req.body.empty());
{ {
const auto &text_value = req.get_file_values("text"); const auto &text = req.form.get_field("text");
EXPECT_EQ(1u, text_value.size()); EXPECT_EQ("default text", text);
auto &text = text_value[0];
EXPECT_TRUE(text.filename.empty());
EXPECT_EQ("default text", text.content);
} }
{ {
const auto &text1_values = req.get_file_values("multi_text1"); const auto &text1_values = req.form.get_fields("multi_text1");
EXPECT_EQ(2u, text1_values.size()); EXPECT_EQ(2u, text1_values.size());
EXPECT_EQ("aaaaa", text1_values[0].content); EXPECT_EQ("aaaaa", text1_values[0]);
EXPECT_EQ("bbbbb", text1_values[1].content); EXPECT_EQ("bbbbb", text1_values[1]);
} }
{ {
const auto &file1_values = req.get_file_values("multi_file1"); const auto &file1_values = req.form.get_files("multi_file1");
EXPECT_EQ(2u, file1_values.size()); EXPECT_EQ(2u, file1_values.size());
auto file1 = file1_values[0]; auto file1 = file1_values[0];
EXPECT_EQ(file1.filename, "hello.txt"); EXPECT_EQ(file1.filename, "hello.txt");
@ -3349,40 +3355,47 @@ protected:
[&](const Request &req, Response &res, [&](const Request &req, Response &res,
const ContentReader &content_reader) { const ContentReader &content_reader) {
if (req.is_multipart_form_data()) { if (req.is_multipart_form_data()) {
MultipartFormDataItems files; std::vector<FormData> items;
content_reader( content_reader(
[&](const MultipartFormData &file) { [&](const FormData &file) {
files.push_back(file); items.push_back(file);
return true; return true;
}, },
[&](const char *data, size_t data_length) { [&](const char *data, size_t data_length) {
files.back().content.append(data, data_length); items.back().content.append(data, data_length);
return true; return true;
}); });
EXPECT_EQ(5u, files.size()); EXPECT_EQ(5u, items.size());
{ {
const auto &file = get_file_value(files, "text1"); const auto &file = get_file_value(items, "text1");
EXPECT_TRUE(file.filename.empty()); EXPECT_TRUE(file.filename.empty());
EXPECT_EQ("text default", file.content); EXPECT_EQ("text default", file.content);
} }
{ {
const auto &file = get_file_value(files, "text2"); const auto &file = get_file_value(items, "text2");
EXPECT_TRUE(file.filename.empty()); EXPECT_TRUE(file.filename.empty());
EXPECT_EQ("aωb", file.content); EXPECT_EQ("aωb", file.content);
} }
{ {
const auto &file = get_file_value(files, "file1"); const auto &file = get_file_value(items, "file1");
EXPECT_EQ("hello.txt", file.filename); EXPECT_EQ("hello.txt", file.filename);
EXPECT_EQ("text/plain", file.content_type); EXPECT_EQ("text/plain", file.content_type);
EXPECT_EQ("h\ne\n\nl\nl\no\n", file.content); EXPECT_EQ("h\ne\n\nl\nl\no\n", file.content);
} }
{ {
const auto &file = get_file_value(files, "file3"); const auto &file = get_file_value(items, "file2");
EXPECT_EQ("world.json", file.filename);
EXPECT_EQ("application/json", file.content_type);
EXPECT_EQ(R"({\n "world": true\n}\n)", file.content);
}
{
const auto &file = get_file_value(items, "file3");
EXPECT_TRUE(file.filename.empty()); EXPECT_TRUE(file.filename.empty());
EXPECT_EQ("application/octet-stream", file.content_type); EXPECT_EQ("application/octet-stream", file.content_type);
EXPECT_EQ(0u, file.content.size()); EXPECT_EQ(0u, file.content.size());
@ -3496,19 +3509,17 @@ protected:
}) })
.Post("/compress-multipart", .Post("/compress-multipart",
[&](const Request &req, Response & /*res*/) { [&](const Request &req, Response & /*res*/) {
EXPECT_EQ(2u, req.files.size()); EXPECT_EQ(2u, req.form.fields.size());
ASSERT_TRUE(!req.has_file("???")); ASSERT_TRUE(!req.form.has_field("???"));
{ {
const auto &file = req.get_file_value("key1"); const auto &text = req.form.get_field("key1");
EXPECT_TRUE(file.filename.empty()); EXPECT_EQ("test", text);
EXPECT_EQ("test", file.content);
} }
{ {
const auto &file = req.get_file_value("key2"); const auto &text = req.form.get_field("key2");
EXPECT_TRUE(file.filename.empty()); EXPECT_EQ("--abcdefg123", text);
EXPECT_EQ("--abcdefg123", file.content);
} }
}) })
#endif #endif
@ -4431,7 +4442,7 @@ TEST_F(ServerTest, HeaderCountSecurityTest) {
} }
TEST_F(ServerTest, MultipartFormData) { TEST_F(ServerTest, MultipartFormData) {
MultipartFormDataItemsForClientInput items = { UploadFormDataItems items = {
{"text1", "text default", "", ""}, {"text1", "text default", "", ""},
{"text2", "aωb", "", ""}, {"text2", "aωb", "", ""},
{"file1", "h\ne\n\nl\nl\no\n", "hello.txt", "text/plain"}, {"file1", "h\ne\n\nl\nl\no\n", "hello.txt", "text/plain"},
@ -4446,7 +4457,7 @@ TEST_F(ServerTest, MultipartFormData) {
} }
TEST_F(ServerTest, MultipartFormDataMultiFileValues) { TEST_F(ServerTest, MultipartFormDataMultiFileValues) {
MultipartFormDataItemsForClientInput items = { UploadFormDataItems items = {
{"text", "default text", "", ""}, {"text", "default text", "", ""},
{"multi_text1", "aaaaa", "", ""}, {"multi_text1", "aaaaa", "", ""},
@ -5386,11 +5397,11 @@ TEST_F(ServerTest, PostContentReceiver) {
} }
TEST_F(ServerTest, PostMultipartFileContentReceiver) { TEST_F(ServerTest, PostMultipartFileContentReceiver) {
MultipartFormDataItemsForClientInput items = { UploadFormDataItems items = {
{"text1", "text default", "", ""}, {"text1", "text default", "", ""},
{"text2", "aωb", "", ""}, {"text2", "aωb", "", ""},
{"file1", "h\ne\n\nl\nl\no\n", "hello.txt", "text/plain"}, {"file1", "h\ne\n\nl\nl\no\n", "hello.txt", "text/plain"},
{"file2", "{\n \"world\", true\n}\n", "world.json", "application/json"}, {"file2", R"({\n "world": true\n}\n)", "world.json", "application/json"},
{"file3", "", "", "application/octet-stream"}, {"file3", "", "", "application/octet-stream"},
}; };
@ -5401,11 +5412,11 @@ TEST_F(ServerTest, PostMultipartFileContentReceiver) {
} }
TEST_F(ServerTest, PostMultipartPlusBoundary) { TEST_F(ServerTest, PostMultipartPlusBoundary) {
MultipartFormDataItemsForClientInput items = { UploadFormDataItems items = {
{"text1", "text default", "", ""}, {"text1", "text default", "", ""},
{"text2", "aωb", "", ""}, {"text2", "aωb", "", ""},
{"file1", "h\ne\n\nl\nl\no\n", "hello.txt", "text/plain"}, {"file1", "h\ne\n\nl\nl\no\n", "hello.txt", "text/plain"},
{"file2", "{\n \"world\", true\n}\n", "world.json", "application/json"}, {"file2", R"({\n "world": true\n}\n)", "world.json", "application/json"},
{"file3", "", "", "application/octet-stream"}, {"file3", "", "", "application/octet-stream"},
}; };
@ -5860,7 +5871,7 @@ TEST_F(ServerTest, NoGzipWithContentReceiver) {
} }
TEST_F(ServerTest, MultipartFormDataGzip) { TEST_F(ServerTest, MultipartFormDataGzip) {
MultipartFormDataItemsForClientInput items = { UploadFormDataItems items = {
{"key1", "test", "", ""}, {"key1", "test", "", ""},
{"key2", "--abcdefg123", "", ""}, {"key2", "--abcdefg123", "", ""},
}; };
@ -6024,7 +6035,7 @@ TEST_F(ServerTest, NoZstdWithContentReceiver) {
// TODO: How to enable zstd ?? // TODO: How to enable zstd ??
TEST_F(ServerTest, MultipartFormDataZstd) { TEST_F(ServerTest, MultipartFormDataZstd) {
MultipartFormDataItemsForClientInput items = { UploadFormDataItems items = {
{"key1", "test", "", ""}, {"key1", "test", "", ""},
{"key2", "--abcdefg123", "", ""}, {"key2", "--abcdefg123", "", ""},
}; };
@ -6059,7 +6070,9 @@ TEST_F(ServerTest, PutWithContentProviderWithZstd) {
// Pre-compression logging tests // Pre-compression logging tests
TEST_F(ServerTest, PreCompressionLogging) { TEST_F(ServerTest, PreCompressionLogging) {
// Test data for compression (matches the actual /compress endpoint content) // Test data for compression (matches the actual /compress endpoint content)
const std::string test_content = "1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890"; const std::string test_content =
"123456789012345678901234567890123456789012345678901234567890123456789012"
"3456789012345678901234567890";
// Variables to capture logging data // Variables to capture logging data
std::string pre_compression_body; std::string pre_compression_body;
@ -6071,17 +6084,19 @@ TEST_F(ServerTest, PreCompressionLogging) {
std::string post_compression_content_encoding; std::string post_compression_content_encoding;
// Set up pre-compression logger // Set up pre-compression logger
svr_.set_pre_compression_logger([&](const Request &req, const Response &res) { svr_.set_pre_compression_logger([&](const Request & /*req*/,
const Response &res) {
pre_compression_body = res.body; pre_compression_body = res.body;
pre_compression_content_type = res.get_header_value("Content-Type"); pre_compression_content_type = res.get_header_value("Content-Type");
pre_compression_content_encoding = res.get_header_value("Content-Encoding"); pre_compression_content_encoding = res.get_header_value("Content-Encoding");
}); });
// Set up post-compression logger // Set up post-compression logger
svr_.set_logger([&](const Request &req, const Response &res) { svr_.set_logger([&](const Request & /*req*/, const Response &res) {
post_compression_body = res.body; post_compression_body = res.body;
post_compression_content_type = res.get_header_value("Content-Type"); post_compression_content_type = res.get_header_value("Content-Type");
post_compression_content_encoding = res.get_header_value("Content-Encoding"); post_compression_content_encoding =
res.get_header_value("Content-Encoding");
}); });
// Test with gzip compression // Test with gzip compression
@ -6098,10 +6113,12 @@ TEST_F(ServerTest, PreCompressionLogging) {
// Verify pre-compression logger captured uncompressed content // Verify pre-compression logger captured uncompressed content
EXPECT_EQ(test_content, pre_compression_body); EXPECT_EQ(test_content, pre_compression_body);
EXPECT_EQ("text/plain", pre_compression_content_type); EXPECT_EQ("text/plain", pre_compression_content_type);
EXPECT_TRUE(pre_compression_content_encoding.empty()); // No encoding header before compression EXPECT_TRUE(pre_compression_content_encoding
.empty()); // No encoding header before compression
// Verify post-compression logger captured compressed content // Verify post-compression logger captured compressed content
EXPECT_NE(test_content, post_compression_body); // Should be different after compression EXPECT_NE(test_content,
post_compression_body); // Should be different after compression
EXPECT_EQ("text/plain", post_compression_content_type); EXPECT_EQ("text/plain", post_compression_content_type);
EXPECT_EQ("gzip", post_compression_content_encoding); EXPECT_EQ("gzip", post_compression_content_encoding);
@ -6110,16 +6127,19 @@ TEST_F(ServerTest, PreCompressionLogging) {
} }
TEST_F(ServerTest, PreCompressionLoggingWithBrotli) { TEST_F(ServerTest, PreCompressionLoggingWithBrotli) {
const std::string test_content = "1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890"; const std::string test_content =
"123456789012345678901234567890123456789012345678901234567890123456789012"
"3456789012345678901234567890";
std::string pre_compression_body; std::string pre_compression_body;
std::string post_compression_body; std::string post_compression_body;
svr_.set_pre_compression_logger([&](const Request &req, const Response &res) { svr_.set_pre_compression_logger(
[&](const Request & /*req*/, const Response &res) {
pre_compression_body = res.body; pre_compression_body = res.body;
}); });
svr_.set_logger([&](const Request &req, const Response &res) { svr_.set_logger([&](const Request & /*req*/, const Response &res) {
post_compression_body = res.body; post_compression_body = res.body;
}); });
@ -6141,16 +6161,19 @@ TEST_F(ServerTest, PreCompressionLoggingWithBrotli) {
} }
TEST_F(ServerTest, PreCompressionLoggingWithoutCompression) { TEST_F(ServerTest, PreCompressionLoggingWithoutCompression) {
const std::string test_content = "1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890"; const std::string test_content =
"123456789012345678901234567890123456789012345678901234567890123456789012"
"3456789012345678901234567890";
std::string pre_compression_body; std::string pre_compression_body;
std::string post_compression_body; std::string post_compression_body;
svr_.set_pre_compression_logger([&](const Request &req, const Response &res) { svr_.set_pre_compression_logger(
[&](const Request & /*req*/, const Response &res) {
pre_compression_body = res.body; pre_compression_body = res.body;
}); });
svr_.set_logger([&](const Request &req, const Response &res) { svr_.set_logger([&](const Request & /*req*/, const Response &res) {
post_compression_body = res.body; post_compression_body = res.body;
}); });
@ -6163,18 +6186,24 @@ TEST_F(ServerTest, PreCompressionLoggingWithoutCompression) {
EXPECT_TRUE(res->get_header_value("Content-Encoding").empty()); EXPECT_TRUE(res->get_header_value("Content-Encoding").empty());
// Pre-compression logger should not be called when no compression is applied // Pre-compression logger should not be called when no compression is applied
EXPECT_TRUE(pre_compression_body.empty()); // Pre-compression logger not called EXPECT_TRUE(
EXPECT_EQ(test_content, post_compression_body); // Post-compression logger captures final content pre_compression_body.empty()); // Pre-compression logger not called
EXPECT_EQ(
test_content,
post_compression_body); // Post-compression logger captures final content
} }
TEST_F(ServerTest, PreCompressionLoggingOnlyPreLogger) { TEST_F(ServerTest, PreCompressionLoggingOnlyPreLogger) {
const std::string test_content = "1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890"; const std::string test_content =
"123456789012345678901234567890123456789012345678901234567890123456789012"
"3456789012345678901234567890";
std::string pre_compression_body; std::string pre_compression_body;
bool pre_logger_called = false; bool pre_logger_called = false;
// Set only pre-compression logger // Set only pre-compression logger
svr_.set_pre_compression_logger([&](const Request &req, const Response &res) { svr_.set_pre_compression_logger(
[&](const Request & /*req*/, const Response &res) {
pre_compression_body = res.body; pre_compression_body = res.body;
pre_logger_called = true; pre_logger_called = true;
}); });
@ -6767,7 +6796,7 @@ void TestMultipartUploadProgress(SetupHandler &&setup_handler,
Client cli(HOST, PORT); Client cli(HOST, PORT);
vector<uint64_t> progress_values; vector<uint64_t> progress_values;
MultipartFormDataItemsForClientInput items = { UploadFormDataItems items = {
{"field1", "value1", "", ""}, {"field1", "value1", "", ""},
{"field2", "longer value for progress tracking test", "", ""}, {"field2", "longer value for progress tracking test", "", ""},
{"file1", "file content data for upload progress", "test.txt", {"file1", "file content data for upload progress", "test.txt",
@ -6788,12 +6817,11 @@ TEST(UploadProgressTest, PostMultipartProgress) {
TestMultipartUploadProgress( TestMultipartUploadProgress(
[](Server &svr) { [](Server &svr) {
svr.Post("/multipart", [](const Request &req, Response &res) { svr.Post("/multipart", [](const Request &req, Response &res) {
EXPECT_FALSE(req.files.empty()); EXPECT_TRUE(!req.form.files.empty() || !req.form.fields.empty());
res.set_content("multipart received", "text/plain"); res.set_content("multipart received", "text/plain");
}); });
}, },
[](Client &cli, const string &endpoint, [](Client &cli, const string &endpoint, const UploadFormDataItems &items,
const MultipartFormDataItemsForClientInput &items,
UploadProgress progress_callback) { UploadProgress progress_callback) {
return cli.Post(endpoint, items, progress_callback); return cli.Post(endpoint, items, progress_callback);
}, },
@ -8631,26 +8659,26 @@ TEST(MultipartFormDataTest, LargeData) {
svr.Post("/post", [&](const Request &req, Response & /*res*/, svr.Post("/post", [&](const Request &req, Response & /*res*/,
const ContentReader &content_reader) { const ContentReader &content_reader) {
if (req.is_multipart_form_data()) { if (req.is_multipart_form_data()) {
MultipartFormDataItems files; std::vector<FormData> items;
content_reader( content_reader(
[&](const MultipartFormData &file) { [&](const FormData &file) {
files.push_back(file); items.push_back(file);
return true; return true;
}, },
[&](const char *data, size_t data_length) { [&](const char *data, size_t data_length) {
files.back().content.append(data, data_length); items.back().content.append(data, data_length);
return true; return true;
}); });
EXPECT_TRUE(std::string(files[0].name) == "document"); EXPECT_TRUE(std::string(items[0].name) == "document");
EXPECT_EQ(size_t(1024 * 1024 * 2), files[0].content.size()); EXPECT_EQ(size_t(1024 * 1024 * 2), items[0].content.size());
EXPECT_TRUE(files[0].filename == "2MB_data"); EXPECT_TRUE(items[0].filename == "2MB_data");
EXPECT_TRUE(files[0].content_type == "application/octet-stream"); EXPECT_TRUE(items[0].content_type == "application/octet-stream");
EXPECT_TRUE(files[1].name == "hello"); EXPECT_TRUE(items[1].name == "hello");
EXPECT_TRUE(files[1].content == "world"); EXPECT_TRUE(items[1].content == "world");
EXPECT_TRUE(files[1].filename == ""); EXPECT_TRUE(items[1].filename == "");
EXPECT_TRUE(files[1].content_type == ""); EXPECT_TRUE(items[1].content_type == "");
} else { } else {
std::string body; std::string body;
content_reader([&](const char *data, size_t data_length) { content_reader([&](const char *data, size_t data_length) {
@ -8677,7 +8705,7 @@ TEST(MultipartFormDataTest, LargeData) {
Client cli("https://localhost:8080"); Client cli("https://localhost:8080");
cli.enable_server_certificate_verification(false); cli.enable_server_certificate_verification(false);
MultipartFormDataItemsForClientInput items{ UploadFormDataItems items{
{"document", buffer.str(), "2MB_data", "application/octet-stream"}, {"document", buffer.str(), "2MB_data", "application/octet-stream"},
{"hello", "world", "", ""}, {"hello", "world", "", ""},
}; };
@ -8719,92 +8747,92 @@ TEST(MultipartFormDataTest, DataProviderItems) {
svr.Post("/post-items", [&](const Request &req, Response & /*res*/, svr.Post("/post-items", [&](const Request &req, Response & /*res*/,
const ContentReader &content_reader) { const ContentReader &content_reader) {
ASSERT_TRUE(req.is_multipart_form_data()); ASSERT_TRUE(req.is_multipart_form_data());
MultipartFormDataItems files; std::vector<FormData> items;
content_reader( content_reader(
[&](const MultipartFormData &file) { [&](const FormData &file) {
files.push_back(file); items.push_back(file);
return true; return true;
}, },
[&](const char *data, size_t data_length) { [&](const char *data, size_t data_length) {
files.back().content.append(data, data_length); items.back().content.append(data, data_length);
return true; return true;
}); });
ASSERT_TRUE(files.size() == 2); ASSERT_TRUE(items.size() == 2);
EXPECT_EQ(std::string(files[0].name), "name1"); EXPECT_EQ(std::string(items[0].name), "name1");
EXPECT_EQ(files[0].content, "Testing123"); EXPECT_EQ(items[0].content, "Testing123");
EXPECT_EQ(files[0].filename, "filename1"); EXPECT_EQ(items[0].filename, "filename1");
EXPECT_EQ(files[0].content_type, "application/octet-stream"); EXPECT_EQ(items[0].content_type, "application/octet-stream");
EXPECT_EQ(files[1].name, "name2"); EXPECT_EQ(items[1].name, "name2");
EXPECT_EQ(files[1].content, "Testing456"); EXPECT_EQ(items[1].content, "Testing456");
EXPECT_EQ(files[1].filename, ""); EXPECT_EQ(items[1].filename, "");
EXPECT_EQ(files[1].content_type, ""); EXPECT_EQ(items[1].content_type, "");
}); });
svr.Post("/post-providers", [&](const Request &req, Response & /*res*/, svr.Post("/post-providers", [&](const Request &req, Response & /*res*/,
const ContentReader &content_reader) { const ContentReader &content_reader) {
ASSERT_TRUE(req.is_multipart_form_data()); ASSERT_TRUE(req.is_multipart_form_data());
MultipartFormDataItems files; std::vector<FormData> items;
content_reader( content_reader(
[&](const MultipartFormData &file) { [&](const FormData &file) {
files.push_back(file); items.push_back(file);
return true; return true;
}, },
[&](const char *data, size_t data_length) { [&](const char *data, size_t data_length) {
files.back().content.append(data, data_length); items.back().content.append(data, data_length);
return true; return true;
}); });
ASSERT_TRUE(files.size() == 2); ASSERT_TRUE(items.size() == 2);
EXPECT_EQ(files[0].name, "name3"); EXPECT_EQ(items[0].name, "name3");
EXPECT_EQ(files[0].content, rand1); EXPECT_EQ(items[0].content, rand1);
EXPECT_EQ(files[0].filename, "filename3"); EXPECT_EQ(items[0].filename, "filename3");
EXPECT_EQ(files[0].content_type, ""); EXPECT_EQ(items[0].content_type, "");
EXPECT_EQ(files[1].name, "name4"); EXPECT_EQ(items[1].name, "name4");
EXPECT_EQ(files[1].content, rand2); EXPECT_EQ(items[1].content, rand2);
EXPECT_EQ(files[1].filename, "filename4"); EXPECT_EQ(items[1].filename, "filename4");
EXPECT_EQ(files[1].content_type, ""); EXPECT_EQ(items[1].content_type, "");
}); });
svr.Post("/post-both", [&](const Request &req, Response & /*res*/, svr.Post("/post-both", [&](const Request &req, Response & /*res*/,
const ContentReader &content_reader) { const ContentReader &content_reader) {
ASSERT_TRUE(req.is_multipart_form_data()); ASSERT_TRUE(req.is_multipart_form_data());
MultipartFormDataItems files; std::vector<FormData> items;
content_reader( content_reader(
[&](const MultipartFormData &file) { [&](const FormData &file) {
files.push_back(file); items.push_back(file);
return true; return true;
}, },
[&](const char *data, size_t data_length) { [&](const char *data, size_t data_length) {
files.back().content.append(data, data_length); items.back().content.append(data, data_length);
return true; return true;
}); });
ASSERT_TRUE(files.size() == 4); ASSERT_TRUE(items.size() == 4);
EXPECT_EQ(std::string(files[0].name), "name1"); EXPECT_EQ(std::string(items[0].name), "name1");
EXPECT_EQ(files[0].content, "Testing123"); EXPECT_EQ(items[0].content, "Testing123");
EXPECT_EQ(files[0].filename, "filename1"); EXPECT_EQ(items[0].filename, "filename1");
EXPECT_EQ(files[0].content_type, "application/octet-stream"); EXPECT_EQ(items[0].content_type, "application/octet-stream");
EXPECT_EQ(files[1].name, "name2"); EXPECT_EQ(items[1].name, "name2");
EXPECT_EQ(files[1].content, "Testing456"); EXPECT_EQ(items[1].content, "Testing456");
EXPECT_EQ(files[1].filename, ""); EXPECT_EQ(items[1].filename, "");
EXPECT_EQ(files[1].content_type, ""); EXPECT_EQ(items[1].content_type, "");
EXPECT_EQ(files[2].name, "name3"); EXPECT_EQ(items[2].name, "name3");
EXPECT_EQ(files[2].content, rand1); EXPECT_EQ(items[2].content, rand1);
EXPECT_EQ(files[2].filename, "filename3"); EXPECT_EQ(items[2].filename, "filename3");
EXPECT_EQ(files[2].content_type, ""); EXPECT_EQ(items[2].content_type, "");
EXPECT_EQ(files[3].name, "name4"); EXPECT_EQ(items[3].name, "name4");
EXPECT_EQ(files[3].content, rand2); EXPECT_EQ(items[3].content, rand2);
EXPECT_EQ(files[3].filename, "filename4"); EXPECT_EQ(items[3].filename, "filename4");
EXPECT_EQ(files[3].content_type, ""); EXPECT_EQ(items[3].content_type, "");
}); });
auto t = std::thread([&]() { svr.listen("localhost", 8080); }); auto t = std::thread([&]() { svr.listen("localhost", 8080); });
@ -8820,7 +8848,7 @@ TEST(MultipartFormDataTest, DataProviderItems) {
Client cli("https://localhost:8080"); Client cli("https://localhost:8080");
cli.enable_server_certificate_verification(false); cli.enable_server_certificate_verification(false);
MultipartFormDataItemsForClientInput items{ UploadFormDataItems items{
{"name1", "Testing123", "filename1", "application/octet-stream"}, {"name1", "Testing123", "filename1", "application/octet-stream"},
{"name2", "Testing456", "", ""}, // not a file {"name2", "Testing456", "", ""}, // not a file
}; };
@ -8831,7 +8859,7 @@ TEST(MultipartFormDataTest, DataProviderItems) {
ASSERT_EQ(StatusCode::OK_200, res->status); ASSERT_EQ(StatusCode::OK_200, res->status);
} }
MultipartFormDataProviderItems providers; FormDataProviderItems providers;
{ {
auto res = auto res =
@ -8979,26 +9007,26 @@ TEST(MultipartFormDataTest, PostCustomBoundary) {
svr.Post("/post_customboundary", [&](const Request &req, Response & /*res*/, svr.Post("/post_customboundary", [&](const Request &req, Response & /*res*/,
const ContentReader &content_reader) { const ContentReader &content_reader) {
if (req.is_multipart_form_data()) { if (req.is_multipart_form_data()) {
MultipartFormDataItems files; std::vector<FormData> items;
content_reader( content_reader(
[&](const MultipartFormData &file) { [&](const FormData &file) {
files.push_back(file); items.push_back(file);
return true; return true;
}, },
[&](const char *data, size_t data_length) { [&](const char *data, size_t data_length) {
files.back().content.append(data, data_length); items.back().content.append(data, data_length);
return true; return true;
}); });
EXPECT_TRUE(std::string(files[0].name) == "document"); EXPECT_TRUE(std::string(items[0].name) == "document");
EXPECT_EQ(size_t(1024 * 1024 * 2), files[0].content.size()); EXPECT_EQ(size_t(1024 * 1024 * 2), items[0].content.size());
EXPECT_TRUE(files[0].filename == "2MB_data"); EXPECT_TRUE(items[0].filename == "2MB_data");
EXPECT_TRUE(files[0].content_type == "application/octet-stream"); EXPECT_TRUE(items[0].content_type == "application/octet-stream");
EXPECT_TRUE(files[1].name == "hello"); EXPECT_TRUE(items[1].name == "hello");
EXPECT_TRUE(files[1].content == "world"); EXPECT_TRUE(items[1].content == "world");
EXPECT_TRUE(files[1].filename == ""); EXPECT_TRUE(items[1].filename == "");
EXPECT_TRUE(files[1].content_type == ""); EXPECT_TRUE(items[1].content_type == "");
} else { } else {
std::string body; std::string body;
content_reader([&](const char *data, size_t data_length) { content_reader([&](const char *data, size_t data_length) {
@ -9025,7 +9053,7 @@ TEST(MultipartFormDataTest, PostCustomBoundary) {
Client cli("https://localhost:8080"); Client cli("https://localhost:8080");
cli.enable_server_certificate_verification(false); cli.enable_server_certificate_verification(false);
MultipartFormDataItemsForClientInput items{ UploadFormDataItems items{
{"document", buffer.str(), "2MB_data", "application/octet-stream"}, {"document", buffer.str(), "2MB_data", "application/octet-stream"},
{"hello", "world", "", ""}, {"hello", "world", "", ""},
}; };
@ -9043,7 +9071,7 @@ TEST(MultipartFormDataTest, PostInvalidBoundaryChars) {
Client cli("https://localhost:8080"); Client cli("https://localhost:8080");
MultipartFormDataItemsForClientInput items{ UploadFormDataItems items{
{"document", buffer.str(), "2MB_data", "application/octet-stream"}, {"document", buffer.str(), "2MB_data", "application/octet-stream"},
{"hello", "world", "", ""}, {"hello", "world", "", ""},
}; };
@ -9062,26 +9090,26 @@ TEST(MultipartFormDataTest, PutFormData) {
svr.Put("/put", [&](const Request &req, const Response & /*res*/, svr.Put("/put", [&](const Request &req, const Response & /*res*/,
const ContentReader &content_reader) { const ContentReader &content_reader) {
if (req.is_multipart_form_data()) { if (req.is_multipart_form_data()) {
MultipartFormDataItems files; std::vector<FormData> items;
content_reader( content_reader(
[&](const MultipartFormData &file) { [&](const FormData &file) {
files.push_back(file); items.push_back(file);
return true; return true;
}, },
[&](const char *data, size_t data_length) { [&](const char *data, size_t data_length) {
files.back().content.append(data, data_length); items.back().content.append(data, data_length);
return true; return true;
}); });
EXPECT_TRUE(std::string(files[0].name) == "document"); EXPECT_TRUE(std::string(items[0].name) == "document");
EXPECT_EQ(size_t(1024 * 1024 * 2), files[0].content.size()); EXPECT_EQ(size_t(1024 * 1024 * 2), items[0].content.size());
EXPECT_TRUE(files[0].filename == "2MB_data"); EXPECT_TRUE(items[0].filename == "2MB_data");
EXPECT_TRUE(files[0].content_type == "application/octet-stream"); EXPECT_TRUE(items[0].content_type == "application/octet-stream");
EXPECT_TRUE(files[1].name == "hello"); EXPECT_TRUE(items[1].name == "hello");
EXPECT_TRUE(files[1].content == "world"); EXPECT_TRUE(items[1].content == "world");
EXPECT_TRUE(files[1].filename == ""); EXPECT_TRUE(items[1].filename == "");
EXPECT_TRUE(files[1].content_type == ""); EXPECT_TRUE(items[1].content_type == "");
} else { } else {
std::string body; std::string body;
content_reader([&](const char *data, size_t data_length) { content_reader([&](const char *data, size_t data_length) {
@ -9108,7 +9136,7 @@ TEST(MultipartFormDataTest, PutFormData) {
Client cli("https://localhost:8080"); Client cli("https://localhost:8080");
cli.enable_server_certificate_verification(false); cli.enable_server_certificate_verification(false);
MultipartFormDataItemsForClientInput items{ UploadFormDataItems items{
{"document", buffer.str(), "2MB_data", "application/octet-stream"}, {"document", buffer.str(), "2MB_data", "application/octet-stream"},
{"hello", "world", "", ""}, {"hello", "world", "", ""},
}; };
@ -9126,26 +9154,26 @@ TEST(MultipartFormDataTest, PutFormDataCustomBoundary) {
[&](const Request &req, const Response & /*res*/, [&](const Request &req, const Response & /*res*/,
const ContentReader &content_reader) { const ContentReader &content_reader) {
if (req.is_multipart_form_data()) { if (req.is_multipart_form_data()) {
MultipartFormDataItems files; std::vector<FormData> items;
content_reader( content_reader(
[&](const MultipartFormData &file) { [&](const FormData &file) {
files.push_back(file); items.push_back(file);
return true; return true;
}, },
[&](const char *data, size_t data_length) { [&](const char *data, size_t data_length) {
files.back().content.append(data, data_length); items.back().content.append(data, data_length);
return true; return true;
}); });
EXPECT_TRUE(std::string(files[0].name) == "document"); EXPECT_TRUE(std::string(items[0].name) == "document");
EXPECT_EQ(size_t(1024 * 1024 * 2), files[0].content.size()); EXPECT_EQ(size_t(1024 * 1024 * 2), items[0].content.size());
EXPECT_TRUE(files[0].filename == "2MB_data"); EXPECT_TRUE(items[0].filename == "2MB_data");
EXPECT_TRUE(files[0].content_type == "application/octet-stream"); EXPECT_TRUE(items[0].content_type == "application/octet-stream");
EXPECT_TRUE(files[1].name == "hello"); EXPECT_TRUE(items[1].name == "hello");
EXPECT_TRUE(files[1].content == "world"); EXPECT_TRUE(items[1].content == "world");
EXPECT_TRUE(files[1].filename == ""); EXPECT_TRUE(items[1].filename == "");
EXPECT_TRUE(files[1].content_type == ""); EXPECT_TRUE(items[1].content_type == "");
} else { } else {
std::string body; std::string body;
content_reader([&](const char *data, size_t data_length) { content_reader([&](const char *data, size_t data_length) {
@ -9172,7 +9200,7 @@ TEST(MultipartFormDataTest, PutFormDataCustomBoundary) {
Client cli("https://localhost:8080"); Client cli("https://localhost:8080");
cli.enable_server_certificate_verification(false); cli.enable_server_certificate_verification(false);
MultipartFormDataItemsForClientInput items{ UploadFormDataItems items{
{"document", buffer.str(), "2MB_data", "application/octet-stream"}, {"document", buffer.str(), "2MB_data", "application/octet-stream"},
{"hello", "world", "", ""}, {"hello", "world", "", ""},
}; };
@ -9191,7 +9219,7 @@ TEST(MultipartFormDataTest, PutInvalidBoundaryChars) {
Client cli("https://localhost:8080"); Client cli("https://localhost:8080");
cli.enable_server_certificate_verification(false); cli.enable_server_certificate_verification(false);
MultipartFormDataItemsForClientInput items{ UploadFormDataItems items{
{"document", buffer.str(), "2MB_data", "application/octet-stream"}, {"document", buffer.str(), "2MB_data", "application/octet-stream"},
{"hello", "world", "", ""}, {"hello", "world", "", ""},
}; };
@ -9208,26 +9236,26 @@ TEST(MultipartFormDataTest, AlternateFilename) {
Server svr; Server svr;
svr.Post("/test", [&](const Request &req, Response &res) { svr.Post("/test", [&](const Request &req, Response &res) {
ASSERT_EQ(3u, req.files.size()); ASSERT_EQ(2u, req.form.files.size());
ASSERT_EQ(1u, req.form.fields.size());
auto it = req.files.begin(); // Test files
ASSERT_EQ("file1", it->second.name); const auto &file1 = req.form.get_file("file1");
ASSERT_EQ("A.txt", it->second.filename); ASSERT_EQ("file1", file1.name);
ASSERT_EQ("text/plain", it->second.content_type); ASSERT_EQ("A.txt", file1.filename);
ASSERT_EQ("Content of a.txt.\r\n", it->second.content); ASSERT_EQ("text/plain", file1.content_type);
ASSERT_EQ("Content of a.txt.\r\n", file1.content);
++it; const auto &file2 = req.form.get_file("file2");
ASSERT_EQ("file2", it->second.name); ASSERT_EQ("file2", file2.name);
ASSERT_EQ("a.html", it->second.filename); ASSERT_EQ("a.html", file2.filename);
ASSERT_EQ("text/html", it->second.content_type); ASSERT_EQ("text/html", file2.content_type);
ASSERT_EQ("<!DOCTYPE html><title>Content of a.html.</title>\r\n", ASSERT_EQ("<!DOCTYPE html><title>Content of a.html.</title>\r\n",
it->second.content); file2.content);
++it; // Test text field
ASSERT_EQ("text", it->second.name); const auto &text = req.form.get_field("text");
ASSERT_EQ("", it->second.filename); ASSERT_EQ("text default", text);
ASSERT_EQ("", it->second.content_type);
ASSERT_EQ("text default", it->second.content);
res.set_content("ok", "text/plain"); res.set_content("ok", "text/plain");
@ -9276,15 +9304,13 @@ TEST(MultipartFormDataTest, CloseDelimiterWithoutCRLF) {
Server svr; Server svr;
svr.Post("/test", [&](const Request &req, Response &) { svr.Post("/test", [&](const Request &req, Response &) {
ASSERT_EQ(2u, req.files.size()); ASSERT_EQ(2u, req.form.fields.size());
auto it = req.files.begin(); const auto &text1 = req.form.get_field("text1");
ASSERT_EQ("text1", it->second.name); ASSERT_EQ("text1", text1);
ASSERT_EQ("text1", it->second.content);
++it; const auto &text2 = req.form.get_field("text2");
ASSERT_EQ("text2", it->second.name); ASSERT_EQ("text2", text2);
ASSERT_EQ("text2", it->second.content);
handled = true; handled = true;
}); });
@ -9322,15 +9348,13 @@ TEST(MultipartFormDataTest, ContentLength) {
Server svr; Server svr;
svr.Post("/test", [&](const Request &req, Response &) { svr.Post("/test", [&](const Request &req, Response &) {
ASSERT_EQ(2u, req.files.size()); ASSERT_EQ(2u, req.form.fields.size());
auto it = req.files.begin(); const auto &text1 = req.form.get_field("text1");
ASSERT_EQ("text1", it->second.name); ASSERT_EQ("text1", text1);
ASSERT_EQ("text1", it->second.content);
++it; const auto &text2 = req.form.get_field("text2");
ASSERT_EQ("text2", it->second.name); ASSERT_EQ("text2", text2);
ASSERT_EQ("text2", it->second.content);
handled = true; handled = true;
}); });
@ -9369,26 +9393,22 @@ TEST(MultipartFormDataTest, AccessPartHeaders) {
Server svr; Server svr;
svr.Post("/test", [&](const Request &req, Response &) { svr.Post("/test", [&](const Request &req, Response &) {
ASSERT_EQ(2u, req.files.size()); ASSERT_EQ(2u, req.form.fields.size());
auto it = req.files.begin(); const auto &text1 = req.form.get_field("text1");
ASSERT_EQ("text1", it->second.name); ASSERT_EQ("text1", text1);
ASSERT_EQ("text1", it->second.content); // TODO: Add header access for text fields if needed
ASSERT_EQ(1U, it->second.headers.count("Content-Length"));
auto content_length = it->second.headers.find("CONTENT-length");
ASSERT_EQ("5", content_length->second);
ASSERT_EQ(3U, it->second.headers.size());
++it; const auto &text2 = req.form.get_field("text2");
ASSERT_EQ("text2", it->second.name); ASSERT_EQ("text2", text2);
ASSERT_EQ("text2", it->second.content); // TODO: Header access for text fields needs to be implemented
auto &headers = it->second.headers; // auto &headers = it->second.headers;
ASSERT_EQ(3U, headers.size()); // ASSERT_EQ(3U, headers.size());
auto custom_header = headers.find("x-whatever"); // auto custom_header = headers.find("x-whatever");
ASSERT_TRUE(custom_header != headers.end()); // ASSERT_TRUE(custom_header != headers.end());
ASSERT_NE("customvalue", custom_header->second); // ASSERT_NE("customvalue", custom_header->second);
ASSERT_EQ("CustomValue", custom_header->second); // ASSERT_EQ("CustomValue", custom_header->second);
ASSERT_TRUE(headers.find("X-Test") == headers.end()); // text1 header // ASSERT_TRUE(headers.find("X-Test") == headers.end()); // text1 header
handled = true; handled = true;
}); });
@ -9432,11 +9452,10 @@ TEST(MultipartFormDataTest, LargeHeader) {
Server svr; Server svr;
svr.Post("/test", [&](const Request &req, Response &) { svr.Post("/test", [&](const Request &req, Response &) {
ASSERT_EQ(1u, req.files.size()); ASSERT_EQ(1u, req.form.fields.size());
auto it = req.files.begin(); const auto &text = req.form.get_field("name1");
ASSERT_EQ("name1", it->second.name); ASSERT_EQ("text1", text);
ASSERT_EQ("text1", it->second.content);
handled = true; handled = true;
}); });