int main(int argc, char **argv) {
srand(time(NULL));
if (argc>1 && strcmp(argv[1], "intercept") == 0)
return do_intercept(atoi(argv[2]), atoi(argv[3]));
if (argc>1 && strcmp(argv[1], "release") == 0)
return do_release(atoi(argv[2]), atoi(argv[3]));
if (argc>1 && strcmp(argv[1], "nonroot") == 0)
return do_nonroot(atoi(argv[2]));
test("insmod interceptor.ko %s", "", system("insmod interceptor.ko") == 0);
test("bad MY_CUSTOM_SYSCALL args%s", "", vsyscall_arg(MY_CUSTOM_SYSCALL, 3, 100, 0, 0) == -EINVAL);
do_intercept(MY_CUSTOM_SYSCALL, -EINVAL);
do_release(MY_CUSTOM_SYSCALL, -EINVAL);
do_intercept(-1, -EINVAL);
do_release(-1, -EINVAL);
do_intercept(__NR_exit, 0);
do_release(__NR_exit, 0);
test_syscall(SYS_open);
/* The above line of code tests SYS_open.
* Feel free to add more tests here for other system calls,
* once you get everything to work; check Linux documentation
* for other syscall number definitions.
*/
test("rmmod interceptor.ko %s", "", system("rmmod interceptor") == 0);
return 0;
}
int main(int argc, char **argv) {
srand(time(NULL));
if (argc > 1 && strcmp(argv[1], "intercept") == 0)
return do_intercept(atoi(argv[2]), atoi(argv[3]));
if (argc > 1 && strcmp(argv[1], "release") == 0)
return do_release(atoi(argv[2]), atoi(argv[3]));
if (argc > 1 && strcmp(argv[1], "start") == 0)
return do_start(atoi(argv[2]), atoi(argv[3]), atoi(argv[4]));
if (argc > 1 && strcmp(argv[1], "stop") == 0)
return do_stop(atoi(argv[2]), atoi(argv[3]), atoi(argv[4]));
if (argc > 1 && strcmp(argv[1], "monitor") == 0)
return test_monitor(atoi(argv[2]), TRUE);
if (argc > 1 && strcmp(argv[1], "nonroot") == 0)
return do_nonroot(atoi(argv[2]));
struct sigaction sa;
sa.sa_flags = SA_SIGINFO;
sigemptyset(&sa.sa_mask);
sa.sa_sigaction = on_quit;
if (sigaction(SIGQUIT, &sa, NULL) == -1)
perror("Cannot register signal handler");
test("insmod interceptor.ko %s", "", system("insmod interceptor.ko") == 0);
test("bad MY_SYSCALL args%s", "", vsyscall_arg(MY_CUSTOM_SYSCALL, 3, 100, 0, 0) == -EINVAL);
do_intercept(MY_CUSTOM_SYSCALL, -EINVAL);
do_release(MY_CUSTOM_SYSCALL, -EINVAL);
do_intercept(-1, -EINVAL);
do_release(-1, -EINVAL);
do_intercept(__NR_exit, 0);
do_release(__NR_exit, 0);
test_syscall(SYS_open);
/* The above line of code tests SYS_open.
Feel free to add more tests here for other system calls,
once you get everything to work; check Linux documentation
for other syscall number definitions. */
do_intercept(SYS_open, 0);
do_start(SYS_open, -1, 0);
test_monitor(SYS_open, TRUE);
test("rmmod interceptor.ko %s", "", system("rmmod interceptor") == 0);
test_monitor(SYS_open, FALSE);
return 0;
}
