src/WINNT/afsd/cm_daemon.c

   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 <afs/param.h>
  11 #include <afs/stds.h>
  12
  13 #include <windows.h>
  14 #include <winsock2.h>
  15 #include <iphlpapi.h>
  16 #include <stdlib.h>
  17 #include <malloc.h>
  18 #include <string.h>
  19
  20 #include "afsd.h"
  21
  22 #include <rx/rx.h>
  23 #include <rx/rx_prototypes.h>
  24 #include <WINNT/afsreg.h>
  25
  26 #include "afsicf.h"
  27
  28 /* in seconds */
  29 long cm_daemonCheckDownInterval  = 180;
  30 long cm_daemonCheckUpInterval    = 600;
  31 long cm_daemonCheckVolInterval   = 3600;
  32 long cm_daemonCheckCBInterval    = 60;
  33 long cm_daemonCheckLockInterval  = 60;
  34 long cm_daemonTokenCheckInterval = 180;
  35
  36 osi_rwlock_t cm_daemonLock;
  37
  38 long cm_bkgQueueCount;          /* # of queued requests */
  39
  40 int cm_bkgWaitingForCount;      /* true if someone's waiting for cm_bkgQueueCount to drop */
  41
  42 cm_bkgRequest_t *cm_bkgListp;           /* first elt in the list of requests */
  43 cm_bkgRequest_t *cm_bkgListEndp;        /* last elt in the list of requests */
  44
  45 static int daemon_ShutdownFlag = 0;
  46
  47 void cm_IpAddrDaemon(long parm)
  48 {
  49     extern void smb_CheckVCs(void);
  50
  51     rx_StartClientThread();
  52
  53     while (daemon_ShutdownFlag == 0) {
  54         DWORD Result = NotifyAddrChange(NULL,NULL);
  55         if (Result == NO_ERROR && daemon_ShutdownFlag == 0) {
  56             osi_Log0(afsd_logp, "cm_IpAddrDaemon CheckDownServers");
  57             Sleep(2500);
  58             cm_ForceNewConnectionsAllServers();
  59             cm_CheckServers(CM_FLAG_CHECKUPSERVERS | CM_FLAG_CHECKDOWNSERVERS, NULL);
  60             smb_CheckVCs();
  61         }
  62     }
  63 }
  64
  65 void cm_BkgDaemon(long parm)
  66 {
  67     cm_bkgRequest_t *rp;
  68
  69     rx_StartClientThread();
  70
  71     lock_ObtainWrite(&cm_daemonLock);
  72     while (daemon_ShutdownFlag == 0) {
  73         if (!cm_bkgListEndp) {
  74             osi_SleepW((LONG_PTR)&cm_bkgListp, &cm_daemonLock);
  75             lock_ObtainWrite(&cm_daemonLock);
  76             continue;
  77         }
  78
  79         /* we found a request */
  80         rp = cm_bkgListEndp;
  81         cm_bkgListEndp = (cm_bkgRequest_t *) osi_QPrev(&rp->q);
  82         osi_QRemove((osi_queue_t **) &cm_bkgListp, &rp->q);
  83         osi_assert(cm_bkgQueueCount-- > 0);
  84         lock_ReleaseWrite(&cm_daemonLock);
  85
  86 #ifdef DEBUG_REFCOUNT
  87         osi_Log2(afsd_logp,"cm_BkgDaemon (before) scp 0x%x ref %d",rp->scp, rp->scp->refCount);
  88 #endif
  89         (*rp->procp)(rp->scp, rp->p1, rp->p2, rp->p3, rp->p4, rp->userp);
  90 #ifdef DEBUG_REFCOUNT
  91         osi_Log2(afsd_logp,"cm_BkgDaemon (after) scp 0x%x ref %d",rp->scp, rp->scp->refCount);
  92 #endif
  93         cm_ReleaseUser(rp->userp);
  94         cm_ReleaseSCache(rp->scp);
  95         free(rp);
  96
  97         lock_ObtainWrite(&cm_daemonLock);
  98     }
  99     lock_ReleaseWrite(&cm_daemonLock);
 100 }
 101
 102 void cm_QueueBKGRequest(cm_scache_t *scp, cm_bkgProc_t *procp, afs_uint32 p1, afs_uint32 p2, afs_uint32 p3, afs_uint32 p4,
 103         cm_user_t *userp)
 104 {
 105     cm_bkgRequest_t *rp;
 106
 107     rp = malloc(sizeof(*rp));
 108     memset(rp, 0, sizeof(*rp));
 109
 110     cm_HoldSCache(scp);
 111     rp->scp = scp;
 112     cm_HoldUser(userp);
 113     rp->userp = userp;
 114     rp->procp = procp;
 115     rp->p1 = p1;
 116     rp->p2 = p2;
 117     rp->p3 = p3;
 118     rp->p4 = p4;
 119
 120     lock_ObtainWrite(&cm_daemonLock);
 121     cm_bkgQueueCount++;
 122     osi_QAdd((osi_queue_t **) &cm_bkgListp, &rp->q);
 123     if (!cm_bkgListEndp)
 124         cm_bkgListEndp = rp;
 125     lock_ReleaseWrite(&cm_daemonLock);
 126
 127     osi_Wakeup((LONG_PTR) &cm_bkgListp);
 128 }
 129
 130 static int
 131 IsWindowsFirewallPresent(void)
 132 {
 133     SC_HANDLE scm;
 134     SC_HANDLE svc;
 135     BOOLEAN flag;
 136     BOOLEAN result = FALSE;
 137     LPQUERY_SERVICE_CONFIG pConfig = NULL;
 138     DWORD BufSize;
 139     LONG status;
 140
 141     /* Open services manager */
 142     scm = OpenSCManager(NULL, NULL, GENERIC_READ);
 143     if (!scm) return FALSE;
 144
 145     /* Open Windows Firewall service */
 146     svc = OpenService(scm, "MpsSvc", SERVICE_QUERY_CONFIG);
 147     if (!svc) {
 148         afsi_log("MpsSvc Service could not be opened for query: 0x%x", GetLastError());
 149         svc = OpenService(scm, "SharedAccess", SERVICE_QUERY_CONFIG);
 150         if (!svc)
 151             afsi_log("SharedAccess Service could not be opened for query: 0x%x", GetLastError());
 152     }
 153     if (!svc)
 154         goto close_scm;
 155
 156     /* Query Windows Firewall service config, first just to get buffer size */
 157     /* Expected to fail, so don't test return value */
 158     (void) QueryServiceConfig(svc, NULL, 0, &BufSize);
 159     status = GetLastError();
 160     if (status != ERROR_INSUFFICIENT_BUFFER)
 161         goto close_svc;
 162
 163     /* Allocate buffer */
 164     pConfig = (LPQUERY_SERVICE_CONFIG)GlobalAlloc(GMEM_FIXED,BufSize);
 165     if (!pConfig)
 166         goto close_svc;
 167
 168     /* Query Windows Firewall service config, this time for real */
 169     flag = QueryServiceConfig(svc, pConfig, BufSize, &BufSize);
 170     if (!flag) {
 171         afsi_log("QueryServiceConfig failed: 0x%x", GetLastError());
 172         goto free_pConfig;
 173     }
 174
 175     /* Is it autostart? */
 176     afsi_log("AutoStart 0x%x", pConfig->dwStartType);
 177     if (pConfig->dwStartType < SERVICE_DEMAND_START)
 178         result = TRUE;
 179
 180   free_pConfig:
 181     GlobalFree(pConfig);
 182   close_svc:
 183     CloseServiceHandle(svc);
 184   close_scm:
 185     CloseServiceHandle(scm);
 186
 187     return result;
 188 }
 189
 190 void
 191 cm_DaemonCheckInit(void)
 192 {
 193     HKEY parmKey;
 194     DWORD dummyLen;
 195     DWORD dummy;
 196     DWORD code;
 197
 198     code = RegOpenKeyEx(HKEY_LOCAL_MACHINE, AFSREG_CLT_SVC_PARAM_SUBKEY,
 199                          0, KEY_QUERY_VALUE, &parmKey);
 200     if (code)
 201         return;
 202
 203     dummyLen = sizeof(DWORD);
 204     code = RegQueryValueEx(parmKey, "DownServerCheckInterval", NULL, NULL,
 205                             (BYTE *) &dummy, &dummyLen);
 206     if (code == ERROR_SUCCESS)
 207         cm_daemonCheckDownInterval = dummy;
 208
 209     dummyLen = sizeof(DWORD);
 210     code = RegQueryValueEx(parmKey, "UpServerCheckInterval", NULL, NULL,
 211                             (BYTE *) &dummy, &dummyLen);
 212     if (code == ERROR_SUCCESS)
 213         cm_daemonCheckUpInterval = dummy;
 214
 215     dummyLen = sizeof(DWORD);
 216     code = RegQueryValueEx(parmKey, "VolumeCheckInterval", NULL, NULL,
 217                             (BYTE *) &dummy, &dummyLen);
 218     if (code == ERROR_SUCCESS)
 219         cm_daemonCheckVolInterval = dummy;
 220
 221     dummyLen = sizeof(DWORD);
 222     code = RegQueryValueEx(parmKey, "CallbackCheckInterval", NULL, NULL,
 223                             (BYTE *) &dummy, &dummyLen);
 224     if (code == ERROR_SUCCESS)
 225         cm_daemonCheckCBInterval = dummy;
 226
 227     dummyLen = sizeof(DWORD);
 228     code = RegQueryValueEx(parmKey, "LockCheckInterval", NULL, NULL,
 229                             (BYTE *) &dummy, &dummyLen);
 230     if (code == ERROR_SUCCESS)
 231         cm_daemonCheckLockInterval = dummy;
 232
 233     dummyLen = sizeof(DWORD);
 234     code = RegQueryValueEx(parmKey, "TokenCheckInterval", NULL, NULL,
 235                             (BYTE *) &dummy, &dummyLen);
 236     if (code == ERROR_SUCCESS)
 237         cm_daemonTokenCheckInterval = dummy;
 238
 239     RegCloseKey(parmKey);
 240 }
 241
 242 /* periodic check daemon */
 243 void cm_Daemon(long parm)
 244 {
 245     time_t now;
 246     time_t lastLockCheck;
 247     time_t lastVolCheck;
 248     time_t lastCBExpirationCheck;
 249     time_t lastDownServerCheck;
 250     time_t lastUpServerCheck;
 251     time_t lastTokenCacheCheck;
 252     char thostName[200];
 253     unsigned long code;
 254     struct hostent *thp;
 255     HMODULE hHookDll;
 256     int configureFirewall = IsWindowsFirewallPresent();
 257
 258     if (!configureFirewall) {
 259         afsi_log("No Windows Firewall detected");
 260     }
 261
 262     /* ping all file servers, up or down, with unauthenticated connection,
 263      * to find out whether we have all our callbacks from the server still.
 264      * Also, ping down VLDBs.
 265      */
 266     /*
 267      * Seed the random number generator with our own address, so that
 268      * clients starting at the same time don't all do vol checks at the
 269      * same time.
 270      */
 271     gethostname(thostName, sizeof(thostName));
 272     thp = gethostbyname(thostName);
 273     if (thp == NULL)    /* In djgpp, gethostname returns the netbios
 274                            name of the machine.  gethostbyname will fail
 275                            looking this up if it differs from DNS name. */
 276         code = 0;
 277     else
 278         memcpy(&code, thp->h_addr_list[0], 4);
 279
 280     srand(ntohl(code));
 281
 282     cm_DaemonCheckInit();
 283
 284     now = osi_Time();
 285     lastVolCheck = now - cm_daemonCheckVolInterval/2 + (rand() % cm_daemonCheckVolInterval);
 286     lastCBExpirationCheck = now - cm_daemonCheckCBInterval/2 + (rand() % cm_daemonCheckCBInterval);
 287     lastLockCheck = now - cm_daemonCheckLockInterval/2 + (rand() % cm_daemonCheckLockInterval);
 288     lastDownServerCheck = now - cm_daemonCheckDownInterval/2 + (rand() % cm_daemonCheckDownInterval);
 289     lastUpServerCheck = now - cm_daemonCheckUpInterval/2 + (rand() % cm_daemonCheckUpInterval);
 290     lastTokenCacheCheck = now - cm_daemonTokenCheckInterval/2 + (rand() % cm_daemonTokenCheckInterval);
 291
 292     while (daemon_ShutdownFlag == 0) {
 293         if (configureFirewall) {
 294             /* Open Microsoft Firewall to allow in port 7001 */
 295             switch (icf_CheckAndAddAFSPorts(AFS_PORTSET_CLIENT)) {
 296             case 0:
 297                 afsi_log("Windows Firewall Configuration succeeded");
 298                 configureFirewall = 0;
 299                 break;
 300             case 1:
 301                 afsi_log("Invalid Windows Firewall Port Set");
 302                 break;
 303             case 2:
 304                 afsi_log("Unable to open Windows Firewall Profile");
 305                 break;
 306             case 3:
 307                 afsi_log("Unable to create/modify Windows Firewall Port entries");
 308                 break;
 309             default:
 310                 afsi_log("Unknown Windows Firewall Configuration error");
 311             }
 312         }
 313
 314         /* find out what time it is */
 315         now = osi_Time();
 316
 317         /* check down servers */
 318         if (now > lastDownServerCheck + cm_daemonCheckDownInterval) {
 319             lastDownServerCheck = now;
 320             osi_Log0(afsd_logp, "cm_Daemon CheckDownServers");
 321             cm_CheckServers(CM_FLAG_CHECKDOWNSERVERS, NULL);
 322             now = osi_Time();
 323         }
 324
 325         /* check up servers */
 326         if (now > lastUpServerCheck + cm_daemonCheckUpInterval) {
 327             lastUpServerCheck = now;
 328             osi_Log0(afsd_logp, "cm_Daemon CheckUpServers");
 329             cm_CheckServers(CM_FLAG_CHECKUPSERVERS, NULL);
 330             now = osi_Time();
 331         }
 332
 333         if (now > lastVolCheck + cm_daemonCheckVolInterval) {
 334             lastVolCheck = now;
 335             cm_CheckVolumes();
 336             now = osi_Time();
 337         }
 338
 339         if (now > lastCBExpirationCheck + cm_daemonCheckCBInterval) {
 340             lastCBExpirationCheck = now;
 341             cm_CheckCBExpiration();
 342             now = osi_Time();
 343         }
 344
 345         if (now > lastLockCheck + cm_daemonCheckLockInterval) {
 346             lastLockCheck = now;
 347             cm_CheckLocks();
 348             now = osi_Time();
 349         }
 350
 351         if (now > lastTokenCacheCheck + cm_daemonTokenCheckInterval) {
 352             lastTokenCacheCheck = now;
 353             cm_CheckTokenCache(now);
 354             now = osi_Time();
 355         }
 356
 357         /* allow an exit to be called prior to stopping the service */
 358         hHookDll = LoadLibrary(AFSD_HOOK_DLL);
 359         if (hHookDll)
 360         {
 361             BOOL hookRc = TRUE;
 362             AfsdDaemonHook daemonHook = ( AfsdDaemonHook ) GetProcAddress(hHookDll, AFSD_DAEMON_HOOK);
 363             if (daemonHook)
 364             {
 365                 hookRc = daemonHook();
 366             }
 367             FreeLibrary(hHookDll);
 368             hHookDll = NULL;
 369
 370             if (hookRc == FALSE)
 371             {
 372                 SetEvent(WaitToTerminate);
 373             }
 374         }
 375
 376         thrd_Sleep(30 * 1000);          /* sleep 30 seconds */
 377         if (daemon_ShutdownFlag == 1)
 378             return;
 379     }
 380 }
 381
 382 void cm_DaemonShutdown(void)
 383 {
 384     daemon_ShutdownFlag = 1;
 385 }
 386
 387 void cm_InitDaemon(int nDaemons)
 388 {
 389     static osi_once_t once;
 390     long pid;
 391     thread_t phandle;
 392     int i;
 393
 394     if (osi_Once(&once)) {
 395         lock_InitializeRWLock(&cm_daemonLock, "cm_daemonLock");
 396         osi_EndOnce(&once);
 397
 398         /* creating IP Address Change monitor daemon */
 399         phandle = thrd_Create((SecurityAttrib) 0, 0,
 400                                (ThreadFunc) cm_IpAddrDaemon, 0, 0, &pid, "cm_IpAddrDaemon");
 401         osi_assert(phandle != NULL);
 402         thrd_CloseHandle(phandle);
 403
 404         /* creating pinging daemon */
 405         phandle = thrd_Create((SecurityAttrib) 0, 0,
 406                                (ThreadFunc) cm_Daemon, 0, 0, &pid, "cm_Daemon");
 407         osi_assert(phandle != NULL);
 408         thrd_CloseHandle(phandle);
 409
 410         for(i=0; i < nDaemons; i++) {
 411             phandle = thrd_Create((SecurityAttrib) 0, 0,
 412                                    (ThreadFunc) cm_BkgDaemon, 0, 0, &pid,
 413                                    "cm_BkgDaemon");
 414             osi_assert(phandle != NULL);
 415             thrd_CloseHandle(phandle);
 416         }
 417     }
 418 }