##// END OF EJS Templates
factotum: initialize global variables to suppress pyflakes warning
Yuya Nishihara -
r21228:be561a62 default
parent child Browse files
Show More
@@ -1,120 +1,122 b''
1 1 # factotum.py - Plan 9 factotum integration for Mercurial
2 2 #
3 3 # Copyright (C) 2012 Steven Stallion <sstallion@gmail.com>
4 4 #
5 5 # This program is free software; you can redistribute it and/or modify it
6 6 # under the terms of the GNU General Public License as published by the
7 7 # Free Software Foundation; either version 2 of the License, or (at your
8 8 # option) any later version.
9 9 #
10 10 # This program is distributed in the hope that it will be useful, but
11 11 # WITHOUT ANY WARRANTY; without even the implied warranty of
12 12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
13 13 # Public License for more details.
14 14 #
15 15 # You should have received a copy of the GNU General Public License along
16 16 # with this program; if not, write to the Free Software Foundation, Inc.,
17 17 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 18
19 19 '''http authentication with factotum
20 20
21 21 This extension allows the factotum(4) facility on Plan 9 from Bell Labs
22 22 platforms to provide authentication information for HTTP access. Configuration
23 23 entries specified in the auth section as well as authentication information
24 24 provided in the repository URL are fully supported. If no prefix is specified,
25 25 a value of "*" will be assumed.
26 26
27 27 By default, keys are specified as::
28 28
29 29 proto=pass service=hg prefix=<prefix> user=<username> !password=<password>
30 30
31 31 If the factotum extension is unable to read the required key, one will be
32 32 requested interactively.
33 33
34 34 A configuration section is available to customize runtime behavior. By
35 35 default, these entries are::
36 36
37 37 [factotum]
38 38 executable = /bin/auth/factotum
39 39 mountpoint = /mnt/factotum
40 40 service = hg
41 41
42 42 The executable entry defines the full path to the factotum binary. The
43 43 mountpoint entry defines the path to the factotum file service. Lastly, the
44 44 service entry controls the service name used when reading keys.
45 45
46 46 '''
47 47
48 48 from mercurial.i18n import _
49 49 from mercurial.url import passwordmgr
50 50 from mercurial import httpconnection, util
51 51 import os, urllib2
52 52
53 53 ERRMAX = 128
54 54
55 _executable = _mountpoint = _service = None
56
55 57 def auth_getkey(self, params):
56 58 if not self.ui.interactive():
57 59 raise util.Abort(_('factotum not interactive'))
58 60 if 'user=' not in params:
59 61 params = '%s user?' % params
60 62 params = '%s !password?' % params
61 63 os.system("%s -g '%s'" % (_executable, params))
62 64
63 65 def auth_getuserpasswd(self, getkey, params):
64 66 params = 'proto=pass %s' % params
65 67 while True:
66 68 fd = os.open('%s/rpc' % _mountpoint, os.O_RDWR)
67 69 try:
68 70 try:
69 71 os.write(fd, 'start %s' % params)
70 72 l = os.read(fd, ERRMAX).split()
71 73 if l[0] == 'ok':
72 74 os.write(fd, 'read')
73 75 l = os.read(fd, ERRMAX).split()
74 76 if l[0] == 'ok':
75 77 return l[1:]
76 78 except (OSError, IOError):
77 79 raise util.Abort(_('factotum not responding'))
78 80 finally:
79 81 os.close(fd)
80 82 getkey(self, params)
81 83
82 84 def monkeypatch_method(cls):
83 85 def decorator(func):
84 86 setattr(cls, func.__name__, func)
85 87 return func
86 88 return decorator
87 89
88 90 @monkeypatch_method(passwordmgr)
89 91 def find_user_password(self, realm, authuri):
90 92 user, passwd = urllib2.HTTPPasswordMgrWithDefaultRealm.find_user_password(
91 93 self, realm, authuri)
92 94 if user and passwd:
93 95 self._writedebug(user, passwd)
94 96 return (user, passwd)
95 97
96 98 prefix = ''
97 99 res = httpconnection.readauthforuri(self.ui, authuri, user)
98 100 if res:
99 101 _, auth = res
100 102 prefix = auth.get('prefix')
101 103 user, passwd = auth.get('username'), auth.get('password')
102 104 if not user or not passwd:
103 105 if not prefix:
104 106 prefix = realm.split(' ')[0].lower()
105 107 params = 'service=%s prefix=%s' % (_service, prefix)
106 108 if user:
107 109 params = '%s user=%s' % (params, user)
108 110 user, passwd = auth_getuserpasswd(self, auth_getkey, params)
109 111
110 112 self.add_password(realm, authuri, user, passwd)
111 113 self._writedebug(user, passwd)
112 114 return (user, passwd)
113 115
114 116 def uisetup(ui):
115 117 global _executable
116 118 _executable = ui.config('factotum', 'executable', '/bin/auth/factotum')
117 119 global _mountpoint
118 120 _mountpoint = ui.config('factotum', 'mountpoint', '/mnt/factotum')
119 121 global _service
120 122 _service = ui.config('factotum', 'service', 'hg')
General Comments 0
You need to be logged in to leave comments. Login now