src/WINNT/afsadmsvr/TaAfsAdmSvrClientNotify.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 "TaAfsAdmSvrClientInternal.h"
  19 #include <WINNT/AfsAppLib.h>
  20
  21
  22 /*
  23  * VARIABLES __________________________________________________________________
  24  *
  25  */
  26
  27 typedef struct
  28    {
  29    ASID idCell;
  30    ASID idObject;
  31    HWND hNotify;
  32    } LISTENER, *LPLISTENER;
  33
  34 static struct
  35    {
  36    LPHASHLIST pListeners;
  37    LPHASHLISTKEY pListenersKeyObject;
  38
  39    HWND *ahActionListeners;
  40    size_t chActionListeners;
  41    } l;
  42
  43
  44 /*
  45  * PROTOTYPES _________________________________________________________________
  46  *
  47  */
  48
  49 BOOL CALLBACK ListenersKeyObject_Compare (LPHASHLISTKEY pKey, PVOID pObject, PVOID pData);
  50 HASHVALUE CALLBACK ListenersKeyObject_HashObject (LPHASHLISTKEY pKey, PVOID pObject);
  51 HASHVALUE CALLBACK ListenersKeyObject_HashData (LPHASHLISTKEY pKey, PVOID pData);
  52
  53
  54 /*
  55  * ROUTINES ___________________________________________________________________
  56  *
  57  */
  58
  59 BOOL AddObjectNotification (HWND hNotify, ASID idCell, ASID idObject)
  60 {
  61    asc_Enter();
  62
  63    if (!l.pListeners)
  64       {
  65       l.pListeners = New (HASHLIST);
  66       l.pListenersKeyObject = l.pListeners->CreateKey (TEXT("idObject"), ListenersKeyObject_Compare, ListenersKeyObject_HashObject, ListenersKeyObject_HashData);
  67       }
  68
  69    LPLISTENER pl = New (LISTENER);
  70    pl->idCell = idCell;
  71    pl->idObject = idObject;
  72    pl->hNotify = hNotify;
  73    l.pListeners->Add (pl);
  74
  75    asc_Leave();
  76    return TRUE;
  77 }
  78
  79
  80 void ClearObjectNotifications (HWND hNotify)
  81 {
  82    asc_Enter();
  83
  84    if (l.pListeners)
  85       {
  86       for (LPENUM pEnum = l.pListeners->FindFirst(); pEnum; pEnum = pEnum->FindNext())
  87          {
  88          LPLISTENER pl = (LPLISTENER)( pEnum->GetObject() );
  89          if (pl->hNotify == hNotify)
  90             {
  91             l.pListeners->Remove (pl);
  92             Delete (pl);
  93             }
  94          }
  95       }
  96
  97    asc_Leave();
  98 }
  99
 100
 101 void TestForNotifications (UINT_PTR idClient, ASID idCell, ASID idObject)
 102 {
 103    if (l.pListeners)
 104       {
 105       // First we'll zip through our list of listeners and
 106       // build an ASIDLIST reflecting the objects in this cell
 107       // for which we're listening.
 108       //
 109       LPASIDLIST pAsidList = NULL;
 110       for (LPENUM pEnum = l.pListeners->FindFirst(); pEnum; pEnum = pEnum->FindNext())
 111          {
 112          LPLISTENER pl = (LPLISTENER)( pEnum->GetObject() );
 113          if (pl->idCell != idCell)
 114             continue;
 115          if (idObject && (pl->idObject != idObject))
 116             continue;
 117
 118          if (!pAsidList)
 119             {
 120             if (!asc_AsidListCreate (&pAsidList))
 121                break;
 122             }
 123          if (!asc_AsidListAddEntry (&pAsidList, pl->idObject, 0))
 124             break;
 125          }
 126
 127       // Then we'll call one of our cache routines, which in turn will tell
 128       // the admin server what version of the properties we have for each
 129       // of these objects; if any have newer properties available, we'll
 130       // get them back--and that will cause us to send out notifications to
 131       // our listeners.
 132       //
 133       if (pAsidList)
 134          {
 135          ULONG status;
 136          (void)RefreshCachedProperties (idClient, idCell, pAsidList, GET_ALL_DATA, &status);
 137          }
 138       }
 139 }
 140
 141
 142 void NotifyObjectListeners (ASID idCell, ASID idObject)
 143 {
 144    // If we get here, our cache of information for the specified object
 145    // has just been updated. Check for listeners who may be interested
 146    // in changes to this object.
 147    //
 148    if (l.pListeners)
 149       {
 150       for (LPENUM pEnum = l.pListenersKeyObject->FindFirst (&idObject); pEnum; pEnum = pEnum->FindNext())
 151          {
 152          LPLISTENER pl = (LPLISTENER)( pEnum->GetObject() );
 153          if (pl->idCell != idCell)
 154             continue;
 155
 156          if (!IsWindow (pl->hNotify))
 157             {
 158             l.pListeners->Remove (pl);
 159             Delete (pl);
 160             continue;
 161             }
 162
 163          PostMessage (pl->hNotify, WM_ASC_NOTIFY_OBJECT, (WPARAM)0, (LPARAM)idObject);
 164          }
 165       }
 166 }
 167
 168
 169 /*
 170  * HASHLIST KEYS ______________________________________________________________
 171  *
 172  */
 173
 174 BOOL CALLBACK ListenersKeyObject_Compare (LPHASHLISTKEY pKey, PVOID pObject, PVOID pData)
 175 {
 176    return (((LPLISTENER)pObject)->idObject == *(ASID*)pData) ? TRUE : FALSE;
 177 }
 178
 179 HASHVALUE CALLBACK ListenersKeyObject_HashObject (LPHASHLISTKEY pKey, PVOID pObject)
 180 {
 181    return ListenersKeyObject_HashData (pKey, &((LPLISTENER)pObject)->idObject);
 182 }
 183
 184 HASHVALUE CALLBACK ListenersKeyObject_HashData (LPHASHLISTKEY pKey, PVOID pData)
 185 {
 186    return (HASHVALUE)*(ASID*)pData;
 187 }
 188
 189
 190
 191 /*
 192  * ACTION NOTIFICATIONS _______________________________________________________
 193  *
 194  */
 195
 196 BOOL SetActionNotification (HWND hNotify, BOOL fSet)
 197 {
 198    BOOL rc = TRUE;
 199    size_t ii = 0;
 200
 201    asc_Enter();
 202
 203    if (!fSet)
 204       {
 205       for (ii = 0; ii < l.chActionListeners; ++ii)
 206          {
 207          if (l.ahActionListeners[ ii ] == hNotify)
 208             l.ahActionListeners[ ii ] = NULL;
 209          }
 210       }
 211    else // (fSet)
 212       {
 213       for (ii = 0; ii < l.chActionListeners; ++ii)
 214          {
 215          if (l.ahActionListeners[ ii ] == NULL)
 216             break;
 217          }
 218       if (!REALLOC (l.ahActionListeners, l.chActionListeners, 1+ii, 1))
 219          {
 220          rc = FALSE;
 221          }
 222       else
 223          {
 224          l.ahActionListeners[ ii ] = hNotify;
 225          }
 226       }
 227
 228    asc_Leave();
 229    return rc;
 230 }
 231
 232
 233 void NotifyActionListeners (LPASACTION pAction, BOOL fFinished)
 234 {
 235    asc_Enter();
 236
 237    for (size_t ii = 0; ii < l.chActionListeners; ++ii)
 238       {
 239       if (IsWindow (l.ahActionListeners[ ii ]))
 240          {
 241          LPASACTION pActionPost = New (ASACTION);
 242          memcpy (pActionPost, pAction, sizeof(ASACTION));
 243          PostMessage (l.ahActionListeners[ ii ], WM_ASC_NOTIFY_ACTION, (WPARAM)fFinished, (LPARAM)pActionPost);
 244          }
 245       }
 246
 247    asc_Leave();
 248 }
 249