From c40174034c0438b0c99c42e0a54d90a48c8761cf Mon Sep 17 00:00:00 2001
From: Cameron Gutman <aicommander@gmail.com>
Date: Wed, 5 Dec 2018 20:17:00 -0800
Subject: [PATCH] Don't kill the app on the host when the session terminates
 unexpectedly

---
 app/streaming/session.cpp | 15 +++++++++++----
 app/streaming/session.h   |  1 +
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/app/streaming/session.cpp b/app/streaming/session.cpp
index 5a3f316d..3a123b56 100644
--- a/app/streaming/session.cpp
+++ b/app/streaming/session.cpp
@@ -73,6 +73,7 @@ void Session::clStageFailed(int stage, long errorCode)
 
 void Session::clConnectionTerminated(long errorCode)
 {
+    s_ActiveSession->m_ConnectionTerminated = true;
     emit s_ActiveSession->displayLaunchError("Connection terminated");
 
     SDL_LogError(SDL_LOG_CATEGORY_APPLICATION,
@@ -287,6 +288,7 @@ Session::Session(NvComputer* computer, NvApp& app, StreamingPreferences *prefere
       m_DisplayOriginX(0),
       m_DisplayOriginY(0),
       m_PendingWindowedTransition(false),
+      m_ConnectionTerminated(false),
       m_OpusDecoder(nullptr),
       m_AudioRenderer(nullptr),
       m_AudioSampleCount(0)
@@ -573,19 +575,24 @@ private:
 
     void run() override
     {
+        // Only quit the running app if our session terminated gracefully
+        bool shouldQuit =
+                !m_Session->m_ConnectionTerminated &&
+                m_Session->m_Preferences->quitAppAfter;
+
         // Notify the UI
-        if (!m_Session->m_Preferences->quitAppAfter) {
-            emit m_Session->sessionFinished();
+        if (shouldQuit) {
+            emit m_Session->quitStarting();
         }
         else {
-            emit m_Session->quitStarting();
+            emit m_Session->sessionFinished();
         }
 
         // Finish cleanup of the connection state
         LiStopConnection();
 
         // Perform a best-effort app quit
-        if (m_Session->m_Preferences->quitAppAfter) {
+        if (shouldQuit) {
             NvHTTP http(m_Session->m_Computer->activeAddress);
 
             // Logging is already done inside NvHTTP
diff --git a/app/streaming/session.h b/app/streaming/session.h
index 965fc4ec..629909c1 100644
--- a/app/streaming/session.h
+++ b/app/streaming/session.h
@@ -120,6 +120,7 @@ private:
     int m_DisplayOriginX;
     int m_DisplayOriginY;
     bool m_PendingWindowedTransition;
+    bool m_ConnectionTerminated;
 
     int m_ActiveVideoFormat;
     int m_ActiveVideoWidth;