瀏覽代碼

Added a malloc fail guard.

Florian DORMONT 11 年之前
父節點
當前提交
8c50a3362f
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      wall.c

+ 4 - 0
wall.c

@@ -7,6 +7,10 @@ Wall *addWall(Wall *list, int d, int h, int id, int line)
 	tmp = list;
 
 	new = malloc(sizeof(Wall));
+	if(new == NULL) {
+		// Uh oh
+		return list;
+	}
 	new->d = d;
 	new->h = h;
 	new->id = id;