kea/src/bin/usermgr/b10-cmdctl-usermgr.py.in

#!@PYTHON@

# Copyright (C) 2010  Internet Systems Consortium.
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SYSTEMS CONSORTIUM
# DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL
# INTERNET SYSTEMS CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT,
# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING
# FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
# NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
# WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

'''
This file implements user management program. The user name and 
its password is appended to csv file.
'''
import random
from hashlib import sha1
import csv
import getpass
import getopt
import sys
import isc.utils.process

isc.utils.process.rename()

VERSION_NUMBER = 'bind10'
DEFAULT_FILE = 'cmdctl-accounts.csv'

def gen_password_hash(password):
    salt = "".join(chr(random.randint(33, 127)) for x in range(64))
    saltedpwd = sha1((password + salt).encode()).hexdigest()
    return salt, saltedpwd

def username_exist(name, filename):
    # The file may doesn't exist.
    exist = False
    csvfile = None
    try:
        csvfile = open(filename)        
        reader = csv.reader(csvfile)
        for row in reader:
            if name == row[0]:
                exist = True               
                break
    except Exception:
        pass
            
    if csvfile:    
        csvfile.close()
    return exist

def save_userinfo(username, pw, salt, filename):
    csvfile = open(filename, 'a')
    writer = csv.writer(csvfile)
    writer.writerow([username, pw, salt])
    csvfile.close()
    print("\n create new account successfully! \n")

def usage():
    print('''Usage: usermgr [options]
           -h, --help \t Show this help message and exit
           -f, --file \t Specify the file to append user name and password
           -v, --version\t Get version number
           ''')           

def main():
    filename = DEFAULT_FILE
    try: 
        opts, args = getopt.getopt(sys.argv[1:], 'f:hv', 
                                   ['file=', 'help', 'version']) 
    except getopt.GetoptError as err: 
        print(err) 
        usage() 
        sys.exit(2)
    for op, param in opts:        
        if op in ('-h', '--help'): 
            usage()
            sys.exit()
        elif op in ('-v', '--version'): 
            print(VERSION_NUMBER)
            sys.exit()                     
        elif op in ('-f', "--file"):
            filename = param
        else: 
            assert False, 'unknown option' 
            usage()
    
    try:
        while True :
            name = input("Desired Login Name:")
            if name == '':
                print("error, user name can't be empty")
                continue

            if username_exist(name, filename):
                 print("user name already exists!")
                 continue

            while True:
                pwd1 = getpass.getpass("Choose a password:")
                pwd2 = getpass.getpass("Re-enter password:")
                if pwd1 != pwd2:
                    print("password is not same, please input again")
                else:
                    break;
            
            salt, pw = gen_password_hash(pwd1)
            save_userinfo(name, pw, salt, filename)
            inputdata = input('continue to create new account by input \'y\' or \'Y\':')
            if inputdata not in ['y', 'Y']:
                break                
                
    except KeyboardInterrupt:
        pass                


if __name__ == '__main__':
    main()