Fixed errors in address generation
[umurmur.git] / src / server.c
index 05f5213abc058575cd22b1591ecb68553188cf3d..5394b09dd545e05ad795c59fbf43f3f19684c303 100644 (file)
 #define TCP_SOCK 0
 #define UDP_SOCK 1
 
+#define TCP6_SOCK 2
+#define UDP6_SOCK 3
+
 /* globals */
 int udpsock;
 bool_t shutdown_server;
 extern char *bindaddr;
+extern char *bindaddr6;
 extern int bindport;
+extern int bindport6;
 
-void Server_run()
+struct sockaddr_storage** Server_setupAddressesAndPorts()
 {
-       int timeout = 1000, rc;
-       struct pollfd *pollfds;
-       int tcpsock, sockopt = 1;
-       struct sockaddr_in sin;
-       int val, clientcount;
-       etimer_t janitorTimer;
-       unsigned short port;
-       in_addr_t inet_address;
+  struct sockaddr_storage** addresses = malloc(2 * sizeof(void*));
 
-       /* 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");
+  struct sockaddr_storage* v4address = calloc(1, sizeof(struct sockaddr_storage));
+  v4address->ss_family = AF_INET;
+  v4address->ss_len = sizeof(struct sockaddr_storage);
+  struct sockaddr_storage* v6address = calloc(1, sizeof(struct sockaddr_storage));
+  v6address->ss_family = AF_INET;
+  v6address->ss_len = sizeof(struct sockaddr_storage);
 
-       /* 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));
+  int error = 0;
 
-       /* Prepare TCP socket */
-       memset(&sin, 0, sizeof(sin));
-       tcpsock = socket(PF_INET, SOCK_STREAM, 0);
-       if (tcpsock < 0)
-               Log_fatal("socket");
-       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 = port;
-       sin.sin_addr.s_addr = inet_address;
+  error = inet_pton(AF_INET, (!bindaddr) ? ((getStrConf(BINDADDR)) ? getStrConf(BINDADDR) : "0.0.0.0")
+                                         : bindaddr, &(((struct sockaddr_in*)v4address)->sin_addr));
+  if (error == 0) Log_fatal("Invalid IPv4 address supplied!");
 
-       rc = bind(tcpsock, (struct sockaddr *) &sin, sizeof (struct sockaddr_in));
-       if (rc < 0) Log_fatal("bind: %s", strerror(errno));
-       rc = listen(tcpsock, 3);
-       if (rc < 0) Log_fatal("listen");
-       fcntl(tcpsock, F_SETFL, O_NONBLOCK);
+  error = inet_pton(AF_INET6, (!bindaddr6) ? ((getStrConf(BINDADDR6)) ? getStrConf(BINDADDR6) : "::")
+                                         : bindaddr6, &(((struct sockaddr_in6*)v6address)->sin6_addr));
+  if (error == 0) Log_fatal("Invalid IPv6 address supplied!");
 
-       pollfds[LISTEN_SOCK].fd = tcpsock;
-       pollfds[LISTEN_SOCK].events = POLLIN;
+  ((struct sockaddr_in*)v4address)->sin_port = htons((bindport) ? bindport : getIntConf(BINDPORT));
+  ((struct sockaddr_in6*)v6address)->sin6_port = htons((bindport) ? bindport : getIntConf(BINDPORT));
 
-       /* Prepare UDP socket */
-       memset(&sin, 0, sizeof(sin));
-       udpsock = socket(PF_INET, SOCK_DGRAM, 0);
-       sin.sin_family = AF_INET;
-       sin.sin_port = port;
-       sin.sin_addr.s_addr = inet_address;
+  addresses[0] = v4address;
+  addresses[1] = v6address;
 
-       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_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_warn("Server: Failed to set TOS for UDP Socket");
+  return addresses;
+}
 
-       fcntl(udpsock, F_SETFL, O_NONBLOCK);
-       pollfds[UDP_SOCK].fd = udpsock;
-       pollfds[UDP_SOCK].events = POLLIN | POLLHUP | POLLERR;
+void Server_runLoop(struct pollfd* pollfds)
+  {
+  int timeout = 1000, rc, clientcount;
+       etimer_t janitorTimer;
 
        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;
                int i;
@@ -189,6 +151,75 @@ void Server_run()
                        }
                }
        }
+  }
+
+void Server_run()
+{
+       int rc;
+       struct pollfd *pollfds;
+       int tcpsock, sockopt = 1;
+       struct sockaddr_in sin;
+       int val;
+       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 */
+  struct sockaddr_storage** addresses = Server_setupAddressesAndPorts();
+
+       /* Prepare TCP socket */
+       memset(&sin, 0, sizeof(sin));
+       tcpsock = socket(PF_INET, SOCK_STREAM, 0);
+       if (tcpsock < 0)
+               Log_fatal("socket");
+       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 = 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);
+       if (rc < 0) Log_fatal("listen");
+       fcntl(tcpsock, F_SETFL, O_NONBLOCK);
+
+       pollfds[LISTEN_SOCK].fd = tcpsock;
+       pollfds[LISTEN_SOCK].events = POLLIN;
+
+       /* Prepare UDP socket */
+       memset(&sin, 0, sizeof(sin));
+       udpsock = socket(PF_INET, SOCK_DGRAM, 0);
+       sin.sin_family = AF_INET;
+       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_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_warn("Server: Failed to set TOS for UDP Socket");
+
+       fcntl(udpsock, F_SETFL, O_NONBLOCK);
+       pollfds[UDP_SOCK].fd = udpsock;
+       pollfds[UDP_SOCK].events = POLLIN | POLLHUP | POLLERR;
+
+       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 */
+  Server_runLoop(pollfds);
 
        /* Disconnect clients */
        Client_disconnect_all();