From d5c4cd3f6ce4023376c550f74656303a425bdcd4 Mon Sep 17 00:00:00 2001 From: Christopher Faylor Date: Fri, 29 Jan 2010 18:34:09 +0000 Subject: * dll_init.cc (remove_dll_atexit): New function. (dll_list::detach): Run any atexit handlers registered in the DLL prior to unloading. --- winsup/cygwin/dll_init.cc | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) (limited to 'winsup/cygwin/dll_init.cc') diff --git a/winsup/cygwin/dll_init.cc b/winsup/cygwin/dll_init.cc index 20cc4f9ce..763964bac 100644 --- a/winsup/cygwin/dll_init.cc +++ b/winsup/cygwin/dll_init.cc @@ -20,6 +20,7 @@ details. */ #include "pinfo.h" #include "cygtls.h" #include +#include extern void __stdcall check_sanity_and_sync (per_process *); @@ -142,6 +143,32 @@ dll_list::alloc (HINSTANCE h, per_process *p, dll_type type) return d; } +/* This function looks for every atexit function registered in the + about-to-be-unloaded DLL and runs it. + + newlib does not provide any method for selectively running elements + from the atexit() queue so we have to roll our own. + + Note that this is not foolproof since a function in the DLL could + register an atexit function outside of the DLL and that should be + run when the DLL detachs. */ +static void +remove_dll_atexit (MEMORY_BASIC_INFORMATION& m) +{ + unsigned char *dll_beg = (unsigned char *) m.AllocationBase; + unsigned char *dll_end = (unsigned char *) m.AllocationBase + m.RegionSize; + struct _atexit *p = _GLOBAL_REENT->_atexit; + for (int n = p->_ind - 1; n >= 0; n--) + { + void (*fn) (void) = p->_fns[n]; + if ((unsigned char *) fn >= dll_beg && (unsigned char *) fn < dll_end) + { + fn (); + p->_fns[n] = NULL; + } + } +} + /* Detach a DLL from the chain. */ void dll_list::detach (void *retaddr) @@ -161,6 +188,7 @@ dll_list::detach (void *retaddr) system_printf ("WARNING: trying to detach an already detached dll ..."); else if (--d->count == 0) { + remove_dll_atexit (m); d->run_dtors (); d->prev->next = d->next; if (d->next) -- cgit v1.2.3