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  }