2 * Copyright 2000, International Business Machines Corporation and others.
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
10 #include <afs/param.h>
23 #include <rx/rx_prototypes.h>
24 #include <WINNT/afsreg.h>
29 long cm_daemonCheckDownInterval = 180;
30 long cm_daemonCheckUpInterval = 240;
31 long cm_daemonCheckVolInterval = 3600;
32 long cm_daemonCheckCBInterval = 60;
33 long cm_daemonCheckVolCBInterval = 0;
34 long cm_daemonCheckLockInterval = 60;
35 long cm_daemonTokenCheckInterval = 180;
36 long cm_daemonCheckOfflineVolInterval = 600;
38 osi_rwlock_t cm_daemonLock;
40 long cm_bkgQueueCount; /* # of queued requests */
42 int cm_bkgWaitingForCount; /* true if someone's waiting for cm_bkgQueueCount to drop */
44 cm_bkgRequest_t *cm_bkgListp; /* first elt in the list of requests */
45 cm_bkgRequest_t *cm_bkgListEndp; /* last elt in the list of requests */
47 static int daemon_ShutdownFlag = 0;
48 static int cm_nDaemons = 0;
49 static time_t lastIPAddrChange = 0;
51 static EVENT_HANDLE cm_Daemon_ShutdownEvent = NULL;
52 static EVENT_HANDLE cm_IPAddrDaemon_ShutdownEvent = NULL;
53 static EVENT_HANDLE cm_BkgDaemon_ShutdownEvent[CM_MAX_DAEMONS] =
54 {NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL};
56 void cm_IpAddrDaemon(long parm)
58 extern void smb_CheckVCs(void);
59 char * name = "cm_IPAddrDaemon_ShutdownEvent";
61 cm_IPAddrDaemon_ShutdownEvent = thrd_CreateEvent(NULL, FALSE, FALSE, name);
62 if ( GetLastError() == ERROR_ALREADY_EXISTS )
63 afsi_log("Event Object Already Exists: %s", name);
65 rx_StartClientThread();
67 while (daemon_ShutdownFlag == 0) {
70 thrd_SetEvent(cm_IPAddrDaemon_ShutdownEvent);
71 Result = NotifyAddrChange(NULL,NULL);
72 if (Result == NO_ERROR && daemon_ShutdownFlag == 0) {
73 lastIPAddrChange = osi_Time();
74 smb_SetLanAdapterChangeDetected();
75 thrd_ResetEvent(cm_IPAddrDaemon_ShutdownEvent);
79 thrd_SetEvent(cm_IPAddrDaemon_ShutdownEvent);
82 void cm_BkgDaemon(void * parm)
87 long daemonID = (long)parm;
89 snprintf(name, sizeof(name), "cm_BkgDaemon_ShutdownEvent%d", daemonID);
91 cm_BkgDaemon_ShutdownEvent[daemonID] = thrd_CreateEvent(NULL, FALSE, FALSE, name);
92 if ( GetLastError() == ERROR_ALREADY_EXISTS )
93 afsi_log("Event Object Already Exists: %s", name);
95 rx_StartClientThread();
97 lock_ObtainWrite(&cm_daemonLock);
98 while (daemon_ShutdownFlag == 0) {
99 if (!cm_bkgListEndp) {
100 osi_SleepW((LONG_PTR)&cm_bkgListp, &cm_daemonLock);
101 lock_ObtainWrite(&cm_daemonLock);
105 /* we found a request */
106 for (rp = cm_bkgListEndp; rp; rp = (cm_bkgRequest_t *) osi_QPrev(&rp->q))
108 if (cm_ServerAvailable(&rp->scp->fid, rp->userp))
112 /* we couldn't find a request that we could process at the current time */
113 lock_ReleaseWrite(&cm_daemonLock);
115 lock_ObtainWrite(&cm_daemonLock);
119 osi_QRemoveHT((osi_queue_t **) &cm_bkgListp, (osi_queue_t **) &cm_bkgListEndp, &rp->q);
120 osi_assertx(cm_bkgQueueCount-- > 0, "cm_bkgQueueCount 0");
121 lock_ReleaseWrite(&cm_daemonLock);
123 osi_Log1(afsd_logp,"cm_BkgDaemon processing request 0x%p", rp);
125 #ifdef DEBUG_REFCOUNT
126 osi_Log2(afsd_logp,"cm_BkgDaemon (before) scp 0x%x ref %d",rp->scp, rp->scp->refCount);
128 code = (*rp->procp)(rp->scp, rp->p1, rp->p2, rp->p3, rp->p4, rp->userp);
129 #ifdef DEBUG_REFCOUNT
130 osi_Log2(afsd_logp,"cm_BkgDaemon (after) scp 0x%x ref %d",rp->scp, rp->scp->refCount);
133 cm_ReleaseUser(rp->userp);
134 cm_ReleaseSCache(rp->scp);
138 lock_ObtainWrite(&cm_daemonLock);
141 case 0: /* success */
142 osi_Log1(afsd_logp,"cm_BkgDaemon SUCCESS: request 0x%p", rp);
144 case CM_ERROR_TIMEDOUT: /* or server restarting */
146 case CM_ERROR_WOULDBLOCK:
147 case CM_ERROR_ALLBUSY:
148 case CM_ERROR_ALLDOWN:
149 case CM_ERROR_ALLOFFLINE:
150 case CM_ERROR_PARTIALWRITE:
151 osi_Log2(afsd_logp,"cm_BkgDaemon re-queueing failed request 0x%p code 0x%x",
154 osi_QAddT((osi_queue_t **) &cm_bkgListp, (osi_queue_t **)&cm_bkgListEndp, &rp->q);
157 osi_Log2(afsd_logp,"cm_BkgDaemon FAILED: request dropped 0x%p code 0x%x",
161 lock_ReleaseWrite(&cm_daemonLock);
163 thrd_SetEvent(cm_BkgDaemon_ShutdownEvent[daemonID]);
167 void cm_QueueBKGRequest(cm_scache_t *scp, cm_bkgProc_t *procp, afs_uint32 p1, afs_uint32 p2, afs_uint32 p3, afs_uint32 p4,
172 rp = malloc(sizeof(*rp));
173 memset(rp, 0, sizeof(*rp));
185 lock_ObtainWrite(&cm_daemonLock);
187 osi_QAdd((osi_queue_t **) &cm_bkgListp, &rp->q);
190 lock_ReleaseWrite(&cm_daemonLock);
192 osi_Wakeup((LONG_PTR) &cm_bkgListp);
196 IsWindowsFirewallPresent(void)
201 BOOLEAN result = FALSE;
202 LPQUERY_SERVICE_CONFIG pConfig = NULL;
206 /* Open services manager */
207 scm = OpenSCManager(NULL, NULL, GENERIC_READ);
208 if (!scm) return FALSE;
210 /* Open Windows Firewall service */
211 svc = OpenService(scm, "MpsSvc", SERVICE_QUERY_CONFIG);
213 afsi_log("MpsSvc Service could not be opened for query: 0x%x", GetLastError());
214 svc = OpenService(scm, "SharedAccess", SERVICE_QUERY_CONFIG);
216 afsi_log("SharedAccess Service could not be opened for query: 0x%x", GetLastError());
221 /* Query Windows Firewall service config, first just to get buffer size */
222 /* Expected to fail, so don't test return value */
223 (void) QueryServiceConfig(svc, NULL, 0, &BufSize);
224 status = GetLastError();
225 if (status != ERROR_INSUFFICIENT_BUFFER)
228 /* Allocate buffer */
229 pConfig = (LPQUERY_SERVICE_CONFIG)GlobalAlloc(GMEM_FIXED,BufSize);
233 /* Query Windows Firewall service config, this time for real */
234 flag = QueryServiceConfig(svc, pConfig, BufSize, &BufSize);
236 afsi_log("QueryServiceConfig failed: 0x%x", GetLastError());
240 /* Is it autostart? */
241 afsi_log("AutoStart 0x%x", pConfig->dwStartType);
242 if (pConfig->dwStartType < SERVICE_DEMAND_START)
248 CloseServiceHandle(svc);
250 CloseServiceHandle(scm);
256 cm_DaemonCheckInit(void)
263 code = RegOpenKeyEx(HKEY_LOCAL_MACHINE, AFSREG_CLT_SVC_PARAM_SUBKEY,
264 0, KEY_QUERY_VALUE, &parmKey);
268 dummyLen = sizeof(DWORD);
269 code = RegQueryValueEx(parmKey, "daemonCheckDownInterval", NULL, NULL,
270 (BYTE *) &dummy, &dummyLen);
271 if (code == ERROR_SUCCESS)
272 cm_daemonCheckDownInterval = dummy;
273 afsi_log("daemonCheckDownInterval is %d", cm_daemonCheckDownInterval);
275 dummyLen = sizeof(DWORD);
276 code = RegQueryValueEx(parmKey, "daemonCheckUpInterval", NULL, NULL,
277 (BYTE *) &dummy, &dummyLen);
278 if (code == ERROR_SUCCESS)
279 cm_daemonCheckUpInterval = dummy;
280 afsi_log("daemonCheckUpInterval is %d", cm_daemonCheckUpInterval);
282 dummyLen = sizeof(DWORD);
283 code = RegQueryValueEx(parmKey, "daemonCheckVolInterval", NULL, NULL,
284 (BYTE *) &dummy, &dummyLen);
285 if (code == ERROR_SUCCESS)
286 cm_daemonCheckVolInterval = dummy;
287 afsi_log("daemonCheckVolInterval is %d", cm_daemonCheckVolInterval);
289 dummyLen = sizeof(DWORD);
290 code = RegQueryValueEx(parmKey, "daemonCheckCBInterval", NULL, NULL,
291 (BYTE *) &dummy, &dummyLen);
292 if (code == ERROR_SUCCESS)
293 cm_daemonCheckCBInterval = dummy;
294 afsi_log("daemonCheckCBInterval is %d", cm_daemonCheckCBInterval);
296 dummyLen = sizeof(DWORD);
297 code = RegQueryValueEx(parmKey, "daemonCheckVolCBInterval", NULL, NULL,
298 (BYTE *) &dummy, &dummyLen);
299 if (code == ERROR_SUCCESS)
300 cm_daemonCheckVolCBInterval = dummy;
301 afsi_log("daemonCheckVolCBInterval is %d", cm_daemonCheckVolCBInterval);
303 dummyLen = sizeof(DWORD);
304 code = RegQueryValueEx(parmKey, "daemonCheckLockInterval", NULL, NULL,
305 (BYTE *) &dummy, &dummyLen);
306 if (code == ERROR_SUCCESS)
307 cm_daemonCheckLockInterval = dummy;
308 afsi_log("daemonCheckLockInterval is %d", cm_daemonCheckLockInterval);
310 dummyLen = sizeof(DWORD);
311 code = RegQueryValueEx(parmKey, "daemonCheckTokenInterval", NULL, NULL,
312 (BYTE *) &dummy, &dummyLen);
313 if (code == ERROR_SUCCESS)
314 cm_daemonTokenCheckInterval = dummy;
315 afsi_log("daemonCheckTokenInterval is %d", cm_daemonTokenCheckInterval);
317 dummyLen = sizeof(DWORD);
318 code = RegQueryValueEx(parmKey, "daemonCheckOfflineVolInterval", NULL, NULL,
319 (BYTE *) &dummy, &dummyLen);
320 if (code == ERROR_SUCCESS)
321 cm_daemonCheckOfflineVolInterval = dummy;
322 afsi_log("daemonCheckOfflineVolInterval is %d", cm_daemonCheckOfflineVolInterval);
324 RegCloseKey(parmKey);
327 /* periodic check daemon */
328 void cm_Daemon(long parm)
331 time_t lastLockCheck;
333 time_t lastCBExpirationCheck;
334 time_t lastVolCBRenewalCheck;
335 time_t lastDownServerCheck;
336 time_t lastUpServerCheck;
337 time_t lastTokenCacheCheck;
338 time_t lastBusyVolCheck;
343 char * name = "cm_Daemon_ShutdownEvent";
344 int configureFirewall = IsWindowsFirewallPresent();
345 int bAddrChangeCheck = 0;
347 cm_Daemon_ShutdownEvent = thrd_CreateEvent(NULL, FALSE, FALSE, name);
348 if ( GetLastError() == ERROR_ALREADY_EXISTS )
349 afsi_log("Event Object Already Exists: %s", name);
351 if (!configureFirewall) {
352 afsi_log("No Windows Firewall detected");
355 /* ping all file servers, up or down, with unauthenticated connection,
356 * to find out whether we have all our callbacks from the server still.
357 * Also, ping down VLDBs.
360 * Seed the random number generator with our own address, so that
361 * clients starting at the same time don't all do vol checks at the
364 gethostname(thostName, sizeof(thostName));
365 thp = gethostbyname(thostName);
366 if (thp == NULL) /* In djgpp, gethostname returns the netbios
367 name of the machine. gethostbyname will fail
368 looking this up if it differs from DNS name. */
371 memcpy(&code, thp->h_addr_list[0], 4);
375 cm_DaemonCheckInit();
378 lastVolCheck = now - cm_daemonCheckVolInterval/2 + (rand() % cm_daemonCheckVolInterval);
379 lastCBExpirationCheck = now - cm_daemonCheckCBInterval/2 + (rand() % cm_daemonCheckCBInterval);
380 if (cm_daemonCheckVolCBInterval)
381 lastVolCBRenewalCheck = now - cm_daemonCheckVolCBInterval/2 + (rand() % cm_daemonCheckVolCBInterval);
382 lastLockCheck = now - cm_daemonCheckLockInterval/2 + (rand() % cm_daemonCheckLockInterval);
383 lastDownServerCheck = now - cm_daemonCheckDownInterval/2 + (rand() % cm_daemonCheckDownInterval);
384 lastUpServerCheck = now - cm_daemonCheckUpInterval/2 + (rand() % cm_daemonCheckUpInterval);
385 lastTokenCacheCheck = now - cm_daemonTokenCheckInterval/2 + (rand() % cm_daemonTokenCheckInterval);
386 lastBusyVolCheck = now - cm_daemonCheckOfflineVolInterval/2 * (rand() % cm_daemonCheckOfflineVolInterval);
388 while (daemon_ShutdownFlag == 0) {
389 /* check to see if the listener threads halted due to network
390 * disconnect or other issues. If so, attempt to restart them.
392 smb_RestartListeners(0);
394 if (daemon_ShutdownFlag == 1)
397 if (configureFirewall) {
398 /* Open Microsoft Firewall to allow in port 7001 */
399 switch (icf_CheckAndAddAFSPorts(AFS_PORTSET_CLIENT)) {
401 afsi_log("Windows Firewall Configuration succeeded");
402 configureFirewall = 0;
405 afsi_log("Invalid Windows Firewall Port Set");
408 afsi_log("Unable to open Windows Firewall Profile");
411 afsi_log("Unable to create/modify Windows Firewall Port entries");
414 afsi_log("Unknown Windows Firewall Configuration error");
418 /* find out what time it is */
421 /* Determine whether an address change took place that we need to respond to */
422 if (bAddrChangeCheck)
423 bAddrChangeCheck = 0;
425 if (lastIPAddrChange != 0 && lastIPAddrChange + 2500 < now) {
426 bAddrChangeCheck = 1;
427 lastIPAddrChange = 0;
430 /* check down servers */
431 if ((bAddrChangeCheck || now > lastDownServerCheck + cm_daemonCheckDownInterval) &&
432 daemon_ShutdownFlag == 0) {
433 lastDownServerCheck = now;
434 osi_Log0(afsd_logp, "cm_Daemon CheckDownServers");
435 cm_CheckServers(CM_FLAG_CHECKDOWNSERVERS, NULL);
436 if (daemon_ShutdownFlag == 1)
441 if (bAddrChangeCheck)
442 cm_ForceNewConnectionsAllServers();
444 /* check up servers */
445 if ((bAddrChangeCheck || now > lastUpServerCheck + cm_daemonCheckUpInterval) &&
446 daemon_ShutdownFlag == 0) {
447 lastUpServerCheck = now;
448 osi_Log0(afsd_logp, "cm_Daemon CheckUpServers");
449 cm_CheckServers(CM_FLAG_CHECKUPSERVERS, NULL);
450 if (daemon_ShutdownFlag == 1)
455 if (bAddrChangeCheck) {
457 cm_VolStatus_Network_Addr_Change();
460 if (now > lastVolCheck + cm_daemonCheckVolInterval &&
461 daemon_ShutdownFlag == 0) {
464 if (daemon_ShutdownFlag == 1)
469 if (cm_daemonCheckVolCBInterval &&
470 now > lastVolCBRenewalCheck + cm_daemonCheckVolCBInterval &&
471 daemon_ShutdownFlag == 0) {
472 lastVolCBRenewalCheck = now;
473 cm_VolumeRenewROCallbacks();
474 if (daemon_ShutdownFlag == 1)
479 if ((bAddrChangeCheck || now > lastBusyVolCheck + cm_daemonCheckOfflineVolInterval) &&
480 daemon_ShutdownFlag == 0) {
482 cm_CheckOfflineVolumes();
483 if (daemon_ShutdownFlag == 1)
488 if (now > lastCBExpirationCheck + cm_daemonCheckCBInterval &&
489 daemon_ShutdownFlag == 0) {
490 lastCBExpirationCheck = now;
491 cm_CheckCBExpiration();
492 if (daemon_ShutdownFlag == 1)
497 if (now > lastLockCheck + cm_daemonCheckLockInterval &&
498 daemon_ShutdownFlag == 0) {
501 if (daemon_ShutdownFlag == 1)
506 if (now > lastTokenCacheCheck + cm_daemonTokenCheckInterval &&
507 daemon_ShutdownFlag == 0) {
508 lastTokenCacheCheck = now;
509 cm_CheckTokenCache(now);
510 if (daemon_ShutdownFlag == 1)
515 /* allow an exit to be called prior to stopping the service */
516 hHookDll = LoadLibrary(AFSD_HOOK_DLL);
520 AfsdDaemonHook daemonHook = ( AfsdDaemonHook ) GetProcAddress(hHookDll, AFSD_DAEMON_HOOK);
523 hookRc = daemonHook();
525 FreeLibrary(hHookDll);
530 SetEvent(WaitToTerminate);
534 if (daemon_ShutdownFlag == 1) {
537 thrd_Sleep(10000); /* sleep 10 seconds */
539 thrd_SetEvent(cm_Daemon_ShutdownEvent);
542 void cm_DaemonShutdown(void)
547 daemon_ShutdownFlag = 1;
548 osi_Wakeup((LONG_PTR) &cm_bkgListp);
550 /* wait for shutdown */
551 if (cm_Daemon_ShutdownEvent)
552 code = thrd_WaitForSingleObject_Event(cm_Daemon_ShutdownEvent, INFINITE);
554 for ( i=0; i<cm_nDaemons; i++) {
555 if (cm_BkgDaemon_ShutdownEvent[i])
556 code = thrd_WaitForSingleObject_Event(cm_BkgDaemon_ShutdownEvent[i], INFINITE);
559 if (cm_IPAddrDaemon_ShutdownEvent)
560 code = thrd_WaitForSingleObject_Event(cm_IPAddrDaemon_ShutdownEvent, INFINITE);
563 void cm_InitDaemon(int nDaemons)
565 static osi_once_t once;
570 cm_nDaemons = (nDaemons > CM_MAX_DAEMONS) ? CM_MAX_DAEMONS : nDaemons;
572 if (osi_Once(&once)) {
573 lock_InitializeRWLock(&cm_daemonLock, "cm_daemonLock");
576 /* creating IP Address Change monitor daemon */
577 phandle = thrd_Create((SecurityAttrib) 0, 0,
578 (ThreadFunc) cm_IpAddrDaemon, 0, 0, &pid, "cm_IpAddrDaemon");
579 osi_assertx(phandle != NULL, "cm_IpAddrDaemon thread creation failure");
580 thrd_CloseHandle(phandle);
582 /* creating pinging daemon */
583 phandle = thrd_Create((SecurityAttrib) 0, 0,
584 (ThreadFunc) cm_Daemon, 0, 0, &pid, "cm_Daemon");
585 osi_assertx(phandle != NULL, "cm_Daemon thread creation failure");
586 thrd_CloseHandle(phandle);
588 for(i=0; i < cm_nDaemons; i++) {
589 phandle = thrd_Create((SecurityAttrib) 0, 0,
590 (ThreadFunc) cm_BkgDaemon, (LPVOID)i, 0, &pid,
592 osi_assertx(phandle != NULL, "cm_BkgDaemon thread creation failure");
593 thrd_CloseHandle(phandle);