Get full coverage of aplwc-repl.

This required wrapping malloc so that it can fail under our control.

FossilOrigin-Name: a27ccb5f02994393a518419bae15fd26ff0016c734d96350ae2137a1d3c0937c

1 files changed, 9 insertions, 10 deletions

diff --git a/src/bin/main.c b/src/bin/main.c
index 2538248..80b2ee0 100644
--- a/src/bin/main.c
+++ b/src/bin/main.c
@@ -1,6 +1,5 @@
 #include <stdio.h>
-#include <stdlib.h>
-#include <aplwc.h>
+#include "aplwc_repl.h"
 
 int
 main(void)
@@ -8,24 +7,24 @@ main(void)
 	struct aplwc *aplwc;
 	uint8_t *data;
 	int ret;
+	size_t memsize;
 
-	data = malloc(APLWC_MIN_MEM);
+	memsize = getenv_ulong("APLWC_MEM", APLWC_MIN_MEM);
+
+	data = malloc(memsize);
 	if(data == NULL){
 		fprintf(stderr, "Couldn't allocate memory\n");
 		return 1;
 	}
 
-	ret = aplwc_new(&aplwc, data, APLWC_MIN_MEM);
+	ret = aplwc_new(&aplwc, data, memsize);
 	if(ret){
-		fprintf(stderr, "Couldn't create new aplwc instance: %s\n", aplwc_errstr(aplwc));
+		fprintf(stderr, "Couldn't create new aplwc instance: %s\n", aplwc_errstr_rc(ret));
+		free(data);
 		return 1;
 	}
 
-	ret = aplwc_destroy(aplwc);
-	if(ret){
-		fprintf(stderr, "Couldn't destroy the aplwc instance: %s\n", aplwc_errstr(aplwc));
-		return 1;
-	}
+	aplwc_destroy(aplwc);
 
 	free(data);