falk-werner_webfuse-provider/test/webfuse_provider/tests/integration/server.cc

#include "webfuse_provider/tests/integration/server.hpp"
#include <thread>
#include <mutex>
#include <sstream>
#include <cstdlib>
#include <cstring>
#include <unistd.h>
#include <sys/stat.h>
#include "webfuse_adapter.h"
#include "webfuse/adapter/impl/server.h"

#define WF_PATH_MAX (100)

extern "C"
{

static void webfuse_test_server_cleanup_mountpoint(
    void * user_data)
{
    char * path = reinterpret_cast<char*>(user_data);
    rmdir(path);
    free(path);
}

static struct wf_mountpoint *
webfuse_test_server_create_mountpoint(
    char const * filesystem,
    void * user_data)
{
    char const * base_dir = reinterpret_cast<char const*>(user_data);
    char path[WF_PATH_MAX];
    snprintf(path, WF_PATH_MAX, "%s/%s", base_dir, filesystem);
    mkdir(path, 0755);
    struct wf_mountpoint * mountpoint = wf_mountpoint_create(path);
    wf_mountpoint_set_userdata(
        mountpoint,
        reinterpret_cast<void*>(strdup(path)),
        &webfuse_test_server_cleanup_mountpoint);

    return mountpoint;
}

}

namespace webfuse_test
{

class Server::Private
{
public:
    Private()
    : is_shutdown_requested(false)
    {
        snprintf(base_dir, WF_PATH_MAX, "%s", "/tmp/webfuse_test_integration_XXXXXX");
        char const * result = mkdtemp(base_dir);
        if (NULL == result)
        {
            throw std::runtime_error("unable to create temp dir");
        }


        config = wf_server_config_create();
        wf_server_config_set_port(config, 0);
        wf_server_config_set_mountpoint_factory(config, 
            &webfuse_test_server_create_mountpoint,
            reinterpret_cast<void*>(base_dir));
        wf_server_config_set_keypath(config, "server-key.pem");
        wf_server_config_set_certpath(config, "server-cert.pem");

        server = wf_server_create(config);

        while (!wf_impl_server_is_operational(server))
        {
            wf_server_service(server);
        }

        thread = std::thread(Run, this);

    }

    ~Private()
    {
        RequestShutdown();
        thread.join();
        rmdir(base_dir);
        wf_server_dispose(server);
        wf_server_config_dispose(config);
    }

    bool IsShutdownRequested()
    {
        std::lock_guard<std::mutex> lock(shutdown_lock);
        return is_shutdown_requested;
    }

    std::string GetUrl(void) const
    {
        int const port = wf_server_get_port(server);
        std::ostringstream stream;
        stream << "wss://localhost:" << port << "/";
        return stream.str();
    }

private:
    void RequestShutdown()
    {
        std::lock_guard<std::mutex> lock(shutdown_lock);
        is_shutdown_requested = true;
        wf_server_interrupt(server);
    }

    static void Run(Server::Private * context)
    {
        while (!context->IsShutdownRequested())
        {
            wf_server_service(context->server);
        }
    }


    std::mutex shutdown_lock;
    std::thread thread;
    bool is_shutdown_requested;


public:
    char base_dir[WF_PATH_MAX];
    wf_server_config * config;
    wf_server * server;
};

Server::Server()
: d(new Server::Private())
{

}

Server::~Server()
{
    delete d;
}

char const * Server::GetBaseDir(void) const
{
    return d->base_dir;
}

std::string Server::GetUrl(void) const
{
    return d->GetUrl();
}


}