You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
208 lines
4.9 KiB
C++
208 lines
4.9 KiB
C++
#ifndef WEBCC_BODY_H_
|
|
#define WEBCC_BODY_H_
|
|
|
|
#include <fstream>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
|
|
#include "boost/filesystem/fstream.hpp"
|
|
#include "boost/filesystem/path.hpp"
|
|
|
|
#include "webcc/common.h"
|
|
|
|
namespace webcc {
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
class Body {
|
|
public:
|
|
Body() = default;
|
|
virtual ~Body() = default;
|
|
|
|
// Get the size in bytes of the body.
|
|
virtual std::size_t GetSize() const {
|
|
return 0;
|
|
}
|
|
|
|
bool IsEmpty() const {
|
|
return GetSize() == 0;
|
|
}
|
|
|
|
#ifdef WEBCC_ENABLE_GZIP
|
|
|
|
// Compress the data with Gzip.
|
|
// If data size <= threshold (1400 bytes), no compression will be taken
|
|
// and false will be simply returned.
|
|
virtual bool Compress() {
|
|
return false;
|
|
}
|
|
|
|
// Decompress the data.
|
|
virtual bool Decompress() {
|
|
return false;
|
|
}
|
|
|
|
#endif // WEBCC_ENABLE_GZIP
|
|
|
|
// Initialize the payload for iteration.
|
|
// Usage:
|
|
// InitPayload();
|
|
// for (auto p = NextPayload(); !p.empty(); p = NextPayload()) {
|
|
// }
|
|
virtual void InitPayload() {
|
|
}
|
|
|
|
// Get the next payload.
|
|
// An empty payload returned indicates the end.
|
|
virtual Payload NextPayload(bool free_previous = false) {
|
|
return {};
|
|
}
|
|
|
|
// Dump to output stream for logging purpose.
|
|
virtual void Dump(std::ostream& os, const std::string& prefix) const {
|
|
}
|
|
};
|
|
|
|
using BodyPtr = std::shared_ptr<Body>;
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
class StringBody : public Body {
|
|
public:
|
|
explicit StringBody(const std::string& data, bool compressed)
|
|
: data_(data), compressed_(compressed) {
|
|
}
|
|
|
|
explicit StringBody(std::string&& data, bool compressed)
|
|
: data_(std::move(data)), compressed_(compressed) {
|
|
}
|
|
|
|
std::size_t GetSize() const override {
|
|
return data_.size();
|
|
}
|
|
|
|
const std::string& data() const {
|
|
return data_;
|
|
}
|
|
|
|
bool compressed() const {
|
|
return compressed_;
|
|
}
|
|
|
|
#ifdef WEBCC_ENABLE_GZIP
|
|
|
|
bool Compress() override;
|
|
|
|
bool Decompress() override;
|
|
|
|
#endif // WEBCC_ENABLE_GZIP
|
|
|
|
void InitPayload() override;
|
|
|
|
Payload NextPayload(bool free_previous = false) override;
|
|
|
|
void Dump(std::ostream& os, const std::string& prefix) const override;
|
|
|
|
private:
|
|
std::string data_;
|
|
|
|
// Is the data compressed?
|
|
bool compressed_;
|
|
|
|
// Index for (not really) iterating the payload.
|
|
std::size_t index_ = 0;
|
|
};
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
// Multi-part form body for request.
|
|
class FormBody : public Body {
|
|
public:
|
|
FormBody(const std::vector<FormPartPtr>& parts, const std::string& boundary);
|
|
|
|
std::size_t GetSize() const override;
|
|
|
|
const std::vector<FormPartPtr>& parts() const {
|
|
return parts_;
|
|
}
|
|
|
|
void InitPayload() override;
|
|
|
|
Payload NextPayload(bool free_previous = false) override;
|
|
|
|
void Dump(std::ostream& os, const std::string& prefix) const override;
|
|
|
|
private:
|
|
void AddBoundary(Payload* payload);
|
|
void AddBoundaryEnd(Payload* payload);
|
|
|
|
void Free(std::size_t index);
|
|
|
|
private:
|
|
std::vector<FormPartPtr> parts_;
|
|
std::string boundary_;
|
|
|
|
// Index for iterating the payload.
|
|
std::size_t index_ = 0;
|
|
};
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
// File body for server to serve a file without loading the whole of it into
|
|
// the memory.
|
|
class FileBody : public Body {
|
|
public:
|
|
// For message to be sent out.
|
|
FileBody(const boost::filesystem::path& path, std::size_t chunk_size);
|
|
|
|
// For message received.
|
|
// No |chunk_size| is needed since you don't iterate the payload of a
|
|
// received message.
|
|
// If |auto_delete| is true, the file will be deleted on destructor unless it
|
|
// is moved to another path (see Move()).
|
|
FileBody(const boost::filesystem::path& path, bool auto_delete = false);
|
|
|
|
~FileBody() override;
|
|
|
|
std::size_t GetSize() const override {
|
|
return size_;
|
|
}
|
|
|
|
void InitPayload() override;
|
|
|
|
Payload NextPayload(bool free_previous = false) override;
|
|
|
|
void Dump(std::ostream& os, const std::string& prefix) const override;
|
|
|
|
const boost::filesystem::path& path() const {
|
|
return path_;
|
|
}
|
|
|
|
// Move (or rename) the file.
|
|
// Used to move the streamed file of the received message to a new place.
|
|
// Applicable to both client and server.
|
|
// After move, the original path will be reset to empty.
|
|
// If |new_path| and |path_| resolve to the same file, do nothing and just
|
|
// return false.
|
|
// If |new_path| resolves to an existing non-directory file, it is removed.
|
|
// If |new_path| resolves to an existing directory, it is removed if empty
|
|
// on ISO/IEC 9945 but is an error on Windows.
|
|
// See boost::filesystem::rename() for more details.
|
|
bool Move(const boost::filesystem::path& new_path);
|
|
|
|
private:
|
|
boost::filesystem::path path_;
|
|
std::size_t chunk_size_;
|
|
bool auto_delete_;
|
|
|
|
std::size_t size_; // File size in bytes
|
|
|
|
boost::filesystem::ifstream ifstream_;
|
|
std::string chunk_;
|
|
};
|
|
|
|
} // namespace webcc
|
|
|
|
#endif // WEBCC_BODY_H_
|