session: add error checking to session creation

Add a check in init_session that returns -1 if malloc cannot create a
new session.

Signed-off-by: Danny Holman <dholman@gymli.org>

1 files changed, 5 insertions, 3 deletions

diff --git a/server/src/session.c b/server/src/session.c
index 2737f4b..74b29a7 100644
--- a/server/src/session.c
+++ b/server/src/session.c
@@ -1,12 +1,14 @@
 #include <session.h>
-#include <util.h>
-#include <mini-rat.h>
+#include <stdlib.h>
+#include <unistd.h>
 
 struct list_head sessions;
-uint16_t next_id = 1;
+int next_id = 1;
 
 int init_session(int socket) {
         struct session *ses = malloc(sizeof(struct session));
+        if (ses == NULL)
+                return -1;
         ses->id = next_id;
         next_id++;
         ses->socket = socket;