From a71427cfcefd9c92af7117153de3a6e055981677 Mon Sep 17 00:00:00 2001 From: LV-426 Date: Tue, 13 Jan 2015 16:57:09 +0200 Subject: add_user_login(): refactoring out of cryp_util (not needed in production) --- src/local/rz_cli_tool.py | 14 +++++++++++++- src/server/crypt_util.py | 12 ------------ 2 files changed, 13 insertions(+), 13 deletions(-) (limited to 'src') diff --git a/src/local/rz_cli_tool.py b/src/local/rz_cli_tool.py index 1dea8559..93cc132a 100644 --- a/src/local/rz_cli_tool.py +++ b/src/local/rz_cli_tool.py @@ -8,6 +8,18 @@ from rz_server import init_config def init_pw_db(cfg, user_pw_list_file): + + def add_user_login(htpasswd_path, salt, u, p): + with open(htpasswd_path, 'rb') as f: + data = f.read() + pw_db = pickle.loads(data) + + with open(htpasswd_path, 'wb') as f: + pw_db[u] = hash_pw(str(p), salt) + pickle.dump(pw_db, f) + + log.info('htpasswd db: added entry: user: %s, pw: %s...' % (u, pw_db[u][:5])) + htpasswd_path = cfg.htpasswd_path if os.path.exists(htpasswd_path): @@ -32,7 +44,7 @@ def init_pw_db(cfg, user_pw_list_file): raise Exception('failed to parse user,pw line: ' + line) user, pw = map(str.strip, kv_arr) - crypt_util.add_user_login(htpasswd_path, salt, user, pw) + add_user_login(htpasswd_path, salt, user, pw) print('added user: u: ' + user) u_count = u_count + 1 print('htpasswd generated: path: %s, user-count: %d' % (htpasswd_path, u_count)) diff --git a/src/server/crypt_util.py b/src/server/crypt_util.py index 7eec4c57..a3b2582d 100644 --- a/src/server/crypt_util.py +++ b/src/server/crypt_util.py @@ -5,18 +5,6 @@ import pickle log = logging.getLogger('rhizi') -def add_user_login(htpasswd_path, salt, u, p): - - with open(htpasswd_path, 'rb') as f: - data = f.read() - pw_db = pickle.loads(data) - - with open(htpasswd_path, 'wb') as f: - pw_db[u] = hash_pw(str(p), salt) - pickle.dump(pw_db, f) - - log.info('htpasswd db: added entry: user: %s, pw: %s...' % (u, pw_db[u][:5])) - def hash_pw(pw_str, salt_str): salt = hashlib.sha512(salt_str).hexdigest() ret = hashlib.sha512(pw_str + salt).hexdigest() -- cgit v1.3.1