xref: /freebsd/libexec/rtld-elf/tests/dlopen_test.c (revision c5796f1572c82b88e8b6a2810c92f30e5ac3e118)
1*c5796f15SKyle Evans /*-
2*c5796f15SKyle Evans  *
3*c5796f15SKyle Evans  * Copyright (C) 2024 Kyle Evans <kevans@FreeBSD.org>
4*c5796f15SKyle Evans  *
5*c5796f15SKyle Evans  * SPDX-License-Identifier: BSD-2-Clause
6*c5796f15SKyle Evans  *
7*c5796f15SKyle Evans  */
8*c5796f15SKyle Evans 
9*c5796f15SKyle Evans #include <dlfcn.h>
10*c5796f15SKyle Evans 
11*c5796f15SKyle Evans #include <atf-c.h>
12*c5796f15SKyle Evans 
13*c5796f15SKyle Evans ATF_TC_WITHOUT_HEAD(dlopen_basic);
ATF_TC_BODY(dlopen_basic,tc)14*c5796f15SKyle Evans ATF_TC_BODY(dlopen_basic, tc)
15*c5796f15SKyle Evans {
16*c5796f15SKyle Evans 	void *hdl, *sym;
17*c5796f15SKyle Evans 
18*c5796f15SKyle Evans 	hdl = dlopen("libthr.so", RTLD_NOW);
19*c5796f15SKyle Evans 	ATF_REQUIRE(hdl != NULL);
20*c5796f15SKyle Evans 
21*c5796f15SKyle Evans 	sym = dlsym(hdl, "pthread_create");
22*c5796f15SKyle Evans 	ATF_REQUIRE(sym != NULL);
23*c5796f15SKyle Evans 
24*c5796f15SKyle Evans 	dlclose(hdl);
25*c5796f15SKyle Evans 
26*c5796f15SKyle Evans 	sym = dlsym(hdl, "pthread_create");
27*c5796f15SKyle Evans 	ATF_REQUIRE(sym == NULL);
28*c5796f15SKyle Evans }
29*c5796f15SKyle Evans 
30*c5796f15SKyle Evans ATF_TC_WITHOUT_HEAD(dlopen_recursing);
ATF_TC_BODY(dlopen_recursing,tc)31*c5796f15SKyle Evans ATF_TC_BODY(dlopen_recursing, tc)
32*c5796f15SKyle Evans {
33*c5796f15SKyle Evans 	void *hdl;
34*c5796f15SKyle Evans 
35*c5796f15SKyle Evans 	/*
36*c5796f15SKyle Evans 	 * If this doesn't crash, we're OK; a regression at one point caused
37*c5796f15SKyle Evans 	 * some infinite recursion here.
38*c5796f15SKyle Evans 	 */
39*c5796f15SKyle Evans 	hdl = dlopen("libthr.so", RTLD_NOW | RTLD_GLOBAL);
40*c5796f15SKyle Evans 	ATF_REQUIRE(hdl != NULL);
41*c5796f15SKyle Evans 
42*c5796f15SKyle Evans 	dlclose(hdl);
43*c5796f15SKyle Evans }
44*c5796f15SKyle Evans 
ATF_TP_ADD_TCS(tp)45*c5796f15SKyle Evans ATF_TP_ADD_TCS(tp)
46*c5796f15SKyle Evans {
47*c5796f15SKyle Evans 
48*c5796f15SKyle Evans 	ATF_TP_ADD_TC(tp, dlopen_basic);
49*c5796f15SKyle Evans 	ATF_TP_ADD_TC(tp, dlopen_recursing);
50*c5796f15SKyle Evans 
51*c5796f15SKyle Evans 	return atf_no_error();
52*c5796f15SKyle Evans }
53