[apple/xnu.git] / tests / text_corruption.c

#include <unistd.h>
#include <stdio.h>

#include <darwintest.h>
#include <darwintest_utils.h>

T_GLOBAL_META(T_META_RUN_CONCURRENTLY(false));

/*
 * No system(3c) on watchOS, so provide our own.
 * returns -1 if fails to run
 * returns 0 if process exits normally.
 * returns +n if process exits due to signal N
 */
static int
my_system(const char *command)
{
	pid_t pid;
	int status = 0;
	int signal = 0;
	int err;
	const char *argv[] = {
		"/bin/sh",
		"-c",
		command,
		NULL
	};

	if (dt_launch_tool(&pid, (char **)(void *)argv, FALSE, NULL, NULL)) {
		return -1;
	}

	err = dt_waitpid(pid, &status, &signal, 30);
	if (err) {
		return 0;
	}

	return signal;
}


/*
 * The tests are run in the following order:
 *
 * - call foo
 * - corrupt foo, then call foo
 * - call foo
 *
 * - call atan
 * - corrupt atan, then call atan
 * - call atan
 *
 * The first and last of each should exit normally. The middle one should exit with SIGILL.
 *
 * atan() was picked as a shared region function that isn't likely used by any normal daemons.
 */
T_DECL(text_corruption_recovery, "test detection/recovery of text corruption",
    T_META_IGNORECRASHES(".*text_corruption_helper.*"),
    T_META_ASROOT(true))
{
	int ret;

	ret = my_system("./text_corruption_helper foo");
	T_QUIET; T_ASSERT_EQ(ret, 0, "First call of foo");

	ret = my_system("./text_corruption_helper Xfoo");
	T_QUIET; T_ASSERT_EQ(ret, SIGILL, "Call of corrupted foo");

	ret = my_system("./text_corruption_helper foo");
	T_QUIET; T_ASSERT_EQ(ret, 0, "Fixed call of foo");

	ret = my_system("./text_corruption_helper atan");
	T_QUIET; T_ASSERT_EQ(ret, 0, "First call of atan");

	ret = my_system("./text_corruption_helper Xatan");
	T_QUIET; T_ASSERT_EQ(ret, SIGILL, "Call of corrupted atan");

	ret = my_system("./text_corruption_helper atan");
	T_QUIET; T_ASSERT_EQ(ret, 0, "Fixed call of atan");
}
Commit	Line	Data
c3c9b80d A	1	#include <unistd.h>
	2	#include <stdio.h>
	3
	4	#include <darwintest.h>
	5	#include <darwintest_utils.h>
	6
	7	T_GLOBAL_META(T_META_RUN_CONCURRENTLY(false));
	8
	9	/*
	10	* No system(3c) on watchOS, so provide our own.
	11	* returns -1 if fails to run
	12	* returns 0 if process exits normally.
	13	* returns +n if process exits due to signal N
	14	*/
	15	static int
	16	my_system(const char *command)
	17	{
	18	pid_t pid;
	19	int status = 0;
	20	int signal = 0;
	21	int err;
	22	const char *argv[] = {
	23	"/bin/sh",
	24	"-c",
	25	command,
	26	NULL
	27	};
	28
	29	if (dt_launch_tool(&pid, (char *)(void )argv, FALSE, NULL, NULL)) {
	30	return -1;
	31	}
	32
	33	err = dt_waitpid(pid, &status, &signal, 30);
	34	if (err) {
	35	return 0;
	36	}
	37
	38	return signal;
	39	}
	40
	41
	42	/*
	43	* The tests are run in the following order:
	44	*
	45	* - call foo
	46	* - corrupt foo, then call foo
	47	* - call foo
	48	*
	49	* - call atan
	50	* - corrupt atan, then call atan
	51	* - call atan
	52	*
	53	* The first and last of each should exit normally. The middle one should exit with SIGILL.
	54	*
	55	* atan() was picked as a shared region function that isn't likely used by any normal daemons.
	56	*/
	57	T_DECL(text_corruption_recovery, "test detection/recovery of text corruption",
	58	T_META_IGNORECRASHES(".text_corruption_helper."),
	59	T_META_ASROOT(true))
	60	{
	61	int ret;
	62
	63	ret = my_system("./text_corruption_helper foo");
	64	T_QUIET; T_ASSERT_EQ(ret, 0, "First call of foo");
65
66	ret = my_system("./text_corruption_helper Xfoo");
67	T_QUIET; T_ASSERT_EQ(ret, SIGILL, "Call of corrupted foo");
68
69	ret = my_system("./text_corruption_helper foo");
70	T_QUIET; T_ASSERT_EQ(ret, 0, "Fixed call of foo");
71
72	ret = my_system("./text_corruption_helper atan");
73	T_QUIET; T_ASSERT_EQ(ret, 0, "First call of atan");
74
75	ret = my_system("./text_corruption_helper Xatan");
76	T_QUIET; T_ASSERT_EQ(ret, SIGILL, "Call of corrupted atan");
77
78	ret = my_system("./text_corruption_helper atan");
79	T_QUIET; T_ASSERT_EQ(ret, 0, "Fixed call of atan");
80	}