From 93f64ffc55d29947614f234a5c6deddcbc2d4e81 Mon Sep 17 00:00:00 2001
From: okan <okan>
Date: Fri, 16 Nov 2012 14:15:48 +0000
Subject: add some checks

---
 xmalloc.c | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'xmalloc.c')

diff --git a/xmalloc.c b/xmalloc.c
index 1d0c581..d0697fa 100644
--- a/xmalloc.c
+++ b/xmalloc.c
@@ -23,6 +23,7 @@
 
 #include <err.h>
 #include <errno.h>
+#include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -35,6 +36,8 @@ xmalloc(size_t siz)
 {
 	void	*p;
 
+	if (siz == 0)
+		errx(1, "xmalloc: zero size");
 	if ((p = malloc(siz)) == NULL)
 		err(1, "malloc");
 
@@ -46,6 +49,10 @@ xcalloc(size_t no, size_t siz)
 {
 	void	*p;
 
+	if (siz == 0 || no == 0)
+		errx(1, "xcalloc: zero size");
+	if (SIZE_MAX / no < siz)
+		errx(1, "xcalloc: no * siz > SIZE_MAX");
 	if ((p = calloc(no, siz)) == NULL)
 		err(1, "calloc");
 
-- 
cgit 1.4.1