From 2fa09dbd95d092a58e6935f48586d1651d6bbfc0 Mon Sep 17 00:00:00 2001
From: Geoffrey McRae <geoff@hostfission.com>
Date: Thu, 5 Aug 2021 21:43:14 +1000
Subject: [PATCH] [client] allow the user to upgrade if the LGMP version
 doesn't match

---
 client/src/main.c | 28 +++++++++++++++++++++++-----
 1 file changed, 23 insertions(+), 5 deletions(-)

diff --git a/client/src/main.c b/client/src/main.c
index 614b2908..7a72bf61 100644
--- a/client/src/main.c
+++ b/client/src/main.c
@@ -796,6 +796,14 @@ static bool tryRenderer(const int index, const LG_RendererParams lgrParams,
   return true;
 }
 
+static void reportBadVersion()
+{
+  DEBUG_BREAK();
+  DEBUG_ERROR("The host application is not compatible with this client");
+  DEBUG_ERROR("This is not a Looking Glass error, do not report this");
+  DEBUG_ERROR("Please install the matching host application for this client");
+}
+
 static int lg_run(void)
 {
   g_cursor.sens = g_params.mouseSens;
@@ -1051,6 +1059,20 @@ restart:
     if ((status = lgmpClientSessionInit(g_state.lgmp, &udataSize, (uint8_t **)&udata)) == LGMP_OK)
       break;
 
+    if (status == LGMP_ERR_INVALID_VERSION)
+    {
+      reportBadVersion();
+      DEBUG_INFO("Waiting for you to upgrade the host application");
+      while (g_state.state == APP_STATE_RUNNING &&
+          lgmpClientSessionInit(g_state.lgmp, &udataSize, (uint8_t **)&udata) != LGMP_OK)
+        g_state.ds->wait(1000);
+
+      if (g_state.state != APP_STATE_RUNNING)
+        return -1;
+
+      status = LGMP_OK;
+    }
+
     if (status != LGMP_ERR_INVALID_SESSION && status != LGMP_ERR_INVALID_MAGIC)
     {
       DEBUG_ERROR("lgmpClientSessionInit Failed: %s", lgmpStatusString(status));
@@ -1084,11 +1106,7 @@ restart:
   const bool magicMatches = memcmp(udata->magic, KVMFR_MAGIC, sizeof(udata->magic)) == 0;
   if (udataSize != sizeof(KVMFR) || !magicMatches || udata->version != KVMFR_VERSION)
   {
-    DEBUG_BREAK();
-    DEBUG_ERROR("The host application is not compatible with this client");
-    DEBUG_ERROR("This is not a Looking Glass error, do not report this");
-    DEBUG_ERROR("Please install the matching host application for this client");
-
+    reportBadVersion();
     if (magicMatches)
     {
       DEBUG_ERROR("Expected KVMFR version %d, got %d", KVMFR_VERSION, udata->version);