1 files changed, 9 insertions, 2 deletions

diff --git a/pam/pam.py b/pam/pam_mfa.py
index 5cb9f4d..85d0a82 100755
--- a/pam/pam.py
+++ b/pam/pam_mfa.py
@@ -19,6 +19,10 @@ RESPONSE_LENGTH = 1
 PAM_SUCCESS = 0
 PAM_AUTH_ERR = 7
 
+def die(msg):
+    print(msg)
+    sys.exit(1)
+
 def parse_arguments():
     # Parse command line arguments
     parser = argparse.ArgumentParser()
@@ -140,10 +144,13 @@ def main():
 
     # Listen for response from MFA server
     # Response will be 0 for authenticated and 1 for denied
-    response = int(connection.recv(RESPONSE_LENGTH).decode(FORMAT))
+    response = connection.recv(RESPONSE_LENGTH).decode(FORMAT)
+    if response == "":
+        # lost connection to server
+        response = 1
 
     # Print success/failure for PAM module
-    print(str(response))
+    print(response)
 
 if __name__ == '__main__':
     main()