diff options
| -rw-r--r-- | Makefile | 9 | ||||
| -rw-r--r-- | nDPIsrvd.c | 97 |
2 files changed, 104 insertions, 2 deletions
@@ -35,11 +35,16 @@ endif RM = rm -f +all: nDPId nDPIsrvd + nDPId: help nDPId.c $(CC) $(PROJECT_CFLAGS) $(CFLAGS) nDPId.c -o $@ $(LDFLAGS) $(LIBS) +nDPIsrvd: nDPIsrvd.c + $(CC) $(PROJECT_CFLAGS) $(CFLAGS) nDPIsrvd.c -o $@ $(LDFLAGS) $(LIBS) + clean: - $(RM) nDPId + $(RM) -f nDPId nDPIsrvd help: @echo 'CC = $(CC)' @@ -51,4 +56,4 @@ help: @echo 'ENABLE_SANITIZER = $(ENABLE_SANITIZER)' @echo 'ENABLE_SANITIZER_THREAD = $(ENABLE_SANITIZER_THREAD)' -.PHONY: help +.PHONY: all clean help diff --git a/nDPIsrvd.c b/nDPIsrvd.c new file mode 100644 index 000000000..59ecba355 --- /dev/null +++ b/nDPIsrvd.c @@ -0,0 +1,97 @@ +#include <arpa/inet.h> +#include <errno.h> +#include <fcntl.h> +#include <netdb.h> +#include <linux/un.h> +#include <stdio.h> +#include <stdlib.h> +#include <stdint.h> +#include <string.h> +#include <syslog.h> +#include <sys/socket.h> +#include <sys/types.h> +#include <unistd.h> + +static char json_sockpath[UNIX_PATH_MAX] = "/tmp/ndpid-collector.sock"; +static char serv_listen_addr[INET6_ADDRSTRLEN] = "127.0.0.1"; +static uint16_t serv_listen_port = 7000; +static int json_sockfd; +static int serv_sockfd; + +static int create_listen_sockets(void) +{ + json_sockfd = socket(AF_UNIX, SOCK_STREAM, 0); + serv_sockfd = socket(AF_INET, SOCK_STREAM, 0); + if (json_sockfd < 0 || serv_sockfd < 0) + { + syslog(LOG_DAEMON | LOG_ERR, "Error opening socket: %s", strerror(errno)); + return 1; + } + + // This helps avoid spurious EADDRINUSE when the previous instance of this + // server died. + int opt = 1; + if (setsockopt(json_sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) < 0 || + setsockopt(serv_sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) < 0) + { + syslog(LOG_DAEMON | LOG_ERR, "setsockopt with SO_REUSEADDR failed: %s", strerror(errno)); + return 1; + } + + struct sockaddr_un json_addr; + json_addr.sun_family = AF_UNIX; + if (snprintf(json_addr.sun_path, sizeof(json_addr.sun_path), "%s", json_sockpath) <= 0) + { + syslog(LOG_DAEMON | LOG_ERR, "snprintf failed: %s", strerror(errno)); + return 1; + } + + struct sockaddr_in serv_addr; + memset(&serv_addr, 0, sizeof(serv_addr)); + serv_addr.sin_family = AF_INET; + serv_addr.sin_addr.s_addr = INADDR_ANY; + serv_addr.sin_port = htons(serv_listen_port); + if (inet_ntop(AF_INET, &serv_addr.sin_addr, &serv_listen_addr[0], INET_ADDRSTRLEN) == NULL) + { + syslog(LOG_DAEMON | LOG_ERR, "Error converting an internet address: %s", strerror(errno)); + return 1; + } + + if (bind(json_sockfd, (struct sockaddr *)&json_addr, sizeof(json_addr)) < 0) + { + syslog(LOG_DAEMON | LOG_ERR, "Error on binding a JSON socket: %s", strerror(errno)); + return 1; + } + + if (bind(serv_sockfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) + { + syslog(LOG_DAEMON | LOG_ERR, "Error on binding the INET socket: %s", strerror(errno)); + return 1; + } + + if (listen(json_sockfd, 16) < 0 || listen(serv_sockfd, 16) < 0) + { + syslog(LOG_DAEMON | LOG_ERR, "Error on listen: %s", strerror(errno)); + return 1; + } + + return 0; +} + +int main(void) +{ + openlog("nDPIsrvd", LOG_CONS | LOG_PERROR, LOG_DAEMON); + + if (create_listen_sockets() != 0) + { + return 1; + } + + getchar(); + + close(json_sockfd); + close(serv_sockfd); + unlink(json_sockpath); + + return 0; +} |