elf/tst-auditmod24b.c

   1 /* Audit modules for tst-audit24b.
   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 TEST_NAME "tst-audit24b"
  26 #define TEST_FUNC "tst_audit24b"
  27
  28 #define AUDIT24_COOKIE     0x1
  29 #define AUDIT24MOD1_COOKIE 0x2
  30 #define AUDIT24MOD2_COOKIE 0x3
  31
  32 unsigned int
  33 la_version (unsigned int version)
  34 {
  35   return LAV_CURRENT;
  36 }
  37
  38 unsigned int
  39 la_objopen (struct link_map *map, Lmid_t lmid, uintptr_t *cookie)
  40 {
  41   const char *p = strrchr (map->l_name, '/');
  42   const char *l_name = p == NULL ? TEST_NAME : p + 1;
  43
  44   uintptr_t ck = -1;
  45   if (strcmp (l_name, TEST_NAME "mod1.so") == 0)
  46     ck = AUDIT24MOD1_COOKIE;
  47   else if (strcmp (l_name, TEST_NAME "mod2.so") == 0)
  48     ck = AUDIT24MOD2_COOKIE;
  49   else if (strcmp (l_name, TEST_NAME) == 0)
  50     ck = AUDIT24_COOKIE;
  51
  52   *cookie = ck;
  53   return ck == -1 ? 0 : LA_FLG_BINDFROM | LA_FLG_BINDTO;
  54 }
  55
  56 static int
  57 tst_func1 (void)
  58 {
  59   return 1;
  60 }
  61
  62 static int
  63 tst_func2 (void)
  64 {
  65   return 2;
  66 }
  67
  68 #if __ELF_NATIVE_CLASS == 64
  69 uintptr_t
  70 la_symbind64 (Elf64_Sym *sym, unsigned int ndx,
  71               uintptr_t *refcook, uintptr_t *defcook,
  72               unsigned int *flags, const char *symname)
  73 #else
  74 uintptr_t
  75 la_symbind32 (Elf32_Sym *sym, unsigned int ndx,
  76               uintptr_t *refcook, uintptr_t *defcook,
  77               unsigned int *flags, const char *symname)
  78 #endif
  79 {
  80   if (*refcook == AUDIT24_COOKIE)
  81     {
  82       if (*defcook == AUDIT24MOD1_COOKIE)
  83           {
  84             if (strcmp (symname, TEST_FUNC "mod1_func1") == 0)
  85               return (uintptr_t) tst_func1;
  86             else if (strcmp (symname, TEST_FUNC "mod1_func2") == 0)
  87               return sym->st_value;
  88             abort ();
  89           }
  90       /* malloc functions.  */
  91       return sym->st_value;
  92     }
  93   else if (*refcook == AUDIT24MOD1_COOKIE)
  94     {
  95       if (*defcook == AUDIT24MOD2_COOKIE
  96           && (strcmp (symname, TEST_FUNC "mod2_func1") == 0))
  97         {
  98           test_symbind_flags (*flags);
  99           return (uintptr_t) tst_func2;
 100         }
 101     }
 102
 103   abort ();
 104 }