You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.4 KiB
75 lines
2.4 KiB
#!/usr/bin/env python
|
|
#
|
|
# Password manager: handles hashing and comparing for passwords used in LikedSavedDownloaderServer
|
|
# I'm no expert so use and trust at your own risk!
|
|
#
|
|
from passlib.context import CryptContext
|
|
|
|
# It's not strictly necessary to import these, but I do it here for PyInstaller
|
|
# (see https://github.com/pyinstaller/pyinstaller/issues/649)
|
|
import argon2
|
|
import cffi
|
|
import configparser
|
|
import passlib.handlers
|
|
import passlib.handlers.argon2
|
|
import passlib.handlers.sha2_crypt
|
|
import passlib.handlers.bcrypt
|
|
|
|
import sys
|
|
import os
|
|
|
|
# Even if this file gets compromised, it'll still be hard to use for anything
|
|
passwordsFilename = "passwords.txt"
|
|
|
|
password_context = CryptContext(
|
|
# Replace this list with the hash(es) you wish to support.
|
|
# this example sets pbkdf2_sha256 as the default,
|
|
# with additional support for reading legacy des_crypt hashes.
|
|
schemes=["argon2", "sha512_crypt", "bcrypt"],
|
|
|
|
# Automatically mark all but first hasher in list as deprecated.
|
|
# (this will be the default in Passlib 2.0)
|
|
deprecated="auto",
|
|
|
|
# Optionally, set the number of rounds that should be used.
|
|
# Appropriate values may vary for different schemes,
|
|
# and the amount of time you wish it to take.
|
|
# Leaving this alone is usually safe, and will use passlib's defaults.
|
|
## pbkdf2_sha256__rounds = 29000,
|
|
)
|
|
|
|
passwords = []
|
|
def cachePasswords():
|
|
global passwords
|
|
passwordsFile = open(passwordsFilename, "r")
|
|
passwords = passwordsFile.readlines()
|
|
passwordsFile.close()
|
|
|
|
def havePasswordsBeenSet():
|
|
return os.path.exists(passwordsFilename)
|
|
|
|
def verify(password):
|
|
if not len(passwords):
|
|
cachePasswords()
|
|
if not len(passwords):
|
|
raise Exception("Tried to verify a password, but {} has no passwords or does not exist!")
|
|
|
|
for storedPassword in passwords:
|
|
if password_context.verify(password, storedPassword[:-1]):
|
|
return True
|
|
return False
|
|
|
|
def createPassword(password):
|
|
passwordHashed = password_context.hash(password)
|
|
|
|
passwordsOut = open(passwordsFilename, "a")
|
|
passwordsOut.write(passwordHashed + "\n")
|
|
passwordsOut.close()
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) != 2:
|
|
print("Wrong number of arguments!\n"
|
|
"PasswordManager: Adds a password to the passwords file.\n"
|
|
"Usage:\n python PasswordManager.py \"your password\"")
|
|
else:
|
|
createPassword(sys.argv[1])
|