blob: 36b1b0acf963136a7d05bac3ee0a2dc8d5206f98 [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
Mike Frysingerebf04a42021-02-23 20:48:04 -050015import functools
Mike Frysingeracf63b22019-06-13 02:24:21 -040016import http.cookiejar as cookielib
Mike Frysinger7b586f22021-02-23 18:38:39 -050017import io
Anthony King85b24ac2014-05-06 15:57:48 +010018import json
Mike Frysingerebf04a42021-02-23 20:48:04 -050019import multiprocessing
David Pursehouse86d973d2012-08-24 10:21:02 +090020import netrc
Shawn O. Pearce2a1ccb22009-04-10 16:51:53 -070021from optparse import SUPPRESS_HELP
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070022import os
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -070023import socket
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070024import sys
Dan Willemsen0745bb22015-08-17 13:41:45 -070025import tempfile
Shawn O. Pearcef6906872009-04-18 10:49:00 -070026import time
Mike Frysingeracf63b22019-06-13 02:24:21 -040027import urllib.error
28import urllib.parse
29import urllib.request
30import xmlrpc.client
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070031
Roy Lee18afd7f2010-05-09 04:32:08 +080032try:
33 import threading as _threading
34except ImportError:
35 import dummy_threading as _threading
36
Shawn O. Pearce97d2b2f2011-09-22 17:23:41 -070037try:
38 import resource
David Pursehouse819827a2020-02-12 15:20:19 +090039
Shawn O. Pearce97d2b2f2011-09-22 17:23:41 -070040 def _rlimit_nofile():
41 return resource.getrlimit(resource.RLIMIT_NOFILE)
42except ImportError:
43 def _rlimit_nofile():
44 return (256, 256)
45
David Rileye0684ad2017-04-05 00:02:59 -070046import event_log
Mike Frysinger347f9ed2021-03-15 14:58:52 -040047from git_command import git_require
David Pursehouseba7bc732015-08-20 16:55:42 +090048from git_config import GetUrlCookieFile
David Pursehoused94aaef2012-09-07 09:52:04 +090049from git_refs import R_HEADS, HEAD
Raman Tenneti6a872c92021-01-14 19:17:50 -080050import git_superproject
Simran Basibdb52712015-08-10 13:23:23 -070051import gitc_utils
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -070052from project import Project
53from project import RemoteSpec
Mike Frysingerebf04a42021-02-23 20:48:04 -050054from command import Command, MirrorSafeCommand, WORKER_BATCH_SIZE
Raman Tenneti1fd7bc22021-02-04 14:39:38 -080055from error import RepoChangedException, GitError, ManifestParseError
Renaud Paquaya65adf72016-11-03 10:37:53 -070056import platform_utils
Shawn O. Pearce350cde42009-04-16 11:21:18 -070057from project import SyncBuffer
Shawn O. Pearce68194f42009-04-10 16:48:52 -070058from progress import Progress
Conley Owens094cdbe2014-01-30 15:09:59 -080059from wrapper import Wrapper
Dan Willemsen5ea32d12015-09-08 13:27:20 -070060from manifest_xml import GitcManifest
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070061
Dave Borowitz67700e92012-10-23 15:00:54 -070062_ONE_DAY_S = 24 * 60 * 60
63
David Pursehouse819827a2020-02-12 15:20:19 +090064
Shawn O. Pearcec95583b2009-03-03 17:47:06 -080065class Sync(Command, MirrorSafeCommand):
Roy Lee18afd7f2010-05-09 04:32:08 +080066 jobs = 1
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070067 common = True
68 helpSummary = "Update working tree to the latest revision"
69 helpUsage = """
70%prog [<project>...]
71"""
72 helpDescription = """
73The '%prog' command synchronizes local project directories
74with the remote repositories specified in the manifest. If a local
75project does not yet exist, it will clone a new local directory from
76the remote repository and set up tracking branches as specified in
77the manifest. If the local project already exists, '%prog'
78will update the remote branches and rebase any new local changes
79on top of the new remote changes.
80
81'%prog' will synchronize all projects listed at the command
82line. Projects can be specified either by name, or by a relative
83or absolute path to the project's local directory. If no projects
84are specified, '%prog' will synchronize all projects listed in
85the manifest.
Shawn O. Pearce3e768c92009-04-10 16:59:36 -070086
87The -d/--detach option can be used to switch specified projects
88back to the manifest revision. This option is especially helpful
89if the project is currently on a topic branch, but the manifest
90revision is temporarily needed.
Shawn O. Pearceeb7af872009-04-21 08:02:04 -070091
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -070092The -s/--smart-sync option can be used to sync to a known good
93build as specified by the manifest-server element in the current
Victor Boivie08c880d2011-04-19 10:32:52 +020094manifest. The -t/--smart-tag option is similar and allows you to
95specify a custom tag/label.
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -070096
David Pursehousecf76b1b2012-09-14 10:31:42 +090097The -u/--manifest-server-username and -p/--manifest-server-password
98options can be used to specify a username and password to authenticate
99with the manifest server when using the -s or -t option.
100
101If -u and -p are not specified when using the -s or -t option, '%prog'
102will attempt to read authentication credentials for the manifest server
103from the user's .netrc file.
104
105'%prog' will not use authentication credentials from -u/-p or .netrc
106if the manifest server specified in the manifest file already includes
107credentials.
108
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400109By default, all projects will be synced. The --fail-fast option can be used
Mike Frysinger7ae210a2020-05-24 14:56:52 -0400110to halt syncing as soon as possible when the first project fails to sync.
Andrei Warkentin5df6de02010-07-02 17:58:31 -0500111
Kevin Degiabaa7f32014-11-12 11:27:45 -0700112The --force-sync option can be used to overwrite existing git
113directories if they have previously been linked to a different
Roger Shimizuac29ac32020-06-06 02:33:40 +0900114object directory. WARNING: This may cause data to be lost since
Kevin Degiabaa7f32014-11-12 11:27:45 -0700115refs may be removed when overwriting.
116
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500117The --force-remove-dirty option can be used to remove previously used
118projects with uncommitted changes. WARNING: This may cause data to be
119lost since uncommitted changes may be removed with projects that no longer
120exist in the manifest.
121
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -0700122The --no-clone-bundle option disables any attempt to use
123$URL/clone.bundle to bootstrap a new Git repository from a
124resumeable bundle file on a content delivery network. This
125may be necessary if there are problems with the local Python
126HTTP client or proxy configuration, but the Git binary works.
127
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800128The --fetch-submodules option enables fetching Git submodules
129of a project from server.
130
David Pursehousef2fad612015-01-29 14:36:28 +0900131The -c/--current-branch option can be used to only fetch objects that
132are on the branch specified by a project's revision.
133
David Pursehouseb1553542014-09-04 21:28:09 +0900134The --optimized-fetch option can be used to only fetch projects that
135are fixed to a sha1 revision if the sha1 revision does not already
136exist locally.
137
David Pursehouse74cfd272015-10-14 10:50:15 +0900138The --prune option can be used to remove any refs that no longer
139exist on the remote.
140
Mike Frysingerb8f7bb02018-10-10 01:05:11 -0400141# SSH Connections
Shawn O. Pearceeb7af872009-04-21 08:02:04 -0700142
143If at least one project remote URL uses an SSH connection (ssh://,
144git+ssh://, or user@host:path syntax) repo will automatically
145enable the SSH ControlMaster option when connecting to that host.
146This feature permits other projects in the same '%prog' session to
147reuse the same SSH tunnel, saving connection setup overheads.
148
149To disable this behavior on UNIX platforms, set the GIT_SSH
150environment variable to 'ssh'. For example:
151
152 export GIT_SSH=ssh
153 %prog
154
Mike Frysingerb8f7bb02018-10-10 01:05:11 -0400155# Compatibility
Shawn O. Pearceeb7af872009-04-21 08:02:04 -0700156
157This feature is automatically disabled on Windows, due to the lack
158of UNIX domain socket support.
159
160This feature is not compatible with url.insteadof rewrites in the
161user's ~/.gitconfig. '%prog' is currently not able to perform the
162rewrite early enough to establish the ControlMaster tunnel.
163
164If the remote SSH daemon is Gerrit Code Review, version 2.0.10 or
165later is required to fix a server side protocol bug.
166
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700167"""
Mike Frysinger6a2400a2021-02-16 01:43:31 -0500168 PARALLEL_JOBS = 1
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700169
Nico Sallembien6623b212010-05-11 12:57:01 -0700170 def _Options(self, p, show_smart=True):
Torne (Richard Coles)7bdbde72012-12-05 10:58:06 +0000171 try:
Mike Frysinger6a2400a2021-02-16 01:43:31 -0500172 self.PARALLEL_JOBS = self.manifest.default.sync_j
Torne (Richard Coles)7bdbde72012-12-05 10:58:06 +0000173 except ManifestParseError:
Mike Frysinger6a2400a2021-02-16 01:43:31 -0500174 pass
175 super()._Options(p)
Shawn O. Pearce6392c872011-09-22 17:44:31 -0700176
Andrei Warkentin5df6de02010-07-02 17:58:31 -0500177 p.add_option('-f', '--force-broken',
Stefan Müller-Klieser46702ed2019-08-30 10:20:15 +0200178 dest='force_broken', action='store_true',
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400179 help='obsolete option (to be deleted in the future)')
180 p.add_option('--fail-fast',
181 dest='fail_fast', action='store_true',
182 help='stop syncing after first error is hit')
Kevin Degiabaa7f32014-11-12 11:27:45 -0700183 p.add_option('--force-sync',
184 dest='force_sync', action='store_true',
185 help="overwrite an existing git directory if it needs to "
186 "point to a different object directory. WARNING: this "
187 "may cause loss of data")
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500188 p.add_option('--force-remove-dirty',
189 dest='force_remove_dirty', action='store_true',
190 help="force remove projects with uncommitted modifications if "
191 "projects no longer exist in the manifest. "
192 "WARNING: this may cause loss of data")
David Pursehouse8f62fb72012-11-14 12:09:38 +0900193 p.add_option('-l', '--local-only',
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700194 dest='local_only', action='store_true',
195 help="only update working tree, don't fetch")
David Pursehouse54a4e602020-02-12 14:31:05 +0900196 p.add_option('--no-manifest-update', '--nmu',
Fredrik de Grootcc960972019-11-22 09:04:31 +0100197 dest='mp_update', action='store_false', default='true',
198 help='use the existing manifest checkout as-is. '
199 '(do not update to the latest revision)')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900200 p.add_option('-n', '--network-only',
Shawn O. Pearce96fdcef2009-04-10 16:29:20 -0700201 dest='network_only', action='store_true',
202 help="fetch only, don't update working tree")
David Pursehouse8f62fb72012-11-14 12:09:38 +0900203 p.add_option('-d', '--detach',
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700204 dest='detach_head', action='store_true',
205 help='detach projects back to manifest revision')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900206 p.add_option('-c', '--current-branch',
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -0700207 dest='current_branch_only', action='store_true',
208 help='fetch only current branch from server')
Mike Frysinger521d01b2020-02-17 01:51:49 -0500209 p.add_option('-v', '--verbose',
210 dest='output_mode', action='store_true',
211 help='show all sync output')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900212 p.add_option('-q', '--quiet',
Mike Frysinger521d01b2020-02-17 01:51:49 -0500213 dest='output_mode', action='store_false',
214 help='only show errors')
Chris Wolfee9dc3b32012-01-26 11:36:18 -0500215 p.add_option('-m', '--manifest-name',
216 dest='manifest_name',
217 help='temporary manifest to use for this sync', metavar='NAME.xml')
Xin Lid79a4bc2020-05-20 16:03:45 -0700218 p.add_option('--clone-bundle', action='store_true',
219 help='enable use of /clone.bundle on HTTP/HTTPS')
220 p.add_option('--no-clone-bundle', dest='clone_bundle', action='store_false',
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -0700221 help='disable use of /clone.bundle on HTTP/HTTPS')
Conley Owens8d070cf2012-11-06 13:14:31 -0800222 p.add_option('-u', '--manifest-server-username', action='store',
223 dest='manifest_server_username',
224 help='username to authenticate with the manifest server')
225 p.add_option('-p', '--manifest-server-password', action='store',
226 dest='manifest_server_password',
227 help='password to authenticate with the manifest server')
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800228 p.add_option('--fetch-submodules',
229 dest='fetch_submodules', action='store_true',
230 help='fetch submodules from server')
Raman Tenneti6a872c92021-01-14 19:17:50 -0800231 p.add_option('--use-superproject', action='store_true',
232 help='use the manifest superproject to sync projects')
Mitchel Humpherys597868b2012-10-29 10:18:34 -0700233 p.add_option('--no-tags',
Mike Frysingerc58ec4d2020-02-17 14:36:08 -0500234 dest='tags', default=True, action='store_false',
Mitchel Humpherys597868b2012-10-29 10:18:34 -0700235 help="don't fetch tags")
David Pursehouseb1553542014-09-04 21:28:09 +0900236 p.add_option('--optimized-fetch',
237 dest='optimized_fetch', action='store_true',
238 help='only fetch projects fixed to sha1 if revision does not exist locally')
George Engelbrecht9bc283e2020-04-02 12:36:09 -0600239 p.add_option('--retry-fetches',
240 default=0, action='store', type='int',
241 help='number of times to retry fetches on transient errors')
David Pursehouse74cfd272015-10-14 10:50:15 +0900242 p.add_option('--prune', dest='prune', action='store_true',
243 help='delete refs that no longer exist on the remote')
Nico Sallembien6623b212010-05-11 12:57:01 -0700244 if show_smart:
245 p.add_option('-s', '--smart-sync',
246 dest='smart_sync', action='store_true',
David Pursehouse79fba682016-04-13 18:03:00 +0900247 help='smart sync using manifest from the latest known good build')
Victor Boivie08c880d2011-04-19 10:32:52 +0200248 p.add_option('-t', '--smart-tag',
249 dest='smart_tag', action='store',
250 help='smart sync using manifest from a known tag')
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700251
Shawn O. Pearcefd89b672009-04-18 11:28:57 -0700252 g = p.add_option_group('repo Version options')
253 g.add_option('--no-repo-verify',
Mike Frysingerc58ec4d2020-02-17 14:36:08 -0500254 dest='repo_verify', default=True, action='store_false',
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700255 help='do not verify repo source code')
Shawn O. Pearcefd89b672009-04-18 11:28:57 -0700256 g.add_option('--repo-upgraded',
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800257 dest='repo_upgraded', action='store_true',
Shawn O. Pearce2a1ccb22009-04-10 16:51:53 -0700258 help=SUPPRESS_HELP)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700259
Raman Tenneti8d43dea2021-02-07 16:30:27 -0800260 def _GetBranch(self):
261 """Returns the branch name for getting the approved manifest."""
262 p = self.manifest.manifestProject
263 b = p.GetBranch(p.CurrentBranch)
264 branch = b.merge
265 if branch.startswith(R_HEADS):
266 branch = branch[len(R_HEADS):]
267 return branch
268
Raman Tenneti2ae44d72021-03-23 15:12:27 -0700269 def _UseSuperproject(self, opt):
270 """Returns True if use-superproject option is enabled"""
271 return (opt.use_superproject or
272 self.manifest.manifestProject.config.GetBoolean(
273 'repo.superproject'))
274
275 def _GetCurrentBranchOnly(self, opt):
276 """Returns True if current-branch or use-superproject options are enabled."""
277 return opt.current_branch_only or self._UseSuperproject(opt)
278
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800279 def _UpdateProjectsRevisionId(self, opt, args):
280 """Update revisionId of every project with the SHA from superproject.
281
282 This function updates each project's revisionId with SHA from superproject.
283 It writes the updated manifest into a file and reloads the manifest from it.
284
285 Args:
286 opt: Program options returned from optparse. See _Options().
287 args: Arguments to pass to GetProjects. See the GetProjects
288 docstring for details.
289
290 Returns:
291 Returns path to the overriding manifest file.
292 """
Raman Tenneti21dce3d2021-02-09 00:26:31 -0800293 superproject = git_superproject.Superproject(self.manifest,
Raman Tennetief99ec02021-03-04 10:29:40 -0800294 self.repodir,
295 quiet=opt.quiet)
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800296 all_projects = self.GetProjects(args,
297 missing_ok=True,
298 submodules_ok=opt.fetch_submodules)
Raman Tenneti21dce3d2021-02-09 00:26:31 -0800299 manifest_path = superproject.UpdateProjectsRevisionId(all_projects)
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800300 if not manifest_path:
301 print('error: Update of revsionId from superproject has failed',
302 file=sys.stderr)
303 sys.exit(1)
304 self._ReloadManifest(manifest_path)
305 return manifest_path
306
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500307 def _FetchProjectList(self, opt, projects):
308 """Main function of the fetch worker.
309
310 The projects we're given share the same underlying git object store, so we
311 have to fetch them in serial.
Roy Lee18afd7f2010-05-09 04:32:08 +0800312
David James8d201162013-10-11 17:03:19 -0700313 Delegates most of the work to _FetchHelper.
314
315 Args:
316 opt: Program options returned from optparse. See _Options().
317 projects: Projects to fetch.
David James8d201162013-10-11 17:03:19 -0700318 """
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500319 return [self._FetchOne(opt, x) for x in projects]
David James8d201162013-10-11 17:03:19 -0700320
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500321 def _FetchOne(self, opt, project):
David James8d201162013-10-11 17:03:19 -0700322 """Fetch git objects for a single project.
323
David Pursehousec1b86a22012-11-14 11:36:51 +0900324 Args:
325 opt: Program options returned from optparse. See _Options().
326 project: Project object for the project to fetch.
David James8d201162013-10-11 17:03:19 -0700327
328 Returns:
329 Whether the fetch was successful.
David Pursehousec1b86a22012-11-14 11:36:51 +0900330 """
David Rileye0684ad2017-04-05 00:02:59 -0700331 start = time.time()
332 success = False
Mike Frysinger7b586f22021-02-23 18:38:39 -0500333 buf = io.StringIO()
David Pursehousec1b86a22012-11-14 11:36:51 +0900334 try:
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500335 success = project.Sync_NetworkHalf(
336 quiet=opt.quiet,
337 verbose=opt.verbose,
338 output_redir=buf,
339 current_branch_only=self._GetCurrentBranchOnly(opt),
340 force_sync=opt.force_sync,
341 clone_bundle=opt.clone_bundle,
342 tags=opt.tags, archive=self.manifest.IsArchive,
343 optimized_fetch=opt.optimized_fetch,
344 retry_fetches=opt.retry_fetches,
345 prune=opt.prune,
346 clone_filter=self.manifest.CloneFilter)
Doug Andersonfc06ced2011-03-16 15:49:18 -0700347
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500348 output = buf.getvalue()
349 if opt.verbose and output:
350 print('\n' + output.rstrip())
Doug Andersonfc06ced2011-03-16 15:49:18 -0700351
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500352 if not success:
353 print('error: Cannot fetch %s from %s'
354 % (project.name, project.remote.url),
355 file=sys.stderr)
356 except Exception as e:
357 print('error: Cannot fetch %s (%s: %s)'
358 % (project.name, type(e).__name__, str(e)), file=sys.stderr)
359 raise
Mike Frysinger7b586f22021-02-23 18:38:39 -0500360
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500361 finish = time.time()
362 return (success, project, start, finish)
David James8d201162013-10-11 17:03:19 -0700363
Mike Frysinger5a033082019-09-23 19:21:20 -0400364 def _Fetch(self, projects, opt, err_event):
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500365 ret = True
366
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700367 fetched = set()
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500368 pm = Progress('Fetching', len(projects), delay=False)
Roy Lee18afd7f2010-05-09 04:32:08 +0800369
David James89ece422014-01-09 18:51:58 -0800370 objdir_project_map = dict()
371 for project in projects:
372 objdir_project_map.setdefault(project.objdir, []).append(project)
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500373 projects_list = list(objdir_project_map.values())
David James8d201162013-10-11 17:03:19 -0700374
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500375 def _ProcessResults(results_sets):
376 ret = True
377 for results in results_sets:
378 for (success, project, start, finish) in results:
379 self._fetch_times.Set(project, finish - start)
380 self.event_log.AddSync(project, event_log.TASK_SYNC_NETWORK,
381 start, finish, success)
382 # Check for any errors before running any more tasks.
383 # ...we'll let existing jobs finish, though.
384 if not success:
385 ret = False
386 else:
387 fetched.add(project.gitdir)
388 pm.update(msg=project.name)
389 if not ret and opt.fail_fast:
390 break
391 return ret
Doug Andersonfc06ced2011-03-16 15:49:18 -0700392
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500393 # NB: Multiprocessing is heavy, so don't spin it up for one job.
394 if len(projects_list) == 1 or opt.jobs == 1:
395 if not _ProcessResults(self._FetchProjectList(opt, x) for x in projects_list):
396 ret = False
397 else:
398 # Favor throughput over responsiveness when quiet. It seems that imap()
399 # will yield results in batches relative to chunksize, so even as the
400 # children finish a sync, we won't see the result until one child finishes
401 # ~chunksize jobs. When using a large --jobs with large chunksize, this
402 # can be jarring as there will be a large initial delay where repo looks
403 # like it isn't doing anything and sits at 0%, but then suddenly completes
404 # a lot of jobs all at once. Since this code is more network bound, we
405 # can accept a bit more CPU overhead with a smaller chunksize so that the
406 # user sees more immediate & continuous feedback.
407 if opt.quiet:
408 chunksize = WORKER_BATCH_SIZE
David James89ece422014-01-09 18:51:58 -0800409 else:
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500410 pm.update(inc=0, msg='warming up')
411 chunksize = 4
412 with multiprocessing.Pool(opt.jobs) as pool:
413 results = pool.imap_unordered(
414 functools.partial(self._FetchProjectList, opt),
415 projects_list,
416 chunksize=chunksize)
417 if not _ProcessResults(results):
418 ret = False
419 pool.close()
Roy Lee18afd7f2010-05-09 04:32:08 +0800420
Shawn O. Pearce68194f42009-04-10 16:48:52 -0700421 pm.end()
Dave Borowitz67700e92012-10-23 15:00:54 -0700422 self._fetch_times.Save()
Dave Borowitz18857212012-10-23 17:02:59 -0700423
Julien Campergue335f5ef2013-10-16 11:02:35 +0200424 if not self.manifest.IsArchive:
Mike Frysinger5a033082019-09-23 19:21:20 -0400425 self._GCProjects(projects, opt, err_event)
Julien Campergue335f5ef2013-10-16 11:02:35 +0200426
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500427 return (ret, fetched)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700428
Mike Frysingerebf04a42021-02-23 20:48:04 -0500429 def _CheckoutOne(self, opt, project):
Xin Li745be2e2019-06-03 11:24:30 -0700430 """Checkout work tree for one project
431
432 Args:
433 opt: Program options returned from optparse. See _Options().
434 project: Project object for the project to checkout.
Xin Li745be2e2019-06-03 11:24:30 -0700435
436 Returns:
437 Whether the fetch was successful.
438 """
Xin Li745be2e2019-06-03 11:24:30 -0700439 start = time.time()
440 syncbuf = SyncBuffer(self.manifest.manifestProject.config,
441 detach_head=opt.detach_head)
442 success = False
443 try:
Mike Frysingerebf04a42021-02-23 20:48:04 -0500444 project.Sync_LocalHalf(syncbuf, force_sync=opt.force_sync)
445 success = syncbuf.Finish()
446 except Exception as e:
447 print('error: Cannot checkout %s: %s: %s' %
448 (project.name, type(e).__name__, str(e)),
449 file=sys.stderr)
450 raise
Xin Li745be2e2019-06-03 11:24:30 -0700451
Mike Frysingerebf04a42021-02-23 20:48:04 -0500452 if not success:
453 print('error: Cannot checkout %s' % (project.name), file=sys.stderr)
454 finish = time.time()
455 return (success, project, start, finish)
Xin Li745be2e2019-06-03 11:24:30 -0700456
Mike Frysingerebf04a42021-02-23 20:48:04 -0500457 def _Checkout(self, all_projects, opt, err_results):
Xin Li745be2e2019-06-03 11:24:30 -0700458 """Checkout projects listed in all_projects
459
460 Args:
461 all_projects: List of all projects that should be checked out.
462 opt: Program options returned from optparse. See _Options().
Mike Frysingerebf04a42021-02-23 20:48:04 -0500463 err_results: A list of strings, paths to git repos where checkout failed.
Xin Li745be2e2019-06-03 11:24:30 -0700464 """
Mike Frysingerebf04a42021-02-23 20:48:04 -0500465 ret = True
Xin Li745be2e2019-06-03 11:24:30 -0700466
Mike Frysingerebf04a42021-02-23 20:48:04 -0500467 # Only checkout projects with worktrees.
468 all_projects = [x for x in all_projects if x.worktree]
Xin Li745be2e2019-06-03 11:24:30 -0700469
Mike Frysingerfbb95a42021-02-23 17:34:35 -0500470 pm = Progress('Checking out', len(all_projects))
Xin Li745be2e2019-06-03 11:24:30 -0700471
Mike Frysingerebf04a42021-02-23 20:48:04 -0500472 def _ProcessResults(results):
473 for (success, project, start, finish) in results:
474 self.event_log.AddSync(project, event_log.TASK_SYNC_LOCAL,
475 start, finish, success)
476 # Check for any errors before running any more tasks.
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500477 # ...we'll let existing jobs finish, though.
Mike Frysingerebf04a42021-02-23 20:48:04 -0500478 if not success:
479 err_results.append(project.relpath)
480 if opt.fail_fast:
481 return False
482 pm.update(msg=project.name)
483 return True
Xin Li745be2e2019-06-03 11:24:30 -0700484
Mike Frysingerebf04a42021-02-23 20:48:04 -0500485 # NB: Multiprocessing is heavy, so don't spin it up for one job.
486 if len(all_projects) == 1 or opt.jobs == 1:
487 if not _ProcessResults(self._CheckoutOne(opt, x) for x in all_projects):
488 ret = False
489 else:
490 with multiprocessing.Pool(opt.jobs) as pool:
491 results = pool.imap_unordered(
492 functools.partial(self._CheckoutOne, opt),
493 all_projects,
494 chunksize=WORKER_BATCH_SIZE)
495 if not _ProcessResults(results):
496 ret = False
497 pool.close()
Xin Li745be2e2019-06-03 11:24:30 -0700498
499 pm.end()
Xin Li745be2e2019-06-03 11:24:30 -0700500
Mike Frysinger511a0e52021-03-12 20:03:51 -0500501 return ret and not err_results
Mike Frysingerebf04a42021-02-23 20:48:04 -0500502
Mike Frysinger5a033082019-09-23 19:21:20 -0400503 def _GCProjects(self, projects, opt, err_event):
Gabe Black2ff30292014-10-09 17:54:35 -0700504 gc_gitdirs = {}
David James8d201162013-10-11 17:03:19 -0700505 for project in projects:
Mike Frysinger5f2b0452020-02-11 03:23:24 -0500506 # Make sure pruning never kicks in with shared projects.
Mike Frysinger979d5bd2020-02-09 02:28:34 -0500507 if (not project.use_git_worktrees and
David Pursehouseaa611a22020-02-20 10:47:26 +0900508 len(project.manifest.GetProjectsWithName(project.name)) > 1):
Anders Björklund2a2da802021-01-18 10:32:36 +0100509 if not opt.quiet:
510 print('%s: Shared project %s found, disabling pruning.' %
511 (project.relpath, project.name))
Mike Frysinger5f2b0452020-02-11 03:23:24 -0500512 if git_require((2, 7, 0)):
Mike Frysingerf81c72e2020-02-19 15:50:00 -0500513 project.EnableRepositoryExtension('preciousObjects')
Mike Frysinger5f2b0452020-02-11 03:23:24 -0500514 else:
515 # This isn't perfect, but it's the best we can do with old git.
516 print('%s: WARNING: shared projects are unreliable when using old '
517 'versions of git; please upgrade to git-2.7.0+.'
518 % (project.relpath,),
519 file=sys.stderr)
520 project.config.SetString('gc.pruneExpire', 'never')
Gabe Black2ff30292014-10-09 17:54:35 -0700521 gc_gitdirs[project.gitdir] = project.bare_git
David James8d201162013-10-11 17:03:19 -0700522
Mike Frysinger6f1c6262020-02-04 00:09:23 -0500523 if multiprocessing:
Dave Borowitz18857212012-10-23 17:02:59 -0700524 cpu_count = multiprocessing.cpu_count()
525 else:
526 cpu_count = 1
527 jobs = min(self.jobs, cpu_count)
528
529 if jobs < 2:
Gabe Black2ff30292014-10-09 17:54:35 -0700530 for bare_git in gc_gitdirs.values():
David James8d201162013-10-11 17:03:19 -0700531 bare_git.gc('--auto')
Dave Borowitz18857212012-10-23 17:02:59 -0700532 return
533
Mike Frysinger0c0e9342019-06-13 12:42:39 -0400534 config = {'pack.threads': cpu_count // jobs if cpu_count > jobs else 1}
Dave Borowitz18857212012-10-23 17:02:59 -0700535
536 threads = set()
537 sem = _threading.Semaphore(jobs)
Dave Borowitz18857212012-10-23 17:02:59 -0700538
David James8d201162013-10-11 17:03:19 -0700539 def GC(bare_git):
Dave Borowitz18857212012-10-23 17:02:59 -0700540 try:
541 try:
David James8d201162013-10-11 17:03:19 -0700542 bare_git.gc('--auto', config=config)
Dave Borowitz18857212012-10-23 17:02:59 -0700543 except GitError:
544 err_event.set()
David Pursehouse145e35b2020-02-12 15:40:47 +0900545 except Exception:
Dave Borowitz18857212012-10-23 17:02:59 -0700546 err_event.set()
547 raise
548 finally:
549 sem.release()
550
Gabe Black2ff30292014-10-09 17:54:35 -0700551 for bare_git in gc_gitdirs.values():
Mike Frysingerbe24a542021-02-23 03:24:12 -0500552 if err_event.is_set() and opt.fail_fast:
Dave Borowitz18857212012-10-23 17:02:59 -0700553 break
554 sem.acquire()
David James8d201162013-10-11 17:03:19 -0700555 t = _threading.Thread(target=GC, args=(bare_git,))
Dave Borowitz18857212012-10-23 17:02:59 -0700556 t.daemon = True
557 threads.add(t)
558 t.start()
559
560 for t in threads:
561 t.join()
562
Tim Kilbourn07669002013-03-08 15:02:49 -0800563 def _ReloadManifest(self, manifest_name=None):
564 if manifest_name:
565 # Override calls _Unload already
566 self.manifest.Override(manifest_name)
567 else:
568 self.manifest._Unload()
569
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500570 def UpdateProjectList(self, opt):
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700571 new_project_paths = []
Colin Cross5acde752012-03-28 20:15:45 -0700572 for project in self.GetProjects(None, missing_ok=True):
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700573 if project.relpath:
574 new_project_paths.append(project.relpath)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700575 file_name = 'project.list'
Mike Frysingere3315bb2021-02-09 23:45:28 -0500576 file_path = os.path.join(self.repodir, file_name)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700577 old_project_paths = []
578
579 if os.path.exists(file_path):
Mike Frysinger3164d402019-11-11 05:40:22 -0500580 with open(file_path, 'r') as fd:
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700581 old_project_paths = fd.read().split('\n')
Kuang-che Wu0d9b16d2019-04-06 00:49:47 +0800582 # In reversed order, so subfolders are deleted before parent folder.
583 for path in sorted(old_project_paths, reverse=True):
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700584 if not path:
585 continue
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700586 if path not in new_project_paths:
David Pursehouse8a68ff92012-09-24 12:15:13 +0900587 # If the path has already been deleted, we don't need to do it
Dan Willemsen43507912016-09-01 16:26:02 -0700588 gitdir = os.path.join(self.manifest.topdir, path, '.git')
589 if os.path.exists(gitdir):
David Pursehousec1b86a22012-11-14 11:36:51 +0900590 project = Project(
David Pursehouseabdf7502020-02-12 14:58:39 +0900591 manifest=self.manifest,
592 name=path,
593 remote=RemoteSpec('origin'),
594 gitdir=gitdir,
595 objdir=gitdir,
Mike Frysingerc0d18662020-02-19 19:19:18 -0500596 use_git_worktrees=os.path.isfile(gitdir),
David Pursehouseabdf7502020-02-12 14:58:39 +0900597 worktree=os.path.join(self.manifest.topdir, path),
598 relpath=path,
599 revisionExpr='HEAD',
600 revisionId=None,
601 groups=None)
Mike Frysingerc0d18662020-02-19 19:19:18 -0500602 if not project.DeleteWorktree(
David Pursehouseaa611a22020-02-20 10:47:26 +0900603 quiet=opt.quiet,
604 force=opt.force_remove_dirty):
Mike Frysingera850ca22019-08-07 17:19:24 -0400605 return 1
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700606
Shawn O. Pearce9fb29ce2009-06-04 20:41:02 -0700607 new_project_paths.sort()
Mike Frysinger3164d402019-11-11 05:40:22 -0500608 with open(file_path, 'w') as fd:
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700609 fd.write('\n'.join(new_project_paths))
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700610 fd.write('\n')
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700611 return 0
612
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400613 def _SmartSyncSetup(self, opt, smart_sync_manifest_path):
614 if not self.manifest.manifest_server:
615 print('error: cannot smart sync: no manifest server defined in '
616 'manifest', file=sys.stderr)
617 sys.exit(1)
618
619 manifest_server = self.manifest.manifest_server
620 if not opt.quiet:
621 print('Using manifest server %s' % manifest_server)
622
David Pursehouseeeff3532020-02-12 11:24:10 +0900623 if '@' not in manifest_server:
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400624 username = None
625 password = None
626 if opt.manifest_server_username and opt.manifest_server_password:
627 username = opt.manifest_server_username
628 password = opt.manifest_server_password
629 else:
630 try:
631 info = netrc.netrc()
632 except IOError:
633 # .netrc file does not exist or could not be opened
634 pass
635 else:
636 try:
637 parse_result = urllib.parse.urlparse(manifest_server)
638 if parse_result.hostname:
639 auth = info.authenticators(parse_result.hostname)
640 if auth:
641 username, _account, password = auth
642 else:
643 print('No credentials found for %s in .netrc'
644 % parse_result.hostname, file=sys.stderr)
645 except netrc.NetrcParseError as e:
646 print('Error parsing .netrc file: %s' % e, file=sys.stderr)
647
648 if (username and password):
649 manifest_server = manifest_server.replace('://', '://%s:%s@' %
650 (username, password),
651 1)
652
653 transport = PersistentTransport(manifest_server)
654 if manifest_server.startswith('persistent-'):
655 manifest_server = manifest_server[len('persistent-'):]
656
657 try:
658 server = xmlrpc.client.Server(manifest_server, transport=transport)
659 if opt.smart_sync:
Raman Tenneti8d43dea2021-02-07 16:30:27 -0800660 branch = self._GetBranch()
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400661
Mike Frysinger56ce3462019-12-04 19:30:48 -0500662 if 'SYNC_TARGET' in os.environ:
Mike Frysingere20da3e2020-03-07 01:53:53 -0500663 target = os.environ['SYNC_TARGET']
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400664 [success, manifest_str] = server.GetApprovedManifest(branch, target)
Mike Frysinger56ce3462019-12-04 19:30:48 -0500665 elif ('TARGET_PRODUCT' in os.environ and
666 'TARGET_BUILD_VARIANT' in os.environ):
Mike Frysingere20da3e2020-03-07 01:53:53 -0500667 target = '%s-%s' % (os.environ['TARGET_PRODUCT'],
668 os.environ['TARGET_BUILD_VARIANT'])
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400669 [success, manifest_str] = server.GetApprovedManifest(branch, target)
670 else:
671 [success, manifest_str] = server.GetApprovedManifest(branch)
672 else:
673 assert(opt.smart_tag)
674 [success, manifest_str] = server.GetManifest(opt.smart_tag)
675
676 if success:
677 manifest_name = os.path.basename(smart_sync_manifest_path)
678 try:
Mike Frysinger3164d402019-11-11 05:40:22 -0500679 with open(smart_sync_manifest_path, 'w') as f:
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400680 f.write(manifest_str)
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400681 except IOError as e:
682 print('error: cannot write manifest to %s:\n%s'
683 % (smart_sync_manifest_path, e),
684 file=sys.stderr)
685 sys.exit(1)
686 self._ReloadManifest(manifest_name)
687 else:
688 print('error: manifest server RPC call failed: %s' %
689 manifest_str, file=sys.stderr)
690 sys.exit(1)
691 except (socket.error, IOError, xmlrpc.client.Fault) as e:
692 print('error: cannot connect to manifest server %s:\n%s'
693 % (self.manifest.manifest_server, e), file=sys.stderr)
694 sys.exit(1)
695 except xmlrpc.client.ProtocolError as e:
696 print('error: cannot connect to manifest server %s:\n%d %s'
697 % (self.manifest.manifest_server, e.errcode, e.errmsg),
698 file=sys.stderr)
699 sys.exit(1)
700
701 return manifest_name
702
Mike Frysingerfb527e32019-08-27 02:34:32 -0400703 def _UpdateManifestProject(self, opt, mp, manifest_name):
704 """Fetch & update the local manifest project."""
705 if not opt.local_only:
706 start = time.time()
Mike Frysinger521d01b2020-02-17 01:51:49 -0500707 success = mp.Sync_NetworkHalf(quiet=opt.quiet, verbose=opt.verbose,
Raman Tenneti2ae44d72021-03-23 15:12:27 -0700708 current_branch_only=self._GetCurrentBranchOnly(opt),
Erwan Yvindc5c4d12019-06-18 13:49:12 +0200709 force_sync=opt.force_sync,
Mike Frysingerc58ec4d2020-02-17 14:36:08 -0500710 tags=opt.tags,
Mike Frysingerfb527e32019-08-27 02:34:32 -0400711 optimized_fetch=opt.optimized_fetch,
George Engelbrecht9bc283e2020-04-02 12:36:09 -0600712 retry_fetches=opt.retry_fetches,
Mike Frysingerfb527e32019-08-27 02:34:32 -0400713 submodules=self.manifest.HasSubmodules,
714 clone_filter=self.manifest.CloneFilter)
715 finish = time.time()
716 self.event_log.AddSync(mp, event_log.TASK_SYNC_NETWORK,
717 start, finish, success)
718
719 if mp.HasChanges:
720 syncbuf = SyncBuffer(mp.config)
721 start = time.time()
722 mp.Sync_LocalHalf(syncbuf, submodules=self.manifest.HasSubmodules)
723 clean = syncbuf.Finish()
724 self.event_log.AddSync(mp, event_log.TASK_SYNC_LOCAL,
725 start, time.time(), clean)
726 if not clean:
727 sys.exit(1)
728 self._ReloadManifest(opt.manifest_name)
729 if opt.jobs is None:
730 self.jobs = self.manifest.default.sync_j
731
Mike Frysingerae6cb082019-08-27 01:10:59 -0400732 def ValidateOptions(self, opt, args):
733 if opt.force_broken:
734 print('warning: -f/--force-broken is now the default behavior, and the '
735 'options are deprecated', file=sys.stderr)
736 if opt.network_only and opt.detach_head:
737 self.OptionParser.error('cannot combine -n and -d')
738 if opt.network_only and opt.local_only:
739 self.OptionParser.error('cannot combine -n and -l')
740 if opt.manifest_name and opt.smart_sync:
741 self.OptionParser.error('cannot combine -m and -s')
742 if opt.manifest_name and opt.smart_tag:
743 self.OptionParser.error('cannot combine -m and -t')
744 if opt.manifest_server_username or opt.manifest_server_password:
745 if not (opt.smart_sync or opt.smart_tag):
746 self.OptionParser.error('-u and -p may only be combined with -s or -t')
747 if None in [opt.manifest_server_username, opt.manifest_server_password]:
748 self.OptionParser.error('both -u and -p must be given')
749
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700750 def Execute(self, opt, args):
Roy Lee18afd7f2010-05-09 04:32:08 +0800751 if opt.jobs:
752 self.jobs = opt.jobs
Shawn O. Pearce97d2b2f2011-09-22 17:23:41 -0700753 if self.jobs > 1:
754 soft_limit, _ = _rlimit_nofile()
Mike Frysinger0c0e9342019-06-13 12:42:39 -0400755 self.jobs = min(self.jobs, (soft_limit - 5) // 3)
Shawn O. Pearce97d2b2f2011-09-22 17:23:41 -0700756
Mike Frysinger521d01b2020-02-17 01:51:49 -0500757 opt.quiet = opt.output_mode is False
758 opt.verbose = opt.output_mode is True
759
Chris Wolfee9dc3b32012-01-26 11:36:18 -0500760 if opt.manifest_name:
761 self.manifest.Override(opt.manifest_name)
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700762
Chirayu Desaia892b102013-06-11 14:18:46 +0530763 manifest_name = opt.manifest_name
David Pursehouse59b41742015-05-07 14:36:09 +0900764 smart_sync_manifest_path = os.path.join(
David Pursehouseabdf7502020-02-12 14:58:39 +0900765 self.manifest.manifestProject.worktree, 'smart_sync_override.xml')
Chirayu Desaia892b102013-06-11 14:18:46 +0530766
Xin Lid79a4bc2020-05-20 16:03:45 -0700767 if opt.clone_bundle is None:
768 opt.clone_bundle = self.manifest.CloneBundle
769
Victor Boivie08c880d2011-04-19 10:32:52 +0200770 if opt.smart_sync or opt.smart_tag:
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400771 manifest_name = self._SmartSyncSetup(opt, smart_sync_manifest_path)
772 else:
David Pursehouse59b41742015-05-07 14:36:09 +0900773 if os.path.isfile(smart_sync_manifest_path):
774 try:
Renaud Paquay010fed72016-11-11 14:25:29 -0800775 platform_utils.remove(smart_sync_manifest_path)
David Pursehouse59b41742015-05-07 14:36:09 +0900776 except OSError as e:
777 print('error: failed to remove existing smart sync override manifest: %s' %
778 e, file=sys.stderr)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700779
Mike Frysinger5a033082019-09-23 19:21:20 -0400780 err_event = _threading.Event()
781
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700782 rp = self.manifest.repoProject
783 rp.PreSync()
Mike Frysinger23c900f2020-02-29 02:52:44 -0500784 cb = rp.CurrentBranch
785 if cb:
786 base = rp.GetBranch(cb).merge
787 if not base or not base.startswith('refs/heads/'):
788 print('warning: repo is not tracking a remote branch, so it will not '
Mike Frysinger58ac1672020-03-14 14:35:26 -0400789 'receive updates; run `repo init --repo-rev=stable` to fix.',
Mike Frysinger23c900f2020-02-29 02:52:44 -0500790 file=sys.stderr)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700791
792 mp = self.manifest.manifestProject
793 mp.PreSync()
794
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800795 if opt.repo_upgraded:
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -0700796 _PostRepoUpgrade(self.manifest, quiet=opt.quiet)
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800797
Fredrik de Grootcc960972019-11-22 09:04:31 +0100798 if not opt.mp_update:
799 print('Skipping update of local manifest project.')
800 else:
801 self._UpdateManifestProject(opt, mp, manifest_name)
Simran Basib9a1b732015-08-20 12:19:28 -0700802
Raman Tenneti2ae44d72021-03-23 15:12:27 -0700803 if self._UseSuperproject(opt):
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800804 manifest_name = self._UpdateProjectsRevisionId(opt, args)
805
Simran Basib9a1b732015-08-20 12:19:28 -0700806 if self.gitc_manifest:
807 gitc_manifest_projects = self.GetProjects(args,
Simran Basib9a1b732015-08-20 12:19:28 -0700808 missing_ok=True)
809 gitc_projects = []
810 opened_projects = []
811 for project in gitc_manifest_projects:
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700812 if project.relpath in self.gitc_manifest.paths and \
813 self.gitc_manifest.paths[project.relpath].old_revision:
814 opened_projects.append(project.relpath)
Simran Basib9a1b732015-08-20 12:19:28 -0700815 else:
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700816 gitc_projects.append(project.relpath)
Simran Basib9a1b732015-08-20 12:19:28 -0700817
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700818 if not args:
819 gitc_projects = None
820
821 if gitc_projects != [] and not opt.local_only:
Simran Basib9a1b732015-08-20 12:19:28 -0700822 print('Updating GITC client: %s' % self.gitc_manifest.gitc_client_name)
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700823 manifest = GitcManifest(self.repodir, self.gitc_manifest.gitc_client_name)
824 if manifest_name:
825 manifest.Override(manifest_name)
826 else:
827 manifest.Override(self.manifest.manifestFile)
828 gitc_utils.generate_gitc_manifest(self.gitc_manifest,
829 manifest,
Simran Basib9a1b732015-08-20 12:19:28 -0700830 gitc_projects)
831 print('GITC client successfully synced.')
832
833 # The opened projects need to be synced as normal, therefore we
834 # generate a new args list to represent the opened projects.
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700835 # TODO: make this more reliable -- if there's a project name/path overlap,
836 # this may choose the wrong project.
David Pursehouse3bcd3052017-07-10 22:42:22 +0900837 args = [os.path.relpath(self.manifest.paths[path].worktree, os.getcwd())
838 for path in opened_projects]
Simran Basib9a1b732015-08-20 12:19:28 -0700839 if not args:
840 return
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800841 all_projects = self.GetProjects(args,
842 missing_ok=True,
843 submodules_ok=opt.fetch_submodules)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700844
Mike Frysinger5a033082019-09-23 19:21:20 -0400845 err_network_sync = False
846 err_update_projects = False
Mike Frysinger5a033082019-09-23 19:21:20 -0400847
Dave Borowitz67700e92012-10-23 15:00:54 -0700848 self._fetch_times = _FetchTimes(self.manifest)
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700849 if not opt.local_only:
Shawn O. Pearcef6906872009-04-18 10:49:00 -0700850 to_fetch = []
851 now = time.time()
Dave Borowitz67700e92012-10-23 15:00:54 -0700852 if _ONE_DAY_S <= (now - rp.LastFetch):
Shawn O. Pearcef6906872009-04-18 10:49:00 -0700853 to_fetch.append(rp)
David Pursehouse8a68ff92012-09-24 12:15:13 +0900854 to_fetch.extend(all_projects)
Dave Borowitz67700e92012-10-23 15:00:54 -0700855 to_fetch.sort(key=self._fetch_times.Get, reverse=True)
Shawn O. Pearcef6906872009-04-18 10:49:00 -0700856
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500857 success, fetched = self._Fetch(to_fetch, opt, err_event)
858 if not success:
859 err_event.set()
Mike Frysinger5a033082019-09-23 19:21:20 -0400860
Mike Frysingerc58ec4d2020-02-17 14:36:08 -0500861 _PostRepoFetch(rp, opt.repo_verify)
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700862 if opt.network_only:
863 # bail out now; the rest touches the working tree
Mike Frysingerbe24a542021-02-23 03:24:12 -0500864 if err_event.is_set():
Mike Frysinger5a033082019-09-23 19:21:20 -0400865 print('\nerror: Exited sync due to fetch errors.\n', file=sys.stderr)
866 sys.exit(1)
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700867 return
868
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800869 # Iteratively fetch missing and/or nested unregistered submodules
870 previously_missing_set = set()
871 while True:
Victor Boivie53a6c5d2013-03-19 12:20:52 +0100872 self._ReloadManifest(manifest_name)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800873 all_projects = self.GetProjects(args,
874 missing_ok=True,
875 submodules_ok=opt.fetch_submodules)
876 missing = []
877 for project in all_projects:
878 if project.gitdir not in fetched:
879 missing.append(project)
880 if not missing:
881 break
882 # Stop us from non-stopped fetching actually-missing repos: If set of
883 # missing repos has not been changed from last fetch, we break.
884 missing_set = set(p.name for p in missing)
885 if previously_missing_set == missing_set:
886 break
887 previously_missing_set = missing_set
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500888 success, new_fetched = self._Fetch(to_fetch, opt, err_event)
889 if not success:
890 err_event.set()
891 fetched.update(new_fetched)
Mike Frysinger5a033082019-09-23 19:21:20 -0400892
893 # If we saw an error, exit with code 1 so that other scripts can check.
Mike Frysingerbe24a542021-02-23 03:24:12 -0500894 if err_event.is_set():
Mike Frysinger5a033082019-09-23 19:21:20 -0400895 err_network_sync = True
896 if opt.fail_fast:
897 print('\nerror: Exited sync due to fetch errors.\n'
898 'Local checkouts *not* updated. Resolve network issues & '
899 'retry.\n'
900 '`repo sync -l` will update some local checkouts.',
901 file=sys.stderr)
902 sys.exit(1)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800903
Julien Campergue335f5ef2013-10-16 11:02:35 +0200904 if self.manifest.IsMirror or self.manifest.IsArchive:
Shawn O. Pearcecd1d7ff2009-06-04 16:15:53 -0700905 # bail out now, we have no working tree
906 return
907
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500908 if self.UpdateProjectList(opt):
Mike Frysinger5a033082019-09-23 19:21:20 -0400909 err_event.set()
910 err_update_projects = True
911 if opt.fail_fast:
912 print('\nerror: Local checkouts *not* updated.', file=sys.stderr)
913 sys.exit(1)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700914
Mike Frysinger5a033082019-09-23 19:21:20 -0400915 err_results = []
Mike Frysingerebf04a42021-02-23 20:48:04 -0500916 # NB: We don't exit here because this is the last step.
917 err_checkout = not self._Checkout(all_projects, opt, err_results)
918 if err_checkout:
919 err_event.set()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700920
Doug Anderson2b8db3c2010-11-01 15:08:06 -0700921 # If there's a notice that's supposed to print at the end of the sync, print
922 # it now...
923 if self.manifest.notice:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700924 print(self.manifest.notice)
Doug Anderson2b8db3c2010-11-01 15:08:06 -0700925
Mike Frysinger5a033082019-09-23 19:21:20 -0400926 # If we saw an error, exit with code 1 so that other scripts can check.
Mike Frysingerbe24a542021-02-23 03:24:12 -0500927 if err_event.is_set():
Mike Frysinger5a033082019-09-23 19:21:20 -0400928 print('\nerror: Unable to fully sync the tree.', file=sys.stderr)
929 if err_network_sync:
930 print('error: Downloading network changes failed.', file=sys.stderr)
931 if err_update_projects:
932 print('error: Updating local project lists failed.', file=sys.stderr)
933 if err_checkout:
934 print('error: Checking out local projects failed.', file=sys.stderr)
935 if err_results:
936 print('Failing repos:\n%s' % '\n'.join(err_results), file=sys.stderr)
937 print('Try re-running with "-j1 --fail-fast" to exit at the first error.',
938 file=sys.stderr)
939 sys.exit(1)
940
Mike Frysingere19d9e12020-02-12 11:23:32 -0500941 if not opt.quiet:
942 print('repo sync has finished successfully.')
943
David Pursehouse819827a2020-02-12 15:20:19 +0900944
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -0700945def _PostRepoUpgrade(manifest, quiet=False):
Conley Owens094cdbe2014-01-30 15:09:59 -0800946 wrapper = Wrapper()
Conley Owensc9129d92012-10-01 16:12:28 -0700947 if wrapper.NeedSetupGnuPG():
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -0700948 wrapper.SetupGnuPG(quiet)
Conley Owensf2fe2d92014-01-29 13:53:43 -0800949 for project in manifest.projects:
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700950 if project.Exists:
951 project.PostRepoUpgrade()
952
David Pursehouse819827a2020-02-12 15:20:19 +0900953
Mike Frysingerc58ec4d2020-02-17 14:36:08 -0500954def _PostRepoFetch(rp, repo_verify=True, verbose=False):
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700955 if rp.HasChanges:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700956 print('info: A new version of repo is available', file=sys.stderr)
Mike Frysinger347f9ed2021-03-15 14:58:52 -0400957 wrapper = Wrapper()
958 try:
959 rev = rp.bare_git.describe(rp.GetRevisionId())
960 except GitError:
961 rev = None
962 _, new_rev = wrapper.check_repo_rev(rp.gitdir, rev, repo_verify=repo_verify)
963 # See if we're held back due to missing signed tag.
964 current_revid = rp.bare_git.rev_parse('HEAD')
965 new_revid = rp.bare_git.rev_parse('--verify', new_rev)
966 if current_revid != new_revid:
967 # We want to switch to the new rev, but also not trash any uncommitted
968 # changes. This helps with local testing/hacking.
969 # If a local change has been made, we will throw that away.
970 # We also have to make sure this will switch to an older commit if that's
971 # the latest tag in order to support release rollback.
972 try:
973 rp.work_git.reset('--keep', new_rev)
974 except GitError as e:
975 sys.exit(str(e))
Sarah Owenscecd1d82012-11-01 22:59:27 -0700976 print('info: Restarting repo with latest version', file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700977 raise RepoChangedException(['--repo-upgraded'])
978 else:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700979 print('warning: Skipped upgrade to unverified version', file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700980 else:
981 if verbose:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700982 print('repo version %s is current' % rp.work_git.describe(HEAD),
983 file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700984
David Pursehouse819827a2020-02-12 15:20:19 +0900985
Dave Borowitz67700e92012-10-23 15:00:54 -0700986class _FetchTimes(object):
Dave Borowitzd9478582012-10-23 16:35:39 -0700987 _ALPHA = 0.5
988
Dave Borowitz67700e92012-10-23 15:00:54 -0700989 def __init__(self, manifest):
Anthony King85b24ac2014-05-06 15:57:48 +0100990 self._path = os.path.join(manifest.repodir, '.repo_fetchtimes.json')
Dave Borowitz67700e92012-10-23 15:00:54 -0700991 self._times = None
Dave Borowitzd9478582012-10-23 16:35:39 -0700992 self._seen = set()
Dave Borowitz67700e92012-10-23 15:00:54 -0700993
994 def Get(self, project):
995 self._Load()
996 return self._times.get(project.name, _ONE_DAY_S)
997
998 def Set(self, project, t):
Dave Borowitzd9478582012-10-23 16:35:39 -0700999 self._Load()
1000 name = project.name
1001 old = self._times.get(name, t)
1002 self._seen.add(name)
1003 a = self._ALPHA
David Pursehouse54a4e602020-02-12 14:31:05 +09001004 self._times[name] = (a * t) + ((1 - a) * old)
Dave Borowitz67700e92012-10-23 15:00:54 -07001005
1006 def _Load(self):
1007 if self._times is None:
1008 try:
Mike Frysinger3164d402019-11-11 05:40:22 -05001009 with open(self._path) as f:
Anthony King85b24ac2014-05-06 15:57:48 +01001010 self._times = json.load(f)
Anthony King85b24ac2014-05-06 15:57:48 +01001011 except (IOError, ValueError):
1012 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08001013 platform_utils.remove(self._path)
Anthony King85b24ac2014-05-06 15:57:48 +01001014 except OSError:
1015 pass
1016 self._times = {}
Dave Borowitz67700e92012-10-23 15:00:54 -07001017
1018 def Save(self):
1019 if self._times is None:
1020 return
Dave Borowitzd9478582012-10-23 16:35:39 -07001021
1022 to_delete = []
1023 for name in self._times:
1024 if name not in self._seen:
1025 to_delete.append(name)
1026 for name in to_delete:
1027 del self._times[name]
1028
Dave Borowitz67700e92012-10-23 15:00:54 -07001029 try:
Mike Frysinger3164d402019-11-11 05:40:22 -05001030 with open(self._path, 'w') as f:
Anthony King85b24ac2014-05-06 15:57:48 +01001031 json.dump(self._times, f, indent=2)
Anthony King85b24ac2014-05-06 15:57:48 +01001032 except (IOError, TypeError):
1033 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08001034 platform_utils.remove(self._path)
Anthony King85b24ac2014-05-06 15:57:48 +01001035 except OSError:
1036 pass
Dan Willemsen0745bb22015-08-17 13:41:45 -07001037
1038# This is a replacement for xmlrpc.client.Transport using urllib2
1039# and supporting persistent-http[s]. It cannot change hosts from
1040# request to request like the normal transport, the real url
1041# is passed during initialization.
David Pursehouse819827a2020-02-12 15:20:19 +09001042
1043
Dan Willemsen0745bb22015-08-17 13:41:45 -07001044class PersistentTransport(xmlrpc.client.Transport):
1045 def __init__(self, orig_host):
1046 self.orig_host = orig_host
1047
1048 def request(self, host, handler, request_body, verbose=False):
1049 with GetUrlCookieFile(self.orig_host, not verbose) as (cookiefile, proxy):
1050 # Python doesn't understand cookies with the #HttpOnly_ prefix
1051 # Since we're only using them for HTTP, copy the file temporarily,
1052 # stripping those prefixes away.
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001053 if cookiefile:
Collin Fijalkoviche1191b32020-02-18 10:57:32 -08001054 tmpcookiefile = tempfile.NamedTemporaryFile(mode='w')
David Pursehouse4c5f74e2015-10-02 11:10:10 +09001055 tmpcookiefile.write("# HTTP Cookie File")
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001056 try:
1057 with open(cookiefile) as f:
1058 for line in f:
1059 if line.startswith("#HttpOnly_"):
1060 line = line[len("#HttpOnly_"):]
1061 tmpcookiefile.write(line)
1062 tmpcookiefile.flush()
Dan Willemsen0745bb22015-08-17 13:41:45 -07001063
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001064 cookiejar = cookielib.MozillaCookieJar(tmpcookiefile.name)
David Pursehouseb1ad2192015-09-30 10:35:43 +09001065 try:
1066 cookiejar.load()
1067 except cookielib.LoadError:
1068 cookiejar = cookielib.CookieJar()
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001069 finally:
1070 tmpcookiefile.close()
1071 else:
1072 cookiejar = cookielib.CookieJar()
Dan Willemsen0745bb22015-08-17 13:41:45 -07001073
1074 proxyhandler = urllib.request.ProxyHandler
1075 if proxy:
1076 proxyhandler = urllib.request.ProxyHandler({
1077 "http": proxy,
David Pursehouse54a4e602020-02-12 14:31:05 +09001078 "https": proxy})
Dan Willemsen0745bb22015-08-17 13:41:45 -07001079
1080 opener = urllib.request.build_opener(
1081 urllib.request.HTTPCookieProcessor(cookiejar),
1082 proxyhandler)
1083
1084 url = urllib.parse.urljoin(self.orig_host, handler)
1085 parse_results = urllib.parse.urlparse(url)
1086
1087 scheme = parse_results.scheme
1088 if scheme == 'persistent-http':
1089 scheme = 'http'
1090 if scheme == 'persistent-https':
1091 # If we're proxying through persistent-https, use http. The
1092 # proxy itself will do the https.
1093 if proxy:
1094 scheme = 'http'
1095 else:
1096 scheme = 'https'
1097
1098 # Parse out any authentication information using the base class
1099 host, extra_headers, _ = self.get_host_info(parse_results.netloc)
1100
1101 url = urllib.parse.urlunparse((
1102 scheme,
1103 host,
1104 parse_results.path,
1105 parse_results.params,
1106 parse_results.query,
1107 parse_results.fragment))
1108
1109 request = urllib.request.Request(url, request_body)
1110 if extra_headers is not None:
1111 for (name, header) in extra_headers:
1112 request.add_header(name, header)
1113 request.add_header('Content-Type', 'text/xml')
1114 try:
1115 response = opener.open(request)
1116 except urllib.error.HTTPError as e:
1117 if e.code == 501:
1118 # We may have been redirected through a login process
1119 # but our POST turned into a GET. Retry.
1120 response = opener.open(request)
1121 else:
1122 raise
1123
1124 p, u = xmlrpc.client.getparser()
1125 while 1:
1126 data = response.read(1024)
1127 if not data:
1128 break
1129 p.feed(data)
1130 p.close()
1131 return u.close()
1132
1133 def close(self):
1134 pass