b/198398480 Handle VM lookup failures (#558)

Map VM lookup failures like access failures and theow an UnauthorizedException.
GoogleCloudPlatform · Sep 1, 2021 · 0bbca5d · 0bbca5d
1 parent fe847c0
commit 0bbca5d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/sources/Google.Solutions.IapTunneling/Iap/Messages.cs b/sources/Google.Solutions.IapTunneling/Iap/Messages.cs
@@ -97,7 +97,9 @@ public enum CloseCode : int
         DESTINATION_READ_FAILED = 4010,
 
         INVALID_DATA = 4013,
-        NOT_AUTHORIZED = 4033
+        NOT_AUTHORIZED = 4033,
+        LOOKUP_FAILED = 4047,
+        LOOKUP_FAILED_RECONNECT = 4051
     }
 
     /// <summary>

diff --git a/sources/Google.Solutions.IapTunneling/Iap/SshRelayStream.cs b/sources/Google.Solutions.IapTunneling/Iap/SshRelayStream.cs
@@ -252,7 +252,9 @@ public async Task TestConnectionAsync(TimeSpan timeout)
                 }
             }
             catch (WebSocketStreamClosedByServerException e)
-                when ((CloseCode)e.CloseStatus == CloseCode.NOT_AUTHORIZED)
+                when ((CloseCode)e.CloseStatus == CloseCode.NOT_AUTHORIZED ||
+                      (CloseCode)e.CloseStatus == CloseCode.LOOKUP_FAILED ||
+                      (CloseCode)e.CloseStatus == CloseCode.LOOKUP_FAILED_RECONNECT)
             {
                 //
                 // Request was rejected by access level or IAM policy.