blob: 1c49b46ee1a4687dee1ca3b8dc8485ed12be97b7 [file] [log] [blame]
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001# Copyright (C) 2008 The Android Open Source Project
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14
LaMont Jonesbdcba7d2022-04-11 22:50:11 +000015import collections
Mike Frysingerebf04a42021-02-23 20:48:04 -050016import functools
Mike Frysingeracf63b22019-06-13 02:24:21 -040017import http.cookiejar as cookielib
Mike Frysinger7b586f22021-02-23 18:38:39 -050018import io
Anthony King85b24ac2014-05-06 15:57:48 +010019import json
Mike Frysingerebf04a42021-02-23 20:48:04 -050020import multiprocessing
David Pursehouse86d973d2012-08-24 10:21:02 +090021import netrc
Shawn O. Pearce2a1ccb22009-04-10 16:51:53 -070022from optparse import SUPPRESS_HELP
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070023import os
LaMont Jones891e8f72022-09-08 20:17:58 +000024import shutil
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -070025import socket
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070026import sys
Dan Willemsen0745bb22015-08-17 13:41:45 -070027import tempfile
Shawn O. Pearcef6906872009-04-18 10:49:00 -070028import time
LaMont Jones1eddca82022-09-01 15:15:04 +000029from typing import NamedTuple
Mike Frysingeracf63b22019-06-13 02:24:21 -040030import urllib.error
31import urllib.parse
32import urllib.request
Mike Frysinger5951e302022-05-20 23:34:44 -040033import xml.parsers.expat
Mike Frysingeracf63b22019-06-13 02:24:21 -040034import xmlrpc.client
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070035
Roy Lee18afd7f2010-05-09 04:32:08 +080036try:
37 import threading as _threading
38except ImportError:
39 import dummy_threading as _threading
40
Shawn O. Pearce97d2b2f2011-09-22 17:23:41 -070041try:
42 import resource
David Pursehouse819827a2020-02-12 15:20:19 +090043
Shawn O. Pearce97d2b2f2011-09-22 17:23:41 -070044 def _rlimit_nofile():
45 return resource.getrlimit(resource.RLIMIT_NOFILE)
46except ImportError:
47 def _rlimit_nofile():
48 return (256, 256)
49
David Rileye0684ad2017-04-05 00:02:59 -070050import event_log
Mike Frysinger347f9ed2021-03-15 14:58:52 -040051from git_command import git_require
David Pursehouseba7bc732015-08-20 16:55:42 +090052from git_config import GetUrlCookieFile
David Pursehoused94aaef2012-09-07 09:52:04 +090053from git_refs import R_HEADS, HEAD
Raman Tenneti6a872c92021-01-14 19:17:50 -080054import git_superproject
Simran Basibdb52712015-08-10 13:23:23 -070055import gitc_utils
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -070056from project import Project
57from project import RemoteSpec
Mike Frysinger355f4392022-07-20 17:15:29 -040058from command import Command, DEFAULT_LOCAL_JOBS, MirrorSafeCommand, WORKER_BATCH_SIZE
Raman Tenneti1fd7bc22021-02-04 14:39:38 -080059from error import RepoChangedException, GitError, ManifestParseError
Renaud Paquaya65adf72016-11-03 10:37:53 -070060import platform_utils
Shawn O. Pearce350cde42009-04-16 11:21:18 -070061from project import SyncBuffer
Shawn O. Pearce68194f42009-04-10 16:48:52 -070062from progress import Progress
LaMont Jones891e8f72022-09-08 20:17:58 +000063from repo_trace import IsTrace, Trace
Mike Frysinger19e409c2021-05-05 19:44:35 -040064import ssh
Conley Owens094cdbe2014-01-30 15:09:59 -080065from wrapper import Wrapper
Dan Willemsen5ea32d12015-09-08 13:27:20 -070066from manifest_xml import GitcManifest
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070067
Dave Borowitz67700e92012-10-23 15:00:54 -070068_ONE_DAY_S = 24 * 60 * 60
LaMont Jones891e8f72022-09-08 20:17:58 +000069# Env var to implicitly turn off object backups.
70REPO_BACKUP_OBJECTS = 'REPO_BACKUP_OBJECTS'
71
72_BACKUP_OBJECTS = os.environ.get(REPO_BACKUP_OBJECTS) != '0'
Dave Borowitz67700e92012-10-23 15:00:54 -070073
David Pursehouse819827a2020-02-12 15:20:19 +090074
LaMont Jones1eddca82022-09-01 15:15:04 +000075class _FetchOneResult(NamedTuple):
76 """_FetchOne return value.
77
78 Attributes:
79 success (bool): True if successful.
80 project (Project): The fetched project.
81 start (float): The starting time.time().
82 finish (float): The ending time.time().
83 remote_fetched (bool): True if the remote was actually queried.
84 """
85 success: bool
86 project: Project
87 start: float
88 finish: float
89 remote_fetched: bool
90
91
92class _FetchResult(NamedTuple):
93 """_Fetch return value.
94
95 Attributes:
96 success (bool): True if successful.
97 projects (set[str]): The names of the git directories of fetched projects.
98 """
99 success: bool
100 projects: set[str]
101
102
103class _FetchMainResult(NamedTuple):
104 """_FetchMain return value.
105
106 Attributes:
107 all_projects (list[Project]): The fetched projects.
108 """
109 all_projects: list[Project]
110
111
112class _CheckoutOneResult(NamedTuple):
113 """_CheckoutOne return value.
114
115 Attributes:
116 success (bool): True if successful.
117 project (Project): The project.
118 start (float): The starting time.time().
119 finish (float): The ending time.time().
120 """
121 success: bool
122 project: Project
123 start: float
124 finish: float
125
126
Shawn O. Pearcec95583b2009-03-03 17:47:06 -0800127class Sync(Command, MirrorSafeCommand):
Mike Frysinger4f210542021-06-14 16:05:19 -0400128 COMMON = True
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000129 MULTI_MANIFEST_SUPPORT = True
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700130 helpSummary = "Update working tree to the latest revision"
131 helpUsage = """
132%prog [<project>...]
133"""
134 helpDescription = """
135The '%prog' command synchronizes local project directories
136with the remote repositories specified in the manifest. If a local
137project does not yet exist, it will clone a new local directory from
138the remote repository and set up tracking branches as specified in
139the manifest. If the local project already exists, '%prog'
140will update the remote branches and rebase any new local changes
141on top of the new remote changes.
142
143'%prog' will synchronize all projects listed at the command
144line. Projects can be specified either by name, or by a relative
145or absolute path to the project's local directory. If no projects
146are specified, '%prog' will synchronize all projects listed in
147the manifest.
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700148
149The -d/--detach option can be used to switch specified projects
150back to the manifest revision. This option is especially helpful
151if the project is currently on a topic branch, but the manifest
152revision is temporarily needed.
Shawn O. Pearceeb7af872009-04-21 08:02:04 -0700153
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700154The -s/--smart-sync option can be used to sync to a known good
155build as specified by the manifest-server element in the current
Victor Boivie08c880d2011-04-19 10:32:52 +0200156manifest. The -t/--smart-tag option is similar and allows you to
157specify a custom tag/label.
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700158
David Pursehousecf76b1b2012-09-14 10:31:42 +0900159The -u/--manifest-server-username and -p/--manifest-server-password
160options can be used to specify a username and password to authenticate
161with the manifest server when using the -s or -t option.
162
163If -u and -p are not specified when using the -s or -t option, '%prog'
164will attempt to read authentication credentials for the manifest server
165from the user's .netrc file.
166
167'%prog' will not use authentication credentials from -u/-p or .netrc
168if the manifest server specified in the manifest file already includes
169credentials.
170
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400171By default, all projects will be synced. The --fail-fast option can be used
Mike Frysinger7ae210a2020-05-24 14:56:52 -0400172to halt syncing as soon as possible when the first project fails to sync.
Andrei Warkentin5df6de02010-07-02 17:58:31 -0500173
Kevin Degiabaa7f32014-11-12 11:27:45 -0700174The --force-sync option can be used to overwrite existing git
175directories if they have previously been linked to a different
Roger Shimizuac29ac32020-06-06 02:33:40 +0900176object directory. WARNING: This may cause data to be lost since
Kevin Degiabaa7f32014-11-12 11:27:45 -0700177refs may be removed when overwriting.
178
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500179The --force-remove-dirty option can be used to remove previously used
180projects with uncommitted changes. WARNING: This may cause data to be
181lost since uncommitted changes may be removed with projects that no longer
182exist in the manifest.
183
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -0700184The --no-clone-bundle option disables any attempt to use
185$URL/clone.bundle to bootstrap a new Git repository from a
186resumeable bundle file on a content delivery network. This
187may be necessary if there are problems with the local Python
188HTTP client or proxy configuration, but the Git binary works.
189
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800190The --fetch-submodules option enables fetching Git submodules
191of a project from server.
192
David Pursehousef2fad612015-01-29 14:36:28 +0900193The -c/--current-branch option can be used to only fetch objects that
194are on the branch specified by a project's revision.
195
David Pursehouseb1553542014-09-04 21:28:09 +0900196The --optimized-fetch option can be used to only fetch projects that
197are fixed to a sha1 revision if the sha1 revision does not already
198exist locally.
199
David Pursehouse74cfd272015-10-14 10:50:15 +0900200The --prune option can be used to remove any refs that no longer
201exist on the remote.
202
Mike Frysingerb8f7bb02018-10-10 01:05:11 -0400203# SSH Connections
Shawn O. Pearceeb7af872009-04-21 08:02:04 -0700204
205If at least one project remote URL uses an SSH connection (ssh://,
206git+ssh://, or user@host:path syntax) repo will automatically
207enable the SSH ControlMaster option when connecting to that host.
208This feature permits other projects in the same '%prog' session to
209reuse the same SSH tunnel, saving connection setup overheads.
210
211To disable this behavior on UNIX platforms, set the GIT_SSH
212environment variable to 'ssh'. For example:
213
214 export GIT_SSH=ssh
215 %prog
216
Mike Frysingerb8f7bb02018-10-10 01:05:11 -0400217# Compatibility
Shawn O. Pearceeb7af872009-04-21 08:02:04 -0700218
219This feature is automatically disabled on Windows, due to the lack
220of UNIX domain socket support.
221
222This feature is not compatible with url.insteadof rewrites in the
223user's ~/.gitconfig. '%prog' is currently not able to perform the
224rewrite early enough to establish the ControlMaster tunnel.
225
226If the remote SSH daemon is Gerrit Code Review, version 2.0.10 or
227later is required to fix a server side protocol bug.
228
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700229"""
Mike Frysinger355f4392022-07-20 17:15:29 -0400230 # A value of 0 means we want parallel jobs, but we'll determine the default
231 # value later on.
232 PARALLEL_JOBS = 0
Shawn O. Pearce6392c872011-09-22 17:44:31 -0700233
Mike Frysinger9180a072021-04-13 14:57:40 -0400234 def _Options(self, p, show_smart=True):
Mike Frysinger49de8ef2021-04-09 00:21:02 -0400235 p.add_option('--jobs-network', default=None, type=int, metavar='JOBS',
Mike Frysinger355f4392022-07-20 17:15:29 -0400236 help='number of network jobs to run in parallel (defaults to --jobs or 1)')
Mike Frysinger49de8ef2021-04-09 00:21:02 -0400237 p.add_option('--jobs-checkout', default=None, type=int, metavar='JOBS',
Mike Frysinger355f4392022-07-20 17:15:29 -0400238 help='number of local checkout jobs to run in parallel (defaults to --jobs or '
239 f'{DEFAULT_LOCAL_JOBS})')
Mike Frysinger49de8ef2021-04-09 00:21:02 -0400240
Andrei Warkentin5df6de02010-07-02 17:58:31 -0500241 p.add_option('-f', '--force-broken',
Stefan Müller-Klieser46702ed2019-08-30 10:20:15 +0200242 dest='force_broken', action='store_true',
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400243 help='obsolete option (to be deleted in the future)')
244 p.add_option('--fail-fast',
245 dest='fail_fast', action='store_true',
246 help='stop syncing after first error is hit')
Kevin Degiabaa7f32014-11-12 11:27:45 -0700247 p.add_option('--force-sync',
248 dest='force_sync', action='store_true',
249 help="overwrite an existing git directory if it needs to "
250 "point to a different object directory. WARNING: this "
251 "may cause loss of data")
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500252 p.add_option('--force-remove-dirty',
253 dest='force_remove_dirty', action='store_true',
254 help="force remove projects with uncommitted modifications if "
255 "projects no longer exist in the manifest. "
256 "WARNING: this may cause loss of data")
David Pursehouse8f62fb72012-11-14 12:09:38 +0900257 p.add_option('-l', '--local-only',
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700258 dest='local_only', action='store_true',
259 help="only update working tree, don't fetch")
David Pursehouse54a4e602020-02-12 14:31:05 +0900260 p.add_option('--no-manifest-update', '--nmu',
Fredrik de Grootcc960972019-11-22 09:04:31 +0100261 dest='mp_update', action='store_false', default='true',
262 help='use the existing manifest checkout as-is. '
263 '(do not update to the latest revision)')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900264 p.add_option('-n', '--network-only',
Shawn O. Pearce96fdcef2009-04-10 16:29:20 -0700265 dest='network_only', action='store_true',
266 help="fetch only, don't update working tree")
David Pursehouse8f62fb72012-11-14 12:09:38 +0900267 p.add_option('-d', '--detach',
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700268 dest='detach_head', action='store_true',
269 help='detach projects back to manifest revision')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900270 p.add_option('-c', '--current-branch',
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -0700271 dest='current_branch_only', action='store_true',
272 help='fetch only current branch from server')
Mike Frysinger73561142021-05-03 01:10:09 -0400273 p.add_option('--no-current-branch',
274 dest='current_branch_only', action='store_false',
275 help='fetch all branches from server')
Chris Wolfee9dc3b32012-01-26 11:36:18 -0500276 p.add_option('-m', '--manifest-name',
277 dest='manifest_name',
278 help='temporary manifest to use for this sync', metavar='NAME.xml')
Xin Lid79a4bc2020-05-20 16:03:45 -0700279 p.add_option('--clone-bundle', action='store_true',
280 help='enable use of /clone.bundle on HTTP/HTTPS')
281 p.add_option('--no-clone-bundle', dest='clone_bundle', action='store_false',
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -0700282 help='disable use of /clone.bundle on HTTP/HTTPS')
Conley Owens8d070cf2012-11-06 13:14:31 -0800283 p.add_option('-u', '--manifest-server-username', action='store',
284 dest='manifest_server_username',
285 help='username to authenticate with the manifest server')
286 p.add_option('-p', '--manifest-server-password', action='store',
287 dest='manifest_server_password',
288 help='password to authenticate with the manifest server')
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800289 p.add_option('--fetch-submodules',
290 dest='fetch_submodules', action='store_true',
291 help='fetch submodules from server')
Raman Tenneti6a872c92021-01-14 19:17:50 -0800292 p.add_option('--use-superproject', action='store_true',
Raman Tenneti62517292021-11-01 14:49:16 -0700293 help='use the manifest superproject to sync projects; implies -c')
Raman Tenneti23ea7542021-05-07 14:01:54 -0700294 p.add_option('--no-use-superproject', action='store_false',
295 dest='use_superproject',
296 help='disable use of manifest superprojects')
Mike Frysinger2273f462021-11-05 15:10:33 -0400297 p.add_option('--tags', action='store_true',
Mike Frysingerd68ed632021-05-03 01:21:35 -0400298 help='fetch tags')
Mitchel Humpherys597868b2012-10-29 10:18:34 -0700299 p.add_option('--no-tags',
Mike Frysingerd68ed632021-05-03 01:21:35 -0400300 dest='tags', action='store_false',
Mike Frysinger2273f462021-11-05 15:10:33 -0400301 help="don't fetch tags (default)")
David Pursehouseb1553542014-09-04 21:28:09 +0900302 p.add_option('--optimized-fetch',
303 dest='optimized_fetch', action='store_true',
304 help='only fetch projects fixed to sha1 if revision does not exist locally')
George Engelbrecht9bc283e2020-04-02 12:36:09 -0600305 p.add_option('--retry-fetches',
306 default=0, action='store', type='int',
307 help='number of times to retry fetches on transient errors')
Mike Frysinger0531a622021-11-05 15:22:01 -0400308 p.add_option('--prune', action='store_true',
309 help='delete refs that no longer exist on the remote (default)')
310 p.add_option('--no-prune', dest='prune', action='store_false',
311 help='do not delete refs that no longer exist on the remote')
Nico Sallembien6623b212010-05-11 12:57:01 -0700312 if show_smart:
313 p.add_option('-s', '--smart-sync',
314 dest='smart_sync', action='store_true',
David Pursehouse79fba682016-04-13 18:03:00 +0900315 help='smart sync using manifest from the latest known good build')
Victor Boivie08c880d2011-04-19 10:32:52 +0200316 p.add_option('-t', '--smart-tag',
317 dest='smart_tag', action='store',
318 help='smart sync using manifest from a known tag')
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700319
Shawn O. Pearcefd89b672009-04-18 11:28:57 -0700320 g = p.add_option_group('repo Version options')
321 g.add_option('--no-repo-verify',
Mike Frysingerc58ec4d2020-02-17 14:36:08 -0500322 dest='repo_verify', default=True, action='store_false',
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700323 help='do not verify repo source code')
Shawn O. Pearcefd89b672009-04-18 11:28:57 -0700324 g.add_option('--repo-upgraded',
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800325 dest='repo_upgraded', action='store_true',
Shawn O. Pearce2a1ccb22009-04-10 16:51:53 -0700326 help=SUPPRESS_HELP)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700327
LaMont Jonesa46047a2022-04-07 21:57:06 +0000328 def _GetBranch(self, manifest_project):
329 """Returns the branch name for getting the approved smartsync manifest.
330
331 Args:
332 manifest_project: the manifestProject to query.
333 """
334 b = manifest_project.GetBranch(manifest_project.CurrentBranch)
Raman Tenneti8d43dea2021-02-07 16:30:27 -0800335 branch = b.merge
336 if branch.startswith(R_HEADS):
337 branch = branch[len(R_HEADS):]
338 return branch
339
LaMont Jonesa46047a2022-04-07 21:57:06 +0000340 def _GetCurrentBranchOnly(self, opt, manifest):
Daniel Anderssond52ca422022-04-01 12:55:38 +0200341 """Returns whether current-branch or use-superproject options are enabled.
342
LaMont Jonesa46047a2022-04-07 21:57:06 +0000343 Args:
344 opt: Program options returned from optparse. See _Options().
345 manifest: The manifest to use.
346
Daniel Anderssond52ca422022-04-01 12:55:38 +0200347 Returns:
348 True if a superproject is requested, otherwise the value of the
349 current_branch option (True, False or None).
350 """
LaMont Jonesa46047a2022-04-07 21:57:06 +0000351 return git_superproject.UseSuperproject(opt.use_superproject, manifest) or opt.current_branch_only
Raman Tenneti2ae44d72021-03-23 15:12:27 -0700352
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000353 def _UpdateProjectsRevisionId(self, opt, args, superproject_logging_data,
354 manifest):
355 """Update revisionId of projects with the commit hash from the superproject.
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800356
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000357 This function updates each project's revisionId with the commit hash from
358 the superproject. It writes the updated manifest into a file and reloads
359 the manifest from it. When appropriate, sub manifests are also processed.
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800360
361 Args:
362 opt: Program options returned from optparse. See _Options().
363 args: Arguments to pass to GetProjects. See the GetProjects
364 docstring for details.
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000365 superproject_logging_data: A dictionary of superproject data to log.
LaMont Jonesa46047a2022-04-07 21:57:06 +0000366 manifest: The manifest to use.
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800367 """
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000368 have_superproject = manifest.superproject or any(
369 m.superproject for m in manifest.all_children)
370 if not have_superproject:
371 return
372
LaMont Jonesff6b1da2022-06-01 21:03:34 +0000373 if opt.local_only and manifest.superproject:
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000374 manifest_path = manifest.superproject.manifest_path
Raman Tennetiae86a462021-07-27 08:54:59 -0700375 if manifest_path:
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000376 self._ReloadManifest(manifest_path, manifest)
377 return
Raman Tennetiae86a462021-07-27 08:54:59 -0700378
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800379 all_projects = self.GetProjects(args,
380 missing_ok=True,
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000381 submodules_ok=opt.fetch_submodules,
382 manifest=manifest,
383 all_manifests=not opt.this_manifest_only)
384
385 per_manifest = collections.defaultdict(list)
386 manifest_paths = {}
387 if opt.this_manifest_only:
388 per_manifest[manifest.path_prefix] = all_projects
Raman Tenneti784e16f2021-06-11 17:29:45 -0700389 else:
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000390 for p in all_projects:
391 per_manifest[p.manifest.path_prefix].append(p)
392
393 superproject_logging_data = {}
394 need_unload = False
395 for m in self.ManifestList(opt):
396 if not m.path_prefix in per_manifest:
397 continue
398 use_super = git_superproject.UseSuperproject(opt.use_superproject, m)
399 if superproject_logging_data:
400 superproject_logging_data['multimanifest'] = True
401 superproject_logging_data.update(
402 superproject=use_super,
403 haslocalmanifests=bool(m.HasLocalManifests),
404 hassuperprojecttag=bool(m.superproject),
405 )
406 if use_super and (m.IsMirror or m.IsArchive):
407 # Don't use superproject, because we have no working tree.
408 use_super = False
409 superproject_logging_data['superproject'] = False
410 superproject_logging_data['noworktree'] = True
411 if opt.use_superproject is not False:
412 print(f'{m.path_prefix}: not using superproject because there is no '
413 'working tree.')
414
415 if not use_super:
416 continue
417 m.superproject.SetQuiet(opt.quiet)
418 print_messages = git_superproject.PrintMessages(opt.use_superproject, m)
419 m.superproject.SetPrintMessages(print_messages)
420 update_result = m.superproject.UpdateProjectsRevisionId(
421 per_manifest[m.path_prefix], git_event_log=self.git_event_log)
422 manifest_path = update_result.manifest_path
423 superproject_logging_data['updatedrevisionid'] = bool(manifest_path)
424 if manifest_path:
425 m.SetManifestOverride(manifest_path)
426 need_unload = True
427 else:
428 if print_messages:
429 print(f'{m.path_prefix}: warning: Update of revisionId from '
430 'superproject has failed, repo sync will not use superproject '
431 'to fetch the source. ',
432 'Please resync with the --no-use-superproject option to avoid '
433 'this repo warning.',
434 file=sys.stderr)
435 if update_result.fatal and opt.use_superproject is not None:
436 sys.exit(1)
437 if need_unload:
438 m.outer_client.manifest.Unload()
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800439
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500440 def _FetchProjectList(self, opt, projects):
441 """Main function of the fetch worker.
442
443 The projects we're given share the same underlying git object store, so we
444 have to fetch them in serial.
Roy Lee18afd7f2010-05-09 04:32:08 +0800445
David James8d201162013-10-11 17:03:19 -0700446 Delegates most of the work to _FetchHelper.
447
448 Args:
449 opt: Program options returned from optparse. See _Options().
450 projects: Projects to fetch.
David James8d201162013-10-11 17:03:19 -0700451 """
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500452 return [self._FetchOne(opt, x) for x in projects]
David James8d201162013-10-11 17:03:19 -0700453
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500454 def _FetchOne(self, opt, project):
David James8d201162013-10-11 17:03:19 -0700455 """Fetch git objects for a single project.
456
David Pursehousec1b86a22012-11-14 11:36:51 +0900457 Args:
458 opt: Program options returned from optparse. See _Options().
459 project: Project object for the project to fetch.
David James8d201162013-10-11 17:03:19 -0700460
461 Returns:
462 Whether the fetch was successful.
David Pursehousec1b86a22012-11-14 11:36:51 +0900463 """
David Rileye0684ad2017-04-05 00:02:59 -0700464 start = time.time()
465 success = False
Mike Frysinger7b586f22021-02-23 18:38:39 -0500466 buf = io.StringIO()
David Pursehousec1b86a22012-11-14 11:36:51 +0900467 try:
LaMont Jones1eddca82022-09-01 15:15:04 +0000468 sync_result = project.Sync_NetworkHalf(
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500469 quiet=opt.quiet,
470 verbose=opt.verbose,
471 output_redir=buf,
LaMont Jonesa46047a2022-04-07 21:57:06 +0000472 current_branch_only=self._GetCurrentBranchOnly(opt, project.manifest),
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500473 force_sync=opt.force_sync,
474 clone_bundle=opt.clone_bundle,
LaMont Jonesa46047a2022-04-07 21:57:06 +0000475 tags=opt.tags, archive=project.manifest.IsArchive,
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500476 optimized_fetch=opt.optimized_fetch,
477 retry_fetches=opt.retry_fetches,
478 prune=opt.prune,
Mike Frysinger339f2df2021-05-06 00:44:42 -0400479 ssh_proxy=self.ssh_proxy,
LaMont Jonesa46047a2022-04-07 21:57:06 +0000480 clone_filter=project.manifest.CloneFilter,
481 partial_clone_exclude=project.manifest.PartialCloneExclude)
LaMont Jones1eddca82022-09-01 15:15:04 +0000482 success = sync_result.success
Doug Andersonfc06ced2011-03-16 15:49:18 -0700483
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500484 output = buf.getvalue()
Mike Frysinger58929732021-07-02 00:29:35 -0400485 if (opt.verbose or not success) and output:
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500486 print('\n' + output.rstrip())
Doug Andersonfc06ced2011-03-16 15:49:18 -0700487
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500488 if not success:
489 print('error: Cannot fetch %s from %s'
490 % (project.name, project.remote.url),
491 file=sys.stderr)
Raman Tennetiad8aa692021-04-15 09:20:51 -0700492 except GitError as e:
493 print('error.GitError: Cannot fetch %s' % str(e), file=sys.stderr)
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500494 except Exception as e:
495 print('error: Cannot fetch %s (%s: %s)'
496 % (project.name, type(e).__name__, str(e)), file=sys.stderr)
497 raise
Mike Frysinger7b586f22021-02-23 18:38:39 -0500498
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500499 finish = time.time()
LaMont Jones1eddca82022-09-01 15:15:04 +0000500 return _FetchOneResult(success, project, start, finish,
501 sync_result.remote_fetched)
David James8d201162013-10-11 17:03:19 -0700502
Mike Frysinger339f2df2021-05-06 00:44:42 -0400503 @classmethod
504 def _FetchInitChild(cls, ssh_proxy):
505 cls.ssh_proxy = ssh_proxy
506
507 def _Fetch(self, projects, opt, err_event, ssh_proxy):
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500508 ret = True
509
Mike Frysinger355f4392022-07-20 17:15:29 -0400510 jobs = opt.jobs_network
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700511 fetched = set()
LaMont Jones1eddca82022-09-01 15:15:04 +0000512 remote_fetched = set()
Mike Frysinger151701e2021-04-13 15:07:21 -0400513 pm = Progress('Fetching', len(projects), delay=False, quiet=opt.quiet)
Roy Lee18afd7f2010-05-09 04:32:08 +0800514
David James89ece422014-01-09 18:51:58 -0800515 objdir_project_map = dict()
516 for project in projects:
517 objdir_project_map.setdefault(project.objdir, []).append(project)
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500518 projects_list = list(objdir_project_map.values())
David James8d201162013-10-11 17:03:19 -0700519
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500520 def _ProcessResults(results_sets):
521 ret = True
522 for results in results_sets:
LaMont Jones1eddca82022-09-01 15:15:04 +0000523 for result in results:
524 success = result.success
525 project = result.project
526 start = result.start
527 finish = result.finish
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500528 self._fetch_times.Set(project, finish - start)
529 self.event_log.AddSync(project, event_log.TASK_SYNC_NETWORK,
530 start, finish, success)
LaMont Jones1eddca82022-09-01 15:15:04 +0000531 if result.remote_fetched:
532 remote_fetched.add(project)
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500533 # Check for any errors before running any more tasks.
534 # ...we'll let existing jobs finish, though.
535 if not success:
536 ret = False
537 else:
538 fetched.add(project.gitdir)
539 pm.update(msg=project.name)
540 if not ret and opt.fail_fast:
541 break
542 return ret
Doug Andersonfc06ced2011-03-16 15:49:18 -0700543
Mike Frysinger339f2df2021-05-06 00:44:42 -0400544 # We pass the ssh proxy settings via the class. This allows multiprocessing
545 # to pickle it up when spawning children. We can't pass it as an argument
546 # to _FetchProjectList below as multiprocessing is unable to pickle those.
547 Sync.ssh_proxy = None
548
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500549 # NB: Multiprocessing is heavy, so don't spin it up for one job.
Mike Frysinger49de8ef2021-04-09 00:21:02 -0400550 if len(projects_list) == 1 or jobs == 1:
Mike Frysinger339f2df2021-05-06 00:44:42 -0400551 self._FetchInitChild(ssh_proxy)
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500552 if not _ProcessResults(self._FetchProjectList(opt, x) for x in projects_list):
553 ret = False
554 else:
555 # Favor throughput over responsiveness when quiet. It seems that imap()
556 # will yield results in batches relative to chunksize, so even as the
557 # children finish a sync, we won't see the result until one child finishes
558 # ~chunksize jobs. When using a large --jobs with large chunksize, this
559 # can be jarring as there will be a large initial delay where repo looks
560 # like it isn't doing anything and sits at 0%, but then suddenly completes
561 # a lot of jobs all at once. Since this code is more network bound, we
562 # can accept a bit more CPU overhead with a smaller chunksize so that the
563 # user sees more immediate & continuous feedback.
564 if opt.quiet:
565 chunksize = WORKER_BATCH_SIZE
David James89ece422014-01-09 18:51:58 -0800566 else:
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500567 pm.update(inc=0, msg='warming up')
568 chunksize = 4
Raman Tenneti4a478ed2021-11-17 18:38:24 -0800569 with multiprocessing.Pool(jobs, initializer=self._FetchInitChild,
570 initargs=(ssh_proxy,)) as pool:
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500571 results = pool.imap_unordered(
572 functools.partial(self._FetchProjectList, opt),
573 projects_list,
574 chunksize=chunksize)
575 if not _ProcessResults(results):
576 ret = False
577 pool.close()
Roy Lee18afd7f2010-05-09 04:32:08 +0800578
Mike Frysinger339f2df2021-05-06 00:44:42 -0400579 # Cleanup the reference now that we're done with it, and we're going to
580 # release any resources it points to. If we don't, later multiprocessing
581 # usage (e.g. checkouts) will try to pickle and then crash.
582 del Sync.ssh_proxy
583
Shawn O. Pearce68194f42009-04-10 16:48:52 -0700584 pm.end()
Dave Borowitz67700e92012-10-23 15:00:54 -0700585 self._fetch_times.Save()
Dave Borowitz18857212012-10-23 17:02:59 -0700586
LaMont Jonesa46047a2022-04-07 21:57:06 +0000587 if not self.outer_client.manifest.IsArchive:
Mike Frysinger5a033082019-09-23 19:21:20 -0400588 self._GCProjects(projects, opt, err_event)
Julien Campergue335f5ef2013-10-16 11:02:35 +0200589
LaMont Jones1eddca82022-09-01 15:15:04 +0000590 return _FetchResult(ret, fetched)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700591
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000592 def _FetchMain(self, opt, args, all_projects, err_event,
593 ssh_proxy, manifest):
Mike Frysingerb4429432021-05-05 20:03:26 -0400594 """The main network fetch loop.
595
596 Args:
597 opt: Program options returned from optparse. See _Options().
598 args: Command line args used to filter out projects.
Peter Kjellerstedtd1776092021-05-19 19:37:23 +0200599 all_projects: List of all projects that should be fetched.
Mike Frysingerb4429432021-05-05 20:03:26 -0400600 err_event: Whether an error was hit while processing.
Mike Frysinger339f2df2021-05-06 00:44:42 -0400601 ssh_proxy: SSH manager for clients & masters.
LaMont Jonesa46047a2022-04-07 21:57:06 +0000602 manifest: The manifest to use.
Peter Kjellerstedtd1776092021-05-19 19:37:23 +0200603
604 Returns:
605 List of all projects that should be checked out.
Mike Frysingerb4429432021-05-05 20:03:26 -0400606 """
LaMont Jonesa46047a2022-04-07 21:57:06 +0000607 rp = manifest.repoProject
Mike Frysingerb4429432021-05-05 20:03:26 -0400608
609 to_fetch = []
610 now = time.time()
611 if _ONE_DAY_S <= (now - rp.LastFetch):
612 to_fetch.append(rp)
613 to_fetch.extend(all_projects)
614 to_fetch.sort(key=self._fetch_times.Get, reverse=True)
615
LaMont Jones1eddca82022-09-01 15:15:04 +0000616 result = self._Fetch(to_fetch, opt, err_event, ssh_proxy)
617 success = result.success
618 fetched = result.projects
Mike Frysingerb4429432021-05-05 20:03:26 -0400619 if not success:
620 err_event.set()
621
622 _PostRepoFetch(rp, opt.repo_verify)
623 if opt.network_only:
624 # bail out now; the rest touches the working tree
625 if err_event.is_set():
626 print('\nerror: Exited sync due to fetch errors.\n', file=sys.stderr)
627 sys.exit(1)
LaMont Jones1eddca82022-09-01 15:15:04 +0000628 return _FetchMainResult([])
Mike Frysingerb4429432021-05-05 20:03:26 -0400629
630 # Iteratively fetch missing and/or nested unregistered submodules
631 previously_missing_set = set()
632 while True:
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000633 self._ReloadManifest(None, manifest)
Mike Frysingerb4429432021-05-05 20:03:26 -0400634 all_projects = self.GetProjects(args,
635 missing_ok=True,
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000636 submodules_ok=opt.fetch_submodules,
637 manifest=manifest,
638 all_manifests=not opt.this_manifest_only)
Mike Frysingerb4429432021-05-05 20:03:26 -0400639 missing = []
640 for project in all_projects:
641 if project.gitdir not in fetched:
642 missing.append(project)
643 if not missing:
644 break
645 # Stop us from non-stopped fetching actually-missing repos: If set of
646 # missing repos has not been changed from last fetch, we break.
647 missing_set = set(p.name for p in missing)
648 if previously_missing_set == missing_set:
649 break
650 previously_missing_set = missing_set
LaMont Jones1eddca82022-09-01 15:15:04 +0000651 result = self._Fetch(missing, opt, err_event, ssh_proxy)
652 success = result.success
653 new_fetched = result.projects
Mike Frysingerb4429432021-05-05 20:03:26 -0400654 if not success:
655 err_event.set()
656 fetched.update(new_fetched)
657
LaMont Jones1eddca82022-09-01 15:15:04 +0000658 return _FetchMainResult(all_projects)
Peter Kjellerstedtd1776092021-05-19 19:37:23 +0200659
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500660 def _CheckoutOne(self, detach_head, force_sync, project):
Xin Li745be2e2019-06-03 11:24:30 -0700661 """Checkout work tree for one project
662
663 Args:
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500664 detach_head: Whether to leave a detached HEAD.
665 force_sync: Force checking out of the repo.
Xin Li745be2e2019-06-03 11:24:30 -0700666 project: Project object for the project to checkout.
Xin Li745be2e2019-06-03 11:24:30 -0700667
668 Returns:
669 Whether the fetch was successful.
670 """
Xin Li745be2e2019-06-03 11:24:30 -0700671 start = time.time()
LaMont Jonesa46047a2022-04-07 21:57:06 +0000672 syncbuf = SyncBuffer(project.manifest.manifestProject.config,
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500673 detach_head=detach_head)
Xin Li745be2e2019-06-03 11:24:30 -0700674 success = False
675 try:
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500676 project.Sync_LocalHalf(syncbuf, force_sync=force_sync)
Mike Frysingerebf04a42021-02-23 20:48:04 -0500677 success = syncbuf.Finish()
Raman Tennetiad8aa692021-04-15 09:20:51 -0700678 except GitError as e:
679 print('error.GitError: Cannot checkout %s: %s' %
680 (project.name, str(e)), file=sys.stderr)
Mike Frysingerebf04a42021-02-23 20:48:04 -0500681 except Exception as e:
682 print('error: Cannot checkout %s: %s: %s' %
683 (project.name, type(e).__name__, str(e)),
684 file=sys.stderr)
685 raise
Xin Li745be2e2019-06-03 11:24:30 -0700686
Mike Frysingerebf04a42021-02-23 20:48:04 -0500687 if not success:
688 print('error: Cannot checkout %s' % (project.name), file=sys.stderr)
689 finish = time.time()
LaMont Jones1eddca82022-09-01 15:15:04 +0000690 return _CheckoutOneResult(success, project, start, finish)
Xin Li745be2e2019-06-03 11:24:30 -0700691
Mike Frysingerebf04a42021-02-23 20:48:04 -0500692 def _Checkout(self, all_projects, opt, err_results):
Xin Li745be2e2019-06-03 11:24:30 -0700693 """Checkout projects listed in all_projects
694
695 Args:
696 all_projects: List of all projects that should be checked out.
697 opt: Program options returned from optparse. See _Options().
Mike Frysingerebf04a42021-02-23 20:48:04 -0500698 err_results: A list of strings, paths to git repos where checkout failed.
Xin Li745be2e2019-06-03 11:24:30 -0700699 """
Mike Frysingerebf04a42021-02-23 20:48:04 -0500700 # Only checkout projects with worktrees.
701 all_projects = [x for x in all_projects if x.worktree]
Xin Li745be2e2019-06-03 11:24:30 -0700702
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500703 def _ProcessResults(pool, pm, results):
704 ret = True
LaMont Jones1eddca82022-09-01 15:15:04 +0000705 for result in results:
706 success = result.success
707 project = result.project
708 start = result.start
709 finish = result.finish
Mike Frysingerebf04a42021-02-23 20:48:04 -0500710 self.event_log.AddSync(project, event_log.TASK_SYNC_LOCAL,
711 start, finish, success)
712 # Check for any errors before running any more tasks.
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500713 # ...we'll let existing jobs finish, though.
Mike Frysingerebf04a42021-02-23 20:48:04 -0500714 if not success:
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500715 ret = False
Mike Frysingerebf04a42021-02-23 20:48:04 -0500716 err_results.append(project.relpath)
717 if opt.fail_fast:
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500718 if pool:
719 pool.close()
720 return ret
Mike Frysingerebf04a42021-02-23 20:48:04 -0500721 pm.update(msg=project.name)
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500722 return ret
Xin Li745be2e2019-06-03 11:24:30 -0700723
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500724 return self.ExecuteInParallel(
Mike Frysinger355f4392022-07-20 17:15:29 -0400725 opt.jobs_checkout,
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500726 functools.partial(self._CheckoutOne, opt.detach_head, opt.force_sync),
727 all_projects,
728 callback=_ProcessResults,
729 output=Progress('Checking out', len(all_projects), quiet=opt.quiet)) and not err_results
Mike Frysingerebf04a42021-02-23 20:48:04 -0500730
LaMont Jonesaefa4d32022-09-15 18:24:56 +0000731 def _backup_cruft(self, bare_git):
732 """Save a copy of any cruft from `git gc`."""
733 # Find any cruft packs in the current gitdir, and save them.
734 # b/221065125 (repo sync complains that objects are missing). This does
735 # not prevent that state, but makes it so that the missing objects are
736 # available.
737 objdir = bare_git._project.objdir
738 pack_dir = os.path.join(objdir, 'pack')
739 bak_dir = os.path.join(objdir, '.repo', 'pack.bak')
740 if not _BACKUP_OBJECTS or not platform_utils.isdir(pack_dir):
741 return
742 saved = []
743 files = set(platform_utils.listdir(pack_dir))
744 to_backup = []
745 for f in files:
746 base, ext = os.path.splitext(f)
747 if base + '.mtimes' in files:
748 to_backup.append(f)
749 if to_backup:
750 os.makedirs(bak_dir, exist_ok=True)
751 for fname in to_backup:
752 bak_fname = os.path.join(bak_dir, fname)
753 if not os.path.exists(bak_fname):
754 saved.append(fname)
755 # Use a tmp file so that we are sure of a complete copy.
756 shutil.copy(os.path.join(pack_dir, fname), bak_fname + '.tmp')
757 shutil.move(bak_fname + '.tmp', bak_fname)
758 if saved:
759 Trace('%s saved %s', bare_git._project.name, ' '.join(saved))
760
Mike Frysinger5a033082019-09-23 19:21:20 -0400761 def _GCProjects(self, projects, opt, err_event):
Mike Frysinger151701e2021-04-13 15:07:21 -0400762 pm = Progress('Garbage collecting', len(projects), delay=False, quiet=opt.quiet)
Mike Frysinger65af2602021-04-08 22:47:44 -0400763 pm.update(inc=0, msg='prescan')
764
Allen Webb4ee4a452021-10-07 10:42:38 -0500765 tidy_dirs = {}
David James8d201162013-10-11 17:03:19 -0700766 for project in projects:
LaMont Jones48ea25c2022-05-20 10:35:04 +0000767 # Make sure pruning never kicks in with shared projects that do not use
768 # alternates to avoid corruption.
Mike Frysinger979d5bd2020-02-09 02:28:34 -0500769 if (not project.use_git_worktrees and
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000770 len(project.manifest.GetProjectsWithName(project.name, all_manifests=True)) > 1):
LaMont Jones48ea25c2022-05-20 10:35:04 +0000771 if project.UseAlternates:
772 # Undo logic set by previous versions of repo.
773 project.config.SetString('extensions.preciousObjects', None)
774 project.config.SetString('gc.pruneExpire', None)
Mike Frysinger5f2b0452020-02-11 03:23:24 -0500775 else:
LaMont Jones48ea25c2022-05-20 10:35:04 +0000776 if not opt.quiet:
777 print('\r%s: Shared project %s found, disabling pruning.' %
778 (project.relpath, project.name))
779 if git_require((2, 7, 0)):
780 project.EnableRepositoryExtension('preciousObjects')
781 else:
782 # This isn't perfect, but it's the best we can do with old git.
783 print('\r%s: WARNING: shared projects are unreliable when using old '
784 'versions of git; please upgrade to git-2.7.0+.'
785 % (project.relpath,),
786 file=sys.stderr)
787 project.config.SetString('gc.pruneExpire', 'never')
Allen Webb669efd02021-10-01 15:25:31 -0500788 project.config.SetString('gc.autoDetach', 'false')
Allen Webb4ee4a452021-10-07 10:42:38 -0500789 # Only call git gc once per objdir, but call pack-refs for the remainder.
790 if project.objdir not in tidy_dirs:
791 tidy_dirs[project.objdir] = (
792 True, # Run a full gc.
793 project.bare_git,
794 )
795 elif project.gitdir not in tidy_dirs:
796 tidy_dirs[project.gitdir] = (
797 False, # Do not run a full gc; just run pack-refs.
798 project.bare_git,
799 )
Mike Frysinger65af2602021-04-08 22:47:44 -0400800
Mike Frysinger355f4392022-07-20 17:15:29 -0400801 jobs = opt.jobs
Dave Borowitz18857212012-10-23 17:02:59 -0700802
LaMont Jones891e8f72022-09-08 20:17:58 +0000803 gc_args = ('--auto', '--cruft')
804 pack_refs_args = ()
Dave Borowitz18857212012-10-23 17:02:59 -0700805 if jobs < 2:
Allen Webb4ee4a452021-10-07 10:42:38 -0500806 for (run_gc, bare_git) in tidy_dirs.values():
Mike Frysinger65af2602021-04-08 22:47:44 -0400807 pm.update(msg=bare_git._project.name)
LaMont Jones891e8f72022-09-08 20:17:58 +0000808
Allen Webb4ee4a452021-10-07 10:42:38 -0500809 if run_gc:
LaMont Jones891e8f72022-09-08 20:17:58 +0000810 bare_git.gc(*gc_args)
Allen Webb4ee4a452021-10-07 10:42:38 -0500811 else:
LaMont Jones891e8f72022-09-08 20:17:58 +0000812 bare_git.pack_refs(*pack_refs_args)
LaMont Jonesaefa4d32022-09-15 18:24:56 +0000813 self._backup_cruft(bare_git)
Mike Frysinger65af2602021-04-08 22:47:44 -0400814 pm.end()
Dave Borowitz18857212012-10-23 17:02:59 -0700815 return
816
Mike Frysinger355f4392022-07-20 17:15:29 -0400817 cpu_count = os.cpu_count()
Mike Frysinger0c0e9342019-06-13 12:42:39 -0400818 config = {'pack.threads': cpu_count // jobs if cpu_count > jobs else 1}
Dave Borowitz18857212012-10-23 17:02:59 -0700819
820 threads = set()
821 sem = _threading.Semaphore(jobs)
Dave Borowitz18857212012-10-23 17:02:59 -0700822
Allen Webb4ee4a452021-10-07 10:42:38 -0500823 def tidy_up(run_gc, bare_git):
Mike Frysinger65af2602021-04-08 22:47:44 -0400824 pm.start(bare_git._project.name)
Dave Borowitz18857212012-10-23 17:02:59 -0700825 try:
826 try:
Allen Webb4ee4a452021-10-07 10:42:38 -0500827 if run_gc:
LaMont Jones891e8f72022-09-08 20:17:58 +0000828 bare_git.gc(*gc_args, config=config)
Allen Webb4ee4a452021-10-07 10:42:38 -0500829 else:
LaMont Jones891e8f72022-09-08 20:17:58 +0000830 bare_git.pack_refs(*pack_refs_args, config=config)
Dave Borowitz18857212012-10-23 17:02:59 -0700831 except GitError:
832 err_event.set()
David Pursehouse145e35b2020-02-12 15:40:47 +0900833 except Exception:
Dave Borowitz18857212012-10-23 17:02:59 -0700834 err_event.set()
835 raise
836 finally:
LaMont Jonesaefa4d32022-09-15 18:24:56 +0000837 self._backup_cruft(bare_git)
Mike Frysinger65af2602021-04-08 22:47:44 -0400838 pm.finish(bare_git._project.name)
Dave Borowitz18857212012-10-23 17:02:59 -0700839 sem.release()
840
Allen Webb4ee4a452021-10-07 10:42:38 -0500841 for (run_gc, bare_git) in tidy_dirs.values():
Mike Frysingerbe24a542021-02-23 03:24:12 -0500842 if err_event.is_set() and opt.fail_fast:
Dave Borowitz18857212012-10-23 17:02:59 -0700843 break
844 sem.acquire()
Allen Webb4ee4a452021-10-07 10:42:38 -0500845 t = _threading.Thread(target=tidy_up, args=(run_gc, bare_git,))
Dave Borowitz18857212012-10-23 17:02:59 -0700846 t.daemon = True
847 threads.add(t)
848 t.start()
849
850 for t in threads:
851 t.join()
Mike Frysinger65af2602021-04-08 22:47:44 -0400852 pm.end()
Dave Borowitz18857212012-10-23 17:02:59 -0700853
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000854 def _ReloadManifest(self, manifest_name, manifest):
Raman Tennetifeb28912021-05-02 19:47:29 -0700855 """Reload the manfiest from the file specified by the |manifest_name|.
856
857 It unloads the manifest if |manifest_name| is None.
858
859 Args:
860 manifest_name: Manifest file to be reloaded.
LaMont Jonesa46047a2022-04-07 21:57:06 +0000861 manifest: The manifest to use.
Raman Tennetifeb28912021-05-02 19:47:29 -0700862 """
Tim Kilbourn07669002013-03-08 15:02:49 -0800863 if manifest_name:
LaMont Jonesa2ff20d2022-04-07 16:49:06 +0000864 # Override calls Unload already
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000865 manifest.Override(manifest_name)
Tim Kilbourn07669002013-03-08 15:02:49 -0800866 else:
LaMont Jonesa46047a2022-04-07 21:57:06 +0000867 manifest.Unload()
Tim Kilbourn07669002013-03-08 15:02:49 -0800868
LaMont Jonesa46047a2022-04-07 21:57:06 +0000869 def UpdateProjectList(self, opt, manifest):
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000870 """Update the cached projects list for |manifest|
871
872 In a multi-manifest checkout, each manifest has its own project.list.
873
874 Args:
875 opt: Program options returned from optparse. See _Options().
876 manifest: The manifest to use.
877
878 Returns:
879 0: success
880 1: failure
881 """
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700882 new_project_paths = []
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000883 for project in self.GetProjects(None, missing_ok=True, manifest=manifest,
884 all_manifests=False):
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700885 if project.relpath:
886 new_project_paths.append(project.relpath)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700887 file_name = 'project.list'
LaMont Jonesa46047a2022-04-07 21:57:06 +0000888 file_path = os.path.join(manifest.subdir, file_name)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700889 old_project_paths = []
890
891 if os.path.exists(file_path):
Mike Frysinger3164d402019-11-11 05:40:22 -0500892 with open(file_path, 'r') as fd:
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700893 old_project_paths = fd.read().split('\n')
Kuang-che Wu0d9b16d2019-04-06 00:49:47 +0800894 # In reversed order, so subfolders are deleted before parent folder.
895 for path in sorted(old_project_paths, reverse=True):
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700896 if not path:
897 continue
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700898 if path not in new_project_paths:
David Pursehouse8a68ff92012-09-24 12:15:13 +0900899 # If the path has already been deleted, we don't need to do it
LaMont Jonesa46047a2022-04-07 21:57:06 +0000900 gitdir = os.path.join(manifest.topdir, path, '.git')
Dan Willemsen43507912016-09-01 16:26:02 -0700901 if os.path.exists(gitdir):
David Pursehousec1b86a22012-11-14 11:36:51 +0900902 project = Project(
LaMont Jonesa46047a2022-04-07 21:57:06 +0000903 manifest=manifest,
David Pursehouseabdf7502020-02-12 14:58:39 +0900904 name=path,
905 remote=RemoteSpec('origin'),
906 gitdir=gitdir,
907 objdir=gitdir,
Mike Frysingerc0d18662020-02-19 19:19:18 -0500908 use_git_worktrees=os.path.isfile(gitdir),
LaMont Jonesa46047a2022-04-07 21:57:06 +0000909 worktree=os.path.join(manifest.topdir, path),
David Pursehouseabdf7502020-02-12 14:58:39 +0900910 relpath=path,
911 revisionExpr='HEAD',
912 revisionId=None,
913 groups=None)
Mike Frysingerc0d18662020-02-19 19:19:18 -0500914 if not project.DeleteWorktree(
David Pursehouseaa611a22020-02-20 10:47:26 +0900915 quiet=opt.quiet,
916 force=opt.force_remove_dirty):
Mike Frysingera850ca22019-08-07 17:19:24 -0400917 return 1
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700918
Shawn O. Pearce9fb29ce2009-06-04 20:41:02 -0700919 new_project_paths.sort()
Mike Frysinger3164d402019-11-11 05:40:22 -0500920 with open(file_path, 'w') as fd:
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700921 fd.write('\n'.join(new_project_paths))
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700922 fd.write('\n')
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700923 return 0
924
LaMont Jonesa46047a2022-04-07 21:57:06 +0000925 def UpdateCopyLinkfileList(self, manifest):
jiajia tanga590e642021-04-25 20:02:02 +0800926 """Save all dests of copyfile and linkfile, and update them if needed.
927
928 Returns:
929 Whether update was successful.
930 """
931 new_paths = {}
932 new_linkfile_paths = []
933 new_copyfile_paths = []
LaMont Jonesbdcba7d2022-04-11 22:50:11 +0000934 for project in self.GetProjects(None, missing_ok=True,
935 manifest=manifest, all_manifests=False):
jiajia tanga590e642021-04-25 20:02:02 +0800936 new_linkfile_paths.extend(x.dest for x in project.linkfiles)
937 new_copyfile_paths.extend(x.dest for x in project.copyfiles)
938
939 new_paths = {
940 'linkfile': new_linkfile_paths,
941 'copyfile': new_copyfile_paths,
942 }
943
944 copylinkfile_name = 'copy-link-files.json'
LaMont Jonesa46047a2022-04-07 21:57:06 +0000945 copylinkfile_path = os.path.join(manifest.subdir, copylinkfile_name)
jiajia tanga590e642021-04-25 20:02:02 +0800946 old_copylinkfile_paths = {}
947
948 if os.path.exists(copylinkfile_path):
949 with open(copylinkfile_path, 'rb') as fp:
950 try:
951 old_copylinkfile_paths = json.load(fp)
Raman Tenneti4a478ed2021-11-17 18:38:24 -0800952 except Exception:
jiajia tanga590e642021-04-25 20:02:02 +0800953 print('error: %s is not a json formatted file.' %
954 copylinkfile_path, file=sys.stderr)
955 platform_utils.remove(copylinkfile_path)
956 return False
957
958 need_remove_files = []
959 need_remove_files.extend(
960 set(old_copylinkfile_paths.get('linkfile', [])) -
961 set(new_linkfile_paths))
962 need_remove_files.extend(
963 set(old_copylinkfile_paths.get('copyfile', [])) -
964 set(new_copyfile_paths))
965
966 for need_remove_file in need_remove_files:
Mike Frysinger9d96f582021-09-28 11:27:24 -0400967 # Try to remove the updated copyfile or linkfile.
968 # So, if the file is not exist, nothing need to do.
969 platform_utils.remove(need_remove_file, missing_ok=True)
jiajia tanga590e642021-04-25 20:02:02 +0800970
971 # Create copy-link-files.json, save dest path of "copyfile" and "linkfile".
972 with open(copylinkfile_path, 'w', encoding='utf-8') as fp:
973 json.dump(new_paths, fp)
974 return True
975
LaMont Jonesa46047a2022-04-07 21:57:06 +0000976 def _SmartSyncSetup(self, opt, smart_sync_manifest_path, manifest):
977 if not manifest.manifest_server:
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400978 print('error: cannot smart sync: no manifest server defined in '
979 'manifest', file=sys.stderr)
980 sys.exit(1)
981
LaMont Jonesa46047a2022-04-07 21:57:06 +0000982 manifest_server = manifest.manifest_server
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400983 if not opt.quiet:
984 print('Using manifest server %s' % manifest_server)
985
David Pursehouseeeff3532020-02-12 11:24:10 +0900986 if '@' not in manifest_server:
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400987 username = None
988 password = None
989 if opt.manifest_server_username and opt.manifest_server_password:
990 username = opt.manifest_server_username
991 password = opt.manifest_server_password
992 else:
993 try:
994 info = netrc.netrc()
995 except IOError:
996 # .netrc file does not exist or could not be opened
997 pass
998 else:
999 try:
1000 parse_result = urllib.parse.urlparse(manifest_server)
1001 if parse_result.hostname:
1002 auth = info.authenticators(parse_result.hostname)
1003 if auth:
1004 username, _account, password = auth
1005 else:
1006 print('No credentials found for %s in .netrc'
1007 % parse_result.hostname, file=sys.stderr)
1008 except netrc.NetrcParseError as e:
1009 print('Error parsing .netrc file: %s' % e, file=sys.stderr)
1010
1011 if (username and password):
1012 manifest_server = manifest_server.replace('://', '://%s:%s@' %
1013 (username, password),
1014 1)
1015
1016 transport = PersistentTransport(manifest_server)
1017 if manifest_server.startswith('persistent-'):
1018 manifest_server = manifest_server[len('persistent-'):]
1019
1020 try:
1021 server = xmlrpc.client.Server(manifest_server, transport=transport)
1022 if opt.smart_sync:
LaMont Jonesa46047a2022-04-07 21:57:06 +00001023 branch = self._GetBranch(manifest.manifestProject)
Mike Frysinger01d6c3c2019-08-27 01:56:43 -04001024
Mike Frysinger56ce3462019-12-04 19:30:48 -05001025 if 'SYNC_TARGET' in os.environ:
Mike Frysingere20da3e2020-03-07 01:53:53 -05001026 target = os.environ['SYNC_TARGET']
Mike Frysinger01d6c3c2019-08-27 01:56:43 -04001027 [success, manifest_str] = server.GetApprovedManifest(branch, target)
Mike Frysinger56ce3462019-12-04 19:30:48 -05001028 elif ('TARGET_PRODUCT' in os.environ and
1029 'TARGET_BUILD_VARIANT' in os.environ):
Mike Frysingere20da3e2020-03-07 01:53:53 -05001030 target = '%s-%s' % (os.environ['TARGET_PRODUCT'],
1031 os.environ['TARGET_BUILD_VARIANT'])
Mike Frysinger01d6c3c2019-08-27 01:56:43 -04001032 [success, manifest_str] = server.GetApprovedManifest(branch, target)
1033 else:
1034 [success, manifest_str] = server.GetApprovedManifest(branch)
1035 else:
1036 assert(opt.smart_tag)
1037 [success, manifest_str] = server.GetManifest(opt.smart_tag)
1038
1039 if success:
1040 manifest_name = os.path.basename(smart_sync_manifest_path)
1041 try:
Mike Frysinger3164d402019-11-11 05:40:22 -05001042 with open(smart_sync_manifest_path, 'w') as f:
Mike Frysinger01d6c3c2019-08-27 01:56:43 -04001043 f.write(manifest_str)
Mike Frysinger01d6c3c2019-08-27 01:56:43 -04001044 except IOError as e:
1045 print('error: cannot write manifest to %s:\n%s'
1046 % (smart_sync_manifest_path, e),
1047 file=sys.stderr)
1048 sys.exit(1)
LaMont Jonesa46047a2022-04-07 21:57:06 +00001049 self._ReloadManifest(manifest_name, manifest)
Mike Frysinger01d6c3c2019-08-27 01:56:43 -04001050 else:
1051 print('error: manifest server RPC call failed: %s' %
1052 manifest_str, file=sys.stderr)
1053 sys.exit(1)
1054 except (socket.error, IOError, xmlrpc.client.Fault) as e:
1055 print('error: cannot connect to manifest server %s:\n%s'
LaMont Jonesa46047a2022-04-07 21:57:06 +00001056 % (manifest.manifest_server, e), file=sys.stderr)
Mike Frysinger01d6c3c2019-08-27 01:56:43 -04001057 sys.exit(1)
1058 except xmlrpc.client.ProtocolError as e:
1059 print('error: cannot connect to manifest server %s:\n%d %s'
LaMont Jonesa46047a2022-04-07 21:57:06 +00001060 % (manifest.manifest_server, e.errcode, e.errmsg),
Mike Frysinger01d6c3c2019-08-27 01:56:43 -04001061 file=sys.stderr)
1062 sys.exit(1)
1063
1064 return manifest_name
1065
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001066 def _UpdateAllManifestProjects(self, opt, mp, manifest_name):
1067 """Fetch & update the local manifest project.
1068
1069 After syncing the manifest project, if the manifest has any sub manifests,
1070 those are recursively processed.
1071
1072 Args:
1073 opt: Program options returned from optparse. See _Options().
1074 mp: the manifestProject to query.
1075 manifest_name: Manifest file to be reloaded.
1076 """
1077 if not mp.standalone_manifest_url:
1078 self._UpdateManifestProject(opt, mp, manifest_name)
1079
1080 if mp.manifest.submanifests:
1081 for submanifest in mp.manifest.submanifests.values():
1082 child = submanifest.repo_client.manifest
1083 child.manifestProject.SyncWithPossibleInit(
1084 submanifest,
1085 current_branch_only=self._GetCurrentBranchOnly(opt, child),
1086 verbose=opt.verbose,
1087 tags=opt.tags,
1088 git_event_log=self.git_event_log,
1089 )
1090 self._UpdateAllManifestProjects(opt, child.manifestProject, None)
1091
Mike Frysingerfb527e32019-08-27 02:34:32 -04001092 def _UpdateManifestProject(self, opt, mp, manifest_name):
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001093 """Fetch & update the local manifest project.
1094
1095 Args:
1096 opt: Program options returned from optparse. See _Options().
1097 mp: the manifestProject to query.
1098 manifest_name: Manifest file to be reloaded.
1099 """
Mike Frysingerfb527e32019-08-27 02:34:32 -04001100 if not opt.local_only:
1101 start = time.time()
Mike Frysinger521d01b2020-02-17 01:51:49 -05001102 success = mp.Sync_NetworkHalf(quiet=opt.quiet, verbose=opt.verbose,
LaMont Jonesa46047a2022-04-07 21:57:06 +00001103 current_branch_only=self._GetCurrentBranchOnly(opt, mp.manifest),
Erwan Yvindc5c4d12019-06-18 13:49:12 +02001104 force_sync=opt.force_sync,
Mike Frysingerc58ec4d2020-02-17 14:36:08 -05001105 tags=opt.tags,
Mike Frysingerfb527e32019-08-27 02:34:32 -04001106 optimized_fetch=opt.optimized_fetch,
George Engelbrecht9bc283e2020-04-02 12:36:09 -06001107 retry_fetches=opt.retry_fetches,
LaMont Jonesa46047a2022-04-07 21:57:06 +00001108 submodules=mp.manifest.HasSubmodules,
1109 clone_filter=mp.manifest.CloneFilter,
1110 partial_clone_exclude=mp.manifest.PartialCloneExclude)
Mike Frysingerfb527e32019-08-27 02:34:32 -04001111 finish = time.time()
1112 self.event_log.AddSync(mp, event_log.TASK_SYNC_NETWORK,
1113 start, finish, success)
1114
1115 if mp.HasChanges:
1116 syncbuf = SyncBuffer(mp.config)
1117 start = time.time()
LaMont Jonesa46047a2022-04-07 21:57:06 +00001118 mp.Sync_LocalHalf(syncbuf, submodules=mp.manifest.HasSubmodules)
Mike Frysingerfb527e32019-08-27 02:34:32 -04001119 clean = syncbuf.Finish()
1120 self.event_log.AddSync(mp, event_log.TASK_SYNC_LOCAL,
1121 start, time.time(), clean)
1122 if not clean:
1123 sys.exit(1)
LaMont Jonesa46047a2022-04-07 21:57:06 +00001124 self._ReloadManifest(manifest_name, mp.manifest)
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001125
Mike Frysingerae6cb082019-08-27 01:10:59 -04001126 def ValidateOptions(self, opt, args):
1127 if opt.force_broken:
1128 print('warning: -f/--force-broken is now the default behavior, and the '
1129 'options are deprecated', file=sys.stderr)
1130 if opt.network_only and opt.detach_head:
1131 self.OptionParser.error('cannot combine -n and -d')
1132 if opt.network_only and opt.local_only:
1133 self.OptionParser.error('cannot combine -n and -l')
1134 if opt.manifest_name and opt.smart_sync:
1135 self.OptionParser.error('cannot combine -m and -s')
1136 if opt.manifest_name and opt.smart_tag:
1137 self.OptionParser.error('cannot combine -m and -t')
1138 if opt.manifest_server_username or opt.manifest_server_password:
1139 if not (opt.smart_sync or opt.smart_tag):
1140 self.OptionParser.error('-u and -p may only be combined with -s or -t')
1141 if None in [opt.manifest_server_username, opt.manifest_server_password]:
1142 self.OptionParser.error('both -u and -p must be given')
1143
Mike Frysinger0531a622021-11-05 15:22:01 -04001144 if opt.prune is None:
1145 opt.prune = True
1146
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001147 def Execute(self, opt, args):
LaMont Jonesa46047a2022-04-07 21:57:06 +00001148 manifest = self.outer_manifest
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001149 if not opt.outer_manifest:
LaMont Jonesa46047a2022-04-07 21:57:06 +00001150 manifest = self.manifest
1151
Chris Wolfee9dc3b32012-01-26 11:36:18 -05001152 if opt.manifest_name:
LaMont Jonesa46047a2022-04-07 21:57:06 +00001153 manifest.Override(opt.manifest_name)
Shawn O. Pearce3e768c92009-04-10 16:59:36 -07001154
Chirayu Desaia892b102013-06-11 14:18:46 +05301155 manifest_name = opt.manifest_name
David Pursehouse59b41742015-05-07 14:36:09 +09001156 smart_sync_manifest_path = os.path.join(
LaMont Jonesa46047a2022-04-07 21:57:06 +00001157 manifest.manifestProject.worktree, 'smart_sync_override.xml')
Chirayu Desaia892b102013-06-11 14:18:46 +05301158
Xin Lid79a4bc2020-05-20 16:03:45 -07001159 if opt.clone_bundle is None:
LaMont Jonesa46047a2022-04-07 21:57:06 +00001160 opt.clone_bundle = manifest.CloneBundle
Xin Lid79a4bc2020-05-20 16:03:45 -07001161
Victor Boivie08c880d2011-04-19 10:32:52 +02001162 if opt.smart_sync or opt.smart_tag:
LaMont Jonesa46047a2022-04-07 21:57:06 +00001163 manifest_name = self._SmartSyncSetup(opt, smart_sync_manifest_path, manifest)
Mike Frysinger01d6c3c2019-08-27 01:56:43 -04001164 else:
David Pursehouse59b41742015-05-07 14:36:09 +09001165 if os.path.isfile(smart_sync_manifest_path):
1166 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08001167 platform_utils.remove(smart_sync_manifest_path)
David Pursehouse59b41742015-05-07 14:36:09 +09001168 except OSError as e:
1169 print('error: failed to remove existing smart sync override manifest: %s' %
1170 e, file=sys.stderr)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -07001171
Mike Frysingerc99322a2021-05-04 15:32:43 -04001172 err_event = multiprocessing.Event()
Mike Frysinger5a033082019-09-23 19:21:20 -04001173
LaMont Jonesa46047a2022-04-07 21:57:06 +00001174 rp = manifest.repoProject
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001175 rp.PreSync()
Mike Frysinger23c900f2020-02-29 02:52:44 -05001176 cb = rp.CurrentBranch
1177 if cb:
1178 base = rp.GetBranch(cb).merge
1179 if not base or not base.startswith('refs/heads/'):
1180 print('warning: repo is not tracking a remote branch, so it will not '
Mike Frysinger58ac1672020-03-14 14:35:26 -04001181 'receive updates; run `repo init --repo-rev=stable` to fix.',
Mike Frysinger23c900f2020-02-29 02:52:44 -05001182 file=sys.stderr)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001183
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001184 for m in self.ManifestList(opt):
LaMont Jones4112c072022-08-24 17:32:25 +00001185 if not m.manifestProject.standalone_manifest_url:
1186 m.manifestProject.PreSync()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001187
LaMont Jones4112c072022-08-24 17:32:25 +00001188 if opt.repo_upgraded:
1189 _PostRepoUpgrade(manifest, quiet=opt.quiet)
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08001190
LaMont Jones4112c072022-08-24 17:32:25 +00001191 mp = manifest.manifestProject
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001192 if opt.mp_update:
1193 self._UpdateAllManifestProjects(opt, mp, manifest_name)
1194 else:
Fredrik de Grootcc960972019-11-22 09:04:31 +01001195 print('Skipping update of local manifest project.')
Simran Basib9a1b732015-08-20 12:19:28 -07001196
Mike Frysinger355f4392022-07-20 17:15:29 -04001197 # Now that the manifests are up-to-date, setup the jobs value.
1198 if opt.jobs is None:
1199 # User has not made a choice, so use the manifest settings.
1200 opt.jobs = mp.default.sync_j
1201 if opt.jobs is not None:
1202 # Neither user nor manifest have made a choice.
1203 if opt.jobs_network is None:
1204 opt.jobs_network = opt.jobs
1205 if opt.jobs_checkout is None:
1206 opt.jobs_checkout = opt.jobs
1207 # Setup defaults if jobs==0.
1208 if not opt.jobs:
1209 if not opt.jobs_network:
1210 opt.jobs_network = 1
1211 if not opt.jobs_checkout:
1212 opt.jobs_checkout = DEFAULT_LOCAL_JOBS
1213 opt.jobs = os.cpu_count()
1214
1215 # Try to stay under user rlimit settings.
1216 #
1217 # Since each worker requires at 3 file descriptors to run `git fetch`, use
1218 # that to scale down the number of jobs. Unfortunately there isn't an easy
1219 # way to determine this reliably as systems change, but it was last measured
1220 # by hand in 2011.
1221 soft_limit, _ = _rlimit_nofile()
1222 jobs_soft_limit = max(1, (soft_limit - 5) // 3)
1223 opt.jobs = min(opt.jobs, jobs_soft_limit)
1224 opt.jobs_network = min(opt.jobs_network, jobs_soft_limit)
1225 opt.jobs_checkout = min(opt.jobs_checkout, jobs_soft_limit)
1226
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001227 superproject_logging_data = {}
1228 self._UpdateProjectsRevisionId(opt, args, superproject_logging_data,
1229 manifest)
Raman Tenneti1fd7bc22021-02-04 14:39:38 -08001230
Simran Basib9a1b732015-08-20 12:19:28 -07001231 if self.gitc_manifest:
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001232 gitc_manifest_projects = self.GetProjects(args, missing_ok=True)
Simran Basib9a1b732015-08-20 12:19:28 -07001233 gitc_projects = []
1234 opened_projects = []
1235 for project in gitc_manifest_projects:
Dan Willemsen5ea32d12015-09-08 13:27:20 -07001236 if project.relpath in self.gitc_manifest.paths and \
1237 self.gitc_manifest.paths[project.relpath].old_revision:
1238 opened_projects.append(project.relpath)
Simran Basib9a1b732015-08-20 12:19:28 -07001239 else:
Dan Willemsen5ea32d12015-09-08 13:27:20 -07001240 gitc_projects.append(project.relpath)
Simran Basib9a1b732015-08-20 12:19:28 -07001241
Dan Willemsen5ea32d12015-09-08 13:27:20 -07001242 if not args:
1243 gitc_projects = None
1244
1245 if gitc_projects != [] and not opt.local_only:
Simran Basib9a1b732015-08-20 12:19:28 -07001246 print('Updating GITC client: %s' % self.gitc_manifest.gitc_client_name)
Dan Willemsen5ea32d12015-09-08 13:27:20 -07001247 manifest = GitcManifest(self.repodir, self.gitc_manifest.gitc_client_name)
1248 if manifest_name:
1249 manifest.Override(manifest_name)
1250 else:
LaMont Jonesa46047a2022-04-07 21:57:06 +00001251 manifest.Override(manifest.manifestFile)
Dan Willemsen5ea32d12015-09-08 13:27:20 -07001252 gitc_utils.generate_gitc_manifest(self.gitc_manifest,
1253 manifest,
Simran Basib9a1b732015-08-20 12:19:28 -07001254 gitc_projects)
1255 print('GITC client successfully synced.')
1256
1257 # The opened projects need to be synced as normal, therefore we
1258 # generate a new args list to represent the opened projects.
Dan Willemsen5ea32d12015-09-08 13:27:20 -07001259 # TODO: make this more reliable -- if there's a project name/path overlap,
1260 # this may choose the wrong project.
LaMont Jonesa46047a2022-04-07 21:57:06 +00001261 args = [os.path.relpath(manifest.paths[path].worktree, os.getcwd())
David Pursehouse3bcd3052017-07-10 22:42:22 +09001262 for path in opened_projects]
Simran Basib9a1b732015-08-20 12:19:28 -07001263 if not args:
1264 return
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001265
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001266 all_projects = self.GetProjects(args,
1267 missing_ok=True,
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001268 submodules_ok=opt.fetch_submodules,
1269 manifest=manifest,
1270 all_manifests=not opt.this_manifest_only)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001271
Mike Frysinger5a033082019-09-23 19:21:20 -04001272 err_network_sync = False
1273 err_update_projects = False
Mike Frysinger5a033082019-09-23 19:21:20 -04001274
LaMont Jonesa46047a2022-04-07 21:57:06 +00001275 self._fetch_times = _FetchTimes(manifest)
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -07001276 if not opt.local_only:
Mike Frysinger339f2df2021-05-06 00:44:42 -04001277 with multiprocessing.Manager() as manager:
1278 with ssh.ProxyManager(manager) as ssh_proxy:
1279 # Initialize the socket dir once in the parent.
1280 ssh_proxy.sock()
LaMont Jones1eddca82022-09-01 15:15:04 +00001281 result = self._FetchMain(opt, args, all_projects, err_event,
1282 ssh_proxy, manifest)
1283 all_projects = result.all_projects
Mike Frysinger339f2df2021-05-06 00:44:42 -04001284
1285 if opt.network_only:
1286 return
Mike Frysinger5a033082019-09-23 19:21:20 -04001287
1288 # If we saw an error, exit with code 1 so that other scripts can check.
Mike Frysingerbe24a542021-02-23 03:24:12 -05001289 if err_event.is_set():
Mike Frysinger5a033082019-09-23 19:21:20 -04001290 err_network_sync = True
1291 if opt.fail_fast:
1292 print('\nerror: Exited sync due to fetch errors.\n'
1293 'Local checkouts *not* updated. Resolve network issues & '
1294 'retry.\n'
1295 '`repo sync -l` will update some local checkouts.',
1296 file=sys.stderr)
1297 sys.exit(1)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001298
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001299 for m in self.ManifestList(opt):
1300 if m.IsMirror or m.IsArchive:
1301 # bail out now, we have no working tree
1302 continue
Shawn O. Pearcecd1d7ff2009-06-04 16:15:53 -07001303
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001304 if self.UpdateProjectList(opt, m):
1305 err_event.set()
1306 err_update_projects = True
1307 if opt.fail_fast:
1308 print('\nerror: Local checkouts *not* updated.', file=sys.stderr)
1309 sys.exit(1)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -07001310
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001311 err_update_linkfiles = not self.UpdateCopyLinkfileList(m)
1312 if err_update_linkfiles:
1313 err_event.set()
1314 if opt.fail_fast:
1315 print('\nerror: Local update copyfile or linkfile failed.', file=sys.stderr)
1316 sys.exit(1)
jiajia tanga590e642021-04-25 20:02:02 +08001317
Mike Frysinger5a033082019-09-23 19:21:20 -04001318 err_results = []
Mike Frysingerebf04a42021-02-23 20:48:04 -05001319 # NB: We don't exit here because this is the last step.
1320 err_checkout = not self._Checkout(all_projects, opt, err_results)
1321 if err_checkout:
1322 err_event.set()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001323
LaMont Jonesbdcba7d2022-04-11 22:50:11 +00001324 printed_notices = set()
1325 # If there's a notice that's supposed to print at the end of the sync,
1326 # print it now... But avoid printing duplicate messages, and preserve
1327 # order.
1328 for m in sorted(self.ManifestList(opt), key=lambda x: x.path_prefix):
1329 if m.notice and m.notice not in printed_notices:
1330 print(m.notice)
1331 printed_notices.add(m.notice)
Doug Anderson2b8db3c2010-11-01 15:08:06 -07001332
Mike Frysinger5a033082019-09-23 19:21:20 -04001333 # If we saw an error, exit with code 1 so that other scripts can check.
Mike Frysingerbe24a542021-02-23 03:24:12 -05001334 if err_event.is_set():
Mike Frysinger5a033082019-09-23 19:21:20 -04001335 print('\nerror: Unable to fully sync the tree.', file=sys.stderr)
1336 if err_network_sync:
1337 print('error: Downloading network changes failed.', file=sys.stderr)
1338 if err_update_projects:
1339 print('error: Updating local project lists failed.', file=sys.stderr)
jiajia tanga590e642021-04-25 20:02:02 +08001340 if err_update_linkfiles:
1341 print('error: Updating copyfiles or linkfiles failed.', file=sys.stderr)
Mike Frysinger5a033082019-09-23 19:21:20 -04001342 if err_checkout:
1343 print('error: Checking out local projects failed.', file=sys.stderr)
1344 if err_results:
1345 print('Failing repos:\n%s' % '\n'.join(err_results), file=sys.stderr)
1346 print('Try re-running with "-j1 --fail-fast" to exit at the first error.',
1347 file=sys.stderr)
1348 sys.exit(1)
1349
Raman Tenneti7954de12021-07-28 14:36:49 -07001350 # Log the previous sync analysis state from the config.
Raman Tenneti6448a4f2021-09-13 17:40:07 -07001351 self.git_event_log.LogDataConfigEvents(mp.config.GetSyncAnalysisStateData(),
1352 'previous_sync_state')
Raman Tenneti7954de12021-07-28 14:36:49 -07001353
1354 # Update and log with the new sync analysis state.
1355 mp.config.UpdateSyncAnalysisState(opt, superproject_logging_data)
Raman Tenneti6448a4f2021-09-13 17:40:07 -07001356 self.git_event_log.LogDataConfigEvents(mp.config.GetSyncAnalysisStateData(),
1357 'current_sync_state')
Raman Tenneti7954de12021-07-28 14:36:49 -07001358
Mike Frysingere19d9e12020-02-12 11:23:32 -05001359 if not opt.quiet:
1360 print('repo sync has finished successfully.')
1361
David Pursehouse819827a2020-02-12 15:20:19 +09001362
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -07001363def _PostRepoUpgrade(manifest, quiet=False):
Mike Frysingerfdeb20f2021-11-14 03:53:04 -05001364 # Link the docs for the internal .repo/ layout for people
1365 link = os.path.join(manifest.repodir, 'internal-fs-layout.md')
1366 if not platform_utils.islink(link):
1367 target = os.path.join('repo', 'docs', 'internal-fs-layout.md')
1368 try:
1369 platform_utils.symlink(target, link)
Raman Tenneti4a478ed2021-11-17 18:38:24 -08001370 except Exception:
Mike Frysingerfdeb20f2021-11-14 03:53:04 -05001371 pass
1372
Conley Owens094cdbe2014-01-30 15:09:59 -08001373 wrapper = Wrapper()
Conley Owensc9129d92012-10-01 16:12:28 -07001374 if wrapper.NeedSetupGnuPG():
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -07001375 wrapper.SetupGnuPG(quiet)
Conley Owensf2fe2d92014-01-29 13:53:43 -08001376 for project in manifest.projects:
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001377 if project.Exists:
1378 project.PostRepoUpgrade()
1379
David Pursehouse819827a2020-02-12 15:20:19 +09001380
Mike Frysingerc58ec4d2020-02-17 14:36:08 -05001381def _PostRepoFetch(rp, repo_verify=True, verbose=False):
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001382 if rp.HasChanges:
Sarah Owenscecd1d82012-11-01 22:59:27 -07001383 print('info: A new version of repo is available', file=sys.stderr)
Mike Frysinger347f9ed2021-03-15 14:58:52 -04001384 wrapper = Wrapper()
1385 try:
1386 rev = rp.bare_git.describe(rp.GetRevisionId())
1387 except GitError:
1388 rev = None
1389 _, new_rev = wrapper.check_repo_rev(rp.gitdir, rev, repo_verify=repo_verify)
1390 # See if we're held back due to missing signed tag.
1391 current_revid = rp.bare_git.rev_parse('HEAD')
1392 new_revid = rp.bare_git.rev_parse('--verify', new_rev)
1393 if current_revid != new_revid:
1394 # We want to switch to the new rev, but also not trash any uncommitted
1395 # changes. This helps with local testing/hacking.
1396 # If a local change has been made, we will throw that away.
1397 # We also have to make sure this will switch to an older commit if that's
1398 # the latest tag in order to support release rollback.
1399 try:
1400 rp.work_git.reset('--keep', new_rev)
1401 except GitError as e:
1402 sys.exit(str(e))
Sarah Owenscecd1d82012-11-01 22:59:27 -07001403 print('info: Restarting repo with latest version', file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001404 raise RepoChangedException(['--repo-upgraded'])
1405 else:
Sarah Owenscecd1d82012-11-01 22:59:27 -07001406 print('warning: Skipped upgrade to unverified version', file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001407 else:
1408 if verbose:
Sarah Owenscecd1d82012-11-01 22:59:27 -07001409 print('repo version %s is current' % rp.work_git.describe(HEAD),
1410 file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001411
David Pursehouse819827a2020-02-12 15:20:19 +09001412
Dave Borowitz67700e92012-10-23 15:00:54 -07001413class _FetchTimes(object):
Dave Borowitzd9478582012-10-23 16:35:39 -07001414 _ALPHA = 0.5
1415
Dave Borowitz67700e92012-10-23 15:00:54 -07001416 def __init__(self, manifest):
Anthony King85b24ac2014-05-06 15:57:48 +01001417 self._path = os.path.join(manifest.repodir, '.repo_fetchtimes.json')
Dave Borowitz67700e92012-10-23 15:00:54 -07001418 self._times = None
Dave Borowitzd9478582012-10-23 16:35:39 -07001419 self._seen = set()
Dave Borowitz67700e92012-10-23 15:00:54 -07001420
1421 def Get(self, project):
1422 self._Load()
1423 return self._times.get(project.name, _ONE_DAY_S)
1424
1425 def Set(self, project, t):
Dave Borowitzd9478582012-10-23 16:35:39 -07001426 self._Load()
1427 name = project.name
1428 old = self._times.get(name, t)
1429 self._seen.add(name)
1430 a = self._ALPHA
David Pursehouse54a4e602020-02-12 14:31:05 +09001431 self._times[name] = (a * t) + ((1 - a) * old)
Dave Borowitz67700e92012-10-23 15:00:54 -07001432
1433 def _Load(self):
1434 if self._times is None:
1435 try:
Mike Frysinger3164d402019-11-11 05:40:22 -05001436 with open(self._path) as f:
Anthony King85b24ac2014-05-06 15:57:48 +01001437 self._times = json.load(f)
Anthony King85b24ac2014-05-06 15:57:48 +01001438 except (IOError, ValueError):
Mike Frysinger9d96f582021-09-28 11:27:24 -04001439 platform_utils.remove(self._path, missing_ok=True)
Anthony King85b24ac2014-05-06 15:57:48 +01001440 self._times = {}
Dave Borowitz67700e92012-10-23 15:00:54 -07001441
1442 def Save(self):
1443 if self._times is None:
1444 return
Dave Borowitzd9478582012-10-23 16:35:39 -07001445
1446 to_delete = []
1447 for name in self._times:
1448 if name not in self._seen:
1449 to_delete.append(name)
1450 for name in to_delete:
1451 del self._times[name]
1452
Dave Borowitz67700e92012-10-23 15:00:54 -07001453 try:
Mike Frysinger3164d402019-11-11 05:40:22 -05001454 with open(self._path, 'w') as f:
Anthony King85b24ac2014-05-06 15:57:48 +01001455 json.dump(self._times, f, indent=2)
Anthony King85b24ac2014-05-06 15:57:48 +01001456 except (IOError, TypeError):
Mike Frysinger9d96f582021-09-28 11:27:24 -04001457 platform_utils.remove(self._path, missing_ok=True)
Dan Willemsen0745bb22015-08-17 13:41:45 -07001458
1459# This is a replacement for xmlrpc.client.Transport using urllib2
1460# and supporting persistent-http[s]. It cannot change hosts from
1461# request to request like the normal transport, the real url
1462# is passed during initialization.
David Pursehouse819827a2020-02-12 15:20:19 +09001463
1464
Dan Willemsen0745bb22015-08-17 13:41:45 -07001465class PersistentTransport(xmlrpc.client.Transport):
1466 def __init__(self, orig_host):
1467 self.orig_host = orig_host
1468
1469 def request(self, host, handler, request_body, verbose=False):
1470 with GetUrlCookieFile(self.orig_host, not verbose) as (cookiefile, proxy):
1471 # Python doesn't understand cookies with the #HttpOnly_ prefix
1472 # Since we're only using them for HTTP, copy the file temporarily,
1473 # stripping those prefixes away.
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001474 if cookiefile:
Collin Fijalkoviche1191b32020-02-18 10:57:32 -08001475 tmpcookiefile = tempfile.NamedTemporaryFile(mode='w')
David Pursehouse4c5f74e2015-10-02 11:10:10 +09001476 tmpcookiefile.write("# HTTP Cookie File")
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001477 try:
1478 with open(cookiefile) as f:
1479 for line in f:
1480 if line.startswith("#HttpOnly_"):
1481 line = line[len("#HttpOnly_"):]
1482 tmpcookiefile.write(line)
1483 tmpcookiefile.flush()
Dan Willemsen0745bb22015-08-17 13:41:45 -07001484
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001485 cookiejar = cookielib.MozillaCookieJar(tmpcookiefile.name)
David Pursehouseb1ad2192015-09-30 10:35:43 +09001486 try:
1487 cookiejar.load()
1488 except cookielib.LoadError:
1489 cookiejar = cookielib.CookieJar()
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001490 finally:
1491 tmpcookiefile.close()
1492 else:
1493 cookiejar = cookielib.CookieJar()
Dan Willemsen0745bb22015-08-17 13:41:45 -07001494
1495 proxyhandler = urllib.request.ProxyHandler
1496 if proxy:
1497 proxyhandler = urllib.request.ProxyHandler({
1498 "http": proxy,
David Pursehouse54a4e602020-02-12 14:31:05 +09001499 "https": proxy})
Dan Willemsen0745bb22015-08-17 13:41:45 -07001500
1501 opener = urllib.request.build_opener(
1502 urllib.request.HTTPCookieProcessor(cookiejar),
1503 proxyhandler)
1504
1505 url = urllib.parse.urljoin(self.orig_host, handler)
1506 parse_results = urllib.parse.urlparse(url)
1507
1508 scheme = parse_results.scheme
1509 if scheme == 'persistent-http':
1510 scheme = 'http'
1511 if scheme == 'persistent-https':
1512 # If we're proxying through persistent-https, use http. The
1513 # proxy itself will do the https.
1514 if proxy:
1515 scheme = 'http'
1516 else:
1517 scheme = 'https'
1518
1519 # Parse out any authentication information using the base class
1520 host, extra_headers, _ = self.get_host_info(parse_results.netloc)
1521
1522 url = urllib.parse.urlunparse((
1523 scheme,
1524 host,
1525 parse_results.path,
1526 parse_results.params,
1527 parse_results.query,
1528 parse_results.fragment))
1529
1530 request = urllib.request.Request(url, request_body)
1531 if extra_headers is not None:
1532 for (name, header) in extra_headers:
1533 request.add_header(name, header)
1534 request.add_header('Content-Type', 'text/xml')
1535 try:
1536 response = opener.open(request)
1537 except urllib.error.HTTPError as e:
1538 if e.code == 501:
1539 # We may have been redirected through a login process
1540 # but our POST turned into a GET. Retry.
1541 response = opener.open(request)
1542 else:
1543 raise
1544
1545 p, u = xmlrpc.client.getparser()
Mike Frysinger5951e302022-05-20 23:34:44 -04001546 # Response should be fairly small, so read it all at once.
1547 # This way we can show it to the user in case of error (e.g. HTML).
1548 data = response.read()
1549 try:
Dan Willemsen0745bb22015-08-17 13:41:45 -07001550 p.feed(data)
Mike Frysinger5951e302022-05-20 23:34:44 -04001551 except xml.parsers.expat.ExpatError as e:
1552 raise IOError(
1553 f'Parsing the manifest failed: {e}\n'
1554 f'Please report this to your manifest server admin.\n'
1555 f'Here is the full response:\n{data.decode("utf-8")}')
Dan Willemsen0745bb22015-08-17 13:41:45 -07001556 p.close()
1557 return u.close()
1558
1559 def close(self):
1560 pass