X-Git-Url: http://git.code-monkey.de/?a=blobdiff_plain;f=src%2Fserver.c;h=827b99b4f3b47050f4b0c068fcadb88f006fecb1;hb=6b301bd735bbbe384057a463ede871616efb5861;hp=7e0c4c6f8d82eb1cef756d194c21f393489f1d88;hpb=4c431fe65269e9b1d452855b9df8cfe80683b691;p=umurmur.git diff --git a/src/server.c b/src/server.c index 7e0c4c6..827b99b 100644 --- a/src/server.c +++ b/src/server.c @@ -1,5 +1,5 @@ -/* Copyright (C) 2009-2012, Martin Johansson - Copyright (C) 2005-2012, Thorvald Natvig +/* Copyright (C) 2009-2014, Martin Johansson + Copyright (C) 2005-2014, Thorvald Natvig All rights reserved. @@ -51,100 +51,62 @@ #define TCP_SOCK 0 #define UDP_SOCK 1 +#define TCP6_SOCK 2 +#define UDP6_SOCK 3 + /* globals */ -int udpsock; +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; - - /* 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** addresses; - /* 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)); - 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; + 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); + + int error = 0; + + 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!"); + + error = inet_pton(AF_INET, (!bindaddr6) ? ((getStrConf(BINDADDR6)) ? getStrConf(BINDADDR6) : "::") + : bindaddr6, &(((struct sockaddr_in6*)v6address)->sin6_addr)); + if (error == 0) Log_fatal("Invalid IPv6 address supplied!"); + + ((struct sockaddr_in*)v4address)->sin_port = htons((bindport) ? bindport : getIntConf(BINDPORT)); + ((struct sockaddr_in6*)v6address)->sin6_port = htons((bindport) ? bindport : getIntConf(BINDPORT)); + + addresses[0] = v4address; + addresses[1] = v6address; + + return addresses; +} + +void Server_runLoop(struct pollfd* pollfds) + { + int timeout = 1000, rc, clientcount; + etimer_t janitorTimer; - /* 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; - Timer_init(&janitorTimer); - - Log_info("uMurmur version %s protocol version %d.%d.%d", - UMURMUR_VERSION, 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; - + pollfds[UDP_SOCK].revents = 0; pollfds[TCP_SOCK].revents = 0; clientcount = Client_getfds(&pollfds[2]); - + timeout = (int)(1000000LL - (int64_t)Timer_elapsed(&janitorTimer)) / 1000LL; if (timeout <= 0) { Client_janitor(); @@ -188,7 +150,76 @@ void Server_run() Client_write_fd(pollfds[i + 2].fd); } } - } + } + } + +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();