blob: 4763fadca77fbd3217580d4097019682ee783987 [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 Frysingerb5d075d2021-03-01 00:56:38 -050054from command import Command, MirrorSafeCommand
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 [...]
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
Mike Frysinger9180a072021-04-13 14:57:40 -0400170 def _CommonOptions(self, p):
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
Mike Frysinger9180a072021-04-13 14:57:40 -0400175 super()._CommonOptions(p)
Shawn O. Pearce6392c872011-09-22 17:44:31 -0700176
Mike Frysinger9180a072021-04-13 14:57:40 -0400177 def _Options(self, p, show_smart=True):
Mike Frysinger49de8ef2021-04-09 00:21:02 -0400178 p.add_option('--jobs-network', default=None, type=int, metavar='JOBS',
179 help='number of network jobs to run in parallel (defaults to --jobs)')
180 p.add_option('--jobs-checkout', default=None, type=int, metavar='JOBS',
181 help='number of local checkout jobs to run in parallel (defaults to --jobs)')
182
Andrei Warkentin5df6de02010-07-02 17:58:31 -0500183 p.add_option('-f', '--force-broken',
Stefan Müller-Klieser46702ed2019-08-30 10:20:15 +0200184 dest='force_broken', action='store_true',
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400185 help='obsolete option (to be deleted in the future)')
186 p.add_option('--fail-fast',
187 dest='fail_fast', action='store_true',
188 help='stop syncing after first error is hit')
Kevin Degiabaa7f32014-11-12 11:27:45 -0700189 p.add_option('--force-sync',
190 dest='force_sync', action='store_true',
191 help="overwrite an existing git directory if it needs to "
192 "point to a different object directory. WARNING: this "
193 "may cause loss of data")
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500194 p.add_option('--force-remove-dirty',
195 dest='force_remove_dirty', action='store_true',
196 help="force remove projects with uncommitted modifications if "
197 "projects no longer exist in the manifest. "
198 "WARNING: this may cause loss of data")
David Pursehouse8f62fb72012-11-14 12:09:38 +0900199 p.add_option('-l', '--local-only',
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700200 dest='local_only', action='store_true',
201 help="only update working tree, don't fetch")
David Pursehouse54a4e602020-02-12 14:31:05 +0900202 p.add_option('--no-manifest-update', '--nmu',
Fredrik de Grootcc960972019-11-22 09:04:31 +0100203 dest='mp_update', action='store_false', default='true',
204 help='use the existing manifest checkout as-is. '
205 '(do not update to the latest revision)')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900206 p.add_option('-n', '--network-only',
Shawn O. Pearce96fdcef2009-04-10 16:29:20 -0700207 dest='network_only', action='store_true',
208 help="fetch only, don't update working tree")
David Pursehouse8f62fb72012-11-14 12:09:38 +0900209 p.add_option('-d', '--detach',
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700210 dest='detach_head', action='store_true',
211 help='detach projects back to manifest revision')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900212 p.add_option('-c', '--current-branch',
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -0700213 dest='current_branch_only', action='store_true',
214 help='fetch only current branch from server')
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,
Raman Tennetif32f2432021-04-12 20:57:25 -0700346 clone_filter=self.manifest.CloneFilter,
347 partial_clone_exclude=self.manifest.PartialCloneExclude)
Doug Andersonfc06ced2011-03-16 15:49:18 -0700348
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500349 output = buf.getvalue()
350 if opt.verbose and output:
351 print('\n' + output.rstrip())
Doug Andersonfc06ced2011-03-16 15:49:18 -0700352
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500353 if not success:
354 print('error: Cannot fetch %s from %s'
355 % (project.name, project.remote.url),
356 file=sys.stderr)
357 except Exception as e:
358 print('error: Cannot fetch %s (%s: %s)'
359 % (project.name, type(e).__name__, str(e)), file=sys.stderr)
360 raise
Mike Frysinger7b586f22021-02-23 18:38:39 -0500361
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500362 finish = time.time()
363 return (success, project, start, finish)
David James8d201162013-10-11 17:03:19 -0700364
Mike Frysinger5a033082019-09-23 19:21:20 -0400365 def _Fetch(self, projects, opt, err_event):
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500366 ret = True
367
Mike Frysinger49de8ef2021-04-09 00:21:02 -0400368 jobs = opt.jobs_network if opt.jobs_network else self.jobs
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700369 fetched = set()
Mike Frysinger151701e2021-04-13 15:07:21 -0400370 pm = Progress('Fetching', len(projects), delay=False, quiet=opt.quiet)
Roy Lee18afd7f2010-05-09 04:32:08 +0800371
David James89ece422014-01-09 18:51:58 -0800372 objdir_project_map = dict()
373 for project in projects:
374 objdir_project_map.setdefault(project.objdir, []).append(project)
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500375 projects_list = list(objdir_project_map.values())
David James8d201162013-10-11 17:03:19 -0700376
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500377 def _ProcessResults(results_sets):
378 ret = True
379 for results in results_sets:
380 for (success, project, start, finish) in results:
381 self._fetch_times.Set(project, finish - start)
382 self.event_log.AddSync(project, event_log.TASK_SYNC_NETWORK,
383 start, finish, success)
384 # Check for any errors before running any more tasks.
385 # ...we'll let existing jobs finish, though.
386 if not success:
387 ret = False
388 else:
389 fetched.add(project.gitdir)
390 pm.update(msg=project.name)
391 if not ret and opt.fail_fast:
392 break
393 return ret
Doug Andersonfc06ced2011-03-16 15:49:18 -0700394
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500395 # NB: Multiprocessing is heavy, so don't spin it up for one job.
Mike Frysinger49de8ef2021-04-09 00:21:02 -0400396 if len(projects_list) == 1 or jobs == 1:
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500397 if not _ProcessResults(self._FetchProjectList(opt, x) for x in projects_list):
398 ret = False
399 else:
400 # Favor throughput over responsiveness when quiet. It seems that imap()
401 # will yield results in batches relative to chunksize, so even as the
402 # children finish a sync, we won't see the result until one child finishes
403 # ~chunksize jobs. When using a large --jobs with large chunksize, this
404 # can be jarring as there will be a large initial delay where repo looks
405 # like it isn't doing anything and sits at 0%, but then suddenly completes
406 # a lot of jobs all at once. Since this code is more network bound, we
407 # can accept a bit more CPU overhead with a smaller chunksize so that the
408 # user sees more immediate & continuous feedback.
409 if opt.quiet:
410 chunksize = WORKER_BATCH_SIZE
David James89ece422014-01-09 18:51:58 -0800411 else:
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500412 pm.update(inc=0, msg='warming up')
413 chunksize = 4
Mike Frysinger49de8ef2021-04-09 00:21:02 -0400414 with multiprocessing.Pool(jobs) as pool:
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500415 results = pool.imap_unordered(
416 functools.partial(self._FetchProjectList, opt),
417 projects_list,
418 chunksize=chunksize)
419 if not _ProcessResults(results):
420 ret = False
421 pool.close()
Roy Lee18afd7f2010-05-09 04:32:08 +0800422
Shawn O. Pearce68194f42009-04-10 16:48:52 -0700423 pm.end()
Dave Borowitz67700e92012-10-23 15:00:54 -0700424 self._fetch_times.Save()
Dave Borowitz18857212012-10-23 17:02:59 -0700425
Julien Campergue335f5ef2013-10-16 11:02:35 +0200426 if not self.manifest.IsArchive:
Mike Frysinger5a033082019-09-23 19:21:20 -0400427 self._GCProjects(projects, opt, err_event)
Julien Campergue335f5ef2013-10-16 11:02:35 +0200428
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500429 return (ret, fetched)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700430
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500431 def _CheckoutOne(self, detach_head, force_sync, project):
Xin Li745be2e2019-06-03 11:24:30 -0700432 """Checkout work tree for one project
433
434 Args:
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500435 detach_head: Whether to leave a detached HEAD.
436 force_sync: Force checking out of the repo.
Xin Li745be2e2019-06-03 11:24:30 -0700437 project: Project object for the project to checkout.
Xin Li745be2e2019-06-03 11:24:30 -0700438
439 Returns:
440 Whether the fetch was successful.
441 """
Xin Li745be2e2019-06-03 11:24:30 -0700442 start = time.time()
443 syncbuf = SyncBuffer(self.manifest.manifestProject.config,
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500444 detach_head=detach_head)
Xin Li745be2e2019-06-03 11:24:30 -0700445 success = False
446 try:
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500447 project.Sync_LocalHalf(syncbuf, force_sync=force_sync)
Mike Frysingerebf04a42021-02-23 20:48:04 -0500448 success = syncbuf.Finish()
449 except Exception as e:
450 print('error: Cannot checkout %s: %s: %s' %
451 (project.name, type(e).__name__, str(e)),
452 file=sys.stderr)
453 raise
Xin Li745be2e2019-06-03 11:24:30 -0700454
Mike Frysingerebf04a42021-02-23 20:48:04 -0500455 if not success:
456 print('error: Cannot checkout %s' % (project.name), file=sys.stderr)
457 finish = time.time()
458 return (success, project, start, finish)
Xin Li745be2e2019-06-03 11:24:30 -0700459
Mike Frysingerebf04a42021-02-23 20:48:04 -0500460 def _Checkout(self, all_projects, opt, err_results):
Xin Li745be2e2019-06-03 11:24:30 -0700461 """Checkout projects listed in all_projects
462
463 Args:
464 all_projects: List of all projects that should be checked out.
465 opt: Program options returned from optparse. See _Options().
Mike Frysingerebf04a42021-02-23 20:48:04 -0500466 err_results: A list of strings, paths to git repos where checkout failed.
Xin Li745be2e2019-06-03 11:24:30 -0700467 """
Mike Frysingerebf04a42021-02-23 20:48:04 -0500468 # Only checkout projects with worktrees.
469 all_projects = [x for x in all_projects if x.worktree]
Xin Li745be2e2019-06-03 11:24:30 -0700470
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500471 def _ProcessResults(pool, pm, results):
472 ret = True
Mike Frysingerebf04a42021-02-23 20:48:04 -0500473 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:
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500479 ret = False
Mike Frysingerebf04a42021-02-23 20:48:04 -0500480 err_results.append(project.relpath)
481 if opt.fail_fast:
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500482 if pool:
483 pool.close()
484 return ret
Mike Frysingerebf04a42021-02-23 20:48:04 -0500485 pm.update(msg=project.name)
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500486 return ret
Xin Li745be2e2019-06-03 11:24:30 -0700487
Mike Frysingerb5d075d2021-03-01 00:56:38 -0500488 return self.ExecuteInParallel(
489 opt.jobs_checkout if opt.jobs_checkout else self.jobs,
490 functools.partial(self._CheckoutOne, opt.detach_head, opt.force_sync),
491 all_projects,
492 callback=_ProcessResults,
493 output=Progress('Checking out', len(all_projects), quiet=opt.quiet)) and not err_results
Mike Frysingerebf04a42021-02-23 20:48:04 -0500494
Mike Frysinger5a033082019-09-23 19:21:20 -0400495 def _GCProjects(self, projects, opt, err_event):
Mike Frysinger151701e2021-04-13 15:07:21 -0400496 pm = Progress('Garbage collecting', len(projects), delay=False, quiet=opt.quiet)
Mike Frysinger65af2602021-04-08 22:47:44 -0400497 pm.update(inc=0, msg='prescan')
498
Gabe Black2ff30292014-10-09 17:54:35 -0700499 gc_gitdirs = {}
David James8d201162013-10-11 17:03:19 -0700500 for project in projects:
Mike Frysinger5f2b0452020-02-11 03:23:24 -0500501 # Make sure pruning never kicks in with shared projects.
Mike Frysinger979d5bd2020-02-09 02:28:34 -0500502 if (not project.use_git_worktrees and
David Pursehouseaa611a22020-02-20 10:47:26 +0900503 len(project.manifest.GetProjectsWithName(project.name)) > 1):
Anders Björklund2a2da802021-01-18 10:32:36 +0100504 if not opt.quiet:
Mike Frysinger65af2602021-04-08 22:47:44 -0400505 print('\r%s: Shared project %s found, disabling pruning.' %
Anders Björklund2a2da802021-01-18 10:32:36 +0100506 (project.relpath, project.name))
Mike Frysinger5f2b0452020-02-11 03:23:24 -0500507 if git_require((2, 7, 0)):
Mike Frysingerf81c72e2020-02-19 15:50:00 -0500508 project.EnableRepositoryExtension('preciousObjects')
Mike Frysinger5f2b0452020-02-11 03:23:24 -0500509 else:
510 # This isn't perfect, but it's the best we can do with old git.
Mike Frysinger65af2602021-04-08 22:47:44 -0400511 print('\r%s: WARNING: shared projects are unreliable when using old '
Mike Frysinger5f2b0452020-02-11 03:23:24 -0500512 'versions of git; please upgrade to git-2.7.0+.'
513 % (project.relpath,),
514 file=sys.stderr)
515 project.config.SetString('gc.pruneExpire', 'never')
Gabe Black2ff30292014-10-09 17:54:35 -0700516 gc_gitdirs[project.gitdir] = project.bare_git
David James8d201162013-10-11 17:03:19 -0700517
Mike Frysinger65af2602021-04-08 22:47:44 -0400518 pm.update(inc=len(projects) - len(gc_gitdirs), msg='warming up')
519
520 cpu_count = os.cpu_count()
Dave Borowitz18857212012-10-23 17:02:59 -0700521 jobs = min(self.jobs, cpu_count)
522
523 if jobs < 2:
Gabe Black2ff30292014-10-09 17:54:35 -0700524 for bare_git in gc_gitdirs.values():
Mike Frysinger65af2602021-04-08 22:47:44 -0400525 pm.update(msg=bare_git._project.name)
David James8d201162013-10-11 17:03:19 -0700526 bare_git.gc('--auto')
Mike Frysinger65af2602021-04-08 22:47:44 -0400527 pm.end()
Dave Borowitz18857212012-10-23 17:02:59 -0700528 return
529
Mike Frysinger0c0e9342019-06-13 12:42:39 -0400530 config = {'pack.threads': cpu_count // jobs if cpu_count > jobs else 1}
Dave Borowitz18857212012-10-23 17:02:59 -0700531
532 threads = set()
533 sem = _threading.Semaphore(jobs)
Dave Borowitz18857212012-10-23 17:02:59 -0700534
David James8d201162013-10-11 17:03:19 -0700535 def GC(bare_git):
Mike Frysinger65af2602021-04-08 22:47:44 -0400536 pm.start(bare_git._project.name)
Dave Borowitz18857212012-10-23 17:02:59 -0700537 try:
538 try:
David James8d201162013-10-11 17:03:19 -0700539 bare_git.gc('--auto', config=config)
Dave Borowitz18857212012-10-23 17:02:59 -0700540 except GitError:
541 err_event.set()
David Pursehouse145e35b2020-02-12 15:40:47 +0900542 except Exception:
Dave Borowitz18857212012-10-23 17:02:59 -0700543 err_event.set()
544 raise
545 finally:
Mike Frysinger65af2602021-04-08 22:47:44 -0400546 pm.finish(bare_git._project.name)
Dave Borowitz18857212012-10-23 17:02:59 -0700547 sem.release()
548
Gabe Black2ff30292014-10-09 17:54:35 -0700549 for bare_git in gc_gitdirs.values():
Mike Frysingerbe24a542021-02-23 03:24:12 -0500550 if err_event.is_set() and opt.fail_fast:
Dave Borowitz18857212012-10-23 17:02:59 -0700551 break
552 sem.acquire()
David James8d201162013-10-11 17:03:19 -0700553 t = _threading.Thread(target=GC, args=(bare_git,))
Dave Borowitz18857212012-10-23 17:02:59 -0700554 t.daemon = True
555 threads.add(t)
556 t.start()
557
558 for t in threads:
559 t.join()
Mike Frysinger65af2602021-04-08 22:47:44 -0400560 pm.end()
Dave Borowitz18857212012-10-23 17:02:59 -0700561
Tim Kilbourn07669002013-03-08 15:02:49 -0800562 def _ReloadManifest(self, manifest_name=None):
563 if manifest_name:
564 # Override calls _Unload already
565 self.manifest.Override(manifest_name)
566 else:
567 self.manifest._Unload()
568
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500569 def UpdateProjectList(self, opt):
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700570 new_project_paths = []
Colin Cross5acde752012-03-28 20:15:45 -0700571 for project in self.GetProjects(None, missing_ok=True):
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700572 if project.relpath:
573 new_project_paths.append(project.relpath)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700574 file_name = 'project.list'
Mike Frysingere3315bb2021-02-09 23:45:28 -0500575 file_path = os.path.join(self.repodir, file_name)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700576 old_project_paths = []
577
578 if os.path.exists(file_path):
Mike Frysinger3164d402019-11-11 05:40:22 -0500579 with open(file_path, 'r') as fd:
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700580 old_project_paths = fd.read().split('\n')
Kuang-che Wu0d9b16d2019-04-06 00:49:47 +0800581 # In reversed order, so subfolders are deleted before parent folder.
582 for path in sorted(old_project_paths, reverse=True):
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700583 if not path:
584 continue
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700585 if path not in new_project_paths:
David Pursehouse8a68ff92012-09-24 12:15:13 +0900586 # If the path has already been deleted, we don't need to do it
Dan Willemsen43507912016-09-01 16:26:02 -0700587 gitdir = os.path.join(self.manifest.topdir, path, '.git')
588 if os.path.exists(gitdir):
David Pursehousec1b86a22012-11-14 11:36:51 +0900589 project = Project(
David Pursehouseabdf7502020-02-12 14:58:39 +0900590 manifest=self.manifest,
591 name=path,
592 remote=RemoteSpec('origin'),
593 gitdir=gitdir,
594 objdir=gitdir,
Mike Frysingerc0d18662020-02-19 19:19:18 -0500595 use_git_worktrees=os.path.isfile(gitdir),
David Pursehouseabdf7502020-02-12 14:58:39 +0900596 worktree=os.path.join(self.manifest.topdir, path),
597 relpath=path,
598 revisionExpr='HEAD',
599 revisionId=None,
600 groups=None)
Mike Frysingerc0d18662020-02-19 19:19:18 -0500601 if not project.DeleteWorktree(
David Pursehouseaa611a22020-02-20 10:47:26 +0900602 quiet=opt.quiet,
603 force=opt.force_remove_dirty):
Mike Frysingera850ca22019-08-07 17:19:24 -0400604 return 1
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700605
Shawn O. Pearce9fb29ce2009-06-04 20:41:02 -0700606 new_project_paths.sort()
Mike Frysinger3164d402019-11-11 05:40:22 -0500607 with open(file_path, 'w') as fd:
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700608 fd.write('\n'.join(new_project_paths))
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700609 fd.write('\n')
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700610 return 0
611
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400612 def _SmartSyncSetup(self, opt, smart_sync_manifest_path):
613 if not self.manifest.manifest_server:
614 print('error: cannot smart sync: no manifest server defined in '
615 'manifest', file=sys.stderr)
616 sys.exit(1)
617
618 manifest_server = self.manifest.manifest_server
619 if not opt.quiet:
620 print('Using manifest server %s' % manifest_server)
621
David Pursehouseeeff3532020-02-12 11:24:10 +0900622 if '@' not in manifest_server:
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400623 username = None
624 password = None
625 if opt.manifest_server_username and opt.manifest_server_password:
626 username = opt.manifest_server_username
627 password = opt.manifest_server_password
628 else:
629 try:
630 info = netrc.netrc()
631 except IOError:
632 # .netrc file does not exist or could not be opened
633 pass
634 else:
635 try:
636 parse_result = urllib.parse.urlparse(manifest_server)
637 if parse_result.hostname:
638 auth = info.authenticators(parse_result.hostname)
639 if auth:
640 username, _account, password = auth
641 else:
642 print('No credentials found for %s in .netrc'
643 % parse_result.hostname, file=sys.stderr)
644 except netrc.NetrcParseError as e:
645 print('Error parsing .netrc file: %s' % e, file=sys.stderr)
646
647 if (username and password):
648 manifest_server = manifest_server.replace('://', '://%s:%s@' %
649 (username, password),
650 1)
651
652 transport = PersistentTransport(manifest_server)
653 if manifest_server.startswith('persistent-'):
654 manifest_server = manifest_server[len('persistent-'):]
655
656 try:
657 server = xmlrpc.client.Server(manifest_server, transport=transport)
658 if opt.smart_sync:
Raman Tenneti8d43dea2021-02-07 16:30:27 -0800659 branch = self._GetBranch()
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400660
Mike Frysinger56ce3462019-12-04 19:30:48 -0500661 if 'SYNC_TARGET' in os.environ:
Mike Frysingere20da3e2020-03-07 01:53:53 -0500662 target = os.environ['SYNC_TARGET']
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400663 [success, manifest_str] = server.GetApprovedManifest(branch, target)
Mike Frysinger56ce3462019-12-04 19:30:48 -0500664 elif ('TARGET_PRODUCT' in os.environ and
665 'TARGET_BUILD_VARIANT' in os.environ):
Mike Frysingere20da3e2020-03-07 01:53:53 -0500666 target = '%s-%s' % (os.environ['TARGET_PRODUCT'],
667 os.environ['TARGET_BUILD_VARIANT'])
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400668 [success, manifest_str] = server.GetApprovedManifest(branch, target)
669 else:
670 [success, manifest_str] = server.GetApprovedManifest(branch)
671 else:
672 assert(opt.smart_tag)
673 [success, manifest_str] = server.GetManifest(opt.smart_tag)
674
675 if success:
676 manifest_name = os.path.basename(smart_sync_manifest_path)
677 try:
Mike Frysinger3164d402019-11-11 05:40:22 -0500678 with open(smart_sync_manifest_path, 'w') as f:
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400679 f.write(manifest_str)
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400680 except IOError as e:
681 print('error: cannot write manifest to %s:\n%s'
682 % (smart_sync_manifest_path, e),
683 file=sys.stderr)
684 sys.exit(1)
685 self._ReloadManifest(manifest_name)
686 else:
687 print('error: manifest server RPC call failed: %s' %
688 manifest_str, file=sys.stderr)
689 sys.exit(1)
690 except (socket.error, IOError, xmlrpc.client.Fault) as e:
691 print('error: cannot connect to manifest server %s:\n%s'
692 % (self.manifest.manifest_server, e), file=sys.stderr)
693 sys.exit(1)
694 except xmlrpc.client.ProtocolError as e:
695 print('error: cannot connect to manifest server %s:\n%d %s'
696 % (self.manifest.manifest_server, e.errcode, e.errmsg),
697 file=sys.stderr)
698 sys.exit(1)
699
700 return manifest_name
701
Mike Frysingerfb527e32019-08-27 02:34:32 -0400702 def _UpdateManifestProject(self, opt, mp, manifest_name):
703 """Fetch & update the local manifest project."""
704 if not opt.local_only:
705 start = time.time()
Mike Frysinger521d01b2020-02-17 01:51:49 -0500706 success = mp.Sync_NetworkHalf(quiet=opt.quiet, verbose=opt.verbose,
Raman Tenneti2ae44d72021-03-23 15:12:27 -0700707 current_branch_only=self._GetCurrentBranchOnly(opt),
Erwan Yvindc5c4d12019-06-18 13:49:12 +0200708 force_sync=opt.force_sync,
Mike Frysingerc58ec4d2020-02-17 14:36:08 -0500709 tags=opt.tags,
Mike Frysingerfb527e32019-08-27 02:34:32 -0400710 optimized_fetch=opt.optimized_fetch,
George Engelbrecht9bc283e2020-04-02 12:36:09 -0600711 retry_fetches=opt.retry_fetches,
Mike Frysingerfb527e32019-08-27 02:34:32 -0400712 submodules=self.manifest.HasSubmodules,
Raman Tennetif32f2432021-04-12 20:57:25 -0700713 clone_filter=self.manifest.CloneFilter,
714 partial_clone_exclude=self.manifest.PartialCloneExclude)
Mike Frysingerfb527e32019-08-27 02:34:32 -0400715 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
Chris Wolfee9dc3b32012-01-26 11:36:18 -0500757 if opt.manifest_name:
758 self.manifest.Override(opt.manifest_name)
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700759
Chirayu Desaia892b102013-06-11 14:18:46 +0530760 manifest_name = opt.manifest_name
David Pursehouse59b41742015-05-07 14:36:09 +0900761 smart_sync_manifest_path = os.path.join(
David Pursehouseabdf7502020-02-12 14:58:39 +0900762 self.manifest.manifestProject.worktree, 'smart_sync_override.xml')
Chirayu Desaia892b102013-06-11 14:18:46 +0530763
Xin Lid79a4bc2020-05-20 16:03:45 -0700764 if opt.clone_bundle is None:
765 opt.clone_bundle = self.manifest.CloneBundle
766
Victor Boivie08c880d2011-04-19 10:32:52 +0200767 if opt.smart_sync or opt.smart_tag:
Mike Frysinger01d6c3c2019-08-27 01:56:43 -0400768 manifest_name = self._SmartSyncSetup(opt, smart_sync_manifest_path)
769 else:
David Pursehouse59b41742015-05-07 14:36:09 +0900770 if os.path.isfile(smart_sync_manifest_path):
771 try:
Renaud Paquay010fed72016-11-11 14:25:29 -0800772 platform_utils.remove(smart_sync_manifest_path)
David Pursehouse59b41742015-05-07 14:36:09 +0900773 except OSError as e:
774 print('error: failed to remove existing smart sync override manifest: %s' %
775 e, file=sys.stderr)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700776
Mike Frysinger5a033082019-09-23 19:21:20 -0400777 err_event = _threading.Event()
778
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700779 rp = self.manifest.repoProject
780 rp.PreSync()
Mike Frysinger23c900f2020-02-29 02:52:44 -0500781 cb = rp.CurrentBranch
782 if cb:
783 base = rp.GetBranch(cb).merge
784 if not base or not base.startswith('refs/heads/'):
785 print('warning: repo is not tracking a remote branch, so it will not '
Mike Frysinger58ac1672020-03-14 14:35:26 -0400786 'receive updates; run `repo init --repo-rev=stable` to fix.',
Mike Frysinger23c900f2020-02-29 02:52:44 -0500787 file=sys.stderr)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700788
789 mp = self.manifest.manifestProject
790 mp.PreSync()
791
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800792 if opt.repo_upgraded:
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -0700793 _PostRepoUpgrade(self.manifest, quiet=opt.quiet)
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800794
Fredrik de Grootcc960972019-11-22 09:04:31 +0100795 if not opt.mp_update:
796 print('Skipping update of local manifest project.')
797 else:
798 self._UpdateManifestProject(opt, mp, manifest_name)
Simran Basib9a1b732015-08-20 12:19:28 -0700799
Raman Tenneti2ae44d72021-03-23 15:12:27 -0700800 if self._UseSuperproject(opt):
Raman Tenneti1fd7bc22021-02-04 14:39:38 -0800801 manifest_name = self._UpdateProjectsRevisionId(opt, args)
802
Simran Basib9a1b732015-08-20 12:19:28 -0700803 if self.gitc_manifest:
804 gitc_manifest_projects = self.GetProjects(args,
Simran Basib9a1b732015-08-20 12:19:28 -0700805 missing_ok=True)
806 gitc_projects = []
807 opened_projects = []
808 for project in gitc_manifest_projects:
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700809 if project.relpath in self.gitc_manifest.paths and \
810 self.gitc_manifest.paths[project.relpath].old_revision:
811 opened_projects.append(project.relpath)
Simran Basib9a1b732015-08-20 12:19:28 -0700812 else:
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700813 gitc_projects.append(project.relpath)
Simran Basib9a1b732015-08-20 12:19:28 -0700814
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700815 if not args:
816 gitc_projects = None
817
818 if gitc_projects != [] and not opt.local_only:
Simran Basib9a1b732015-08-20 12:19:28 -0700819 print('Updating GITC client: %s' % self.gitc_manifest.gitc_client_name)
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700820 manifest = GitcManifest(self.repodir, self.gitc_manifest.gitc_client_name)
821 if manifest_name:
822 manifest.Override(manifest_name)
823 else:
824 manifest.Override(self.manifest.manifestFile)
825 gitc_utils.generate_gitc_manifest(self.gitc_manifest,
826 manifest,
Simran Basib9a1b732015-08-20 12:19:28 -0700827 gitc_projects)
828 print('GITC client successfully synced.')
829
830 # The opened projects need to be synced as normal, therefore we
831 # generate a new args list to represent the opened projects.
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700832 # TODO: make this more reliable -- if there's a project name/path overlap,
833 # this may choose the wrong project.
David Pursehouse3bcd3052017-07-10 22:42:22 +0900834 args = [os.path.relpath(self.manifest.paths[path].worktree, os.getcwd())
835 for path in opened_projects]
Simran Basib9a1b732015-08-20 12:19:28 -0700836 if not args:
837 return
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800838 all_projects = self.GetProjects(args,
839 missing_ok=True,
840 submodules_ok=opt.fetch_submodules)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700841
Mike Frysinger5a033082019-09-23 19:21:20 -0400842 err_network_sync = False
843 err_update_projects = False
Mike Frysinger5a033082019-09-23 19:21:20 -0400844
Dave Borowitz67700e92012-10-23 15:00:54 -0700845 self._fetch_times = _FetchTimes(self.manifest)
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700846 if not opt.local_only:
Shawn O. Pearcef6906872009-04-18 10:49:00 -0700847 to_fetch = []
848 now = time.time()
Dave Borowitz67700e92012-10-23 15:00:54 -0700849 if _ONE_DAY_S <= (now - rp.LastFetch):
Shawn O. Pearcef6906872009-04-18 10:49:00 -0700850 to_fetch.append(rp)
David Pursehouse8a68ff92012-09-24 12:15:13 +0900851 to_fetch.extend(all_projects)
Dave Borowitz67700e92012-10-23 15:00:54 -0700852 to_fetch.sort(key=self._fetch_times.Get, reverse=True)
Shawn O. Pearcef6906872009-04-18 10:49:00 -0700853
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500854 success, fetched = self._Fetch(to_fetch, opt, err_event)
855 if not success:
856 err_event.set()
Mike Frysinger5a033082019-09-23 19:21:20 -0400857
Mike Frysingerc58ec4d2020-02-17 14:36:08 -0500858 _PostRepoFetch(rp, opt.repo_verify)
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700859 if opt.network_only:
860 # bail out now; the rest touches the working tree
Mike Frysingerbe24a542021-02-23 03:24:12 -0500861 if err_event.is_set():
Mike Frysinger5a033082019-09-23 19:21:20 -0400862 print('\nerror: Exited sync due to fetch errors.\n', file=sys.stderr)
863 sys.exit(1)
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700864 return
865
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800866 # Iteratively fetch missing and/or nested unregistered submodules
867 previously_missing_set = set()
868 while True:
Victor Boivie53a6c5d2013-03-19 12:20:52 +0100869 self._ReloadManifest(manifest_name)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800870 all_projects = self.GetProjects(args,
871 missing_ok=True,
872 submodules_ok=opt.fetch_submodules)
873 missing = []
874 for project in all_projects:
875 if project.gitdir not in fetched:
876 missing.append(project)
877 if not missing:
878 break
879 # Stop us from non-stopped fetching actually-missing repos: If set of
880 # missing repos has not been changed from last fetch, we break.
881 missing_set = set(p.name for p in missing)
882 if previously_missing_set == missing_set:
883 break
884 previously_missing_set = missing_set
Mike Frysingerb2fa30a2021-02-24 00:15:32 -0500885 success, new_fetched = self._Fetch(to_fetch, opt, err_event)
886 if not success:
887 err_event.set()
888 fetched.update(new_fetched)
Mike Frysinger5a033082019-09-23 19:21:20 -0400889
890 # If we saw an error, exit with code 1 so that other scripts can check.
Mike Frysingerbe24a542021-02-23 03:24:12 -0500891 if err_event.is_set():
Mike Frysinger5a033082019-09-23 19:21:20 -0400892 err_network_sync = True
893 if opt.fail_fast:
894 print('\nerror: Exited sync due to fetch errors.\n'
895 'Local checkouts *not* updated. Resolve network issues & '
896 'retry.\n'
897 '`repo sync -l` will update some local checkouts.',
898 file=sys.stderr)
899 sys.exit(1)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800900
Julien Campergue335f5ef2013-10-16 11:02:35 +0200901 if self.manifest.IsMirror or self.manifest.IsArchive:
Shawn O. Pearcecd1d7ff2009-06-04 16:15:53 -0700902 # bail out now, we have no working tree
903 return
904
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500905 if self.UpdateProjectList(opt):
Mike Frysinger5a033082019-09-23 19:21:20 -0400906 err_event.set()
907 err_update_projects = True
908 if opt.fail_fast:
909 print('\nerror: Local checkouts *not* updated.', file=sys.stderr)
910 sys.exit(1)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700911
Mike Frysinger5a033082019-09-23 19:21:20 -0400912 err_results = []
Mike Frysingerebf04a42021-02-23 20:48:04 -0500913 # NB: We don't exit here because this is the last step.
914 err_checkout = not self._Checkout(all_projects, opt, err_results)
915 if err_checkout:
916 err_event.set()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700917
Doug Anderson2b8db3c2010-11-01 15:08:06 -0700918 # If there's a notice that's supposed to print at the end of the sync, print
919 # it now...
920 if self.manifest.notice:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700921 print(self.manifest.notice)
Doug Anderson2b8db3c2010-11-01 15:08:06 -0700922
Mike Frysinger5a033082019-09-23 19:21:20 -0400923 # If we saw an error, exit with code 1 so that other scripts can check.
Mike Frysingerbe24a542021-02-23 03:24:12 -0500924 if err_event.is_set():
Mike Frysinger5a033082019-09-23 19:21:20 -0400925 print('\nerror: Unable to fully sync the tree.', file=sys.stderr)
926 if err_network_sync:
927 print('error: Downloading network changes failed.', file=sys.stderr)
928 if err_update_projects:
929 print('error: Updating local project lists failed.', file=sys.stderr)
930 if err_checkout:
931 print('error: Checking out local projects failed.', file=sys.stderr)
932 if err_results:
933 print('Failing repos:\n%s' % '\n'.join(err_results), file=sys.stderr)
934 print('Try re-running with "-j1 --fail-fast" to exit at the first error.',
935 file=sys.stderr)
936 sys.exit(1)
937
Mike Frysingere19d9e12020-02-12 11:23:32 -0500938 if not opt.quiet:
939 print('repo sync has finished successfully.')
940
David Pursehouse819827a2020-02-12 15:20:19 +0900941
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -0700942def _PostRepoUpgrade(manifest, quiet=False):
Conley Owens094cdbe2014-01-30 15:09:59 -0800943 wrapper = Wrapper()
Conley Owensc9129d92012-10-01 16:12:28 -0700944 if wrapper.NeedSetupGnuPG():
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -0700945 wrapper.SetupGnuPG(quiet)
Conley Owensf2fe2d92014-01-29 13:53:43 -0800946 for project in manifest.projects:
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700947 if project.Exists:
948 project.PostRepoUpgrade()
949
David Pursehouse819827a2020-02-12 15:20:19 +0900950
Mike Frysingerc58ec4d2020-02-17 14:36:08 -0500951def _PostRepoFetch(rp, repo_verify=True, verbose=False):
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700952 if rp.HasChanges:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700953 print('info: A new version of repo is available', file=sys.stderr)
Mike Frysinger347f9ed2021-03-15 14:58:52 -0400954 wrapper = Wrapper()
955 try:
956 rev = rp.bare_git.describe(rp.GetRevisionId())
957 except GitError:
958 rev = None
959 _, new_rev = wrapper.check_repo_rev(rp.gitdir, rev, repo_verify=repo_verify)
960 # See if we're held back due to missing signed tag.
961 current_revid = rp.bare_git.rev_parse('HEAD')
962 new_revid = rp.bare_git.rev_parse('--verify', new_rev)
963 if current_revid != new_revid:
964 # We want to switch to the new rev, but also not trash any uncommitted
965 # changes. This helps with local testing/hacking.
966 # If a local change has been made, we will throw that away.
967 # We also have to make sure this will switch to an older commit if that's
968 # the latest tag in order to support release rollback.
969 try:
970 rp.work_git.reset('--keep', new_rev)
971 except GitError as e:
972 sys.exit(str(e))
Sarah Owenscecd1d82012-11-01 22:59:27 -0700973 print('info: Restarting repo with latest version', file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700974 raise RepoChangedException(['--repo-upgraded'])
975 else:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700976 print('warning: Skipped upgrade to unverified version', file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700977 else:
978 if verbose:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700979 print('repo version %s is current' % rp.work_git.describe(HEAD),
980 file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700981
David Pursehouse819827a2020-02-12 15:20:19 +0900982
Dave Borowitz67700e92012-10-23 15:00:54 -0700983class _FetchTimes(object):
Dave Borowitzd9478582012-10-23 16:35:39 -0700984 _ALPHA = 0.5
985
Dave Borowitz67700e92012-10-23 15:00:54 -0700986 def __init__(self, manifest):
Anthony King85b24ac2014-05-06 15:57:48 +0100987 self._path = os.path.join(manifest.repodir, '.repo_fetchtimes.json')
Dave Borowitz67700e92012-10-23 15:00:54 -0700988 self._times = None
Dave Borowitzd9478582012-10-23 16:35:39 -0700989 self._seen = set()
Dave Borowitz67700e92012-10-23 15:00:54 -0700990
991 def Get(self, project):
992 self._Load()
993 return self._times.get(project.name, _ONE_DAY_S)
994
995 def Set(self, project, t):
Dave Borowitzd9478582012-10-23 16:35:39 -0700996 self._Load()
997 name = project.name
998 old = self._times.get(name, t)
999 self._seen.add(name)
1000 a = self._ALPHA
David Pursehouse54a4e602020-02-12 14:31:05 +09001001 self._times[name] = (a * t) + ((1 - a) * old)
Dave Borowitz67700e92012-10-23 15:00:54 -07001002
1003 def _Load(self):
1004 if self._times is None:
1005 try:
Mike Frysinger3164d402019-11-11 05:40:22 -05001006 with open(self._path) as f:
Anthony King85b24ac2014-05-06 15:57:48 +01001007 self._times = json.load(f)
Anthony King85b24ac2014-05-06 15:57:48 +01001008 except (IOError, ValueError):
1009 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08001010 platform_utils.remove(self._path)
Anthony King85b24ac2014-05-06 15:57:48 +01001011 except OSError:
1012 pass
1013 self._times = {}
Dave Borowitz67700e92012-10-23 15:00:54 -07001014
1015 def Save(self):
1016 if self._times is None:
1017 return
Dave Borowitzd9478582012-10-23 16:35:39 -07001018
1019 to_delete = []
1020 for name in self._times:
1021 if name not in self._seen:
1022 to_delete.append(name)
1023 for name in to_delete:
1024 del self._times[name]
1025
Dave Borowitz67700e92012-10-23 15:00:54 -07001026 try:
Mike Frysinger3164d402019-11-11 05:40:22 -05001027 with open(self._path, 'w') as f:
Anthony King85b24ac2014-05-06 15:57:48 +01001028 json.dump(self._times, f, indent=2)
Anthony King85b24ac2014-05-06 15:57:48 +01001029 except (IOError, TypeError):
1030 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08001031 platform_utils.remove(self._path)
Anthony King85b24ac2014-05-06 15:57:48 +01001032 except OSError:
1033 pass
Dan Willemsen0745bb22015-08-17 13:41:45 -07001034
1035# This is a replacement for xmlrpc.client.Transport using urllib2
1036# and supporting persistent-http[s]. It cannot change hosts from
1037# request to request like the normal transport, the real url
1038# is passed during initialization.
David Pursehouse819827a2020-02-12 15:20:19 +09001039
1040
Dan Willemsen0745bb22015-08-17 13:41:45 -07001041class PersistentTransport(xmlrpc.client.Transport):
1042 def __init__(self, orig_host):
1043 self.orig_host = orig_host
1044
1045 def request(self, host, handler, request_body, verbose=False):
1046 with GetUrlCookieFile(self.orig_host, not verbose) as (cookiefile, proxy):
1047 # Python doesn't understand cookies with the #HttpOnly_ prefix
1048 # Since we're only using them for HTTP, copy the file temporarily,
1049 # stripping those prefixes away.
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001050 if cookiefile:
Collin Fijalkoviche1191b32020-02-18 10:57:32 -08001051 tmpcookiefile = tempfile.NamedTemporaryFile(mode='w')
David Pursehouse4c5f74e2015-10-02 11:10:10 +09001052 tmpcookiefile.write("# HTTP Cookie File")
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001053 try:
1054 with open(cookiefile) as f:
1055 for line in f:
1056 if line.startswith("#HttpOnly_"):
1057 line = line[len("#HttpOnly_"):]
1058 tmpcookiefile.write(line)
1059 tmpcookiefile.flush()
Dan Willemsen0745bb22015-08-17 13:41:45 -07001060
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001061 cookiejar = cookielib.MozillaCookieJar(tmpcookiefile.name)
David Pursehouseb1ad2192015-09-30 10:35:43 +09001062 try:
1063 cookiejar.load()
1064 except cookielib.LoadError:
1065 cookiejar = cookielib.CookieJar()
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001066 finally:
1067 tmpcookiefile.close()
1068 else:
1069 cookiejar = cookielib.CookieJar()
Dan Willemsen0745bb22015-08-17 13:41:45 -07001070
1071 proxyhandler = urllib.request.ProxyHandler
1072 if proxy:
1073 proxyhandler = urllib.request.ProxyHandler({
1074 "http": proxy,
David Pursehouse54a4e602020-02-12 14:31:05 +09001075 "https": proxy})
Dan Willemsen0745bb22015-08-17 13:41:45 -07001076
1077 opener = urllib.request.build_opener(
1078 urllib.request.HTTPCookieProcessor(cookiejar),
1079 proxyhandler)
1080
1081 url = urllib.parse.urljoin(self.orig_host, handler)
1082 parse_results = urllib.parse.urlparse(url)
1083
1084 scheme = parse_results.scheme
1085 if scheme == 'persistent-http':
1086 scheme = 'http'
1087 if scheme == 'persistent-https':
1088 # If we're proxying through persistent-https, use http. The
1089 # proxy itself will do the https.
1090 if proxy:
1091 scheme = 'http'
1092 else:
1093 scheme = 'https'
1094
1095 # Parse out any authentication information using the base class
1096 host, extra_headers, _ = self.get_host_info(parse_results.netloc)
1097
1098 url = urllib.parse.urlunparse((
1099 scheme,
1100 host,
1101 parse_results.path,
1102 parse_results.params,
1103 parse_results.query,
1104 parse_results.fragment))
1105
1106 request = urllib.request.Request(url, request_body)
1107 if extra_headers is not None:
1108 for (name, header) in extra_headers:
1109 request.add_header(name, header)
1110 request.add_header('Content-Type', 'text/xml')
1111 try:
1112 response = opener.open(request)
1113 except urllib.error.HTTPError as e:
1114 if e.code == 501:
1115 # We may have been redirected through a login process
1116 # but our POST turned into a GET. Retry.
1117 response = opener.open(request)
1118 else:
1119 raise
1120
1121 p, u = xmlrpc.client.getparser()
1122 while 1:
1123 data = response.read(1024)
1124 if not data:
1125 break
1126 p.feed(data)
1127 p.close()
1128 return u.close()
1129
1130 def close(self):
1131 pass