First version of pwgen
This commit is contained in:
Fredrik Eriksson
parent
498c0eefd0
commit
932322011b
115
bin/pwgen
Normal file
115
bin/pwgen
Normal file
@ -0,0 +1,115 @@
|
||||
#!/usr/bin/python3
|
||||
import argparse
|
||||
import os
|
||||
import sys
|
||||
|
||||
import pwgen
|
||||
|
||||
default_config_file=os.path.expanduser('~/.pwgen.cfg')
|
||||
|
||||
def main():
|
||||
|
||||
parser = argparse.ArgumentParser(description='Generate passwords')
|
||||
parser.add_argument(
|
||||
'--generate-config', '-g',
|
||||
action='store_true',
|
||||
help='Generate configuration file and then exit')
|
||||
parser.add_argument(
|
||||
'--config-file', '-c',
|
||||
help='Configuration file to use',
|
||||
default=default_config_file)
|
||||
|
||||
parser.add_argument(
|
||||
'--myspell-dir', '-i',
|
||||
help='Directory containing myspell dictionaries')
|
||||
|
||||
parser.add_argument(
|
||||
'--lang', '-l',
|
||||
help='Dictionary language to use')
|
||||
parser.add_argument(
|
||||
'--word-min-char', '-m',
|
||||
type=int,
|
||||
help='Minimum number of characters in a word')
|
||||
parser.add_argument(
|
||||
'--word-max-char', '-M',
|
||||
type=int,
|
||||
help='Maximum number of characters in a word')
|
||||
|
||||
parser.add_argument(
|
||||
'--words', '-w',
|
||||
type=int,
|
||||
help='Number of words to use in the passphrase')
|
||||
parser.add_argument(
|
||||
'--capitalize', '-C',
|
||||
choices=('true', 'false', 'random'),
|
||||
help='Capitalize the words')
|
||||
parser.add_argument(
|
||||
'--separators', '-s',
|
||||
help='Possible characters to use as separators')
|
||||
parser.add_argument(
|
||||
'--trailing-digits', '-d',
|
||||
type=int,
|
||||
help='Number of digits at the end of the passphrase')
|
||||
parser.add_argument(
|
||||
'--leading-digits', '-D',
|
||||
type=int,
|
||||
help='Number of digits at the start of the passphrase')
|
||||
parser.add_argument(
|
||||
'--special-chars', '-S',
|
||||
help='Possible characters to use as extra special characters')
|
||||
parser.add_argument(
|
||||
'--trailing-chars', '-p',
|
||||
type=int,
|
||||
help='Number of special characters to add at the end of the passphrase')
|
||||
parser.add_argument(
|
||||
'--leading-chars', '-P',
|
||||
type=int,
|
||||
help='Number of special characters to add at the start of the passphrase')
|
||||
parser.add_argument(
|
||||
'--passwords', '-n',
|
||||
type=int,
|
||||
help='Number of passwords to generate')
|
||||
parser.add_argument(
|
||||
'--max-length', '-L',
|
||||
type=int,
|
||||
help="Maximum length of the generated passwords. Full-knowledge "\
|
||||
"entropy calculation doesn't work when this is set.")
|
||||
|
||||
|
||||
|
||||
args = vars(parser.parse_args())
|
||||
|
||||
config_file = args['config_file']
|
||||
del args['config_file']
|
||||
|
||||
if not os.path.isfile(config_file):
|
||||
print("Missing configuration file; generating a new at {}".format(config_file))
|
||||
conf = pwgen.update_config()
|
||||
pwgen.save_config(config_file, conf)
|
||||
|
||||
conf = pwgen.get_config(config_file)
|
||||
save_config = args['generate_config']
|
||||
del args['generate_config']
|
||||
|
||||
conf = pwgen.update_config(config=conf, **args)
|
||||
|
||||
if save_config:
|
||||
print("Updating configuration file at {}".format(config_file))
|
||||
pwgen.save_config(config_file, conf)
|
||||
sys.exit(0)
|
||||
|
||||
pwds, seen_entropy = pwgen.generate_passwords(conf)
|
||||
|
||||
print("Generated {} passwords".format(len(pwds)))
|
||||
if seen_entropy:
|
||||
print("Full-knowledge entropy is {0:.2g}".format(seen_entropy))
|
||||
else:
|
||||
print("Unable to calculate full-knowledge entropy since max_length is used")
|
||||
print("Blind entropy\tPassword")
|
||||
print("========================")
|
||||
for pw in pwds.keys():
|
||||
print("{:.5n}\t\t{}".format(pwds[pw]['entropy'], pw))
|
||||
print("========================")
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
194
pwgen/__init__.py
Normal file
194
pwgen/__init__.py
Normal file
@ -0,0 +1,194 @@
|
||||
#!/usr/bin/python3
|
||||
|
||||
import math
|
||||
import os
|
||||
import sys
|
||||
#import secrets as random
|
||||
import random
|
||||
|
||||
if sys.version_info[0] < 3:
|
||||
import ConfigParser as configparser
|
||||
else:
|
||||
import configparser
|
||||
|
||||
if sys.version_info >= (3, 6):
|
||||
do_seed = False
|
||||
import secrets as random
|
||||
else:
|
||||
do_seed = True
|
||||
import random
|
||||
|
||||
class PasswordLengthError(Exception):
|
||||
pass
|
||||
|
||||
def save_config(target_file, config):
|
||||
path = os.path.expanduser(target_file)
|
||||
with open(path, 'w') as f:
|
||||
config.write(f, False)
|
||||
|
||||
def update_config(
|
||||
config=None,
|
||||
|
||||
myspell_dir='/usr/share/hunspell',
|
||||
lang='en_US',
|
||||
word_min_char=2,
|
||||
word_max_char=0,
|
||||
|
||||
words=4,
|
||||
capitalize='random',
|
||||
separators=',.- _=:',
|
||||
trailing_digits=2,
|
||||
leading_digits=0,
|
||||
special_chars='!?.,:-_$/@',
|
||||
trailing_chars=1,
|
||||
leading_chars=0,
|
||||
passwords=5,
|
||||
max_length=0
|
||||
):
|
||||
if not config:
|
||||
conf = configparser.ConfigParser()
|
||||
else:
|
||||
conf = config
|
||||
|
||||
def set_if_defined(conf, section, var, val):
|
||||
if val is not None:
|
||||
conf.set(section, var, str(val))
|
||||
|
||||
if not conf.has_section('dictionary'):
|
||||
conf.add_section('dictionary')
|
||||
|
||||
set_if_defined(conf, 'dictionary', 'myspell_dir', myspell_dir)
|
||||
set_if_defined(conf, 'dictionary', 'lang', lang)
|
||||
set_if_defined(conf, 'dictionary', 'word_min_char', word_min_char)
|
||||
set_if_defined(conf, 'dictionary', 'word_max_char', word_max_char)
|
||||
|
||||
if not conf.has_section('passwords'):
|
||||
conf.add_section('passwords')
|
||||
set_if_defined(conf, 'passwords', 'words', words)
|
||||
set_if_defined(conf, 'passwords', 'capitalize', capitalize)
|
||||
set_if_defined(conf, 'passwords', 'separators', separators)
|
||||
set_if_defined(conf, 'passwords', 'trailing_digits', trailing_digits)
|
||||
set_if_defined(conf, 'passwords', 'leading_digits', leading_digits)
|
||||
set_if_defined(conf, 'passwords', 'special_chars', special_chars)
|
||||
set_if_defined(conf, 'passwords', 'trailing_chars', trailing_chars)
|
||||
set_if_defined(conf, 'passwords', 'leading_chars', leading_chars)
|
||||
set_if_defined(conf, 'passwords', 'passwords', passwords)
|
||||
set_if_defined(conf, 'passwords', 'max_length', max_length)
|
||||
return conf
|
||||
|
||||
|
||||
def get_config(f_name):
|
||||
conf = configparser.ConfigParser()
|
||||
conf.read(f_name)
|
||||
return conf
|
||||
|
||||
def _read_dictionary(f_name, word_min_chars, word_max_chars):
|
||||
words = set()
|
||||
chars = 0
|
||||
with open(f_name, 'r') as f:
|
||||
for line in f:
|
||||
if not line:
|
||||
continue
|
||||
first_char = line[0]
|
||||
if first_char in '1234567890,.-:':
|
||||
continue
|
||||
word = line.split('/', 1)[0]
|
||||
word = word.strip() # remove newlines
|
||||
last_char = word[-1]
|
||||
if last_char in '-':
|
||||
continue
|
||||
if word_min_chars and len(word) < word_min_chars:
|
||||
continue
|
||||
if word_max_chars and len(word) > word_max_chars:
|
||||
continue
|
||||
words.add(word)
|
||||
chars += len(word)
|
||||
return {'words': list(words), 'wordlength': int(chars/len(words))}
|
||||
|
||||
|
||||
def generate_passwords(conf):
|
||||
|
||||
nr_of_words = conf.getint('passwords', 'words')
|
||||
nr_of_ld = conf.getint('passwords', 'leading_digits')
|
||||
nr_of_td = conf.getint('passwords', 'trailing_digits')
|
||||
nr_of_lc = conf.getint('passwords', 'leading_chars')
|
||||
nr_of_tc = conf.getint('passwords', 'trailing_chars')
|
||||
special_chars = conf.get('passwords', 'special_chars')
|
||||
separators = conf.get('passwords', 'separators')
|
||||
capitalize = conf.get('passwords', 'capitalize')
|
||||
max_len = conf.getint('passwords', 'max_length')
|
||||
|
||||
if do_seed:
|
||||
random.seed()
|
||||
res = {}
|
||||
dict_data = _read_dictionary(
|
||||
os.path.join(conf.get('dictionary', 'myspell_dir'), '{}.dic'.format(conf.get('dictionary', 'lang'))),
|
||||
conf.getint('dictionary', 'word_min_char'),
|
||||
conf.getint('dictionary', 'word_max_char'))
|
||||
|
||||
words = dict_data['words']
|
||||
word_length = dict_data['wordlength']
|
||||
approx_pwd_length = nr_of_words*word_length+nr_of_ld+nr_of_td+nr_of_lc+nr_of_tc+nr_of_words-1
|
||||
if max_len and approx_pwd_length > max_len:
|
||||
raise PasswordLengthError(
|
||||
"Password would be ~{} charactes long but max_len is {}; "\
|
||||
"please adjust max_lenght, words and/or the "\
|
||||
"trailing/leading settings.".format(
|
||||
approx_pwd_length,
|
||||
max_len)
|
||||
)
|
||||
|
||||
capitalize_entropy = 1
|
||||
if capitalize == 'random':
|
||||
capitalize_entropy = 2
|
||||
# seen entropy is calculated from the password rules.
|
||||
# At the moment setting max_length breaks this calculation as not all
|
||||
# combinations of words are possible.
|
||||
if max_len:
|
||||
seen_entropy = False
|
||||
else:
|
||||
seen_entropy = math.log(
|
||||
len(words)**nr_of_words *\
|
||||
len(separators) *\
|
||||
10**nr_of_ld *\
|
||||
10**nr_of_td *\
|
||||
len(special_chars)**nr_of_lc *\
|
||||
len(special_chars)**nr_of_tc *\
|
||||
capitalize_entropy
|
||||
, 2)
|
||||
|
||||
while len(res) < conf.getint('passwords', 'passwords'):
|
||||
separator = random.choice(separators)
|
||||
my_words = []
|
||||
|
||||
for i in range(nr_of_words):
|
||||
word = random.choice(words)
|
||||
if capitalize == 'random':
|
||||
if random.choice((True, False)):
|
||||
word = word.capitalize()
|
||||
elif capitalize == 'true':
|
||||
word = word.capitalize()
|
||||
my_words.append(word)
|
||||
|
||||
base_pwd = separator.join(my_words)
|
||||
leading_digits = ''.join(random.choice('1234567890') for i in range(nr_of_ld))
|
||||
trailing_digits = ''.join(random.choice('1234567890') for i in range(nr_of_td))
|
||||
leading_chars = ''.join(random.choice(special_chars) for i in range(nr_of_lc))
|
||||
trailing_chars = ''.join(random.choice(special_chars) for i in range(nr_of_tc))
|
||||
pwd = '{ld}{lc}{base}{tc}{td}'.format(
|
||||
lc=leading_chars,
|
||||
ld=leading_digits,
|
||||
base=base_pwd,
|
||||
td=trailing_digits,
|
||||
tc=trailing_chars)
|
||||
# blind entropy is calculated only by the number of unique characters
|
||||
# and password length
|
||||
blind_entropy = math.log(len(''.join(set(pwd)))**len(pwd), 2)
|
||||
|
||||
if max_len and len(pwd) > max_len:
|
||||
continue
|
||||
res[pwd] = {
|
||||
'length': len(pwd),
|
||||
'entropy': blind_entropy,
|
||||
}
|
||||
return (res, seen_entropy)
|
||||
34
setup.py
Normal file
34
setup.py
Normal file
@ -0,0 +1,34 @@
|
||||
#!/usr/bin/env python
|
||||
from os import environ
|
||||
|
||||
try:
|
||||
from setuptools import setup
|
||||
except ImportError:
|
||||
from distutils import setup
|
||||
|
||||
import pwgen
|
||||
|
||||
version = '0.1'
|
||||
|
||||
setup(
|
||||
name='pwgen',
|
||||
version=str(version),
|
||||
description="Passphrase generator",
|
||||
author="Fredrik Eriksson",
|
||||
author_email="pwgen@wb9.se",
|
||||
url="https://github.com/fredrik-eriksson/pwgen",
|
||||
platforms=['any'],
|
||||
license='BSD',
|
||||
packages=['pwgen'],
|
||||
classifiers=[
|
||||
'Development Status :: 1 - Planning',
|
||||
'Environment :: Console',
|
||||
'Intended Audience :: System Administrators',
|
||||
'License :: OSI Approved :: BSD License',
|
||||
'Programming Language :: Python :: 2',
|
||||
'Programming Language :: Python :: 3',
|
||||
'Topic :: Utilities',
|
||||
],
|
||||
keywords='password passphrase',
|
||||
scripts=['bin/pwgen']
|
||||
)
|
||||
Loading…
Reference in New Issue
Block a user