def console(): """ Console for managing users (add, remove, change password) """ def useradd(username, password): storage.add_user(username, password) def userdel(username): storage.del_user(username) def chpass(username, password): storage.change_key(username, password) banner = "\n".join([ "Usage:", "========", "useradd(username, password) - add user", "userdel(username) - remove user", "chpass(username, password) - change passphrase", "exit() or ^D - exit" ]) DB.auth_module.register_auth() DB.encrypter.register_class(default=True) init_crypto(passphrase=_passphrase) storage = client_storage(_sock, username=_username, password=_passphrase, realm="ZERO") embed(banner1=banner)
def run(username, passphrase, sock): logging.basicConfig() username = str(username) passphrase = str(passphrase) sock = str(sock) if not sock.startswith("/"): sock = (sock.split(":")[0], int(sock.split(":")[1])) DB.auth_module.register_auth() DB.encrypter.register_class(default=True) init_crypto(passphrase=passphrase) def useradd(username, password): storage.add_user(username, password) def userdel(username): storage.del_user(username) def chpass(username, password): storage.change_key(username, password) print "Usage:" print "========" print "useradd(username, password) - add user" print "userdel(username) - remove user" print "chpass(username, password) - change passphrase" print "exit() or ^D - exit" storage = client_storage(sock, username=username, password=passphrase, realm="ZERO") embed(display_banner=False)
def test_utilities(): AES256Encrypter.register_class(default=True) init_crypto(passphrase="Hello world") test_text = b"Test text " * 100 encrypted_text = encrypt(test_text) assert encrypted_text.startswith(b".e") assert decrypt(encrypted_text) == test_text
def _init_default_crypto(self, **kw): encrypters = self.encrypter if not isinstance(encrypters, (list, tuple)): encrypters = [self.encrypter] elif not encrypters: encrypters = [] if encrypters: encrypters[0].register_class(default=True) for e in encrypters[1:]: e.register_class(default=False) if self.compressor: self.compressor.register(default=True) init_crypto(**kw)
def _init_default_crypto(self, passphrase=None): encrypters = self.encrypter if not isinstance(encrypters, (list, tuple)): encrypters = [self.encrypter] elif not encrypters: encrypters = [] if encrypters: encrypters[0].register_class(default=True) for e in encrypters[1:]: e.register_class(default=False) if self.compressor: self.compressor.register(default=True) init_crypto(passphrase=passphrase)
def _init_default_crypto(self, passphrase=None): if self.encrypter: self.encrypter.register_class(default=True) if self.compressor: self.compressor.register(default=True) init_crypto(passphrase=passphrase)