src/WINNT/afsapplib/al_pump.cpp

   1 /*
   2  * Copyright 2000, International Business Machines Corporation and others.
   3  * All Rights Reserved.
   4  *
   5  * This software has been released under the terms of the IBM Public
   6  * License.  For details, see the LICENSE file in the top-level source
   7  * directory or online at http://www.openafs.org/dl/license10.html
   8  */
   9
  10 #include <winsock2.h>
  11 #include <ws2tcpip.h>
  12
  13 extern "C" {
  14 #include <afs/param.h>
  15 #include <afs/stds.h>
  16 }
  17
  18 #include <WINNT/afsapplib.h>
  19
  20
  21 /*
  22  * DEFINITIONS ________________________________________________________________
  23  *
  24  */
  25
  26 #define cREALLOC_MODELESS 16
  27 #define cREALLOC_WINDOWDATA_FIELDS 4
  28 #define cREALLOC_WINDOWDATA_WINDOWS 16
  29 #define cREALLOC_WINDOWDATA_DATA 8
  30
  31 #define GWD_IS_MODELESS  TEXT("afsapplib/al_pump.cpp - is window modeless?")
  32
  33
  34 /*
  35  * VARIABLES __________________________________________________________________
  36  *
  37  */
  38
  39          // Modeless-dialog support
  40          //
  41 static HWND  *aModeless = NULL;
  42 static size_t cModeless = 0;
  43 static LPCRITICAL_SECTION pcsPump = NULL;
  44
  45 static void (*g_fnPump)(MSG *lpm) = NULL;
  46
  47          // Window-data support
  48          //
  49 static struct
  50    {
  51    TCHAR szField[ cchNAME ];
  52    } *aFields = NULL;
  53 static size_t cFields = 0;
  54
  55 static struct
  56    {
  57    HWND hWnd;
  58    UINT_PTR *adwData;
  59    size_t cdwData;
  60    } *aWindows = NULL;
  61 static size_t cWindows = 0;
  62 static LPCRITICAL_SECTION pcsData = NULL;
  63
  64
  65 /*
  66  * DIALOG ROUTINES ____________________________________________________________
  67  *
  68  */
  69
  70 HRESULT CALLBACK Modeless_HookProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp)
  71 {
  72    PVOID oldProc = Subclass_FindNextHook (hDlg, Modeless_HookProc);
  73
  74    if (msg == WM_DESTROY)
  75       {
  76       EnterCriticalSection (pcsPump);
  77       for (size_t ii = 0; ii < cModeless; ++ii)
  78          {
  79          if (aModeless[ ii ] == hDlg)
  80             aModeless[ ii ] = 0;
  81          }
  82       LeaveCriticalSection (pcsPump);
  83       Subclass_RemoveHook (hDlg, Modeless_HookProc);
  84       }
  85
  86    if (oldProc)
  87       return CallWindowProc ((WNDPROC)oldProc, hDlg, msg, wp, lp);
  88    else
  89       return DefWindowProc (hDlg, msg, wp, lp);
  90 }
  91
  92
  93 BOOL AfsAppLib_IsModelessDialogMessage (MSG *lpm)
  94 {
  95    BOOL rc = FALSE;
  96
  97    if (pcsPump)
  98       {
  99       EnterCriticalSection (pcsPump);
 100
 101       for (size_t ii = 0; (!rc) && (ii < cModeless); ++ii)
 102          {
 103          if (aModeless[ ii ] != 0)
 104             {
 105             if (IsPropSheet (aModeless[ ii ]))
 106                {
 107                if (!PropSheet_GetCurrentPageHwnd(aModeless[ ii ]))
 108                   DestroyWindow (aModeless[ ii ]);
 109                if (PropSheet_IsDialogMessage(aModeless[ ii ], lpm))
 110                   rc = TRUE;
 111                }
 112             else
 113                {
 114                if (IsDialogMessage (aModeless[ ii ], lpm))
 115                   rc = TRUE;
 116                }
 117             }
 118          }
 119
 120       LeaveCriticalSection (pcsPump);
 121       }
 122
 123    return rc;
 124 }
 125
 126
 127 void AfsAppLib_RegisterModelessDialog (HWND hDlg)
 128 {
 129    if (pcsPump == NULL)
 130       {
 131       pcsPump = New (CRITICAL_SECTION);
 132       InitializeCriticalSection (pcsPump);
 133       }
 134
 135    EnterCriticalSection (pcsPump);
 136
 137    size_t ii;
 138    for (ii = 0; ii < cModeless; ++ii)
 139       {
 140       if (aModeless[ ii ] == hDlg)
 141          break;
 142       }
 143    if (ii == cModeless)
 144       {
 145       for (ii = 0; ii < cModeless; ++ii)
 146          {
 147          if (aModeless[ ii ] == 0)
 148             break;
 149          }
 150       if (ii == cModeless)
 151          {
 152          (void)REALLOC (aModeless, cModeless, 1+ii, cREALLOC_MODELESS);
 153          }
 154       if (ii < cModeless)
 155          {
 156          aModeless[ ii ] = hDlg;
 157          Subclass_AddHook (hDlg, Modeless_HookProc);
 158          }
 159       }
 160
 161    LeaveCriticalSection (pcsPump);
 162 }
 163
 164
 165 void AfsAppLib_SetPumpRoutine (void (*fnPump)(MSG *lpm))
 166 {
 167    g_fnPump = fnPump;
 168 }
 169
 170
 171 void AfsAppLib_MainPump (void)
 172 {
 173    MSG msg;
 174
 175    while (GetMessage (&msg, NULL, 0, 0))
 176       {
 177       if (AfsAppLib_IsModelessDialogMessage (&msg))
 178          continue;
 179
 180       if (g_fnPump)
 181          (*g_fnPump)(&msg);
 182       else
 183          {
 184          TranslateMessage (&msg);
 185          DispatchMessage (&msg);
 186          }
 187       }
 188 }
 189
 190
 191 /*
 192  * WINDOW-DATA ________________________________________________________________
 193  *
 194  */
 195
 196 HRESULT CALLBACK WindowData_HookProc (HWND hWnd, UINT msg, WPARAM wp, LPARAM lp)
 197 {
 198    PVOID oldProc = Subclass_FindNextHook (hWnd, WindowData_HookProc);
 199
 200    if (msg == WM_DESTROY)
 201       {
 202       EnterCriticalSection (pcsData);
 203       for (size_t ii = 0; ii < cWindows; ++ii)
 204          {
 205          if (aWindows[ ii ].hWnd == hWnd)
 206             {
 207             if (aWindows[ ii ].adwData)
 208                Free (aWindows[ ii ].adwData);
 209             memset (&aWindows[ii], 0x00, sizeof(aWindows[ii]));
 210             }
 211          }
 212       LeaveCriticalSection (pcsData);
 213       Subclass_RemoveHook (hWnd, WindowData_HookProc);
 214       }
 215
 216    if (oldProc)
 217       return CallWindowProc ((WNDPROC)oldProc, hWnd, msg, wp, lp);
 218    else
 219       return DefWindowProc (hWnd, msg, wp, lp);
 220 }
 221
 222
 223 size_t GetWindowDataField (LPTSTR pszField)
 224 {
 225    size_t rc = (size_t)-1;
 226
 227    if (pcsData == NULL)
 228       {
 229       pcsData = New (CRITICAL_SECTION);
 230       InitializeCriticalSection (pcsData);
 231       }
 232    EnterCriticalSection (pcsData);
 233
 234    size_t ii;
 235    for (ii = 0; ii < cFields; ++ii)
 236       {
 237       if (!lstrcmpi (aFields[ ii ].szField, pszField))
 238          break;
 239       }
 240    if (ii == cFields)
 241       {
 242       for (ii = 0; ii < cFields; ++ii)
 243          {
 244          if (aFields[ ii ].szField[0] == TEXT('\0'))
 245             break;
 246          }
 247       if (ii == cFields)
 248          {
 249          (void)REALLOC (aFields, cFields, 1+ii, cREALLOC_WINDOWDATA_FIELDS);
 250          }
 251       if (ii < cFields)
 252          {
 253          lstrcpy (aFields[ ii ].szField, pszField);
 254          }
 255       }
 256    if (ii < cFields)
 257       {
 258       rc = ii;
 259       }
 260
 261    LeaveCriticalSection (pcsData);
 262    return rc;
 263 }
 264
 265
 266 UINT_PTR GetWindowData (HWND hWnd, LPTSTR pszField)
 267 {
 268    UINT_PTR rc = 0;
 269
 270    if (pcsData == NULL)
 271       {
 272       pcsData = New (CRITICAL_SECTION);
 273       InitializeCriticalSection (pcsData);
 274       }
 275    EnterCriticalSection (pcsData);
 276
 277    size_t iField;
 278    if ((iField = GetWindowDataField (pszField)) != (size_t)-1)
 279       {
 280       for (size_t ii = 0; ii < cWindows; ++ii)
 281          {
 282          if (aWindows[ ii ].hWnd == hWnd)
 283             {
 284             if (iField < aWindows[ ii ].cdwData)
 285                rc = aWindows[ ii ].adwData[ iField ];
 286             break;
 287             }
 288          }
 289       }
 290
 291    LeaveCriticalSection (pcsData);
 292    return rc;
 293 }
 294
 295
 296 UINT_PTR SetWindowData (HWND hWnd, LPTSTR pszField, UINT_PTR dwNewData)
 297 {
 298    UINT_PTR rc = 0;
 299
 300    if (pcsData == NULL)
 301       {
 302       pcsData = New (CRITICAL_SECTION);
 303       InitializeCriticalSection (pcsData);
 304       }
 305    EnterCriticalSection (pcsData);
 306
 307    size_t iField;
 308    if ((iField = GetWindowDataField (pszField)) != (size_t)-1)
 309       {
 310       size_t ii;
 311       for (ii = 0; ii < cWindows; ++ii)
 312          {
 313          if (aWindows[ ii ].hWnd == hWnd)
 314             {
 315             if (iField < aWindows[ ii ].cdwData)
 316                rc = aWindows[ ii ].adwData[ iField ];
 317             break;
 318             }
 319          }
 320       if (ii == cWindows)
 321          {
 322          for (ii = 0; ii < cWindows; ++ii)
 323             {
 324             if (aWindows[ ii ].hWnd == 0)
 325                break;
 326             }
 327          if (ii == cWindows)
 328             {
 329             (void)REALLOC (aWindows, cWindows, 1+ii, cREALLOC_WINDOWDATA_WINDOWS);
 330             }
 331          }
 332       if (ii < cWindows)
 333          {
 334          if (aWindows[ ii ].hWnd == 0)
 335             {
 336             aWindows[ ii ].hWnd = hWnd;
 337             Subclass_AddHook (hWnd, WindowData_HookProc);
 338             }
 339          if ((dwNewData) && (iField >= aWindows[ ii ].cdwData))
 340             {
 341             (void)REALLOC (aWindows[ ii ].adwData, aWindows[ ii ].cdwData, 1+iField, cREALLOC_WINDOWDATA_DATA);
 342             }
 343          if (iField < aWindows[ ii ].cdwData)
 344             {
 345             aWindows[ ii ].adwData[ iField ] = dwNewData;
 346             }
 347          }
 348       }
 349
 350    LeaveCriticalSection (pcsData);
 351    return rc;
 352 }
 353