X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/09b895cb4a954494063eb17d3dc302a654fd99e5..fcdd53359135f790b85728c4254b97095a56dad8:/src/msw/app.cpp?ds=sidebyside diff --git a/src/msw/app.cpp b/src/msw/app.cpp index 5012573d09..646470fe6f 100644 --- a/src/msw/app.cpp +++ b/src/msw/app.cpp @@ -50,6 +50,8 @@ #include "wx/dynlib.h" #include "wx/evtloop.h" #include "wx/thread.h" +#include "wx/scopeguard.h" +#include "wx/vector.h" #include "wx/msw/private.h" #include "wx/msw/dc.h" @@ -103,6 +105,9 @@ typedef HRESULT (CALLBACK* DLLGETVERSIONPROC)(DLLVERSIONINFO *); #endif // defined(DLLVERSIONINFO) +#ifndef ATTACH_PARENT_PROCESS + #define ATTACH_PARENT_PROCESS ((DWORD)-1) +#endif // --------------------------------------------------------------------------- // global variables @@ -112,19 +117,23 @@ extern void wxSetKeyboardHook(bool doIt); #endif -// NB: all "NoRedraw" classes must have the same names as the "normal" classes -// with NR suffix - wxWindow::MSWCreate() supposes this -#ifdef __WXWINCE__ -WXDLLIMPEXP_CORE wxChar *wxCanvasClassName; -WXDLLIMPEXP_CORE wxChar *wxCanvasClassNameNR; -#else -WXDLLIMPEXP_CORE const wxChar *wxCanvasClassName = NULL; -WXDLLIMPEXP_CORE const wxChar *wxCanvasClassNameNR = NULL; -#endif -WXDLLIMPEXP_CORE const wxChar *wxMDIFrameClassName = NULL; -WXDLLIMPEXP_CORE const wxChar *wxMDIFrameClassNameNoRedraw = NULL; -WXDLLIMPEXP_CORE const wxChar *wxMDIChildFrameClassName = NULL; -WXDLLIMPEXP_CORE const wxChar *wxMDIChildFrameClassNameNoRedraw = NULL; +namespace +{ + +struct ClassRegInfo +{ + // the base name of the class: this is used to construct the unique name in + // RegisterClassWithUniqueNames() + wxString basename; + + // the name of the registered class with and without CS_[HV]REDRAW styles + wxString regname, + regnameNR; +}; + +wxVector gs_regClassesInfo; + +} // anonymous namespace // ---------------------------------------------------------------------------- // private functions @@ -282,6 +291,290 @@ wxEventLoopBase* wxGUIAppTraits::CreateEventLoop() return new wxEventLoop; } +// --------------------------------------------------------------------------- +// Stuff for using console from the GUI applications +// --------------------------------------------------------------------------- + +#ifndef __WXWINCE__ + +#include + +namespace +{ + +/* + Helper class to manipulate console from a GUI app. + + Notice that console output is available in the GUI app only if: + - AttachConsole() returns TRUE (which means it never works under pre-XP) + - we have a valid STD_ERROR_HANDLE + - command history hasn't been changed since our startup + + To check if all these conditions are verified, you need to simple call + IsOkToUse(). It will check the first two conditions above the first time it + is called (and if this fails, the subsequent calls will return immediately) + and also recheck the last one every time it is called. + */ +class wxConsoleStderr +{ +public: + // default ctor does nothing, call Init() before using this class + wxConsoleStderr() + { + m_hStderr = INVALID_HANDLE_VALUE; + m_historyLen = + m_dataLen = + m_dataLine = 0; + + m_ok = -1; + } + + ~wxConsoleStderr() + { + if ( m_hStderr != INVALID_HANDLE_VALUE ) + { + if ( !::FreeConsole() ) + { + wxLogLastError(_T("FreeConsole")); + } + } + } + + // return true if we were successfully initialized and there had been no + // console activity which would interfere with our output since then + bool IsOkToUse() const + { + if ( m_ok == -1 ) + { + wxConsoleStderr * const self = const_cast(this); + self->m_ok = self->DoInit(); + + // no need to call IsHistoryUnchanged() as we just initialized + // m_history anyhow + return m_ok == 1; + } + + return m_ok && IsHistoryUnchanged(); + } + + + // output the provided text on the console, return true if ok + bool Write(const wxString& text); + +private: + // called by Init() once only to do the real initialization + bool DoInit(); + + // retrieve the command line history into the provided buffer and return + // its length + int GetCommandHistory(wxWxCharBuffer& buf) const; + + // check if the console history has changed + bool IsHistoryUnchanged() const; + + int m_ok; // initially -1, set to true or false by Init() + + wxDynamicLibrary m_dllKernel32; + + HANDLE m_hStderr; // console handle, if it's valid we must call + // FreeConsole() (even if m_ok != 1) + + wxWxCharBuffer m_history; // command history on startup + int m_historyLen; // length command history buffer + + wxCharBuffer m_data; // data between empty line and cursor position + int m_dataLen; // length data buffer + int m_dataLine; // line offset + + typedef DWORD (WINAPI *GetConsoleCommandHistory_t)(LPTSTR sCommands, + DWORD nBufferLength, + LPCTSTR sExeName); + typedef DWORD (WINAPI *GetConsoleCommandHistoryLength_t)(LPCTSTR sExeName); + + GetConsoleCommandHistory_t m_pfnGetConsoleCommandHistory; + GetConsoleCommandHistoryLength_t m_pfnGetConsoleCommandHistoryLength; + + DECLARE_NO_COPY_CLASS(wxConsoleStderr) +}; + +bool wxConsoleStderr::DoInit() +{ + HANDLE hStderr = ::GetStdHandle(STD_ERROR_HANDLE); + + if ( hStderr == INVALID_HANDLE_VALUE || !hStderr ) + return false; + + if ( !m_dllKernel32.Load(_T("kernel32.dll")) ) + return false; + + typedef BOOL (WINAPI *AttachConsole_t)(DWORD dwProcessId); + AttachConsole_t wxDL_INIT_FUNC(pfn, AttachConsole, m_dllKernel32); + + if ( !pfnAttachConsole || !pfnAttachConsole(ATTACH_PARENT_PROCESS) ) + return false; + + // console attached, set m_hStderr now to ensure that we free it in the + // dtor + m_hStderr = hStderr; + + wxDL_INIT_FUNC_AW(m_pfn, GetConsoleCommandHistory, m_dllKernel32); + if ( !m_pfnGetConsoleCommandHistory ) + return false; + + wxDL_INIT_FUNC_AW(m_pfn, GetConsoleCommandHistoryLength, m_dllKernel32); + if ( !m_pfnGetConsoleCommandHistoryLength ) + return false; + + // remember the current command history to be able to compare with it later + // in IsHistoryUnchanged() + m_historyLen = GetCommandHistory(m_history); + if ( !m_history ) + return false; + + + // now find the first blank line above the current position + CONSOLE_SCREEN_BUFFER_INFO csbi; + + if ( !::GetConsoleScreenBufferInfo(m_hStderr, &csbi) ) + { + wxLogLastError(_T("GetConsoleScreenBufferInfo")); + return false; + } + + COORD pos; + pos.X = 0; + pos.Y = csbi.dwCursorPosition.Y + 1; + + // we decide that a line is empty if first 4 characters are spaces + DWORD ret; + char buf[4]; + do + { + pos.Y--; + if ( !::ReadConsoleOutputCharacterA(m_hStderr, buf, WXSIZEOF(buf), + pos, &ret) ) + { + wxLogLastError(_T("ReadConsoleOutputCharacterA")); + return false; + } + } while ( wxStrncmp(" ", buf, WXSIZEOF(buf)) != 0 ); + + // calculate line offset and length of data + m_dataLine = csbi.dwCursorPosition.Y - pos.Y; + m_dataLen = m_dataLine*csbi.dwMaximumWindowSize.X + csbi.dwCursorPosition.X; + + if ( m_dataLen > 0 ) + { + m_data.extend(m_dataLen); + if ( !::ReadConsoleOutputCharacterA(m_hStderr, m_data.data(), m_dataLen, + pos, &ret) ) + { + wxLogLastError(_T("ReadConsoleOutputCharacterA")); + return false; + } + } + + return true; +} + +int wxConsoleStderr::GetCommandHistory(wxWxCharBuffer& buf) const +{ + // these functions are internal and may only be called by cmd.exe + static const wxChar *CMD_EXE = _T("cmd.exe"); + + const int len = m_pfnGetConsoleCommandHistoryLength(CMD_EXE); + if ( len ) + { + buf.extend(len); + + int len2 = m_pfnGetConsoleCommandHistory(buf.data(), len, CMD_EXE); + +#if !wxUSE_UNICODE + // there seems to be a bug in the GetConsoleCommandHistoryA(), it + // returns the length of Unicode string and not ANSI one + len2 /= 2; +#endif // !wxUSE_UNICODE + + if ( len2 != len ) + { + wxFAIL_MSG( _T("failed getting history?") ); + } + } + + return len; +} + +bool wxConsoleStderr::IsHistoryUnchanged() const +{ + wxASSERT_MSG( m_ok == 1, _T("shouldn't be called if not initialized") ); + + // get (possibly changed) command history + wxWxCharBuffer history; + const int historyLen = GetCommandHistory(history); + + // and compare it with the original one + return historyLen == m_historyLen && history && + memcmp(m_history, history, historyLen) == 0; +} + +bool wxConsoleStderr::Write(const wxString& text) +{ + wxASSERT_MSG( m_hStderr != INVALID_HANDLE_VALUE, + _T("should only be called if Init() returned true") ); + + // get current position + CONSOLE_SCREEN_BUFFER_INFO csbi; + if ( !::GetConsoleScreenBufferInfo(m_hStderr, &csbi) ) + { + wxLogLastError(_T("GetConsoleScreenBufferInfo")); + return false; + } + + // and calculate new position (where is empty line) + csbi.dwCursorPosition.X = 0; + csbi.dwCursorPosition.Y -= m_dataLine; + + if ( !::SetConsoleCursorPosition(m_hStderr, csbi.dwCursorPosition) ) + { + wxLogLastError(_T("SetConsoleCursorPosition")); + return false; + } + + DWORD ret; + if ( !::FillConsoleOutputCharacter(m_hStderr, _T(' '), m_dataLen, + csbi.dwCursorPosition, &ret) ) + { + wxLogLastError(_T("FillConsoleOutputCharacter")); + return false; + } + + if ( !::WriteConsole(m_hStderr, text.wx_str(), text.length(), &ret, NULL) ) + { + wxLogLastError(_T("WriteConsole")); + return false; + } + + WriteConsoleA(m_hStderr, m_data, m_dataLen, &ret, 0); + + return true; +} + +wxConsoleStderr s_consoleStderr; + +} // anonymous namespace + +bool wxGUIAppTraits::CanUseStderr() +{ + return s_consoleStderr.IsOkToUse(); +} + +bool wxGUIAppTraits::WriteToStderr(const wxString& text) +{ + return s_consoleStderr.IsOkToUse() && s_consoleStderr.Write(text); +} + +#endif // !__WXWINCE__ + // =========================================================================== // wxApp implementation // =========================================================================== @@ -323,20 +616,6 @@ bool wxApp::Initialize(int& argc, wxChar **argv) // ensure that base cleanup is done if we return too early wxCallBaseCleanup callBaseCleanup(this); -#ifdef __WXWINCE__ - wxString tmp = GetAppName(); - tmp += wxT("ClassName"); - wxCanvasClassName = wxStrdup( tmp.wc_str() ); - tmp += wxT("NR"); - wxCanvasClassNameNR = wxStrdup( tmp.wc_str() ); - HWND hWnd = FindWindow( wxCanvasClassNameNR, NULL ); - if (hWnd) - { - SetForegroundWindow( (HWND)(((DWORD)hWnd)|0x01) ); - return false; - } -#endif - #if !defined(__WXMICROWIN__) InitCommonControls(); #endif // !defined(__WXMICROWIN__) @@ -353,8 +632,6 @@ bool wxApp::Initialize(int& argc, wxChar **argv) wxOleInitialize(); - RegisterWindowClasses(); - #if !defined(__WXMICROWIN__) && !defined(__WXWINCE__) wxSetKeyboardHook(true); #endif @@ -365,143 +642,106 @@ bool wxApp::Initialize(int& argc, wxChar **argv) } // --------------------------------------------------------------------------- -// RegisterWindowClasses +// Win32 window class registration // --------------------------------------------------------------------------- -// This function registers the given class name and stores a pointer to a -// heap-allocated copy of it at the specified location, it must be deleted -// later. -static void RegisterAndStoreClassName(const wxString& uniqueClassName, - const wxChar **className, - WNDCLASS *lpWndClass) +/* static */ +const wxChar *wxApp::GetRegisteredClassName(const wxChar *name, + int bgBrushCol, + int extraStyles) { - const size_t length = uniqueClassName.length() + 1; // for trailing NUL - wxChar *newChars = new wxChar[length]; - wxStrncpy(newChars, uniqueClassName, length); - *className = newChars; - lpWndClass->lpszClassName = *className; - - if ( !::RegisterClass(lpWndClass) ) + const size_t count = gs_regClassesInfo.size(); + for ( size_t n = 0; n < count; n++ ) { - wxLogLastError(wxString::Format(wxT("RegisterClass(%s)"), newChars)); + if ( gs_regClassesInfo[n].basename == name ) + return gs_regClassesInfo[n].regname.c_str(); } -} - -// This function registers the class defined by the provided WNDCLASS struct -// contents using a unique name constructed from the specified base name and -// and a suffix unique to this library instance. It also stores the generated -// unique names for normal and "no redraw" versions of the class in the -// provided variables, caller must delete their contents later. -static void RegisterClassWithUniqueNames(const wxString& baseName, - const wxChar **className, - const wxChar **classNameNR, - WNDCLASS *lpWndClass) -{ - // for each class we register one with CS_(V|H)REDRAW style and one - // without for windows created with wxNO_FULL_REDRAW_ON_REPAINT flag - static const long styleNormal = CS_HREDRAW | CS_VREDRAW | CS_DBLCLKS; - static const long styleNoRedraw = CS_DBLCLKS; - - const wxString uniqueSuffix(wxString::Format(wxT("@%p"), className)); - - wxString uniqueClassName(baseName + uniqueSuffix); - lpWndClass->style = styleNormal; - RegisterAndStoreClassName(uniqueClassName, className, lpWndClass); - // NB: remember that code elsewhere supposes that no redraw class names - // use the same names as normal classes with "NR" suffix so we must put - // "NR" at the end instead of using more natural baseName+"NR"+suffix - wxString uniqueClassNameNR(uniqueClassName + wxT("NR")); - lpWndClass->style = styleNoRedraw; - RegisterAndStoreClassName(uniqueClassNameNR, classNameNR, lpWndClass); -} - -// TODO we should only register classes really used by the app. For this it -// would be enough to just delay the class registration until an attempt -// to create a window of this class is made. -bool wxApp::RegisterWindowClasses() -{ + // we need to register this class WNDCLASS wndclass; wxZeroMemory(wndclass); - // the fields which are common to all classes wndclass.lpfnWndProc = (WNDPROC)wxWndProc; wndclass.hInstance = wxhInstance; - wndclass.hCursor = ::LoadCursor((HINSTANCE)NULL, IDC_ARROW); - - // register the class for all normal windows and "no redraw" frames - wndclass.hbrBackground = (HBRUSH)(COLOR_BTNFACE + 1); - RegisterClassWithUniqueNames(wxT("wxWindowClass"), - &wxCanvasClassName, - &wxCanvasClassNameNR, - &wndclass); - - // Register the MDI frame window class and "no redraw" MDI frame - wndclass.hbrBackground = (HBRUSH)NULL; // paint MDI frame ourselves - RegisterClassWithUniqueNames(wxT("wxMDIFrameClass"), - &wxMDIFrameClassName, - &wxMDIFrameClassNameNoRedraw, - &wndclass); - - // Register the MDI child frame window class and "no redraw" MDI child frame - wndclass.hbrBackground = (HBRUSH)(COLOR_WINDOW + 1); - RegisterClassWithUniqueNames(wxT("wxMDIChildFrameClass"), - &wxMDIChildFrameClassName, - &wxMDIChildFrameClassNameNoRedraw, - &wndclass); - - return true; -} - -// --------------------------------------------------------------------------- -// UnregisterWindowClasses -// --------------------------------------------------------------------------- - -// This function unregisters the class with the given name and frees memory -// allocated for it by RegisterAndStoreClassName(). -static bool UnregisterAndFreeClassName(const wxChar **ppClassName) -{ - bool retval = true; - - if ( !::UnregisterClass(*ppClassName, wxhInstance) ) + wndclass.hCursor = ::LoadCursor(NULL, IDC_ARROW); + wndclass.hbrBackground = (HBRUSH)wxUIntToPtr(bgBrushCol + 1); + wndclass.style = CS_HREDRAW | CS_VREDRAW | CS_DBLCLKS | extraStyles; + + + ClassRegInfo regClass; + regClass.basename = name; + + // constuct a unique suffix to allow registering the class with the same + // base name in a main application using wxWidgets and a DLL using + // wxWidgets loaded into its address space: as gs_regClassesInfo variable + // is different in them, we're going to obtain a unique prefix by using its + // address here + regClass.regname = regClass.basename + + wxString::Format(wxT("@%p"), &gs_regClassesInfo); + wndclass.lpszClassName = regClass.regname.wx_str(); + if ( !::RegisterClass(&wndclass) ) { - wxLogLastError( - wxString::Format(wxT("UnregisterClass(%s)"), *ppClassName)); + wxLogLastError(wxString::Format(wxT("RegisterClass(%s)"), + regClass.regname)); + return NULL; + } - retval = false; + // NB: remember that code elsewhere supposes that no redraw class names + // use the same names as normal classes with "NR" suffix so we must put + // "NR" at the end instead of using more natural basename+"NR"+suffix + regClass.regnameNR = regClass.regname + GetNoRedrawClassSuffix(); + wndclass.style &= ~(CS_HREDRAW | CS_VREDRAW); + wndclass.lpszClassName = regClass.regnameNR.wx_str(); + if ( !::RegisterClass(&wndclass) ) + { + wxLogLastError(wxString::Format(wxT("RegisterClass(%s)"), + regClass.regname)); + ::UnregisterClass(regClass.regname.c_str(), wxhInstance); + return NULL; } - delete [] (wxChar*) *ppClassName; - *ppClassName = NULL; + gs_regClassesInfo.push_back(regClass); - return retval; + // take care to return the pointer which will remain valid after the + // function returns (it could be invalidated later if new elements are + // added to the vector and it's reallocated but this shouldn't matter as + // this pointer should be used right now, not stored) + return gs_regClassesInfo.back().regname.wx_str(); } -bool wxApp::UnregisterWindowClasses() +bool wxApp::IsRegisteredClassName(const wxString& name) { - bool retval = true; - -#ifndef __WXMICROWIN__ - if ( !UnregisterAndFreeClassName(&wxMDIFrameClassName) ) - retval = false; - - if ( !UnregisterAndFreeClassName(&wxMDIFrameClassNameNoRedraw) ) - retval = false; - - if ( !UnregisterAndFreeClassName(&wxMDIChildFrameClassName) ) - retval = false; + const size_t count = gs_regClassesInfo.size(); + for ( size_t n = 0; n < count; n++ ) + { + if ( gs_regClassesInfo[n].regname == name || + gs_regClassesInfo[n].regnameNR == name ) + return true; + } - if ( !UnregisterAndFreeClassName(&wxMDIChildFrameClassNameNoRedraw) ) - retval = false; + return false; +} - if ( !UnregisterAndFreeClassName(&wxCanvasClassName) ) - retval = false; +void wxApp::UnregisterWindowClasses() +{ + const size_t count = gs_regClassesInfo.size(); + for ( size_t n = 0; n < count; n++ ) + { + const ClassRegInfo& regClass = gs_regClassesInfo[n]; + if ( !::UnregisterClass(regClass.regname.c_str(), wxhInstance) ) + { + wxLogLastError(wxString::Format(wxT("UnregisterClass(%s)"), + regClass.regname)); + } - if ( !UnregisterAndFreeClassName(&wxCanvasClassNameNR) ) - retval = false; -#endif // __WXMICROWIN__ + if ( !::UnregisterClass(regClass.regnameNR.c_str(), wxhInstance) ) + { + wxLogLastError(wxString::Format(wxT("UnregisterClass(%s)"), + regClass.regnameNR)); + } + } - return retval; + gs_regClassesInfo.clear(); } void wxApp::CleanUp() @@ -523,11 +763,6 @@ void wxApp::CleanUp() // which case the registration will fail after the first time if we don't // unregister the classes now UnregisterWindowClasses(); - -#ifdef __WXWINCE__ - free( wxCanvasClassName ); - free( wxCanvasClassNameNR ); -#endif } // ---------------------------------------------------------------------------- @@ -678,8 +913,11 @@ int wxApp::GetComCtl32Version() // we're prepared to handle the errors wxLogNull noLog; - // the DLL should really be available - wxDynamicLibrary dllComCtl32(_T("comctl32.dll"), wxDL_VERBATIM); + // we don't want to load comctl32.dll, it should be already loaded but, + // depending on the OS version and the presence of the manifest, it can + // be either v5 or v6 and instead of trying to guess it just get the + // handle of the already loaded version + wxLoadedDLL dllComCtl32(_T("comctl32.dll")); if ( !dllComCtl32.IsLoaded() ) { s_verComCtl32 = 0; @@ -780,12 +1018,6 @@ bool wxApp::Yield(bool onlyIfNeeded) // MT-FIXME static bool s_inYield = false; -#if wxUSE_LOG - // disable log flushing from here because a call to wxYield() shouldn't - // normally result in message boxes popping up &c - wxLog::Suspend(); -#endif // wxUSE_LOG - if ( s_inYield ) { if ( !onlyIfNeeded ) @@ -796,7 +1028,20 @@ bool wxApp::Yield(bool onlyIfNeeded) return false; } + // set the flag and don't forget to reset it before returning s_inYield = true; + wxON_BLOCK_EXIT_SET(s_inYield, false); + + +#if wxUSE_LOG + // disable log flushing from here because a call to wxYield() shouldn't + // normally result in message boxes popping up &c + wxLog::Suspend(); + + // ensure the logs will be flashed again when we exit + wxON_BLOCK_EXIT0(wxLog::Resume); +#endif // wxUSE_LOG + // we don't want to process WM_QUIT from here - it should be processed in // the main event loop in order to stop it @@ -816,13 +1061,6 @@ bool wxApp::Yield(bool onlyIfNeeded) // if there are pending events, we must process them. ProcessPendingEvents(); -#if wxUSE_LOG - // let the logs be flashed again - wxLog::Resume(); -#endif // wxUSE_LOG - - s_inYield = false; - return true; }