glibc/elf/tst-dlopen-tlsreinitmod3.c
Florian Weimer 5097cd344f elf: Avoid re-initializing already allocated TLS in dlopen (bug 31717)
The old code used l_init_called as an indicator for whether TLS
initialization was complete.  However, it is possible that
TLS for an object is initialized, written to, and then dlopen
for this object is called again, and l_init_called is not true at
this point.  Previously, this resulted in TLS being initialized
twice, discarding any interim writes (technically introducing a
use-after-free bug even).

This commit introduces an explicit per-object flag, l_tls_in_slotinfo.
It indicates whether _dl_add_to_slotinfo has been called for this
object.  This flag is used to avoid double-initialization of TLS.
In update_tls_slotinfo, the first_static_tls micro-optimization
is removed because preserving the initalization flag for subsequent
use by the second loop for static TLS is a bit complicated, and
another per-object flag does not seem to be worth it.  Furthermore,
the l_init_called flag is dropped from the second loop (for static
TLS initialization) because l_need_tls_init on its own prevents
double-initialization.

The remaining l_init_called usage in resize_scopes and update_scopes
is just an optimization due to the use of scope_has_map, so it is
not changed in this commit.

The isupper check ensures that libc.so.6 is TLS is not reverted.
Such a revert happens if l_need_tls_init is not cleared in
_dl_allocate_tls_init for the main_thread case, now that
l_init_called is not checked anymore in update_tls_slotinfo
in elf/dl-open.c.

Reported-by: Jonathon Anderson <janderson@rice.edu>
Reviewed-by: Carlos O'Donell <carlos@redhat.com>
2024-08-05 18:26:52 +02:00

103 lines
3.1 KiB
C

/* Test that dlopen preserves already accessed TLS (bug 31717), module 3.
Copyright (C) 2024 Free Software Foundation, Inc.
This file is part of the GNU C Library.
The GNU C Library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
The GNU C Library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with the GNU C Library; if not, see
<https://www.gnu.org/licenses/>. */
#include <dlfcn.h>
#include <stdbool.h>
#include <stdio.h>
#include <unistd.h>
/* Used to verify from the main program that the test ran. */
bool tlsreinitmod3_tested;
/* This TLS variable must not revert back to the initial state after
dlopen. */
static __thread int tlsreinitmod3_state = 1;
/* Set from the ELF constructor during dlopen. */
static bool tlsreinitmod3_constructed;
/* Second half of test, behind a compiler barrier. The compiler
barrier is necessary to prevent carrying over TLS address
information from call_tlsreinitmod3 to call_tlsreinitmod3_tail. */
void call_tlsreinitmod3_tail (void *self) __attribute__ ((weak));
/* Called from tst-dlopen-tlsreinitmod2.so. */
void
call_tlsreinitmod3 (void)
{
printf ("info: call_tlsreinitmod3 invoked (state=%d)\n",
tlsreinitmod3_state);
if (tlsreinitmod3_constructed)
{
puts ("error: call_tlsreinitmod3 called after ELF constructor");
fflush (stdout);
/* Cannot rely on test harness due to dynamic linking. */
_exit (1);
}
tlsreinitmod3_state = 2;
/* Self-dlopen. This will run the ELF constructor. */
void *self = dlopen ("tst-dlopen-tlsreinitmod3.so", RTLD_NOW);
if (self == NULL)
{
printf ("error: dlopen: %s\n", dlerror ());
fflush (stdout);
/* Cannot rely on test harness due to dynamic linking. */
_exit (1);
}
call_tlsreinitmod3_tail (self);
}
void
call_tlsreinitmod3_tail (void *self)
{
printf ("info: dlopen returned in tlsreinitmod3 (state=%d)\n",
tlsreinitmod3_state);
if (!tlsreinitmod3_constructed)
{
puts ("error: dlopen did not call tlsreinitmod3 ELF constructor");
fflush (stdout);
/* Cannot rely on test harness due to dynamic linking. */
_exit (1);
}
if (tlsreinitmod3_state != 2)
{
puts ("error: TLS state reverted in tlsreinitmod3");
fflush (stdout);
/* Cannot rely on test harness due to dynamic linking. */
_exit (1);
}
dlclose (self);
/* Signal test completion to the main program. */
tlsreinitmod3_tested = true;
}
static void __attribute__ ((constructor))
tlsreinitmod3_init (void)
{
puts ("info: constructor of tst-dlopen-tlsreinitmod3.so invoked");
tlsreinitmod3_constructed = true;
}