tcrt/src/init.c

   1 /*=============================================================================
   2         init.c : tinycrt initialization (tiny crt)
   3
   4         DO NOT use ltcg optimizations in this file under penalty of death
   5         by TLS corruption.
   6 =============================================================================*/
   7
   8 #include <signal.h>
   9 #include <windows.h>
  10 #include "init.h"
  11
  12 ///////////////////////////////////////////////////////////////////////////////
  13 #pragma comment(linker, "/merge:.CRT=.rdata")
  14 #pragma comment(linker, "/defaultlib:kernel32.lib")
  15 #pragma comment(linker, "/disallowlib:libc.lib")
  16 #pragma comment(linker, "/disallowlib:libcd.lib")
  17 #pragma comment(linker, "/disallowlib:libcmt.lib")
  18 #pragma comment(linker, "/disallowlib:libcmtd.lib")
  19 #pragma comment(linker, "/disallowlib:msvcrt.lib")
  20 #pragma comment(linker, "/disallowlib:msvcrtd.lib")
  21
  22 #if defined (_WIN64) && defined (_M_IA64)
  23 #       pragma section(".base", long, read)
  24         _CRTALLOC(".base")
  25 #endif
  26
  27 _CRTALLOC(".CRT$XIA") _PIFV __xi_a[] = { NULL };
  28 _CRTALLOC(".CRT$XIZ") _PIFV __xi_z[] = { NULL };
  29 _CRTALLOC(".CRT$XCA") _PVFV __xc_a[] = { NULL };
  30 _CRTALLOC(".CRT$XCZ") _PVFV __xc_z[] = { NULL };
  31 _CRTALLOC(".CRT$XPA") _PVFV __xp_a[] = { NULL };
  32 _CRTALLOC(".CRT$XPZ") _PVFV __xp_z[] = { NULL };
  33 _CRTALLOC(".CRT$XTA") _PVFV __xt_a[] = { NULL };
  34 _CRTALLOC(".CRT$XTZ") _PVFV __xt_z[] = { NULL };
  35
  36 ///////////////////////////////////////////////////////////////////////////////
  37 static BOOL WINAPI __dyn_tls_dtor(HANDLE handle, DWORD reason, LPVOID reserved);
  38 static BOOL WINAPI __dyn_tls_init(HANDLE handle, DWORD reason, LPVOID reserved);
  39
  40 ///////////////////////////////////////////////////////////////////////////////
  41 ULONG _tls_index = 0;
  42
  43 #pragma data_seg(".tls")
  44 #if defined (_M_IA64) || defined (_M_AMD64)
  45         _CRTALLOC(".tls")
  46 #endif
  47
  48 char _tls_start = 0;
  49
  50 #pragma data_seg(".tls$ZZZ")
  51 #if defined (_M_IA64) || defined (_M_AMD64)
  52         _CRTALLOC(".tls$ZZZ")
  53 #endif
  54
  55 char _tls_end = 0;
  56
  57 #pragma data_seg()
  58
  59 _CRTALLOC(".CRT$XLA") PIMAGE_TLS_CALLBACK __xl_a = 0;
  60 _CRTALLOC(".CRT$XLZ") PIMAGE_TLS_CALLBACK __xl_z = 0;
  61
  62 #ifdef _WIN64
  63         _CRTALLOC(".rdata$T") const IMAGE_TLS_DIRECTORY64 _tls_used = {
  64                 (ULONGLONG) &_tls_start,        // start of tls data
  65                 (ULONGLONG) &_tls_end,          // end of tls data
  66                 (ULONGLONG) &_tls_index,        // address of tls_index
  67                 (ULONGLONG) (&__xl_a+1),        // pointer to call back array
  68                 (ULONG) 0,                      // size of tls zero fill
  69                 (ULONG) 0                       // characteristics
  70         };
  71 #else
  72         _CRTALLOC(".rdata$T") const IMAGE_TLS_DIRECTORY _tls_used = {
  73                 (ULONG)(ULONG_PTR) &_tls_start, // start of tls data
  74                 (ULONG)(ULONG_PTR) &_tls_end,   // end of tls data
  75                 (ULONG)(ULONG_PTR) &_tls_index, // address of tls_index
  76                 (ULONG)(ULONG_PTR) (&__xl_a+1), // pointer to call back array
  77                 (ULONG) 0,                      // size of tls zero fill
  78                 (ULONG) 0                       // characteristics
  79         };
  80 #endif
  81
  82 static _CRTALLOC(".CRT$XDA") _PVFV __xd_a = 0;
  83 static _CRTALLOC(".CRT$XDZ") _PVFV __xd_z = 0;
  84
  85 static _CRTALLOC(".CRT$XLC") PIMAGE_TLS_CALLBACK __xl_c = __dyn_tls_init;
  86 static _CRTALLOC(".CRT$XLD") PIMAGE_TLS_CALLBACK __xl_d = __dyn_tls_dtor;
  87
  88 #define FUNCS_PER_NODE  30
  89
  90 typedef struct TlsDtorNode {
  91     int count;
  92     struct TlsDtorNode *next;
  93     _PVFV funcs[FUNCS_PER_NODE];
  94 } TlsDtorNode;
  95
  96 static __declspec(thread) TlsDtorNode*  dtor_list;
  97 static __declspec(thread) TlsDtorNode   dtor_list_head;
  98
  99 atexit_entry* g_atexit_list;
 100
 101 ///////////////////////////////////////////////////////////////////////////////
 102 static BOOL WINAPI __dyn_tls_init(HANDLE handle, DWORD reason, LPVOID reserved)
 103 {
 104         _PVFV *pfunc;
 105
 106         if (reason != DLL_THREAD_ATTACH) {
 107                 return TRUE;
 108         }
 109
 110         #pragma warning(push)
 111         #pragma warning(disable:26000)
 112         for(pfunc = &__xd_a + 1; pfunc != &__xd_z; ++pfunc) {
 113                 if (*pfunc != NULL) {
 114                         (*pfunc)();
 115                 }
 116         }
 117         #pragma warning(pop)
 118
 119         return TRUE;
 120 }
 121
 122 int __cdecl __tlregdtor(_PVFV func)
 123 {
 124         if (dtor_list == NULL) {
 125                 dtor_list = &dtor_list_head;
 126                 dtor_list_head.count = 0;
 127         }
 128         else if (dtor_list->count == FUNCS_PER_NODE) {
 129                 TlsDtorNode* pnode = (TlsDtorNode*) malloc(sizeof(TlsDtorNode));
 130                 if (pnode == NULL)
 131                         return -1;
 132                 pnode->count = 0;
 133                 pnode->next = dtor_list;
 134                 dtor_list = pnode;
 135                 dtor_list->count = 0;
 136         }
 137         dtor_list->funcs[dtor_list->count++] = func;
 138         return 0;
 139 }
 140
 141 static BOOL WINAPI __dyn_tls_dtor(HANDLE handle, DWORD reason, LPVOID reserved)
 142 {
 143         TlsDtorNode *pnode, *pnext;
 144         int i;
 145
 146         if (reason != DLL_THREAD_DETACH && reason != DLL_PROCESS_DETACH) {
 147                 return TRUE;
 148         }
 149
 150         for (pnode = dtor_list; pnode != NULL; pnode = pnext) {
 151                 for (i = pnode->count - 1; i >= 0; --i) {
 152                         if (pnode->funcs[i] != NULL) {
 153                                 (*pnode->funcs[i])();
 154                         }
 155                 }
 156                 pnext = pnode->next;
 157                 if (pnext != NULL) {
 158                         free(pnode);
 159                 }
 160         }
 161
 162         return TRUE;
 163 }
 164
 165 static void __atexit(void)
 166 {
 167         atexit_entry* next = g_atexit_list;
 168
 169         while (next != NULL) {
 170                 next->fn();
 171                 next = next->next;
 172         }
 173         if (g_tcrt_sigaction_slot[SIGTERM].sa_handler != SIG_DFL)
 174                 g_tcrt_sigaction_slot[SIGTERM].sa_handler(SIGTERM);
 175 }
 176
 177 ///////////////////////////////////////////////////////////////////////////////
 178 void run_ctors(void)
 179 {
 180         _PIFV* ii;
 181         _PVFV* ic;
 182         int ret;
 183
 184         __dyn_tls_init(NULL, DLL_THREAD_ATTACH, NULL);
 185         for(ii = __xi_a; ii < __xi_z; ii++) {
 186                 if( *ii != NULL ) {
 187                         ret = (**ii)();
 188                         if( ret != 0 ) {
 189                                 ExitProcess(ret);
 190                         }
 191                 }
 192         }
 193
 194         for(ic = __xc_a; ic < __xc_z; ic++) {
 195                 if( *ic != NULL ) {
 196                         (**ic)();
 197                 }
 198         }
 199 }
 200
 201 void run_dtors(void)
 202 {
 203         _PVFV* ip;
 204         _PVFV* it;
 205
 206         for(ip = __xp_a; ip < __xp_z; ip++) {
 207                 if( *ip != NULL ) {
 208                         (**ip)();
 209                 }
 210         }
 211
 212         for(it = __xt_a; it < __xt_z; it++) {
 213                 if( *it != NULL ) {
 214                         (**it)();
 215                 }
 216         }
 217
 218         __atexit();
 219 }
 220
 221 // EOF ////////////////////////////////////////////////////////////////////////