Added Biometric Authentication to unlock the PwManager. Closes IMGIIT…

…Roorkee#28
Akshat1276 · Jan 5, 2025 · 5f66d36 · 5f66d36
1 parent bb75c6a
commit 5f66d36
Showing 1 changed file with 125 additions and 64 deletions.
diff --git a/main.py b/main.py
@@ -1,10 +1,9 @@
 from manager import PasswordManager
 import pyperclip
+import platform
 import sys
 import select
 import time
-
-
 def get_input_with_timeout(prompt, timeout=60):
     print(prompt, end='', flush=True)
     # For Windows
@@ -40,76 +39,138 @@ def validate_key_loaded(pm : PasswordManager):
         return False
     return True
 
+# Windows:
+# Install the winbio library, which is a Python wrapper for the Windows Biometric Framework.
+# Ensure your fingerprint device is configured in Windows Hello.
+
+# macOS:
+# Install pyobjc (pip install pyobjc).
+# This script uses the LocalAuthentication framework via PyObjC.
+
+def authenticate_windowsHello():
+    try:
+        import winbio  # Windows Biometric Framework library
+        bio = winbio.WinBio()
+        print("Place your finger on the fingerprint reader...")
+        result = bio.verify()
+        if result:
+            print("Authentication successful!")
+            return True
+        else:
+            print("Authentication failed.")
+            return False
+    except ImportError:
+        print("Windows Biometric Framework (winbio) is not installed.")
+        return False
+
+def authenticate_touchId():
+    try:
+        import objc
+        from Foundation import NSBundle
+        from LocalAuthentication import LAContext, LAPolicyDeviceOwnerAuthenticationWithBiometrics
+
+        context = LAContext.alloc().init()
+        if context.canEvaluatePolicy_error_(LAPolicyDeviceOwnerAuthenticationWithBiometrics, None)[0]:
+            success, error = context.evaluatePolicy_localizedReason_reply_(
+                LAPolicyDeviceOwnerAuthenticationWithBiometrics,
+                "Authenticate to proceed.",
+                None
+            )
+            if success:
+                print("Authentication successful!")
+                return True
+            else:
+                print("Authentication failed.")
+                return False
+        else:
+            print("Touch ID is not available on this device.")
+            return False
+    except ImportError:
+        print("PyObjC is not installed.")
+        return False
+
 def main():
-    password = {
+    system = platform.system()
+    if system == "Windows":
+        authenticated = authenticate_windowsHello()
+    elif system == "Darwin":  # macOS
+        authenticated = authenticate_touchId()
+    else:
+        print(f"Unsupported operating system: {system}")
+        sys.exit(1)
+
+    if authenticated:
+        print("Running the main code...")
+        password = {
         "gmail": "password1",
         "facebook": "password2",
         "twitter": "password3"
-    }
+        }
 
-    pm = PasswordManager()
-
-    print("""What would you like to do?
-          1. Create a new key
-          2. Load an existing key
-          3. Create a new password file
-          4. Load an existing password file
-          5. Add a password
-          6. Get a password
-          7. List all sites
-          q. Quit
-          """)
-
-    done = False
-    while not done:
-        choice = get_input_with_timeout("Enter choice: ")
-        if choice is None:  # Timeout occurred
-            done = True
-            continue
-
-        choice = choice.strip().lower()
-        if choice == '1':
-            path = input("Enter key file path: ").strip()
-            pm.create_key(path)
-        elif choice == '2':
-            path = input("Enter key file path: ").strip()
-            pm.load_key(path)
-        elif choice == '3' and validate_key_loaded(pm):
-            path = input("Enter password file path: ").strip()
-            pm.create_password_file(path, password)
-        elif choice == '4' and validate_key_loaded(pm):
-            path = input("Enter password file path: ").strip()
-            pm.load_password_file(path)
-        elif choice == '5' and validate_key_loaded(pm):
-            site = input("Enter site: ").strip()
-            password = input("Enter password: ").strip()
-            if pm.validate_strength(password):
-                print("added successfully")
-            else:
-                print("WARNING: This password is weak, It is recommended to set a stronger password")
-                print("- Password should be more than 8 characters long")
-                print("- Password should have alphanumeric characters, capital letters and special characters")
-            pm.add_password(site, password)
+        pm = PasswordManager()
 
-        elif choice == '6' and validate_key_loaded(pm):
+        print("""What would you like to do?
+            1. Create a new key
+            2. Load an existing key
+            3. Create a new password file
+            4. Load an existing password file
+            5. Add a password
+            6. Get a password
+            7. List all sites
+            q. Quit
+            """)
+
+        done = False
+        while not done:
+            choice = get_input_with_timeout("Enter choice: ")
+            if choice is None:  # Timeout occurred
+                done = True
+                continue
+
+            choice = choice.strip().lower()
+            if choice == '1':
+                path = input("Enter key file path: ").strip()
+                pm.create_key(path)
+            elif choice == '2':
+                path = input("Enter key file path: ").strip()
+                pm.load_key(path)
+            elif choice == '3' and validate_key_loaded(pm):
+                path = input("Enter password file path: ").strip()
+                pm.create_password_file(path, password)
+            elif choice == '4' and validate_key_loaded(pm):
+                path = input("Enter password file path: ").strip()
+                pm.load_password_file(path)
+            elif choice == '5' and validate_key_loaded(pm):
+                site = input("Enter site: ").strip()
+                password = input("Enter password: ").strip()
+                if pm.validate_strength(password):
+                    print("added successfully")
+                else:
+                    print("WARNING: This password is weak, It is recommended to set a stronger password")
+                    print("- Password should be more than 8 characters long")
+                    print("- Password should have alphanumeric characters, capital letters and special characters")
+                pm.add_password(site, password)
 
-            site = input("Enter site: ").strip()
-            res = pm.get_password(site)
-            print(f"Password for {site}: {res}")
-            if(res != "Password not found."):
-                pyperclip.copy(pm.get_password(site))
-                print("Password copied to clipboard.")
-
-        elif choice == '7':
-            print("Saved Sites:")
-            for site in pm.password_dict:
-                print(site)
-        elif choice == 'q':
-            done = True
-            print("Goodbye!")
-        else:
-            print("Invalid choice. Please try again.")
+            elif choice == '6' and validate_key_loaded(pm):
 
+                site = input("Enter site: ").strip()
+                res = pm.get_password(site)
+                print(f"Password for {site}: {res}")
+                if(res != "Password not found."):
+                    pyperclip.copy(pm.get_password(site))
+                    print("Password copied to clipboard.")
 
+            elif choice == '7':
+                print("Saved Sites:")
+                for site in pm.password_dict:
+                    print(site)
+            elif choice == 'q':
+                done = True
+                print("Goodbye!")
+            else:
+                print("Invalid choice. Please try again.")
+    else:
+        print("Exiting due to failed authentication.")
+
 if __name__ == '__main__':
     main()