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.
120 lines
3.2 KiB
C++
120 lines
3.2 KiB
C++
#include "webcc/utility.h"
|
|
|
|
#include <algorithm>
|
|
#include <ctime>
|
|
#include <fstream>
|
|
#include <iomanip> // for put_time
|
|
#include <iostream>
|
|
#include <sstream>
|
|
|
|
#include "boost/filesystem/fstream.hpp"
|
|
|
|
#include "webcc/string.h"
|
|
#include "webcc/version.h"
|
|
|
|
namespace bfs = boost::filesystem;
|
|
|
|
namespace webcc {
|
|
namespace utility {
|
|
|
|
const std::string& UserAgent() {
|
|
static const std::string s_user_agent = std::string("Webcc/") + WEBCC_VERSION;
|
|
return s_user_agent;
|
|
}
|
|
|
|
std::string HttpDate() {
|
|
std::time_t t = std::time(nullptr);
|
|
tm* gmt = std::gmtime(&t);
|
|
|
|
// Either put_time() or strftime() could format the date as expected, but they
|
|
// are both locale dependent!
|
|
//
|
|
// std::stringstream ss;
|
|
// ss << std::put_time(gmt, "%a, %d %b %Y %H:%M:%S") << " GMT";
|
|
// return ss.str();
|
|
//
|
|
// char buf[26];
|
|
// std::strftime(buf, 26, "%a, %d %b %Y %H:%M:%S", gmt);
|
|
|
|
static const char* const kDays[7] = { "Sun", "Mon", "Tue", "Wed",
|
|
"Thu", "Fri", "Sat" };
|
|
|
|
static const char* const kMonths[12] = { "Jan", "Feb", "Mar", "Apr",
|
|
"May", "Jun", "Jul", "Aug",
|
|
"Sep", "Oct", "Nov", "Dec" };
|
|
|
|
char buf[26];
|
|
|
|
std::snprintf(buf, 26, "%s, %.2i %s %i %.2i:%.2i:%.2i", kDays[gmt->tm_wday],
|
|
gmt->tm_mday, kMonths[gmt->tm_mon], gmt->tm_year + 1900,
|
|
gmt->tm_hour, gmt->tm_min, gmt->tm_sec);
|
|
|
|
return std::string(buf) + " GMT";
|
|
}
|
|
|
|
std::size_t TellSize(const bfs::path& path) {
|
|
// Flag "ate": seek to the end of stream immediately after open.
|
|
bfs::ifstream stream{ path, std::ios::binary | std::ios::ate };
|
|
if (stream.fail()) {
|
|
return kInvalidLength;
|
|
}
|
|
return static_cast<std::size_t>(stream.tellg());
|
|
}
|
|
|
|
bool ReadFile(const bfs::path& path, std::string* output) {
|
|
// Flag "ate": seek to the end of stream immediately after open.
|
|
bfs::ifstream stream{ path, std::ios::binary | std::ios::ate };
|
|
if (stream.fail()) {
|
|
return false;
|
|
}
|
|
|
|
auto size = stream.tellg();
|
|
output->resize(static_cast<std::size_t>(size), '\0');
|
|
stream.seekg(std::ios::beg);
|
|
stream.read(&(*output)[0], size);
|
|
if (stream.fail()) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void DumpByLine(const std::string& data, std::ostream& os,
|
|
const std::string& prefix) {
|
|
std::vector<boost::string_view> lines;
|
|
Split(data, '\n', false, &lines);
|
|
|
|
std::size_t size = 0;
|
|
|
|
for (const auto& line : lines) {
|
|
os << prefix;
|
|
|
|
if (line.size() + size > kMaxDumpSize) {
|
|
os.write(line.data(), kMaxDumpSize - size);
|
|
os << "..." << std::endl;
|
|
break;
|
|
} else {
|
|
os << line << std::endl;
|
|
size += line.size();
|
|
}
|
|
}
|
|
}
|
|
|
|
void PrintEndpoint(std::ostream& ostream,
|
|
const boost::asio::ip::tcp::endpoint& endpoint) {
|
|
ostream << endpoint;
|
|
if (endpoint.protocol() == boost::asio::ip::tcp::v4()) {
|
|
ostream << ", v4";
|
|
} else if (endpoint.protocol() == boost::asio::ip::tcp::v6()) {
|
|
ostream << ", v6";
|
|
}
|
|
}
|
|
|
|
std::string EndpointToString(const boost::asio::ip::tcp::endpoint& endpoint) {
|
|
std::stringstream ss;
|
|
PrintEndpoint(ss, endpoint);
|
|
return ss.str();
|
|
}
|
|
|
|
} // namespace utility
|
|
} // namespace webcc
|