1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
|
/* Test getenv from a signal handler interrupting environment updates.
Copyright (C) 2024-2025 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 <array_length.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <support/check.h>
#include <support/support.h>
#include <support/xthread.h>
#include <support/xsignal.h>
/* Set to false by the main thread after doing all the setenv
calls. */
static bool running = true;
/* Used to synchronize the start of signal sending. */
static pthread_barrier_t barrier;
/* Identity of the main thread. */
static pthread_t main_thread;
/* Send SIGUSR1 signals to main_thread. */
static void *
signal_thread (void *ignored)
{
xpthread_barrier_wait (&barrier);
while (__atomic_load_n (&running, __ATOMIC_RELAXED))
xpthread_kill (main_thread, SIGUSR1);
return NULL;
}
/* Call getenv from a signal handler. */
static void
signal_handler (int signo)
{
TEST_COMPARE_STRING (getenv ("unset_variable"), NULL);
char *value = getenv ("set_variable");
TEST_VERIFY (strncmp (value, "value", strlen ("value")) == 0);
}
static int
do_test (void)
{
/* Added to the environment using putenv. */
char *variables[30];
for (int i = 0; i < array_length (variables); ++i)
variables[i] = xasprintf ("v%d=%d", i, i);
xsignal (SIGUSR1, signal_handler);
TEST_COMPARE (setenv ("set_variable", "value", 1), 0);
xraise (SIGUSR1);
main_thread = pthread_self ();
xpthread_barrier_init (&barrier, NULL, 2);
pthread_t thr = xpthread_create (NULL, signal_thread, NULL);
xpthread_barrier_wait (&barrier);
for (int i = 0; i < array_length (variables); ++i)
{
char buf[30];
TEST_COMPARE (setenv ("temporary_variable", "1", 1), 0);
snprintf (buf, sizeof (buf), "V%d", i);
TEST_COMPARE (setenv (buf, buf + 1, 1), 0);
TEST_COMPARE (putenv (variables[i]), 0);
snprintf (buf, sizeof (buf), "value%d", i);
TEST_COMPARE (setenv ("set_variable", buf, 1), 0);
TEST_COMPARE (unsetenv ("temporary_variable"), 0);
}
__atomic_store_n (&running, false, __ATOMIC_RELAXED);
xpthread_join (thr);
xpthread_barrier_destroy (&barrier);
for (int i = 0; i < array_length (variables); ++i)
free (variables[i]);
return 0;
}
#include <support/test-driver.c>
|