##// END OF EJS Templates
filelog: allow censored files to contain padding data...
Mike Edgar -
r24117:9cfd7c4f default
parent child Browse files
Show More
@@ -1,109 +1,109 b''
1 1 # filelog.py - file history class for mercurial
2 2 #
3 3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 4 #
5 5 # This software may be used and distributed according to the terms of the
6 6 # GNU General Public License version 2 or any later version.
7 7
8 8 import error, revlog
9 9 import re
10 10
11 11 _mdre = re.compile('\1\n')
12 12 def parsemeta(text):
13 13 """return (metadatadict, keylist, metadatasize)"""
14 14 # text can be buffer, so we can't use .startswith or .index
15 15 if text[:2] != '\1\n':
16 16 return None, None
17 17 s = _mdre.search(text, 2).start()
18 18 mtext = text[2:s]
19 19 meta = {}
20 20 for l in mtext.splitlines():
21 21 k, v = l.split(": ", 1)
22 22 meta[k] = v
23 23 return meta, (s + 2)
24 24
25 25 def packmeta(meta, text):
26 26 keys = sorted(meta.iterkeys())
27 27 metatext = "".join("%s: %s\n" % (k, meta[k]) for k in keys)
28 28 return "\1\n%s\1\n%s" % (metatext, text)
29 29
30 30 def _censoredtext(text):
31 31 m, offs = parsemeta(text)
32 return m and "censored" in m and not text[offs:]
32 return m and "censored" in m
33 33
34 34 class filelog(revlog.revlog):
35 35 def __init__(self, opener, path):
36 36 super(filelog, self).__init__(opener,
37 37 "/".join(("data", path + ".i")))
38 38
39 39 def read(self, node):
40 40 t = self.revision(node)
41 41 if not t.startswith('\1\n'):
42 42 return t
43 43 s = t.index('\1\n', 2)
44 44 return t[s + 2:]
45 45
46 46 def add(self, text, meta, transaction, link, p1=None, p2=None):
47 47 if meta or text.startswith('\1\n'):
48 48 text = packmeta(meta, text)
49 49 return self.addrevision(text, transaction, link, p1, p2)
50 50
51 51 def renamed(self, node):
52 52 if self.parents(node)[0] != revlog.nullid:
53 53 return False
54 54 t = self.revision(node)
55 55 m = parsemeta(t)[0]
56 56 if m and "copy" in m:
57 57 return (m["copy"], revlog.bin(m["copyrev"]))
58 58 return False
59 59
60 60 def size(self, rev):
61 61 """return the size of a given revision"""
62 62
63 63 # for revisions with renames, we have to go the slow way
64 64 node = self.node(rev)
65 65 if self.renamed(node):
66 66 return len(self.read(node))
67 67 if self._iscensored(rev):
68 68 return 0
69 69
70 70 # XXX if self.read(node).startswith("\1\n"), this returns (size+4)
71 71 return super(filelog, self).size(rev)
72 72
73 73 def cmp(self, node, text):
74 74 """compare text with a given file revision
75 75
76 76 returns True if text is different than what is stored.
77 77 """
78 78
79 79 t = text
80 80 if text.startswith('\1\n'):
81 81 t = '\1\n\1\n' + text
82 82
83 83 samehashes = not super(filelog, self).cmp(node, t)
84 84 if samehashes:
85 85 return False
86 86
87 87 # censored files compare against the empty file
88 88 if self._iscensored(self.rev(node)):
89 89 return text != ''
90 90
91 91 # renaming a file produces a different hash, even if the data
92 92 # remains unchanged. Check if it's the case (slow):
93 93 if self.renamed(node):
94 94 t2 = self.read(node)
95 95 return t2 != text
96 96
97 97 return True
98 98
99 99 def checkhash(self, text, p1, p2, node, rev=None):
100 100 try:
101 101 super(filelog, self).checkhash(text, p1, p2, node, rev=rev)
102 102 except error.RevlogError:
103 103 if _censoredtext(text):
104 104 raise error.CensoredNodeError(self.indexfile, node)
105 105 raise
106 106
107 107 def _iscensored(self, rev):
108 108 """Check if a file revision is censored."""
109 109 return self.flags(rev) & revlog.REVIDX_ISCENSORED
General Comments 0
You need to be logged in to leave comments. Login now