mirror of
https://github.com/Xahau/xahaud.git
synced 2025-12-06 17:27:52 +00:00
d8dea963fa
6d5547a Set version to 1.0.0-b34 6fab138 Fix and tidy up CMake build scripts: ccefa54 Set version to 1.0.0-b33 32afe41 Set internal state correctly when writing frames: fe3e20b Add write_frames unit test 578dcd0 Add decorator unit test aaa3733 Use fwrite return value in file_body df66165 Require Visual Studio 2015 Update 3 or later b8e5a21 Set version to 1.0.0-b32 ffb1758 Update CMake scripts for finding packages: b893749 Remove http Writer suspend and resume feature (API Change): 27864fb Add io_service completion invariants tests eba05a7 Set version to 1.0.0-b31 484bcef Fix badge markdown in README.md 5663bea Add missing dynabuf_readstream member 0d7a551 Tidy up build settings 0fd4030 Move the handler, don't copy it git-subtree-dir: src/beast git-subtree-split: 6d5547a32c50ec95832c4779311502555ab0ee1f
98 lines
2.3 KiB
C++
98 lines
2.3 KiB
C++
//
|
|
// Copyright (c) 2013-2017 Vinnie Falco (vinnie dot falco at gmail dot com)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
#ifndef BEAST_EXAMPLE_FILE_BODY_H_INCLUDED
|
|
#define BEAST_EXAMPLE_FILE_BODY_H_INCLUDED
|
|
|
|
#include <beast/core/error.hpp>
|
|
#include <beast/http/message.hpp>
|
|
#include <boost/asio/buffer.hpp>
|
|
#include <boost/assert.hpp>
|
|
#include <boost/filesystem.hpp>
|
|
#include <cstdio>
|
|
#include <cstdint>
|
|
|
|
namespace beast {
|
|
namespace http {
|
|
|
|
struct file_body
|
|
{
|
|
using value_type = std::string;
|
|
|
|
class writer
|
|
{
|
|
std::uint64_t size_ = 0;
|
|
std::uint64_t offset_ = 0;
|
|
std::string const& path_;
|
|
FILE* file_ = nullptr;
|
|
char buf_[4096];
|
|
std::size_t buf_len_;
|
|
|
|
public:
|
|
writer(writer const&) = delete;
|
|
writer& operator=(writer const&) = delete;
|
|
|
|
template<bool isRequest, class Fields>
|
|
writer(message<isRequest,
|
|
file_body, Fields> const& m) noexcept
|
|
: path_(m.body)
|
|
{
|
|
}
|
|
|
|
~writer()
|
|
{
|
|
if(file_)
|
|
fclose(file_);
|
|
}
|
|
|
|
void
|
|
init(error_code& ec) noexcept
|
|
{
|
|
file_ = fopen(path_.c_str(), "rb");
|
|
if(! file_)
|
|
ec = error_code{errno,
|
|
system_category()};
|
|
else
|
|
size_ = boost::filesystem::file_size(path_);
|
|
}
|
|
|
|
std::uint64_t
|
|
content_length() const noexcept
|
|
{
|
|
return size_;
|
|
}
|
|
|
|
template<class WriteFunction>
|
|
bool
|
|
write(error_code& ec, WriteFunction&& wf) noexcept
|
|
{
|
|
if(size_ - offset_ < sizeof(buf_))
|
|
buf_len_ = static_cast<std::size_t>(
|
|
size_ - offset_);
|
|
else
|
|
buf_len_ = sizeof(buf_);
|
|
auto const nread = fread(
|
|
buf_, 1, sizeof(buf_), file_);
|
|
if(ferror(file_))
|
|
{
|
|
ec = error_code(errno,
|
|
system_category());
|
|
return true;
|
|
}
|
|
BOOST_ASSERT(nread != 0);
|
|
offset_ += nread;
|
|
wf(boost::asio::buffer(buf_, nread));
|
|
return offset_ >= size_;
|
|
}
|
|
};
|
|
};
|
|
|
|
} // http
|
|
} // beast
|
|
|
|
#endif
|