Tidied server.c

This commit is contained in:
James Booth
2015-05-19 00:55:20 +01:00
parent abc78757b6
commit 27f30aaa8a

View File

@@ -26,7 +26,8 @@
static XMPPClient *client; static XMPPClient *client;
static void _shutdown(void); static void _shutdown(void);
static void _shutdown_sig(int sig); static void _signal_handler(int sig);
static int listen_socket; static int listen_socket;
void void
@@ -35,24 +36,25 @@ write_stream(const char * const stanza)
int sent = 0; int sent = 0;
int to_send = strlen(stanza); int to_send = strlen(stanza);
char *marker = (char*)stanza; char *marker = (char*)stanza;
while (to_send > 0 && ((sent = write(client->sock, marker, to_send)) > 0)) { while (to_send > 0 && ((sent = write(client->sock, marker, to_send)) > 0)) {
to_send -= sent; to_send -= sent;
marker += sent; marker += sent;
} }
log_println("SENT: %s", stanza); log_println("SENT: %s", stanza);
} }
int int
read_stream(void) read_stream(void)
{ {
int read_size;
char buf[2]; char buf[2];
memset(buf, 0, sizeof(buf)); memset(buf, 0, sizeof(buf));
GString *stream = g_string_new(""); GString *stream = g_string_new("");
errno = 0; errno = 0;
while (TRUE) { while (TRUE) {
read_size = recv(client->sock, buf, 1, 0); int read_size = recv(client->sock, buf, 1, 0);
// client disconnect // client disconnect
if (read_size == 0) { if (read_size == 0) {
@@ -79,7 +81,7 @@ read_stream(void)
} }
} }
// read a byte, feed parser // success, feed parser with byte
log_print_chars("%c", buf[0]); log_print_chars("%c", buf[0]);
parser_feed(buf, 1); parser_feed(buf, 1);
g_string_append_len(stream, buf, read_size); g_string_append_len(stream, buf, read_size);
@@ -100,9 +102,11 @@ stream_start_callback(void)
{ {
log_print_chars("\n"); log_print_chars("\n");
log_println("--> Stream start callback fired"); log_println("--> Stream start callback fired");
write_stream(XML_START); write_stream(XML_START);
write_stream(STREAM_RESP); write_stream(STREAM_RESP);
write_stream(FEATURES); write_stream(FEATURES);
log_print("RECV: "); log_print("RECV: ");
} }
@@ -149,14 +153,14 @@ _start_server_cb(void* userdata)
{ {
struct sockaddr_in client_addr; struct sockaddr_in client_addr;
// set socket to nonblocking mode // listen socket non blocking
int res = fcntl(listen_socket, F_SETFL, fcntl(listen_socket, F_GETFL, 0) | O_NONBLOCK); int res = fcntl(listen_socket, F_SETFL, fcntl(listen_socket, F_GETFL, 0) | O_NONBLOCK);
if (res == -1) { if (res == -1) {
log_println("Error setting nonblocking on listen socket: %s", strerror(errno)); log_println("Error setting nonblocking on listen socket: %s", strerror(errno));
exit(0); exit(0);
} }
// set socket recv timeout // listen socket read timeout
struct timeval timeout; struct timeval timeout;
timeout.tv_sec = 0; timeout.tv_sec = 0;
timeout.tv_usec = 1000 * 10; timeout.tv_usec = 1000 * 10;
@@ -167,7 +171,7 @@ _start_server_cb(void* userdata)
exit(0); exit(0);
} }
// connection accept // wait for connection
int c = sizeof(struct sockaddr_in); int c = sizeof(struct sockaddr_in);
int client_socket; int client_socket;
errno = 0; errno = 0;
@@ -179,14 +183,14 @@ _start_server_cb(void* userdata)
errno = 0; errno = 0;
} }
// set socket to nonblocking mode // client socket non blocking
res = fcntl(client_socket, F_SETFL, fcntl(client_socket, F_GETFL, 0) | O_NONBLOCK); res = fcntl(client_socket, F_SETFL, fcntl(client_socket, F_GETFL, 0) | O_NONBLOCK);
if (res == -1) { if (res == -1) {
log_println("Error setting nonblocking on client socket: %s", strerror(errno)); log_println("Error setting nonblocking on client socket: %s", strerror(errno));
exit(0); exit(0);
} }
// set socket recv timeout // client socket read timeout
errno = 0; errno = 0;
res = setsockopt(client_socket, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout, sizeof(timeout)); res = setsockopt(client_socket, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout, sizeof(timeout));
if (res < 0) { if (res < 0) {
@@ -210,8 +214,9 @@ server_run(int port)
log_init(); log_init();
log_println("Starting on port: %d...", port); log_println("Starting on port: %d...", port);
signal(SIGSTOP, _shutdown_sig); // signal handlers
signal(SIGINT, _shutdown_sig); signal(SIGSTOP, _signal_handler);
signal(SIGINT, _signal_handler);
atexit(_shutdown); atexit(_shutdown);
// create listen socket // create listen socket
@@ -247,6 +252,7 @@ server_run(int port)
prime_init(); prime_init();
// start client processor thread
pthread_t server_thread; pthread_t server_thread;
int res = pthread_create(&server_thread, NULL, _start_server_cb, NULL); int res = pthread_create(&server_thread, NULL, _start_server_cb, NULL);
if (res != 0) { if (res != 0) {
@@ -260,11 +266,11 @@ server_run(int port)
static void static void
_shutdown(void) _shutdown(void)
{ {
_shutdown_sig(-1); _signal_handler(-1);
} }
static void static void
_shutdown_sig(int sig) _signal_handler(int sig)
{ {
log_println("SHUTDOWN"); log_println("SHUTDOWN");
// stanza_show_all(); // stanza_show_all();