elf/tst-auditmod24a.c

   1 /* Audit modules for tst-audit24a.
   2    Copyright (C) 2022-2023 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 (map->l_name, '/');
  49   const char *l_name = p == NULL ? TEST_NAME : p + 1;
  50
  51   uintptr_t ck = -1;
  52   if (strcmp (l_name, TEST_MOD "mod1.so") == 0)
  53     ck = AUDIT24MOD1_COOKIE;
  54   else if (strcmp (l_name, TEST_MOD "mod2.so") == 0)
  55     ck = AUDIT24MOD2_COOKIE;
  56   else if (strcmp (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);
  94
  95           if (strcmp (symname, TEST_FUNC "mod1_func1") == 0)
  96             return (uintptr_t) tst_func1;
  97           else if (strcmp (symname, TEST_FUNC "mod1_func2") == 0)
  98             return sym->st_value;
  99           abort ();
 100         }
 101       if (*defcook == AUDIT24MOD2_COOKIE
 102           && (strcmp (symname, TEST_FUNC "mod2_func1") == 0))
 103         {
 104           test_symbind_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 }