Removing copy on shutdown

jkotalik · jkotalik · commit b4ea78a27380 · 2020-12-07T13:45:53.000-08:00
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/AspNetCore/applicationinfo.cpp b/src/Servers/IIS/AspNetCoreModuleV2/AspNetCore/applicationinfo.cpp
@@ -93,7 +93,6 @@ APPLICATION_INFO::CreateApplication(IHttpContext& pHttpContext)
             errorContext.statusCode = 500i16;
             errorContext.subStatusCode = 0i16;
 
-
             const auto hr = TryCreateApplication(pHttpContext, options, errorContext);
 
             if (FAILED_LOG(hr))
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/CommonLibTests/inprocess_application_tests.cpp b/src/Servers/IIS/AspNetCoreModuleV2/CommonLibTests/inprocess_application_tests.cpp
@@ -39,7 +39,7 @@ namespace InprocessTests
             {"InProcessExeLocation", exePath.data()}
         };
 
-        IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), parameters.data(), 1);
+        IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), parameters.data(), 1, std::wstring());
 
         ASSERT_STREQ(app->QueryExeLocation().c_str(), L"hello");
     }
@@ -59,7 +59,7 @@ namespace InprocessTests
             .WillByDefault(testing::Return(L""));
 
         auto requestHandlerConfig = std::unique_ptr<InProcessOptions>(MockInProcessOptions::CreateConfig());
-        IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), nullptr, 0);
+        IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), nullptr, 0, std::wstring());
 
         ASSERT_STREQ(app->QueryApplicationVirtualPath().c_str(), L"/SECTION5");
     }
@@ -79,7 +79,7 @@ namespace InprocessTests
             .WillByDefault(testing::Return(L""));
 
         auto requestHandlerConfig = std::unique_ptr<InProcessOptions>(MockInProcessOptions::CreateConfig());
-        IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), nullptr, 0);
+        IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), nullptr, 0, std::wstring());
 
         ASSERT_STREQ(app->QueryApplicationVirtualPath().c_str(), L"/");
     }
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/ShuttingDownApplication.h b/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/ShuttingDownApplication.h
@@ -31,8 +31,8 @@ class ShuttingDownHandler : public REQUEST_HANDLER
 class ShuttingDownApplication : public InProcessApplicationBase
 {
 public:
-    ShuttingDownApplication(IHttpServer& pHttpServer, IHttpApplication& pHttpApplication)
-        : InProcessApplicationBase(pHttpServer, pHttpApplication, std::wstring())
+    ShuttingDownApplication(IHttpServer& pHttpServer, IHttpApplication& pHttpApplication, std::wstring shadowCopy)
+        : InProcessApplicationBase(pHttpServer, pHttpApplication, shadowCopy)
     {
     }
 
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/StartupExceptionApplication.h b/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/StartupExceptionApplication.h
@@ -18,14 +18,15 @@ class StartupExceptionApplication : public InProcessApplicationBase
         const std::string& errorPageContent,
         USHORT statusCode,
         USHORT subStatusCode,
-        const std::string& statusText)
+        const std::string& statusText,
+        std::wstring shadowCopy)
         : m_disableLogs(disableLogs),
         m_HR(hr),
         m_error(errorPageContent),
         m_statusCode(statusCode),
         m_subStatusCode(subStatusCode),
         m_statusText(std::move(statusText)),
-        InProcessApplicationBase(pServer, pApplication, std::wstring())
+        InProcessApplicationBase(pServer, pApplication, shadowCopy)
     {
     }
 
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/dllmain.cpp b/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/dllmain.cpp
@@ -103,6 +103,7 @@ CreateApplication(
 {
     TraceContextScope traceScope(FindParameter<IHttpTraceContext*>("TraceContext", pParameters, nParameters));
     const auto pSite = FindParameter<IHttpSite*>("Site", pParameters, nParameters);
+    const auto shadowCopyPtr = FindParameter<PCWSTR>("ShadowCopyDirectory", pParameters, nParameters);
 
     try
     {
@@ -113,7 +114,7 @@ CreateApplication(
         // means that server is shutting does and request arrived in the meantime
         if (g_fInProcessApplicationCreated)
         {
-            *ppApplication = new ShuttingDownApplication(*pServer, *pHttpApplication);
+            *ppApplication = new ShuttingDownApplication(*pServer, *pHttpApplication, shadowCopyPtr);
             return S_OK;
         }
 
@@ -128,7 +129,7 @@ CreateApplication(
         errorContext.generalErrorType = "ASP.NET Core app failed to start";
         errorContext.errorReason = "<ul><li>The app failed to start</li><li>The app started but then stopped</li><li>The app started but threw an exception during startup</li></ul>";
 
-        if (!FAILED_LOG(hr = IN_PROCESS_APPLICATION::Start(*pServer, pSite, *pHttpApplication, pParameters, nParameters, inProcessApplication, errorContext)))
+        if (!FAILED_LOG(hr = IN_PROCESS_APPLICATION::Start(*pServer, pSite, *pHttpApplication, pParameters, nParameters, inProcessApplication, errorContext, shadowCopyPtr)))
         {
             *ppApplication = inProcessApplication.release();
         }
@@ -154,7 +155,8 @@ CreateApplication(
                 content,
                 errorContext.statusCode,
                 errorContext.subStatusCode,
-                "Internal Server Error");
+                "Internal Server Error",
+                shadowCopyPtr);
 
             RETURN_IF_FAILED(pErrorApplication->StartMonitoringAppOffline());
             *ppApplication = pErrorApplication.release();
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/inprocessapplication.cpp b/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/inprocessapplication.cpp
@@ -21,10 +21,9 @@ IN_PROCESS_APPLICATION::IN_PROCESS_APPLICATION(
     IHttpApplication& pApplication,
     std::unique_ptr<InProcessOptions> pConfig,
     APPLICATION_PARAMETER* pParameters,
-    DWORD                  nParameters) :
-    InProcessApplicationBase(pHttpServer,
-        pApplication,
-        FindParameter<PCWSTR>("ShadowCopyDirectory", pParameters, nParameters)),
+    DWORD                  nParameters,
+    std::wstring shadowCopy) :
+    InProcessApplicationBase(pHttpServer, pApplication, shadowCopy),
     m_Initialized(false),
     m_blockManagedCallbacks(true),
     m_waitForShutdown(true),
@@ -250,7 +249,7 @@ IN_PROCESS_APPLICATION::ExecuteApplication()
                 LOG_INFOF(L"Setting dll directory to %s", currentDirectory.c_str());
             }
 
-            // I think I still want this to be the current directory
+            // Keep as original directory
             LOG_LAST_ERROR_IF(!SetCurrentDirectory(this->QueryApplicationPhysicalPath().c_str()));
 
             LOG_INFOF(L"Setting current directory to %s", this->QueryApplicationPhysicalPath().c_str());
@@ -414,14 +413,15 @@ HRESULT IN_PROCESS_APPLICATION::Start(
     APPLICATION_PARAMETER* pParameters,
     DWORD nParameters,
     std::unique_ptr<IN_PROCESS_APPLICATION, IAPPLICATION_DELETER>& application,
-    ErrorContext& errorContext)
+    ErrorContext& errorContext,
+    std::wstring shadowCopy)
 {
     try
     {
         std::unique_ptr<InProcessOptions> options;
         THROW_IF_FAILED(InProcessOptions::Create(pServer, pSite, pHttpApplication, options));
         application = std::unique_ptr<IN_PROCESS_APPLICATION, IAPPLICATION_DELETER>(
-            new IN_PROCESS_APPLICATION(pServer, pHttpApplication, std::move(options), pParameters, nParameters));
+            new IN_PROCESS_APPLICATION(pServer, pHttpApplication, std::move(options), pParameters, nParameters, shadowCopy));
         THROW_IF_FAILED(application->LoadManagedApplication(errorContext));
         return S_OK;
     }
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/inprocessapplication.h b/src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/inprocessapplication.h
@@ -23,7 +23,8 @@ class IN_PROCESS_APPLICATION : public InProcessApplicationBase
         IHttpApplication& pApplication,
         std::unique_ptr<InProcessOptions> pConfig,
         APPLICATION_PARAMETER *pParameters,
-        DWORD                  nParameters);
+        DWORD                  nParameters,
+        std::wstring shadowCopy);
 
     ~IN_PROCESS_APPLICATION();
 
@@ -114,7 +115,8 @@ class IN_PROCESS_APPLICATION : public InProcessApplicationBase
         APPLICATION_PARAMETER* pParameters,
         DWORD nParameters,
         std::unique_ptr<IN_PROCESS_APPLICATION, IAPPLICATION_DELETER>& application,
-        ErrorContext& errorContext);
+        ErrorContext& errorContext,
+        std::wstring shadowCopy);
 
 private:
     struct ExecuteClrContext: std::enable_shared_from_this<ExecuteClrContext>
@@ -178,7 +180,6 @@ class IN_PROCESS_APPLICATION : public InProcessApplicationBase
 
     inline static const LPCSTR      s_exeLocationParameterName = "InProcessExeLocation";
 
-
     VOID
     UnexpectedThreadExit(const ExecuteClrContext& context) const;
 
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/OutOfProcessRequestHandler/dllmain.cpp b/src/Servers/IIS/AspNetCoreModuleV2/OutOfProcessRequestHandler/dllmain.cpp
@@ -252,6 +252,7 @@ CreateApplication(
 {
     TraceContextScope traceScope(FindParameter<IHttpTraceContext*>("TraceContext", pParameters, nParameters));
     auto pSite = FindParameter<IHttpSite*>("Site", pParameters, nParameters);
+    auto shadowCopy = FindParameter<PCWSTR> ("ShadowCopyDirectory", pParameters, nParameters);
 
     InitializeGlobalConfiguration(pServer);
 
@@ -261,7 +262,7 @@ CreateApplication(
 
     RETURN_IF_FAILED(EnsureOutOfProcessInitializtion(pHttpApplication));
 
-    std::unique_ptr<OUT_OF_PROCESS_APPLICATION> pApplication = std::make_unique<OUT_OF_PROCESS_APPLICATION>(*pHttpApplication, std::move(pRequestHandlerConfig));
+    std::unique_ptr<OUT_OF_PROCESS_APPLICATION> pApplication = std::make_unique<OUT_OF_PROCESS_APPLICATION>(*pHttpApplication, std::move(pRequestHandlerConfig), shadowCopy);
 
     RETURN_IF_FAILED(pApplication->Initialize());
     RETURN_IF_FAILED(pApplication->StartMonitoringAppOffline());
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/OutOfProcessRequestHandler/outprocessapplication.cpp b/src/Servers/IIS/AspNetCoreModuleV2/OutOfProcessRequestHandler/outprocessapplication.cpp
@@ -7,8 +7,9 @@
 
 OUT_OF_PROCESS_APPLICATION::OUT_OF_PROCESS_APPLICATION(
     IHttpApplication& pApplication,
-    std::unique_ptr<REQUESTHANDLER_CONFIG> pConfig) :
-    AppOfflineTrackingApplication(pApplication, std::wstring()),
+    std::unique_ptr<REQUESTHANDLER_CONFIG> pConfig,
+    std::wstring shadowCopy) :
+    AppOfflineTrackingApplication(pApplication, shadowCopy),
     m_fWebSocketSupported(WEBSOCKET_STATUS::WEBSOCKET_UNKNOWN),
     m_pConfig(std::move(pConfig))
 {
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/OutOfProcessRequestHandler/outprocessapplication.h b/src/Servers/IIS/AspNetCoreModuleV2/OutOfProcessRequestHandler/outprocessapplication.h
@@ -17,7 +17,8 @@ class OUT_OF_PROCESS_APPLICATION : public AppOfflineTrackingApplication
 public:
     OUT_OF_PROCESS_APPLICATION(
         IHttpApplication& pApplication,
-        std::unique_ptr<REQUESTHANDLER_CONFIG> pConfig);
+        std::unique_ptr<REQUESTHANDLER_CONFIG> pConfig,
+        std::wstring shadowCopy);
 
     __override
     ~OUT_OF_PROCESS_APPLICATION() override;
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/RequestHandlerLib/AppOfflineTrackingApplication.cpp b/src/Servers/IIS/AspNetCoreModuleV2/RequestHandlerLib/AppOfflineTrackingApplication.cpp
@@ -45,7 +45,6 @@ HRESULT AppOfflineTrackingApplication::StartMonitoringAppOflineImpl()
     RETURN_IF_FAILED(m_fileWatcher->Create(m_applicationPath.c_str(),
         L"app_offline.htm",
         /* fTrackDllChanges */ !m_shadowCopyDirectory.empty(),
-        m_shadowCopyDirectory,
         this));
 
     return S_OK;
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/RequestHandlerLib/filewatcher.cpp b/src/Servers/IIS/AspNetCoreModuleV2/RequestHandlerLib/filewatcher.cpp
@@ -18,7 +18,7 @@ FILE_WATCHER::FILE_WATCHER() :
 FILE_WATCHER::~FILE_WATCHER()
 {
     StopMonitor();
-    WaitForMonitor(20);
+    WaitForMonitor(100);
 }
 
 void FILE_WATCHER::WaitForMonitor(DWORD dwRetryCounter)
@@ -59,11 +59,9 @@ FILE_WATCHER::Create(
     _In_ PCWSTR                  pszDirectoryToMonitor,
     _In_ PCWSTR                  pszFileNameToMonitor,
     _In_ bool                    fTrackDllChanges,
-    _In_ std::wstring            shadowCopyPath,
     _In_ AppOfflineTrackingApplication *pApplication
 )
 {
-    _shadowCopyPath = shadowCopyPath;
     RETURN_LAST_ERROR_IF_NULL(m_hCompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0));
 
     RETURN_LAST_ERROR_IF_NULL(m_hChangeNotificationThread = CreateThread(NULL,
@@ -288,15 +286,12 @@ HRESULT
     {
         // Reference application before
         _pApplication->ReferenceApplication();
-        if (fDllChanged)
-        {
-            // wait for all file changes to complete
-            PostQueuedCompletionStatus(m_hCompletionPort, 0, FILE_WATCHER_SHUTDOWN_KEY, NULL);
-            WaitForMonitor(100); // 5 seconds here.
+        //if (fDllChanged)
+        //{
+        //    // wait for all file changes to complete
+        //    PostQueuedCompletionStatus(m_hCompletionPort, 0, FILE_WATCHER_SHUTDOWN_KEY, NULL);
+        //}
 
-            // Copy contents before shutdown
-            RETURN_IF_FAILED(Environment::CopyToDirectory(_shadowCopyPath, _strDirectoryName.QueryStr(), false));
-        }
         RETURN_LAST_ERROR_IF(!QueueUserWorkItem(RunNotificationCallback, _pApplication.get(), WT_EXECUTEDEFAULT));
     }
 
diff --git a/src/Servers/IIS/AspNetCoreModuleV2/RequestHandlerLib/filewatcher.h b/src/Servers/IIS/AspNetCoreModuleV2/RequestHandlerLib/filewatcher.h
@@ -29,7 +29,6 @@ class FILE_WATCHER{
         _In_ PCWSTR                  pszDirectoryToMonitor,
         _In_ PCWSTR                  pszFileNameToMonitor,
         _In_ bool                    fTrackDllChanges,
-        _In_ std::wstring            shadowCopyPath,
         _In_ AppOfflineTrackingApplication *pApplication
     );
 
@@ -58,7 +57,6 @@ class FILE_WATCHER{
     STRU                    _strFullName;
     LONG                    _lStopMonitorCalled {};
     bool                    _fTrackDllChanges;
-    std::wstring            _shadowCopyPath;
     OVERLAPPED              _overlapped;
     std::unique_ptr<AppOfflineTrackingApplication, IAPPLICATION_DELETER> _pApplication;
 };

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@ namespace InprocessTests`
`39`	`39`	`{"InProcessExeLocation", exePath.data()}`
`40`	`40`	`};`
`41`	`41`
`42`		`- IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), parameters.data(), 1);`
	`42`	`+ IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), parameters.data(), 1, std::wstring());`
`43`	`43`
`44`	`44`	`ASSERT_STREQ(app->QueryExeLocation().c_str(), L"hello");`
`45`	`45`	`}`
`@@ -59,7 +59,7 @@ namespace InprocessTests`
`59`	`59`	`.WillByDefault(testing::Return(L""));`
`60`	`60`
`61`	`61`	`auto requestHandlerConfig = std::unique_ptr<InProcessOptions>(MockInProcessOptions::CreateConfig());`
`62`		`- IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), nullptr, 0);`
	`62`	`+ IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), nullptr, 0, std::wstring());`
`63`	`63`
`64`	`64`	`ASSERT_STREQ(app->QueryApplicationVirtualPath().c_str(), L"/SECTION5");`
`65`	`65`	`}`
`@@ -79,7 +79,7 @@ namespace InprocessTests`
`79`	`79`	`.WillByDefault(testing::Return(L""));`
`80`	`80`
`81`	`81`	`auto requestHandlerConfig = std::unique_ptr<InProcessOptions>(MockInProcessOptions::CreateConfig());`
`82`		`- IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), nullptr, 0);`
	`82`	`+ IN_PROCESS_APPLICATION *app = new IN_PROCESS_APPLICATION(server, application, std::move(requestHandlerConfig), nullptr, 0, std::wstring());`
`83`	`83`
`84`	`84`	`ASSERT_STREQ(app->QueryApplicationVirtualPath().c_str(), L"/");`
`85`	`85`	`}`
Original file line number	Diff line number	Diff line change
`@@ -31,8 +31,8 @@ class ShuttingDownHandler : public REQUEST_HANDLER`
`31`	`31`	`class ShuttingDownApplication : public InProcessApplicationBase`
`32`	`32`	`{`
`33`	`33`	`public:`
`34`		`- ShuttingDownApplication(IHttpServer& pHttpServer, IHttpApplication& pHttpApplication)`
`35`		`- : InProcessApplicationBase(pHttpServer, pHttpApplication, std::wstring())`
	`34`	`+ ShuttingDownApplication(IHttpServer& pHttpServer, IHttpApplication& pHttpApplication, std::wstring shadowCopy)`
	`35`	`+ : InProcessApplicationBase(pHttpServer, pHttpApplication, shadowCopy)`
`36`	`36`	`{`
`37`	`37`	`}`
`38`	`38`
Original file line number	Diff line number	Diff line change
`@@ -103,6 +103,7 @@ CreateApplication(`
`103`	`103`	`{`
`104`	`104`	`TraceContextScope traceScope(FindParameter<IHttpTraceContext*>("TraceContext", pParameters, nParameters));`
`105`	`105`	`const auto pSite = FindParameter<IHttpSite*>("Site", pParameters, nParameters);`
	`106`	`+ const auto shadowCopyPtr = FindParameter<PCWSTR>("ShadowCopyDirectory", pParameters, nParameters);`
`106`	`107`
`107`	`108`	`try`
`108`	`109`	`{`
`@@ -113,7 +114,7 @@ CreateApplication(`
`113`	`114`	`// means that server is shutting does and request arrived in the meantime`
`114`	`115`	`if (g_fInProcessApplicationCreated)`
`115`	`116`	`{`
`116`		`- ppApplication = new ShuttingDownApplication(pServer, *pHttpApplication);`
	`117`	`+ ppApplication = new ShuttingDownApplication(pServer, *pHttpApplication, shadowCopyPtr);`
`117`	`118`	`return S_OK;`
`118`	`119`	`}`
`119`	`120`
`@@ -128,7 +129,7 @@ CreateApplication(`
`128`	`129`	`errorContext.generalErrorType = "ASP.NET Core app failed to start";`
`129`	`130`	`errorContext.errorReason = "<ul><li>The app failed to start</li><li>The app started but then stopped</li><li>The app started but threw an exception during startup</li></ul>";`
`130`	`131`
`131`		`- if (!FAILED_LOG(hr = IN_PROCESS_APPLICATION::Start(pServer, pSite, pHttpApplication, pParameters, nParameters, inProcessApplication, errorContext)))`
	`132`	`+ if (!FAILED_LOG(hr = IN_PROCESS_APPLICATION::Start(pServer, pSite, pHttpApplication, pParameters, nParameters, inProcessApplication, errorContext, shadowCopyPtr)))`
`132`	`133`	`{`
`133`	`134`	`*ppApplication = inProcessApplication.release();`
`134`	`135`	`}`
`@@ -154,7 +155,8 @@ CreateApplication(`
`154`	`155`	`content,`
`155`	`156`	`errorContext.statusCode,`
`156`	`157`	`errorContext.subStatusCode,`
`157`		`- "Internal Server Error");`
	`158`	`+ "Internal Server Error",`
	`159`	`+ shadowCopyPtr);`
`158`	`160`
`159`	`161`	`RETURN_IF_FAILED(pErrorApplication->StartMonitoringAppOffline());`
`160`	`162`	`*ppApplication = pErrorApplication.release();`