node sticky sessions fix

Azure · Feb 19, 2017 · 9b05ea4 · 9b05ea4
1 parent 8f9d43a
commit 9b05ea4
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/iisnode/cnodeprocessmanager.cpp b/src/iisnode/cnodeprocessmanager.cpp
@@ -135,11 +135,19 @@ int CNodeProcessManager::ExtractStickySessionsProcess( PCSTR pszCookie )
                     pEnd = pStart;
                     while (*pEnd) /* Works because end-of-string and FALSE are identical. */
                     {
+                        if((pEnd - pStart) >= sizeof(acProcess))
+                        {
+                            break;
+                        }
                         pEnd++;
                     }
                 }
-                memcpy(acProcess, pStart, pEnd - pStart); // copy result
-                return atoi(acProcess);
+
+                if((pEnd - pStart) < sizeof(acProcess))
+                {
+                    memcpy(acProcess, pStart, pEnd - pStart); // copy result
+                    return atoi(acProcess);
+                }
             }
         }
     }