~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/check.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-10-13 07:04:50 UTC
  • mfrom: (5447.2.2 work)
  • Revision ID: pqm@pqm.ubuntu.com-20101013070450-xmn9cpnli5qnmrt8
(vila) Tweak the release process based on ML discussion (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2004, 2005 by Martin Pool
2
 
# Copyright (C) 2005 by Canonical Ltd
3
 
 
 
1
# Copyright (C) 2005, 2006 Canonical Ltd
 
2
#
4
3
# This program is free software; you can redistribute it and/or modify
5
4
# it under the terms of the GNU General Public License as published by
6
5
# the Free Software Foundation; either version 2 of the License, or
7
6
# (at your option) any later version.
8
 
 
 
7
#
9
8
# This program is distributed in the hope that it will be useful,
10
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
11
# GNU General Public License for more details.
13
 
 
 
12
#
14
13
# You should have received a copy of the GNU General Public License
15
14
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
16
 
18
17
# TODO: Check ancestries are correct for every revision: includes
19
18
# every committed so far, and in a reasonable order.
22
21
 
23
22
# TODO: Check for extra files in the control directory.
24
23
 
25
 
# TODO: Check revision, inventory and entry objects have all 
 
24
# TODO: Check revision, inventory and entry objects have all
26
25
# required fields.
27
26
 
28
27
# TODO: Get every revision in the revision-store even if they're not
29
28
# referenced by history and make sure they're all valid.
30
29
 
 
30
# TODO: Perhaps have a way to record errors other than by raising exceptions;
 
31
# would perhaps be enough to accumulate exception objects in a list without
 
32
# raising them.  If there's more than one exception it'd be good to see them
 
33
# all.
 
34
 
 
35
"""Checking of bzr objects.
 
36
 
 
37
check_refs is a concept used for optimising check. Objects that depend on other
 
38
objects (e.g. tree on repository) can list the objects they would be requesting
 
39
so that when the dependent object is checked, matches can be pulled out and
 
40
evaluated in-line rather than re-reading the same data many times.
 
41
check_refs are tuples (kind, value). Currently defined kinds are:
 
42
* 'trees', where value is a revid and the looked up objects are revision trees.
 
43
* 'lefthand-distance', where value is a revid and the looked up objects are the
 
44
  distance along the lefthand path to NULL for that revid.
 
45
* 'revision-existence', where value is a revid, and the result is True or False
 
46
  indicating that the revision was found/not found.
 
47
"""
 
48
 
 
49
from bzrlib import errors
 
50
from bzrlib.branch import Branch
 
51
from bzrlib.bzrdir import BzrDir
 
52
from bzrlib.revision import NULL_REVISION
 
53
from bzrlib.symbol_versioning import deprecated_function, deprecated_in
 
54
from bzrlib.trace import note
31
55
import bzrlib.ui
32
 
from bzrlib.trace import note, warning
33
 
from bzrlib.osutils import rename, sha_string, fingerprint_file
34
 
from bzrlib.trace import mutter
35
 
from bzrlib.errors import BzrCheckError, NoSuchRevision
36
 
from bzrlib.inventory import ROOT_ID
37
 
from bzrlib.branch import gen_root_id
38
 
 
 
56
from bzrlib.workingtree import WorkingTree
39
57
 
40
58
class Check(object):
41
 
    """Check a branch"""
42
 
 
43
 
    def __init__(self, branch):
44
 
        self.branch = branch
45
 
        self.inventory_weave = branch._get_inventory_weave()
46
 
        self.checked_text_cnt = 0
 
59
    """Check a repository"""
 
60
 
 
61
    # The Check object interacts with InventoryEntry.check, etc.
 
62
 
 
63
    def __init__(self, repository, check_repo=True):
 
64
        self.repository = repository
47
65
        self.checked_rev_cnt = 0
48
 
        self.ghosts = []
49
 
        self.repeated_text_cnt = 0
 
66
        self.ghosts = set()
50
67
        self.missing_parent_links = {}
51
68
        self.missing_inventory_sha_cnt = 0
52
69
        self.missing_revision_cnt = 0
53
 
        # maps (file-id, version) -> sha1
54
 
        self.checked_texts = {}
 
70
        self.checked_weaves = set()
 
71
        self.unreferenced_versions = set()
 
72
        self.inconsistent_parents = []
 
73
        self.rich_roots = repository.supports_rich_root()
 
74
        self.text_key_references = {}
 
75
        self.check_repo = check_repo
 
76
        self.other_results = []
 
77
        # Plain text lines to include in the report
 
78
        self._report_items = []
 
79
        # Keys we are looking for; may be large and need spilling to disk.
 
80
        # key->(type(revision/inventory/text/signature/map), sha1, first-referer)
 
81
        self.pending_keys = {}
 
82
        # Ancestors map for all of revisions being checked; while large helper
 
83
        # functions we call would create it anyway, so better to have once and
 
84
        # keep.
 
85
        self.ancestors = {}
55
86
 
56
 
    def check(self):
57
 
        self.branch.lock_read()
58
 
        self.progress = bzrlib.ui.ui_factory.progress_bar()
 
87
    def check(self, callback_refs=None, check_repo=True):
 
88
        if callback_refs is None:
 
89
            callback_refs = {}
 
90
        self.repository.lock_read()
 
91
        self.progress = bzrlib.ui.ui_factory.nested_progress_bar()
59
92
        try:
60
 
            self.history = self.branch.revision_history()
61
 
            if not len(self.history):
62
 
                # nothing to see here
63
 
                return
64
 
            if not self.branch.revision_store.listable():
65
 
                raise BzrCheckError("Branch must be local")
66
 
            self.planned_revisions = set(self.branch.revision_store)
67
 
            inventoried = set(self.inventory_weave.names())
68
 
            awol = self.planned_revisions - inventoried
69
 
            if len(awol) > 0:
70
 
                raise BzrCheckError('Stored revisions missing from inventory'
71
 
                    '{%s}' % ','.join([f for f in awol]))
72
 
    
73
 
            for revno, rev_id in enumerate(self.planned_revisions):
74
 
                self.progress.update('checking revision', revno+1,
75
 
                                     len(self.planned_revisions))
76
 
                self.check_one_rev(rev_id)
 
93
            self.progress.update('check', 0, 4)
 
94
            if self.check_repo:
 
95
                self.progress.update('checking revisions', 0)
 
96
                self.check_revisions()
 
97
                self.progress.update('checking commit contents', 1)
 
98
                self.repository._check_inventories(self)
 
99
                self.progress.update('checking file graphs', 2)
 
100
                # check_weaves is done after the revision scan so that
 
101
                # revision index is known to be valid.
 
102
                self.check_weaves()
 
103
            self.progress.update('checking branches and trees', 3)
 
104
            if callback_refs:
 
105
                repo = self.repository
 
106
                # calculate all refs, and callback the objects requesting them.
 
107
                refs = {}
 
108
                wanting_items = set()
 
109
                # Current crude version calculates everything and calls
 
110
                # everything at once. Doing a queue and popping as things are
 
111
                # satisfied would be cheaper on memory [but few people have
 
112
                # huge numbers of working trees today. TODO: fix before
 
113
                # landing].
 
114
                distances = set()
 
115
                existences = set()
 
116
                for ref, wantlist in callback_refs.iteritems():
 
117
                    wanting_items.update(wantlist)
 
118
                    kind, value = ref
 
119
                    if kind == 'trees':
 
120
                        refs[ref] = repo.revision_tree(value)
 
121
                    elif kind == 'lefthand-distance':
 
122
                        distances.add(value)
 
123
                    elif kind == 'revision-existence':
 
124
                        existences.add(value)
 
125
                    else:
 
126
                        raise AssertionError(
 
127
                            'unknown ref kind for ref %s' % ref)
 
128
                node_distances = repo.get_graph().find_lefthand_distances(distances)
 
129
                for key, distance in node_distances.iteritems():
 
130
                    refs[('lefthand-distance', key)] = distance
 
131
                    if key in existences and distance > 0:
 
132
                        refs[('revision-existence', key)] = True
 
133
                        existences.remove(key)
 
134
                parent_map = repo.get_graph().get_parent_map(existences)
 
135
                for key in parent_map:
 
136
                    refs[('revision-existence', key)] = True
 
137
                    existences.remove(key)
 
138
                for key in existences:
 
139
                    refs[('revision-existence', key)] = False
 
140
                for item in wanting_items:
 
141
                    if isinstance(item, WorkingTree):
 
142
                        item._check(refs)
 
143
                    if isinstance(item, Branch):
 
144
                        self.other_results.append(item.check(refs))
77
145
        finally:
78
 
            self.progress.clear()
79
 
            self.branch.unlock()
 
146
            self.progress.finished()
 
147
            self.repository.unlock()
 
148
 
 
149
    def _check_revisions(self, revisions_iterator):
 
150
        """Check revision objects by decorating a generator.
 
151
 
 
152
        :param revisions_iterator: An iterator of(revid, Revision-or-None).
 
153
        :return: A generator of the contents of revisions_iterator.
 
154
        """
 
155
        self.planned_revisions = set()
 
156
        for revid, revision in revisions_iterator:
 
157
            yield revid, revision
 
158
            self._check_one_rev(revid, revision)
 
159
        # Flatten the revisions we found to guarantee consistent later
 
160
        # iteration.
 
161
        self.planned_revisions = list(self.planned_revisions)
 
162
        # TODO: extract digital signatures as items to callback on too.
 
163
 
 
164
    def check_revisions(self):
 
165
        """Scan revisions, checking data directly available as we go."""
 
166
        revision_iterator = self.repository._iter_revisions(None)
 
167
        revision_iterator = self._check_revisions(revision_iterator)
 
168
        # We read the all revisions here:
 
169
        # - doing this allows later code to depend on the revision index.
 
170
        # - we can fill out existence flags at this point
 
171
        # - we can read the revision inventory sha at this point
 
172
        # - we can check properties and serialisers etc.
 
173
        if not self.repository.revision_graph_can_have_wrong_parents():
 
174
            # The check against the index isn't needed.
 
175
            self.revs_with_bad_parents_in_index = None
 
176
            for thing in revision_iterator:
 
177
                pass
 
178
        else:
 
179
            bad_revisions = self.repository._find_inconsistent_revision_parents(
 
180
                revision_iterator)
 
181
            self.revs_with_bad_parents_in_index = list(bad_revisions)
80
182
 
81
183
    def report_results(self, verbose):
82
 
        note('checked branch %s format %d',
83
 
             self.branch.base, 
84
 
             self.branch._branch_format)
 
184
        if self.check_repo:
 
185
            self._report_repo_results(verbose)
 
186
        for result in self.other_results:
 
187
            result.report_results(verbose)
85
188
 
 
189
    def _report_repo_results(self, verbose):
 
190
        note('checked repository %s format %s',
 
191
            self.repository.user_url,
 
192
            self.repository._format)
86
193
        note('%6d revisions', self.checked_rev_cnt)
87
 
        note('%6d unique file texts', self.checked_text_cnt)
88
 
        note('%6d repeated file texts', self.repeated_text_cnt)
 
194
        note('%6d file-ids', len(self.checked_weaves))
 
195
        if verbose:
 
196
            note('%6d unreferenced text versions',
 
197
                len(self.unreferenced_versions))
 
198
        if verbose and len(self.unreferenced_versions):
 
199
                for file_id, revision_id in self.unreferenced_versions:
 
200
                    note('unreferenced version: {%s} in %s', revision_id,
 
201
                        file_id)
89
202
        if self.missing_inventory_sha_cnt:
90
203
            note('%6d revisions are missing inventory_sha1',
91
204
                 self.missing_inventory_sha_cnt)
98
211
                for ghost in self.ghosts:
99
212
                    note('      %s', ghost)
100
213
        if len(self.missing_parent_links):
101
 
            note('%6d revisions missing parents in ancestry', 
 
214
            note('%6d revisions missing parents in ancestry',
102
215
                 len(self.missing_parent_links))
103
216
            if verbose:
104
217
                for link, linkers in self.missing_parent_links.items():
105
218
                    note('      %s should be in the ancestry for:', link)
106
219
                    for linker in linkers:
107
220
                        note('       * %s', linker)
108
 
 
109
 
    def check_one_rev(self, rev_id):
110
 
        """Check one revision.
111
 
 
112
 
        rev_id - the one to check
113
 
 
114
 
        last_rev_id - the previous one on the mainline, if any.
 
221
        if len(self.inconsistent_parents):
 
222
            note('%6d inconsistent parents', len(self.inconsistent_parents))
 
223
            if verbose:
 
224
                for info in self.inconsistent_parents:
 
225
                    revision_id, file_id, found_parents, correct_parents = info
 
226
                    note('      * %s version %s has parents %r '
 
227
                         'but should have %r'
 
228
                         % (file_id, revision_id, found_parents,
 
229
                             correct_parents))
 
230
        if self.revs_with_bad_parents_in_index:
 
231
            note('%6d revisions have incorrect parents in the revision index',
 
232
                 len(self.revs_with_bad_parents_in_index))
 
233
            if verbose:
 
234
                for item in self.revs_with_bad_parents_in_index:
 
235
                    revision_id, index_parents, actual_parents = item
 
236
                    note(
 
237
                        '       %s has wrong parents in index: '
 
238
                        '%r should be %r',
 
239
                        revision_id, index_parents, actual_parents)
 
240
        for item in self._report_items:
 
241
            note(item)
 
242
 
 
243
    def _check_one_rev(self, rev_id, rev):
 
244
        """Cross-check one revision.
 
245
 
 
246
        :param rev_id: A revision id to check.
 
247
        :param rev: A revision or None to indicate a missing revision.
115
248
        """
116
 
 
117
 
        # mutter('    revision {%s}', rev_id)
118
 
        branch = self.branch
119
 
        try:
120
 
            rev_history_position = self.history.index(rev_id)
121
 
        except ValueError:
122
 
            rev_history_position = None
123
 
        last_rev_id = None
124
 
        if rev_history_position:
125
 
            rev = branch.get_revision(rev_id)
126
 
            if rev_history_position > 0:
127
 
                last_rev_id = self.history[rev_history_position - 1]
128
 
        else:
129
 
            rev = branch.get_revision(rev_id)
130
 
                
131
249
        if rev.revision_id != rev_id:
132
 
            raise BzrCheckError('wrong internal revision id in revision {%s}'
133
 
                                % rev_id)
134
 
 
135
 
        # check the previous history entry is a parent of this entry
136
 
        if rev.parent_ids:
137
 
            if last_rev_id is not None:
138
 
                for parent_id in rev.parent_ids:
139
 
                    if parent_id == last_rev_id:
140
 
                        break
141
 
                else:
142
 
                    raise BzrCheckError("previous revision {%s} not listed among "
143
 
                                        "parents of {%s}"
144
 
                                        % (last_rev_id, rev_id))
145
 
            for parent in rev.parent_ids:
146
 
                if not parent in self.planned_revisions:
147
 
                    missing_links = self.missing_parent_links.get(parent, [])
148
 
                    missing_links.append(rev_id)
149
 
                    self.missing_parent_links[parent] = missing_links
150
 
                    # list based so slow, TODO have a planned_revisions list and set.
151
 
                    if self.branch.has_revision(parent):
152
 
                        missing_ancestry = self.branch.get_ancestry(parent)
153
 
                        for missing in missing_ancestry:
154
 
                            if (missing is not None 
155
 
                                and missing not in self.planned_revisions):
156
 
                                self.planned_revisions.append(missing)
157
 
                    else:
158
 
                        self.ghosts.append(rev_id)
159
 
        elif last_rev_id:
160
 
            raise BzrCheckError("revision {%s} has no parents listed "
161
 
                                "but preceded by {%s}"
162
 
                                % (rev_id, last_rev_id))
163
 
 
164
 
        if rev.inventory_sha1:
165
 
            inv_sha1 = branch.get_inventory_sha1(rev_id)
166
 
            if inv_sha1 != rev.inventory_sha1:
167
 
                raise BzrCheckError('Inventory sha1 hash doesn\'t match'
168
 
                    ' value in revision {%s}' % rev_id)
169
 
        else:
170
 
            missing_inventory_sha_cnt += 1
171
 
            mutter("no inventory_sha1 on revision {%s}", rev_id)
172
 
        self._check_revision_tree(rev_id)
 
250
            self._report_items.append(
 
251
                'Mismatched internal revid {%s} and index revid {%s}' % (
 
252
                rev.revision_id, rev_id))
 
253
            rev_id = rev.revision_id
 
254
        # Check this revision tree etc, and count as seen when we encounter a
 
255
        # reference to it.
 
256
        self.planned_revisions.add(rev_id)
 
257
        # It is not a ghost
 
258
        self.ghosts.discard(rev_id)
 
259
        # Count all parents as ghosts if we haven't seen them yet.
 
260
        for parent in rev.parent_ids:
 
261
            if not parent in self.planned_revisions:
 
262
                self.ghosts.add(parent)
 
263
        
 
264
        self.ancestors[rev_id] = tuple(rev.parent_ids) or (NULL_REVISION,)
 
265
        self.add_pending_item(rev_id, ('inventories', rev_id), 'inventory',
 
266
            rev.inventory_sha1)
173
267
        self.checked_rev_cnt += 1
174
268
 
175
 
    def _check_revision_tree(self, rev_id):
176
 
        tree = self.branch.revision_tree(rev_id)
177
 
        inv = tree.inventory
178
 
        seen_ids = {}
179
 
        for file_id in inv:
180
 
            if file_id in seen_ids:
181
 
                raise BzrCheckError('duplicated file_id {%s} '
182
 
                                    'in inventory for revision {%s}'
183
 
                                    % (file_id, rev_id))
184
 
            seen_ids[file_id] = True
185
 
        for file_id in inv:
186
 
            ie = inv[file_id]
187
 
            ie.check(self, rev_id, inv, tree)
188
 
        seen_names = {}
189
 
        for path, ie in inv.iter_entries():
190
 
            if path in seen_names:
191
 
                raise BzrCheckError('duplicated path %s '
192
 
                                    'in inventory for revision {%s}'
193
 
                                    % (path, rev_id))
194
 
            seen_names[path] = True
195
 
 
196
 
 
 
269
    def add_pending_item(self, referer, key, kind, sha1):
 
270
        """Add a reference to a sha1 to be cross checked against a key.
 
271
 
 
272
        :param referer: The referer that expects key to have sha1.
 
273
        :param key: A storage key e.g. ('texts', 'foo@bar-20040504-1234')
 
274
        :param kind: revision/inventory/text/map/signature
 
275
        :param sha1: A hex sha1 or None if no sha1 is known.
 
276
        """
 
277
        existing = self.pending_keys.get(key)
 
278
        if existing:
 
279
            if sha1 != existing[1]:
 
280
                self._report_items.append('Multiple expected sha1s for %s. {%s}'
 
281
                    ' expects {%s}, {%s} expects {%s}', (
 
282
                    key, referer, sha1, existing[1], existing[0]))
 
283
        else:
 
284
            self.pending_keys[key] = (kind, sha1, referer)
 
285
 
 
286
    def check_weaves(self):
 
287
        """Check all the weaves we can get our hands on.
 
288
        """
 
289
        weave_ids = []
 
290
        storebar = bzrlib.ui.ui_factory.nested_progress_bar()
 
291
        try:
 
292
            self._check_weaves(storebar)
 
293
        finally:
 
294
            storebar.finished()
 
295
 
 
296
    def _check_weaves(self, storebar):
 
297
        storebar.update('text-index', 0, 2)
 
298
        if self.repository._format.fast_deltas:
 
299
            # We haven't considered every fileid instance so far.
 
300
            weave_checker = self.repository._get_versioned_file_checker(
 
301
                ancestors=self.ancestors)
 
302
        else:
 
303
            weave_checker = self.repository._get_versioned_file_checker(
 
304
                text_key_references=self.text_key_references,
 
305
                ancestors=self.ancestors)
 
306
        storebar.update('file-graph', 1)
 
307
        result = weave_checker.check_file_version_parents(
 
308
            self.repository.texts)
 
309
        self.checked_weaves = weave_checker.file_ids
 
310
        bad_parents, unused_versions = result
 
311
        bad_parents = bad_parents.items()
 
312
        for text_key, (stored_parents, correct_parents) in bad_parents:
 
313
            # XXX not ready for id join/split operations.
 
314
            weave_id = text_key[0]
 
315
            revision_id = text_key[-1]
 
316
            weave_parents = tuple([parent[-1] for parent in stored_parents])
 
317
            correct_parents = tuple([parent[-1] for parent in correct_parents])
 
318
            self.inconsistent_parents.append(
 
319
                (revision_id, weave_id, weave_parents, correct_parents))
 
320
        self.unreferenced_versions.update(unused_versions)
 
321
 
 
322
    def _add_entry_to_text_key_references(self, inv, entry):
 
323
        if not self.rich_roots and entry.name == '':
 
324
            return
 
325
        key = (entry.file_id, entry.revision)
 
326
        self.text_key_references.setdefault(key, False)
 
327
        if entry.revision == inv.revision_id:
 
328
            self.text_key_references[key] = True
 
329
 
 
330
 
 
331
@deprecated_function(deprecated_in((1,6,0)))
197
332
def check(branch, verbose):
198
 
    """Run consistency checks on a branch."""
199
 
    checker = Check(branch)
200
 
    checker.check()
201
 
    checker.report_results(verbose)
 
333
    """Run consistency checks on a branch.
 
334
 
 
335
    Results are reported through logging.
 
336
 
 
337
    Deprecated in 1.6.  Please use check_dwim instead.
 
338
 
 
339
    :raise BzrCheckError: if there's a consistency error.
 
340
    """
 
341
    check_branch(branch, verbose)
 
342
 
 
343
 
 
344
@deprecated_function(deprecated_in((1,16,0)))
 
345
def check_branch(branch, verbose):
 
346
    """Run consistency checks on a branch.
 
347
 
 
348
    Results are reported through logging.
 
349
 
 
350
    :raise BzrCheckError: if there's a consistency error.
 
351
    """
 
352
    branch.lock_read()
 
353
    try:
 
354
        needed_refs = {}
 
355
        for ref in branch._get_check_refs():
 
356
            needed_refs.setdefault(ref, []).append(branch)
 
357
        result = branch.repository.check([branch.last_revision()], needed_refs)
 
358
        branch_result = result.other_results[0]
 
359
    finally:
 
360
        branch.unlock()
 
361
    branch_result.report_results(verbose)
 
362
 
 
363
 
 
364
def scan_branch(branch, needed_refs, to_unlock):
 
365
    """Scan a branch for refs.
 
366
 
 
367
    :param branch:  The branch to schedule for checking.
 
368
    :param needed_refs: Refs we are accumulating.
 
369
    :param to_unlock: The unlock list accumulating.
 
370
    """
 
371
    note("Checking branch at '%s'." % (branch.base,))
 
372
    branch.lock_read()
 
373
    to_unlock.append(branch)
 
374
    branch_refs = branch._get_check_refs()
 
375
    for ref in branch_refs:
 
376
        reflist = needed_refs.setdefault(ref, [])
 
377
        reflist.append(branch)
 
378
 
 
379
 
 
380
def scan_tree(base_tree, tree, needed_refs, to_unlock):
 
381
    """Scan a tree for refs.
 
382
 
 
383
    :param base_tree: The original tree check opened, used to detect duplicate
 
384
        tree checks.
 
385
    :param tree:  The tree to schedule for checking.
 
386
    :param needed_refs: Refs we are accumulating.
 
387
    :param to_unlock: The unlock list accumulating.
 
388
    """
 
389
    if base_tree is not None and tree.basedir == base_tree.basedir:
 
390
        return
 
391
    note("Checking working tree at '%s'." % (tree.basedir,))
 
392
    tree.lock_read()
 
393
    to_unlock.append(tree)
 
394
    tree_refs = tree._get_check_refs()
 
395
    for ref in tree_refs:
 
396
        reflist = needed_refs.setdefault(ref, [])
 
397
        reflist.append(tree)
 
398
 
 
399
 
 
400
def check_dwim(path, verbose, do_branch=False, do_repo=False, do_tree=False):
 
401
    """Check multiple objects.
 
402
 
 
403
    If errors occur they are accumulated and reported as far as possible, and
 
404
    an exception raised at the end of the process.
 
405
    """
 
406
    try:
 
407
        base_tree, branch, repo, relpath = \
 
408
                        BzrDir.open_containing_tree_branch_or_repository(path)
 
409
    except errors.NotBranchError:
 
410
        base_tree = branch = repo = None
 
411
 
 
412
    to_unlock = []
 
413
    needed_refs= {}
 
414
    try:
 
415
        if base_tree is not None:
 
416
            # If the tree is a lightweight checkout we won't see it in
 
417
            # repo.find_branches - add now.
 
418
            if do_tree:
 
419
                scan_tree(None, base_tree, needed_refs, to_unlock)
 
420
            branch = base_tree.branch
 
421
        if branch is not None:
 
422
            # We have a branch
 
423
            if repo is None:
 
424
                # The branch is in a shared repository
 
425
                repo = branch.repository
 
426
        if repo is not None:
 
427
            repo.lock_read()
 
428
            to_unlock.append(repo)
 
429
            branches = repo.find_branches(using=True)
 
430
            saw_tree = False
 
431
            if do_branch or do_tree:
 
432
                for branch in branches:
 
433
                    if do_tree:
 
434
                        try:
 
435
                            tree = branch.bzrdir.open_workingtree()
 
436
                            saw_tree = True
 
437
                        except (errors.NotLocalUrl, errors.NoWorkingTree):
 
438
                            pass
 
439
                        else:
 
440
                            scan_tree(base_tree, tree, needed_refs, to_unlock)
 
441
                    if do_branch:
 
442
                        scan_branch(branch, needed_refs, to_unlock)
 
443
            if do_branch and not branches:
 
444
                note("No branch found at specified location.")
 
445
            if do_tree and base_tree is None and not saw_tree:
 
446
                note("No working tree found at specified location.")
 
447
            if do_repo or do_branch or do_tree:
 
448
                if do_repo:
 
449
                    note("Checking repository at '%s'."
 
450
                         % (repo.user_url,))
 
451
                result = repo.check(None, callback_refs=needed_refs,
 
452
                    check_repo=do_repo)
 
453
                result.report_results(verbose)
 
454
        else:
 
455
            if do_tree:
 
456
                note("No working tree found at specified location.")
 
457
            if do_branch:
 
458
                note("No branch found at specified location.")
 
459
            if do_repo:
 
460
                note("No repository found at specified location.")
 
461
    finally:
 
462
        for thing in to_unlock:
 
463
            thing.unlock()