From 9867d0c92cd8522c7d8083e1e71f34206e068071 Mon Sep 17 00:00:00 2001
From: Evan Pratten <evan@ewpratten.com>
Date: Fri, 17 Nov 2023 10:58:12 -0500
Subject: [PATCH] Add a fix for bad TOTP codes

---
 scripts/guru-vpn | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/guru-vpn b/scripts/guru-vpn
index b974c6b..fa5c189 100755
--- a/scripts/guru-vpn
+++ b/scripts/guru-vpn
@@ -71,6 +71,9 @@ def handle_connect(args: argparse.Namespace) -> int:
     # Fetch the credentials from the Yubikey
     oath_code = get_oath_code("Guru")
     print(f"Using OATH code: {oath_code}")
+    if not len(str(oath_code)) == 6:
+        print("Invalid OATH code length. Try again in a minute.", file=sys.stderr)
+        return 1
 
     # Construct the one-time password
     password = f"{base_password}{oath_code}"