1/* Check DT_AUDIT with dlmopen.
2 Copyright (C) 2021-2022 Free Software Foundation, Inc.
3 This file is part of the GNU C Library.
4
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Lesser General Public
7 License as published by the Free Software Foundation; either
8 version 2.1 of the License, or (at your option) any later version.
9
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public
16 License along with the GNU C Library; if not, see
17 <https://www.gnu.org/licenses/>. */
18
19#include <array_length.h>
20#include <getopt.h>
21#include <string.h>
22#include <stdlib.h>
23#include <unistd.h>
24#include <gnu/lib-names.h>
25#include <support/capture_subprocess.h>
26#include <support/check.h>
27#include <support/xdlfcn.h>
28#include <support/xstdio.h>
29#include <support/support.h>
30
31static int restart;
32#define CMDLINE_OPTIONS \
33 { "restart", no_argument, &restart, 1 },
34
35static int
36handle_restart (void)
37{
38 {
39 void *h = xdlmopen (LM_ID_NEWLM, LIBC_SO, RTLD_NOW);
40
41 pid_t (*s) (void) = xdlsym (handle: h, symbol: "getpid");
42 TEST_COMPARE (s (), getpid ());
43
44 xdlclose (handle: h);
45 }
46
47 {
48 void *h = xdlmopen (LM_ID_NEWLM, filename: "tst-audit18mod.so", RTLD_NOW);
49
50 int (*foo) (void) = xdlsym (handle: h, symbol: "foo");
51 TEST_COMPARE (foo (), 10);
52
53 xdlclose (handle: h);
54 }
55
56 return 0;
57}
58
59static int
60do_test (int argc, char *argv[])
61{
62 /* We must have either:
63 - One our fource parameters left if called initially:
64 + path to ld.so optional
65 + "--library-path" optional
66 + the library path optional
67 + the application name */
68
69 if (restart)
70 return handle_restart ();
71
72 char *spargv[9];
73 int i = 0;
74 for (; i < argc - 1; i++)
75 spargv[i] = argv[i + 1];
76 spargv[i++] = (char *) "--direct";
77 spargv[i++] = (char *) "--restart";
78 spargv[i] = NULL;
79
80 setenv (name: "LD_AUDIT", value: "tst-auditmod18.so", replace: 0);
81 struct support_capture_subprocess result
82 = support_capture_subprogram (file: spargv[0], argv: spargv);
83 support_capture_subprocess_check (&result, context: "tst-audit18", status_or_signal: 0, allowed: sc_allow_stderr);
84
85 struct
86 {
87 const char *name;
88 bool found;
89 } audit_iface[] =
90 {
91 { "la_version", false },
92 { "la_objsearch", false },
93 { "la_activity", false },
94 { "la_objopen", false },
95 { "la_objclose", false },
96 { "la_preinit", false },
97#if __WORDSIZE == 32
98 { "la_symbind32", false },
99#elif __WORDSIZE == 64
100 { "la_symbind64", false },
101#endif
102 };
103
104 /* Some hooks are called more than once but the test only check if any
105 is called at least once. */
106 FILE *out = fmemopen (s: result.err.buffer, len: result.err.length, modes: "r");
107 TEST_VERIFY (out != NULL);
108 char *buffer = NULL;
109 size_t buffer_length = 0;
110 while (xgetline (lineptr: &buffer, n: &buffer_length, stream: out))
111 {
112 for (int i = 0; i < array_length (audit_iface); i++)
113 if (strncmp (s1: buffer, s2: audit_iface[i].name,
114 n: strlen (s: audit_iface[i].name)) == 0)
115 audit_iface[i].found = true;
116 }
117 free (ptr: buffer);
118 xfclose (out);
119
120 for (int i = 0; i < array_length (audit_iface); i++)
121 TEST_COMPARE (audit_iface[i].found, true);
122
123 support_capture_subprocess_free (&result);
124
125 return 0;
126}
127
128#define TEST_FUNCTION_ARGV do_test
129#include <support/test-driver.c>
130

source code of glibc/elf/tst-audit18.c