X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/b1ac3b56e6b5b7342d69aa3b33744c345edb3d1e..f9133b32002a600cb48dd05fbcce1ab3ed684dba:/src/common/init.cpp diff --git a/src/common/init.cpp b/src/common/init.cpp index 6182c6734f..c25c12cd86 100644 --- a/src/common/init.cpp +++ b/src/common/init.cpp @@ -27,207 +27,463 @@ #include "wx/app.h" #include "wx/debug.h" #include "wx/filefn.h" + #include "wx/log.h" + #include "wx/thread.h" #endif -#include "wx/module.h" +#include "wx/init.h" -// ---------------------------------------------------------------------------- -// global vars -// ---------------------------------------------------------------------------- +#include "wx/ptr_scpd.h" +#include "wx/module.h" -WXDLLEXPORT wxApp *wxTheApp = NULL; +#if defined(__WXMSW__) && defined(__WXDEBUG__) + #include "wx/msw/msvcrt.h" -wxAppInitializerFunction - wxAppBase::m_appInitFn = (wxAppInitializerFunction)NULL; + static struct EnableMemLeakChecking + { + EnableMemLeakChecking() + { + // do check for memory leaks on program exit (another useful flag + // is _CRTDBG_DELAY_FREE_MEM_DF which doesn't free deallocated + // memory which may be used to simulate low-memory condition) + wxCrtSetDbgFlag(_CRTDBG_LEAK_CHECK_DF); + } + } gs_enableLeakChecks; +#endif // __WXMSW__ && __WXDEBUG__ // ---------------------------------------------------------------------------- // private classes // ---------------------------------------------------------------------------- -class /* no WXDLLEXPORT */ wxConsoleApp : public wxApp +// we need a dummy app object if the user doesn't want to create a real one +class wxDummyConsoleApp : public wxAppConsole { public: - virtual int OnRun() { wxFAIL_MSG(wxT("unreachable")); return 0; } - virtual bool ProcessIdle() { return TRUE; } + virtual int OnRun() { wxFAIL_MSG( _T("unreachable code") ); return 0; } +}; + +// we need a special kind of auto pointer to wxApp which not only deletes the +// pointer it holds in its dtor but also resets wxTheApp +wxDECLARE_SCOPED_PTR(wxApp, wxAppPtrBase); +wxDEFINE_SCOPED_PTR(wxApp, wxAppPtrBase); + +class wxAppPtr : public wxAppPtrBase +{ +public: + wxEXPLICIT wxAppPtr(wxApp *ptr = NULL) : wxAppPtrBase(ptr) { } + ~wxAppPtr() + { + if ( get() ) + { + // the pointer is going to be deleted in the base class dtor, don't + // leave the dangling pointer! + wxTheApp = NULL; + } + } + + void Set(wxApp *ptr) + { + reset(ptr); + + wxTheApp = ptr; + } +}; + +// class to ensure that wxAppBase::CleanUp() is called if our Initialize() +// fails +class wxCallAppCleanup +{ +public: + wxCallAppCleanup(wxApp *app) : m_app(app) { } + ~wxCallAppCleanup() { if ( m_app ) m_app->CleanUp(); } + + void Dismiss() { m_app = NULL; } + +private: + wxApp *m_app; +}; + +// another tiny class which simply exists to ensure that wxEntryCleanup is +// always called +class wxCleanupOnExit +{ +public: + ~wxCleanupOnExit() { wxEntryCleanup(); } }; // ---------------------------------------------------------------------------- // private functions // ---------------------------------------------------------------------------- -static bool DoInit(); -static void DoCleanUp(); +// suppress warnings about unused variables +static inline void Use(void *) { } + +#define WX_SUPPRESS_UNUSED_WARN(x) Use(&x) // ---------------------------------------------------------------------------- -// private vars +// initialization data // ---------------------------------------------------------------------------- -static size_t gs_nInitCount = 0; +static struct InitData +{ + InitData() + { + nInitCount = 0; + +#if wxUSE_UNICODE + argc = 0; + // argv = NULL; -- not even really needed +#endif // wxUSE_UNICODE + } + + // critical section protecting this struct + wxCRIT_SECT_DECLARE_MEMBER(csInit); + + // number of times wxInitialize() was called minus the number of times + // wxUninitialize() was + size_t nInitCount; + +#if wxUSE_UNICODE + int argc; + + // if we receive the command line arguments as ASCII and have to convert + // them to Unicode ourselves (this is the case under Unix but not Windows, + // for example), we remember the converted argv here because we'll have to + // free it when doing cleanup to avoid memory leaks + wchar_t **argv; +#endif // wxUSE_UNICODE +} gs_initData; // ============================================================================ // implementation // ============================================================================ // ---------------------------------------------------------------------------- -// stubs for some GUI functions +// command line arguments ANSI -> Unicode conversion // ---------------------------------------------------------------------------- -void WXDLLEXPORT wxExit() +#if wxUSE_UNICODE + +static void ConvertArgsToUnicode(int argc, char **argv) { - abort(); + gs_initData.argv = new wchar_t *[argc + 1]; + for ( int i = 0; i < argc; i++ ) + { + gs_initData.argv[i] = wxStrdup(wxConvLocal.cMB2WX(argv[i])); + } + + gs_initData.argv[argc] = NULL; } -// Yield to other apps/messages -void WXDLLEXPORT wxWakeUpIdle() +static void FreeConvertedArgs() { - // do nothing + if ( gs_initData.argv ) + { + for ( int i = 0; i < gs_initData.argc; i++ ) + { + free(gs_initData.argv[i]); + } + + delete [] gs_initData.argv; + gs_initData.argv = NULL; + } } +#endif // wxUSE_UNICODE + // ---------------------------------------------------------------------------- -// wxBase-specific functions +// start up // ---------------------------------------------------------------------------- -bool WXDLLEXPORT wxInitialize() +// initialization which is always done (not customizable) before wxApp creation +static bool DoCommonPreInit() { - if ( gs_nInitCount ) + wxClassInfo::InitializeClasses(); + + return true; +} + +// non customizable initialization done after wxApp creation and initialization +static bool DoCommonPostInit() +{ + wxModule::RegisterModules(); + + return wxModule::InitializeModules(); +} + +bool wxEntryStart(int& argc, wxChar **argv) +{ + // do minimal, always necessary, initialization + // -------------------------------------------- + + // initialize wxRTTI + if ( !DoCommonPreInit() ) { - // already initialized - return TRUE; + return false; } - wxASSERT_MSG( !wxTheApp, - wxT("either call wxInitialize or create app, not both!") ); - if ( !DoInit() ) + // first of all, we need an application object + // ------------------------------------------- + + // the user might have already created it himself somehow + wxAppPtr app(wxTheApp); + if ( !app.get() ) { - return FALSE; + // if not, he might have used IMPLEMENT_APP() to give us a function to + // create it + wxAppInitializerFunction fnCreate = wxApp::GetInitializerFunction(); + + if ( fnCreate ) + { + // he did, try to create the custom wxApp object + app.Set((*fnCreate)()); + } } - wxTheApp = new wxConsoleApp; + if ( !app.get() ) + { + // either IMPLEMENT_APP() was not used at all or it failed -- in any + // case we still need something + // + // NB: cast is needed because for the backwards-compatibility reasons + // wxTheApp is really a wxApp and not just wxAppConsole... + app.Set((wxApp *)new wxDummyConsoleApp); + } + + + // wxApp initialization: this can be customized + // -------------------------------------------- - if ( !wxTheApp ) + if ( !wxTheApp->Initialize(argc, argv) ) { - return FALSE; + return false; } - wxTheApp->DoInit(); - gs_nInitCount++; + wxCallAppCleanup callAppCleanup(wxTheApp); + + // for compatibility call the old initialization function too + if ( !wxTheApp->OnInitGui() ) + return false; + + + // common initialization after wxTheApp creation + // --------------------------------------------- + + if ( !DoCommonPostInit() ) + return false; + + + // prevent the smart pointer from destroying its contents + app.release(); + + // and the cleanup object from doing cleanup + callAppCleanup.Dismiss(); - return TRUE; + return true; } -void WXDLLEXPORT wxUninitialize() +#if wxUSE_UNICODE + +// we provide a wxEntryStart() wrapper taking "char *" pointer too +bool wxEntryStart(int& argc, char **argv) { - if ( !--gs_nInitCount ) + ConvertArgsToUnicode(argc, argv); + + if ( !wxEntryStart(argc, gs_initData.argv) ) { - DoCleanUp(); + FreeConvertedArgs(); + + return false; } + + return true; } -int wxEntry(int argc, char **argv) +#endif // wxUSE_UNICODE + +// ---------------------------------------------------------------------------- +// clean up +// ---------------------------------------------------------------------------- + +// cleanup done before destroying wxTheApp +static void DoCommonPreCleanup() { - // library initialization - if ( !DoInit() ) +#if wxUSE_LOG + // flush the logged messages if any and install a 'safer' log target: the + // default one (wxLogGui) can't be used after the resources are freed just + // below and the user supplied one might be even more unsafe (using any + // wxWindows GUI function is unsafe starting from now) + wxLog::DontCreateOnDemand(); + + // this will flush the old messages if any + delete wxLog::SetActiveTarget(new wxLogStderr); +#endif // wxUSE_LOG + + wxModule::CleanUpModules(); +} + +// cleanup done after destroying wxTheApp +static void DoCommonPostCleanup() +{ + wxClassInfo::CleanUpClasses(); + + // we can't do this in wxApp itself because it doesn't know if argv had + // been allocated +#if wxUSE_UNICODE + FreeConvertedArgs(); +#endif // wxUSE_UNICODE + +#if wxUSE_LOG + // and now delete the last logger as well + delete wxLog::SetActiveTarget(NULL); +#endif // wxUSE_LOG +} + +void wxEntryCleanup() +{ + DoCommonPreCleanup(); + + + // delete the application object + if ( wxTheApp ) { - return -1; + wxTheApp->CleanUp(); + + delete wxTheApp; + wxTheApp = NULL; } - // create the app - if ( !wxTheApp ) - { - wxCHECK_MSG( wxApp::GetInitializerFunction(), -1, - wxT("No application object: use IMPLEMENT_APP macro.") ); - wxAppInitializerFunction fnCreate = wxApp::GetInitializerFunction(); + DoCommonPostCleanup(); - wxTheApp = (wxApp *)fnCreate(); + // check for memory leaks +#if (defined(__WXDEBUG__) && wxUSE_MEMORY_TRACING) || wxUSE_DEBUG_CONTEXT + if (wxDebugContext::CountObjectsLeft(TRUE) > 0) + { + wxLogDebug(wxT("There were memory leaks.\n")); + wxDebugContext::Dump(); + wxDebugContext::PrintStatistics(); } +#endif // Debug + +} - wxCHECK_MSG( wxTheApp, -1, wxT("wxWindows error: no application object") ); +// ---------------------------------------------------------------------------- +// wxEntry +// ---------------------------------------------------------------------------- - // app preinitialization - wxTheApp->argc = argc; +#if !defined(__WXMSW__) || !wxUSE_ON_FATAL_EXCEPTION + #define wxEntryReal wxEntry +#endif // !(__WXMSW__ && wxUSE_ON_FATAL_EXCEPTION) -#if wxUSE_UNICODE - wxTheApp->argv = new wxChar*[argc+1]; - int mb_argc = 0; - while (mb_argc < argc) +int wxEntryReal(int& argc, wxChar **argv) +{ + // library initialization + if ( !wxEntryStart(argc, argv) ) { - wxTheApp->argv[mb_argc] = wxStrdup(wxConvLocal.cMB2WX(argv[mb_argc])); - mb_argc++; + return -1; } - wxTheApp->argv[mb_argc] = (wxChar *)NULL; -#else - wxTheApp->argv = argv; -#endif - wxString name = wxFileNameFromPath(argv[0]); - wxStripExtension(name); - wxTheApp->SetAppName(name); + // if wxEntryStart succeeded, we must call wxEntryCleanup even if the code + // below returns or throws + wxCleanupOnExit cleanupOnExit; - int retValue = 0; + WX_SUPPRESS_UNUSED_WARN(cleanupOnExit); // app initialization - if ( !wxTheApp->OnInit() ) - retValue = -1; + if ( !wxTheApp->CallOnInit() ) + { + // don't call OnExit() if OnInit() failed + return -1; + } // app execution - if ( retValue == 0 ) + int retValue = wxTheApp->OnRun(); + + // why should we do this? it doesn't close all window, just one of them and + // this shouldn't be necessary anyhow... +#if 0 + // close any remaining windows + wxWindow *topWindow = wxTheApp->GetTopWindow(); + if ( topWindow ) { - retValue = wxTheApp->OnRun(); + // forcibly delete the window. + topWindow->Destroy(); - // app clean up - wxTheApp->OnExit(); + // collect the dead objects + wxTheApp->DeletePendingObjects(); } +#endif // 0 - // library clean up - DoCleanUp(); + // app clean up + wxTheApp->OnExit(); return retValue; } -// ---------------------------------------------------------------------------- -// private functions -// ---------------------------------------------------------------------------- +// wrap real wxEntry in a try-except block to be able to call +// OnFatalException() if necessary +#if defined(__WXMSW__) && wxUSE_ON_FATAL_EXCEPTION -static bool DoInit() -{ - wxClassInfo::InitializeClasses(); +extern unsigned long wxGlobalSEHandler(); - wxModule::RegisterModules(); - if ( !wxModule::InitializeModules() ) +int wxEntry(int& argc, wxChar **argv) +{ + __try { - return FALSE; + return wxEntryReal(argc, argv); } + __except ( wxGlobalSEHandler() ) + { + ::ExitProcess(3); // the same exit code as abort() - return TRUE; +#if !defined(_MSC_VER) || _MSC_VER < 1300 + // this code is unreachable but put it here to suppress warnings + // from some compilers + return -1; +#endif + } } -static void DoCleanUp() +#endif // __WXMSW__ && wxUSE_ON_FATAL_EXCEPTION + +#if wxUSE_UNICODE + +// as with wxEntryStart, we provide an ANSI wrapper +int wxEntry(int& argc, char **argv) { -#if wxUSE_LOG - // flush the logged messages if any - wxLog *log = wxLog::GetActiveTarget(); - if (log != NULL && log->HasPendingMessages()) - log->Flush(); - - // continuing to use user defined log target is unsafe from now on because - // some resources may be already unavailable, so replace it by something - // more safe - wxLog::DontCreateOnDemand(); - delete wxLog::SetActiveTarget(new wxLogStderr); -#endif // wxUSE_LOG + ConvertArgsToUnicode(argc, argv); - wxModule::CleanUpModules(); + return wxEntry(argc, gs_initData.argv); +} - wxClassInfo::CleanUpClasses(); +#endif // wxUSE_UNICODE - // delete the application object - delete wxTheApp; - wxTheApp = (wxApp *)NULL; +// ---------------------------------------------------------------------------- +// wxInitialize/wxUninitialize +// ---------------------------------------------------------------------------- -#if wxUSE_LOG - // and now delete the last logger as well - delete wxLog::SetActiveTarget(NULL); -#endif // wxUSE_LOG +bool wxInitialize(int argc, wxChar **argv) +{ + wxCRIT_SECT_LOCKER(lockInit, gs_initData.csInit); + + if ( gs_initData.nInitCount++ ) + { + // already initialized + return true; + } + + return wxEntryStart(argc, argv); +} + +void wxUninitialize() +{ + wxCRIT_SECT_LOCKER(lockInit, gs_initData.csInit); + + if ( !--gs_initData.nInitCount ) + { + wxEntryCleanup(); + } } +