1 | /* Audit modules for tst-audit24a. |
2 | Copyright (C) 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 <link.h> |
20 | #include <inttypes.h> |
21 | #include <stdlib.h> |
22 | #include <string.h> |
23 | #include <tst-auditmod24.h> |
24 | |
25 | #define AUDIT24_COOKIE 0x1 |
26 | #define AUDIT24MOD1_COOKIE 0x2 |
27 | #define AUDIT24MOD2_COOKIE 0x3 |
28 | |
29 | #ifndef TEST_NAME |
30 | # define TEST_NAME "tst-audit24a" |
31 | #endif |
32 | #ifndef TEST_MOD |
33 | # define TEST_MOD TEST_NAME |
34 | #endif |
35 | #ifndef TEST_FUNC |
36 | # define TEST_FUNC "tst_audit24a" |
37 | #endif |
38 | |
39 | unsigned int |
40 | la_version (unsigned int version) |
41 | { |
42 | return LAV_CURRENT; |
43 | } |
44 | |
45 | unsigned int |
46 | la_objopen (struct link_map *map, Lmid_t lmid, uintptr_t *cookie) |
47 | { |
48 | const char *p = strrchr (s: map->l_name, c: '/'); |
49 | const char *l_name = p == NULL ? TEST_NAME : p + 1; |
50 | |
51 | uintptr_t ck = -1; |
52 | if (strcmp (s1: l_name, TEST_MOD "mod1.so" ) == 0) |
53 | ck = AUDIT24MOD1_COOKIE; |
54 | else if (strcmp (s1: l_name, TEST_MOD "mod2.so" ) == 0) |
55 | ck = AUDIT24MOD2_COOKIE; |
56 | else if (strcmp (s1: l_name, TEST_NAME) == 0) |
57 | ck = AUDIT24_COOKIE; |
58 | |
59 | *cookie = ck; |
60 | return ck == -1 ? 0 : LA_FLG_BINDFROM | LA_FLG_BINDTO; |
61 | } |
62 | |
63 | static int |
64 | tst_func1 (void) |
65 | { |
66 | return 1; |
67 | } |
68 | |
69 | static int |
70 | tst_func2 (void) |
71 | { |
72 | return 10; |
73 | } |
74 | |
75 | #if __ELF_NATIVE_CLASS == 64 |
76 | uintptr_t |
77 | la_symbind64 (Elf64_Sym *sym, unsigned int ndx, |
78 | uintptr_t *refcook, uintptr_t *defcook, |
79 | unsigned int *flags, const char *symname) |
80 | #else |
81 | uintptr_t |
82 | la_symbind32 (Elf32_Sym *sym, unsigned int ndx, |
83 | uintptr_t *refcook, uintptr_t *defcook, |
84 | unsigned int *flags, const char *symname) |
85 | #endif |
86 | { |
87 | if (*refcook == AUDIT24_COOKIE) |
88 | { |
89 | if (*defcook == AUDIT24MOD1_COOKIE) |
90 | { |
91 | /* Check if bind-now symbols are advertised to not call the PLT |
92 | hooks. */ |
93 | test_symbind_flags (flags: *flags); |
94 | |
95 | if (strcmp (s1: symname, TEST_FUNC "mod1_func1" ) == 0) |
96 | return (uintptr_t) tst_func1; |
97 | else if (strcmp (s1: symname, TEST_FUNC "mod1_func2" ) == 0) |
98 | return sym->st_value; |
99 | abort (); |
100 | } |
101 | if (*defcook == AUDIT24MOD2_COOKIE |
102 | && (strcmp (s1: symname, TEST_FUNC "mod2_func1" ) == 0)) |
103 | { |
104 | test_symbind_flags (flags: *flags); |
105 | |
106 | return (uintptr_t) tst_func2; |
107 | } |
108 | |
109 | /* malloc functions. */ |
110 | return sym->st_value; |
111 | } |
112 | |
113 | if (symname[0] != '\0') |
114 | abort (); |
115 | return sym->st_value; |
116 | } |
117 | |