{

webserver* ws;

struct MHD_Daemon* daemon;

daemon_item(webserver* ws, struct MHD_Daemon* daemon):

ws(ws),

daemon(daemon)

{

}

~daemon_item()

{

MHD_stop_daemon (this->daemon);

}

{

bool operator() (const std::pair<int, int>& left,

const std::pair<int, int>& right

) const

{

return left.second < right.second;

}

{

#ifndef __MINGW32__

struct sigaction oldsig;

struct sigaction sig;

sig.sa_handler = &catcher;

sigemptyset (&sig.sa_mask);

#ifdef SA_INTERRUPT

sig.sa_flags = SA_INTERRUPT; /* SunOS */

#else //SA_INTERRUPT

sig.sa_flags = SA_RESTART;

#endif //SA_INTERRUPTT

if (0 != sigaction (SIGPIPE, &sig, &oldsig))

fprintf (stderr,

gettext("Failed to install SIGPIPE handler: %s\n"),

strerror (errno)

);

#endif

port(params._port),

start_method(params._start_method),

max_threads(params._max_threads),

max_connections(params._max_connections),

memory_limit(params._memory_limit),

connection_timeout(params._connection_timeout),

per_IP_connection_limit(params._per_IP_connection_limit),

log_access(params._log_access),

log_error(params._log_error),

validator(params._validator),

unescaper(params._unescaper),

bind_address(params._bind_address),

bind_socket(params._bind_socket),

max_thread_stack_size(params._max_thread_stack_size),

use_ssl(params._use_ssl),

use_ipv6(params._use_ipv6),

debug(params._debug),

pedantic(params._pedantic),

https_mem_key(params._https_mem_key),

https_mem_cert(params._https_mem_cert),

https_mem_trust(params._https_mem_trust),

https_priorities(params._https_priorities),

cred_type(params._cred_type),

digest_auth_random(params._digest_auth_random),

nonce_nc_size(params._nonce_nc_size),

running(false),

default_policy(params._default_policy),

basic_auth_enabled(params._basic_auth_enabled),

digest_auth_enabled(params._digest_auth_enabled),

regex_checking(params._regex_checking),

ban_system_enabled(params._ban_system_enabled),

post_process_enabled(params._post_process_enabled),

single_resource(params._single_resource),

not_found_resource(params._not_found_resource),

method_not_allowed_resource(params._method_not_allowed_resource),

method_not_acceptable_resource(params._method_not_acceptable_resource),

internal_error_resource(params._internal_error_resource),

next_to_choose(0),

internal_comet_manager(new details::comet_manager())

{

if(single_resource != 0x0)

this->single_resource = true;

else

this->single_resource = false;

ignore_sigpipe();

pthread_mutex_init(&mutexwait, NULL);

pthread_rwlock_init(&runguard, NULL);

pthread_cond_init(&mutexcond, NULL);

pthread_rwlock_init(&cache_guard, NULL);

void *cls,

struct MHD_Connection *connection,

void **con_cls,

enum MHD_RequestTerminationCode toe

)

{

details::modded_request* mr = static_cast<details::modded_request*>(*con_cls);

if (mr != 0x0)

{

if(mr->ws != 0x0)

{

mr->ws->internal_comet_manager->complete_request(mr->dhrs->connection_id);

}

if(mr->dhrs.res != 0x0 && mr->dhrs->ca != 0x0)

mr->dhrs->ca(mr->dhrs->closure_data);

delete mr;

mr = 0x0;

}

const std::string& resource,

details::http_resource_mirror hrm,

bool family

)

{

if(method_not_acceptable_resource)

hrm.method_not_acceptable_resource = method_not_acceptable_resource;

details::http_endpoint idx(resource, family, true, regex_checking);

pair<map<details::http_endpoint, details::http_resource_mirror>::iterator, bool> result = registered_resources.insert(

map<details::http_endpoint, details::http_resource_mirror>::value_type(idx, hrm)

);

if(result.second)

{

registered_resources_str.insert(

pair<string, details::http_resource_mirror*>(idx.get_url_complete(), &(result.first->second))

);

}

return result.second;

{

int sock_cloexec = SOCK_CLOEXEC;

int ctype = SOCK_STREAM | sock_cloexec;

/* use SOCK_STREAM rather than ai_socktype: some getaddrinfo

MHD_socket fd = socket(domain, ctype, protocol);

#ifdef _WINDOWS

if (fd == INVALID_SOCKET)

#else

if ((fd == -1) &&

(errno == EINVAL || errno == EPROTONOSUPPORT) && (sock_cloexec != 0)

)

#endif

{

fd = socket(domain, type, protocol);

}

return fd;

{

struct {

MHD_OptionItem operator ()(

enum MHD_OPTION opt,

intptr_t val,

void *ptr = 0

)

{

MHD_OptionItem x = {opt, val, ptr};

return x;

}

} gen;

vector<struct MHD_OptionItem> iov;

iov.push_back(gen(MHD_OPTION_NOTIFY_COMPLETED,

(intptr_t) &request_completed,

NULL

));

iov.push_back(gen(MHD_OPTION_URI_LOG_CALLBACK, (intptr_t) &uri_log, this));

iov.push_back(gen(MHD_OPTION_EXTERNAL_LOGGER, (intptr_t) &error_log, this));

iov.push_back(gen(MHD_OPTION_UNESCAPE_CALLBACK,

(intptr_t) &unescaper_func,

this)

);

iov.push_back(gen(MHD_OPTION_CONNECTION_TIMEOUT, connection_timeout));

if(bind_address != 0x0)

iov.push_back(gen(MHD_OPTION_SOCK_ADDR, (intptr_t) bind_address));

if(bind_socket != 0)

iov.push_back(gen(MHD_OPTION_LISTEN_SOCKET, bind_socket));

if(! (start_method == http_utils::INTERNAL_SELECT))

{

if(max_threads != 0)

iov.push_back(gen(MHD_OPTION_THREAD_POOL_SIZE, max_threads));

}

if(max_connections != 0)

iov.push_back(gen(MHD_OPTION_CONNECTION_LIMIT, max_connections));

if(memory_limit != 0)

iov.push_back(gen(MHD_OPTION_CONNECTION_MEMORY_LIMIT, memory_limit));

if(per_IP_connection_limit != 0)

iov.push_back(gen(MHD_OPTION_PER_IP_CONNECTION_LIMIT,

per_IP_connection_limit)

);

if(max_thread_stack_size != 0)

iov.push_back(gen(MHD_OPTION_THREAD_STACK_SIZE, max_thread_stack_size));

if(nonce_nc_size != 0)

iov.push_back(gen(MHD_OPTION_NONCE_NC_SIZE, nonce_nc_size));

if(use_ssl)

iov.push_back(gen(MHD_OPTION_HTTPS_MEM_KEY,

0,

(void*)https_mem_key.c_str())

);

if(use_ssl)

iov.push_back(gen(MHD_OPTION_HTTPS_MEM_CERT,

0,

(void*)https_mem_cert.c_str())

);

if(https_mem_trust != "" && use_ssl)

iov.push_back(gen(MHD_OPTION_HTTPS_MEM_TRUST,

0,

(void*)https_mem_trust.c_str())

);

if(https_priorities != "" && use_ssl)

iov.push_back(gen(MHD_OPTION_HTTPS_PRIORITIES,

0,

(void*)https_priorities.c_str())

);

if(digest_auth_random != "")

iov.push_back(gen(MHD_OPTION_DIGEST_AUTH_RANDOM,

digest_auth_random.size(),

(char*)digest_auth_random.c_str())

);

#ifdef HAVE_GNUTLS

if(cred_type != http_utils::NONE)

iov.push_back(gen(MHD_OPTION_HTTPS_CRED_TYPE, cred_type));

#endif //HAVE_GNUTLS

if(start_method == http_utils::INTERNAL_SELECT)

{

int on = 1;

bool bind_settled = true;

if(!bind_socket)

{

bind_settled = false;

struct sockaddr_in servaddr4;

#if HAVE_INET6

struct sockaddr_in6 servaddr6;

#endif

const struct sockaddr *servaddr = NULL;

socklen_t addrlen;

#if HAVE_INET6

if (0 != (options & MHD_USE_IPv6))

addrlen = sizeof (struct sockaddr_in6);

else

#endif

addrlen = sizeof (struct ::sockaddr_in);

#if HAVE_INET6

if (0 != (options & MHD_USE_IPv6))

{

memset (&servaddr6, 0, sizeof (struct sockaddr_in6));

servaddr6.sin6_family = AF_INET6;

servaddr6.sin6_port = htons (port);

#if HAVE_SOCKADDR_IN_SIN_LEN

servaddr6.sin6_len = sizeof (struct sockaddr_in6);

#endif

servaddr = (struct sockaddr *) &servaddr6;

}

else

#endif

{

memset (&servaddr4, 0, sizeof (struct ::sockaddr_in));

servaddr4.sin_family = AF_INET;

servaddr4.sin_port = htons (port);

#if HAVE_SOCKADDR_IN_SIN_LEN

servaddr4.sin_len = sizeof (struct ::sockaddr_in);

#endif

servaddr = (struct sockaddr *) &servaddr4;

}

if (use_ipv6)

bind_socket = create_socket (PF_INET6, SOCK_STREAM, 0);

else

bind_socket = create_socket (PF_INET, SOCK_STREAM, 0);

setsockopt (bind_socket,

SOL_SOCKET,

SO_REUSEADDR,

(const char*) &on, sizeof (on));

if(use_ipv6)

{

#ifdef IPPROTO_IPV6

#ifdef IPV6_V6ONLY

setsockopt (bind_socket,

IPPROTO_IPV6, IPV6_V6ONLY,

(const char*) &on, sizeof (on)

);

#endif

#endif

}

bind(bind_socket, servaddr, addrlen);

}

#ifdef _WINDOWS

unsigned long ioarg = 1;

ioctlsocket(bind_socket, FIONBIO, &ioarg);

#else

int flags = fcntl (bind_socket, F_GETFL);

flags |= O_NONBLOCK;

fcntl (bind_socket, F_SETFL, flags);

#endif

if(!bind_settled)

listen(bind_socket, 1);

iov.push_back(gen(MHD_OPTION_LISTEN_SOCKET, bind_socket));

}

iov.push_back(gen(MHD_OPTION_END, 0, NULL ));

int start_conf = start_method;

if(use_ssl)

start_conf |= MHD_USE_SSL;

if(use_ipv6)

start_conf |= MHD_USE_IPv6;

if(debug)

start_conf |= MHD_USE_DEBUG;

if(pedantic)

start_conf |= MHD_USE_PEDANTIC_CHECKS;

int num_threads = 1;

if(max_threads > num_threads)

num_threads = max_threads;

this->running = true;

if(start_method == http_utils::INTERNAL_SELECT)

{

int i;

try {

for(i = 0; i < num_threads; i++)

{

struct MHD_Daemon* daemon = MHD_start_daemon

(

start_conf, this->port, &policy_callback, this,

&answer_to_connection, this, MHD_OPTION_ARRAY,

&iov[0], MHD_OPTION_END

);

if(NULL == daemon)

{

cout << gettext("Unable to connect daemon to port: ") << this->port << endl;

throw ::httpserver::webserver_exception();

}

details::daemon_item* di = new details::daemon_item(this, daemon);

daemons.push_back(di);

//RUN SELECT THREADS

pthread_t t;

threads.push_back(t);

if(pthread_create(

&threads[i],

NULL,

&webserver::select,

static_cast<void*>(di)

))

{

throw ::httpserver::webserver_exception();

}

}

} catch (::httpserver::webserver_exception &e) {

this->running = false;

for (;i >= 0; --i) {

pthread_kill(threads[i], 9);

}

threads.clear();

throw e;

}

}

else

{

struct MHD_Daemon* daemon = MHD_start_daemon

(

start_conf, this->port, &policy_callback, this,

&answer_to_connection, this, MHD_OPTION_ARRAY,

&iov[0], MHD_OPTION_END

);

if(NULL == daemon)

{

cout << gettext("Unable to connect daemon to port: ") <<

this->port << endl;

throw ::httpserver::webserver_exception();

}

details::daemon_item* di = new details::daemon_item(this, daemon);

daemons.push_back(di);

}

bool value_onclose = false;

if(blocking)

{

pthread_mutex_lock(&mutexwait);

while(blocking && running)

pthread_cond_wait(&mutexcond, &mutexwait);

pthread_mutex_unlock(&mutexwait);

value_onclose = true;

}

return value_onclose;

{

if(this->running)

{

pthread_mutex_lock(&mutexwait);

this->running = false;

pthread_cond_signal(&mutexcond);

pthread_mutex_unlock(&mutexwait);

for(unsigned int i = 0; i < threads.size(); ++i)

{

void* t_res;

pthread_join(threads[i], &t_res);

free(t_res);

}

threads.clear();

typedef vector<details::daemon_item*>::const_iterator daemon_item_it;

for(daemon_item_it it = daemons.begin(); it != daemons.end(); ++it)

delete *it;

daemons.clear();

return true;

}

else

{

return false;

}

{

details::http_endpoint he(resource);

this->registered_resources.erase(he);

this->registered_resources.erase(he.url_complete);

{

ip_representation t_ip(ip);

set<ip_representation>::iterator it = this->bans.find(t_ip);

if(it != this->bans.end() && (t_ip.weight() < (*it).weight()))

{

this->bans.erase(it);

this->bans.insert(t_ip);

}

else

this->bans.insert(t_ip);

{

ip_representation t_ip(ip);

set<ip_representation>::iterator it = this->allowances.find(t_ip);

if(it != this->allowances.end() && (t_ip.weight() < (*it).weight()))

{

this->allowances.erase(it);

this->allowances.insert(t_ip);

}

else

this->allowances.insert(t_ip);

void *cls,

enum MHD_ValueKind kind,

const char *key,

const char *value

)

{

http_request* dhr = static_cast<http_request*>(cls);

dhr->set_header(key, value);

return MHD_YES;

void *cls,

enum MHD_ValueKind kind,

const char *key,

const char *value

)

{

http_request* dhr = static_cast<http_request*>(cls);

dhr->set_cookie(key, value);

return MHD_YES;

void *cls,

enum MHD_ValueKind kind,

const char *key,

const char *value

)

{

http_request* dhr = static_cast<http_request*>(cls);

dhr->set_footer(key, value);

return MHD_YES;

void *cls,

enum MHD_ValueKind kind,

const char *key,

const char *arg_value

)

{

details::modded_request* mr = static_cast<details::modded_request*>(cls);

char* value = (char*) ((arg_value == NULL) ? "" : arg_value);

{

char buf[strlen(key) + strlen(value) + 3];

if(mr->dhr->querystring == "")

{

snprintf(buf, sizeof buf, "?%s=%s", key, value);

mr->dhr->querystring = buf;

}

else

{

snprintf(buf, sizeof buf, "&%s=%s", key, value);

mr->dhr->querystring += string(buf);

}

}

size_t size = internal_unescaper((void*) mr->ws, value);

mr->dhr->set_arg(key, string(value, size));

return MHD_YES;

{

if((static_cast<webserver*>(cls))->ban_system_enabled)

{

if((((static_cast<webserver*>(cls))->default_policy == http_utils::ACCEPT) &&

((static_cast<webserver*>(cls))->bans.count(addr)) &&

(!(static_cast<webserver*>(cls))->allowances.count(addr))

) ||

(((static_cast<webserver*>(cls))->default_policy == http_utils::REJECT)

&& ((!(static_cast<webserver*>(cls))->allowances.count(addr)) ||

((static_cast<webserver*>(cls))->bans.count(addr)))

))

return MHD_NO;

}

return MHD_YES;

{

webserver* dws = static_cast<webserver*>(cls);

if(dws->log_error != 0x0)

dws->log_error(fmt);

{

if(dws->log_access != 0x0)

dws->log_access(uri);

{

// THIS IS USED TO AVOID AN UNESCAPING OF URL BEFORE THE ANSWER.

// IT IS DUE TO A BOGUS ON libmicrohttpd (V0.99) THAT PRODUCING A

// STRING CONTAINING '\0' AFTER AN UNESCAPING, IS UNABLE TO PARSE

// ARGS WITH get_connection_values FUNC OR lookup FUNC.

return strlen(s);

{

if(s[0] == 0) return 0;

webserver* dws = static_cast<webserver*>(cls);

if(dws->unescaper != 0x0)

{

dws->unescaper(s);

return strlen(s);

}

else

{

return http_unescape(s);

}

const char *key,

const char *filename,

const char *content_type,

const char *transfer_encoding,

const char *data, uint64_t off, size_t size

)

{

struct details::modded_request* mr = (struct details::modded_request*) cls;

mr->dhr->set_arg(key, data, size);

return MHD_YES;

http_response** dhrs,

details::modded_request* mr

)

{

if(not_found_resource != 0x0)

not_found_resource(*mr->dhr, dhrs);

else

*dhrs = new http_response(http_response_builder(NOT_FOUND_ERROR, http_utils::http_not_found).string_response());

struct MHD_Connection *connection)

{

int ret;

struct MHD_Response *response;

/* unsupported HTTP method */

response = MHD_create_response_from_buffer (strlen (NOT_METHOD_ERROR),

(void *) NOT_METHOD_ERROR,

MHD_RESPMEM_PERSISTENT);

ret = MHD_queue_response (connection,

MHD_HTTP_NOT_ACCEPTABLE,

response);

MHD_add_response_header (response,

MHD_HTTP_HEADER_CONTENT_ENCODING,

"text/plain");

MHD_destroy_response (response);

return ret;

http_response** dhrs,

details::modded_request* mr

)

{

if(method_not_acceptable_resource != 0x0)

method_not_allowed_resource(*mr->dhr, dhrs);

else

*dhrs = new http_response(http_response_builder(METHOD_ERROR, http_utils::http_method_not_allowed).string_response());

http_response** dhrs,

details::modded_request* mr,

bool force_our

)

{

if(internal_error_resource != 0x0 && !force_our)

internal_error_resource(*mr->dhr, dhrs);

else

*dhrs = new http_response(http_response_builder(GENERIC_ERROR, http_utils::http_internal_server_error).string_response());

MHD_Connection* connection, const char* method,

const char* version, struct details::modded_request* mr

)

{

http_request req;

mr->dhr = &(req);

return complete_request(connection, mr, version, method);

MHD_Connection* connection,

struct details::modded_request* mr

)

{

mr->second = true;

mr->dhr = new http_request();

const char *encoding = MHD_lookup_connection_value (

connection,

MHD_HEADER_KIND,

http_utils::http_header_content_type.c_str()

);

if(encoding != 0x0)

mr->dhr->set_header(http_utils::http_header_content_type, encoding);

if ( post_process_enabled &&

(

0x0 != encoding &&

((0 == strncasecmp (

MHD_HTTP_POST_ENCODING_FORM_URLENCODED,

encoding,

strlen (MHD_HTTP_POST_ENCODING_FORM_URLENCODED)

)

))

)

)

{

mr->pp = MHD_create_post_processor (

connection,

1024,

&post_iterator,

mr

);

}

else

{

mr->pp = NULL;

}

return MHD_YES;

MHD_Connection* connection, const char* method,

const char* version, const char* upload_data,

size_t* upload_data_size, struct details::modded_request* mr

)

{

if ( 0 != *upload_data_size)

{

#ifdef DEBUG

cout << "Writing content: " << upload_data << endl;

#endif //DEBUG

mr->dhr->grow_content(upload_data, *upload_data_size);

if (mr->pp != NULL)

{

MHD_post_process(mr->pp, upload_data, *upload_data_size);

}

*upload_data_size = 0;

return MHD_YES;

}

return complete_request(connection, mr, version, method);

MHD_Connection* connection,

struct details::modded_request* mr,

const char* version,

const char* method,

char* user,

char* pass,

char* digested_user

)

{

mr->ws = this;

MHD_get_connection_values (

connection,

MHD_GET_ARGUMENT_KIND,

&build_request_args,

(void*) mr

);

MHD_get_connection_values (

connection,

MHD_HEADER_KIND,

&build_request_header,

(void*) mr->dhr

);

MHD_get_connection_values (

connection,

MHD_FOOTER_KIND,

&build_request_footer,

(void*) mr->dhr