242 lines
8.6 KiB
C++
242 lines
8.6 KiB
C++
//
|
|
// Created by WolverinDEV on 29/04/2020.
|
|
//
|
|
|
|
#include <files/FileServer.h>
|
|
#include <log/LogUtils.h>
|
|
|
|
#include <experimental/filesystem>
|
|
#include <local_server/clnpath.h>
|
|
#include <event2/thread.h>
|
|
#include <include/files/Config.h>
|
|
#include <local_server/HTTPUtils.h>
|
|
|
|
namespace fs = std::experimental::filesystem;
|
|
|
|
using namespace ts::server;
|
|
|
|
struct Nothing {};
|
|
|
|
template <typename ErrorType, typename ResponseType>
|
|
inline void print_fs_response(const std::string& message, const std::shared_ptr<file::ExecuteResponse<file::filesystem::DetailedError<ErrorType>, ResponseType>>& response) {
|
|
if(response->status == file::ExecuteStatus::ERROR)
|
|
logError(LOG_FT, "{}: {} => {}", message, (int) response->error().error_type, response->error().error_message);
|
|
else if(response->status == file::ExecuteStatus::SUCCESS)
|
|
logMessage(LOG_FT, "{}: success", message);
|
|
else
|
|
logWarning(LOG_FT, "Unknown response state ({})!", (int) response->status);
|
|
}
|
|
|
|
template <typename ErrorType, typename ResponseType>
|
|
inline void print_ft_response(const std::string& message, const std::shared_ptr<file::ExecuteResponse<ErrorType, ResponseType>>& response) {
|
|
if(response->status == file::ExecuteStatus::ERROR)
|
|
logError(LOG_FT, "{}: {} => {}", message, (int) response->error().error_type, response->error().error_message);
|
|
else if(response->status == file::ExecuteStatus::SUCCESS)
|
|
logMessage(LOG_FT, "{}: success", message);
|
|
else
|
|
logWarning(LOG_FT, "Unknown response state ({})!", (int) response->status);
|
|
}
|
|
|
|
inline void print_query(const std::string& message, const file::filesystem::AbstractProvider::directory_query_response_t& response) {
|
|
if(response.status == file::ExecuteStatus::ERROR)
|
|
logError(LOG_FT, "{}: {} => {}", message, (int) response.error().error_type, response.error().error_message);
|
|
else if(response.status == file::ExecuteStatus::SUCCESS) {
|
|
const auto& entries = response.response();
|
|
logMessage(LOG_FT, "{}: Found {} entries", message, entries.size());
|
|
for(auto& entry : entries) {
|
|
if(entry.type == file::filesystem::DirectoryEntry::FILE)
|
|
logMessage(LOG_FT, " - File {}", entry.name);
|
|
else if(entry.type == file::filesystem::DirectoryEntry::DIRECTORY)
|
|
logMessage(LOG_FT, " - Directory {}", entry.name);
|
|
else
|
|
logMessage(LOG_FT, " - Unknown {}", entry.name);
|
|
logMessage(LOG_FT, " Write timestamp: {}", std::chrono::floor<std::chrono::seconds>(entry.modified_at.time_since_epoch()).count());
|
|
logMessage(LOG_FT, " Size: {}", entry.size);
|
|
}
|
|
} else
|
|
logWarning(LOG_FT, "{}: Unknown response state ({})!", message, (int) response.status);
|
|
}
|
|
|
|
EVP_PKEY* ssl_generate_key() {
|
|
auto key = std::unique_ptr<EVP_PKEY, decltype(&EVP_PKEY_free)>(EVP_PKEY_new(), ::EVP_PKEY_free);
|
|
|
|
auto rsa = RSA_new();
|
|
auto e = std::unique_ptr<BIGNUM, decltype(&BN_free)>(BN_new(), ::BN_free);
|
|
BN_set_word(e.get(), RSA_F4);
|
|
if(!RSA_generate_key_ex(rsa, 2048, e.get(), nullptr)) return nullptr;
|
|
EVP_PKEY_assign_RSA(key.get(), rsa);
|
|
return key.release();
|
|
}
|
|
|
|
X509* ssl_generate_certificate(EVP_PKEY* key) {
|
|
auto cert = X509_new();
|
|
X509_set_pubkey(cert, key);
|
|
|
|
ASN1_INTEGER_set(X509_get_serialNumber(cert), 3);
|
|
X509_gmtime_adj(X509_get_notBefore(cert), 0);
|
|
X509_gmtime_adj(X509_get_notAfter(cert), 31536000L);
|
|
|
|
X509_NAME* name = nullptr;
|
|
name = X509_get_subject_name(cert);
|
|
//for(const auto& subject : this->subjects)
|
|
// X509_NAME_add_entry_by_txt(name, subject.first.c_str(), MBSTRING_ASC, (unsigned char *) subject.second.c_str(), subject.second.length(), -1, 0);
|
|
X509_set_subject_name(cert, name);
|
|
|
|
name = X509_get_issuer_name(cert);
|
|
//for(const auto& subject : this->issues)
|
|
// X509_NAME_add_entry_by_txt(name, subject.first.c_str(), MBSTRING_ASC, (unsigned char *) subject.second.c_str(), subject.second.length(), -1, 0);
|
|
|
|
X509_set_issuer_name(cert, name);
|
|
|
|
X509_sign(cert, key, EVP_sha512());
|
|
return cert;
|
|
}
|
|
|
|
int main() {
|
|
evthread_use_pthreads();
|
|
{
|
|
std::map<std::string, std::string> query{};
|
|
http::parse_url_parameters("http://www.example.org/suche?stichwort=wiki&no-arg&1arg=&ausgabe=liste&test=test#bla=d&blub=c", query);
|
|
for(const auto& [key, value] : query)
|
|
std::cout << key << " => " << value << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
auto log_config = std::make_shared<logger::LoggerConfig>();
|
|
log_config->terminalLevel = spdlog::level::trace;
|
|
logger::setup(log_config);
|
|
|
|
std::string error{};
|
|
if(!file::initialize(error)) {
|
|
logError(LOG_FT, "Failed to initialize file server: {}", error);
|
|
return 0;
|
|
}
|
|
logMessage(LOG_FT, "File server started");
|
|
auto instance = file::server();
|
|
|
|
|
|
{
|
|
auto options = std::make_shared<pipes::SSL::Options>();
|
|
options->verbose_io = true;
|
|
options->context_method = SSLv23_method();
|
|
options->free_unused_keypairs = false;
|
|
|
|
{
|
|
std::shared_ptr<EVP_PKEY> pkey{ssl_generate_key(), ::EVP_PKEY_free};
|
|
std::shared_ptr<X509> cert{ssl_generate_certificate(&*pkey), ::X509_free};
|
|
|
|
options->default_keypair({pkey, cert});
|
|
}
|
|
file::config::ssl_option_supplier = [options]{
|
|
return options;
|
|
};
|
|
}
|
|
|
|
#if 0
|
|
auto& fs = instance->file_system();
|
|
{
|
|
auto response = fs.initialize_server(0);
|
|
response->wait();
|
|
print_fs_response("Server init result", response);
|
|
if(response->status != file::ExecuteStatus::SUCCESS)
|
|
return 0;
|
|
}
|
|
|
|
|
|
{
|
|
auto response = fs.create_channel_directory(0, 2, "/");
|
|
response->wait();
|
|
print_fs_response("Channel dir create A", response);
|
|
}
|
|
|
|
|
|
{
|
|
auto response = fs.create_channel_directory(0, 2, "/test-folder/");
|
|
response->wait();
|
|
print_fs_response("Channel dir create B", response);
|
|
}
|
|
|
|
|
|
{
|
|
auto response = fs.create_channel_directory(0, 2, "../test-folder/");
|
|
response->wait();
|
|
print_fs_response("Channel dir create C", response);
|
|
}
|
|
|
|
|
|
{
|
|
auto response = fs.create_channel_directory(0, 2, "./test-folder/../test-folder-2");
|
|
response->wait();
|
|
print_fs_response("Channel dir create D", response);
|
|
}
|
|
|
|
{
|
|
auto response = fs.query_channel_directory(0, 2, "/");
|
|
response->wait();
|
|
print_query("Channel query", *response);
|
|
}
|
|
|
|
{
|
|
auto response = fs.query_icon_directory(0);
|
|
response->wait();
|
|
print_query("Icons", *response);
|
|
}
|
|
|
|
{
|
|
auto response = fs.query_avatar_directory(0);
|
|
response->wait();
|
|
print_query("Avatars", *response);
|
|
}
|
|
|
|
{
|
|
auto response = fs.rename_channel_file(0, 2, "./test-folder/../test-folder-2", "./test-folder/../test-folder-3");
|
|
response->wait();
|
|
print_fs_response("Folder rename A", response);
|
|
}
|
|
|
|
{
|
|
auto response = fs.rename_channel_file(0, 2, "./test-folder/../test-folder-3", "./test-folder/../test-folder-2");
|
|
response->wait();
|
|
print_fs_response("Folder rename B", response);
|
|
}
|
|
#endif
|
|
|
|
#if 0
|
|
auto& ft = instance->file_transfer();
|
|
|
|
ft.callback_transfer_finished = [](const std::shared_ptr<file::transfer::Transfer>& transfer) {
|
|
logMessage(0, "Transfer finished");
|
|
};
|
|
|
|
ft.callback_transfer_started = [](const std::shared_ptr<file::transfer::Transfer>& transfer) {
|
|
logMessage(0, "Transfer started");
|
|
};
|
|
|
|
ft.callback_transfer_aborted = [](const std::shared_ptr<file::transfer::Transfer>& transfer, const transfer::TransferStatistics&, const file::transfer::TransferError& error) {
|
|
logMessage(0, "Transfer aborted ({}/{})", (int) error.error_type, error.error_message);
|
|
};
|
|
|
|
ft.callback_transfer_statistics = [](const std::shared_ptr<file::transfer::Transfer>& transfer, const file::transfer::TransferStatistics& stats) {
|
|
logMessage(0, "Transfer stats. New file bytes: {}, delta bytes send {}", stats.delta_file_bytes_transferred, stats.delta_network_bytes_send);
|
|
};
|
|
|
|
{
|
|
auto response = ft.initialize_channel_transfer(file::transfer::Transfer::DIRECTION_UPLOAD, 0, 2, {
|
|
"test2.txt",
|
|
false,
|
|
4,
|
|
120,
|
|
32
|
|
});
|
|
response->wait();
|
|
print_ft_response("Download test.txt", response);
|
|
if(response->succeeded())
|
|
logMessage(LOG_FT, "Download key: {}", std::string{response->response()->transfer_key, TRANSFER_KEY_LENGTH});
|
|
}
|
|
#endif
|
|
|
|
std::this_thread::sleep_for(std::chrono::seconds{120});
|
|
//TODO: Test file locking
|
|
file::finalize();
|
|
return 0;
|
|
} |