blob: be5e3131dad47acfce800bece340610ff362f0bd [file] [log] [blame]
David Pursehouse8898e2f2012-11-14 07:51:03 +09001#!/usr/bin/env python
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002#
3# Copyright (C) 2008 The Android Open Source Project
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9# http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
Sarah Owenscecd1d82012-11-01 22:59:27 -070017from __future__ import print_function
JoonCheol Parke9860722012-10-11 02:31:44 +090018import getpass
Conley Owensc9129d92012-10-01 16:12:28 -070019import imp
Shawn O. Pearcebd0312a2011-09-19 10:04:23 -070020import netrc
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070021import optparse
22import os
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070023import sys
Shawn O. Pearce3a0e7822011-09-22 17:06:41 -070024import time
David Pursehouse59bbb582013-05-17 10:49:33 +090025
26from pyversion import is_python3
27if is_python3():
Sarah Owens1f7627f2012-10-31 09:21:55 -070028 import urllib.request
29else:
David Pursehouse59bbb582013-05-17 10:49:33 +090030 import urllib2
Sarah Owens1f7627f2012-10-31 09:21:55 -070031 urllib = imp.new_module('urllib')
32 urllib.request = urllib2
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070033
Carlos Aguado1242e602014-02-03 13:48:47 +010034try:
35 import kerberos
36except ImportError:
37 kerberos = None
38
Mike Frysinger902665b2014-12-22 15:17:59 -050039from color import SetDefaultColoring
David Rileye0684ad2017-04-05 00:02:59 -070040import event_log
Shawn O. Pearcead3193a2009-04-18 09:54:51 -070041from trace import SetTrace
Shawn O. Pearce334851e2011-09-19 08:05:31 -070042from git_command import git, GitCommand
Doug Anderson0048b692010-12-21 13:39:23 -080043from git_config import init_ssh, close_ssh
Shawn O. Pearcec95583b2009-03-03 17:47:06 -080044from command import InteractiveCommand
45from command import MirrorSafeCommand
Dan Willemsen79360642015-08-31 15:45:06 -070046from command import GitcAvailableCommand, GitcClientCommand
Shawn O. Pearceecff4f12011-11-29 15:01:33 -080047from subcmds.version import Version
Shawn O. Pearce7965f9f2008-10-29 15:20:02 -070048from editor import Editor
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -070049from error import DownloadError
Jarkko Pöyry87ea5912015-06-19 15:39:25 -070050from error import InvalidProjectGroupsError
Shawn O. Pearce559b8462009-03-02 12:56:08 -080051from error import ManifestInvalidRevisionError
David Pursehouse0b8df7b2012-11-13 09:51:57 +090052from error import ManifestParseError
Conley Owens75ee0572012-11-15 17:33:11 -080053from error import NoManifestException
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070054from error import NoSuchProjectError
55from error import RepoChangedException
Simran Basib9a1b732015-08-20 12:19:28 -070056import gitc_utils
57from manifest_xml import GitcManifest, XmlManifest
Renaud Paquaye8595e92016-11-01 15:51:59 -070058from pager import RunPager, TerminatePager
Conley Owens094cdbe2014-01-30 15:09:59 -080059from wrapper import WrapperPath, Wrapper
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070060
David Pursehouse5c6eeac2012-10-11 16:44:48 +090061from subcmds import all_commands
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070062
David Pursehouse59bbb582013-05-17 10:49:33 +090063if not is_python3():
Chirayu Desai217ea7d2013-03-01 19:14:38 +053064 input = raw_input
Chirayu Desai217ea7d2013-03-01 19:14:38 +053065
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070066global_options = optparse.OptionParser(
67 usage="repo [-p|--paginate|--no-pager] COMMAND [ARGS]"
68 )
69global_options.add_option('-p', '--paginate',
70 dest='pager', action='store_true',
71 help='display command output in the pager')
72global_options.add_option('--no-pager',
73 dest='no_pager', action='store_true',
74 help='disable the pager')
Mike Frysinger902665b2014-12-22 15:17:59 -050075global_options.add_option('--color',
76 choices=('auto', 'always', 'never'), default=None,
77 help='control color usage: auto, always, never')
Shawn O. Pearce0ed2bd12009-03-09 18:26:31 -070078global_options.add_option('--trace',
79 dest='trace', action='store_true',
80 help='trace git command execution')
Shawn O. Pearce3a0e7822011-09-22 17:06:41 -070081global_options.add_option('--time',
82 dest='time', action='store_true',
83 help='time repo command execution')
Shawn O. Pearce47c1a632009-03-02 18:24:23 -080084global_options.add_option('--version',
85 dest='show_version', action='store_true',
86 help='display this version of repo')
David Rileye0684ad2017-04-05 00:02:59 -070087global_options.add_option('--event-log',
88 dest='event_log', action='store',
89 help='filename of event log to append timeline to')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070090
91class _Repo(object):
92 def __init__(self, repodir):
93 self.repodir = repodir
94 self.commands = all_commands
Mike Lockwood2bf9db02009-07-14 15:23:39 -040095 # add 'branch' as an alias for 'branches'
96 all_commands['branch'] = all_commands['branches']
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070097
98 def _Run(self, argv):
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -040099 result = 0
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700100 name = None
101 glob = []
102
Sarah Owensa6053d52012-11-01 13:36:50 -0700103 for i in range(len(argv)):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700104 if not argv[i].startswith('-'):
105 name = argv[i]
106 if i > 0:
107 glob = argv[:i]
108 argv = argv[i + 1:]
109 break
110 if not name:
111 glob = argv
112 name = 'help'
113 argv = []
David Pursehouse8a68ff92012-09-24 12:15:13 +0900114 gopts, _gargs = global_options.parse_args(glob)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700115
Shawn O. Pearce0ed2bd12009-03-09 18:26:31 -0700116 if gopts.trace:
Shawn O. Pearcead3193a2009-04-18 09:54:51 -0700117 SetTrace()
Shawn O. Pearce47c1a632009-03-02 18:24:23 -0800118 if gopts.show_version:
119 if name == 'help':
120 name = 'version'
121 else:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700122 print('fatal: invalid usage of --version', file=sys.stderr)
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400123 return 1
Shawn O. Pearce47c1a632009-03-02 18:24:23 -0800124
Mike Frysinger902665b2014-12-22 15:17:59 -0500125 SetDefaultColoring(gopts.color)
126
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700127 try:
128 cmd = self.commands[name]
129 except KeyError:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700130 print("repo: '%s' is not a repo command. See 'repo help'." % name,
131 file=sys.stderr)
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400132 return 1
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700133
134 cmd.repodir = self.repodir
Shawn O. Pearcec8a300f2009-05-18 13:19:57 -0700135 cmd.manifest = XmlManifest(cmd.repodir)
Simran Basib9a1b732015-08-20 12:19:28 -0700136 cmd.gitc_manifest = None
137 gitc_client_name = gitc_utils.parse_clientdir(os.getcwd())
138 if gitc_client_name:
139 cmd.gitc_manifest = GitcManifest(cmd.repodir, gitc_client_name)
140 cmd.manifest.isGitcClient = True
141
Shawn O. Pearce7965f9f2008-10-29 15:20:02 -0700142 Editor.globalConfig = cmd.manifest.globalConfig
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700143
Shawn O. Pearcec95583b2009-03-03 17:47:06 -0800144 if not isinstance(cmd, MirrorSafeCommand) and cmd.manifest.IsMirror:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700145 print("fatal: '%s' requires a working directory" % name,
146 file=sys.stderr)
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400147 return 1
Shawn O. Pearcec95583b2009-03-03 17:47:06 -0800148
Dan Willemsen79360642015-08-31 15:45:06 -0700149 if isinstance(cmd, GitcAvailableCommand) and not gitc_utils.get_gitc_manifest_dir():
Dan Willemsen9ff2ece2015-08-31 15:45:06 -0700150 print("fatal: '%s' requires GITC to be available" % name,
151 file=sys.stderr)
152 return 1
153
Dan Willemsen79360642015-08-31 15:45:06 -0700154 if isinstance(cmd, GitcClientCommand) and not gitc_client_name:
155 print("fatal: '%s' requires a GITC client" % name,
156 file=sys.stderr)
157 return 1
158
Dan Sandler53e902a2014-03-09 13:20:02 -0400159 try:
160 copts, cargs = cmd.OptionParser.parse_args(argv)
161 copts = cmd.ReadEnvironmentOptions(copts)
162 except NoManifestException as e:
163 print('error: in `%s`: %s' % (' '.join([name] + argv), str(e)),
164 file=sys.stderr)
165 print('error: manifest missing or unreadable -- please run init',
166 file=sys.stderr)
167 return 1
Shawn O. Pearcedb45da12009-04-18 13:49:13 -0700168
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700169 if not gopts.no_pager and not isinstance(cmd, InteractiveCommand):
170 config = cmd.manifest.globalConfig
171 if gopts.pager:
172 use_pager = True
173 else:
174 use_pager = config.GetBoolean('pager.%s' % name)
175 if use_pager is None:
Shawn O. Pearcedb45da12009-04-18 13:49:13 -0700176 use_pager = cmd.WantPager(copts)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700177 if use_pager:
178 RunPager(config)
179
Conley Owens7ba25be2012-11-14 14:18:06 -0800180 start = time.time()
David Rileye0684ad2017-04-05 00:02:59 -0700181 cmd_event = cmd.event_log.Add(name, event_log.TASK_COMMAND, start)
182 cmd.event_log.SetParent(cmd_event)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700183 try:
Conley Owens7ba25be2012-11-14 14:18:06 -0800184 result = cmd.Execute(copts, cargs)
Dan Sandler53e902a2014-03-09 13:20:02 -0400185 except (DownloadError, ManifestInvalidRevisionError,
186 NoManifestException) as e:
187 print('error: in `%s`: %s' % (' '.join([name] + argv), str(e)),
188 file=sys.stderr)
189 if isinstance(e, NoManifestException):
190 print('error: manifest missing or unreadable -- please run init',
191 file=sys.stderr)
Conley Owens75ee0572012-11-15 17:33:11 -0800192 result = 1
Sarah Owensa5be53f2012-09-09 15:37:57 -0700193 except NoSuchProjectError as e:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700194 if e.name:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700195 print('error: project %s not found' % e.name, file=sys.stderr)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700196 else:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700197 print('error: no project in current directory', file=sys.stderr)
Conley Owens7ba25be2012-11-14 14:18:06 -0800198 result = 1
Jarkko Pöyry87ea5912015-06-19 15:39:25 -0700199 except InvalidProjectGroupsError as e:
200 if e.name:
201 print('error: project group must be enabled for project %s' % e.name, file=sys.stderr)
202 else:
203 print('error: project group must be enabled for the project in the current directory', file=sys.stderr)
204 result = 1
David Rileyaa900212017-04-05 13:50:52 -0700205 except SystemExit as e:
206 if e.code:
207 result = e.code
208 raise
Conley Owens7ba25be2012-11-14 14:18:06 -0800209 finally:
David Rileye0684ad2017-04-05 00:02:59 -0700210 finish = time.time()
211 elapsed = finish - start
Conley Owens7ba25be2012-11-14 14:18:06 -0800212 hours, remainder = divmod(elapsed, 3600)
213 minutes, seconds = divmod(remainder, 60)
214 if gopts.time:
215 if hours == 0:
216 print('real\t%dm%.3fs' % (minutes, seconds), file=sys.stderr)
217 else:
218 print('real\t%dh%dm%.3fs' % (hours, minutes, seconds),
219 file=sys.stderr)
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400220
David Rileye0684ad2017-04-05 00:02:59 -0700221 cmd.event_log.FinishEvent(cmd_event, finish,
222 result is None or result == 0)
223 if gopts.event_log:
224 cmd.event_log.Write(os.path.abspath(
225 os.path.expanduser(gopts.event_log)))
226
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400227 return result
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700228
Conley Owens094cdbe2014-01-30 15:09:59 -0800229
Shawn O. Pearce334851e2011-09-19 08:05:31 -0700230def _MyRepoPath():
231 return os.path.dirname(__file__)
232
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700233
234def _CheckWrapperVersion(ver, repo_path):
235 if not repo_path:
236 repo_path = '~/bin/repo'
237
238 if not ver:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700239 print('no --wrapper-version argument', file=sys.stderr)
David Pursehouse8a68ff92012-09-24 12:15:13 +0900240 sys.exit(1)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700241
Conley Owens094cdbe2014-01-30 15:09:59 -0800242 exp = Wrapper().VERSION
David Pursehouse7e6dd2d2012-10-25 12:40:51 +0900243 ver = tuple(map(int, ver.split('.')))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700244 if len(ver) == 1:
245 ver = (0, ver[0])
246
David Pursehouse7e6dd2d2012-10-25 12:40:51 +0900247 exp_str = '.'.join(map(str, exp))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700248 if exp[0] > ver[0] or ver < (0, 4):
Sarah Owenscecd1d82012-11-01 22:59:27 -0700249 print("""
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700250!!! A new repo command (%5s) is available. !!!
251!!! You must upgrade before you can continue: !!!
252
253 cp %s %s
Conley Owens094cdbe2014-01-30 15:09:59 -0800254""" % (exp_str, WrapperPath(), repo_path), file=sys.stderr)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700255 sys.exit(1)
256
257 if exp > ver:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700258 print("""
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700259... A new repo command (%5s) is available.
260... You should upgrade soon:
261
262 cp %s %s
Conley Owens094cdbe2014-01-30 15:09:59 -0800263""" % (exp_str, WrapperPath(), repo_path), file=sys.stderr)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700264
Mickaël Salaün2f6ab7f2012-09-30 00:37:55 +0200265def _CheckRepoDir(repo_dir):
266 if not repo_dir:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700267 print('no --repo-dir argument', file=sys.stderr)
David Pursehouse8a68ff92012-09-24 12:15:13 +0900268 sys.exit(1)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700269
270def _PruneOptions(argv, opt):
271 i = 0
272 while i < len(argv):
273 a = argv[i]
274 if a == '--':
275 break
276 if a.startswith('--'):
277 eq = a.find('=')
278 if eq > 0:
279 a = a[0:eq]
280 if not opt.has_option(a):
281 del argv[i]
282 continue
283 i += 1
284
Shawn O. Pearce334851e2011-09-19 08:05:31 -0700285_user_agent = None
286
287def _UserAgent():
288 global _user_agent
289
290 if _user_agent is None:
291 py_version = sys.version_info
292
293 os_name = sys.platform
294 if os_name == 'linux2':
295 os_name = 'Linux'
296 elif os_name == 'win32':
297 os_name = 'Win32'
298 elif os_name == 'cygwin':
299 os_name = 'Cygwin'
300 elif os_name == 'darwin':
301 os_name = 'Darwin'
302
303 p = GitCommand(
304 None, ['describe', 'HEAD'],
305 cwd = _MyRepoPath(),
306 capture_stdout = True)
307 if p.Wait() == 0:
308 repo_version = p.stdout
309 if len(repo_version) > 0 and repo_version[-1] == '\n':
310 repo_version = repo_version[0:-1]
311 if len(repo_version) > 0 and repo_version[0] == 'v':
312 repo_version = repo_version[1:]
313 else:
314 repo_version = 'unknown'
315
316 _user_agent = 'git-repo/%s (%s) git/%s Python/%d.%d.%d' % (
317 repo_version,
318 os_name,
David Pursehouse7e6dd2d2012-10-25 12:40:51 +0900319 '.'.join(map(str, git.version_tuple())),
Shawn O. Pearce334851e2011-09-19 08:05:31 -0700320 py_version[0], py_version[1], py_version[2])
321 return _user_agent
322
Sarah Owens1f7627f2012-10-31 09:21:55 -0700323class _UserAgentHandler(urllib.request.BaseHandler):
Shawn O. Pearce334851e2011-09-19 08:05:31 -0700324 def http_request(self, req):
325 req.add_header('User-Agent', _UserAgent())
326 return req
327
328 def https_request(self, req):
329 req.add_header('User-Agent', _UserAgent())
330 return req
331
JoonCheol Parke9860722012-10-11 02:31:44 +0900332def _AddPasswordFromUserInput(handler, msg, req):
David Pursehousec1b86a22012-11-14 11:36:51 +0900333 # If repo could not find auth info from netrc, try to get it from user input
334 url = req.get_full_url()
335 user, password = handler.passwd.find_user_password(None, url)
336 if user is None:
337 print(msg)
338 try:
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530339 user = input('User: ')
David Pursehousec1b86a22012-11-14 11:36:51 +0900340 password = getpass.getpass()
341 except KeyboardInterrupt:
342 return
343 handler.passwd.add_password(None, url, user, password)
JoonCheol Parke9860722012-10-11 02:31:44 +0900344
Sarah Owens1f7627f2012-10-31 09:21:55 -0700345class _BasicAuthHandler(urllib.request.HTTPBasicAuthHandler):
JoonCheol Parke9860722012-10-11 02:31:44 +0900346 def http_error_401(self, req, fp, code, msg, headers):
347 _AddPasswordFromUserInput(self, msg, req)
Sarah Owens1f7627f2012-10-31 09:21:55 -0700348 return urllib.request.HTTPBasicAuthHandler.http_error_401(
JoonCheol Parke9860722012-10-11 02:31:44 +0900349 self, req, fp, code, msg, headers)
350
Shawn O. Pearcefab96c62011-10-11 12:00:38 -0700351 def http_error_auth_reqed(self, authreq, host, req, headers):
352 try:
Shawn O. Pearcedf5ee522011-10-11 14:05:21 -0700353 old_add_header = req.add_header
354 def _add_header(name, val):
355 val = val.replace('\n', '')
356 old_add_header(name, val)
357 req.add_header = _add_header
Sarah Owens1f7627f2012-10-31 09:21:55 -0700358 return urllib.request.AbstractBasicAuthHandler.http_error_auth_reqed(
Shawn O. Pearcefab96c62011-10-11 12:00:38 -0700359 self, authreq, host, req, headers)
360 except:
Shawn O. Pearcedf5ee522011-10-11 14:05:21 -0700361 reset = getattr(self, 'reset_retry_count', None)
362 if reset is not None:
363 reset()
Shawn O. Pearceb6605392011-10-11 15:58:07 -0700364 elif getattr(self, 'retried', None):
365 self.retried = 0
Shawn O. Pearcefab96c62011-10-11 12:00:38 -0700366 raise
367
Sarah Owens1f7627f2012-10-31 09:21:55 -0700368class _DigestAuthHandler(urllib.request.HTTPDigestAuthHandler):
JoonCheol Parke9860722012-10-11 02:31:44 +0900369 def http_error_401(self, req, fp, code, msg, headers):
370 _AddPasswordFromUserInput(self, msg, req)
Sarah Owens1f7627f2012-10-31 09:21:55 -0700371 return urllib.request.HTTPDigestAuthHandler.http_error_401(
JoonCheol Parke9860722012-10-11 02:31:44 +0900372 self, req, fp, code, msg, headers)
373
Xiaodong Xuae0a36c2012-01-31 11:10:09 +0800374 def http_error_auth_reqed(self, auth_header, host, req, headers):
375 try:
376 old_add_header = req.add_header
377 def _add_header(name, val):
378 val = val.replace('\n', '')
379 old_add_header(name, val)
380 req.add_header = _add_header
Sarah Owens1f7627f2012-10-31 09:21:55 -0700381 return urllib.request.AbstractDigestAuthHandler.http_error_auth_reqed(
Xiaodong Xuae0a36c2012-01-31 11:10:09 +0800382 self, auth_header, host, req, headers)
383 except:
384 reset = getattr(self, 'reset_retry_count', None)
385 if reset is not None:
386 reset()
387 elif getattr(self, 'retried', None):
388 self.retried = 0
389 raise
390
Carlos Aguado1242e602014-02-03 13:48:47 +0100391class _KerberosAuthHandler(urllib.request.BaseHandler):
392 def __init__(self):
393 self.retried = 0
394 self.context = None
395 self.handler_order = urllib.request.BaseHandler.handler_order - 50
396
David Pursehouse65b0ba52018-06-24 16:21:51 +0900397 def http_error_401(self, req, fp, code, msg, headers):
Carlos Aguado1242e602014-02-03 13:48:47 +0100398 host = req.get_host()
399 retry = self.http_error_auth_reqed('www-authenticate', host, req, headers)
400 return retry
401
402 def http_error_auth_reqed(self, auth_header, host, req, headers):
403 try:
404 spn = "HTTP@%s" % host
405 authdata = self._negotiate_get_authdata(auth_header, headers)
406
407 if self.retried > 3:
408 raise urllib.request.HTTPError(req.get_full_url(), 401,
409 "Negotiate auth failed", headers, None)
410 else:
411 self.retried += 1
412
413 neghdr = self._negotiate_get_svctk(spn, authdata)
414 if neghdr is None:
415 return None
416
417 req.add_unredirected_header('Authorization', neghdr)
418 response = self.parent.open(req)
419
420 srvauth = self._negotiate_get_authdata(auth_header, response.info())
421 if self._validate_response(srvauth):
422 return response
423 except kerberos.GSSError:
424 return None
425 except:
426 self.reset_retry_count()
427 raise
428 finally:
429 self._clean_context()
430
431 def reset_retry_count(self):
432 self.retried = 0
433
434 def _negotiate_get_authdata(self, auth_header, headers):
435 authhdr = headers.get(auth_header, None)
436 if authhdr is not None:
437 for mech_tuple in authhdr.split(","):
438 mech, __, authdata = mech_tuple.strip().partition(" ")
439 if mech.lower() == "negotiate":
440 return authdata.strip()
441 return None
442
443 def _negotiate_get_svctk(self, spn, authdata):
444 if authdata is None:
445 return None
446
447 result, self.context = kerberos.authGSSClientInit(spn)
448 if result < kerberos.AUTH_GSS_COMPLETE:
449 return None
450
451 result = kerberos.authGSSClientStep(self.context, authdata)
452 if result < kerberos.AUTH_GSS_CONTINUE:
453 return None
454
455 response = kerberos.authGSSClientResponse(self.context)
456 return "Negotiate %s" % response
457
458 def _validate_response(self, authdata):
459 if authdata is None:
460 return None
461 result = kerberos.authGSSClientStep(self.context, authdata)
462 if result == kerberos.AUTH_GSS_COMPLETE:
463 return True
464 return None
465
466 def _clean_context(self):
467 if self.context is not None:
468 kerberos.authGSSClientClean(self.context)
469 self.context = None
470
Shawn O. Pearce014d0602011-09-11 12:57:15 -0700471def init_http():
Shawn O. Pearce334851e2011-09-19 08:05:31 -0700472 handlers = [_UserAgentHandler()]
473
Sarah Owens1f7627f2012-10-31 09:21:55 -0700474 mgr = urllib.request.HTTPPasswordMgrWithDefaultRealm()
Shawn O. Pearcebd0312a2011-09-19 10:04:23 -0700475 try:
476 n = netrc.netrc()
477 for host in n.hosts:
478 p = n.hosts[host]
Xiaodong Xuae0a36c2012-01-31 11:10:09 +0800479 mgr.add_password(p[1], 'http://%s/' % host, p[0], p[2])
480 mgr.add_password(p[1], 'https://%s/' % host, p[0], p[2])
Shawn O. Pearcebd0312a2011-09-19 10:04:23 -0700481 except netrc.NetrcParseError:
482 pass
Shawn O. Pearce7b947de2011-09-23 11:50:31 -0700483 except IOError:
484 pass
Shawn O. Pearcefab96c62011-10-11 12:00:38 -0700485 handlers.append(_BasicAuthHandler(mgr))
Xiaodong Xuae0a36c2012-01-31 11:10:09 +0800486 handlers.append(_DigestAuthHandler(mgr))
Carlos Aguado1242e602014-02-03 13:48:47 +0100487 if kerberos:
488 handlers.append(_KerberosAuthHandler())
Shawn O. Pearcebd0312a2011-09-19 10:04:23 -0700489
Shawn O. Pearce014d0602011-09-11 12:57:15 -0700490 if 'http_proxy' in os.environ:
491 url = os.environ['http_proxy']
Sarah Owens1f7627f2012-10-31 09:21:55 -0700492 handlers.append(urllib.request.ProxyHandler({'http': url, 'https': url}))
Shawn O. Pearce334851e2011-09-19 08:05:31 -0700493 if 'REPO_CURL_VERBOSE' in os.environ:
Sarah Owens1f7627f2012-10-31 09:21:55 -0700494 handlers.append(urllib.request.HTTPHandler(debuglevel=1))
495 handlers.append(urllib.request.HTTPSHandler(debuglevel=1))
496 urllib.request.install_opener(urllib.request.build_opener(*handlers))
Shawn O. Pearce014d0602011-09-11 12:57:15 -0700497
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700498def _Main(argv):
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400499 result = 0
500
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700501 opt = optparse.OptionParser(usage="repo wrapperinfo -- ...")
502 opt.add_option("--repo-dir", dest="repodir",
503 help="path to .repo/")
504 opt.add_option("--wrapper-version", dest="wrapper_version",
505 help="version of the wrapper script")
506 opt.add_option("--wrapper-path", dest="wrapper_path",
507 help="location of the wrapper script")
508 _PruneOptions(argv, opt)
509 opt, argv = opt.parse_args(argv)
510
511 _CheckWrapperVersion(opt.wrapper_version, opt.wrapper_path)
512 _CheckRepoDir(opt.repodir)
513
Shawn O. Pearceecff4f12011-11-29 15:01:33 -0800514 Version.wrapper_version = opt.wrapper_version
515 Version.wrapper_path = opt.wrapper_path
516
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700517 repo = _Repo(opt.repodir)
518 try:
Shawn O. Pearcefb231612009-04-10 18:53:46 -0700519 try:
Doug Anderson0048b692010-12-21 13:39:23 -0800520 init_ssh()
Shawn O. Pearce014d0602011-09-11 12:57:15 -0700521 init_http()
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400522 result = repo._Run(argv) or 0
Shawn O. Pearcefb231612009-04-10 18:53:46 -0700523 finally:
524 close_ssh()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700525 except KeyboardInterrupt:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700526 print('aborted by user', file=sys.stderr)
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400527 result = 1
David Pursehouse0b8df7b2012-11-13 09:51:57 +0900528 except ManifestParseError as mpe:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700529 print('fatal: %s' % mpe, file=sys.stderr)
David Pursehouse0b8df7b2012-11-13 09:51:57 +0900530 result = 1
Sarah Owensa5be53f2012-09-09 15:37:57 -0700531 except RepoChangedException as rce:
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800532 # If repo changed, re-exec ourselves.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700533 #
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800534 argv = list(sys.argv)
535 argv.extend(rce.extra_args)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700536 try:
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800537 os.execv(__file__, argv)
Sarah Owensa5be53f2012-09-09 15:37:57 -0700538 except OSError as e:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700539 print('fatal: cannot restart repo after upgrade', file=sys.stderr)
540 print('fatal: %s' % e, file=sys.stderr)
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400541 result = 128
542
Renaud Paquaye8595e92016-11-01 15:51:59 -0700543 TerminatePager()
Daniel Sandler3ce2a6b2011-04-29 09:59:12 -0400544 sys.exit(result)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700545
546if __name__ == '__main__':
547 _Main(sys.argv[1:])