#include <curl/curl.h>

#include <map>

#include <string>

#include "httpserver.hpp"

#include "httpserver/http_utils.hpp"

#include "littletest.hpp"

using std::shared_ptr;

using httpserver::webserver;

using httpserver::create_webserver;

using httpserver::http_resource;

using httpserver::http_response;

using httpserver::string_response;

using httpserver::http_request;

using httpserver::http::http_utils;

size_t writefunc(void *ptr, size_t size, size_t nmemb, std::string *s) {

s->append(reinterpret_cast<char*>(ptr), size*nmemb);

return size*nmemb;

}

class ok_resource : public http_resource {

};

LT_BEGIN_SUITE(ban_system_suite)

void set_up() {

}

void tear_down() {

}

LT_END_SUITE(ban_system_suite)

LT_BEGIN_AUTO_TEST(ban_system_suite, accept_default_ban_blocks)

webserver ws = create_webserver(8080).default_policy(http_utils::ACCEPT);

ws.start(false);

ok_resource resource;

ws.register_resource("base", &resource);

curl_global_init(CURL_GLOBAL_ALL);

{

std::string s;

CURL *curl = curl_easy_init();

CURLcode res;

curl_easy_setopt(curl, CURLOPT_URL, "localhost:8080/base");

curl_easy_setopt(curl, CURLOPT_HTTPGET, 1L);

curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, writefunc);

curl_easy_setopt(curl, CURLOPT_WRITEDATA, &s);

res = curl_easy_perform(curl);

LT_ASSERT_EQ(res, 0);

LT_CHECK_EQ(s, "OK");

curl_easy_cleanup(curl);

}

{

ws.ban_ip("127.0.0.1");

CURL *curl = curl_easy_init();

CURLcode res;

curl_easy_setopt(curl, CURLOPT_URL, "localhost:8080/base");

curl_easy_setopt(curl, CURLOPT_HTTPGET, 1L);

res = curl_easy_perform(curl);

LT_ASSERT_NEQ(res, 0);

curl_easy_cleanup(curl);

}

{

ws.unban_ip("127.0.0.1");

std::string s;

CURL *curl = curl_easy_init();

CURLcode res;

curl_easy_setopt(curl, CURLOPT_URL, "localhost:8080/base");

curl_easy_setopt(curl, CURLOPT_HTTPGET, 1L);

curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, writefunc);

curl_easy_setopt(curl, CURLOPT_WRITEDATA, &s);

res = curl_easy_perform(curl);

LT_ASSERT_EQ(res, 0);

LT_CHECK_EQ(s, "OK");

curl_easy_cleanup(curl);

}

curl_global_cleanup();

ws.stop();

LT_END_AUTO_TEST(accept_default_ban_blocks)

LT_BEGIN_AUTO_TEST(ban_system_suite, reject_default_allow_passes)

webserver ws = create_webserver(8080).default_policy(http_utils::REJECT);

ws.start(false);

ok_resource resource;

ws.register_resource("base", &resource);

curl_global_init(CURL_GLOBAL_ALL);

{

CURL *curl = curl_easy_init();

CURLcode res;

curl_easy_setopt(curl, CURLOPT_URL, "localhost:8080/base");

curl_easy_setopt(curl, CURLOPT_HTTPGET, 1L);

res = curl_easy_perform(curl);

LT_ASSERT_NEQ(res, 0);

curl_easy_cleanup(curl);

}

{

ws.allow_ip("127.0.0.1");

std::string s;

CURL *curl = curl_easy_init();

CURLcode res;

curl_easy_setopt(curl, CURLOPT_URL, "localhost:8080/base");

curl_easy_setopt(curl, CURLOPT_HTTPGET, 1L);

curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, writefunc);

curl_easy_setopt(curl, CURLOPT_WRITEDATA, &s);

res = curl_easy_perform(curl);

LT_ASSERT_EQ(res, 0);

LT_CHECK_EQ(s, "OK");

curl_easy_cleanup(curl);

}

{

ws.disallow_ip("127.0.0.1");

CURL *curl = curl_easy_init();

CURLcode res;

curl_easy_setopt(curl, CURLOPT_URL, "localhost:8080/base");

curl_easy_setopt(curl, CURLOPT_HTTPGET, 1L);

res = curl_easy_perform(curl);

LT_ASSERT_NEQ(res, 0);

curl_easy_cleanup(curl);

}

curl_global_cleanup();

ws.stop();

LT_END_AUTO_TEST(reject_default_allow_passes)

LT_BEGIN_AUTO_TEST_ENV()

AUTORUN_TESTS()

LT_END_AUTO_TEST_ENV()