git » nmdb » commit 2103844

diff --git a/libnmdb/test2c.c b/libnmdb/test2c.c
index 0d98824..6570c1a 100644
--- a/libnmdb/test2c.c
+++ b/libnmdb/test2c.c
@@ -25,6 +25,11 @@ int main(int argc, char **argv)
 	times = atoi(argv[1]);
 	ksize = atoi(argv[2]);
 	vsize = atoi(argv[3]);
+	if (times < 1 || ksize < 1 || vsize < 1) {
+		printf("Error: TIMES, KSIZE and VSIZE must be >= 1\n");
+		return 1;
+	}
+
 	key = malloc(ksize);
 	memset(key, 0, ksize);
 	val = malloc(vsize);
diff --git a/libnmdb/test2d.c b/libnmdb/test2d.c
index 8ee48b5..e55b698 100644
--- a/libnmdb/test2d.c
+++ b/libnmdb/test2d.c
@@ -25,6 +25,11 @@ int main(int argc, char **argv)
 	times = atoi(argv[1]);
 	ksize = atoi(argv[2]);
 	vsize = atoi(argv[3]);
+	if (times < 1 || ksize < 1 || vsize < 1) {
+		printf("Error: TIMES, KSIZE and VSIZE must be >= 1\n");
+		return 1;
+	}
+
 	key = malloc(ksize);
 	memset(key, 0, ksize);
 	val = malloc(vsize);