Mike Frysinger | f601376 | 2019-06-13 02:30:51 -0400 | [diff] [blame] | 1 | # -*- coding:utf-8 -*- |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 2 | # |
| 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 Owens | cecd1d8 | 2012-11-01 22:59:27 -0700 | [diff] [blame] | 17 | from __future__ import print_function |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 18 | import os |
Anders Björklund | 8e0fe19 | 2020-02-18 14:08:35 +0100 | [diff] [blame] | 19 | import re |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 20 | import sys |
| 21 | import subprocess |
Shawn O. Pearce | fb23161 | 2009-04-10 18:53:46 -0700 | [diff] [blame] | 22 | import tempfile |
Shawn O. Pearce | ca8c32c | 2010-05-11 18:21:33 -0700 | [diff] [blame] | 23 | from signal import SIGTERM |
Renaud Paquay | 2e70291 | 2016-11-01 11:23:38 -0700 | [diff] [blame] | 24 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 25 | from error import GitError |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 26 | from git_refs import HEAD |
Renaud Paquay | 2e70291 | 2016-11-01 11:23:38 -0700 | [diff] [blame] | 27 | import platform_utils |
Mike Frysinger | 8a11f6f | 2019-08-27 00:26:15 -0400 | [diff] [blame] | 28 | from repo_trace import REPO_TRACE, IsTrace, Trace |
Conley Owens | ff0a3c8 | 2014-01-30 14:46:03 -0800 | [diff] [blame] | 29 | from wrapper import Wrapper |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 30 | |
| 31 | GIT = 'git' |
Mike Frysinger | 82caef6 | 2020-02-11 18:51:08 -0500 | [diff] [blame] | 32 | # NB: These do not need to be kept in sync with the repo launcher script. |
| 33 | # These may be much newer as it allows the repo launcher to roll between |
| 34 | # different repo releases while source versions might require a newer git. |
| 35 | # |
| 36 | # The soft version is when we start warning users that the version is old and |
| 37 | # we'll be dropping support for it. We'll refuse to work with versions older |
| 38 | # than the hard version. |
| 39 | # |
| 40 | # git-1.7 is in (EOL) Ubuntu Precise. git-1.9 is in Ubuntu Trusty. |
| 41 | MIN_GIT_VERSION_SOFT = (1, 9, 1) |
| 42 | MIN_GIT_VERSION_HARD = (1, 7, 2) |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 43 | GIT_DIR = 'GIT_DIR' |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 44 | |
| 45 | LAST_GITDIR = None |
| 46 | LAST_CWD = None |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 47 | |
Shawn O. Pearce | fb23161 | 2009-04-10 18:53:46 -0700 | [diff] [blame] | 48 | _ssh_proxy_path = None |
| 49 | _ssh_sock_path = None |
Shawn O. Pearce | ca8c32c | 2010-05-11 18:21:33 -0700 | [diff] [blame] | 50 | _ssh_clients = [] |
Anders Björklund | 8e0fe19 | 2020-02-18 14:08:35 +0100 | [diff] [blame] | 51 | _ssh_version = None |
| 52 | |
| 53 | |
| 54 | def _run_ssh_version(): |
| 55 | """run ssh -V to display the version number""" |
| 56 | return subprocess.check_output(['ssh', '-V'], stderr=subprocess.STDOUT).decode() |
| 57 | |
| 58 | |
| 59 | def _parse_ssh_version(ver_str=None): |
| 60 | """parse a ssh version string into a tuple""" |
| 61 | if ver_str is None: |
| 62 | ver_str = _run_ssh_version() |
| 63 | m = re.match(r'^OpenSSH_([0-9.]+)(p[0-9]+)?\s', ver_str) |
| 64 | if m: |
| 65 | return tuple(int(x) for x in m.group(1).split('.')) |
| 66 | else: |
| 67 | return () |
| 68 | |
| 69 | |
| 70 | def ssh_version(): |
| 71 | """return ssh version as a tuple""" |
| 72 | global _ssh_version |
| 73 | if _ssh_version is None: |
| 74 | try: |
| 75 | _ssh_version = _parse_ssh_version() |
| 76 | except subprocess.CalledProcessError: |
| 77 | print('fatal: unable to detect ssh version', file=sys.stderr) |
| 78 | sys.exit(1) |
| 79 | return _ssh_version |
Shawn O. Pearce | fb23161 | 2009-04-10 18:53:46 -0700 | [diff] [blame] | 80 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 81 | |
Nico Sallembien | 1c85f4e | 2010-04-27 14:35:27 -0700 | [diff] [blame] | 82 | def ssh_sock(create=True): |
Shawn O. Pearce | fb23161 | 2009-04-10 18:53:46 -0700 | [diff] [blame] | 83 | global _ssh_sock_path |
| 84 | if _ssh_sock_path is None: |
| 85 | if not create: |
| 86 | return None |
Mickaël Salaün | 2f6ab7f | 2012-09-30 00:37:55 +0200 | [diff] [blame] | 87 | tmp_dir = '/tmp' |
| 88 | if not os.path.exists(tmp_dir): |
| 89 | tmp_dir = tempfile.gettempdir() |
Anders Björklund | 8e0fe19 | 2020-02-18 14:08:35 +0100 | [diff] [blame] | 90 | if ssh_version() < (6, 7): |
| 91 | tokens = '%r@%h:%p' |
| 92 | else: |
| 93 | tokens = '%C' # hash of %l%h%p%r |
Shawn O. Pearce | fb23161 | 2009-04-10 18:53:46 -0700 | [diff] [blame] | 94 | _ssh_sock_path = os.path.join( |
David Pursehouse | abdf750 | 2020-02-12 14:58:39 +0900 | [diff] [blame] | 95 | tempfile.mkdtemp('', 'ssh-', tmp_dir), |
Anders Björklund | 8e0fe19 | 2020-02-18 14:08:35 +0100 | [diff] [blame] | 96 | 'master-' + tokens) |
Shawn O. Pearce | fb23161 | 2009-04-10 18:53:46 -0700 | [diff] [blame] | 97 | return _ssh_sock_path |
| 98 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 99 | |
Shawn O. Pearce | fb23161 | 2009-04-10 18:53:46 -0700 | [diff] [blame] | 100 | def _ssh_proxy(): |
| 101 | global _ssh_proxy_path |
| 102 | if _ssh_proxy_path is None: |
| 103 | _ssh_proxy_path = os.path.join( |
David Pursehouse | abdf750 | 2020-02-12 14:58:39 +0900 | [diff] [blame] | 104 | os.path.dirname(__file__), |
| 105 | 'git_ssh') |
Shawn O. Pearce | fb23161 | 2009-04-10 18:53:46 -0700 | [diff] [blame] | 106 | return _ssh_proxy_path |
| 107 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 108 | |
Shawn O. Pearce | ca8c32c | 2010-05-11 18:21:33 -0700 | [diff] [blame] | 109 | def _add_ssh_client(p): |
| 110 | _ssh_clients.append(p) |
| 111 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 112 | |
Shawn O. Pearce | ca8c32c | 2010-05-11 18:21:33 -0700 | [diff] [blame] | 113 | def _remove_ssh_client(p): |
| 114 | try: |
| 115 | _ssh_clients.remove(p) |
| 116 | except ValueError: |
| 117 | pass |
| 118 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 119 | |
Shawn O. Pearce | ca8c32c | 2010-05-11 18:21:33 -0700 | [diff] [blame] | 120 | def terminate_ssh_clients(): |
| 121 | global _ssh_clients |
| 122 | for p in _ssh_clients: |
| 123 | try: |
| 124 | os.kill(p.pid, SIGTERM) |
| 125 | p.wait() |
| 126 | except OSError: |
| 127 | pass |
| 128 | _ssh_clients = [] |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 129 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 130 | |
Shawn O. Pearce | 334851e | 2011-09-19 08:05:31 -0700 | [diff] [blame] | 131 | _git_version = None |
| 132 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 133 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 134 | class _GitCall(object): |
Shawn O. Pearce | 334851e | 2011-09-19 08:05:31 -0700 | [diff] [blame] | 135 | def version_tuple(self): |
| 136 | global _git_version |
Shawn O. Pearce | 334851e | 2011-09-19 08:05:31 -0700 | [diff] [blame] | 137 | if _git_version is None: |
Mike Frysinger | ca540ae | 2019-07-10 15:42:30 -0400 | [diff] [blame] | 138 | _git_version = Wrapper().ParseGitVersion() |
Conley Owens | ff0a3c8 | 2014-01-30 14:46:03 -0800 | [diff] [blame] | 139 | if _git_version is None: |
Mike Frysinger | ca540ae | 2019-07-10 15:42:30 -0400 | [diff] [blame] | 140 | print('fatal: unable to detect git version', file=sys.stderr) |
Shawn O. Pearce | 334851e | 2011-09-19 08:05:31 -0700 | [diff] [blame] | 141 | sys.exit(1) |
| 142 | return _git_version |
| 143 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 144 | def __getattr__(self, name): |
David Pursehouse | 54a4e60 | 2020-02-12 14:31:05 +0900 | [diff] [blame] | 145 | name = name.replace('_', '-') |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 146 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 147 | def fun(*cmdv): |
| 148 | command = [name] |
| 149 | command.extend(cmdv) |
| 150 | return GitCommand(None, command).Wait() == 0 |
| 151 | return fun |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 152 | |
| 153 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 154 | git = _GitCall() |
| 155 | |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 156 | |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 157 | def RepoSourceVersion(): |
| 158 | """Return the version of the repo.git tree.""" |
| 159 | ver = getattr(RepoSourceVersion, 'version', None) |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 160 | |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 161 | # We avoid GitCommand so we don't run into circular deps -- GitCommand needs |
| 162 | # to initialize version info we provide. |
| 163 | if ver is None: |
| 164 | env = GitCommand._GetBasicEnv() |
| 165 | |
| 166 | proj = os.path.dirname(os.path.abspath(__file__)) |
| 167 | env[GIT_DIR] = os.path.join(proj, '.git') |
| 168 | |
| 169 | p = subprocess.Popen([GIT, 'describe', HEAD], stdout=subprocess.PIPE, |
| 170 | env=env) |
| 171 | if p.wait() == 0: |
| 172 | ver = p.stdout.read().strip().decode('utf-8') |
| 173 | if ver.startswith('v'): |
| 174 | ver = ver[1:] |
| 175 | else: |
| 176 | ver = 'unknown' |
| 177 | setattr(RepoSourceVersion, 'version', ver) |
| 178 | |
| 179 | return ver |
| 180 | |
| 181 | |
| 182 | class UserAgent(object): |
| 183 | """Mange User-Agent settings when talking to external services |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 184 | |
| 185 | We follow the style as documented here: |
| 186 | https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/User-Agent |
| 187 | """ |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 188 | |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 189 | _os = None |
| 190 | _repo_ua = None |
Mike Frysinger | 2f0951b | 2019-07-10 17:13:46 -0400 | [diff] [blame] | 191 | _git_ua = None |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 192 | |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 193 | @property |
| 194 | def os(self): |
| 195 | """The operating system name.""" |
| 196 | if self._os is None: |
| 197 | os_name = sys.platform |
| 198 | if os_name.lower().startswith('linux'): |
| 199 | os_name = 'Linux' |
| 200 | elif os_name == 'win32': |
| 201 | os_name = 'Win32' |
| 202 | elif os_name == 'cygwin': |
| 203 | os_name = 'Cygwin' |
| 204 | elif os_name == 'darwin': |
| 205 | os_name = 'Darwin' |
| 206 | self._os = os_name |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 207 | |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 208 | return self._os |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 209 | |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 210 | @property |
| 211 | def repo(self): |
| 212 | """The UA when connecting directly from repo.""" |
| 213 | if self._repo_ua is None: |
| 214 | py_version = sys.version_info |
| 215 | self._repo_ua = 'git-repo/%s (%s) git/%s Python/%d.%d.%d' % ( |
| 216 | RepoSourceVersion(), |
| 217 | self.os, |
| 218 | git.version_tuple().full, |
| 219 | py_version.major, py_version.minor, py_version.micro) |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 220 | |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 221 | return self._repo_ua |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 222 | |
Mike Frysinger | 2f0951b | 2019-07-10 17:13:46 -0400 | [diff] [blame] | 223 | @property |
| 224 | def git(self): |
| 225 | """The UA when running git.""" |
| 226 | if self._git_ua is None: |
| 227 | self._git_ua = 'git/%s (%s) git-repo/%s' % ( |
| 228 | git.version_tuple().full, |
| 229 | self.os, |
| 230 | RepoSourceVersion()) |
| 231 | |
| 232 | return self._git_ua |
| 233 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 234 | |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 235 | user_agent = UserAgent() |
Mike Frysinger | 369814b | 2019-07-10 17:10:07 -0400 | [diff] [blame] | 236 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 237 | |
Xin Li | 745be2e | 2019-06-03 11:24:30 -0700 | [diff] [blame] | 238 | def git_require(min_version, fail=False, msg=''): |
Shawn O. Pearce | 334851e | 2011-09-19 08:05:31 -0700 | [diff] [blame] | 239 | git_version = git.version_tuple() |
| 240 | if min_version <= git_version: |
Shawn O. Pearce | 2ec00b9 | 2009-06-12 09:32:50 -0700 | [diff] [blame] | 241 | return True |
| 242 | if fail: |
David Pursehouse | 7e6dd2d | 2012-10-25 12:40:51 +0900 | [diff] [blame] | 243 | need = '.'.join(map(str, min_version)) |
Xin Li | 745be2e | 2019-06-03 11:24:30 -0700 | [diff] [blame] | 244 | if msg: |
| 245 | msg = ' for ' + msg |
| 246 | print('fatal: git %s or later required%s' % (need, msg), file=sys.stderr) |
Shawn O. Pearce | 2ec00b9 | 2009-06-12 09:32:50 -0700 | [diff] [blame] | 247 | sys.exit(1) |
| 248 | return False |
| 249 | |
David Pursehouse | 819827a | 2020-02-12 15:20:19 +0900 | [diff] [blame] | 250 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 251 | class GitCommand(object): |
| 252 | def __init__(self, |
| 253 | project, |
| 254 | cmdv, |
David Pursehouse | e5913ae | 2020-02-12 13:56:59 +0900 | [diff] [blame] | 255 | bare=False, |
| 256 | provide_stdin=False, |
| 257 | capture_stdout=False, |
| 258 | capture_stderr=False, |
Mike Frysinger | 31990f0 | 2020-02-17 01:35:18 -0500 | [diff] [blame] | 259 | merge_output=False, |
David Pursehouse | e5913ae | 2020-02-12 13:56:59 +0900 | [diff] [blame] | 260 | disable_editor=False, |
| 261 | ssh_proxy=False, |
| 262 | cwd=None, |
| 263 | gitdir=None): |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 264 | env = self._GetBasicEnv() |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 265 | |
John L. Villalovos | 9c76f67 | 2015-03-16 20:49:10 -0700 | [diff] [blame] | 266 | # If we are not capturing std* then need to print it. |
| 267 | self.tee = {'stdout': not capture_stdout, 'stderr': not capture_stderr} |
| 268 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 269 | if disable_editor: |
Mike Frysinger | 56ce346 | 2019-12-04 19:30:48 -0500 | [diff] [blame] | 270 | env['GIT_EDITOR'] = ':' |
Shawn O. Pearce | fb23161 | 2009-04-10 18:53:46 -0700 | [diff] [blame] | 271 | if ssh_proxy: |
Mike Frysinger | 56ce346 | 2019-12-04 19:30:48 -0500 | [diff] [blame] | 272 | env['REPO_SSH_SOCK'] = ssh_sock() |
| 273 | env['GIT_SSH'] = _ssh_proxy() |
| 274 | env['GIT_SSH_VARIANT'] = 'ssh' |
Shawn O. Pearce | 62d0b10 | 2012-06-05 15:11:15 -0700 | [diff] [blame] | 275 | if 'http_proxy' in env and 'darwin' == sys.platform: |
Shawn O. Pearce | 337aee0 | 2012-06-13 10:40:46 -0700 | [diff] [blame] | 276 | s = "'http.proxy=%s'" % (env['http_proxy'],) |
Shawn O. Pearce | 62d0b10 | 2012-06-05 15:11:15 -0700 | [diff] [blame] | 277 | p = env.get('GIT_CONFIG_PARAMETERS') |
| 278 | if p is not None: |
| 279 | s = p + ' ' + s |
Mike Frysinger | 56ce346 | 2019-12-04 19:30:48 -0500 | [diff] [blame] | 280 | env['GIT_CONFIG_PARAMETERS'] = s |
Dan Willemsen | 466b8c4 | 2015-11-25 13:26:39 -0800 | [diff] [blame] | 281 | if 'GIT_ALLOW_PROTOCOL' not in env: |
Mike Frysinger | 56ce346 | 2019-12-04 19:30:48 -0500 | [diff] [blame] | 282 | env['GIT_ALLOW_PROTOCOL'] = ( |
| 283 | 'file:git:http:https:ssh:persistent-http:persistent-https:sso:rpc') |
| 284 | env['GIT_HTTP_USER_AGENT'] = user_agent.git |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 285 | |
| 286 | if project: |
| 287 | if not cwd: |
| 288 | cwd = project.worktree |
| 289 | if not gitdir: |
| 290 | gitdir = project.gitdir |
| 291 | |
| 292 | command = [GIT] |
| 293 | if bare: |
| 294 | if gitdir: |
Mike Frysinger | 56ce346 | 2019-12-04 19:30:48 -0500 | [diff] [blame] | 295 | env[GIT_DIR] = gitdir |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 296 | cwd = None |
John L. Villalovos | 9c76f67 | 2015-03-16 20:49:10 -0700 | [diff] [blame] | 297 | command.append(cmdv[0]) |
| 298 | # Need to use the --progress flag for fetch/clone so output will be |
| 299 | # displayed as by default git only does progress output if stderr is a TTY. |
| 300 | if sys.stderr.isatty() and cmdv[0] in ('fetch', 'clone'): |
| 301 | if '--progress' not in cmdv and '--quiet' not in cmdv: |
| 302 | command.append('--progress') |
| 303 | command.extend(cmdv[1:]) |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 304 | |
| 305 | if provide_stdin: |
| 306 | stdin = subprocess.PIPE |
| 307 | else: |
| 308 | stdin = None |
| 309 | |
John L. Villalovos | 9c76f67 | 2015-03-16 20:49:10 -0700 | [diff] [blame] | 310 | stdout = subprocess.PIPE |
Mike Frysinger | 31990f0 | 2020-02-17 01:35:18 -0500 | [diff] [blame] | 311 | stderr = subprocess.STDOUT if merge_output else subprocess.PIPE |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 312 | |
Shawn O. Pearce | ad3193a | 2009-04-18 09:54:51 -0700 | [diff] [blame] | 313 | if IsTrace(): |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 314 | global LAST_CWD |
| 315 | global LAST_GITDIR |
| 316 | |
| 317 | dbg = '' |
| 318 | |
| 319 | if cwd and LAST_CWD != cwd: |
| 320 | if LAST_GITDIR or LAST_CWD: |
| 321 | dbg += '\n' |
| 322 | dbg += ': cd %s\n' % cwd |
| 323 | LAST_CWD = cwd |
| 324 | |
| 325 | if GIT_DIR in env and LAST_GITDIR != env[GIT_DIR]: |
| 326 | if LAST_GITDIR or LAST_CWD: |
| 327 | dbg += '\n' |
| 328 | dbg += ': export GIT_DIR=%s\n' % env[GIT_DIR] |
| 329 | LAST_GITDIR = env[GIT_DIR] |
| 330 | |
| 331 | dbg += ': ' |
| 332 | dbg += ' '.join(command) |
| 333 | if stdin == subprocess.PIPE: |
| 334 | dbg += ' 0<|' |
| 335 | if stdout == subprocess.PIPE: |
| 336 | dbg += ' 1>|' |
| 337 | if stderr == subprocess.PIPE: |
| 338 | dbg += ' 2>|' |
Mike Frysinger | 31990f0 | 2020-02-17 01:35:18 -0500 | [diff] [blame] | 339 | elif stderr == subprocess.STDOUT: |
| 340 | dbg += ' 2>&1' |
Shawn O. Pearce | ad3193a | 2009-04-18 09:54:51 -0700 | [diff] [blame] | 341 | Trace('%s', dbg) |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 342 | |
| 343 | try: |
| 344 | p = subprocess.Popen(command, |
David Pursehouse | e5913ae | 2020-02-12 13:56:59 +0900 | [diff] [blame] | 345 | cwd=cwd, |
| 346 | env=env, |
| 347 | stdin=stdin, |
| 348 | stdout=stdout, |
| 349 | stderr=stderr) |
Sarah Owens | a5be53f | 2012-09-09 15:37:57 -0700 | [diff] [blame] | 350 | except Exception as e: |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 351 | raise GitError('%s: %s' % (command[1], e)) |
| 352 | |
Shawn O. Pearce | ca8c32c | 2010-05-11 18:21:33 -0700 | [diff] [blame] | 353 | if ssh_proxy: |
| 354 | _add_ssh_client(p) |
| 355 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 356 | self.process = p |
| 357 | self.stdin = p.stdin |
| 358 | |
Mike Frysinger | 71b0f31 | 2019-09-30 22:39:49 -0400 | [diff] [blame] | 359 | @staticmethod |
| 360 | def _GetBasicEnv(): |
| 361 | """Return a basic env for running git under. |
| 362 | |
| 363 | This is guaranteed to be side-effect free. |
| 364 | """ |
| 365 | env = os.environ.copy() |
| 366 | for key in (REPO_TRACE, |
| 367 | GIT_DIR, |
| 368 | 'GIT_ALTERNATE_OBJECT_DIRECTORIES', |
| 369 | 'GIT_OBJECT_DIRECTORY', |
| 370 | 'GIT_WORK_TREE', |
| 371 | 'GIT_GRAFT_FILE', |
| 372 | 'GIT_INDEX_FILE'): |
| 373 | env.pop(key, None) |
| 374 | return env |
| 375 | |
The Android Open Source Project | cf31fe9 | 2008-10-21 07:00:00 -0700 | [diff] [blame] | 376 | def Wait(self): |
Shawn O. Pearce | ca8c32c | 2010-05-11 18:21:33 -0700 | [diff] [blame] | 377 | try: |
Ulrik Sjölin | 498fe90 | 2011-09-11 22:59:37 +0200 | [diff] [blame] | 378 | p = self.process |
John L. Villalovos | 9c76f67 | 2015-03-16 20:49:10 -0700 | [diff] [blame] | 379 | rc = self._CaptureOutput() |
Shawn O. Pearce | ca8c32c | 2010-05-11 18:21:33 -0700 | [diff] [blame] | 380 | finally: |
| 381 | _remove_ssh_client(p) |
| 382 | return rc |
John L. Villalovos | 9c76f67 | 2015-03-16 20:49:10 -0700 | [diff] [blame] | 383 | |
| 384 | def _CaptureOutput(self): |
| 385 | p = self.process |
Renaud Paquay | 2e70291 | 2016-11-01 11:23:38 -0700 | [diff] [blame] | 386 | s_in = platform_utils.FileDescriptorStreams.create() |
| 387 | s_in.add(p.stdout, sys.stdout, 'stdout') |
Mike Frysinger | 31990f0 | 2020-02-17 01:35:18 -0500 | [diff] [blame] | 388 | if p.stderr is not None: |
| 389 | s_in.add(p.stderr, sys.stderr, 'stderr') |
John L. Villalovos | 9c76f67 | 2015-03-16 20:49:10 -0700 | [diff] [blame] | 390 | self.stdout = '' |
| 391 | self.stderr = '' |
| 392 | |
Renaud Paquay | 2e70291 | 2016-11-01 11:23:38 -0700 | [diff] [blame] | 393 | while not s_in.is_done: |
| 394 | in_ready = s_in.select() |
John L. Villalovos | 9c76f67 | 2015-03-16 20:49:10 -0700 | [diff] [blame] | 395 | for s in in_ready: |
Renaud Paquay | 2e70291 | 2016-11-01 11:23:38 -0700 | [diff] [blame] | 396 | buf = s.read() |
John L. Villalovos | 9c76f67 | 2015-03-16 20:49:10 -0700 | [diff] [blame] | 397 | if not buf: |
| 398 | s_in.remove(s) |
| 399 | continue |
Anthony King | 6cfc68e | 2015-06-03 16:39:32 +0100 | [diff] [blame] | 400 | if not hasattr(buf, 'encode'): |
| 401 | buf = buf.decode() |
John L. Villalovos | 9c76f67 | 2015-03-16 20:49:10 -0700 | [diff] [blame] | 402 | if s.std_name == 'stdout': |
| 403 | self.stdout += buf |
| 404 | else: |
| 405 | self.stderr += buf |
| 406 | if self.tee[s.std_name]: |
| 407 | s.dest.write(buf) |
| 408 | s.dest.flush() |
| 409 | return p.wait() |