security-and-privacy/uts-server
security-and-privacy
/
uts-server
mirror of https://github.com/kakwa/uts-server
1
0
Fork 0
Code Issues Releases Wiki Activity

adding error handling in case civetweb fails to start

This commit is contained in:
kakwa 2016-09-11 20:07:14 +02:00
parent 517535b920
commit bac488ae30
2 changed files with 24 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/lib/http.c
View File

@ -16,6 +16,10 @@
#include <time.h>
#include "http.h"
struct tuser_data {
char *first_message;
};
static char *rand_string(char *str, size_t size) {
const char charset[] = "1234567890ABCDEF";
if (size) {
@ -77,7 +81,8 @@ void log_request(const struct mg_request_info *request_info, char *request_id,
if (strcasecmp(request_info->http_headers[i].name, "User-Agent") == 0) {
user_agent = request_info->http_headers[i].value;
}
if (strcasecmp(request_info->http_headers[i].name, "Content-Type") == 0) {
if (strcasecmp(request_info->http_headers[i].name, "Content-Type") ==
0) {
content_type = request_info->http_headers[i].value;
}
}
@ -205,6 +210,7 @@ int rfc3161_handler(struct mg_connection *conn, void *context) {
int http_server_start(char *conffile, char *conf_wd, bool stdout_dbg) {
struct mg_context *ctx;
struct mg_callbacks callbacks;
struct tuser_data user_data;
rfc3161_context *ct = (rfc3161_context *)calloc(1, sizeof(rfc3161_context));
ct->stdout_dbg = stdout_dbg;
@ -217,8 +223,11 @@ int http_server_start(char *conffile, char *conf_wd, bool stdout_dbg) {
// NULL.
memset(&callbacks, 0, sizeof(callbacks));
memset(&user_data, 0, sizeof(user_data));
// Start the web server.
ctx = mg_start(&callbacks, NULL, ct->http_options);
ctx = mg_start(&callbacks, &user_data, ct->http_options);
if (ctx != NULL) {
mg_set_request_handler(ctx, "/", rfc3161_handler, (void *)ct);
// Wait until some signals are received
@ -226,6 +235,12 @@ int http_server_start(char *conffile, char *conf_wd, bool stdout_dbg) {
sleep(1);
}
// getchar();
} else {
uts_logger(ct, LOG_CRIT, "Failed to start uts-server: %s",
((user_data.first_message == NULL)
? "unknown reason"
: user_data.first_message));
}
// Stop the server.
mg_stop(ctx);