~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/check.py

  • Committer: Patch Queue Manager
  • Date: 2014-04-09 13:36:25 UTC
  • mfrom: (6592.1.2 1303879-py27-issues)
  • Revision ID: pqm@pqm.ubuntu.com-20140409133625-s24spv3kha2w2860
(vila) Fix python-2.7.6 test failures. (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

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