Fix context RefCount when debugger is enabled v0.2.26

Azure · Feb 21, 2017 · 6478fde · 6478fde
1 parent c53c71e
commit 6478fde
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/iisnode/cprotocolbridge.cpp b/src/iisnode/cprotocolbridge.cpp
@@ -481,6 +481,7 @@ HRESULT CProtocolBridge::InitiateRequest(CNodeHttpStoredContext* context)
     IHttpContext* child = NULL;
     BOOL completionExpected;
     BOOL fCompletionPosted = FALSE;
+    BOOL fReference = FALSE;
 
     // determine what the target path of the request is
 
@@ -541,12 +542,14 @@ HRESULT CProtocolBridge::InitiateRequest(CNodeHttpStoredContext* context)
         context->SetNextProcessor(CProtocolBridge::ChildContextCompleted);
 
         context->ReferenceNodeHttpStoredContext();
+        fReference = TRUE;
 
         CheckError(context->GetHttpContext()->ExecuteRequest(TRUE, child, 0, NULL, &completionExpected));
         if (!completionExpected)
         {
             CProtocolBridge::ChildContextCompleted(S_OK, 0, context->GetOverlapped(), &fCompletionPosted);
             context->DereferenceNodeHttpStoredContext();
+            fReference = FALSE;
         }
     }
     else
@@ -565,6 +568,11 @@ HRESULT CProtocolBridge::InitiateRequest(CNodeHttpStoredContext* context)
         context->SetChildContext(NULL);
     }
 
+    if(fReference)
+    {
+        context->DereferenceNodeHttpStoredContext();
+    }
+
     return hr;
 }