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