Update copyright year.

[umurmur.git] / src / server.c

diff --git a/src/server.c b/src/server.c
index 7cabfb37aca8e5926f06db84577d923732fae87e..439b6c3e775adf90b8b4d388fb29058a6c575d15 100644 (file)
--- a/src/server.c
+++ b/src/server.c
@@ -1,5 +1,5 @@
-/* Copyright (C) 2009, Martin Johansson <martin@fatbob.nu>
-   Copyright (C) 2005-2009, Thorvald Natvig <thorvald@natvig.com>
+/* Copyright (C) 2009-2013, Martin Johansson <martin@fatbob.nu>
+   Copyright (C) 2005-2013, Thorvald Natvig <thorvald@natvig.com>
 
    All rights reserved.
 
@@ -39,18 +39,23 @@
 #include <unistd.h>
 #include <fcntl.h>
 #include <errno.h>
+#include <stdlib.h>
 
 #include "client.h"
 #include "conf.h"
 #include "log.h"
 #include "timer.h"
+#include "version.h"
 
 #define LISTEN_SOCK 0
 #define TCP_SOCK 0
 #define UDP_SOCK 1
 
-int udpsock; /* XXX restructure! */
+/* globals */
+int udpsock; 
 bool_t shutdown_server;
+extern char *bindaddr;
+extern int bindport;
 
 void Server_run()
 {
@@ -60,11 +65,27 @@ void Server_run()
        struct sockaddr_in sin;
        int val, clientcount;
        etimer_t janitorTimer;
-
+       unsigned short port;
+       in_addr_t inet_address;
+       
        /* max clients + listen sock + udp sock + client connecting that will be disconnected */
        pollfds = malloc((getIntConf(MAX_CLIENTS) + 3) * sizeof(struct pollfd));
        if (pollfds == NULL)
                Log_fatal("out of memory");
+
+       /* Figure out bind address and port */
+       if (bindport != 0)
+               port = htons(bindport);
+       else
+               port = htons(getIntConf(BINDPORT));
+       
+       if (bindaddr != NULL && inet_addr(bindaddr) != -1)
+               inet_address = inet_addr(bindaddr);
+       else if (inet_addr(getStrConf(BINDADDR)) !=  -1)
+               inet_address = inet_addr(getStrConf(BINDADDR));
+       else
+               inet_address = inet_addr("0.0.0.0");
+       Log_info("Bind to %s:%hu", inet_address == 0 ? "*" : inet_ntoa(*((struct in_addr *)&inet_address)), ntohs(port));
        
        /* Prepare TCP socket */
        memset(&sin, 0, sizeof(sin));
@@ -74,8 +95,9 @@ void Server_run()
        if (setsockopt(tcpsock, SOL_SOCKET, SO_REUSEADDR, &sockopt, sizeof(int)) != 0)
                Log_fatal("setsockopt: %s", strerror(errno));
        sin.sin_family = AF_INET;
-       sin.sin_port = htons(getIntConf(BINDPORT));
-       sin.sin_addr.s_addr = inet_addr(getStrConf(BINDADDR)) ==  -1 ? inet_addr("0.0.0.0") : inet_addr(getStrConf(BINDADDR));
+       sin.sin_port = port;    
+       sin.sin_addr.s_addr = inet_address;
+       
        rc = bind(tcpsock, (struct sockaddr *) &sin, sizeof (struct sockaddr_in));
        if (rc < 0) Log_fatal("bind: %s", strerror(errno));
        rc = listen(tcpsock, 3);
@@ -89,19 +111,20 @@ void Server_run()
        memset(&sin, 0, sizeof(sin));
        udpsock = socket(PF_INET, SOCK_DGRAM, 0);
        sin.sin_family = AF_INET;
-       sin.sin_port = htons(getIntConf(BINDPORT));
-       sin.sin_addr.s_addr = inet_addr(getStrConf(BINDADDR)) ==  -1 ? inet_addr("0.0.0.0") : inet_addr(getStrConf(BINDADDR));
+       sin.sin_port = port;
+       sin.sin_addr.s_addr = inet_address;
+       
        rc = bind(udpsock, (struct sockaddr *) &sin, sizeof (struct sockaddr_in));
        if (rc < 0)
                Log_fatal("bind %d %s: %s", getIntConf(BINDPORT), getStrConf(BINDADDR), strerror(errno));
        val = 0xe0;
        rc = setsockopt(udpsock, IPPROTO_IP, IP_TOS, &val, sizeof(val));
        if (rc < 0)
-               Log_fatal("Server: Failed to set TOS for UDP Socket");
+               Log_warn("Server: Failed to set TOS for UDP Socket");
        val = 0x80;
        rc = setsockopt(udpsock, IPPROTO_IP, IP_TOS, &val, sizeof(val));
        if (rc < 0)
-               Log_fatal("Server: Failed to set TOS for UDP Socket");
+               Log_warn("Server: Failed to set TOS for UDP Socket");
        
        fcntl(udpsock, F_SETFL, O_NONBLOCK);
        pollfds[UDP_SOCK].fd = udpsock;
@@ -109,6 +132,10 @@ void Server_run()
        
        Timer_init(&janitorTimer);
        
+       Log_info("uMurmur version %s ('%s') protocol version %d.%d.%d",
+                UMURMUR_VERSION, UMURMUR_CODENAME, PROTVER_MAJOR, PROTVER_MINOR, PROTVER_PATCH);
+       Log_info("Visit http://code.google.com/p/umurmur/");
+       
        /* Main server loop */
        while (!shutdown_server) {
                struct sockaddr_in remote;
@@ -144,9 +171,10 @@ void Server_run()
                        tcpfd = accept(pollfds[LISTEN_SOCK].fd, (struct sockaddr*)&remote, &addrlen);
                        fcntl(tcpfd, F_SETFL, O_NONBLOCK);
                        setsockopt(tcpfd, IPPROTO_TCP, TCP_NODELAY, (char *) &flag, sizeof(int));
-                       Log_info("Connection from %s port %d\n", inet_ntoa(remote.sin_addr),
-                                        ntohs(remote.sin_port));
-                       Client_add(tcpfd, &remote);
+                       Log_debug("Connection from %s port %d\n", inet_ntoa(remote.sin_addr),
+                                         ntohs(remote.sin_port));
+                       if (Client_add(tcpfd, &remote) < 0)
+                               close(tcpfd);
                }
 
                if (pollfds[UDP_SOCK].revents) {