X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/6f20c5486114167def8c5ce0f9144c222f619f9c..9b69526274b023fa1460b29a92bea8bf82e4703f:/src/common/containr.cpp?ds=sidebyside

diff --git a/src/common/containr.cpp b/src/common/containr.cpp
index 57c9361752..5ad3a7f006 100644
--- a/src/common/containr.cpp
+++ b/src/common/containr.cpp
@@ -50,18 +50,21 @@ wxControlContainer::wxControlContainer(wxWindow *winParent)
 
 void wxControlContainer::SetLastFocus(wxWindow *win)
 {
-    // find the last _immediate_ child which got focus
-    while ( win )
+    // if we're setting the focus
+    if ( win )
     {
-        wxWindow *winParent = win->GetParent();
-        if ( winParent == m_winParent )
-            break;
+        // find the last _immediate_ child which got focus but be prepared to
+        // handle the case when win == m_winParent as well
+        wxWindow *winParent = win;
+        while ( winParent != m_winParent )
+        {
+            win = winParent;
+            winParent = win->GetParent();
+        }
 
-        win = winParent;
+        wxASSERT_MSG( win, _T("attempt to set last focus to not a child?") );
     }
 
-    wxASSERT_MSG( win, _T("attempt to set last focus to not a child?") );
-
     m_winLastFocused = win;
 }