Abstract socket creation and binding behind the HttpServer API.

src/HttpServer.c

@@ -26,6 +26,11 @@
 #include <pthread.h>
 #include <stdlib.h>
 #include <unistd.h>
+#include <stdio.h>
+
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
 
 struct HttpServer
 {
@@ -41,16 +46,43 @@ struct HttpServer
 };
 
 HttpServer *
-HttpServerCreate(int socketDesc, unsigned int nThreads, HttpHandler * requestHandler, void *handlerArgs)
+HttpServerCreate(unsigned short port, unsigned int nThreads, HttpHandler * requestHandler, void *handlerArgs)
 {
     HttpServer *server = malloc(sizeof(HttpServer));
+    struct sockaddr_in sa = {0};
 
     if (!server)
     {
         return NULL;
     }
 
-    server->sd = socketDesc;
+    server->sd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
+
+    if (server->sd < 0)
+    {
+        free(server);
+        return NULL;
+    }
+
+    sa.sin_family = AF_INET;
+    sa.sin_port = port;
+    sa.sin_addr.s_addr = htonl(INADDR_ANY);
+
+    if (bind(server->sd, (struct sockaddr *) & sa, sizeof(sa)) < 0)
+    {
+        close(server->sd);
+        free(server);
+        return NULL;
+    }
+
+    /* TODO: Make this a user-tunable parameter? */
+    if (listen(server->sd, 32) < 0)
+    {
+        close(server->sd);
+        free(server);
+        return NULL;
+    }
+
     server->nThreads = nThreads;
     server->requestHandler = requestHandler;
     server->handlerArgs = handlerArgs;
@@ -63,10 +95,14 @@ HttpServerCreate(int socketDesc, unsigned int nThreads, HttpHandler * requestHan
 void
 HttpServerFree(HttpServer * server)
 {
-    free(server);
+    if (!server)
+    {
+        return;
     }
 
-#include <stdio.h>
+    close(server->sd);
+    free(server);
+}
 
 static void *
 HttpServerEventThread(void *args)

src/Telodendria.c

@@ -31,10 +31,6 @@
 #include <grp.h>
 #include <pwd.h>
 
-#include <sys/socket.h>
-#include <netinet/in.h>
-#include <arpa/inet.h>
-
 #include <TelodendriaConfig.h>
 #include <Log.h>
 #include <HashMap.h>
@@ -93,31 +89,6 @@ TelodendriaPrintUsage(LogConfig * lc)
     Log(lc, LOG_MESSAGE, "  -h           Print this usage, then exit.");
 }
 
-static int
-TelodendriaBindSocket(unsigned short port)
-{
-    struct sockaddr_in remote = {0};
-    int s = socket(AF_INET, SOCK_STREAM, 0);
-
-    if (s < 0)
-    {
-        /* Unable to create the socket */
-        return -1;
-    }
-
-    remote.sin_family = AF_INET;
-    remote.sin_port = port;
-    remote.sin_addr.s_addr = htonl(INADDR_ANY);
-
-    if (bind(s, (struct sockaddr *) & remote, sizeof(remote)) < 0)
-    {
-        /* Unable to bind the socket */
-        return -1;
-    }
-
-    return s;
-}
-
 int
 main(int argc, char **argv)
 {
@@ -141,9 +112,6 @@ main(int argc, char **argv)
     struct passwd *userInfo;
     struct group *groupInfo;
 
-    /* Networking */
-    int httpSocket = -1;
-
     /* Signal handling */
     struct sigaction sigAction;
 
@@ -327,10 +295,11 @@ main(int argc, char **argv)
     }
 
     /* Bind the socket before possibly dropping permissions */
-    httpSocket = TelodendriaBindSocket(tConfig->listenPort);
-    if (httpSocket < 0)
+    httpServer = HttpServerCreate(tConfig->listenPort, tConfig->threads, TelodendriaHttpHandler, NULL);
+    if (!httpServer)
     {
-        Log(lc, LOG_ERROR, "Unable to bind to port %d: %s", strerror(errno));
+        Log(lc, LOG_ERROR, "Unable to create HTTP server on port %d: %s",
+            tConfig->listenPort, strerror(errno));
         exit = EXIT_FAILURE;
         goto finish;
     }
@@ -385,14 +354,6 @@ main(int argc, char **argv)
 
     Log(lc, LOG_TASK, "Starting server...");
 
-    httpServer = HttpServerCreate(httpSocket, tConfig->threads, TelodendriaHttpHandler, NULL);
-    if (!httpServer)
-    {
-        Log(lc, LOG_ERROR, "Unable to create HTTP server.");
-        exit = EXIT_FAILURE;
-        goto finish;
-    }
-
     if (!HttpServerStart(httpServer))
     {
         Log(lc, LOG_ERROR, "Unable to start HTTP server.");
@@ -400,7 +361,7 @@ main(int argc, char **argv)
         goto finish;
     }
 
-    Log(lc, LOG_MESSAGE, "Ready.");
+    Log(lc, LOG_MESSAGE, "Listening on port: %d", tConfig->listenPort);
 
     sigAction.sa_handler = TelodendriaSignalHandler;
     sigfillset(&sigAction.sa_mask);
@@ -419,10 +380,6 @@ main(int argc, char **argv)
 
 finish:
     Log(lc, LOG_TASK, "Shutting down...");
-    if (httpSocket > 0)
-    {
-        close(httpSocket);
-    }
     if (httpServer)
     {
         HttpServerFree(httpServer);

src/include/HttpServer.h

@@ -31,7 +31,7 @@ typedef struct HttpServer HttpServer;
 typedef void (HttpHandler) (HttpRequest *, HttpResponse *, void *);
 
 extern HttpServer *
- HttpServerCreate(int, unsigned int, HttpHandler *, void *);
+ HttpServerCreate(unsigned short, unsigned int, HttpHandler *, void *);
 
 extern void
  HttpServerFree(HttpServer *);