small changes
This commit is contained in:
70
src/async.c
Normal file
70
src/async.c
Normal file
@ -0,0 +1,70 @@
|
|||||||
|
|
||||||
|
|
||||||
|
#include "logger.h"
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <poll.h>
|
||||||
|
#include <signal.h>
|
||||||
|
#include <errno.h>
|
||||||
|
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
int fd;
|
||||||
|
short events;
|
||||||
|
int flags;
|
||||||
|
void (*cb)(void *);
|
||||||
|
void *arg;
|
||||||
|
void (*err_cb)(void *);
|
||||||
|
void *err_arg;
|
||||||
|
} evt_listen_t;
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
int n;
|
||||||
|
evt_listen_t q[256];
|
||||||
|
} listen_queue_t;
|
||||||
|
|
||||||
|
|
||||||
|
static listen_queue_t listen1;
|
||||||
|
static listen_queue_t listen2;
|
||||||
|
listen_queue_t *listen = &listen1;
|
||||||
|
|
||||||
|
volatile sig_atomic_t alive = 1;
|
||||||
|
|
||||||
|
int async(int fd, int events, int flags, void (*cb)(void *), void *arg, void (*err_cb)(void *), void *err_arg) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
void async_thread(void) {
|
||||||
|
|
||||||
|
int num_fds = 0;
|
||||||
|
struct pollfd fds[256];
|
||||||
|
|
||||||
|
// main event loop
|
||||||
|
while (alive) {
|
||||||
|
// swap listen queue
|
||||||
|
listen_queue_t *l = listen;
|
||||||
|
listen = (listen == &listen1) ? &listen2 : &listen1;
|
||||||
|
|
||||||
|
// fill fds with newly added
|
||||||
|
for (int i = 0; i < l->n; i++, num_fds++) {
|
||||||
|
fds[num_fds].fd = l->q[i].fd;
|
||||||
|
fds[num_fds].events = l->q[i].events;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ready_fds = poll(fds, num_fds, -1);
|
||||||
|
if (ready_fds < 0) {
|
||||||
|
if (errno == EINTR) {
|
||||||
|
// interrupt
|
||||||
|
continue;
|
||||||
|
} else {
|
||||||
|
// other error
|
||||||
|
critical("Unable to poll for events");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 0; i < num_fds; i++) {
|
||||||
|
// TODO
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
13
src/async.h
Normal file
13
src/async.h
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
|
||||||
|
|
||||||
|
#ifndef SESIMOS_ASYNC_H
|
||||||
|
#define SESIMOS_ASYNC_H
|
||||||
|
|
||||||
|
#define async_read(fd, cb, arg, err_cb, err_arg) async(fd, 0, 0, cb, arg, err, err_arg)
|
||||||
|
#define async_read_keep(fd, cb, arg, err_cb, err_arg) async(fd, 0, 0, cb, arg, err, err_arg)
|
||||||
|
|
||||||
|
int async(int fd, int events, int flags, void (*cb)(void *), void *arg, void (*err_cb)(void *), void *err_arg);
|
||||||
|
|
||||||
|
void async_thread(void);
|
||||||
|
|
||||||
|
#endif //SESIMOS_ASYNC_H
|
@ -34,6 +34,8 @@
|
|||||||
volatile sig_atomic_t server_keep_alive = 1;
|
volatile sig_atomic_t server_keep_alive = 1;
|
||||||
struct timeval client_timeout = {.tv_sec = CLIENT_TIMEOUT, .tv_usec = 0};
|
struct timeval client_timeout = {.tv_sec = CLIENT_TIMEOUT, .tv_usec = 0};
|
||||||
|
|
||||||
|
static const char *color_table[] = {"\x1B[31m", "\x1B[32m", "\x1B[33m", "\x1B[34m", "\x1B[35m", "\x1B[36m"};
|
||||||
|
|
||||||
host_config *get_host_config(const char *host) {
|
host_config *get_host_config(const char *host) {
|
||||||
for (int i = 0; i < CONFIG_MAX_HOST_CONFIG; i++) {
|
for (int i = 0; i < CONFIG_MAX_HOST_CONFIG; i++) {
|
||||||
host_config *hc = &config->hosts[i];
|
host_config *hc = &config->hosts[i];
|
||||||
@ -814,8 +816,6 @@ int client_handler(sock *client, unsigned long client_num) {
|
|||||||
client_ctx_t ctx;
|
client_ctx_t ctx;
|
||||||
char log_client_prefix[256], log_conn_prefix[512];
|
char log_client_prefix[256], log_conn_prefix[512];
|
||||||
|
|
||||||
char *color_table[] = {"\x1B[31m", "\x1B[32m", "\x1B[33m", "\x1B[34m", "\x1B[35m", "\x1B[36m"};
|
|
||||||
|
|
||||||
signal(SIGINT, client_terminate);
|
signal(SIGINT, client_terminate);
|
||||||
signal(SIGTERM, client_terminate);
|
signal(SIGTERM, client_terminate);
|
||||||
|
|
||||||
|
22
src/server.c
22
src/server.c
@ -49,8 +49,10 @@ static int ssl_servername_cb(SSL *ssl, int *ad, void *arg) {
|
|||||||
return SSL_TLSEXT_ERR_OK;
|
return SSL_TLSEXT_ERR_OK;
|
||||||
}
|
}
|
||||||
|
|
||||||
void destroy(int sig) {
|
void terminate_forcefully(int sig) {
|
||||||
critical("Terminating forcefully!");
|
fprintf(stderr, "\n");
|
||||||
|
notice("Terminating forcefully!");
|
||||||
|
|
||||||
int status = 0;
|
int status = 0;
|
||||||
int ret;
|
int ret;
|
||||||
int kills = 0;
|
int kills = 0;
|
||||||
@ -71,20 +73,20 @@ void destroy(int sig) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (kills > 0) {
|
if (kills > 0) {
|
||||||
critical("Killed %i child process(es)", kills);
|
notice("Killed %i child process(es)", kills);
|
||||||
}
|
}
|
||||||
cache_unload();
|
cache_unload();
|
||||||
config_unload();
|
config_unload();
|
||||||
exit(2);
|
exit(2);
|
||||||
}
|
}
|
||||||
|
|
||||||
void terminate(int sig) {
|
void terminate_gracefully(int sig) {
|
||||||
fprintf(stderr, "\n");
|
fprintf(stderr, "\n");
|
||||||
notice("Terminating gracefully...");
|
notice("Terminating gracefully...");
|
||||||
active = 0;
|
|
||||||
|
|
||||||
signal(SIGINT, destroy);
|
active = 0;
|
||||||
signal(SIGTERM, destroy);
|
signal(SIGINT, terminate_forcefully);
|
||||||
|
signal(SIGTERM, terminate_forcefully);
|
||||||
|
|
||||||
for (int i = 0; i < NUM_SOCKETS; i++) {
|
for (int i = 0; i < NUM_SOCKETS; i++) {
|
||||||
shutdown(sockets[i], SHUT_RDWR);
|
shutdown(sockets[i], SHUT_RDWR);
|
||||||
@ -231,8 +233,8 @@ int main(int argc, const char *argv[]) {
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
signal(SIGINT, terminate);
|
signal(SIGINT, terminate_gracefully);
|
||||||
signal(SIGTERM, terminate);
|
signal(SIGTERM, terminate_gracefully);
|
||||||
|
|
||||||
if ((ret = geoip_init(geoip_dir)) != 0) {
|
if ((ret = geoip_init(geoip_dir)) != 0) {
|
||||||
if (ret == -1) {
|
if (ret == -1) {
|
||||||
@ -306,7 +308,7 @@ int main(int argc, const char *argv[]) {
|
|||||||
ready_sockets_num = poll(poll_fds, NUM_SOCKETS, 1000);
|
ready_sockets_num = poll(poll_fds, NUM_SOCKETS, 1000);
|
||||||
if (ready_sockets_num < 0) {
|
if (ready_sockets_num < 0) {
|
||||||
critical("Unable to poll sockets");
|
critical("Unable to poll sockets");
|
||||||
terminate(0);
|
terminate_gracefully(0);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user