sal/osl/w32/dllentry.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19
  20 #include <systools/win32/uwinapi.h>
  21 #include <process.h>
  22 #include <tlhelp32.h>
  23 #include <rpc.h>
  24 #include <winsock.h>
  25 #ifdef _DEBUG
  26 #include <crtdbg.h>
  27 #endif
  28 #include <osl/diagnose.h>
  29 #include <sal/types.h>
  30 #include <float.h>
  31
  32 #include <osl/mutex.h>
  33 #include <osl/thread.h>
  34
  35 #include "file_url.hxx"
  36 #include <rtllifecycle.h>
  37
  38 #include "thread.hxx"
  39
  40 /*
  41 This is needed because DllMain is called after static constructors. A DLL's
  42 startup and shutdown sequence looks like this:
  43
  44 _pRawDllMain()
  45 _CRT_INIT()
  46 DllMain()
  47 ...
  48 DllMain()
  49 _CRT_INIT()
  50 _pRawDllMain()
  51
  52 */
  53
  54 extern "C" {
  55
  56 static BOOL WINAPI RawDllMain( HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved );
  57 BOOL (WINAPI *_pRawDllMain)(HINSTANCE, DWORD, LPVOID) = RawDllMain;
  58
  59 }
  60
  61 static BOOL WINAPI RawDllMain( HINSTANCE, DWORD fdwReason, LPVOID )
  62 {
  63     switch (fdwReason)
  64     {
  65         case DLL_PROCESS_ATTACH:
  66             {
  67 #ifdef _DEBUG
  68                 WCHAR buf[64];
  69                 DWORD const res = GetEnvironmentVariableW(L"SAL_NO_ASSERT_DIALOGS", buf, SAL_N_ELEMENTS(buf));
  70                 if (res && res < SAL_N_ELEMENTS(buf))
  71                 {
  72                     // disable the dialog on abort()
  73                     _CrtSetReportFile(_CRT_ERROR, _CRTDBG_FILE_STDERR);
  74                     _CrtSetReportMode(_CRT_ERROR, (_CRTDBG_MODE_DEBUG|_CRTDBG_MODE_FILE));
  75                     // not sure which assertions this affects
  76                     _CrtSetReportFile(_CRT_ASSERT, _CRTDBG_FILE_STDERR);
  77                     _CrtSetReportMode(_CRT_ASSERT, (_CRTDBG_MODE_DEBUG|_CRTDBG_MODE_FILE));
  78                     // disable the dialog on assert(false)
  79                     _set_error_mode(_OUT_TO_STDERR);
  80                 }
  81 #endif
  82
  83 #if OSL_DEBUG_LEVEL < 2
  84                 /* Suppress file error messages from system like "Floppy A: not inserted" */
  85                 SetErrorMode( SEM_NOOPENFILEERRORBOX | SEM_FAILCRITICALERRORS );
  86 #endif
  87
  88                 //We disable floating point exceptions. This is the usual state at program startup
  89                 //but on Windows 98 and ME this is not always the case.
  90                 _control87(_MCW_EM, _MCW_EM);
  91                 break;
  92             }
  93
  94         case DLL_PROCESS_DETACH:
  95             WSACleanup( );
  96
  97             /*
  98
  99             On a product build memory management finalization might
 100             cause a crash without assertion (assertions off) if heap is
 101             corrupted. But a crash report won't help here because at
 102             this point all other threads have been terminated and only
 103             ntdll is on the stack. No chance to find the reason for the
 104             corrupted heap if so.
 105
 106             So annoying the user with a crash report is completely useless.
 107
 108             */
 109
 110 #ifndef DBG_UTIL
 111             __try
 112 #endif
 113             {
 114                 /* cleanup locale hashtable */
 115                 rtl_locale_fini();
 116
 117                 /* finalize memory management */
 118                 rtl_cache_fini();
 119                 rtl_arena_fini();
 120             }
 121 #ifndef DBG_UTIL
 122             __except( EXCEPTION_EXECUTE_HANDLER )
 123             {
 124             }
 125 #endif
 126             break;
 127     }
 128
 129     return TRUE;
 130 }
 131
 132 static DWORD GetParentProcessId()
 133 {
 134     DWORD   dwParentProcessId = 0;
 135     HANDLE  hSnapshot = CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0 );
 136
 137     if ( IsValidHandle( hSnapshot ) )
 138     {
 139         PROCESSENTRY32  pe;
 140         bool            fSuccess;
 141
 142         ZeroMemory( &pe, sizeof(pe) );
 143         pe.dwSize = sizeof(pe);
 144         fSuccess = Process32First( hSnapshot, &pe );
 145
 146         while( fSuccess )
 147         {
 148             if ( GetCurrentProcessId() == pe.th32ProcessID )
 149             {
 150                 dwParentProcessId = pe.th32ParentProcessID;
 151                 break;
 152             }
 153
 154             fSuccess = Process32Next( hSnapshot, &pe );
 155         }
 156
 157         CloseHandle( hSnapshot );
 158     }
 159
 160     return dwParentProcessId;
 161 }
 162
 163 static unsigned __stdcall ParentMonitorThreadProc(void* lpParam)
 164 {
 165     DWORD_PTR dwParentProcessId = reinterpret_cast<DWORD_PTR>(lpParam);
 166
 167     HANDLE  hParentProcess = OpenProcess( SYNCHRONIZE, FALSE, dwParentProcessId );
 168
 169     osl_setThreadName("headless ParentMonitorThread");
 170
 171     if ( IsValidHandle( hParentProcess ) )
 172     {
 173         if ( WAIT_OBJECT_0 == WaitForSingleObject( hParentProcess, INFINITE ) )
 174         {
 175             TerminateProcess( GetCurrentProcess(), 0 );
 176         }
 177         CloseHandle( hParentProcess );
 178     }
 179     return 0;
 180 }
 181
 182 extern "C"
 183 BOOL WINAPI DllMain( HINSTANCE, DWORD fdwReason, LPVOID )
 184 {
 185     switch (fdwReason)
 186     {
 187         case DLL_PROCESS_ATTACH:
 188         {
 189             WCHAR szBuffer[64];
 190
 191             // This code will attach the process to its parent process
 192             // if the parent process had set the environment variable.
 193             // The corresponding code (setting the environment variable)
 194             // is desktop/win32/source/officeloader.cxx
 195
 196             DWORD dwResult = GetEnvironmentVariableW( L"ATTACHED_PARENT_PROCESSID", szBuffer, SAL_N_ELEMENTS(szBuffer) );
 197
 198             if ( dwResult && dwResult < SAL_N_ELEMENTS(szBuffer) )
 199             {
 200                 DWORD_PTR dwParentProcessId = static_cast<DWORD_PTR>(_wtol( szBuffer ));
 201
 202                 if ( dwParentProcessId && GetParentProcessId() == dwParentProcessId )
 203                 {
 204                     // No error check, it works or it does not
 205                     // Thread should only be started for headless mode, see desktop/win32/source/officeloader.cxx
 206                     HANDLE hThread
 207                         = reinterpret_cast<HANDLE>(_beginthreadex(nullptr, 0, ParentMonitorThreadProc,
 208                                        reinterpret_cast<LPVOID>(dwParentProcessId), 0, nullptr));
 209                     // Note: calling CreateThread in DllMain is discouraged
 210                     // but this is only done in the headless mode and in
 211                     // that case no other threads should be running at startup
 212                     // when sal3.dll is loaded; also there is no
 213                     // synchronization with the spawned thread, so there
 214                     // does not appear to be a real risk of deadlock here
 215                     if (hThread)
 216                         CloseHandle(hThread);
 217                 }
 218             }
 219
 220             return TRUE;
 221         }
 222
 223         case DLL_THREAD_ATTACH:
 224             break;
 225
 226         case DLL_THREAD_DETACH:
 227             osl_callThreadKeyCallbackOnThreadDetach( );
 228             break;
 229     }
 230
 231     return TRUE;
 232 }
 233
 234 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */