1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (C) 2006-2018 Edgewall Software
# Copyright (C) 2006 Matthew Good <matt@matt-good.net>
# All rights reserved.
#
# This software is licensed as described in the file COPYING, which
# you should have received as part of this distribution. The terms
# are also available at http://trac.edgewall.org/wiki/TracLicense.
#
# This software consists of voluntary contributions made by many
# individuals. For the exact contribution history, see the revision
# history and logs, available at http://trac.edgewall.org/log/.
#
# Author: Matthew Good <matt@matt-good.net>
from __future__ import print_function
import errno
import fileinput
import sys
from getpass import getpass
from hashlib import md5
from optparse import OptionParser
def ask_pass():
pass1 = getpass('New password: ')
pass2 = getpass('Re-type new password: ')
if pass1 != pass2:
print("They don't match, sorry", file=sys.stderr)
sys.exit(1)
return pass1
def get_digest(userprefix, password=None):
if password is None:
password = ask_pass()
return make_digest(userprefix, password)
def make_digest(userprefix, password):
return userprefix + md5(userprefix + password).hexdigest()
usage = "%prog [-c] [-b] passwordfile realm username"
parser = OptionParser(usage=usage)
parser.add_option('-c', action='store_true', dest='create', default=False,
help="Create a new file")
parser.add_option('-b', action='store_true', dest='batch', default=False,
help="Batch mode, password on the commandline.")
if len(sys.argv) <= 1:
parser.print_help()
sys.exit(1)
opts, args = parser.parse_args()
try:
if opts.batch:
filename, realm, username, password = args
else:
filename, realm, username = args
password = None
except ValueError:
parser.error("Wrong number of arguments")
prefix = '%s:%s:' % (username, realm)
if opts.create:
try:
f = open(filename, 'w')
except EnvironmentError as e:
if e.errno == errno.EACCES:
print("Unable to update file", filename, file=sys.stderr)
sys.exit(1)
else:
raise
try:
print(get_digest(prefix, password), file=f)
finally:
f.close()
else:
try:
matched = False
for line in fileinput.input(filename, inplace=True):
if line.startswith(prefix):
if not matched:
print(get_digest(prefix, password))
matched = True
else:
print(line.rstrip())
if not matched:
with open(filename, 'a') as f:
print(get_digest(prefix, password), file=f)
except EnvironmentError as e:
if e.errno == errno.ENOENT:
print("Could not open passwd file %s for reading." % filename,
file=sys.stderr)
print("Use -c option to create a new one.", file=sys.stderr)
sys.exit(1)
elif e.errno == errno.EACCES:
print("Unable to update file", filename, file=sys.stderr)
sys.exit(1)
else:
raise
|
