portugalfuturista/rizin
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
rizin/test/unit/test_subprocess.c
Riccardo Schirone 53bf5c497f SPDX Copyright text for all files based on history 
- Add LICENSES directory
- Download additional licenses used
- Add .reuse directory
- Add doc about SPDX/reuse

Co-authored-by: Florian Märkl <info@florianmaerkl.de>
2021-03-05 19:39:15 +08:00

262 lines
No EOL
9.8 KiB
C
Raw Blame History

// SPDX-FileCopyrightText: 2021 ret2libc <sirmy15@gmail.com>
// SPDX-License-Identifier: LGPL-3.0-only
#include <rz_util.h>
#include "minunit.h"
#define UT_TIMEOUT 100000000
static char tmp_path[1000];
const char *get_auxiliary_path(const char *s) {
char *p = rz_sys_pid_to_path(rz_sys_getpid());
char *pp = (char *)rz_str_lchr(p, RZ_SYS_DIR[0]);
if (pp) {
*pp = '\0';
}
snprintf(tmp_path, sizeof(tmp_path), "%s%s%s%s%s", p, RZ_SYS_DIR, "auxiliary", RZ_SYS_DIR, s);
free(p);
return tmp_path;
}
bool test_noargs_noinput_outerr(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-helloworld");
RzSubprocess *sp = rz_subprocess_start(exe_path, NULL, 0, NULL, NULL, 0);
mu_assert_notnull(sp, "the subprocess should be created");
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_streq(spo->out, "Hello World\n", "hello world string should be print on stdout");
mu_assert_streq(spo->err, "This is on err\n", "stderr should be found");
mu_assert_eq(spo->ret, 0, "return value is 0");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool test_args(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-helloworld");
const char *args[] = { "rizin" };
RzSubprocess *sp = rz_subprocess_start(exe_path, args, 1, NULL, NULL, 0);
mu_assert_notnull(sp, "the subprocess should be created");
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_streq(spo->out, "Hello rizin\n", "rizin arg should be passed and printed");
mu_assert_eq(spo->ret, 0, "return value is 0");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool test_env(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-helloworld");
const char *envvars[] = { "YOUVAR" };
const char *envvals[] = { "Rizin Project" };
RzSubprocess *sp = rz_subprocess_start(exe_path, NULL, 0, envvars, envvals, 1);
mu_assert_notnull(sp, "the subprocess should be created");
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_streq(spo->out, "Hello Rizin Project\n", "YOUVAR env var should be passed and printed on stdout");
mu_assert_eq(spo->ret, 0, "return value is 0");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool test_stdin(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-stdin");
RzSubprocess *sp = rz_subprocess_start(exe_path, NULL, 0, NULL, NULL, 0);
mu_assert_notnull(sp, "the subprocess should be created");
const char *input = "3\n10\n";
rz_subprocess_stdin_write(sp, (const ut8 *)input, strlen(input));
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_streq(spo->out, "13\n", "the sum should be printed on stdout");
mu_assert_eq(spo->ret, 13, "return value is the sum, 13");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool test_multi(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-stdin");
RzSubprocess *sp1 = rz_subprocess_start(exe_path, NULL, 0, NULL, NULL, 0);
RzSubprocess *sp2 = rz_subprocess_start(exe_path, NULL, 0, NULL, NULL, 0);
RzSubprocess *sp3 = rz_subprocess_start(exe_path, NULL, 0, NULL, NULL, 0);
const char *input1 = "3\n10\n";
const char *input2 = "1\n2\n";
const char *input3 = "5\n7\n";
rz_subprocess_stdin_write(sp1, (const ut8 *)input1, strlen(input1));
rz_subprocess_wait(sp1, UT_TIMEOUT);
rz_subprocess_stdin_write(sp2, (const ut8 *)input2, strlen(input2));
rz_subprocess_stdin_write(sp3, (const ut8 *)input3, strlen(input3));
rz_subprocess_wait(sp2, UT_TIMEOUT);
rz_subprocess_wait(sp3, UT_TIMEOUT);
RzSubprocessOutput *spo1 = rz_subprocess_drain(sp1);
mu_assert_eq(spo1->ret, 13, "return value is the sum");
RzSubprocessOutput *spo2 = rz_subprocess_drain(sp2);
mu_assert_eq(spo2->ret, 3, "return value is the sum");
RzSubprocessOutput *spo3 = rz_subprocess_drain(sp3);
mu_assert_eq(spo3->ret, 12, "return value is the sum");
rz_subprocess_output_free(spo1);
rz_subprocess_output_free(spo2);
rz_subprocess_output_free(spo3);
rz_subprocess_free(sp1);
rz_subprocess_free(sp2);
rz_subprocess_free(sp3);
rz_subprocess_fini();
mu_end;
}
bool test_specialchar_args(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-multiargs");
const char *args[] = { "ri$in", " awesome project ", "'single $quoted'", "\"double $quoted\"", "{", "(", "]", "[" };
RzSubprocess *sp = rz_subprocess_start(exe_path, args, RZ_ARRAY_SIZE(args), NULL, NULL, 0);
mu_assert_notnull(sp, "the subprocess should be created");
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_streq(spo->out, "Hello ri$in awesome project 'single $quoted' \"double $quoted\" { ( ] [\n", "all args should be correctly printed");
mu_assert_eq(spo->ret, 0, "return value is 0");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool test_nopipes(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-helloworld");
RzSubprocessOpt opt = { 0 };
opt.file = exe_path;
opt.stdin_pipe = RZ_SUBPROCESS_PIPE_NONE;
opt.stdout_pipe = RZ_SUBPROCESS_PIPE_NONE;
opt.stderr_pipe = RZ_SUBPROCESS_PIPE_NONE;
RzSubprocess *sp = rz_subprocess_start_opt(&opt);
mu_assert_notnull(sp, "the subprocess should be created");
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_streq(spo->out, "", "hello world string should be not intercepted on stdout");
mu_assert_streq(spo->err, "", "stderr should not be intercepted");
mu_assert_eq(spo->ret, 0, "return value is 0");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool test_stdoutonly(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-helloworld");
RzSubprocessOpt opt = { 0 };
opt.file = exe_path;
opt.stdin_pipe = RZ_SUBPROCESS_PIPE_NONE;
opt.stdout_pipe = RZ_SUBPROCESS_PIPE_CREATE;
opt.stderr_pipe = RZ_SUBPROCESS_PIPE_NONE;
RzSubprocess *sp = rz_subprocess_start_opt(&opt);
mu_assert_notnull(sp, "the subprocess should be created");
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_streq(spo->out, "Hello World\n", "hello world string should be on stdout");
mu_assert_streq(spo->err, "", "stderr should not be intercepted");
mu_assert_eq(spo->ret, 0, "return value is 0");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool test_stderronly(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-helloworld");
RzSubprocessOpt opt = { 0 };
opt.file = exe_path;
opt.stdin_pipe = RZ_SUBPROCESS_PIPE_NONE;
opt.stdout_pipe = RZ_SUBPROCESS_PIPE_NONE;
opt.stderr_pipe = RZ_SUBPROCESS_PIPE_CREATE;
RzSubprocess *sp = rz_subprocess_start_opt(&opt);
mu_assert_notnull(sp, "the subprocess should be created");
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_streq(spo->out, "", "stdout should not be intercepted");
mu_assert_streq(spo->err, "This is on err\n", "stderr should not be intercepted");
mu_assert_eq(spo->ret, 0, "return value is 0");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool test_stdoutstderr(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-helloworld");
RzSubprocessOpt opt = { 0 };
opt.file = exe_path;
opt.stdin_pipe = RZ_SUBPROCESS_PIPE_NONE;
opt.stdout_pipe = RZ_SUBPROCESS_PIPE_CREATE;
opt.stderr_pipe = RZ_SUBPROCESS_PIPE_STDOUT;
RzSubprocess *sp = rz_subprocess_start_opt(&opt);
mu_assert_notnull(sp, "the subprocess should be created");
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_strcontains(spo->out, "Hello World\n", "stdout should be captured in out");
mu_assert_strcontains(spo->out, "This is on err\n", "stderr should be captured in out");
mu_assert_streq(spo->err, "", "stderr should not be intercepted");
mu_assert_eq(spo->ret, 0, "return value is 0");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool test_interactive(void) {
rz_subprocess_init();
const char *exe_path = get_auxiliary_path("subprocess-interactive");
RzSubprocessOpt opt = { 0 };
opt.file = exe_path;
opt.stdin_pipe = RZ_SUBPROCESS_PIPE_CREATE;
opt.stdout_pipe = RZ_SUBPROCESS_PIPE_CREATE;
opt.stderr_pipe = RZ_SUBPROCESS_PIPE_NONE;
RzSubprocess *sp = rz_subprocess_start_opt(&opt);
mu_assert_notnull(sp, "the subprocess should be created");
rz_subprocess_stdin_write(sp, (const ut8 *)"3\n", strlen("3\n"));
rz_subprocess_stdin_write(sp, (const ut8 *)"5\n", strlen("5\n"));
RzStrBuf *sb = rz_subprocess_stdout_readline(sp, UT_TIMEOUT);
int c = atoi(rz_strbuf_get(sb));
char buf[100];
snprintf(buf, sizeof(buf), "%d\n", 3 + 5 + c);
rz_subprocess_stdin_write(sp, (const ut8 *)buf, strlen(buf));
rz_subprocess_wait(sp, UT_TIMEOUT);
RzSubprocessOutput *spo = rz_subprocess_drain(sp);
mu_assert_streq(spo->out, "Right\n", "A Good message should be returned");
mu_assert_eq(spo->ret, 0, "subprocess exited in the right way");
rz_subprocess_output_free(spo);
rz_subprocess_free(sp);
rz_subprocess_fini();
mu_end;
}
bool all_tests() {
mu_run_test(test_noargs_noinput_outerr);
mu_run_test(test_args);
mu_run_test(test_env);
mu_run_test(test_stdin);
mu_run_test(test_multi);
mu_run_test(test_specialchar_args);
mu_run_test(test_nopipes);
mu_run_test(test_stdoutonly);
mu_run_test(test_stderronly);
mu_run_test(test_stdoutstderr);
mu_run_test(test_interactive);
return tests_passed != tests_run;
}
mu_main(all_tests)
Reference in a new issue View git blame Copy permalink