| 1 | /* Check DT_AUDIT with dlmopen. |
| 2 | Copyright (C) 2021-2024 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 | |
| 31 | static int restart; |
| 32 | #define CMDLINE_OPTIONS \ |
| 33 | { "restart", no_argument, &restart, 1 }, |
| 34 | |
| 35 | static int |
| 36 | handle_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 | |
| 59 | static int |
| 60 | do_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 | |