~bzr-pqm/bzr/bzr.dev

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
# Copyright (C) 2004, 2005 by Martin Pool
# Copyright (C) 2005 by Canonical Ltd

# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA



######################################################################
# consistency checks

import sys

from trace import mutter
from errors import bailout
import osutils

def check(branch, progress=True):
    out = sys.stdout

    # TODO: factor out
    if not (hasattr(out, 'isatty') and out.isatty()):
        progress=False

    if progress:
        def p(m):
            mutter('checking ' + m)
            out.write('\rchecking: %-50.50s' % m)
            out.flush()
    else:
        def p(m):
            mutter('checking ' + m)

    p('history of %r' % branch.base)
    last_ptr = None
    checked_revs = {}
    
    history = branch.revision_history()
    revno = 0
    revcount = len(history)

    checked_texts = {}
    
    for rid in history:
        revno += 1
        p('revision %d/%d' % (revno, revcount))
        mutter('    revision {%s}' % rid)
        rev = branch.get_revision(rid)
        if rev.revision_id != rid:
            bailout('wrong internal revision id in revision {%s}' % rid)
        if rev.precursor != last_ptr:
            bailout('mismatched precursor in revision {%s}' % rid)
        last_ptr = rid
        if rid in checked_revs:
            bailout('repeated revision {%s}' % rid)
        checked_revs[rid] = True

        ## TODO: Check all the required fields are present on the revision.

        inv = branch.get_inventory(rev.inventory_id)
        seen_ids = {}
        seen_names = {}

        p('revision %d/%d file ids' % (revno, revcount))
        for file_id in inv:
            if file_id in seen_ids:
                bailout('duplicated file_id {%s} in inventory for revision {%s}'
                        % (file_id, rid))
            seen_ids[file_id] = True

        i = 0
        len_inv = len(inv)
        for file_id in inv:
            i += 1
            if (i % 100) == 0:
                p('revision %d/%d file text %d/%d' % (revno, revcount, i, len_inv))

            ie = inv[file_id]

            if ie.parent_id != None:
                if ie.parent_id not in seen_ids:
                    bailout('missing parent {%s} in inventory for revision {%s}'
                            % (ie.parent_id, rid))

            if ie.kind == 'file':
                if ie.text_id in checked_texts:
                    fp = checked_texts[ie.text_id]
                else:
                    if not ie.text_id in branch.text_store:
                        bailout('text {%s} not in text_store' % ie.text_id)

                    tf = branch.text_store[ie.text_id]
                    fp = osutils.fingerprint_file(tf)
                    checked_texts[ie.text_id] = fp

                if ie.text_size != fp['size']:
                    bailout('text {%s} wrong size' % ie.text_id)
                if ie.text_sha1 != fp['sha1']:
                    bailout('text {%s} wrong sha1' % ie.text_id)
            elif ie.kind == 'directory':
                if ie.text_sha1 != None or ie.text_size != None or ie.text_id != None:
                    bailout('directory {%s} has text in revision {%s}'
                            % (file_id, rid))

        p('revision %d/%d file paths' % (revno, revcount))
        for path, ie in inv.iter_entries():
            if path in seen_names:
                bailout('duplicated path %r in inventory for revision {%s}' % (path, revid))
            seen_names[path] = True


    p('done')
    if progress:
        print 
    print 'checked %d revisions, %d file texts' % (revcount, len(checked_texts))