summary refs log tree commit diff
path: root/filter.c
diff options
context:
space:
mode:
authorJason A. Donenfeld <Jason@zx2c4.com>2014-01-14 18:07:23 +0100
committerJason A. Donenfeld <Jason@zx2c4.com>2014-01-14 18:09:52 +0100
commit6ca734da8fb246ad2272826331e0d56428b96fa1 (patch)
tree01abfea3df45135b1114996de29093b6eed7023a /filter.c
parentce56d89a2662549acd178292450798f5ffcd4bc6 (diff)
filter: allow returning exit code from filter
Filters can now indicate a status back to cgit by means of the exit code
for exec, or the return value from close for Lua.

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
Diffstat (limited to 'filter.c')
-rw-r--r--filter.c14
1 files changed, 9 insertions, 5 deletions
diff --git a/filter.c b/filter.c
index 4d4acaf..0cce7bb 100644
--- a/filter.c
+++ b/filter.c
@@ -106,7 +106,7 @@ static int open_exec_filter(struct cgit_filter *base, va_list ap)
 static int close_exec_filter(struct cgit_filter *base)
 {
 	struct cgit_exec_filter *filter = (struct cgit_exec_filter *)base;
-	int i, exit_status;
+	int i, exit_status = 0;
 
 	chk_non_negative(dup2(filter->old_stdout, STDOUT_FILENO),
 		"Unable to restore STDOUT");
@@ -114,14 +114,14 @@ static int close_exec_filter(struct cgit_filter *base)
 	if (filter->pid < 0)
 		goto done;
 	waitpid(filter->pid, &exit_status, 0);
-	if (WIFEXITED(exit_status) && !WEXITSTATUS(exit_status))
+	if (WIFEXITED(exit_status))
 		goto done;
 	die("Subprocess %s exited abnormally", filter->cmd);
 
 done:
 	for (i = 0; i < filter->base.argument_count; i++)
 		filter->argv[i + 1] = NULL;
-	return 0;
+	return WEXITSTATUS(exit_status);
 
 }
 
@@ -315,10 +315,14 @@ static int close_lua_filter(struct cgit_filter *base)
 	int ret = 0;
 
 	lua_getglobal(filter->lua_state, "filter_close");
-	if (lua_pcall(filter->lua_state, 0, 0, 0)) {
+	if (lua_pcall(filter->lua_state, 0, 1, 0)) {
 		error_lua_filter(filter);
-		ret = 1;
+		ret = -1;
+	} else {
+		ret = lua_tonumber(filter->lua_state, -1);
+		lua_pop(filter->lua_state, 1);
 	}
+
 	unhook_write();
 	return ret;
 }