Merge pull request #2743 from andoriyaprashant/branch31

Multi-Factor Authentication Generator script added

Author: avinashkranjan

Multi-Factor Authentication (MFA) Generator/README.md

@@ -0,0 +1,83 @@
+# Multi-Factor Authentication (MFA) Generator Script
+
+This is a Python script that provides a command-line interface for generating and verifying Time-Based One-Time Passwords (TOTPs) for Multi-Factor Authentication (MFA). The script uses the `pyotp` library for TOTP calculations.
+
+## Features
+
+- Generate TOTP codes using different hash algorithms, code lengths, and time intervals.
+- Verify TOTP codes and set a verification window for code validity.
+- Save and load secret keys from JSON files for easier management.
+
+## Prerequisites
+
+Before using the script, make sure you have Python installed. You can install the required `pyotp` library using the following command:
+
+```bash
+pip install pyotp
+```
+
+## Usage
+
+1. Generate TOTP Code
+
+To generate a TOTP code, run:
+
+```bash
+python multi_factor_auth.py --generate --algorithm SHA1 --digits 6 --interval 30
+```
+
+2. Verify TOTP Code
+
+To verify a TOTP code, run:
+
+```bash
+python multi_factor_auth.py --verify CODE_TO_VERIFY --algorithm SHA1 --digits 6 --interval 30 --window 1
+```
+
+Replace CODE_TO_VERIFY with the TOTP code you want to verify.
+
+3. Save and Load Secret Keys
+
+To save a secret key to a JSON file, run:
+
+```bash
+python multi_factor_auth.py --save secret.json
+```
+
+To load a secret key from a JSON file, run:
+```bash
+python multi_factor_auth.py --load
+```
+
+## Options
+
+- --generate: Generate a TOTP code.
+- --verify: Verify a TOTP code.
+- --algorithm: Choose hash algorithm (SHA1, SHA256, SHA512).
+- --digits: Set number of digits in TOTP code (6 or 8).
+- --interval: Set time interval for TOTP code generation.
+- --window: Set verification window for TOTP codes.
+- --save: Save secret key to a JSON file.
+- --load: Load secret key from a JSON file.
+
+## Example
+
+Generate a TOTP code:
+
+```bash
+python multi_factor_auth.py --generate --algorithm SHA256 --digits 8 --interval 60
+```
+
+Verify a TOTP code:
+
+```bash
+python multi_factor_auth.py --verify 123456 --algorithm SHA256 --digits 8 --interval 60 --window 2
+```
+
+## Note
+
+This script is provided for educational purposes. When using it for real-world applications, make sure to follow security best practices and ensure proper handling of secret keys.
+
+## Contributing
+
+If you have any ideas, improvements, or bug fixes, feel free to open an issue or submit a pull request. We appreciate your contributions!

Multi-Factor Authentication (MFA) Generator/multi_factor_auth.py

@@ -0,0 +1,69 @@
+import pyotp
+import argparse
+import json
+
+SECRET_STORAGE_FILE = "secret.json"
+
+def generate_totp(secret_key, algorithm='SHA1', digits=6, interval=30):
+    totp = pyotp.TOTP(secret_key, digits=digits, interval=interval, digest=algorithm)
+    return totp.now()
+
+def verify_totp(secret_key, code, algorithm='SHA1', digits=6, interval=30, window=1):
+    totp = pyotp.TOTP(secret_key, digits=digits, interval=interval, digest=algorithm)
+    return totp.verify(code, valid_window=window)
+
+def save_secret(secret_key, filename):
+    data = {'secret_key': secret_key}
+    with open(filename, 'w') as file:
+        json.dump(data, file)
+    return f'Secret key saved to {filename}'
+
+def load_secret(filename):
+    try:
+        with open(filename, 'r') as file:
+            data = json.load(file)
+            return data['secret_key']
+    except FileNotFoundError:
+        print(f'Secret key file "{filename}" not found.')
+        return None
+
+def main():
+    parser = argparse.ArgumentParser(description='Multi-Factor Authentication (MFA) Generator')
+    parser.add_argument('--generate', action='store_true', help='Generate a TOTP code')
+    parser.add_argument('--verify', help='Verify a TOTP code')
+    parser.add_argument('--algorithm', choices=['SHA1', 'SHA256', 'SHA512'], default='SHA1', help='Hash algorithm for TOTP')
+    parser.add_argument('--digits', type=int, choices=[6, 8], default=6, help='Number of digits in TOTP code')
+    parser.add_argument('--interval', type=int, default=30, help='Time interval for TOTP code generation')
+    parser.add_argument('--window', type=int, default=1, help='Verification window for TOTP codes')
+    parser.add_argument('--save', help='Save secret key to a file')
+    parser.add_argument('--load', action='store_true', help='Load secret key from a file')
+    args = parser.parse_args()
+
+    if args.load:
+        secret_key = load_secret(SECRET_STORAGE_FILE)
+        if secret_key is None:
+            return
+    elif args.generate or args.verify:
+        secret_key = input('Enter your secret key: ').strip()
+        if args.save:
+            save_result = save_secret(secret_key, args.save)
+            print(save_result)
+    else:
+        print('Please specify either --generate or --verify.')
+        return
+
+    if args.generate:
+        code = generate_totp(secret_key, args.algorithm, args.digits, args.interval)
+        print(f'Generated TOTP code: {code}')
+
+    if args.verify:
+        code_to_verify = args.verify
+        result = verify_totp(secret_key, code_to_verify, args.algorithm, args.digits, args.interval, args.window)
+        if result:
+            print('TOTP code is valid.')
+        else:
+            print('TOTP code is NOT valid.')
+
+if __name__ == '__main__':
+    main()
+

Multi-Factor Authentication (MFA) Generator/requirements.txt

@@ -0,0 +1,2 @@
+pyotp
+argparse