1 files changed, 23 insertions, 14 deletions

diff --git a/src/bin/main.c b/src/bin/main.c
index 99149f6..42ab963 100644
--- a/src/bin/main.c
+++ b/src/bin/main.c
@@ -4,7 +4,8 @@
 #include <errno.h>
 #include "aplwc_repl.h"
 
-static void *my_alloc(size_t min, size_t max, size_t *actual)
+static void *
+my_alloc(size_t min, size_t max, size_t *actual)
 {
 	UNUSED_PARAMETER(max);
 
@@ -12,23 +13,29 @@ static void *my_alloc(size_t min, size_t max, size_t *actual)
 	return malloc(min);
 }
 
+static void
+my_dealloc(void *p)
+{
+	free(p);
+}
+
 int
 main(int argc, char *argv[])
 {
 	struct aplwc_fns fns;
 	struct aplwc *aplwc;
-	struct aplwc_line *line;
-	struct aplwc_value *value;
+	enum aplwc_rc ret;
+	AplwcLine line;
+	AplwcValue value;
 	char *input, *src;
-	int ret, exit_code, line_number;
+	int exit_code, line_number;
 	size_t input_size, src_size, i;
 	int opt, xflag, input_file;
 	long tmp;
 	FILE *f;
 
-	input = NULL;
-	line = NULL;
-	value = NULL;
+	aplwc_clear_line(&line);
+	aplwc_clear_value(&value);
 	xflag = 0;
 	input_file = 0;
 	exit_code = 0;
@@ -36,6 +43,7 @@ main(int argc, char *argv[])
 
 	memset(&fns, 0, sizeof(fns));
 	fns.alloc = my_alloc;
+	fns.dealloc = my_dealloc;
 
 	while((opt = getopt(argc, argv, "x")) != -1){
 		switch(opt){
@@ -61,7 +69,7 @@ usage:
 	else
 		goto usage;
 
-	ret = aplwc_new(&aplwc, fns);
+	aplwc = aplwc_new(fns, &ret);
 	if(ret){
 		fprintf(stderr, "Couldn't create new aplwc instance: %s\n", aplwc_errstr_rc(ret));
 		return 1;
@@ -117,19 +125,20 @@ file_size:
 	for(;;){
 		if(input)
 			free(input);
-		if(line)
+		if(aplwc_valid_line(line))
 			aplwc_free_line(line);
-		if(value)
+		if(aplwc_valid_value(value))
 			aplwc_free_value(value);
 
 		input = NULL;
 		if(getline(&input, &input_size, stdin) < 0){
 			free(input);
+			input = NULL;
 			break;
 		}else{
 			line_number++;
 			line = aplwc_read(aplwc, input);
-			if(line == NULL){
+			if(!aplwc_valid_line(line)){
 line_error:
 				fprintf(stderr, "error on line %d: %s: %s\n", line_number, aplwc_errstr(aplwc), input);
 				if(input_file){
@@ -139,7 +148,7 @@ line_error:
 				continue;
 			}
 			value = aplwc_eval(line);
-			if(value == NULL)
+			if(!aplwc_valid_value(value))
 				goto line_error;
 			fprintf(stdout, "%s\n", aplwc_value_string(value));
 		}
@@ -148,9 +157,9 @@ line_error:
 exit:
 	if(input)
 		free(input);
-	if(line)
+	if(aplwc_valid_line(line))
 		aplwc_free_line(line);
-	if(value)
+	if(aplwc_valid_value(value))
 		aplwc_free_value(value);
 
 	aplwc_destroy(aplwc);