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