Page MenuHomeFreeBSD
Files F150382027

D41320.1792259616.diff
No OneTemporary
Actions

Size
4 KB
Referenced Files
None
Subscribers
None

D41320.1792259616.diff
View Options

diff --git a/tests/sys/vm/Makefile b/tests/sys/vm/Makefile
--- a/tests/sys/vm/Makefile
+++ b/tests/sys/vm/Makefile
@@ -17,4 +17,15 @@
PROGS+= mmap_map_32bit_helper
.endif
+ATF_TESTS_C+= stack_dt_need_test
+ATF_TESTS_C+= stack_dl_load_test
+
+LDFLAGS.stack_dt_need_test+= -Wl,-rpath,${TESTSDIR} -L${.OBJDIR}/soxstack
+LDADD.stack_dt_need_test+= -lsoxstack
+LDFLAGS.stack_dl_load_test+= -Wl,-rpath,${TESTSDIR}
+
+stack_dt_need_test:soxstack
+
+SUBDIR+= soxstack
+
.include <bsd.test.mk>
diff --git a/tests/sys/vm/soxstack/Makefile b/tests/sys/vm/soxstack/Makefile
new file mode 100644
--- /dev/null
+++ b/tests/sys/vm/soxstack/Makefile
@@ -0,0 +1,17 @@
+# $FreeBSD$
+
+SHLIB= soxstack
+SHLIB_NAME= libsoxstack.so
+SHLIB_MAJOR= 1
+
+WITHOUT_STATIC=
+WITHOUT_PROFILE=
+WITHOUT_PIC=
+
+SRCS= soxstack.c
+LDFLAGS+= -Wl,-z,execstack
+LIBADD+= procstat
+
+LIBDIR= ${TESTSBASE}/sys/vm
+
+.include <bsd.lib.mk>
diff --git a/tests/sys/vm/soxstack/soxstack.c b/tests/sys/vm/soxstack/soxstack.c
new file mode 100644
--- /dev/null
+++ b/tests/sys/vm/soxstack/soxstack.c
@@ -0,0 +1,51 @@
+/*-
+ * Copyright (c) 2023 Dmitry Chagin <dchagin@FreeBSD.org>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#include <sys/param.h>
+#include <sys/sysctl.h>
+#include <sys/user.h>
+
+#include <assert.h>
+#include <libprocstat.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+int checkstack(void);
+
+#define _STACK_FLAG_GROWS KVME_FLAG_GROWS_UP | KVME_FLAG_GROWS_DOWN
+int
+checkstack(void)
+{
+ struct kinfo_vmentry *freep, *kve;
+ struct kinfo_proc *p;
+ struct procstat *prstat;
+ uint64_t stack;
+ int i, cnt;
+
+ prstat = procstat_open_sysctl();
+ assert(prstat != NULL);
+ p = procstat_getprocs(prstat, KERN_PROC_PID, getpid(), &cnt);
+ assert(p != NULL);
+ freep = procstat_getvmmap(prstat, p, &cnt);
+ assert(freep != NULL);
+
+ stack = (uint64_t)&i;
+ for (i = 0; i < cnt; i++) {
+ kve = &freep[i];
+ if (stack < kve->kve_start || stack > kve->kve_end)
+ continue;
+ if ((kve->kve_flags & _STACK_FLAG_GROWS) != 0 &&
+ (kve->kve_protection & KVME_PROT_EXEC) != 0)
+ stack = 0;
+ break;
+ }
+
+ free(freep);
+ procstat_freeprocs(prstat, p);
+ procstat_close(prstat);
+ return (stack != 0);
+}
diff --git a/tests/sys/vm/stack_dl_load_test.c b/tests/sys/vm/stack_dl_load_test.c
new file mode 100644
--- /dev/null
+++ b/tests/sys/vm/stack_dl_load_test.c
@@ -0,0 +1,64 @@
+/*-
+ * Copyright (c) 2023 Dmitry Chagin <dchagin@FreeBSD.org>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#include <sys/systm.h>
+#include <vm/vm_param.h>
+
+#include <atf-c.h>
+#include <dlfcn.h>
+
+static int jumpstack0(void) __noinline;
+static int jumpstack1(void) __noinline;
+
+static int (*socheckstack)(void) = NULL;
+
+static int
+checkstack(void)
+{
+ void *fh;
+
+ if (socheckstack == NULL) {
+ fh = dlopen("libsoxstack.so", RTLD_LAZY);
+ ATF_REQUIRE(fh != NULL);
+ socheckstack = dlsym(fh, "checkstack");
+ ATF_REQUIRE(socheckstack != NULL);
+ }
+ return (socheckstack());
+}
+
+static int
+jumpstack0(void)
+{
+ char stack[SGROWSIZ];
+
+ explicit_bzero(stack, sizeof(stack));
+ return (checkstack());
+}
+
+static int
+jumpstack1(void)
+{
+ char stack[SGROWSIZ * 2];
+
+ explicit_bzero(stack, sizeof(stack));
+ return (checkstack());
+}
+
+ATF_TC_WITHOUT_HEAD(dl_load_test);
+ATF_TC_BODY(dl_load_test, tc)
+{
+
+ ATF_REQUIRE(jumpstack0() == 0);
+ ATF_REQUIRE(jumpstack1() == 0);
+}
+
+ATF_TP_ADD_TCS(tp)
+{
+
+ ATF_TP_ADD_TC(tp, dl_load_test);
+
+ return (atf_no_error());
+}
diff --git a/tests/sys/vm/stack_dt_need_test.c b/tests/sys/vm/stack_dt_need_test.c
new file mode 100644
--- /dev/null
+++ b/tests/sys/vm/stack_dt_need_test.c
@@ -0,0 +1,50 @@
+/*-
+ * Copyright (c) 2023 Dmitry Chagin <dchagin@FreeBSD.org>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#include <sys/systm.h>
+#include <vm/vm_param.h>
+
+#include <atf-c.h>
+
+extern int checkstack(void);
+
+static int jumpstack0(void) __noinline;
+static int jumpstack1(void) __noinline;
+
+
+static int
+jumpstack0(void)
+{
+ char stack[SGROWSIZ];
+
+ explicit_bzero(stack, sizeof(stack));
+ return (checkstack());
+}
+
+static int
+jumpstack1(void)
+{
+ char stack[SGROWSIZ * 2];
+
+ explicit_bzero(stack, sizeof(stack));
+ return (checkstack());
+}
+
+ATF_TC_WITHOUT_HEAD(dt_need_test);
+ATF_TC_BODY(dt_need_test, tc)
+{
+
+ ATF_REQUIRE(jumpstack0() == 0);
+ ATF_REQUIRE(jumpstack1() == 0);
+}
+
+ATF_TP_ADD_TCS(tp)
+{
+
+ ATF_TP_ADD_TC(tp, dt_need_test);
+
+ return (atf_no_error());
+}

File Metadata

Mime Type
text/plain
Expires
Sat, Oct 17, 5:53 PM (5 h, 31 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
30473844
Default Alt Text
D41320.1792259616.diff (4 KB)

Event Timeline