From 89c48e763ef042c9faee5b7c1434dbcaf12da399 Mon Sep 17 00:00:00 2001
From: Danny Holman <dholman@gymli.org>
Date: Wed, 15 Feb 2023 22:48:35 -0600
Subject: 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>
---
 server/src/session.c | 8 +++++---
 1 file 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;
-- 
cgit v1.2.3