Cygwin: redefine how to recognize forkee state

So far the global variable in_forkee only indicated if the
process is the child process during fork(2) itself.

However, we need an indicator accessible from plain C code
in newlib, allowing to check for a process being a forked
process all the time, after fork(2) succeeded.

Redefine bool in_forkee to int __in_forkee to allow exposing
it to newlib.  Redefine how it indicates fork state (not
forked, forking, forked).

Signed-off-by: Corinna Vinschen <corinna@vinschen.de>
This commit is contained in:
Corinna Vinschen 2024-01-29 13:33:05 +01:00
parent 1c13ca67b5
commit 1f68e88f0d
5 changed files with 27 additions and 16 deletions

View File

@ -64,7 +64,7 @@ do_global_dtors ()
static void
do_global_ctors (void (**in_pfunc)(), int force)
{
if (!force && in_forkee)
if (!force && __in_forkee == FORKING)
return; // inherit constructed stuff from parent pid
/* Run ctors backwards, so skip the first entry and find how many
@ -532,7 +532,7 @@ get_cygwin_startup_info ()
switch (res->type)
{
case _CH_FORK:
in_forkee = true;
__in_forkee = FORKING;
should_be_cb = sizeof (child_info_fork);
fallthrough;
case _CH_SPAWN:
@ -803,7 +803,7 @@ dll_crt0_1 (void *)
_my_tls.incyg++;
/* Inherit "parent" exec'ed process sigmask */
if (spawn_info && !in_forkee)
if (spawn_info && __in_forkee != FORKING)
_my_tls.sigmask = spawn_info->sigmask;
if (dynamically_loaded)
@ -840,7 +840,7 @@ dll_crt0_1 (void *)
/* Initialize pthread mainthread when not forked and it is safe to call new,
otherwise it is reinitalized in fixup_after_fork */
if (!in_forkee)
if (__in_forkee != FORKING)
{
pthread::init_mainthread ();
_pei386_runtime_relocator (user_data);
@ -851,7 +851,7 @@ dll_crt0_1 (void *)
#endif
cygbench ("pre-forkee");
if (in_forkee)
if (__in_forkee == FORKING)
{
/* Make sure to restore the TEB's stack info. If guardsize is -1 the
stack has been provided by the application and must not be deallocated
@ -1018,7 +1018,7 @@ _dll_crt0 ()
under our own control and avoids collision with the OS. */
if (!dynamically_loaded)
{
if (!in_forkee)
if (__in_forkee != FORKING)
{
/* Must be static since it's referenced after the stack and frame
pointer registers have been changed. */
@ -1056,7 +1056,7 @@ void
dll_crt0 (per_process *uptr)
{
/* Set the local copy of the pointer into the user space. */
if (!in_forkee && uptr && uptr != user_data)
if (__in_forkee != FORKING && uptr && uptr != user_data)
{
memcpy (user_data, uptr, per_process_overwrite);
*(user_data->impure_ptr_ptr) = _GLOBAL_REENT;
@ -1246,7 +1246,9 @@ extern "C" void
vapi_fatal (const char *fmt, va_list ap)
{
char buf[4096];
int n = __small_sprintf (buf, "%P: *** fatal error %s- ", in_forkee ? "in forked process " : "");
int n = __small_sprintf (buf, "%P: *** fatal error %s- ",
(__in_forkee == FORKING)
? "in forked process " : "");
__small_vsprintf (buf + n, fmt, ap);
va_end (ap);
strace.prntf (_STRACE_SYSTEM, NULL, "%s", buf);

View File

@ -39,7 +39,7 @@ static bool dll_global_dtors_recorded;
/* We need the in_load_after_fork flag so dll_dllcrt0_1 can decide at fork
time if this is a linked DLL or a dynamically loaded DLL. In either case,
both, cygwin_finished_initializing and in_forkee are true, so they are not
both, cygwin_finished_initializing and __in_forkee are true, so they are not
sufficient to discern the situation. */
static bool NO_COPY in_load_after_fork;
@ -161,7 +161,7 @@ dll_global_dtors ()
/* Don't attempt to call destructors if we're still in fork processing
since that likely means fork is failing and everything will not have been
set up. */
if (in_forkee)
if (__in_forkee == FORKING)
return;
int recorded = dll_global_dtors_recorded;
dll_global_dtors_recorded = false;
@ -205,7 +205,7 @@ dll::init ()
int ret = 1;
/* Don't run constructors or the "main" if we've forked. */
if (!in_forkee)
if (__in_forkee != FORKING)
{
/* global contructors */
p.run_ctors ();
@ -564,7 +564,7 @@ dll_list::detach (void *retaddr)
/* Don't attempt to call destructors if we're still in fork processing
since that likely means fork is failing and everything will not have been
set up. */
if (!myself || in_forkee)
if (!myself || __in_forkee == FORKING)
return;
guard (true);
if ((d = find (retaddr)))

View File

@ -639,7 +639,7 @@ dofork (void **proc, bool *with_forkables)
else
{
res = grouped.child (stackp);
in_forkee = false;
__in_forkee = FORKED;
ischild = true; /* might have been reset by fork mem copy */
}
}

View File

@ -73,8 +73,6 @@ bool wincmdln;
winsym_t allow_winsymlinks = WSYM_default;
bool disable_pcon;
bool NO_COPY in_forkee;
/* Taken from BSD libc:
This variable is zero until a process has created a pthread. It is used
to avoid calling locking functions in libc when they are not required.
@ -183,6 +181,17 @@ extern "C" {
/* impure_ptr */ _GLOBAL_REENT,
};
int _check_for_executable = true;
/* This was a bool initially, just indicating if we're in the forked
child during fork(2). However, we need an indicator accessible from
plain C we can ask if we're in a forked child even after fork(2)
finished. Therefore redefined how we use this variable. */
enum {
NOT_FORKED = 0,
FORKING = 1,
FORKED = 2
};
int NO_COPY __in_forkee;
};
int NO_COPY __api_fatal_exit_val = 1;

View File

@ -154,7 +154,7 @@ user_heap_info::init ()
if ((reserve_size -= page_const) < allocsize)
break;
}
if (!p && in_forkee && !fork_info->abort (NULL))
if (!p && __in_forkee == FORKING && !fork_info->abort (NULL))
api_fatal ("couldn't allocate heap, %E, base %p, top %p, "
"reserve_size %ld, allocsize %ld, page_const %d",
base, top,