#include <afs/param.h>
#include <afs/stds.h>
-#ifndef DJGPP
#include <windows.h>
#include <winsock2.h>
-#else
-#include <netdb.h>
-#endif /* !DJGPP */
+#include <iphlpapi.h>
#include <stdlib.h>
#include <malloc.h>
#include <string.h>
+#include "afsd.h"
+
#include <rx/rx.h>
+#include <rx/rx_prototypes.h>
+#include <WINNT/afsreg.h>
-#include "afsd.h"
+#include "afsicf.h"
-long cm_daemonCheckInterval = 30;
+/* in seconds */
+long cm_daemonCheckDownInterval = 180;
+long cm_daemonCheckUpInterval = 600;
+long cm_daemonCheckVolInterval = 3600;
+long cm_daemonCheckCBInterval = 60;
+long cm_daemonCheckLockInterval = 60;
long cm_daemonTokenCheckInterval = 180;
osi_rwlock_t cm_daemonLock;
cm_bkgRequest_t *cm_bkgListp; /* first elt in the list of requests */
cm_bkgRequest_t *cm_bkgListEndp; /* last elt in the list of requests */
+static int daemon_ShutdownFlag = 0;
+
+void cm_IpAddrDaemon(long parm)
+{
+ extern void smb_CheckVCs(void);
+
+ rx_StartClientThread();
+
+ while (daemon_ShutdownFlag == 0) {
+ DWORD Result = NotifyAddrChange(NULL,NULL);
+ if (Result == NO_ERROR && daemon_ShutdownFlag == 0) {
+ osi_Log0(afsd_logp, "cm_IpAddrDaemon CheckDownServers");
+ Sleep(2500);
+ cm_ForceNewConnectionsAllServers();
+ cm_CheckServers(CM_FLAG_CHECKUPSERVERS | CM_FLAG_CHECKDOWNSERVERS, NULL);
+ smb_CheckVCs();
+ }
+ }
+}
+
void cm_BkgDaemon(long parm)
{
- cm_bkgRequest_t *rp;
-
- lock_ObtainWrite(&cm_daemonLock);
- while(1) {
- if (!cm_bkgListEndp) {
- osi_SleepW((long) &cm_bkgListp, &cm_daemonLock);
- lock_ObtainWrite(&cm_daemonLock);
- continue;
- }
-
- /* we found a request */
- rp = cm_bkgListEndp;
- cm_bkgListEndp = (cm_bkgRequest_t *) osi_QPrev(&rp->q);
- osi_QRemove((osi_queue_t **) &cm_bkgListp, &rp->q);
- osi_assert(cm_bkgQueueCount-- > 0);
- lock_ReleaseWrite(&cm_daemonLock);
+ cm_bkgRequest_t *rp;
+
+ rx_StartClientThread();
+
+ lock_ObtainWrite(&cm_daemonLock);
+ while (daemon_ShutdownFlag == 0) {
+ if (!cm_bkgListEndp) {
+ osi_SleepW((LONG_PTR)&cm_bkgListp, &cm_daemonLock);
+ lock_ObtainWrite(&cm_daemonLock);
+ continue;
+ }
- (*rp->procp)(rp->scp, rp->p1, rp->p2, rp->p3, rp->p4, rp->userp);
+ /* we found a request */
+ rp = cm_bkgListEndp;
+ cm_bkgListEndp = (cm_bkgRequest_t *) osi_QPrev(&rp->q);
+ osi_QRemove((osi_queue_t **) &cm_bkgListp, &rp->q);
+ osi_assert(cm_bkgQueueCount-- > 0);
+ lock_ReleaseWrite(&cm_daemonLock);
+
+ osi_Log2(afsd_logp,"cm_BkgDaemon (before) scp 0x%x ref %d",rp->scp, rp->scp->refCount);
+
+ (*rp->procp)(rp->scp, rp->p1, rp->p2, rp->p3, rp->p4, rp->userp);
- cm_ReleaseUser(rp->userp);
- cm_ReleaseSCache(rp->scp);
- free(rp);
+ osi_Log2(afsd_logp,"cm_BkgDaemon (after) scp 0x%x ref %d",rp->scp, rp->scp->refCount);
- lock_ObtainWrite(&cm_daemonLock);
- }
+ cm_ReleaseUser(rp->userp);
+ cm_ReleaseSCache(rp->scp);
+ free(rp);
+
+ lock_ObtainWrite(&cm_daemonLock);
+ }
+ lock_ReleaseWrite(&cm_daemonLock);
}
-void cm_QueueBKGRequest(cm_scache_t *scp, cm_bkgProc_t *procp, long p1, long p2, long p3, long p4,
+void cm_QueueBKGRequest(cm_scache_t *scp, cm_bkgProc_t *procp, afs_uint32 p1, afs_uint32 p2, afs_uint32 p3, afs_uint32 p4,
cm_user_t *userp)
{
- cm_bkgRequest_t *rp;
+ cm_bkgRequest_t *rp;
rp = malloc(sizeof(*rp));
memset(rp, 0, sizeof(*rp));
rp->p4 = p4;
lock_ObtainWrite(&cm_daemonLock);
- cm_bkgQueueCount++;
+ cm_bkgQueueCount++;
osi_QAdd((osi_queue_t **) &cm_bkgListp, &rp->q);
- if (!cm_bkgListEndp) cm_bkgListEndp = rp;
+ if (!cm_bkgListEndp)
+ cm_bkgListEndp = rp;
lock_ReleaseWrite(&cm_daemonLock);
- osi_Wakeup((long) &cm_bkgListp);
+ osi_Wakeup((LONG_PTR) &cm_bkgListp);
+}
+
+static int
+IsWindowsFirewallPresent(void)
+{
+ SC_HANDLE scm;
+ SC_HANDLE svc;
+ BOOLEAN flag;
+ BOOLEAN result = FALSE;
+ LPQUERY_SERVICE_CONFIG pConfig = NULL;
+ DWORD BufSize;
+ LONG status;
+
+ /* Open services manager */
+ scm = OpenSCManager(NULL, NULL, GENERIC_READ);
+ if (!scm) return FALSE;
+
+ /* Open Windows Firewall service */
+ svc = OpenService(scm, "MpsSvc", SERVICE_QUERY_CONFIG);
+ if (!svc) {
+ afsi_log("MpsSvc Service could not be opened for query: 0x%x", GetLastError());
+ svc = OpenService(scm, "SharedAccess", SERVICE_QUERY_CONFIG);
+ if (!svc)
+ afsi_log("SharedAccess Service could not be opened for query: 0x%x", GetLastError());
+ }
+ if (!svc)
+ goto close_scm;
+
+ /* Query Windows Firewall service config, first just to get buffer size */
+ /* Expected to fail, so don't test return value */
+ (void) QueryServiceConfig(svc, NULL, 0, &BufSize);
+ status = GetLastError();
+ if (status != ERROR_INSUFFICIENT_BUFFER)
+ goto close_svc;
+
+ /* Allocate buffer */
+ pConfig = (LPQUERY_SERVICE_CONFIG)GlobalAlloc(GMEM_FIXED,BufSize);
+ if (!pConfig)
+ goto close_svc;
+
+ /* Query Windows Firewall service config, this time for real */
+ flag = QueryServiceConfig(svc, pConfig, BufSize, &BufSize);
+ if (!flag) {
+ afsi_log("QueryServiceConfig failed: 0x%x", GetLastError());
+ goto free_pConfig;
+ }
+
+ /* Is it autostart? */
+ afsi_log("AutoStart 0x%x", pConfig->dwStartType);
+ if (pConfig->dwStartType < SERVICE_DEMAND_START)
+ result = TRUE;
+
+ free_pConfig:
+ GlobalFree(pConfig);
+ close_svc:
+ CloseServiceHandle(svc);
+ close_scm:
+ CloseServiceHandle(scm);
+
+ return result;
+}
+
+void
+cm_DaemonCheckInit(void)
+{
+ HKEY parmKey;
+ DWORD dummyLen;
+ DWORD dummy;
+ DWORD code;
+
+ code = RegOpenKeyEx(HKEY_LOCAL_MACHINE, AFSREG_CLT_SVC_PARAM_SUBKEY,
+ 0, KEY_QUERY_VALUE, &parmKey);
+ if (code)
+ return;
+
+ dummyLen = sizeof(DWORD);
+ code = RegQueryValueEx(parmKey, "DownServerCheckInterval", NULL, NULL,
+ (BYTE *) &dummy, &dummyLen);
+ if (code == ERROR_SUCCESS)
+ cm_daemonCheckDownInterval = dummy;
+
+ dummyLen = sizeof(DWORD);
+ code = RegQueryValueEx(parmKey, "UpServerCheckInterval", NULL, NULL,
+ (BYTE *) &dummy, &dummyLen);
+ if (code == ERROR_SUCCESS)
+ cm_daemonCheckUpInterval = dummy;
+
+ dummyLen = sizeof(DWORD);
+ code = RegQueryValueEx(parmKey, "VolumeCheckInterval", NULL, NULL,
+ (BYTE *) &dummy, &dummyLen);
+ if (code == ERROR_SUCCESS)
+ cm_daemonCheckVolInterval = dummy;
+
+ dummyLen = sizeof(DWORD);
+ code = RegQueryValueEx(parmKey, "CallbackCheckInterval", NULL, NULL,
+ (BYTE *) &dummy, &dummyLen);
+ if (code == ERROR_SUCCESS)
+ cm_daemonCheckCBInterval = dummy;
+
+ dummyLen = sizeof(DWORD);
+ code = RegQueryValueEx(parmKey, "LockCheckInterval", NULL, NULL,
+ (BYTE *) &dummy, &dummyLen);
+ if (code == ERROR_SUCCESS)
+ cm_daemonCheckLockInterval = dummy;
+
+ dummyLen = sizeof(DWORD);
+ code = RegQueryValueEx(parmKey, "TokenCheckInterval", NULL, NULL,
+ (BYTE *) &dummy, &dummyLen);
+ if (code == ERROR_SUCCESS)
+ cm_daemonTokenCheckInterval = dummy;
+
+ RegCloseKey(parmKey);
}
/* periodic check daemon */
void cm_Daemon(long parm)
{
- long now;
- long lastLockCheck;
- long lastVolCheck;
- long lastCBExpirationCheck;
- long lastDownServerCheck;
- long lastUpServerCheck;
- long lastTokenCacheCheck;
- char thostName[200];
- long code;
- struct hostent *thp;
-
- /* ping all file servers, up or down, with unauthenticated connection,
- * to find out whether we have all our callbacks from the server still.
- * Also, ping down VLDBs.
- */
- /*
- * Seed the random number generator with our own address, so that
- * clients starting at the same time don't all do vol checks at the
- * same time.
- */
- gethostname(thostName, sizeof(thostName));
- thp = gethostbyname(thostName);
- if (thp == NULL) /* In djgpp, gethostname returns the netbios
- name of the machine. gethostbyname will fail
- looking this up if it differs from DNS name. */
- code = 0;
- else
- memcpy(&code, thp->h_addr_list[0], 4);
- srand(ntohl(code));
-
- now = osi_Time();
- lastVolCheck = now - 1800 + (rand() % 3600);
- lastCBExpirationCheck = now - 60 + (rand() % 60);
- lastLockCheck = now - 60 + (rand() % 60);
- lastDownServerCheck = now - cm_daemonCheckInterval/2 + (rand() % cm_daemonCheckInterval);
- lastUpServerCheck = now - 1800 + (rand() % 3600);
- lastTokenCacheCheck = now - cm_daemonTokenCheckInterval/2 + (rand() % cm_daemonTokenCheckInterval);
-
- while (1) {
- thrd_Sleep(30 * 1000); /* sleep 30 seconds */
-
- /* find out what time it is */
- now = osi_Time();
-
- /* check down servers */
- if (now > lastDownServerCheck + cm_daemonCheckInterval) {
- lastDownServerCheck = now;
- cm_CheckServers(CM_FLAG_CHECKDOWNSERVERS, NULL);
- }
-
- /* check up servers */
- if (now > lastUpServerCheck + 3600) {
- lastUpServerCheck = now;
- cm_CheckServers(CM_FLAG_CHECKUPSERVERS, NULL);
- }
-
- if (now > lastVolCheck + 3600) {
- lastVolCheck = now;
- cm_CheckVolumes();
- }
-
- if (now > lastCBExpirationCheck + 60) {
- lastCBExpirationCheck = now;
- cm_CheckCBExpiration();
- }
-
- if (now > lastLockCheck + 60) {
- lastLockCheck = now;
- cm_CheckLocks();
- }
-
- if (now > lastTokenCacheCheck + cm_daemonTokenCheckInterval) {
- lastTokenCacheCheck = now;
- cm_CheckTokenCache(now);
- }
+ time_t now;
+ time_t lastLockCheck;
+ time_t lastVolCheck;
+ time_t lastCBExpirationCheck;
+ time_t lastDownServerCheck;
+ time_t lastUpServerCheck;
+ time_t lastTokenCacheCheck;
+ char thostName[200];
+ unsigned long code;
+ struct hostent *thp;
+ HMODULE hHookDll;
+ int configureFirewall = IsWindowsFirewallPresent();
+
+ if (!configureFirewall) {
+ afsi_log("No Windows Firewall detected");
+ }
+
+ /* ping all file servers, up or down, with unauthenticated connection,
+ * to find out whether we have all our callbacks from the server still.
+ * Also, ping down VLDBs.
+ */
+ /*
+ * Seed the random number generator with our own address, so that
+ * clients starting at the same time don't all do vol checks at the
+ * same time.
+ */
+ gethostname(thostName, sizeof(thostName));
+ thp = gethostbyname(thostName);
+ if (thp == NULL) /* In djgpp, gethostname returns the netbios
+ name of the machine. gethostbyname will fail
+ looking this up if it differs from DNS name. */
+ code = 0;
+ else
+ memcpy(&code, thp->h_addr_list[0], 4);
+
+ srand(ntohl(code));
+
+ cm_DaemonCheckInit();
+
+ now = osi_Time();
+ lastVolCheck = now - cm_daemonCheckVolInterval/2 + (rand() % cm_daemonCheckVolInterval);
+ lastCBExpirationCheck = now - cm_daemonCheckCBInterval/2 + (rand() % cm_daemonCheckCBInterval);
+ lastLockCheck = now - cm_daemonCheckLockInterval/2 + (rand() % cm_daemonCheckLockInterval);
+ lastDownServerCheck = now - cm_daemonCheckDownInterval/2 + (rand() % cm_daemonCheckDownInterval);
+ lastUpServerCheck = now - cm_daemonCheckUpInterval/2 + (rand() % cm_daemonCheckUpInterval);
+ lastTokenCacheCheck = now - cm_daemonTokenCheckInterval/2 + (rand() % cm_daemonTokenCheckInterval);
+
+ while (daemon_ShutdownFlag == 0) {
+ if (configureFirewall) {
+ /* Open Microsoft Firewall to allow in port 7001 */
+ switch (icf_CheckAndAddAFSPorts(AFS_PORTSET_CLIENT)) {
+ case 0:
+ afsi_log("Windows Firewall Configuration succeeded");
+ configureFirewall = 0;
+ break;
+ case 1:
+ afsi_log("Invalid Windows Firewall Port Set");
+ break;
+ case 2:
+ afsi_log("Unable to open Windows Firewall Profile");
+ break;
+ case 3:
+ afsi_log("Unable to create/modify Windows Firewall Port entries");
+ break;
+ default:
+ afsi_log("Unknown Windows Firewall Configuration error");
+ }
+ }
+
+ /* find out what time it is */
+ now = osi_Time();
+
+ /* check down servers */
+ if (now > lastDownServerCheck + cm_daemonCheckDownInterval) {
+ lastDownServerCheck = now;
+ osi_Log0(afsd_logp, "cm_Daemon CheckDownServers");
+ cm_CheckServers(CM_FLAG_CHECKDOWNSERVERS, NULL);
+ now = osi_Time();
+ }
+
+ /* check up servers */
+ if (now > lastUpServerCheck + cm_daemonCheckUpInterval) {
+ lastUpServerCheck = now;
+ osi_Log0(afsd_logp, "cm_Daemon CheckUpServers");
+ cm_CheckServers(CM_FLAG_CHECKUPSERVERS, NULL);
+ now = osi_Time();
+ }
+
+ if (now > lastVolCheck + cm_daemonCheckVolInterval) {
+ lastVolCheck = now;
+ cm_CheckVolumes();
+ now = osi_Time();
}
+
+ if (now > lastCBExpirationCheck + cm_daemonCheckCBInterval) {
+ lastCBExpirationCheck = now;
+ cm_CheckCBExpiration();
+ now = osi_Time();
+ }
+
+ if (now > lastLockCheck + cm_daemonCheckLockInterval) {
+ lastLockCheck = now;
+ cm_CheckLocks();
+ now = osi_Time();
+ }
+
+ if (now > lastTokenCacheCheck + cm_daemonTokenCheckInterval) {
+ lastTokenCacheCheck = now;
+ cm_CheckTokenCache(now);
+ now = osi_Time();
+ }
+
+ /* allow an exit to be called prior to stopping the service */
+ hHookDll = LoadLibrary(AFSD_HOOK_DLL);
+ if (hHookDll)
+ {
+ BOOL hookRc = TRUE;
+ AfsdDaemonHook daemonHook = ( AfsdDaemonHook ) GetProcAddress(hHookDll, AFSD_DAEMON_HOOK);
+ if (daemonHook)
+ {
+ hookRc = daemonHook();
+ }
+ FreeLibrary(hHookDll);
+ hHookDll = NULL;
+
+ if (hookRc == FALSE)
+ {
+ SetEvent(WaitToTerminate);
+ }
+ }
+
+ thrd_Sleep(30 * 1000); /* sleep 30 seconds */
+ if (daemon_ShutdownFlag == 1)
+ return;
+ }
+}
+
+void cm_DaemonShutdown(void)
+{
+ daemon_ShutdownFlag = 1;
}
void cm_InitDaemon(int nDaemons)
{
- static osi_once_t once;
- long pid;
- thread_t phandle;
- int i;
+ static osi_once_t once;
+ long pid;
+ thread_t phandle;
+ int i;
- if (osi_Once(&once)) {
- lock_InitializeRWLock(&cm_daemonLock, "cm_daemonLock");
- osi_EndOnce(&once);
-
- /* creating pinging daemon */
- phandle = thrd_Create((SecurityAttrib) 0, 0,
- (ThreadFunc) cm_Daemon, 0, 0, &pid, "cm_Daemon");
- osi_assert(phandle != NULL);
-
- thrd_CloseHandle(phandle);
- for(i=0; i < nDaemons; i++) {
- phandle = thrd_Create((SecurityAttrib) 0, 0,
- (ThreadFunc) cm_BkgDaemon, 0, 0, &pid,
- "cm_BkgDaemon");
- osi_assert(phandle != NULL);
- thrd_CloseHandle(phandle);
- }
+ if (osi_Once(&once)) {
+ lock_InitializeRWLock(&cm_daemonLock, "cm_daemonLock");
+ osi_EndOnce(&once);
+
+ /* creating IP Address Change monitor daemon */
+ phandle = thrd_Create((SecurityAttrib) 0, 0,
+ (ThreadFunc) cm_IpAddrDaemon, 0, 0, &pid, "cm_IpAddrDaemon");
+ osi_assert(phandle != NULL);
+ thrd_CloseHandle(phandle);
+
+ /* creating pinging daemon */
+ phandle = thrd_Create((SecurityAttrib) 0, 0,
+ (ThreadFunc) cm_Daemon, 0, 0, &pid, "cm_Daemon");
+ osi_assert(phandle != NULL);
+ thrd_CloseHandle(phandle);
+
+ for(i=0; i < nDaemons; i++) {
+ phandle = thrd_Create((SecurityAttrib) 0, 0,
+ (ThreadFunc) cm_BkgDaemon, 0, 0, &pid,
+ "cm_BkgDaemon");
+ osi_assert(phandle != NULL);
+ thrd_CloseHandle(phandle);
}
+ }
}