elf/tst-audit23.c

   1 /* Check for expected la_objopen and la_objeclose for all objects.
   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 <array_length.h>
  20 #include <errno.h>
  21 #include <getopt.h>
  22 #include <link.h>
  23 #include <limits.h>
  24 #include <inttypes.h>
  25 #include <gnu/lib-names.h>
  26 #include <string.h>
  27 #include <stdlib.h>
  28 #include <support/capture_subprocess.h>
  29 #include <support/check.h>
  30 #include <support/xstdio.h>
  31 #include <support/xdlfcn.h>
  32 #include <support/support.h>
  33
  34 static int restart;
  35 #define CMDLINE_OPTIONS \
  36   { "restart", no_argument, &restart, 1 },
  37
  38 static int
  39 handle_restart (void)
  40 {
  41   xdlopen ("tst-audit23mod.so", RTLD_NOW);
  42   xdlmopen (LM_ID_NEWLM, LIBC_SO, RTLD_NOW);
  43
  44   return 0;
  45 }
  46
  47 static inline bool
  48 startswith (const char *str, const char *pre)
  49 {
  50   size_t lenpre = strlen (pre);
  51   size_t lenstr = strlen (str);
  52   return lenstr >= lenpre && memcmp (pre, str, lenpre) == 0;
  53 }
  54
  55 static inline bool
  56 is_vdso (const char *str)
  57 {
  58   return startswith (str, "linux-gate")
  59          || startswith (str, "linux-vdso");
  60 }
  61
  62 static int
  63 do_test (int argc, char *argv[])
  64 {
  65   /* We must have either:
  66      - One or four parameters left if called initially:
  67        + path to ld.so         optional
  68        + "--library-path"      optional
  69        + the library path      optional
  70        + the application name  */
  71   if (restart)
  72     return handle_restart ();
  73
  74   char *spargv[9];
  75   TEST_VERIFY_EXIT (((argc - 1) + 3) < array_length (spargv));
  76   int i = 0;
  77   for (; i < argc - 1; i++)
  78     spargv[i] = argv[i + 1];
  79   spargv[i++] = (char *) "--direct";
  80   spargv[i++] = (char *) "--restart";
  81   spargv[i] = NULL;
  82
  83   setenv ("LD_AUDIT", "tst-auditmod23.so", 0);
  84   struct support_capture_subprocess result
  85     = support_capture_subprogram (spargv[0], spargv);
  86   support_capture_subprocess_check (&result, "tst-audit22", 0, sc_allow_stderr);
  87
  88   /* The expected la_objopen/la_objclose:
  89      1. executable
  90      2. loader
  91      3. libc.so
  92      4. tst-audit23mod.so
  93      5. libc.so (LM_ID_NEWLM).
  94      6. vdso (optional and ignored).  */
  95   enum { max_objs = 6 };
  96   struct la_obj_t
  97   {
  98     char *lname;
  99     uintptr_t laddr;
 100     Lmid_t lmid;
 101     bool closed;
 102   } objs[max_objs] = { [0 ... max_objs-1] = { .closed = false } };
 103   size_t nobjs = 0;
 104
 105   /* The expected namespaces are one for the audit module, one for the
 106      application, and another for the dlmopen on handle_restart.  */
 107   enum { max_ns = 3 };
 108   uintptr_t acts[max_ns] = { 0 };
 109   size_t nacts = 0;
 110   int last_act = -1;
 111   uintptr_t last_act_cookie = -1;
 112   bool seen_first_objclose = false;
 113
 114   FILE *out = fmemopen (result.err.buffer, result.err.length, "r");
 115   TEST_VERIFY (out != NULL);
 116   char *buffer = NULL;
 117   size_t buffer_length = 0;
 118   while (xgetline (&buffer, &buffer_length, out))
 119     {
 120       if (startswith (buffer, "la_activity: "))
 121         {
 122           uintptr_t cookie;
 123           int this_act;
 124           int r = sscanf (buffer, "la_activity: %d %"SCNxPTR"", &this_act,
 125                           &cookie);
 126           TEST_COMPARE (r, 2);
 127
 128           /* The cookie identifies the object at the head of the link map,
 129              so we only add a new namespace if it changes from the previous
 130              one.  This works since dlmopen is the last in the test body.  */
 131           if (cookie != last_act_cookie && last_act_cookie != -1)
 132             TEST_COMPARE (last_act, LA_ACT_CONSISTENT);
 133
 134           if (this_act == LA_ACT_ADD && acts[nacts] != cookie)
 135             {
 136               acts[nacts++] = cookie;
 137               last_act_cookie = cookie;
 138             }
 139           /* The LA_ACT_DELETE is called in the reverse order of LA_ACT_ADD
 140              at program termination (if the tests adds a dlclose or a library
 141              with extra dependencies this will need to be adapted).  */
 142           else if (this_act == LA_ACT_DELETE)
 143             {
 144               last_act_cookie = acts[--nacts];
 145               TEST_COMPARE (acts[nacts], cookie);
 146               acts[nacts] = 0;
 147             }
 148           else if (this_act == LA_ACT_CONSISTENT)
 149             {
 150               TEST_COMPARE (cookie, last_act_cookie);
 151
 152               /* LA_ACT_DELETE must always be followed by an la_objclose.  */
 153               if (last_act == LA_ACT_DELETE)
 154                 TEST_COMPARE (seen_first_objclose, true);
 155               else
 156                 TEST_COMPARE (last_act, LA_ACT_ADD);
 157             }
 158
 159           last_act = this_act;
 160           seen_first_objclose = false;
 161         }
 162       else if (startswith (buffer, "la_objopen: "))
 163         {
 164           char *lname;
 165           uintptr_t laddr;
 166           Lmid_t lmid;
 167           uintptr_t cookie;
 168           int r = sscanf (buffer, "la_objopen: %"SCNxPTR"  %ms %"SCNxPTR" %ld",
 169                           &cookie, &lname, &laddr, &lmid);
 170           TEST_COMPARE (r, 4);
 171
 172           /* la_objclose is not triggered by vDSO because glibc does not
 173              unload it.  */
 174           if (is_vdso (lname))
 175             continue;
 176           if (nobjs == max_objs)
 177             FAIL_EXIT1 ("non expected la_objopen: %s %"PRIxPTR" %ld",
 178                         lname, laddr, lmid);
 179           objs[nobjs].lname = lname;
 180           objs[nobjs].laddr = laddr;
 181           objs[nobjs].lmid = lmid;
 182           objs[nobjs].closed = false;
 183           nobjs++;
 184
 185           /* This indirectly checks that la_objopen always comes before
 186              la_objclose btween la_activity calls.  */
 187           seen_first_objclose = false;
 188         }
 189       else if (startswith (buffer, "la_objclose: "))
 190         {
 191           char *lname;
 192           uintptr_t laddr;
 193           Lmid_t lmid;
 194           uintptr_t cookie;
 195           int r = sscanf (buffer, "la_objclose: %"SCNxPTR" %ms %"SCNxPTR" %ld",
 196                           &cookie, &lname, &laddr, &lmid);
 197           TEST_COMPARE (r, 4);
 198
 199           for (size_t i = 0; i < nobjs; i++)
 200             {
 201               if (strcmp (lname, objs[i].lname) == 0 && lmid == objs[i].lmid)
 202                 {
 203                   TEST_COMPARE (objs[i].closed, false);
 204                   objs[i].closed = true;
 205                   break;
 206                 }
 207             }
 208
 209           /* la_objclose should be called after la_activity(LA_ACT_DELETE) for
 210              the closed object's namespace.  */
 211           TEST_COMPARE (last_act, LA_ACT_DELETE);
 212           if (!seen_first_objclose)
 213             {
 214               TEST_COMPARE (last_act_cookie, cookie);
 215               seen_first_objclose = true;
 216             }
 217         }
 218     }
 219
 220   for (size_t i = 0; i < nobjs; i++)
 221     {
 222       TEST_COMPARE (objs[i].closed, true);
 223       free (objs[i].lname);
 224     }
 225
 226   /* la_activity(LA_ACT_CONSISTENT) should be the last callback received.
 227      Since only one link map may be not-CONSISTENT at a time, this also
 228      ensures la_activity(LA_ACT_CONSISTENT) is the last callback received
 229      for every namespace.  */
 230   TEST_COMPARE (last_act, LA_ACT_CONSISTENT);
 231
 232   free (buffer);
 233   xfclose (out);
 234
 235   return 0;
 236 }
 237
 238 #define TEST_FUNCTION_ARGV do_test
 239 #include <support/test-driver.c>