snake null

cv9/snake.c

@@ -3,9 +3,6 @@
 
 struct snake* add_snake(struct snake* snake,int x,int y){
     struct snake* hlavicka = calloc(1,sizeof(struct snake));
-    if (hlavicka == NULL) {
-        return NULL; 
-    }
 
     hlavicka->x = x;
     hlavicka->y = y;
@@ -16,7 +13,18 @@ struct snake* add_snake(struct snake* snake,int x,int y){
 }
 
 struct snake* remove_snake(struct snake* snake){
+    if (snake == NULL) {
         return NULL; 
+    }
+    
+    struct snake* nova_hlavicka = snake->next;
+    free(snake);
+
+    if (nova_hlavicka != NULL) {
+        nova_hlavicka->next = NULL;
+    }
+
+    return nova_hlavicka;
 }
 
 void free_snake(struct snake* sn){