← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/check.py

  • Committer: mbp at sourcefrog
  • Date: 2005-03-30 23:01:29 UTC
  • Revision ID: mbp@sourcefrog.net-20050330230129-8c4a1f77e7d534d9
experiment in writing XML by hand, not through ElementTree

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/check.py
17
17
 
18
18
 
19
19
 
20
 
 
21
 
def check(branch):
22
 
    """Run consistency checks on a branch.
23
 
 
24
 
    TODO: Also check non-mainline revisions mentioned as parents.
25
 
 
26
 
    TODO: Check for extra files in the control directory.
27
 
    """
28
 
    from bzrlib.trace import mutter
29
 
    from bzrlib.errors import BzrCheckError
30
 
    from bzrlib.osutils import fingerprint_file
31
 
    from bzrlib.progress import ProgressBar
32
 
 
33
 
    branch.lock_read()
34
 
 
35
 
    try:
36
 
        pb = ProgressBar(show_spinner=True)
37
 
        last_rev_id = None
38
 
 
39
 
        missing_inventory_sha_cnt = 0
40
 
        missing_revision_sha_cnt = 0
41
 
 
42
 
        history = branch.revision_history()
43
 
        revno = 0
44
 
        revcount = len(history)
45
 
        mismatch_inv_id = []
46
 
 
47
 
        # for all texts checked, text_id -> sha1
48
 
        checked_texts = {}
49
 
 
50
 
        for rev_id in history:
51
 
            revno += 1
52
 
            pb.update('checking revision', revno, revcount)
53
 
            mutter('    revision {%s}' % rev_id)
54
 
            rev = branch.get_revision(rev_id)
55
 
            if rev.revision_id != rev_id:
56
 
                raise BzrCheckError('wrong internal revision id in revision {%s}'
57
 
                                    % rev_id)
58
 
 
59
 
            # check the previous history entry is a parent of this entry
60
 
            if rev.parents:
61
 
                if last_rev_id is None:
62
 
                    raise BzrCheckError("revision {%s} has %d parents, but is the "
63
 
                                        "start of the branch"
64
 
                                        % (rev_id, len(rev.parents)))
65
 
                for prr in rev.parents:
66
 
                    if prr.revision_id == last_rev_id:
67
 
                        break
 
20
######################################################################
 
21
# consistency checks
 
22
 
 
23
import sys
 
24
from sets import Set
 
25
 
 
26
import bzrlib
 
27
from trace import mutter
 
28
from errors import bailout
 
29
import osutils
 
30
 
 
31
def check(branch, progress=True):
 
32
    out = sys.stdout
 
33
 
 
34
    if progress:
 
35
        def p(m):
 
36
            mutter('checking ' + m)
 
37
            out.write('\rchecking: %-50.50s' % m)
 
38
            out.flush()
 
39
    else:
 
40
        def p(m):
 
41
            mutter('checking ' + m)
 
42
 
 
43
    p('history of %r' % branch.base)
 
44
    last_ptr = None
 
45
    checked_revs = Set()
 
46
    
 
47
    history = branch.revision_history()
 
48
    revno = 0
 
49
    revcount = len(history)
 
50
 
 
51
    checked_texts = {}
 
52
    
 
53
    for rid in history:
 
54
        revno += 1
 
55
        p('revision %d/%d' % (revno, revcount))
 
56
        mutter('    revision {%s}' % rid)
 
57
        rev = branch.get_revision(rid)
 
58
        if rev.revision_id != rid:
 
59
            bailout('wrong internal revision id in revision {%s}' % rid)
 
60
        if rev.precursor != last_ptr:
 
61
            bailout('mismatched precursor in revision {%s}' % rid)
 
62
        last_ptr = rid
 
63
        if rid in checked_revs:
 
64
            bailout('repeated revision {%s}' % rid)
 
65
        checked_revs.add(rid)
 
66
 
 
67
        ## TODO: Check all the required fields are present on the revision.
 
68
 
 
69
        inv = branch.get_inventory(rev.inventory_id)
 
70
        seen_ids = Set()
 
71
        seen_names = Set()
 
72
 
 
73
        p('revision %d/%d file ids' % (revno, revcount))
 
74
        for file_id in inv:
 
75
            if file_id in seen_ids:
 
76
                bailout('duplicated file_id {%s} in inventory for revision {%s}'
 
77
                        % (file_id, revid))
 
78
            seen_ids.add(file_id)
 
79
 
 
80
        i = 0
 
81
        len_inv = len(inv)
 
82
        for file_id in inv:
 
83
            i += 1
 
84
            if (i % 100) == 0:
 
85
                p('revision %d/%d file text %d/%d' % (revno, revcount, i, len_inv))
 
86
 
 
87
            ie = inv[file_id]
 
88
 
 
89
            if ie.parent_id != None:
 
90
                if ie.parent_id not in seen_ids:
 
91
                    bailout('missing parent {%s} in inventory for revision {%s}'
 
92
                            % (ie.parent_id, revid))
 
93
 
 
94
            if ie.kind == 'file':
 
95
                if ie.text_id in checked_texts:
 
96
                    fp = checked_texts[ie.text_id]
68
97
                else:
69
 
                    raise BzrCheckError("previous revision {%s} not listed among "
70
 
                                        "parents of {%s}"
71
 
                                        % (last_rev_id, rev_id))
72
 
 
73
 
                for prr in rev.parents:
74
 
                    if prr.revision_sha1 is None:
75
 
                        missing_revision_sha_cnt += 1
76
 
                        continue
77
 
                    prid = prr.revision_id
78
 
                    actual_sha = branch.get_revision_sha1(prid)
79
 
                    if prr.revision_sha1 != actual_sha:
80
 
                        raise BzrCheckError("mismatched revision sha1 for "
81
 
                                            "parent {%s} of {%s}: %s vs %s"
82
 
                                            % (prid, rev_id,
83
 
                                               prr.revision_sha1, actual_sha))
84
 
            elif last_rev_id:
85
 
                raise BzrCheckError("revision {%s} has no parents listed but preceded "
86
 
                                    "by {%s}"
87
 
                                    % (rev_id, last_rev_id))
88
 
 
89
 
            if rev.inventory_id != rev_id:
90
 
                mismatch_inv_id.append(rev_id)
91
 
 
92
 
            ## TODO: Check all the required fields are present on the revision.
93
 
 
94
 
            if rev.inventory_sha1:
95
 
                inv_sha1 = branch.get_inventory_sha1(rev.inventory_id)
96
 
                if inv_sha1 != rev.inventory_sha1:
97
 
                    raise BzrCheckError('Inventory sha1 hash doesn\'t match'
98
 
                        ' value in revision {%s}' % rev_id)
99
 
            else:
100
 
                missing_inventory_sha_cnt += 1
101
 
                mutter("no inventory_sha1 on revision {%s}" % rev_id)
102
 
 
103
 
            inv = branch.get_inventory(rev.inventory_id)
104
 
            seen_ids = {}
105
 
            seen_names = {}
106
 
 
107
 
            ## p('revision %d/%d file ids' % (revno, revcount))
108
 
            for file_id in inv:
109
 
                if file_id in seen_ids:
110
 
                    raise BzrCheckError('duplicated file_id {%s} '
111
 
                                        'in inventory for revision {%s}'
112
 
                                        % (file_id, rev_id))
113
 
                seen_ids[file_id] = True
114
 
 
115
 
            i = 0
116
 
            for file_id in inv:
117
 
                i += 1
118
 
                if i & 31 == 0:
119
 
                    pb.tick()
120
 
 
121
 
                ie = inv[file_id]
122
 
 
123
 
                if ie.parent_id != None:
124
 
                    if ie.parent_id not in seen_ids:
125
 
                        raise BzrCheckError('missing parent {%s} in inventory for revision {%s}'
126
 
                                % (ie.parent_id, rev_id))
127
 
 
128
 
                if ie.kind == 'file':
129
 
                    if ie.text_id in checked_texts:
130
 
                        fp = checked_texts[ie.text_id]
131
 
                    else:
132
 
                        if not ie.text_id in branch.text_store:
133
 
                            raise BzrCheckError('text {%s} not in text_store' % ie.text_id)
134
 
 
135
 
                        tf = branch.text_store[ie.text_id]
136
 
                        fp = fingerprint_file(tf)
137
 
                        checked_texts[ie.text_id] = fp
138
 
 
139
 
                    if ie.text_size != fp['size']:
140
 
                        raise BzrCheckError('text {%s} wrong size' % ie.text_id)
141
 
                    if ie.text_sha1 != fp['sha1']:
142
 
                        raise BzrCheckError('text {%s} wrong sha1' % ie.text_id)
143
 
                elif ie.kind == 'directory':
144
 
                    if ie.text_sha1 != None or ie.text_size != None or ie.text_id != None:
145
 
                        raise BzrCheckError('directory {%s} has text in revision {%s}'
146
 
                                % (file_id, rev_id))
147
 
 
148
 
            pb.tick()
149
 
            for path, ie in inv.iter_entries():
150
 
                if path in seen_names:
151
 
                    raise BzrCheckError('duplicated path %s '
152
 
                                        'in inventory for revision {%s}'
153
 
                                        % (path, rev_id))
154
 
            seen_names[path] = True
155
 
            last_rev_id = rev_id
156
 
 
157
 
    finally:
158
 
        branch.unlock()
159
 
 
160
 
    pb.clear()
161
 
 
162
 
    print 'checked %d revisions, %d file texts' % (revcount, len(checked_texts))
163
 
    
164
 
    if missing_inventory_sha_cnt:
165
 
        print '%d revisions are missing inventory_sha1' % missing_inventory_sha_cnt
166
 
 
167
 
    if missing_revision_sha_cnt:
168
 
        print '%d parent links are missing revision_sha1' % missing_revision_sha_cnt
169
 
 
170
 
    if (missing_inventory_sha_cnt
171
 
        or missing_revision_sha_cnt):
172
 
        print '  (use "bzr upgrade" to fix them)'
173
 
 
174
 
    if mismatch_inv_id:
175
 
        print '%d revisions have mismatched inventory ids:' % len(mismatch_inv_id)
176
 
        for rev_id in mismatch_inv_id:
177
 
            print '  ', rev_id
 
98
                    if not ie.text_id in branch.text_store:
 
99
                        bailout('text {%s} not in text_store' % ie.text_id)
 
100
 
 
101
                    tf = branch.text_store[ie.text_id]
 
102
                    fp = osutils.fingerprint_file(tf)
 
103
                    checked_texts[ie.text_id] = fp
 
104
 
 
105
                if ie.text_size != fp['size']:
 
106
                    bailout('text {%s} wrong size' % ie.text_id)
 
107
                if ie.text_sha1 != fp['sha1']:
 
108
                    bailout('text {%s} wrong sha1' % ie.text_id)
 
109
            elif ie.kind == 'directory':
 
110
                if ie.text_sha1 != None or ie.text_size != None or ie.text_id != None:
 
111
                    bailout('directory {%s} has text in revision {%s}'
 
112
                            % (file_id, revid))
 
113
 
 
114
        p('revision %d/%d file paths' % (revno, revcount))
 
115
        for path, ie in inv.iter_entries():
 
116
            if path in seen_names:
 
117
                bailout('duplicated path %r in inventory for revision {%s}' % (path, revid))
 
118
            seen_names.add(path)
 
119
 
 
120
 
 
121
    p('done')
 
122
    if progress:
 
123
        print 
 
124
 
 
125