blob: f1fe88d81ef80b2b0d940a42223aa46988f3d326 [file] [log] [blame]
Mike Frysingerf6013762019-06-13 02:30:51 -04001# -*- coding:utf-8 -*-
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002#
3# Copyright (C) 2008 The Android Open Source Project
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9# http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
Sarah Owenscecd1d82012-11-01 22:59:27 -070017from __future__ import print_function
Anthony King85b24ac2014-05-06 15:57:48 +010018import json
David Pursehouse86d973d2012-08-24 10:21:02 +090019import netrc
Shawn O. Pearce2a1ccb22009-04-10 16:51:53 -070020from optparse import SUPPRESS_HELP
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070021import os
22import re
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -070023import socket
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070024import subprocess
25import sys
Dan Willemsen0745bb22015-08-17 13:41:45 -070026import tempfile
Shawn O. Pearcef6906872009-04-18 10:49:00 -070027import time
David Pursehouse59bbb582013-05-17 10:49:33 +090028
29from pyversion import is_python3
30if is_python3():
Dan Willemsen0745bb22015-08-17 13:41:45 -070031 import http.cookiejar as cookielib
32 import urllib.error
Chirayu Desai217ea7d2013-03-01 19:14:38 +053033 import urllib.parse
Dan Willemsen0745bb22015-08-17 13:41:45 -070034 import urllib.request
David Pursehouse59bbb582013-05-17 10:49:33 +090035 import xmlrpc.client
36else:
Dan Willemsen0745bb22015-08-17 13:41:45 -070037 import cookielib
Chirayu Desai217ea7d2013-03-01 19:14:38 +053038 import imp
Dan Willemsen0745bb22015-08-17 13:41:45 -070039 import urllib2
Chirayu Desai217ea7d2013-03-01 19:14:38 +053040 import urlparse
David Pursehouse59bbb582013-05-17 10:49:33 +090041 import xmlrpclib
Chirayu Desai217ea7d2013-03-01 19:14:38 +053042 urllib = imp.new_module('urllib')
Dan Willemsen0745bb22015-08-17 13:41:45 -070043 urllib.error = urllib2
Chirayu Desaidb2ad9d2013-06-11 13:42:25 +053044 urllib.parse = urlparse
Dan Willemsen0745bb22015-08-17 13:41:45 -070045 urllib.request = urllib2
Chirayu Desai217ea7d2013-03-01 19:14:38 +053046 xmlrpc = imp.new_module('xmlrpc')
47 xmlrpc.client = xmlrpclib
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070048
Roy Lee18afd7f2010-05-09 04:32:08 +080049try:
50 import threading as _threading
51except ImportError:
52 import dummy_threading as _threading
53
Shawn O. Pearce97d2b2f2011-09-22 17:23:41 -070054try:
55 import resource
56 def _rlimit_nofile():
57 return resource.getrlimit(resource.RLIMIT_NOFILE)
58except ImportError:
59 def _rlimit_nofile():
60 return (256, 256)
61
Dave Borowitz18857212012-10-23 17:02:59 -070062try:
63 import multiprocessing
64except ImportError:
65 multiprocessing = None
66
David Rileye0684ad2017-04-05 00:02:59 -070067import event_log
Dave Borowitze2152672012-10-31 12:24:38 -070068from git_command import GIT, git_require
David Pursehouseba7bc732015-08-20 16:55:42 +090069from git_config import GetUrlCookieFile
David Pursehoused94aaef2012-09-07 09:52:04 +090070from git_refs import R_HEADS, HEAD
Simran Basibdb52712015-08-10 13:23:23 -070071import gitc_utils
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -070072from project import Project
73from project import RemoteSpec
Shawn O. Pearcec95583b2009-03-03 17:47:06 -080074from command import Command, MirrorSafeCommand
Torne (Richard Coles)7bdbde72012-12-05 10:58:06 +000075from error import RepoChangedException, GitError, ManifestParseError
Renaud Paquaya65adf72016-11-03 10:37:53 -070076import platform_utils
Shawn O. Pearce350cde42009-04-16 11:21:18 -070077from project import SyncBuffer
Shawn O. Pearce68194f42009-04-10 16:48:52 -070078from progress import Progress
Conley Owens094cdbe2014-01-30 15:09:59 -080079from wrapper import Wrapper
Dan Willemsen5ea32d12015-09-08 13:27:20 -070080from manifest_xml import GitcManifest
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070081
Dave Borowitz67700e92012-10-23 15:00:54 -070082_ONE_DAY_S = 24 * 60 * 60
83
Doug Andersonfc06ced2011-03-16 15:49:18 -070084class _FetchError(Exception):
85 """Internal error thrown in _FetchHelper() when we don't want stack trace."""
86 pass
87
Xin Li745be2e2019-06-03 11:24:30 -070088class _CheckoutError(Exception):
89 """Internal error thrown in _CheckoutOne() when we don't want stack trace."""
90
Shawn O. Pearcec95583b2009-03-03 17:47:06 -080091class Sync(Command, MirrorSafeCommand):
Roy Lee18afd7f2010-05-09 04:32:08 +080092 jobs = 1
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070093 common = True
94 helpSummary = "Update working tree to the latest revision"
95 helpUsage = """
96%prog [<project>...]
97"""
98 helpDescription = """
99The '%prog' command synchronizes local project directories
100with the remote repositories specified in the manifest. If a local
101project does not yet exist, it will clone a new local directory from
102the remote repository and set up tracking branches as specified in
103the manifest. If the local project already exists, '%prog'
104will update the remote branches and rebase any new local changes
105on top of the new remote changes.
106
107'%prog' will synchronize all projects listed at the command
108line. Projects can be specified either by name, or by a relative
109or absolute path to the project's local directory. If no projects
110are specified, '%prog' will synchronize all projects listed in
111the manifest.
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700112
113The -d/--detach option can be used to switch specified projects
114back to the manifest revision. This option is especially helpful
115if the project is currently on a topic branch, but the manifest
116revision is temporarily needed.
Shawn O. Pearceeb7af872009-04-21 08:02:04 -0700117
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700118The -s/--smart-sync option can be used to sync to a known good
119build as specified by the manifest-server element in the current
Victor Boivie08c880d2011-04-19 10:32:52 +0200120manifest. The -t/--smart-tag option is similar and allows you to
121specify a custom tag/label.
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700122
David Pursehousecf76b1b2012-09-14 10:31:42 +0900123The -u/--manifest-server-username and -p/--manifest-server-password
124options can be used to specify a username and password to authenticate
125with the manifest server when using the -s or -t option.
126
127If -u and -p are not specified when using the -s or -t option, '%prog'
128will attempt to read authentication credentials for the manifest server
129from the user's .netrc file.
130
131'%prog' will not use authentication credentials from -u/-p or .netrc
132if the manifest server specified in the manifest file already includes
133credentials.
134
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400135By default, all projects will be synced. The --fail-fast option can be used
136to halt syncing as soon as possible when the the first project fails to sync.
Andrei Warkentin5df6de02010-07-02 17:58:31 -0500137
Kevin Degiabaa7f32014-11-12 11:27:45 -0700138The --force-sync option can be used to overwrite existing git
139directories if they have previously been linked to a different
140object direcotry. WARNING: This may cause data to be lost since
141refs may be removed when overwriting.
142
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500143The --force-remove-dirty option can be used to remove previously used
144projects with uncommitted changes. WARNING: This may cause data to be
145lost since uncommitted changes may be removed with projects that no longer
146exist in the manifest.
147
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -0700148The --no-clone-bundle option disables any attempt to use
149$URL/clone.bundle to bootstrap a new Git repository from a
150resumeable bundle file on a content delivery network. This
151may be necessary if there are problems with the local Python
152HTTP client or proxy configuration, but the Git binary works.
153
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800154The --fetch-submodules option enables fetching Git submodules
155of a project from server.
156
David Pursehousef2fad612015-01-29 14:36:28 +0900157The -c/--current-branch option can be used to only fetch objects that
158are on the branch specified by a project's revision.
159
David Pursehouseb1553542014-09-04 21:28:09 +0900160The --optimized-fetch option can be used to only fetch projects that
161are fixed to a sha1 revision if the sha1 revision does not already
162exist locally.
163
David Pursehouse74cfd272015-10-14 10:50:15 +0900164The --prune option can be used to remove any refs that no longer
165exist on the remote.
166
Mike Frysingerb8f7bb02018-10-10 01:05:11 -0400167# SSH Connections
Shawn O. Pearceeb7af872009-04-21 08:02:04 -0700168
169If at least one project remote URL uses an SSH connection (ssh://,
170git+ssh://, or user@host:path syntax) repo will automatically
171enable the SSH ControlMaster option when connecting to that host.
172This feature permits other projects in the same '%prog' session to
173reuse the same SSH tunnel, saving connection setup overheads.
174
175To disable this behavior on UNIX platforms, set the GIT_SSH
176environment variable to 'ssh'. For example:
177
178 export GIT_SSH=ssh
179 %prog
180
Mike Frysingerb8f7bb02018-10-10 01:05:11 -0400181# Compatibility
Shawn O. Pearceeb7af872009-04-21 08:02:04 -0700182
183This feature is automatically disabled on Windows, due to the lack
184of UNIX domain socket support.
185
186This feature is not compatible with url.insteadof rewrites in the
187user's ~/.gitconfig. '%prog' is currently not able to perform the
188rewrite early enough to establish the ControlMaster tunnel.
189
190If the remote SSH daemon is Gerrit Code Review, version 2.0.10 or
191later is required to fix a server side protocol bug.
192
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700193"""
194
Nico Sallembien6623b212010-05-11 12:57:01 -0700195 def _Options(self, p, show_smart=True):
Torne (Richard Coles)7bdbde72012-12-05 10:58:06 +0000196 try:
197 self.jobs = self.manifest.default.sync_j
198 except ManifestParseError:
199 self.jobs = 1
Shawn O. Pearce6392c872011-09-22 17:44:31 -0700200
Andrei Warkentin5df6de02010-07-02 17:58:31 -0500201 p.add_option('-f', '--force-broken',
Stefan Müller-Klieser46702ed2019-08-30 10:20:15 +0200202 dest='force_broken', action='store_true',
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400203 help='obsolete option (to be deleted in the future)')
204 p.add_option('--fail-fast',
205 dest='fail_fast', action='store_true',
206 help='stop syncing after first error is hit')
Kevin Degiabaa7f32014-11-12 11:27:45 -0700207 p.add_option('--force-sync',
208 dest='force_sync', action='store_true',
209 help="overwrite an existing git directory if it needs to "
210 "point to a different object directory. WARNING: this "
211 "may cause loss of data")
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500212 p.add_option('--force-remove-dirty',
213 dest='force_remove_dirty', action='store_true',
214 help="force remove projects with uncommitted modifications if "
215 "projects no longer exist in the manifest. "
216 "WARNING: this may cause loss of data")
David Pursehouse8f62fb72012-11-14 12:09:38 +0900217 p.add_option('-l', '--local-only',
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700218 dest='local_only', action='store_true',
219 help="only update working tree, don't fetch")
David Pursehouse8f62fb72012-11-14 12:09:38 +0900220 p.add_option('-n', '--network-only',
Shawn O. Pearce96fdcef2009-04-10 16:29:20 -0700221 dest='network_only', action='store_true',
222 help="fetch only, don't update working tree")
David Pursehouse8f62fb72012-11-14 12:09:38 +0900223 p.add_option('-d', '--detach',
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700224 dest='detach_head', action='store_true',
225 help='detach projects back to manifest revision')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900226 p.add_option('-c', '--current-branch',
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -0700227 dest='current_branch_only', action='store_true',
228 help='fetch only current branch from server')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900229 p.add_option('-q', '--quiet',
Shawn O. Pearce16614f82010-10-29 12:05:43 -0700230 dest='quiet', action='store_true',
231 help='be more quiet')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900232 p.add_option('-j', '--jobs',
Roy Lee18afd7f2010-05-09 04:32:08 +0800233 dest='jobs', action='store', type='int',
Shawn O. Pearce6392c872011-09-22 17:44:31 -0700234 help="projects to fetch simultaneously (default %d)" % self.jobs)
Chris Wolfee9dc3b32012-01-26 11:36:18 -0500235 p.add_option('-m', '--manifest-name',
236 dest='manifest_name',
237 help='temporary manifest to use for this sync', metavar='NAME.xml')
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -0700238 p.add_option('--no-clone-bundle',
239 dest='no_clone_bundle', action='store_true',
240 help='disable use of /clone.bundle on HTTP/HTTPS')
Conley Owens8d070cf2012-11-06 13:14:31 -0800241 p.add_option('-u', '--manifest-server-username', action='store',
242 dest='manifest_server_username',
243 help='username to authenticate with the manifest server')
244 p.add_option('-p', '--manifest-server-password', action='store',
245 dest='manifest_server_password',
246 help='password to authenticate with the manifest server')
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800247 p.add_option('--fetch-submodules',
248 dest='fetch_submodules', action='store_true',
249 help='fetch submodules from server')
Mitchel Humpherys597868b2012-10-29 10:18:34 -0700250 p.add_option('--no-tags',
251 dest='no_tags', action='store_true',
252 help="don't fetch tags")
David Pursehouseb1553542014-09-04 21:28:09 +0900253 p.add_option('--optimized-fetch',
254 dest='optimized_fetch', action='store_true',
255 help='only fetch projects fixed to sha1 if revision does not exist locally')
David Pursehouse74cfd272015-10-14 10:50:15 +0900256 p.add_option('--prune', dest='prune', action='store_true',
257 help='delete refs that no longer exist on the remote')
Nico Sallembien6623b212010-05-11 12:57:01 -0700258 if show_smart:
259 p.add_option('-s', '--smart-sync',
260 dest='smart_sync', action='store_true',
David Pursehouse79fba682016-04-13 18:03:00 +0900261 help='smart sync using manifest from the latest known good build')
Victor Boivie08c880d2011-04-19 10:32:52 +0200262 p.add_option('-t', '--smart-tag',
263 dest='smart_tag', action='store',
264 help='smart sync using manifest from a known tag')
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700265
Shawn O. Pearcefd89b672009-04-18 11:28:57 -0700266 g = p.add_option_group('repo Version options')
267 g.add_option('--no-repo-verify',
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700268 dest='no_repo_verify', action='store_true',
269 help='do not verify repo source code')
Shawn O. Pearcefd89b672009-04-18 11:28:57 -0700270 g.add_option('--repo-upgraded',
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800271 dest='repo_upgraded', action='store_true',
Shawn O. Pearce2a1ccb22009-04-10 16:51:53 -0700272 help=SUPPRESS_HELP)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700273
Andrew Wheeler7f1ccfb2016-06-17 16:51:07 -0500274 def _FetchProjectList(self, opt, projects, sem, *args, **kwargs):
Xin Li745be2e2019-06-03 11:24:30 -0700275 """Main function of the fetch threads.
Roy Lee18afd7f2010-05-09 04:32:08 +0800276
David James8d201162013-10-11 17:03:19 -0700277 Delegates most of the work to _FetchHelper.
278
279 Args:
280 opt: Program options returned from optparse. See _Options().
281 projects: Projects to fetch.
Andrew Wheeler7f1ccfb2016-06-17 16:51:07 -0500282 sem: We'll release() this semaphore when we exit so that another thread
283 can be started up.
David James89ece422014-01-09 18:51:58 -0800284 *args, **kwargs: Remaining arguments to pass to _FetchHelper. See the
David James8d201162013-10-11 17:03:19 -0700285 _FetchHelper docstring for details.
286 """
Andrew Wheeler7f1ccfb2016-06-17 16:51:07 -0500287 try:
288 for project in projects:
289 success = self._FetchHelper(opt, project, *args, **kwargs)
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400290 if not success and opt.fail_fast:
Andrew Wheeler7f1ccfb2016-06-17 16:51:07 -0500291 break
292 finally:
293 sem.release()
David James8d201162013-10-11 17:03:19 -0700294
Xin Li745be2e2019-06-03 11:24:30 -0700295 def _FetchHelper(self, opt, project, lock, fetched, pm, err_event,
296 clone_filter):
David James8d201162013-10-11 17:03:19 -0700297 """Fetch git objects for a single project.
298
David Pursehousec1b86a22012-11-14 11:36:51 +0900299 Args:
300 opt: Program options returned from optparse. See _Options().
301 project: Project object for the project to fetch.
302 lock: Lock for accessing objects that are shared amongst multiple
303 _FetchHelper() threads.
304 fetched: set object that we will add project.gitdir to when we're done
305 (with our lock held).
306 pm: Instance of a Project object. We will call pm.update() (with our
307 lock held).
David Pursehousec1b86a22012-11-14 11:36:51 +0900308 err_event: We'll set this event in the case of an error (after printing
309 out info about the error).
Xin Li745be2e2019-06-03 11:24:30 -0700310 clone_filter: Filter for use in a partial clone.
David James8d201162013-10-11 17:03:19 -0700311
312 Returns:
313 Whether the fetch was successful.
David Pursehousec1b86a22012-11-14 11:36:51 +0900314 """
315 # We'll set to true once we've locked the lock.
316 did_lock = False
Doug Andersonfc06ced2011-03-16 15:49:18 -0700317
Chirayu Desaifef4ae72013-04-12 14:54:32 +0530318 if not opt.quiet:
319 print('Fetching project %s' % project.name)
320
David Pursehousec1b86a22012-11-14 11:36:51 +0900321 # Encapsulate everything in a try/except/finally so that:
322 # - We always set err_event in the case of an exception.
David Pursehousec1b86a22012-11-14 11:36:51 +0900323 # - We always make sure we unlock the lock if we locked it.
David Rileye0684ad2017-04-05 00:02:59 -0700324 start = time.time()
325 success = False
David Pursehousec1b86a22012-11-14 11:36:51 +0900326 try:
Doug Andersonfc06ced2011-03-16 15:49:18 -0700327 try:
David Pursehousec1b86a22012-11-14 11:36:51 +0900328 success = project.Sync_NetworkHalf(
329 quiet=opt.quiet,
330 current_branch_only=opt.current_branch_only,
Kevin Degiabaa7f32014-11-12 11:27:45 -0700331 force_sync=opt.force_sync,
Mitchel Humpherys597868b2012-10-29 10:18:34 -0700332 clone_bundle=not opt.no_clone_bundle,
David Pursehouseb1553542014-09-04 21:28:09 +0900333 no_tags=opt.no_tags, archive=self.manifest.IsArchive,
David Pursehouse74cfd272015-10-14 10:50:15 +0900334 optimized_fetch=opt.optimized_fetch,
Xin Li745be2e2019-06-03 11:24:30 -0700335 prune=opt.prune,
336 clone_filter=clone_filter)
David Pursehousec1b86a22012-11-14 11:36:51 +0900337 self._fetch_times.Set(project, time.time() - start)
Doug Andersonfc06ced2011-03-16 15:49:18 -0700338
David Pursehousec1b86a22012-11-14 11:36:51 +0900339 # Lock around all the rest of the code, since printing, updating a set
340 # and Progress.update() are not thread safe.
341 lock.acquire()
342 did_lock = True
Doug Andersonfc06ced2011-03-16 15:49:18 -0700343
David Pursehousec1b86a22012-11-14 11:36:51 +0900344 if not success:
Hu Xiuyune9becc02015-11-25 15:52:26 +0800345 err_event.set()
Marc Herbertffb4b892017-04-04 22:03:53 -0700346 print('error: Cannot fetch %s from %s'
347 % (project.name, project.remote.url),
348 file=sys.stderr)
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400349 if opt.fail_fast:
David Pursehousec1b86a22012-11-14 11:36:51 +0900350 raise _FetchError()
Doug Andersonfc06ced2011-03-16 15:49:18 -0700351
David Pursehousec1b86a22012-11-14 11:36:51 +0900352 fetched.add(project.gitdir)
353 pm.update()
354 except _FetchError:
Hu Xiuyune9becc02015-11-25 15:52:26 +0800355 pass
Dan Sandlerc5cd4332015-07-31 09:37:53 -0400356 except Exception as e:
357 print('error: Cannot fetch %s (%s: %s)' \
358 % (project.name, type(e).__name__, str(e)), file=sys.stderr)
David Pursehousec1b86a22012-11-14 11:36:51 +0900359 err_event.set()
360 raise
361 finally:
362 if did_lock:
363 lock.release()
David Rileye0684ad2017-04-05 00:02:59 -0700364 finish = time.time()
365 self.event_log.AddSync(project, event_log.TASK_SYNC_NETWORK,
366 start, finish, success)
Roy Lee18afd7f2010-05-09 04:32:08 +0800367
David James8d201162013-10-11 17:03:19 -0700368 return success
369
Shawn O. Pearce16614f82010-10-29 12:05:43 -0700370 def _Fetch(self, projects, opt):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700371 fetched = set()
David James89ece422014-01-09 18:51:58 -0800372 lock = _threading.Lock()
Tim Schumacher913327f2017-06-05 15:01:41 +0200373 pm = Progress('Fetching projects', len(projects),
Tim Schumacher7be072e2017-06-28 18:29:23 +0200374 print_newline=not(opt.quiet),
375 always_print_percentage=opt.quiet)
Roy Lee18afd7f2010-05-09 04:32:08 +0800376
David James89ece422014-01-09 18:51:58 -0800377 objdir_project_map = dict()
378 for project in projects:
379 objdir_project_map.setdefault(project.objdir, []).append(project)
David James8d201162013-10-11 17:03:19 -0700380
David James89ece422014-01-09 18:51:58 -0800381 threads = set()
382 sem = _threading.Semaphore(self.jobs)
383 err_event = _threading.Event()
384 for project_list in objdir_project_map.values():
385 # Check for any errors before running any more tasks.
386 # ...we'll let existing threads finish, though.
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400387 if err_event.isSet() and opt.fail_fast:
David James89ece422014-01-09 18:51:58 -0800388 break
Doug Andersonfc06ced2011-03-16 15:49:18 -0700389
David James89ece422014-01-09 18:51:58 -0800390 sem.acquire()
391 kwargs = dict(opt=opt,
392 projects=project_list,
Andrew Wheeler7f1ccfb2016-06-17 16:51:07 -0500393 sem=sem,
David James89ece422014-01-09 18:51:58 -0800394 lock=lock,
395 fetched=fetched,
396 pm=pm,
Xin Li745be2e2019-06-03 11:24:30 -0700397 err_event=err_event,
398 clone_filter=self.manifest.CloneFilter)
David James89ece422014-01-09 18:51:58 -0800399 if self.jobs > 1:
David James8d201162013-10-11 17:03:19 -0700400 t = _threading.Thread(target = self._FetchProjectList,
David James89ece422014-01-09 18:51:58 -0800401 kwargs = kwargs)
David 'Digit' Turnere2126652012-09-05 10:35:06 +0200402 # Ensure that Ctrl-C will not freeze the repo process.
403 t.daemon = True
Roy Lee18afd7f2010-05-09 04:32:08 +0800404 threads.add(t)
405 t.start()
David James89ece422014-01-09 18:51:58 -0800406 else:
407 self._FetchProjectList(**kwargs)
Roy Lee18afd7f2010-05-09 04:32:08 +0800408
David James89ece422014-01-09 18:51:58 -0800409 for t in threads:
410 t.join()
Roy Lee18afd7f2010-05-09 04:32:08 +0800411
David James89ece422014-01-09 18:51:58 -0800412 # If we saw an error, exit with code 1 so that other scripts can check.
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400413 if err_event.isSet() and opt.fail_fast:
David James89ece422014-01-09 18:51:58 -0800414 print('\nerror: Exited sync due to fetch errors', file=sys.stderr)
415 sys.exit(1)
Doug Andersonfc06ced2011-03-16 15:49:18 -0700416
Shawn O. Pearce68194f42009-04-10 16:48:52 -0700417 pm.end()
Dave Borowitz67700e92012-10-23 15:00:54 -0700418 self._fetch_times.Save()
Dave Borowitz18857212012-10-23 17:02:59 -0700419
Julien Campergue335f5ef2013-10-16 11:02:35 +0200420 if not self.manifest.IsArchive:
421 self._GCProjects(projects)
422
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700423 return fetched
424
Xin Li745be2e2019-06-03 11:24:30 -0700425 def _CheckoutWorker(self, opt, sem, project, *args, **kwargs):
426 """Main function of the fetch threads.
427
428 Delegates most of the work to _CheckoutOne.
429
430 Args:
431 opt: Program options returned from optparse. See _Options().
432 projects: Projects to fetch.
433 sem: We'll release() this semaphore when we exit so that another thread
434 can be started up.
435 *args, **kwargs: Remaining arguments to pass to _CheckoutOne. See the
436 _CheckoutOne docstring for details.
437 """
438 try:
Mike Frysingera34186e2019-08-07 18:07:31 -0400439 return self._CheckoutOne(opt, project, *args, **kwargs)
Xin Li745be2e2019-06-03 11:24:30 -0700440 finally:
441 sem.release()
442
443 def _CheckoutOne(self, opt, project, lock, pm, err_event):
444 """Checkout work tree for one project
445
446 Args:
447 opt: Program options returned from optparse. See _Options().
448 project: Project object for the project to checkout.
449 lock: Lock for accessing objects that are shared amongst multiple
450 _CheckoutWorker() threads.
451 pm: Instance of a Project object. We will call pm.update() (with our
452 lock held).
453 err_event: We'll set this event in the case of an error (after printing
454 out info about the error).
455
456 Returns:
457 Whether the fetch was successful.
458 """
459 # We'll set to true once we've locked the lock.
460 did_lock = False
461
462 if not opt.quiet:
463 print('Checking out project %s' % project.name)
464
465 # Encapsulate everything in a try/except/finally so that:
466 # - We always set err_event in the case of an exception.
467 # - We always make sure we unlock the lock if we locked it.
468 start = time.time()
469 syncbuf = SyncBuffer(self.manifest.manifestProject.config,
470 detach_head=opt.detach_head)
471 success = False
472 try:
473 try:
474 project.Sync_LocalHalf(syncbuf, force_sync=opt.force_sync)
475 success = syncbuf.Finish()
476
477 # Lock around all the rest of the code, since printing, updating a set
478 # and Progress.update() are not thread safe.
479 lock.acquire()
480 did_lock = True
481
482 if not success:
483 err_event.set()
484 print('error: Cannot checkout %s' % (project.name),
485 file=sys.stderr)
486 raise _CheckoutError()
487
488 pm.update()
489 except _CheckoutError:
490 pass
491 except Exception as e:
492 print('error: Cannot checkout %s: %s: %s' %
493 (project.name, type(e).__name__, str(e)),
494 file=sys.stderr)
495 err_event.set()
496 raise
497 finally:
498 if did_lock:
499 lock.release()
500 finish = time.time()
501 self.event_log.AddSync(project, event_log.TASK_SYNC_LOCAL,
502 start, finish, success)
503
504 return success
505
506 def _Checkout(self, all_projects, opt):
507 """Checkout projects listed in all_projects
508
509 Args:
510 all_projects: List of all projects that should be checked out.
511 opt: Program options returned from optparse. See _Options().
512 """
513
514 # Perform checkouts in multiple threads when we are using partial clone.
515 # Without partial clone, all needed git objects are already downloaded,
516 # in this situation it's better to use only one process because the checkout
517 # would be mostly disk I/O; with partial clone, the objects are only
518 # downloaded when demanded (at checkout time), which is similar to the
519 # Sync_NetworkHalf case and parallelism would be helpful.
520 if self.manifest.CloneFilter:
521 syncjobs = self.jobs
522 else:
523 syncjobs = 1
524
525 lock = _threading.Lock()
526 pm = Progress('Syncing work tree', len(all_projects))
527
528 threads = set()
529 sem = _threading.Semaphore(syncjobs)
530 err_event = _threading.Event()
531
532 for project in all_projects:
533 # Check for any errors before running any more tasks.
534 # ...we'll let existing threads finish, though.
Mike Frysingerd9e5cf02019-08-26 03:12:55 -0400535 if err_event.isSet() and opt.fail_fast:
Xin Li745be2e2019-06-03 11:24:30 -0700536 break
537
538 sem.acquire()
539 if project.worktree:
540 kwargs = dict(opt=opt,
541 sem=sem,
542 project=project,
543 lock=lock,
544 pm=pm,
545 err_event=err_event)
546 if syncjobs > 1:
547 t = _threading.Thread(target=self._CheckoutWorker,
548 kwargs=kwargs)
549 # Ensure that Ctrl-C will not freeze the repo process.
550 t.daemon = True
551 threads.add(t)
552 t.start()
553 else:
554 self._CheckoutWorker(**kwargs)
555
556 for t in threads:
557 t.join()
558
559 pm.end()
560 # If we saw an error, exit with code 1 so that other scripts can check.
561 if err_event.isSet():
562 print('\nerror: Exited sync due to checkout errors', file=sys.stderr)
563 sys.exit(1)
564
Dave Borowitz18857212012-10-23 17:02:59 -0700565 def _GCProjects(self, projects):
Gabe Black2ff30292014-10-09 17:54:35 -0700566 gc_gitdirs = {}
David James8d201162013-10-11 17:03:19 -0700567 for project in projects:
Gabe Black2ff30292014-10-09 17:54:35 -0700568 if len(project.manifest.GetProjectsWithName(project.name)) > 1:
569 print('Shared project %s found, disabling pruning.' % project.name)
570 project.bare_git.config('--replace-all', 'gc.pruneExpire', 'never')
571 gc_gitdirs[project.gitdir] = project.bare_git
David James8d201162013-10-11 17:03:19 -0700572
Dave Borowitze2152672012-10-31 12:24:38 -0700573 has_dash_c = git_require((1, 7, 2))
574 if multiprocessing and has_dash_c:
Dave Borowitz18857212012-10-23 17:02:59 -0700575 cpu_count = multiprocessing.cpu_count()
576 else:
577 cpu_count = 1
578 jobs = min(self.jobs, cpu_count)
579
580 if jobs < 2:
Gabe Black2ff30292014-10-09 17:54:35 -0700581 for bare_git in gc_gitdirs.values():
David James8d201162013-10-11 17:03:19 -0700582 bare_git.gc('--auto')
Dave Borowitz18857212012-10-23 17:02:59 -0700583 return
584
Mike Frysinger0c0e9342019-06-13 12:42:39 -0400585 config = {'pack.threads': cpu_count // jobs if cpu_count > jobs else 1}
Dave Borowitz18857212012-10-23 17:02:59 -0700586
587 threads = set()
588 sem = _threading.Semaphore(jobs)
589 err_event = _threading.Event()
590
David James8d201162013-10-11 17:03:19 -0700591 def GC(bare_git):
Dave Borowitz18857212012-10-23 17:02:59 -0700592 try:
593 try:
David James8d201162013-10-11 17:03:19 -0700594 bare_git.gc('--auto', config=config)
Dave Borowitz18857212012-10-23 17:02:59 -0700595 except GitError:
596 err_event.set()
597 except:
598 err_event.set()
599 raise
600 finally:
601 sem.release()
602
Gabe Black2ff30292014-10-09 17:54:35 -0700603 for bare_git in gc_gitdirs.values():
Dave Borowitz18857212012-10-23 17:02:59 -0700604 if err_event.isSet():
605 break
606 sem.acquire()
David James8d201162013-10-11 17:03:19 -0700607 t = _threading.Thread(target=GC, args=(bare_git,))
Dave Borowitz18857212012-10-23 17:02:59 -0700608 t.daemon = True
609 threads.add(t)
610 t.start()
611
612 for t in threads:
613 t.join()
614
615 if err_event.isSet():
Sarah Owenscecd1d82012-11-01 22:59:27 -0700616 print('\nerror: Exited sync due to gc errors', file=sys.stderr)
Dave Borowitz18857212012-10-23 17:02:59 -0700617 sys.exit(1)
618
Tim Kilbourn07669002013-03-08 15:02:49 -0800619 def _ReloadManifest(self, manifest_name=None):
620 if manifest_name:
621 # Override calls _Unload already
622 self.manifest.Override(manifest_name)
623 else:
624 self.manifest._Unload()
625
Dan Willemsen43507912016-09-01 16:26:02 -0700626 def _DeleteProject(self, path):
627 print('Deleting obsolete path %s' % path, file=sys.stderr)
628
629 # Delete the .git directory first, so we're less likely to have a partially
630 # working git repository around. There shouldn't be any git projects here,
631 # so rmtree works.
632 try:
Renaud Paquaya65adf72016-11-03 10:37:53 -0700633 platform_utils.rmtree(os.path.join(path, '.git'))
Renaud Paquaybed8b622018-09-27 10:46:58 -0700634 except OSError as e:
635 print('Failed to remove %s (%s)' % (os.path.join(path, '.git'), str(e)), file=sys.stderr)
Dan Willemsen43507912016-09-01 16:26:02 -0700636 print('error: Failed to delete obsolete path %s' % path, file=sys.stderr)
637 print(' remove manually, then run sync again', file=sys.stderr)
Mike Frysingera850ca22019-08-07 17:19:24 -0400638 return 1
Dan Willemsen43507912016-09-01 16:26:02 -0700639
640 # Delete everything under the worktree, except for directories that contain
641 # another git project
642 dirs_to_remove = []
643 failed = False
Renaud Paquaybed8b622018-09-27 10:46:58 -0700644 for root, dirs, files in platform_utils.walk(path):
Dan Willemsen43507912016-09-01 16:26:02 -0700645 for f in files:
646 try:
Renaud Paquay010fed72016-11-11 14:25:29 -0800647 platform_utils.remove(os.path.join(root, f))
Renaud Paquaybed8b622018-09-27 10:46:58 -0700648 except OSError as e:
649 print('Failed to remove %s (%s)' % (os.path.join(root, f), str(e)), file=sys.stderr)
Dan Willemsen43507912016-09-01 16:26:02 -0700650 failed = True
651 dirs[:] = [d for d in dirs
652 if not os.path.lexists(os.path.join(root, d, '.git'))]
653 dirs_to_remove += [os.path.join(root, d) for d in dirs
654 if os.path.join(root, d) not in dirs_to_remove]
655 for d in reversed(dirs_to_remove):
Renaud Paquay227ad2e2016-11-01 14:37:13 -0700656 if platform_utils.islink(d):
Dan Willemseneceeb1b2016-09-25 18:24:27 -0700657 try:
Renaud Paquay010fed72016-11-11 14:25:29 -0800658 platform_utils.remove(d)
Renaud Paquaybed8b622018-09-27 10:46:58 -0700659 except OSError as e:
660 print('Failed to remove %s (%s)' % (os.path.join(root, d), str(e)), file=sys.stderr)
Dan Willemseneceeb1b2016-09-25 18:24:27 -0700661 failed = True
Renaud Paquaybed8b622018-09-27 10:46:58 -0700662 elif len(platform_utils.listdir(d)) == 0:
Dan Willemsen43507912016-09-01 16:26:02 -0700663 try:
Renaud Paquaybed8b622018-09-27 10:46:58 -0700664 platform_utils.rmdir(d)
665 except OSError as e:
666 print('Failed to remove %s (%s)' % (os.path.join(root, d), str(e)), file=sys.stderr)
Dan Willemsen43507912016-09-01 16:26:02 -0700667 failed = True
668 continue
669 if failed:
670 print('error: Failed to delete obsolete path %s' % path, file=sys.stderr)
671 print(' remove manually, then run sync again', file=sys.stderr)
Mike Frysingera850ca22019-08-07 17:19:24 -0400672 return 1
Dan Willemsen43507912016-09-01 16:26:02 -0700673
674 # Try deleting parent dirs if they are empty
675 project_dir = path
676 while project_dir != self.manifest.topdir:
Renaud Paquaybed8b622018-09-27 10:46:58 -0700677 if len(platform_utils.listdir(project_dir)) == 0:
678 platform_utils.rmdir(project_dir)
Dan Willemsen43507912016-09-01 16:26:02 -0700679 else:
680 break
681 project_dir = os.path.dirname(project_dir)
682
683 return 0
684
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500685 def UpdateProjectList(self, opt):
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700686 new_project_paths = []
Colin Cross5acde752012-03-28 20:15:45 -0700687 for project in self.GetProjects(None, missing_ok=True):
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700688 if project.relpath:
689 new_project_paths.append(project.relpath)
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700690 file_name = 'project.list'
691 file_path = os.path.join(self.manifest.repodir, file_name)
692 old_project_paths = []
693
694 if os.path.exists(file_path):
695 fd = open(file_path, 'r')
696 try:
697 old_project_paths = fd.read().split('\n')
698 finally:
699 fd.close()
Kuang-che Wu0d9b16d2019-04-06 00:49:47 +0800700 # In reversed order, so subfolders are deleted before parent folder.
701 for path in sorted(old_project_paths, reverse=True):
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700702 if not path:
703 continue
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700704 if path not in new_project_paths:
David Pursehouse8a68ff92012-09-24 12:15:13 +0900705 # If the path has already been deleted, we don't need to do it
Dan Willemsen43507912016-09-01 16:26:02 -0700706 gitdir = os.path.join(self.manifest.topdir, path, '.git')
707 if os.path.exists(gitdir):
David Pursehousec1b86a22012-11-14 11:36:51 +0900708 project = Project(
709 manifest = self.manifest,
710 name = path,
711 remote = RemoteSpec('origin'),
David James8d201162013-10-11 17:03:19 -0700712 gitdir = gitdir,
713 objdir = gitdir,
David Pursehousec1b86a22012-11-14 11:36:51 +0900714 worktree = os.path.join(self.manifest.topdir, path),
715 relpath = path,
716 revisionExpr = 'HEAD',
717 revisionId = None,
718 groups = None)
Anthonyf3fdf822009-09-26 13:38:52 -0400719
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500720 if project.IsDirty() and opt.force_remove_dirty:
721 print('WARNING: Removing dirty project "%s": uncommitted changes '
722 'erased' % project.relpath, file=sys.stderr)
723 self._DeleteProject(project.worktree)
724 elif project.IsDirty():
David Pursehouse2f9e7e42013-03-05 17:26:46 +0900725 print('error: Cannot remove project "%s": uncommitted changes '
David Pursehousec1b86a22012-11-14 11:36:51 +0900726 'are present' % project.relpath, file=sys.stderr)
727 print(' commit changes, then run sync again',
728 file=sys.stderr)
Mike Frysingera850ca22019-08-07 17:19:24 -0400729 return 1
Dan Willemsen43507912016-09-01 16:26:02 -0700730 elif self._DeleteProject(project.worktree):
Mike Frysingera850ca22019-08-07 17:19:24 -0400731 return 1
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700732
Shawn O. Pearce9fb29ce2009-06-04 20:41:02 -0700733 new_project_paths.sort()
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700734 fd = open(file_path, 'w')
735 try:
736 fd.write('\n'.join(new_project_paths))
Shawn O. Pearce3a68bb42009-06-04 16:18:09 -0700737 fd.write('\n')
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700738 finally:
739 fd.close()
740 return 0
741
Mike Frysingerae6cb082019-08-27 01:10:59 -0400742 def ValidateOptions(self, opt, args):
743 if opt.force_broken:
744 print('warning: -f/--force-broken is now the default behavior, and the '
745 'options are deprecated', file=sys.stderr)
746 if opt.network_only and opt.detach_head:
747 self.OptionParser.error('cannot combine -n and -d')
748 if opt.network_only and opt.local_only:
749 self.OptionParser.error('cannot combine -n and -l')
750 if opt.manifest_name and opt.smart_sync:
751 self.OptionParser.error('cannot combine -m and -s')
752 if opt.manifest_name and opt.smart_tag:
753 self.OptionParser.error('cannot combine -m and -t')
754 if opt.manifest_server_username or opt.manifest_server_password:
755 if not (opt.smart_sync or opt.smart_tag):
756 self.OptionParser.error('-u and -p may only be combined with -s or -t')
757 if None in [opt.manifest_server_username, opt.manifest_server_password]:
758 self.OptionParser.error('both -u and -p must be given')
759
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700760 def Execute(self, opt, args):
Roy Lee18afd7f2010-05-09 04:32:08 +0800761 if opt.jobs:
762 self.jobs = opt.jobs
Shawn O. Pearce97d2b2f2011-09-22 17:23:41 -0700763 if self.jobs > 1:
764 soft_limit, _ = _rlimit_nofile()
Mike Frysinger0c0e9342019-06-13 12:42:39 -0400765 self.jobs = min(self.jobs, (soft_limit - 5) // 3)
Shawn O. Pearce97d2b2f2011-09-22 17:23:41 -0700766
Chris Wolfee9dc3b32012-01-26 11:36:18 -0500767 if opt.manifest_name:
768 self.manifest.Override(opt.manifest_name)
Shawn O. Pearce3e768c92009-04-10 16:59:36 -0700769
Chirayu Desaia892b102013-06-11 14:18:46 +0530770 manifest_name = opt.manifest_name
David Pursehouse59b41742015-05-07 14:36:09 +0900771 smart_sync_manifest_name = "smart_sync_override.xml"
772 smart_sync_manifest_path = os.path.join(
773 self.manifest.manifestProject.worktree, smart_sync_manifest_name)
Chirayu Desaia892b102013-06-11 14:18:46 +0530774
Victor Boivie08c880d2011-04-19 10:32:52 +0200775 if opt.smart_sync or opt.smart_tag:
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700776 if not self.manifest.manifest_server:
David Pursehouse2f9e7e42013-03-05 17:26:46 +0900777 print('error: cannot smart sync: no manifest server defined in '
Sarah Owenscecd1d82012-11-01 22:59:27 -0700778 'manifest', file=sys.stderr)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700779 sys.exit(1)
David Pursehouse86d973d2012-08-24 10:21:02 +0900780
781 manifest_server = self.manifest.manifest_server
David Pursehousefb99c712013-09-25 11:09:34 +0900782 if not opt.quiet:
783 print('Using manifest server %s' % manifest_server)
David Pursehousecf76b1b2012-09-14 10:31:42 +0900784
David Pursehouse86d973d2012-08-24 10:21:02 +0900785 if not '@' in manifest_server:
David Pursehousecf76b1b2012-09-14 10:31:42 +0900786 username = None
787 password = None
788 if opt.manifest_server_username and opt.manifest_server_password:
789 username = opt.manifest_server_username
790 password = opt.manifest_server_password
David Pursehouse86d973d2012-08-24 10:21:02 +0900791 else:
792 try:
David Pursehousecf76b1b2012-09-14 10:31:42 +0900793 info = netrc.netrc()
794 except IOError:
David Pursehouseba7bc732015-08-20 16:55:42 +0900795 # .netrc file does not exist or could not be opened
796 pass
David Pursehouse86d973d2012-08-24 10:21:02 +0900797 else:
David Pursehousecf76b1b2012-09-14 10:31:42 +0900798 try:
Chirayu Desaidb2ad9d2013-06-11 13:42:25 +0530799 parse_result = urllib.parse.urlparse(manifest_server)
David Pursehousecf76b1b2012-09-14 10:31:42 +0900800 if parse_result.hostname:
David Pursehouseba7bc732015-08-20 16:55:42 +0900801 auth = info.authenticators(parse_result.hostname)
802 if auth:
803 username, _account, password = auth
804 else:
805 print('No credentials found for %s in .netrc'
806 % parse_result.hostname, file=sys.stderr)
Sarah Owensa5be53f2012-09-09 15:37:57 -0700807 except netrc.NetrcParseError as e:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700808 print('Error parsing .netrc file: %s' % e, file=sys.stderr)
David Pursehousecf76b1b2012-09-14 10:31:42 +0900809
810 if (username and password):
811 manifest_server = manifest_server.replace('://', '://%s:%s@' %
812 (username, password),
813 1)
David Pursehouse86d973d2012-08-24 10:21:02 +0900814
Dan Willemsen0745bb22015-08-17 13:41:45 -0700815 transport = PersistentTransport(manifest_server)
816 if manifest_server.startswith('persistent-'):
817 manifest_server = manifest_server[len('persistent-'):]
818
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700819 try:
Dan Willemsen0745bb22015-08-17 13:41:45 -0700820 server = xmlrpc.client.Server(manifest_server, transport=transport)
Victor Boivie08c880d2011-04-19 10:32:52 +0200821 if opt.smart_sync:
822 p = self.manifest.manifestProject
823 b = p.GetBranch(p.CurrentBranch)
824 branch = b.merge
825 if branch.startswith(R_HEADS):
826 branch = branch[len(R_HEADS):]
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700827
Victor Boivie08c880d2011-04-19 10:32:52 +0200828 env = os.environ.copy()
Jeff Davidson5cf16602014-10-02 10:13:38 -0700829 if 'SYNC_TARGET' in env:
830 target = env['SYNC_TARGET']
831 [success, manifest_str] = server.GetApprovedManifest(branch, target)
832 elif 'TARGET_PRODUCT' in env and 'TARGET_BUILD_VARIANT' in env:
Victor Boivie08c880d2011-04-19 10:32:52 +0200833 target = '%s-%s' % (env['TARGET_PRODUCT'],
834 env['TARGET_BUILD_VARIANT'])
835 [success, manifest_str] = server.GetApprovedManifest(branch, target)
836 else:
837 [success, manifest_str] = server.GetApprovedManifest(branch)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700838 else:
Victor Boivie08c880d2011-04-19 10:32:52 +0200839 assert(opt.smart_tag)
840 [success, manifest_str] = server.GetManifest(opt.smart_tag)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700841
842 if success:
David Pursehouse59b41742015-05-07 14:36:09 +0900843 manifest_name = smart_sync_manifest_name
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700844 try:
David Pursehouse59b41742015-05-07 14:36:09 +0900845 f = open(smart_sync_manifest_path, 'w')
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700846 try:
847 f.write(manifest_str)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700848 finally:
849 f.close()
David Pursehouse727cc3e2015-05-07 14:16:49 +0900850 except IOError as e:
851 print('error: cannot write manifest to %s:\n%s'
David Pursehouse59b41742015-05-07 14:36:09 +0900852 % (smart_sync_manifest_path, e),
Sarah Owenscecd1d82012-11-01 22:59:27 -0700853 file=sys.stderr)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700854 sys.exit(1)
Victor Boivie53a6c5d2013-03-19 12:20:52 +0100855 self._ReloadManifest(manifest_name)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700856 else:
David Pursehouse351fe2c2013-09-25 17:54:26 +0900857 print('error: manifest server RPC call failed: %s' %
858 manifest_str, file=sys.stderr)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700859 sys.exit(1)
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530860 except (socket.error, IOError, xmlrpc.client.Fault) as e:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700861 print('error: cannot connect to manifest server %s:\n%s'
862 % (self.manifest.manifest_server, e), file=sys.stderr)
David Pursehousebd489c42012-08-23 10:21:26 +0900863 sys.exit(1)
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530864 except xmlrpc.client.ProtocolError as e:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700865 print('error: cannot connect to manifest server %s:\n%d %s'
866 % (self.manifest.manifest_server, e.errcode, e.errmsg),
867 file=sys.stderr)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700868 sys.exit(1)
David Pursehouse59b41742015-05-07 14:36:09 +0900869 else: # Not smart sync or smart tag mode
870 if os.path.isfile(smart_sync_manifest_path):
871 try:
Renaud Paquay010fed72016-11-11 14:25:29 -0800872 platform_utils.remove(smart_sync_manifest_path)
David Pursehouse59b41742015-05-07 14:36:09 +0900873 except OSError as e:
874 print('error: failed to remove existing smart sync override manifest: %s' %
875 e, file=sys.stderr)
Nico Sallembiena1bfd2c2010-04-06 10:40:01 -0700876
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700877 rp = self.manifest.repoProject
878 rp.PreSync()
879
880 mp = self.manifest.manifestProject
881 mp.PreSync()
882
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800883 if opt.repo_upgraded:
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -0700884 _PostRepoUpgrade(self.manifest, quiet=opt.quiet)
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -0800885
Nico Sallembien9bb18162009-12-07 15:38:01 -0800886 if not opt.local_only:
David Rileye0684ad2017-04-05 00:02:59 -0700887 start = time.time()
888 success = mp.Sync_NetworkHalf(quiet=opt.quiet,
889 current_branch_only=opt.current_branch_only,
890 no_tags=opt.no_tags,
891 optimized_fetch=opt.optimized_fetch,
Xin Li745be2e2019-06-03 11:24:30 -0700892 submodules=self.manifest.HasSubmodules,
893 clone_filter=self.manifest.CloneFilter)
David Rileye0684ad2017-04-05 00:02:59 -0700894 finish = time.time()
895 self.event_log.AddSync(mp, event_log.TASK_SYNC_NETWORK,
896 start, finish, success)
Nico Sallembien9bb18162009-12-07 15:38:01 -0800897
898 if mp.HasChanges:
899 syncbuf = SyncBuffer(mp.config)
David Rileye0684ad2017-04-05 00:02:59 -0700900 start = time.time()
Martin Kellye4e94d22017-03-21 16:05:12 -0700901 mp.Sync_LocalHalf(syncbuf, submodules=self.manifest.HasSubmodules)
David Rileye0684ad2017-04-05 00:02:59 -0700902 clean = syncbuf.Finish()
903 self.event_log.AddSync(mp, event_log.TASK_SYNC_LOCAL,
904 start, time.time(), clean)
905 if not clean:
Nico Sallembien9bb18162009-12-07 15:38:01 -0800906 sys.exit(1)
Victor Boivie53a6c5d2013-03-19 12:20:52 +0100907 self._ReloadManifest(manifest_name)
Shawn O. Pearcec4657962011-09-26 09:08:01 -0700908 if opt.jobs is None:
909 self.jobs = self.manifest.default.sync_j
Simran Basib9a1b732015-08-20 12:19:28 -0700910
Simran Basib9a1b732015-08-20 12:19:28 -0700911 if self.gitc_manifest:
912 gitc_manifest_projects = self.GetProjects(args,
Simran Basib9a1b732015-08-20 12:19:28 -0700913 missing_ok=True)
914 gitc_projects = []
915 opened_projects = []
916 for project in gitc_manifest_projects:
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700917 if project.relpath in self.gitc_manifest.paths and \
918 self.gitc_manifest.paths[project.relpath].old_revision:
919 opened_projects.append(project.relpath)
Simran Basib9a1b732015-08-20 12:19:28 -0700920 else:
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700921 gitc_projects.append(project.relpath)
Simran Basib9a1b732015-08-20 12:19:28 -0700922
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700923 if not args:
924 gitc_projects = None
925
926 if gitc_projects != [] and not opt.local_only:
Simran Basib9a1b732015-08-20 12:19:28 -0700927 print('Updating GITC client: %s' % self.gitc_manifest.gitc_client_name)
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700928 manifest = GitcManifest(self.repodir, self.gitc_manifest.gitc_client_name)
929 if manifest_name:
930 manifest.Override(manifest_name)
931 else:
932 manifest.Override(self.manifest.manifestFile)
933 gitc_utils.generate_gitc_manifest(self.gitc_manifest,
934 manifest,
Simran Basib9a1b732015-08-20 12:19:28 -0700935 gitc_projects)
936 print('GITC client successfully synced.')
937
938 # The opened projects need to be synced as normal, therefore we
939 # generate a new args list to represent the opened projects.
Dan Willemsen5ea32d12015-09-08 13:27:20 -0700940 # TODO: make this more reliable -- if there's a project name/path overlap,
941 # this may choose the wrong project.
David Pursehouse3bcd3052017-07-10 22:42:22 +0900942 args = [os.path.relpath(self.manifest.paths[path].worktree, os.getcwd())
943 for path in opened_projects]
Simran Basib9a1b732015-08-20 12:19:28 -0700944 if not args:
945 return
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800946 all_projects = self.GetProjects(args,
947 missing_ok=True,
948 submodules_ok=opt.fetch_submodules)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700949
Dave Borowitz67700e92012-10-23 15:00:54 -0700950 self._fetch_times = _FetchTimes(self.manifest)
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700951 if not opt.local_only:
Shawn O. Pearcef6906872009-04-18 10:49:00 -0700952 to_fetch = []
953 now = time.time()
Dave Borowitz67700e92012-10-23 15:00:54 -0700954 if _ONE_DAY_S <= (now - rp.LastFetch):
Shawn O. Pearcef6906872009-04-18 10:49:00 -0700955 to_fetch.append(rp)
David Pursehouse8a68ff92012-09-24 12:15:13 +0900956 to_fetch.extend(all_projects)
Dave Borowitz67700e92012-10-23 15:00:54 -0700957 to_fetch.sort(key=self._fetch_times.Get, reverse=True)
Shawn O. Pearcef6906872009-04-18 10:49:00 -0700958
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800959 fetched = self._Fetch(to_fetch, opt)
Shawn O. Pearcee756c412009-04-13 11:51:15 -0700960 _PostRepoFetch(rp, opt.no_repo_verify)
Shawn O. Pearceb1562fa2009-04-10 17:04:08 -0700961 if opt.network_only:
962 # bail out now; the rest touches the working tree
963 return
964
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800965 # Iteratively fetch missing and/or nested unregistered submodules
966 previously_missing_set = set()
967 while True:
Victor Boivie53a6c5d2013-03-19 12:20:52 +0100968 self._ReloadManifest(manifest_name)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800969 all_projects = self.GetProjects(args,
970 missing_ok=True,
971 submodules_ok=opt.fetch_submodules)
972 missing = []
973 for project in all_projects:
974 if project.gitdir not in fetched:
975 missing.append(project)
976 if not missing:
977 break
978 # Stop us from non-stopped fetching actually-missing repos: If set of
979 # missing repos has not been changed from last fetch, we break.
980 missing_set = set(p.name for p in missing)
981 if previously_missing_set == missing_set:
982 break
983 previously_missing_set = missing_set
984 fetched.update(self._Fetch(missing, opt))
985
Julien Campergue335f5ef2013-10-16 11:02:35 +0200986 if self.manifest.IsMirror or self.manifest.IsArchive:
Shawn O. Pearcecd1d7ff2009-06-04 16:15:53 -0700987 # bail out now, we have no working tree
988 return
989
Oleksii Okolielovd3c0f592018-12-17 19:23:44 -0500990 if self.UpdateProjectList(opt):
Jaikumar Ganesh4f2517f2009-06-01 21:10:33 -0700991 sys.exit(1)
992
Xin Li745be2e2019-06-03 11:24:30 -0700993 self._Checkout(all_projects, opt)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700994
Doug Anderson2b8db3c2010-11-01 15:08:06 -0700995 # If there's a notice that's supposed to print at the end of the sync, print
996 # it now...
997 if self.manifest.notice:
Sarah Owenscecd1d82012-11-01 22:59:27 -0700998 print(self.manifest.notice)
Doug Anderson2b8db3c2010-11-01 15:08:06 -0700999
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -07001000def _PostRepoUpgrade(manifest, quiet=False):
Conley Owens094cdbe2014-01-30 15:09:59 -08001001 wrapper = Wrapper()
Conley Owensc9129d92012-10-01 16:12:28 -07001002 if wrapper.NeedSetupGnuPG():
Shawn O. Pearce80d2ceb2012-10-26 12:23:05 -07001003 wrapper.SetupGnuPG(quiet)
Conley Owensf2fe2d92014-01-29 13:53:43 -08001004 for project in manifest.projects:
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001005 if project.Exists:
1006 project.PostRepoUpgrade()
1007
1008def _PostRepoFetch(rp, no_repo_verify=False, verbose=False):
1009 if rp.HasChanges:
Sarah Owenscecd1d82012-11-01 22:59:27 -07001010 print('info: A new version of repo is available', file=sys.stderr)
1011 print(file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001012 if no_repo_verify or _VerifyTag(rp):
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001013 syncbuf = SyncBuffer(rp.config)
1014 rp.Sync_LocalHalf(syncbuf)
1015 if not syncbuf.Finish():
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001016 sys.exit(1)
Sarah Owenscecd1d82012-11-01 22:59:27 -07001017 print('info: Restarting repo with latest version', file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001018 raise RepoChangedException(['--repo-upgraded'])
1019 else:
Sarah Owenscecd1d82012-11-01 22:59:27 -07001020 print('warning: Skipped upgrade to unverified version', file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001021 else:
1022 if verbose:
Sarah Owenscecd1d82012-11-01 22:59:27 -07001023 print('repo version %s is current' % rp.work_git.describe(HEAD),
1024 file=sys.stderr)
Shawn O. Pearcee756c412009-04-13 11:51:15 -07001025
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001026def _VerifyTag(project):
1027 gpg_dir = os.path.expanduser('~/.repoconfig/gnupg')
1028 if not os.path.exists(gpg_dir):
Sarah Owenscecd1d82012-11-01 22:59:27 -07001029 print('warning: GnuPG was not available during last "repo init"\n'
1030 'warning: Cannot automatically authenticate repo."""',
1031 file=sys.stderr)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001032 return True
1033
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001034 try:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001035 cur = project.bare_git.describe(project.GetRevisionId())
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001036 except GitError:
1037 cur = None
1038
1039 if not cur \
1040 or re.compile(r'^.*-[0-9]{1,}-g[0-9a-f]{1,}$').match(cur):
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001041 rev = project.revisionExpr
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001042 if rev.startswith(R_HEADS):
1043 rev = rev[len(R_HEADS):]
1044
Sarah Owenscecd1d82012-11-01 22:59:27 -07001045 print(file=sys.stderr)
1046 print("warning: project '%s' branch '%s' is not signed"
1047 % (project.name, rev), file=sys.stderr)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001048 return False
1049
Shawn O. Pearcef18cb762010-12-07 11:41:05 -08001050 env = os.environ.copy()
1051 env['GIT_DIR'] = project.gitdir.encode()
1052 env['GNUPGHOME'] = gpg_dir.encode()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001053
1054 cmd = [GIT, 'tag', '-v', cur]
1055 proc = subprocess.Popen(cmd,
1056 stdout = subprocess.PIPE,
1057 stderr = subprocess.PIPE,
1058 env = env)
1059 out = proc.stdout.read()
1060 proc.stdout.close()
1061
1062 err = proc.stderr.read()
1063 proc.stderr.close()
1064
1065 if proc.wait() != 0:
Sarah Owenscecd1d82012-11-01 22:59:27 -07001066 print(file=sys.stderr)
1067 print(out, file=sys.stderr)
1068 print(err, file=sys.stderr)
1069 print(file=sys.stderr)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001070 return False
1071 return True
Dave Borowitz67700e92012-10-23 15:00:54 -07001072
David Rileye0684ad2017-04-05 00:02:59 -07001073
Dave Borowitz67700e92012-10-23 15:00:54 -07001074class _FetchTimes(object):
Dave Borowitzd9478582012-10-23 16:35:39 -07001075 _ALPHA = 0.5
1076
Dave Borowitz67700e92012-10-23 15:00:54 -07001077 def __init__(self, manifest):
Anthony King85b24ac2014-05-06 15:57:48 +01001078 self._path = os.path.join(manifest.repodir, '.repo_fetchtimes.json')
Dave Borowitz67700e92012-10-23 15:00:54 -07001079 self._times = None
Dave Borowitzd9478582012-10-23 16:35:39 -07001080 self._seen = set()
Dave Borowitz67700e92012-10-23 15:00:54 -07001081
1082 def Get(self, project):
1083 self._Load()
1084 return self._times.get(project.name, _ONE_DAY_S)
1085
1086 def Set(self, project, t):
Dave Borowitzd9478582012-10-23 16:35:39 -07001087 self._Load()
1088 name = project.name
1089 old = self._times.get(name, t)
1090 self._seen.add(name)
1091 a = self._ALPHA
1092 self._times[name] = (a*t) + ((1-a) * old)
Dave Borowitz67700e92012-10-23 15:00:54 -07001093
1094 def _Load(self):
1095 if self._times is None:
1096 try:
Anthony King85b24ac2014-05-06 15:57:48 +01001097 f = open(self._path)
Dave Borowitz67700e92012-10-23 15:00:54 -07001098 try:
Anthony King85b24ac2014-05-06 15:57:48 +01001099 self._times = json.load(f)
1100 finally:
1101 f.close()
1102 except (IOError, ValueError):
1103 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08001104 platform_utils.remove(self._path)
Anthony King85b24ac2014-05-06 15:57:48 +01001105 except OSError:
1106 pass
1107 self._times = {}
Dave Borowitz67700e92012-10-23 15:00:54 -07001108
1109 def Save(self):
1110 if self._times is None:
1111 return
Dave Borowitzd9478582012-10-23 16:35:39 -07001112
1113 to_delete = []
1114 for name in self._times:
1115 if name not in self._seen:
1116 to_delete.append(name)
1117 for name in to_delete:
1118 del self._times[name]
1119
Dave Borowitz67700e92012-10-23 15:00:54 -07001120 try:
Anthony King85b24ac2014-05-06 15:57:48 +01001121 f = open(self._path, 'w')
Dave Borowitz67700e92012-10-23 15:00:54 -07001122 try:
Anthony King85b24ac2014-05-06 15:57:48 +01001123 json.dump(self._times, f, indent=2)
1124 finally:
1125 f.close()
1126 except (IOError, TypeError):
1127 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08001128 platform_utils.remove(self._path)
Anthony King85b24ac2014-05-06 15:57:48 +01001129 except OSError:
1130 pass
Dan Willemsen0745bb22015-08-17 13:41:45 -07001131
1132# This is a replacement for xmlrpc.client.Transport using urllib2
1133# and supporting persistent-http[s]. It cannot change hosts from
1134# request to request like the normal transport, the real url
1135# is passed during initialization.
1136class PersistentTransport(xmlrpc.client.Transport):
1137 def __init__(self, orig_host):
1138 self.orig_host = orig_host
1139
1140 def request(self, host, handler, request_body, verbose=False):
1141 with GetUrlCookieFile(self.orig_host, not verbose) as (cookiefile, proxy):
1142 # Python doesn't understand cookies with the #HttpOnly_ prefix
1143 # Since we're only using them for HTTP, copy the file temporarily,
1144 # stripping those prefixes away.
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001145 if cookiefile:
1146 tmpcookiefile = tempfile.NamedTemporaryFile()
David Pursehouse4c5f74e2015-10-02 11:10:10 +09001147 tmpcookiefile.write("# HTTP Cookie File")
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001148 try:
1149 with open(cookiefile) as f:
1150 for line in f:
1151 if line.startswith("#HttpOnly_"):
1152 line = line[len("#HttpOnly_"):]
1153 tmpcookiefile.write(line)
1154 tmpcookiefile.flush()
Dan Willemsen0745bb22015-08-17 13:41:45 -07001155
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001156 cookiejar = cookielib.MozillaCookieJar(tmpcookiefile.name)
David Pursehouseb1ad2192015-09-30 10:35:43 +09001157 try:
1158 cookiejar.load()
1159 except cookielib.LoadError:
1160 cookiejar = cookielib.CookieJar()
Dan Willemsen3010e5b2015-08-20 10:09:20 -07001161 finally:
1162 tmpcookiefile.close()
1163 else:
1164 cookiejar = cookielib.CookieJar()
Dan Willemsen0745bb22015-08-17 13:41:45 -07001165
1166 proxyhandler = urllib.request.ProxyHandler
1167 if proxy:
1168 proxyhandler = urllib.request.ProxyHandler({
1169 "http": proxy,
1170 "https": proxy })
1171
1172 opener = urllib.request.build_opener(
1173 urllib.request.HTTPCookieProcessor(cookiejar),
1174 proxyhandler)
1175
1176 url = urllib.parse.urljoin(self.orig_host, handler)
1177 parse_results = urllib.parse.urlparse(url)
1178
1179 scheme = parse_results.scheme
1180 if scheme == 'persistent-http':
1181 scheme = 'http'
1182 if scheme == 'persistent-https':
1183 # If we're proxying through persistent-https, use http. The
1184 # proxy itself will do the https.
1185 if proxy:
1186 scheme = 'http'
1187 else:
1188 scheme = 'https'
1189
1190 # Parse out any authentication information using the base class
1191 host, extra_headers, _ = self.get_host_info(parse_results.netloc)
1192
1193 url = urllib.parse.urlunparse((
1194 scheme,
1195 host,
1196 parse_results.path,
1197 parse_results.params,
1198 parse_results.query,
1199 parse_results.fragment))
1200
1201 request = urllib.request.Request(url, request_body)
1202 if extra_headers is not None:
1203 for (name, header) in extra_headers:
1204 request.add_header(name, header)
1205 request.add_header('Content-Type', 'text/xml')
1206 try:
1207 response = opener.open(request)
1208 except urllib.error.HTTPError as e:
1209 if e.code == 501:
1210 # We may have been redirected through a login process
1211 # but our POST turned into a GET. Retry.
1212 response = opener.open(request)
1213 else:
1214 raise
1215
1216 p, u = xmlrpc.client.getparser()
1217 while 1:
1218 data = response.read(1024)
1219 if not data:
1220 break
1221 p.feed(data)
1222 p.close()
1223 return u.close()
1224
1225 def close(self):
1226 pass
1227