X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/e2478fde622a16d25c66690af353dfdc37e7b582..2a4a928df7ec1b05c457b088f8719ba6f67ee041:/src/common/init.cpp diff --git a/src/common/init.cpp b/src/common/init.cpp index b3cd9aa66d..561b726ccd 100644 --- a/src/common/init.cpp +++ b/src/common/init.cpp @@ -1,5 +1,5 @@ ///////////////////////////////////////////////////////////////////////////// -// Name: common/init.cpp +// Name: src/common/init.cpp // Purpose: initialisation for the library // Author: Vadim Zeitlin // Modified by: @@ -20,197 +20,515 @@ #include "wx/wxprec.h" #ifdef __BORLANDC__ - #pragma hdrstop + #pragma hdrstop #endif //__BORLANDC__ #ifndef WX_PRECOMP #include "wx/app.h" - #include "wx/debug.h" #include "wx/filefn.h" #include "wx/log.h" + #include "wx/intl.h" + #include "wx/module.h" #endif -#include "wx/module.h" +#include "wx/init.h" +#include "wx/thread.h" + +#include "wx/scopedptr.h" +#include "wx/except.h" + +#if defined(__WXMSW__) + #include "wx/msw/private.h" + #include "wx/msw/msvcrt.h" + + #ifdef wxCrtSetDbgFlag + static struct EnableMemLeakChecking + { + EnableMemLeakChecking() + { + // 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 // wxCrtSetDbgFlag +#endif // __WXMSW__ // ---------------------------------------------------------------------------- // private classes // ---------------------------------------------------------------------------- // we need a dummy app object if the user doesn't want to create a real one -class wxDummyConsoleApp : public wxApp +class wxDummyConsoleApp : public wxAppConsole { public: - virtual int OnRun() { wxFAIL_MSG( _T("unreachable code") ); return 0; } + wxDummyConsoleApp() { } + + virtual int OnRun() { wxFAIL_MSG( wxT("unreachable code") ); return 0; } + virtual bool DoYield(bool, long) { return true; } + + wxDECLARE_NO_COPY_CLASS(wxDummyConsoleApp); +}; + +// we need a special kind of auto pointer to wxApp which not only deletes the +// pointer it holds in its dtor but also resets the global application pointer +wxDECLARE_SCOPED_PTR(wxAppConsole, wxAppPtrBase) +wxDEFINE_SCOPED_PTR(wxAppConsole, wxAppPtrBase) + +class wxAppPtr : public wxAppPtrBase +{ +public: + wxEXPLICIT wxAppPtr(wxAppConsole *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! + wxApp::SetInstance(NULL); + } + } + + void Set(wxAppConsole *ptr) + { + reset(ptr); + + wxApp::SetInstance(ptr); + } + + wxDECLARE_NO_COPY_CLASS(wxAppPtr); +}; + +// class to ensure that wxAppBase::CleanUp() is called if our Initialize() +// fails +class wxCallAppCleanup +{ +public: + wxCallAppCleanup(wxAppConsole *app) : m_app(app) { } + ~wxCallAppCleanup() { if ( m_app ) m_app->CleanUp(); } + + void Dismiss() { m_app = NULL; } + +private: + wxAppConsole *m_app; }; // ---------------------------------------------------------------------------- // 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 + + wxDECLARE_NO_COPY_CLASS(InitData); +} gs_initData; // ============================================================================ // implementation // ============================================================================ // ---------------------------------------------------------------------------- -// wxBase-specific functions +// command line arguments ANSI -> Unicode conversion // ---------------------------------------------------------------------------- -bool WXDLLEXPORT wxInitialize() +#if wxUSE_UNICODE + +static void ConvertArgsToUnicode(int argc, char **argv) { - if ( gs_nInitCount ) + gs_initData.argv = new wchar_t *[argc + 1]; + int wargc = 0; + for ( int i = 0; i < argc; i++ ) { - // already initialized - return TRUE; +#ifdef __DARWIN__ + wxWCharBuffer buf(wxConvFileName->cMB2WX(argv[i])); +#else + wxWCharBuffer buf(wxConvLocal.cMB2WX(argv[i])); +#endif + if ( !buf ) + { + wxLogWarning(_("Command line argument %d couldn't be converted to Unicode and will be ignored."), + i); + } + else // converted ok + { + gs_initData.argv[wargc++] = wxStrdup(buf); + } } - wxASSERT_MSG( !wxTheApp, - wxT("either call wxInitialize or create app, not both!") ); + gs_initData.argc = wargc; + gs_initData.argv[wargc] = NULL; +} - if ( !DoInit() ) +static void FreeConvertedArgs() +{ + if ( gs_initData.argv ) { - return FALSE; + for ( int i = 0; i < gs_initData.argc; i++ ) + { + free(gs_initData.argv[i]); + } + + wxDELETEA(gs_initData.argv); + gs_initData.argc = 0; } +} + +#endif // wxUSE_UNICODE - wxTheApp = new wxDummyConsoleApp; +// ---------------------------------------------------------------------------- +// start up +// ---------------------------------------------------------------------------- + +// initialization which is always done (not customizable) before wxApp creation +static bool DoCommonPreInit() +{ +#if wxUSE_LOG + // Reset logging in case we were cleaned up and are being reinitialized. + wxLog::DoCreateOnDemand(); + + // force wxLog to create a log target now: we do it because wxTheApp + // doesn't exist yet so wxLog will create a special log target which is + // safe to use even when the GUI is not available while without this call + // we could create wxApp in wxEntryStart() below, then log an error about + // e.g. failure to establish connection to the X server and wxLog would + // send it to wxLogGui (because wxTheApp does exist already) which, of + // course, can't be used in this case + // + // notice also that this does nothing if the user had set up a custom log + // target before -- which is fine as we want to give him this possibility + // (as it's impossible to override logging by overriding wxAppTraits:: + // CreateLogTarget() before wxApp is created) and we just assume he knows + // what he is doing + wxLog::GetActiveTarget(); +#endif // wxUSE_LOG - if ( !wxTheApp ) +#ifdef __WXMSW__ + // GUI applications obtain HINSTANCE in their WinMain() but we also need to + // initialize the global wxhInstance variable for the console programs as + // they may need it too, so set it here if it wasn't done yet + if ( !wxGetInstance() ) { - return FALSE; + wxSetInstance(::GetModuleHandle(NULL)); } +#endif // __WXMSW__ - gs_nInitCount++; - - return TRUE; + return true; } -void WXDLLEXPORT wxUninitialize() +// non customizable initialization done after wxApp creation and initialization +static bool DoCommonPostInit() { - if ( !--gs_nInitCount ) + wxModule::RegisterModules(); + + if ( !wxModule::InitializeModules() ) { - DoCleanUp(); + wxLogError(_("Initialization failed in post init, aborting.")); + return false; } + + return true; } -int wxEntry(int argc, char **argv) +bool wxEntryStart(int& argc, wxChar **argv) { - // library initialization - if ( !DoInit() ) + // do minimal, always necessary, initialization + // -------------------------------------------- + + // initialize wxRTTI + if ( !DoCommonPreInit() ) + return false; + + + // first of all, we need an application object + // ------------------------------------------- + + // the user might have already created it himself somehow + wxAppPtr app(wxTheApp); + if ( !app.get() ) { - return -1; + // 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)()); + } } - // create the app - if ( !wxTheApp ) + if ( !app.get() ) { - wxCHECK_MSG( wxApp::GetInitializerFunction(), -1, - wxT("No application object: use IMPLEMENT_APP macro.") ); + // either IMPLEMENT_APP() was not used at all or it failed -- in any + // case we still need something + app.Set(new wxDummyConsoleApp); + } - wxAppInitializerFunction fnCreate = wxApp::GetInitializerFunction(); - wxTheApp = (wxApp *)fnCreate(); - } + // wxApp initialization: this can be customized + // -------------------------------------------- + + if ( !app->Initialize(argc, argv) ) + return false; + + // remember, possibly modified (e.g. due to removal of toolkit-specific + // parameters), command line arguments in member variables + app->argc = argc; + app->argv = argv; + + wxCallAppCleanup callAppCleanup(app.get()); - wxCHECK_MSG( wxTheApp, -1, wxT("wxWindows error: no application object") ); - // app preinitialization - wxTheApp->argc = argc; + // 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(); + +#if wxUSE_LOG + // now that we have a valid wxApp (wxLogGui would have crashed if we used + // it before now), we can delete the temporary sink we had created for the + // initialization messages -- the next time logging function is called, the + // sink will be recreated but this time wxAppTraits will be used + delete wxLog::SetActiveTarget(NULL); +#endif // wxUSE_LOG + + return true; +} #if wxUSE_UNICODE - wxTheApp->argv = new wxChar*[argc+1]; - for ( int mb_argc = 0; mb_argc < argc; mb_argc++ ) + +// we provide a wxEntryStart() wrapper taking "char *" pointer too +bool wxEntryStart(int& argc, char **argv) +{ + ConvertArgsToUnicode(argc, argv); + + if ( !wxEntryStart(gs_initData.argc, gs_initData.argv) ) { - wxTheApp->argv[mb_argc] = wxStrdup(wxConvLocal.cMB2WX(argv[mb_argc])); + FreeConvertedArgs(); + + return false; } - wxTheApp->argv[mb_argc] = (wxChar *)NULL; -#else - wxTheApp->argv = argv; -#endif - wxString name = wxFileNameFromPath(wxTheApp->argv[0]); - wxStripExtension(name); - wxTheApp->SetAppName(name); + return true; +} + +#endif // wxUSE_UNICODE + +// ---------------------------------------------------------------------------- +// clean up +// ---------------------------------------------------------------------------- - int retValue = 0; +// cleanup done before destroying wxTheApp +static void DoCommonPreCleanup() +{ +#if wxUSE_LOG + // flush the logged messages if any and don't use the current probably + // unsafe log target any more: the default one (wxLogGui) can't be used + // after the resources are freed which happens when we return and the user + // supplied one might be even more unsafe (using any wxWidgets GUI function + // is unsafe starting from now) + // + // notice that wxLog will still recreate a default log target if any + // messages are logged but that one will be safe to use until the very end + delete wxLog::SetActiveTarget(NULL); +#endif // wxUSE_LOG +} - // app initialization - if ( !wxTheApp->OnInit() ) - retValue = -1; +// cleanup done after destroying wxTheApp +static void DoCommonPostCleanup() +{ + wxModule::CleanUpModules(); + + // 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 + + // use Set(NULL) and not Get() to avoid creating a message output object on + // demand when we just want to delete it + delete wxMessageOutput::Set(NULL); + +#if wxUSE_LOG + // and now delete the last logger as well + // + // we still don't disable log target auto-vivification even if any log + // objects created now will result in memory leaks because it seems better + // to leak memory which doesn't matter much considering the application is + // exiting anyhow than to not show messages which could still be logged + // from the user code (e.g. static dtors and such) + delete wxLog::SetActiveTarget(NULL); +#endif // wxUSE_LOG +} + +void wxEntryCleanup() +{ + DoCommonPreCleanup(); - // app execution - if ( retValue == 0 ) - { - retValue = wxTheApp->OnRun(); - // app clean up - wxTheApp->OnExit(); + // delete the application object + if ( wxTheApp ) + { + wxTheApp->CleanUp(); + + // reset the global pointer to it to NULL before destroying it as in + // some circumstances this can result in executing the code using + // wxTheApp and using half-destroyed object is no good + wxAppConsole * const app = wxApp::GetInstance(); + wxApp::SetInstance(NULL); + delete app; } - // library clean up - DoCleanUp(); - return retValue; + DoCommonPostCleanup(); } // ---------------------------------------------------------------------------- -// private functions +// wxEntry // ---------------------------------------------------------------------------- -static bool DoInit() +// for MSW the real wxEntry is defined in msw/main.cpp +#ifndef __WXMSW__ + #define wxEntryReal wxEntry +#endif // !__WXMSW__ + +int wxEntryReal(int& argc, wxChar **argv) { - wxClassInfo::InitializeClasses(); + // library initialization + wxInitializer initializer(argc, argv); - wxModule::RegisterModules(); - if ( !wxModule::InitializeModules() ) + if ( !initializer.IsOk() ) { - return FALSE; +#if wxUSE_LOG + // flush any log messages explaining why we failed + delete wxLog::SetActiveTarget(NULL); +#endif + return -1; } - return TRUE; + wxTRY + { + // app initialization + if ( !wxTheApp->CallOnInit() ) + { + // don't call OnExit() if OnInit() failed + return -1; + } + + // ensure that OnExit() is called if OnInit() had succeeded + class CallOnExit + { + public: + ~CallOnExit() { wxTheApp->OnExit(); } + } callOnExit; + + WX_SUPPRESS_UNUSED_WARN(callOnExit); + + // app execution + return wxTheApp->OnRun(); + } + wxCATCH_ALL( wxTheApp->OnUnhandledException(); return -1; ) } -static void DoCleanUp() +#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(gs_initData.argc, gs_initData.argv); +} - wxClassInfo::CleanUpClasses(); +#endif // wxUSE_UNICODE + +// ---------------------------------------------------------------------------- +// wxInitialize/wxUninitialize +// ---------------------------------------------------------------------------- + +bool wxInitialize() +{ + return wxInitialize(0, (wxChar**)NULL); +} + +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); +} - // TODO: this should really be done in ~wxApp #if wxUSE_UNICODE - for ( int mb_argc = 0; mb_argc < wxTheApp->argc; mb_argc++ ) +bool wxInitialize(int argc, char **argv) +{ + wxCRIT_SECT_LOCKER(lockInit, gs_initData.csInit); + + if ( gs_initData.nInitCount++ ) { - free(wxTheApp->argv[mb_argc]); + // already initialized + return true; } + + return wxEntryStart(argc, argv); +} #endif // wxUSE_UNICODE - // delete the application object - delete wxTheApp; - wxTheApp = (wxApp *)NULL; +void wxUninitialize() +{ + wxCRIT_SECT_LOCKER(lockInit, gs_initData.csInit); -#if wxUSE_LOG - // and now delete the last logger as well - delete wxLog::SetActiveTarget(NULL); -#endif // wxUSE_LOG + if ( --gs_initData.nInitCount == 0 ) + { + wxEntryCleanup(); + } } -