From 79b97bfa9f78ec1dea3c4cf0dc7de0dabaf3b211 Mon Sep 17 00:00:00 2001
From: Jeffrey Altman <jaltman@mit.edu>
Date: Tue, 7 Dec 2004 06:16:22 +0000
Subject: [PATCH] STABLE14-windows-shutdown-20041205

Synchronize the shutdown of all smb_Server threads by waiting on a
shutdown event before continuing

Do not lock a buffer object if the attempt to obtain the buffer failed.


(cherry picked from commit fe97ddf25b550365fd54a3300fa7388cf71d8631)
---
 src/WINNT/afsd/cm_vnodeops.c |  7 +++++--
 src/WINNT/afsd/smb.c         | 33 ++++++++++++++++++++++++++++-----
 2 files changed, 33 insertions(+), 7 deletions(-)

diff --git a/src/WINNT/afsd/cm_vnodeops.c b/src/WINNT/afsd/cm_vnodeops.c
index 2ee374494..9267a2a31 100644
--- a/src/WINNT/afsd/cm_vnodeops.c
+++ b/src/WINNT/afsd/cm_vnodeops.c
@@ -552,10 +552,13 @@ long cm_ApplyDir(cm_scache_t *scp, cm_DirFuncp_t funcp, void *parmp,
             lock_ObtainRead(&scp->bufCreateLock);
             code = buf_Get(scp, &thyper, &bufferp);
             lock_ReleaseRead(&scp->bufCreateLock);
+			if (code) {
+				/* if buf_Get() fails we do not have a buffer object to lock */
+                bufferp = NULL;
+                break;
+			}
 
             lock_ObtainMutex(&bufferp->mx);
-            if (code) 
-                break;
             bufferOffset = thyper;
 
             /* now get the data in the cache */
diff --git a/src/WINNT/afsd/smb.c b/src/WINNT/afsd/smb.c
index af73b6242..6ace391c0 100644
--- a/src/WINNT/afsd/smb.c
+++ b/src/WINNT/afsd/smb.c
@@ -95,6 +95,7 @@ LSA_STRING smb_lsaLogonOrigin;
 #define NCBmax MAXIMUM_WAIT_OBJECTS
 EVENT_HANDLE NCBavails[NCBmax], NCBevents[NCBmax];
 EVENT_HANDLE **NCBreturns;
+EVENT_HANDLE *smb_ServerShutdown;
 DWORD NCBsessions[NCBmax];
 NCB *NCBs[NCBmax];
 struct smb_packet *bufs[NCBmax];
@@ -2862,9 +2863,13 @@ void smb_Daemon(void *parmp)
 {
     afs_uint32 count = 0;
 
-    while(1) {
+    while(smbShutdownFlag == 0) {
         count++;
         thrd_Sleep(10000);
+
+        if (smbShutdownFlag == 1)
+            break;
+        
         if ((count % 72) == 0)	{	/* every five minutes */
             struct tm myTime;
             time_t old_localZero = smb_localZero;
@@ -6665,9 +6670,10 @@ void smb_Server(VOID *parmp)
 
         /* terminate silently if shutdown flag is set */
         if (code == WAIT_OBJECT_0) {
-            if (smbShutdownFlag == 1)
+            if (smbShutdownFlag == 1) {
+                thrd_SetEvent(smb_ServerShutdown[myIdx]);
                 break;
-            else
+            } else
                 continue;
         }
 
@@ -7535,6 +7541,15 @@ void smb_Init(osi_log_t *logp, char *snamep, int useV3, int LANadapt,
         NCBreturns[i] = malloc(NCBmax * sizeof(EVENT_HANDLE));
         NCBreturns[i][0] = retHandle;
     }
+
+    smb_ServerShutdown = malloc(smb_NumServerThreads * sizeof(EVENT_HANDLE));
+    for (i = 0; i < smb_NumServerThreads; i++) {
+        sprintf(eventName, "smb_ServerShutdown[%d]", i);
+        smb_ServerShutdown[i] = thrd_CreateEvent(NULL, FALSE, FALSE, eventName);
+        if ( GetLastError() == ERROR_ALREADY_EXISTS )
+            afsi_log("Event Object Already Exists: %s", eventName);
+    }
+
     for (i = 1; i <= nThreads; i++)
         InitNCBslot(i);
     numNCBs = nThreads + 1;
@@ -7831,13 +7846,21 @@ void smb_Shutdown(void)
     }
 
     /* Trigger the shutdown of all SMB threads */
-    for (i = 0; i < numSessions; i++)
+    for (i = 0; i < smb_NumServerThreads; i++)
         thrd_SetEvent(NCBreturns[i][0]);
 
     thrd_SetEvent(NCBevents[0]);
     thrd_SetEvent(SessionEvents[0]);
     thrd_SetEvent(NCBavails[0]);
-    thrd_Sleep(1000);
+
+    for (i = 0;i < smb_NumServerThreads; i++) {
+        DWORD code = thrd_WaitForSingleObject_Event(smb_ServerShutdown[i], INFINITE);
+        if (code == WAIT_OBJECT_0) {
+            continue;
+        } else {
+            afsi_log("smb_Shutdown[%d] wait error",i);
+        }
+    }
 }
 
 /* Get the UNC \\<servername>\<sharename> prefix. */
-- 
2.39.5