lib/cpp-httplib
1
0
Fork 0
mirror of synced 2025-06-12 07:41:53 +03:00
Code Activity

Fix #319

Browse Source
This commit is contained in:
yhirose
2020-01-07 23:27:29 -05:00
parent 68aeb4a06a
commit 2f72845008
2 changed files with 48 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
httplib.h
View File

@ -1897,24 +1897,20 @@ inline ssize_t write_content(Stream &strm, ContentProvider content_provider,
};
data_sink.done = [&](void) { written_length = -1; };
content_provider(offset, end_offset - offset,
// [&](const char *d, size_t l) {
// offset += l;
// written_length = strm.write(d, l);
// },
// [&](void) { written_length = -1; }
data_sink);
content_provider(offset, end_offset - offset, data_sink);
if (written_length < 0) { return written_length; }
}
return static_cast<ssize_t>(offset - begin_offset);
}
template <typename T>
inline ssize_t write_content_chunked(Stream &strm,
ContentProvider content_provider) {
ContentProvider content_provider,
T is_shutting_down) {
size_t offset = 0;
auto data_available = true;
ssize_t total_written_length = 0;
while (data_available) {
while (data_available && !is_shutting_down()) {
ssize_t written_length = 0;
DataSink data_sink;
@ -1931,21 +1927,7 @@ inline ssize_t write_content_chunked(Stream &strm,
written_length = strm.write("0\r\n\r\n");
};
content_provider(
offset, 0,
// [&](const char *d, size_t l) {
// data_available = l > 0;
// offset += l;
//
// // Emit chunked response header and footer for each chunk
// auto chunk = from_i_to_hex(l) + "\r\n" + std::string(d, l) +
// "\r\n"; written_length = strm.write(chunk);
// },
// [&](void) {
// data_available = false;
// written_length = strm.write("0\r\n\r\n");
// }
data_sink);
content_provider(offset, 0, data_sink);
if (written_length < 0) { return written_length; }
total_written_length += written_length;
@ -3088,7 +3070,8 @@ Server::write_content_with_provider(Stream &strm, const Request &req,
}
}
} else {
if (detail::write_content_chunked(strm, res.content_provider) < 0) {
auto is_shutting_down = [this]() { return this->svr_sock_ == INVALID_SOCKET; };
if (detail::write_content_chunked(strm, res.content_provider, is_shutting_down) < 0) {
return false;
}
}