~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/testament.py

Remove all uses of compare_trees and replace with Tree.changes_from throughout bzrlib.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005 by Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
"""Testament - a summary of a revision for signing.
 
18
 
 
19
A testament can be defined as "something that serves as tangible 
 
20
proof or evidence."  In bzr we use them to allow people to certify
 
21
particular revisions as authentic.  
 
22
 
 
23
The goal is that if two revisions are semantically equal, then they will
 
24
have a byte-for-byte equal testament.  We can define different versions of
 
25
"semantically equal" by using different testament classes; e.g. one that
 
26
includes or ignores file-ids.
 
27
 
 
28
We sign a testament rather than the revision XML itself for several reasons.
 
29
The most important is that the form in which the revision is stored
 
30
internally is designed for that purpose, and contains information which need
 
31
not be attested to by the signer.  For example the inventory contains the
 
32
last-changed revision for a file, but this is not necessarily something the
 
33
user cares to sign.
 
34
 
 
35
Having unnecessary fields signed makes the signatures brittle when the same
 
36
revision is stored in different branches or when the format is upgraded.
 
37
 
 
38
Handling upgrades is another motivation for using testaments separate from
 
39
the stored revision.  We would like to be able to compare a signature
 
40
generated from an old-format tree to newer tree, or vice versa.  This could
 
41
be done by comparing the revisions but that makes it unclear about exactly
 
42
what is being compared or not.
 
43
 
 
44
Different signing keys might indicate different levels of trust; we can in
 
45
the future extend this to allow signatures indicating not just that a
 
46
particular version is authentic but that it has other properties.
 
47
 
 
48
The signature can be applied to either the full testament or to just a
 
49
hash of it.
 
50
 
 
51
Testament format 1
 
52
~~~~~~~~~~~~~~~~~~
 
53
 
 
54
* timestamps are given as integers to avoid rounding errors
 
55
* parents given in lexicographical order
 
56
* indented-text form similar to log; intended to be human readable
 
57
* paths are given with forward slashes
 
58
* files are named using paths for ease of comparison/debugging
 
59
* the testament uses unix line-endings (\n)
 
60
"""
 
61
 
 
62
# XXX: At the moment, clients trust that the graph described in a weave
 
63
# is accurate, but that's not covered by the testament.  Perhaps the best
 
64
# fix is when verifying a revision to make sure that every file mentioned 
 
65
# in the revision has compatible ancestry links.
 
66
 
 
67
# TODO: perhaps write timestamp in a more readable form
 
68
 
 
69
# TODO: Perhaps these should just be different formats in which inventories/
 
70
# revisions can be serialized.
 
71
 
 
72
from copy import copy
 
73
from cStringIO import StringIO
 
74
import string
 
75
from sha import sha
 
76
 
 
77
from bzrlib.osutils import contains_whitespace, contains_linebreaks
 
78
 
 
79
class Testament(object):
 
80
    """Reduced summary of a revision.
 
81
 
 
82
    Testaments can be 
 
83
 
 
84
      - produced from a revision
 
85
      - written to a stream
 
86
      - loaded from a stream
 
87
      - compared to a revision
 
88
    """
 
89
 
 
90
    long_header = 'bazaar-ng testament version 1\n'
 
91
    short_header = 'bazaar-ng testament short form 1\n'
 
92
 
 
93
    @classmethod
 
94
    def from_revision(cls, repository, revision_id):
 
95
        """Produce a new testament from a historical revision"""
 
96
        rev = repository.get_revision(revision_id)
 
97
        inventory = repository.get_inventory(revision_id)
 
98
        return cls(rev, inventory)
 
99
 
 
100
    def __init__(self, rev, inventory):
 
101
        """Create a new testament for rev using inventory."""
 
102
        self.revision_id = str(rev.revision_id)
 
103
        self.committer = rev.committer
 
104
        self.timezone = rev.timezone or 0
 
105
        self.timestamp = rev.timestamp
 
106
        self.message = rev.message
 
107
        self.parent_ids = rev.parent_ids[:]
 
108
        self.inventory = inventory
 
109
        self.revprops = copy(rev.properties)
 
110
        assert not contains_whitespace(self.revision_id)
 
111
        assert not contains_linebreaks(self.committer)
 
112
 
 
113
    def as_text_lines(self):
 
114
        """Yield text form as a sequence of lines.
 
115
 
 
116
        The result is returned in utf-8, because it should be signed or
 
117
        hashed in that encoding.
 
118
        """
 
119
        r = []
 
120
        a = r.append
 
121
        a(self.long_header)
 
122
        a('revision-id: %s\n' % self.revision_id)
 
123
        a('committer: %s\n' % self.committer)
 
124
        a('timestamp: %d\n' % self.timestamp)
 
125
        a('timezone: %d\n' % self.timezone)
 
126
        # inventory length contains the root, which is not shown here
 
127
        a('parents:\n')
 
128
        for parent_id in sorted(self.parent_ids):
 
129
            assert not contains_whitespace(parent_id)
 
130
            a('  %s\n' % parent_id)
 
131
        a('message:\n')
 
132
        for l in self.message.splitlines():
 
133
            a('  %s\n' % l)
 
134
        a('inventory:\n')
 
135
        entries = self.inventory.iter_entries()
 
136
        entries.next()
 
137
        for path, ie in entries:
 
138
            a(self._entry_to_line(path, ie))
 
139
        r.extend(self._revprops_to_lines())
 
140
        if __debug__:
 
141
            for l in r:
 
142
                assert isinstance(l, basestring), \
 
143
                    '%r of type %s is not a plain string' % (l, type(l))
 
144
        return [line.encode('utf-8') for line in r]
 
145
 
 
146
    def _escape_path(self, path):
 
147
        assert not contains_linebreaks(path)
 
148
        return unicode(path.replace('\\', '/').replace(' ', '\ ')).encode('utf-8')
 
149
 
 
150
    def _entry_to_line(self, path, ie):
 
151
        """Turn an inventory entry into a testament line"""
 
152
        l = '  ' + str(ie.kind)
 
153
        l += ' ' + self._escape_path(path)
 
154
        assert not contains_whitespace(ie.file_id)
 
155
        l += ' ' + unicode(ie.file_id).encode('utf-8')
 
156
        if ie.kind == 'file':
 
157
            # TODO: avoid switching on kind
 
158
            assert ie.text_sha1
 
159
            l += ' ' + ie.text_sha1
 
160
        elif ie.kind == 'symlink':
 
161
            assert ie.symlink_target
 
162
            l += ' ' + self._escape_path(ie.symlink_target)
 
163
        l += '\n'
 
164
        return l.decode('utf-8')
 
165
 
 
166
    def as_text(self):
 
167
        return ''.join(self.as_text_lines())
 
168
 
 
169
    def as_short_text(self):
 
170
        """Return short digest-based testament."""
 
171
        return (self.short_header + 
 
172
                'revision-id: %s\n'
 
173
                'sha1: %s\n'
 
174
                % (self.revision_id, self.as_sha1()))
 
175
 
 
176
    def _revprops_to_lines(self):
 
177
        """Pack up revision properties."""
 
178
        if not self.revprops:
 
179
            return []
 
180
        r = ['properties:\n']
 
181
        for name, value in sorted(self.revprops.items()):
 
182
            assert isinstance(name, str)
 
183
            assert not contains_whitespace(name)
 
184
            r.append('  %s:\n' % name)
 
185
            for line in value.splitlines():
 
186
                if not isinstance(line, str):
 
187
                    line = line.encode('utf-8')
 
188
                r.append('    %s\n' % line)
 
189
        return r
 
190
 
 
191
    def as_sha1(self):
 
192
        s = sha()
 
193
        map(s.update, self.as_text_lines())
 
194
        return s.hexdigest()
 
195
 
 
196
 
 
197
class StrictTestament(Testament):
 
198
    """This testament format is for use as a checksum in changesets"""
 
199
 
 
200
    long_header = 'bazaar-ng testament version 2.1\n'
 
201
    short_header = 'bazaar-ng testament short form 2.1\n'
 
202
    def _entry_to_line(self, path, ie):
 
203
        l = Testament._entry_to_line(self, path, ie)[:-1]
 
204
        l += ' ' + ie.revision.decode('utf-8')
 
205
        l += {True: ' yes\n', False: ' no\n'}[ie.executable]
 
206
        return l