##// END OF EJS Templates
revlog: detect incomplete revlog reads...
Gregory Szorc -
r40660:87a87255 default
parent child Browse files
Show More
@@ -1,2548 +1,2567 b''
1 # revlog.py - storage back-end for mercurial
1 # revlog.py - storage back-end for mercurial
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 """Storage back-end for Mercurial.
8 """Storage back-end for Mercurial.
9
9
10 This provides efficient delta storage with O(1) retrieve and append
10 This provides efficient delta storage with O(1) retrieve and append
11 and O(changes) merge between branches.
11 and O(changes) merge between branches.
12 """
12 """
13
13
14 from __future__ import absolute_import
14 from __future__ import absolute_import
15
15
16 import collections
16 import collections
17 import contextlib
17 import contextlib
18 import errno
18 import errno
19 import os
19 import os
20 import struct
20 import struct
21 import zlib
21 import zlib
22
22
23 # import stuff from node for others to import from revlog
23 # import stuff from node for others to import from revlog
24 from .node import (
24 from .node import (
25 bin,
25 bin,
26 hex,
26 hex,
27 nullhex,
27 nullhex,
28 nullid,
28 nullid,
29 nullrev,
29 nullrev,
30 short,
30 short,
31 wdirfilenodeids,
31 wdirfilenodeids,
32 wdirhex,
32 wdirhex,
33 wdirid,
33 wdirid,
34 wdirrev,
34 wdirrev,
35 )
35 )
36 from .i18n import _
36 from .i18n import _
37 from .revlogutils.constants import (
37 from .revlogutils.constants import (
38 FLAG_GENERALDELTA,
38 FLAG_GENERALDELTA,
39 FLAG_INLINE_DATA,
39 FLAG_INLINE_DATA,
40 REVIDX_DEFAULT_FLAGS,
40 REVIDX_DEFAULT_FLAGS,
41 REVIDX_ELLIPSIS,
41 REVIDX_ELLIPSIS,
42 REVIDX_EXTSTORED,
42 REVIDX_EXTSTORED,
43 REVIDX_FLAGS_ORDER,
43 REVIDX_FLAGS_ORDER,
44 REVIDX_ISCENSORED,
44 REVIDX_ISCENSORED,
45 REVIDX_KNOWN_FLAGS,
45 REVIDX_KNOWN_FLAGS,
46 REVIDX_RAWTEXT_CHANGING_FLAGS,
46 REVIDX_RAWTEXT_CHANGING_FLAGS,
47 REVLOGV0,
47 REVLOGV0,
48 REVLOGV1,
48 REVLOGV1,
49 REVLOGV1_FLAGS,
49 REVLOGV1_FLAGS,
50 REVLOGV2,
50 REVLOGV2,
51 REVLOGV2_FLAGS,
51 REVLOGV2_FLAGS,
52 REVLOG_DEFAULT_FLAGS,
52 REVLOG_DEFAULT_FLAGS,
53 REVLOG_DEFAULT_FORMAT,
53 REVLOG_DEFAULT_FORMAT,
54 REVLOG_DEFAULT_VERSION,
54 REVLOG_DEFAULT_VERSION,
55 )
55 )
56 from .thirdparty import (
56 from .thirdparty import (
57 attr,
57 attr,
58 )
58 )
59 from . import (
59 from . import (
60 ancestor,
60 ancestor,
61 dagop,
61 dagop,
62 error,
62 error,
63 mdiff,
63 mdiff,
64 policy,
64 policy,
65 pycompat,
65 pycompat,
66 repository,
66 repository,
67 templatefilters,
67 templatefilters,
68 util,
68 util,
69 )
69 )
70 from .revlogutils import (
70 from .revlogutils import (
71 deltas as deltautil,
71 deltas as deltautil,
72 )
72 )
73 from .utils import (
73 from .utils import (
74 interfaceutil,
74 interfaceutil,
75 storageutil,
75 storageutil,
76 stringutil,
76 stringutil,
77 )
77 )
78
78
79 # blanked usage of all the name to prevent pyflakes constraints
79 # blanked usage of all the name to prevent pyflakes constraints
80 # We need these name available in the module for extensions.
80 # We need these name available in the module for extensions.
81 REVLOGV0
81 REVLOGV0
82 REVLOGV1
82 REVLOGV1
83 REVLOGV2
83 REVLOGV2
84 FLAG_INLINE_DATA
84 FLAG_INLINE_DATA
85 FLAG_GENERALDELTA
85 FLAG_GENERALDELTA
86 REVLOG_DEFAULT_FLAGS
86 REVLOG_DEFAULT_FLAGS
87 REVLOG_DEFAULT_FORMAT
87 REVLOG_DEFAULT_FORMAT
88 REVLOG_DEFAULT_VERSION
88 REVLOG_DEFAULT_VERSION
89 REVLOGV1_FLAGS
89 REVLOGV1_FLAGS
90 REVLOGV2_FLAGS
90 REVLOGV2_FLAGS
91 REVIDX_ISCENSORED
91 REVIDX_ISCENSORED
92 REVIDX_ELLIPSIS
92 REVIDX_ELLIPSIS
93 REVIDX_EXTSTORED
93 REVIDX_EXTSTORED
94 REVIDX_DEFAULT_FLAGS
94 REVIDX_DEFAULT_FLAGS
95 REVIDX_FLAGS_ORDER
95 REVIDX_FLAGS_ORDER
96 REVIDX_KNOWN_FLAGS
96 REVIDX_KNOWN_FLAGS
97 REVIDX_RAWTEXT_CHANGING_FLAGS
97 REVIDX_RAWTEXT_CHANGING_FLAGS
98
98
99 parsers = policy.importmod(r'parsers')
99 parsers = policy.importmod(r'parsers')
100
100
101 # Aliased for performance.
101 # Aliased for performance.
102 _zlibdecompress = zlib.decompress
102 _zlibdecompress = zlib.decompress
103
103
104 # max size of revlog with inline data
104 # max size of revlog with inline data
105 _maxinline = 131072
105 _maxinline = 131072
106 _chunksize = 1048576
106 _chunksize = 1048576
107
107
108 # Store flag processors (cf. 'addflagprocessor()' to register)
108 # Store flag processors (cf. 'addflagprocessor()' to register)
109 _flagprocessors = {
109 _flagprocessors = {
110 REVIDX_ISCENSORED: None,
110 REVIDX_ISCENSORED: None,
111 }
111 }
112
112
113 # Flag processors for REVIDX_ELLIPSIS.
113 # Flag processors for REVIDX_ELLIPSIS.
114 def ellipsisreadprocessor(rl, text):
114 def ellipsisreadprocessor(rl, text):
115 return text, False
115 return text, False
116
116
117 def ellipsiswriteprocessor(rl, text):
117 def ellipsiswriteprocessor(rl, text):
118 return text, False
118 return text, False
119
119
120 def ellipsisrawprocessor(rl, text):
120 def ellipsisrawprocessor(rl, text):
121 return False
121 return False
122
122
123 ellipsisprocessor = (
123 ellipsisprocessor = (
124 ellipsisreadprocessor,
124 ellipsisreadprocessor,
125 ellipsiswriteprocessor,
125 ellipsiswriteprocessor,
126 ellipsisrawprocessor,
126 ellipsisrawprocessor,
127 )
127 )
128
128
129 def addflagprocessor(flag, processor):
129 def addflagprocessor(flag, processor):
130 """Register a flag processor on a revision data flag.
130 """Register a flag processor on a revision data flag.
131
131
132 Invariant:
132 Invariant:
133 - Flags need to be defined in REVIDX_KNOWN_FLAGS and REVIDX_FLAGS_ORDER,
133 - Flags need to be defined in REVIDX_KNOWN_FLAGS and REVIDX_FLAGS_ORDER,
134 and REVIDX_RAWTEXT_CHANGING_FLAGS if they can alter rawtext.
134 and REVIDX_RAWTEXT_CHANGING_FLAGS if they can alter rawtext.
135 - Only one flag processor can be registered on a specific flag.
135 - Only one flag processor can be registered on a specific flag.
136 - flagprocessors must be 3-tuples of functions (read, write, raw) with the
136 - flagprocessors must be 3-tuples of functions (read, write, raw) with the
137 following signatures:
137 following signatures:
138 - (read) f(self, rawtext) -> text, bool
138 - (read) f(self, rawtext) -> text, bool
139 - (write) f(self, text) -> rawtext, bool
139 - (write) f(self, text) -> rawtext, bool
140 - (raw) f(self, rawtext) -> bool
140 - (raw) f(self, rawtext) -> bool
141 "text" is presented to the user. "rawtext" is stored in revlog data, not
141 "text" is presented to the user. "rawtext" is stored in revlog data, not
142 directly visible to the user.
142 directly visible to the user.
143 The boolean returned by these transforms is used to determine whether
143 The boolean returned by these transforms is used to determine whether
144 the returned text can be used for hash integrity checking. For example,
144 the returned text can be used for hash integrity checking. For example,
145 if "write" returns False, then "text" is used to generate hash. If
145 if "write" returns False, then "text" is used to generate hash. If
146 "write" returns True, that basically means "rawtext" returned by "write"
146 "write" returns True, that basically means "rawtext" returned by "write"
147 should be used to generate hash. Usually, "write" and "read" return
147 should be used to generate hash. Usually, "write" and "read" return
148 different booleans. And "raw" returns a same boolean as "write".
148 different booleans. And "raw" returns a same boolean as "write".
149
149
150 Note: The 'raw' transform is used for changegroup generation and in some
150 Note: The 'raw' transform is used for changegroup generation and in some
151 debug commands. In this case the transform only indicates whether the
151 debug commands. In this case the transform only indicates whether the
152 contents can be used for hash integrity checks.
152 contents can be used for hash integrity checks.
153 """
153 """
154 _insertflagprocessor(flag, processor, _flagprocessors)
154 _insertflagprocessor(flag, processor, _flagprocessors)
155
155
156 def _insertflagprocessor(flag, processor, flagprocessors):
156 def _insertflagprocessor(flag, processor, flagprocessors):
157 if not flag & REVIDX_KNOWN_FLAGS:
157 if not flag & REVIDX_KNOWN_FLAGS:
158 msg = _("cannot register processor on unknown flag '%#x'.") % (flag)
158 msg = _("cannot register processor on unknown flag '%#x'.") % (flag)
159 raise error.ProgrammingError(msg)
159 raise error.ProgrammingError(msg)
160 if flag not in REVIDX_FLAGS_ORDER:
160 if flag not in REVIDX_FLAGS_ORDER:
161 msg = _("flag '%#x' undefined in REVIDX_FLAGS_ORDER.") % (flag)
161 msg = _("flag '%#x' undefined in REVIDX_FLAGS_ORDER.") % (flag)
162 raise error.ProgrammingError(msg)
162 raise error.ProgrammingError(msg)
163 if flag in flagprocessors:
163 if flag in flagprocessors:
164 msg = _("cannot register multiple processors on flag '%#x'.") % (flag)
164 msg = _("cannot register multiple processors on flag '%#x'.") % (flag)
165 raise error.Abort(msg)
165 raise error.Abort(msg)
166 flagprocessors[flag] = processor
166 flagprocessors[flag] = processor
167
167
168 def getoffset(q):
168 def getoffset(q):
169 return int(q >> 16)
169 return int(q >> 16)
170
170
171 def gettype(q):
171 def gettype(q):
172 return int(q & 0xFFFF)
172 return int(q & 0xFFFF)
173
173
174 def offset_type(offset, type):
174 def offset_type(offset, type):
175 if (type & ~REVIDX_KNOWN_FLAGS) != 0:
175 if (type & ~REVIDX_KNOWN_FLAGS) != 0:
176 raise ValueError('unknown revlog index flags')
176 raise ValueError('unknown revlog index flags')
177 return int(int(offset) << 16 | type)
177 return int(int(offset) << 16 | type)
178
178
179 @attr.s(slots=True, frozen=True)
179 @attr.s(slots=True, frozen=True)
180 class _revisioninfo(object):
180 class _revisioninfo(object):
181 """Information about a revision that allows building its fulltext
181 """Information about a revision that allows building its fulltext
182 node: expected hash of the revision
182 node: expected hash of the revision
183 p1, p2: parent revs of the revision
183 p1, p2: parent revs of the revision
184 btext: built text cache consisting of a one-element list
184 btext: built text cache consisting of a one-element list
185 cachedelta: (baserev, uncompressed_delta) or None
185 cachedelta: (baserev, uncompressed_delta) or None
186 flags: flags associated to the revision storage
186 flags: flags associated to the revision storage
187
187
188 One of btext[0] or cachedelta must be set.
188 One of btext[0] or cachedelta must be set.
189 """
189 """
190 node = attr.ib()
190 node = attr.ib()
191 p1 = attr.ib()
191 p1 = attr.ib()
192 p2 = attr.ib()
192 p2 = attr.ib()
193 btext = attr.ib()
193 btext = attr.ib()
194 textlen = attr.ib()
194 textlen = attr.ib()
195 cachedelta = attr.ib()
195 cachedelta = attr.ib()
196 flags = attr.ib()
196 flags = attr.ib()
197
197
198 @interfaceutil.implementer(repository.irevisiondelta)
198 @interfaceutil.implementer(repository.irevisiondelta)
199 @attr.s(slots=True)
199 @attr.s(slots=True)
200 class revlogrevisiondelta(object):
200 class revlogrevisiondelta(object):
201 node = attr.ib()
201 node = attr.ib()
202 p1node = attr.ib()
202 p1node = attr.ib()
203 p2node = attr.ib()
203 p2node = attr.ib()
204 basenode = attr.ib()
204 basenode = attr.ib()
205 flags = attr.ib()
205 flags = attr.ib()
206 baserevisionsize = attr.ib()
206 baserevisionsize = attr.ib()
207 revision = attr.ib()
207 revision = attr.ib()
208 delta = attr.ib()
208 delta = attr.ib()
209 linknode = attr.ib(default=None)
209 linknode = attr.ib(default=None)
210
210
211 @interfaceutil.implementer(repository.iverifyproblem)
211 @interfaceutil.implementer(repository.iverifyproblem)
212 @attr.s(frozen=True)
212 @attr.s(frozen=True)
213 class revlogproblem(object):
213 class revlogproblem(object):
214 warning = attr.ib(default=None)
214 warning = attr.ib(default=None)
215 error = attr.ib(default=None)
215 error = attr.ib(default=None)
216 node = attr.ib(default=None)
216 node = attr.ib(default=None)
217
217
218 # index v0:
218 # index v0:
219 # 4 bytes: offset
219 # 4 bytes: offset
220 # 4 bytes: compressed length
220 # 4 bytes: compressed length
221 # 4 bytes: base rev
221 # 4 bytes: base rev
222 # 4 bytes: link rev
222 # 4 bytes: link rev
223 # 20 bytes: parent 1 nodeid
223 # 20 bytes: parent 1 nodeid
224 # 20 bytes: parent 2 nodeid
224 # 20 bytes: parent 2 nodeid
225 # 20 bytes: nodeid
225 # 20 bytes: nodeid
226 indexformatv0 = struct.Struct(">4l20s20s20s")
226 indexformatv0 = struct.Struct(">4l20s20s20s")
227 indexformatv0_pack = indexformatv0.pack
227 indexformatv0_pack = indexformatv0.pack
228 indexformatv0_unpack = indexformatv0.unpack
228 indexformatv0_unpack = indexformatv0.unpack
229
229
230 class revlogoldindex(list):
230 class revlogoldindex(list):
231 def __getitem__(self, i):
231 def __getitem__(self, i):
232 if i == -1:
232 if i == -1:
233 return (0, 0, 0, -1, -1, -1, -1, nullid)
233 return (0, 0, 0, -1, -1, -1, -1, nullid)
234 return list.__getitem__(self, i)
234 return list.__getitem__(self, i)
235
235
236 class revlogoldio(object):
236 class revlogoldio(object):
237 def __init__(self):
237 def __init__(self):
238 self.size = indexformatv0.size
238 self.size = indexformatv0.size
239
239
240 def parseindex(self, data, inline):
240 def parseindex(self, data, inline):
241 s = self.size
241 s = self.size
242 index = []
242 index = []
243 nodemap = {nullid: nullrev}
243 nodemap = {nullid: nullrev}
244 n = off = 0
244 n = off = 0
245 l = len(data)
245 l = len(data)
246 while off + s <= l:
246 while off + s <= l:
247 cur = data[off:off + s]
247 cur = data[off:off + s]
248 off += s
248 off += s
249 e = indexformatv0_unpack(cur)
249 e = indexformatv0_unpack(cur)
250 # transform to revlogv1 format
250 # transform to revlogv1 format
251 e2 = (offset_type(e[0], 0), e[1], -1, e[2], e[3],
251 e2 = (offset_type(e[0], 0), e[1], -1, e[2], e[3],
252 nodemap.get(e[4], nullrev), nodemap.get(e[5], nullrev), e[6])
252 nodemap.get(e[4], nullrev), nodemap.get(e[5], nullrev), e[6])
253 index.append(e2)
253 index.append(e2)
254 nodemap[e[6]] = n
254 nodemap[e[6]] = n
255 n += 1
255 n += 1
256
256
257 return revlogoldindex(index), nodemap, None
257 return revlogoldindex(index), nodemap, None
258
258
259 def packentry(self, entry, node, version, rev):
259 def packentry(self, entry, node, version, rev):
260 if gettype(entry[0]):
260 if gettype(entry[0]):
261 raise error.RevlogError(_('index entry flags need revlog '
261 raise error.RevlogError(_('index entry flags need revlog '
262 'version 1'))
262 'version 1'))
263 e2 = (getoffset(entry[0]), entry[1], entry[3], entry[4],
263 e2 = (getoffset(entry[0]), entry[1], entry[3], entry[4],
264 node(entry[5]), node(entry[6]), entry[7])
264 node(entry[5]), node(entry[6]), entry[7])
265 return indexformatv0_pack(*e2)
265 return indexformatv0_pack(*e2)
266
266
267 # index ng:
267 # index ng:
268 # 6 bytes: offset
268 # 6 bytes: offset
269 # 2 bytes: flags
269 # 2 bytes: flags
270 # 4 bytes: compressed length
270 # 4 bytes: compressed length
271 # 4 bytes: uncompressed length
271 # 4 bytes: uncompressed length
272 # 4 bytes: base rev
272 # 4 bytes: base rev
273 # 4 bytes: link rev
273 # 4 bytes: link rev
274 # 4 bytes: parent 1 rev
274 # 4 bytes: parent 1 rev
275 # 4 bytes: parent 2 rev
275 # 4 bytes: parent 2 rev
276 # 32 bytes: nodeid
276 # 32 bytes: nodeid
277 indexformatng = struct.Struct(">Qiiiiii20s12x")
277 indexformatng = struct.Struct(">Qiiiiii20s12x")
278 indexformatng_pack = indexformatng.pack
278 indexformatng_pack = indexformatng.pack
279 versionformat = struct.Struct(">I")
279 versionformat = struct.Struct(">I")
280 versionformat_pack = versionformat.pack
280 versionformat_pack = versionformat.pack
281 versionformat_unpack = versionformat.unpack
281 versionformat_unpack = versionformat.unpack
282
282
283 # corresponds to uncompressed length of indexformatng (2 gigs, 4-byte
283 # corresponds to uncompressed length of indexformatng (2 gigs, 4-byte
284 # signed integer)
284 # signed integer)
285 _maxentrysize = 0x7fffffff
285 _maxentrysize = 0x7fffffff
286
286
287 class revlogio(object):
287 class revlogio(object):
288 def __init__(self):
288 def __init__(self):
289 self.size = indexformatng.size
289 self.size = indexformatng.size
290
290
291 def parseindex(self, data, inline):
291 def parseindex(self, data, inline):
292 # call the C implementation to parse the index data
292 # call the C implementation to parse the index data
293 index, cache = parsers.parse_index2(data, inline)
293 index, cache = parsers.parse_index2(data, inline)
294 return index, getattr(index, 'nodemap', None), cache
294 return index, getattr(index, 'nodemap', None), cache
295
295
296 def packentry(self, entry, node, version, rev):
296 def packentry(self, entry, node, version, rev):
297 p = indexformatng_pack(*entry)
297 p = indexformatng_pack(*entry)
298 if rev == 0:
298 if rev == 0:
299 p = versionformat_pack(version) + p[4:]
299 p = versionformat_pack(version) + p[4:]
300 return p
300 return p
301
301
302 class revlog(object):
302 class revlog(object):
303 """
303 """
304 the underlying revision storage object
304 the underlying revision storage object
305
305
306 A revlog consists of two parts, an index and the revision data.
306 A revlog consists of two parts, an index and the revision data.
307
307
308 The index is a file with a fixed record size containing
308 The index is a file with a fixed record size containing
309 information on each revision, including its nodeid (hash), the
309 information on each revision, including its nodeid (hash), the
310 nodeids of its parents, the position and offset of its data within
310 nodeids of its parents, the position and offset of its data within
311 the data file, and the revision it's based on. Finally, each entry
311 the data file, and the revision it's based on. Finally, each entry
312 contains a linkrev entry that can serve as a pointer to external
312 contains a linkrev entry that can serve as a pointer to external
313 data.
313 data.
314
314
315 The revision data itself is a linear collection of data chunks.
315 The revision data itself is a linear collection of data chunks.
316 Each chunk represents a revision and is usually represented as a
316 Each chunk represents a revision and is usually represented as a
317 delta against the previous chunk. To bound lookup time, runs of
317 delta against the previous chunk. To bound lookup time, runs of
318 deltas are limited to about 2 times the length of the original
318 deltas are limited to about 2 times the length of the original
319 version data. This makes retrieval of a version proportional to
319 version data. This makes retrieval of a version proportional to
320 its size, or O(1) relative to the number of revisions.
320 its size, or O(1) relative to the number of revisions.
321
321
322 Both pieces of the revlog are written to in an append-only
322 Both pieces of the revlog are written to in an append-only
323 fashion, which means we never need to rewrite a file to insert or
323 fashion, which means we never need to rewrite a file to insert or
324 remove data, and can use some simple techniques to avoid the need
324 remove data, and can use some simple techniques to avoid the need
325 for locking while reading.
325 for locking while reading.
326
326
327 If checkambig, indexfile is opened with checkambig=True at
327 If checkambig, indexfile is opened with checkambig=True at
328 writing, to avoid file stat ambiguity.
328 writing, to avoid file stat ambiguity.
329
329
330 If mmaplargeindex is True, and an mmapindexthreshold is set, the
330 If mmaplargeindex is True, and an mmapindexthreshold is set, the
331 index will be mmapped rather than read if it is larger than the
331 index will be mmapped rather than read if it is larger than the
332 configured threshold.
332 configured threshold.
333
333
334 If censorable is True, the revlog can have censored revisions.
334 If censorable is True, the revlog can have censored revisions.
335 """
335 """
336 def __init__(self, opener, indexfile, datafile=None, checkambig=False,
336 def __init__(self, opener, indexfile, datafile=None, checkambig=False,
337 mmaplargeindex=False, censorable=False):
337 mmaplargeindex=False, censorable=False):
338 """
338 """
339 create a revlog object
339 create a revlog object
340
340
341 opener is a function that abstracts the file opening operation
341 opener is a function that abstracts the file opening operation
342 and can be used to implement COW semantics or the like.
342 and can be used to implement COW semantics or the like.
343 """
343 """
344 self.indexfile = indexfile
344 self.indexfile = indexfile
345 self.datafile = datafile or (indexfile[:-2] + ".d")
345 self.datafile = datafile or (indexfile[:-2] + ".d")
346 self.opener = opener
346 self.opener = opener
347 # When True, indexfile is opened with checkambig=True at writing, to
347 # When True, indexfile is opened with checkambig=True at writing, to
348 # avoid file stat ambiguity.
348 # avoid file stat ambiguity.
349 self._checkambig = checkambig
349 self._checkambig = checkambig
350 self._censorable = censorable
350 self._censorable = censorable
351 # 3-tuple of (node, rev, text) for a raw revision.
351 # 3-tuple of (node, rev, text) for a raw revision.
352 self._revisioncache = None
352 self._revisioncache = None
353 # Maps rev to chain base rev.
353 # Maps rev to chain base rev.
354 self._chainbasecache = util.lrucachedict(100)
354 self._chainbasecache = util.lrucachedict(100)
355 # 2-tuple of (offset, data) of raw data from the revlog at an offset.
355 # 2-tuple of (offset, data) of raw data from the revlog at an offset.
356 self._chunkcache = (0, '')
356 self._chunkcache = (0, '')
357 # How much data to read and cache into the raw revlog data cache.
357 # How much data to read and cache into the raw revlog data cache.
358 self._chunkcachesize = 65536
358 self._chunkcachesize = 65536
359 self._maxchainlen = None
359 self._maxchainlen = None
360 self._deltabothparents = True
360 self._deltabothparents = True
361 self.index = []
361 self.index = []
362 # Mapping of partial identifiers to full nodes.
362 # Mapping of partial identifiers to full nodes.
363 self._pcache = {}
363 self._pcache = {}
364 # Mapping of revision integer to full node.
364 # Mapping of revision integer to full node.
365 self._nodecache = {nullid: nullrev}
365 self._nodecache = {nullid: nullrev}
366 self._nodepos = None
366 self._nodepos = None
367 self._compengine = 'zlib'
367 self._compengine = 'zlib'
368 self._maxdeltachainspan = -1
368 self._maxdeltachainspan = -1
369 self._withsparseread = False
369 self._withsparseread = False
370 self._sparserevlog = False
370 self._sparserevlog = False
371 self._srdensitythreshold = 0.50
371 self._srdensitythreshold = 0.50
372 self._srmingapsize = 262144
372 self._srmingapsize = 262144
373
373
374 # Make copy of flag processors so each revlog instance can support
374 # Make copy of flag processors so each revlog instance can support
375 # custom flags.
375 # custom flags.
376 self._flagprocessors = dict(_flagprocessors)
376 self._flagprocessors = dict(_flagprocessors)
377
377
378 mmapindexthreshold = None
378 mmapindexthreshold = None
379 v = REVLOG_DEFAULT_VERSION
379 v = REVLOG_DEFAULT_VERSION
380 opts = getattr(opener, 'options', None)
380 opts = getattr(opener, 'options', None)
381 if opts is not None:
381 if opts is not None:
382 if 'revlogv2' in opts:
382 if 'revlogv2' in opts:
383 # version 2 revlogs always use generaldelta.
383 # version 2 revlogs always use generaldelta.
384 v = REVLOGV2 | FLAG_GENERALDELTA | FLAG_INLINE_DATA
384 v = REVLOGV2 | FLAG_GENERALDELTA | FLAG_INLINE_DATA
385 elif 'revlogv1' in opts:
385 elif 'revlogv1' in opts:
386 if 'generaldelta' in opts:
386 if 'generaldelta' in opts:
387 v |= FLAG_GENERALDELTA
387 v |= FLAG_GENERALDELTA
388 else:
388 else:
389 v = 0
389 v = 0
390 if 'chunkcachesize' in opts:
390 if 'chunkcachesize' in opts:
391 self._chunkcachesize = opts['chunkcachesize']
391 self._chunkcachesize = opts['chunkcachesize']
392 if 'maxchainlen' in opts:
392 if 'maxchainlen' in opts:
393 self._maxchainlen = opts['maxchainlen']
393 self._maxchainlen = opts['maxchainlen']
394 if 'deltabothparents' in opts:
394 if 'deltabothparents' in opts:
395 self._deltabothparents = opts['deltabothparents']
395 self._deltabothparents = opts['deltabothparents']
396 self._lazydeltabase = bool(opts.get('lazydeltabase', False))
396 self._lazydeltabase = bool(opts.get('lazydeltabase', False))
397 if 'compengine' in opts:
397 if 'compengine' in opts:
398 self._compengine = opts['compengine']
398 self._compengine = opts['compengine']
399 if 'maxdeltachainspan' in opts:
399 if 'maxdeltachainspan' in opts:
400 self._maxdeltachainspan = opts['maxdeltachainspan']
400 self._maxdeltachainspan = opts['maxdeltachainspan']
401 if mmaplargeindex and 'mmapindexthreshold' in opts:
401 if mmaplargeindex and 'mmapindexthreshold' in opts:
402 mmapindexthreshold = opts['mmapindexthreshold']
402 mmapindexthreshold = opts['mmapindexthreshold']
403 self._sparserevlog = bool(opts.get('sparse-revlog', False))
403 self._sparserevlog = bool(opts.get('sparse-revlog', False))
404 withsparseread = bool(opts.get('with-sparse-read', False))
404 withsparseread = bool(opts.get('with-sparse-read', False))
405 # sparse-revlog forces sparse-read
405 # sparse-revlog forces sparse-read
406 self._withsparseread = self._sparserevlog or withsparseread
406 self._withsparseread = self._sparserevlog or withsparseread
407 if 'sparse-read-density-threshold' in opts:
407 if 'sparse-read-density-threshold' in opts:
408 self._srdensitythreshold = opts['sparse-read-density-threshold']
408 self._srdensitythreshold = opts['sparse-read-density-threshold']
409 if 'sparse-read-min-gap-size' in opts:
409 if 'sparse-read-min-gap-size' in opts:
410 self._srmingapsize = opts['sparse-read-min-gap-size']
410 self._srmingapsize = opts['sparse-read-min-gap-size']
411 if opts.get('enableellipsis'):
411 if opts.get('enableellipsis'):
412 self._flagprocessors[REVIDX_ELLIPSIS] = ellipsisprocessor
412 self._flagprocessors[REVIDX_ELLIPSIS] = ellipsisprocessor
413
413
414 # revlog v0 doesn't have flag processors
414 # revlog v0 doesn't have flag processors
415 for flag, processor in opts.get(b'flagprocessors', {}).iteritems():
415 for flag, processor in opts.get(b'flagprocessors', {}).iteritems():
416 _insertflagprocessor(flag, processor, self._flagprocessors)
416 _insertflagprocessor(flag, processor, self._flagprocessors)
417
417
418 if self._chunkcachesize <= 0:
418 if self._chunkcachesize <= 0:
419 raise error.RevlogError(_('revlog chunk cache size %r is not '
419 raise error.RevlogError(_('revlog chunk cache size %r is not '
420 'greater than 0') % self._chunkcachesize)
420 'greater than 0') % self._chunkcachesize)
421 elif self._chunkcachesize & (self._chunkcachesize - 1):
421 elif self._chunkcachesize & (self._chunkcachesize - 1):
422 raise error.RevlogError(_('revlog chunk cache size %r is not a '
422 raise error.RevlogError(_('revlog chunk cache size %r is not a '
423 'power of 2') % self._chunkcachesize)
423 'power of 2') % self._chunkcachesize)
424
424
425 self._loadindex(v, mmapindexthreshold)
425 self._loadindex(v, mmapindexthreshold)
426
426
427 def _loadindex(self, v, mmapindexthreshold):
427 def _loadindex(self, v, mmapindexthreshold):
428 indexdata = ''
428 indexdata = ''
429 self._initempty = True
429 self._initempty = True
430 try:
430 try:
431 with self._indexfp() as f:
431 with self._indexfp() as f:
432 if (mmapindexthreshold is not None and
432 if (mmapindexthreshold is not None and
433 self.opener.fstat(f).st_size >= mmapindexthreshold):
433 self.opener.fstat(f).st_size >= mmapindexthreshold):
434 indexdata = util.buffer(util.mmapread(f))
434 indexdata = util.buffer(util.mmapread(f))
435 else:
435 else:
436 indexdata = f.read()
436 indexdata = f.read()
437 if len(indexdata) > 0:
437 if len(indexdata) > 0:
438 v = versionformat_unpack(indexdata[:4])[0]
438 v = versionformat_unpack(indexdata[:4])[0]
439 self._initempty = False
439 self._initempty = False
440 except IOError as inst:
440 except IOError as inst:
441 if inst.errno != errno.ENOENT:
441 if inst.errno != errno.ENOENT:
442 raise
442 raise
443
443
444 self.version = v
444 self.version = v
445 self._inline = v & FLAG_INLINE_DATA
445 self._inline = v & FLAG_INLINE_DATA
446 self._generaldelta = v & FLAG_GENERALDELTA
446 self._generaldelta = v & FLAG_GENERALDELTA
447 flags = v & ~0xFFFF
447 flags = v & ~0xFFFF
448 fmt = v & 0xFFFF
448 fmt = v & 0xFFFF
449 if fmt == REVLOGV0:
449 if fmt == REVLOGV0:
450 if flags:
450 if flags:
451 raise error.RevlogError(_('unknown flags (%#04x) in version %d '
451 raise error.RevlogError(_('unknown flags (%#04x) in version %d '
452 'revlog %s') %
452 'revlog %s') %
453 (flags >> 16, fmt, self.indexfile))
453 (flags >> 16, fmt, self.indexfile))
454 elif fmt == REVLOGV1:
454 elif fmt == REVLOGV1:
455 if flags & ~REVLOGV1_FLAGS:
455 if flags & ~REVLOGV1_FLAGS:
456 raise error.RevlogError(_('unknown flags (%#04x) in version %d '
456 raise error.RevlogError(_('unknown flags (%#04x) in version %d '
457 'revlog %s') %
457 'revlog %s') %
458 (flags >> 16, fmt, self.indexfile))
458 (flags >> 16, fmt, self.indexfile))
459 elif fmt == REVLOGV2:
459 elif fmt == REVLOGV2:
460 if flags & ~REVLOGV2_FLAGS:
460 if flags & ~REVLOGV2_FLAGS:
461 raise error.RevlogError(_('unknown flags (%#04x) in version %d '
461 raise error.RevlogError(_('unknown flags (%#04x) in version %d '
462 'revlog %s') %
462 'revlog %s') %
463 (flags >> 16, fmt, self.indexfile))
463 (flags >> 16, fmt, self.indexfile))
464 else:
464 else:
465 raise error.RevlogError(_('unknown version (%d) in revlog %s') %
465 raise error.RevlogError(_('unknown version (%d) in revlog %s') %
466 (fmt, self.indexfile))
466 (fmt, self.indexfile))
467
467
468 self._storedeltachains = True
468 self._storedeltachains = True
469
469
470 self._io = revlogio()
470 self._io = revlogio()
471 if self.version == REVLOGV0:
471 if self.version == REVLOGV0:
472 self._io = revlogoldio()
472 self._io = revlogoldio()
473 try:
473 try:
474 d = self._io.parseindex(indexdata, self._inline)
474 d = self._io.parseindex(indexdata, self._inline)
475 except (ValueError, IndexError):
475 except (ValueError, IndexError):
476 raise error.RevlogError(_("index %s is corrupted") %
476 raise error.RevlogError(_("index %s is corrupted") %
477 self.indexfile)
477 self.indexfile)
478 self.index, nodemap, self._chunkcache = d
478 self.index, nodemap, self._chunkcache = d
479 if nodemap is not None:
479 if nodemap is not None:
480 self.nodemap = self._nodecache = nodemap
480 self.nodemap = self._nodecache = nodemap
481 if not self._chunkcache:
481 if not self._chunkcache:
482 self._chunkclear()
482 self._chunkclear()
483 # revnum -> (chain-length, sum-delta-length)
483 # revnum -> (chain-length, sum-delta-length)
484 self._chaininfocache = {}
484 self._chaininfocache = {}
485 # revlog header -> revlog compressor
485 # revlog header -> revlog compressor
486 self._decompressors = {}
486 self._decompressors = {}
487
487
488 @util.propertycache
488 @util.propertycache
489 def _compressor(self):
489 def _compressor(self):
490 return util.compengines[self._compengine].revlogcompressor()
490 return util.compengines[self._compengine].revlogcompressor()
491
491
492 def _indexfp(self, mode='r'):
492 def _indexfp(self, mode='r'):
493 """file object for the revlog's index file"""
493 """file object for the revlog's index file"""
494 args = {r'mode': mode}
494 args = {r'mode': mode}
495 if mode != 'r':
495 if mode != 'r':
496 args[r'checkambig'] = self._checkambig
496 args[r'checkambig'] = self._checkambig
497 if mode == 'w':
497 if mode == 'w':
498 args[r'atomictemp'] = True
498 args[r'atomictemp'] = True
499 return self.opener(self.indexfile, **args)
499 return self.opener(self.indexfile, **args)
500
500
501 def _datafp(self, mode='r'):
501 def _datafp(self, mode='r'):
502 """file object for the revlog's data file"""
502 """file object for the revlog's data file"""
503 return self.opener(self.datafile, mode=mode)
503 return self.opener(self.datafile, mode=mode)
504
504
505 @contextlib.contextmanager
505 @contextlib.contextmanager
506 def _datareadfp(self, existingfp=None):
506 def _datareadfp(self, existingfp=None):
507 """file object suitable to read data"""
507 """file object suitable to read data"""
508 if existingfp is not None:
508 if existingfp is not None:
509 yield existingfp
509 yield existingfp
510 else:
510 else:
511 if self._inline:
511 if self._inline:
512 func = self._indexfp
512 func = self._indexfp
513 else:
513 else:
514 func = self._datafp
514 func = self._datafp
515 with func() as fp:
515 with func() as fp:
516 yield fp
516 yield fp
517
517
518 def tip(self):
518 def tip(self):
519 return self.node(len(self.index) - 1)
519 return self.node(len(self.index) - 1)
520 def __contains__(self, rev):
520 def __contains__(self, rev):
521 return 0 <= rev < len(self)
521 return 0 <= rev < len(self)
522 def __len__(self):
522 def __len__(self):
523 return len(self.index)
523 return len(self.index)
524 def __iter__(self):
524 def __iter__(self):
525 return iter(pycompat.xrange(len(self)))
525 return iter(pycompat.xrange(len(self)))
526 def revs(self, start=0, stop=None):
526 def revs(self, start=0, stop=None):
527 """iterate over all rev in this revlog (from start to stop)"""
527 """iterate over all rev in this revlog (from start to stop)"""
528 return storageutil.iterrevs(len(self), start=start, stop=stop)
528 return storageutil.iterrevs(len(self), start=start, stop=stop)
529
529
530 @util.propertycache
530 @util.propertycache
531 def nodemap(self):
531 def nodemap(self):
532 if self.index:
532 if self.index:
533 # populate mapping down to the initial node
533 # populate mapping down to the initial node
534 node0 = self.index[0][7] # get around changelog filtering
534 node0 = self.index[0][7] # get around changelog filtering
535 self.rev(node0)
535 self.rev(node0)
536 return self._nodecache
536 return self._nodecache
537
537
538 def hasnode(self, node):
538 def hasnode(self, node):
539 try:
539 try:
540 self.rev(node)
540 self.rev(node)
541 return True
541 return True
542 except KeyError:
542 except KeyError:
543 return False
543 return False
544
544
545 def candelta(self, baserev, rev):
545 def candelta(self, baserev, rev):
546 """whether two revisions (baserev, rev) can be delta-ed or not"""
546 """whether two revisions (baserev, rev) can be delta-ed or not"""
547 # Disable delta if either rev requires a content-changing flag
547 # Disable delta if either rev requires a content-changing flag
548 # processor (ex. LFS). This is because such flag processor can alter
548 # processor (ex. LFS). This is because such flag processor can alter
549 # the rawtext content that the delta will be based on, and two clients
549 # the rawtext content that the delta will be based on, and two clients
550 # could have a same revlog node with different flags (i.e. different
550 # could have a same revlog node with different flags (i.e. different
551 # rawtext contents) and the delta could be incompatible.
551 # rawtext contents) and the delta could be incompatible.
552 if ((self.flags(baserev) & REVIDX_RAWTEXT_CHANGING_FLAGS)
552 if ((self.flags(baserev) & REVIDX_RAWTEXT_CHANGING_FLAGS)
553 or (self.flags(rev) & REVIDX_RAWTEXT_CHANGING_FLAGS)):
553 or (self.flags(rev) & REVIDX_RAWTEXT_CHANGING_FLAGS)):
554 return False
554 return False
555 return True
555 return True
556
556
557 def clearcaches(self):
557 def clearcaches(self):
558 self._revisioncache = None
558 self._revisioncache = None
559 self._chainbasecache.clear()
559 self._chainbasecache.clear()
560 self._chunkcache = (0, '')
560 self._chunkcache = (0, '')
561 self._pcache = {}
561 self._pcache = {}
562
562
563 try:
563 try:
564 self._nodecache.clearcaches()
564 self._nodecache.clearcaches()
565 except AttributeError:
565 except AttributeError:
566 self._nodecache = {nullid: nullrev}
566 self._nodecache = {nullid: nullrev}
567 self._nodepos = None
567 self._nodepos = None
568
568
569 def rev(self, node):
569 def rev(self, node):
570 try:
570 try:
571 return self._nodecache[node]
571 return self._nodecache[node]
572 except TypeError:
572 except TypeError:
573 raise
573 raise
574 except error.RevlogError:
574 except error.RevlogError:
575 # parsers.c radix tree lookup failed
575 # parsers.c radix tree lookup failed
576 if node == wdirid or node in wdirfilenodeids:
576 if node == wdirid or node in wdirfilenodeids:
577 raise error.WdirUnsupported
577 raise error.WdirUnsupported
578 raise error.LookupError(node, self.indexfile, _('no node'))
578 raise error.LookupError(node, self.indexfile, _('no node'))
579 except KeyError:
579 except KeyError:
580 # pure python cache lookup failed
580 # pure python cache lookup failed
581 n = self._nodecache
581 n = self._nodecache
582 i = self.index
582 i = self.index
583 p = self._nodepos
583 p = self._nodepos
584 if p is None:
584 if p is None:
585 p = len(i) - 1
585 p = len(i) - 1
586 else:
586 else:
587 assert p < len(i)
587 assert p < len(i)
588 for r in pycompat.xrange(p, -1, -1):
588 for r in pycompat.xrange(p, -1, -1):
589 v = i[r][7]
589 v = i[r][7]
590 n[v] = r
590 n[v] = r
591 if v == node:
591 if v == node:
592 self._nodepos = r - 1
592 self._nodepos = r - 1
593 return r
593 return r
594 if node == wdirid or node in wdirfilenodeids:
594 if node == wdirid or node in wdirfilenodeids:
595 raise error.WdirUnsupported
595 raise error.WdirUnsupported
596 raise error.LookupError(node, self.indexfile, _('no node'))
596 raise error.LookupError(node, self.indexfile, _('no node'))
597
597
598 # Accessors for index entries.
598 # Accessors for index entries.
599
599
600 # First tuple entry is 8 bytes. First 6 bytes are offset. Last 2 bytes
600 # First tuple entry is 8 bytes. First 6 bytes are offset. Last 2 bytes
601 # are flags.
601 # are flags.
602 def start(self, rev):
602 def start(self, rev):
603 return int(self.index[rev][0] >> 16)
603 return int(self.index[rev][0] >> 16)
604
604
605 def flags(self, rev):
605 def flags(self, rev):
606 return self.index[rev][0] & 0xFFFF
606 return self.index[rev][0] & 0xFFFF
607
607
608 def length(self, rev):
608 def length(self, rev):
609 return self.index[rev][1]
609 return self.index[rev][1]
610
610
611 def rawsize(self, rev):
611 def rawsize(self, rev):
612 """return the length of the uncompressed text for a given revision"""
612 """return the length of the uncompressed text for a given revision"""
613 l = self.index[rev][2]
613 l = self.index[rev][2]
614 if l >= 0:
614 if l >= 0:
615 return l
615 return l
616
616
617 t = self.revision(rev, raw=True)
617 t = self.revision(rev, raw=True)
618 return len(t)
618 return len(t)
619
619
620 def size(self, rev):
620 def size(self, rev):
621 """length of non-raw text (processed by a "read" flag processor)"""
621 """length of non-raw text (processed by a "read" flag processor)"""
622 # fast path: if no "read" flag processor could change the content,
622 # fast path: if no "read" flag processor could change the content,
623 # size is rawsize. note: ELLIPSIS is known to not change the content.
623 # size is rawsize. note: ELLIPSIS is known to not change the content.
624 flags = self.flags(rev)
624 flags = self.flags(rev)
625 if flags & (REVIDX_KNOWN_FLAGS ^ REVIDX_ELLIPSIS) == 0:
625 if flags & (REVIDX_KNOWN_FLAGS ^ REVIDX_ELLIPSIS) == 0:
626 return self.rawsize(rev)
626 return self.rawsize(rev)
627
627
628 return len(self.revision(rev, raw=False))
628 return len(self.revision(rev, raw=False))
629
629
630 def chainbase(self, rev):
630 def chainbase(self, rev):
631 base = self._chainbasecache.get(rev)
631 base = self._chainbasecache.get(rev)
632 if base is not None:
632 if base is not None:
633 return base
633 return base
634
634
635 index = self.index
635 index = self.index
636 iterrev = rev
636 iterrev = rev
637 base = index[iterrev][3]
637 base = index[iterrev][3]
638 while base != iterrev:
638 while base != iterrev:
639 iterrev = base
639 iterrev = base
640 base = index[iterrev][3]
640 base = index[iterrev][3]
641
641
642 self._chainbasecache[rev] = base
642 self._chainbasecache[rev] = base
643 return base
643 return base
644
644
645 def linkrev(self, rev):
645 def linkrev(self, rev):
646 return self.index[rev][4]
646 return self.index[rev][4]
647
647
648 def parentrevs(self, rev):
648 def parentrevs(self, rev):
649 try:
649 try:
650 entry = self.index[rev]
650 entry = self.index[rev]
651 except IndexError:
651 except IndexError:
652 if rev == wdirrev:
652 if rev == wdirrev:
653 raise error.WdirUnsupported
653 raise error.WdirUnsupported
654 raise
654 raise
655
655
656 return entry[5], entry[6]
656 return entry[5], entry[6]
657
657
658 # fast parentrevs(rev) where rev isn't filtered
658 # fast parentrevs(rev) where rev isn't filtered
659 _uncheckedparentrevs = parentrevs
659 _uncheckedparentrevs = parentrevs
660
660
661 def node(self, rev):
661 def node(self, rev):
662 try:
662 try:
663 return self.index[rev][7]
663 return self.index[rev][7]
664 except IndexError:
664 except IndexError:
665 if rev == wdirrev:
665 if rev == wdirrev:
666 raise error.WdirUnsupported
666 raise error.WdirUnsupported
667 raise
667 raise
668
668
669 # Derived from index values.
669 # Derived from index values.
670
670
671 def end(self, rev):
671 def end(self, rev):
672 return self.start(rev) + self.length(rev)
672 return self.start(rev) + self.length(rev)
673
673
674 def parents(self, node):
674 def parents(self, node):
675 i = self.index
675 i = self.index
676 d = i[self.rev(node)]
676 d = i[self.rev(node)]
677 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
677 return i[d[5]][7], i[d[6]][7] # map revisions to nodes inline
678
678
679 def chainlen(self, rev):
679 def chainlen(self, rev):
680 return self._chaininfo(rev)[0]
680 return self._chaininfo(rev)[0]
681
681
682 def _chaininfo(self, rev):
682 def _chaininfo(self, rev):
683 chaininfocache = self._chaininfocache
683 chaininfocache = self._chaininfocache
684 if rev in chaininfocache:
684 if rev in chaininfocache:
685 return chaininfocache[rev]
685 return chaininfocache[rev]
686 index = self.index
686 index = self.index
687 generaldelta = self._generaldelta
687 generaldelta = self._generaldelta
688 iterrev = rev
688 iterrev = rev
689 e = index[iterrev]
689 e = index[iterrev]
690 clen = 0
690 clen = 0
691 compresseddeltalen = 0
691 compresseddeltalen = 0
692 while iterrev != e[3]:
692 while iterrev != e[3]:
693 clen += 1
693 clen += 1
694 compresseddeltalen += e[1]
694 compresseddeltalen += e[1]
695 if generaldelta:
695 if generaldelta:
696 iterrev = e[3]
696 iterrev = e[3]
697 else:
697 else:
698 iterrev -= 1
698 iterrev -= 1
699 if iterrev in chaininfocache:
699 if iterrev in chaininfocache:
700 t = chaininfocache[iterrev]
700 t = chaininfocache[iterrev]
701 clen += t[0]
701 clen += t[0]
702 compresseddeltalen += t[1]
702 compresseddeltalen += t[1]
703 break
703 break
704 e = index[iterrev]
704 e = index[iterrev]
705 else:
705 else:
706 # Add text length of base since decompressing that also takes
706 # Add text length of base since decompressing that also takes
707 # work. For cache hits the length is already included.
707 # work. For cache hits the length is already included.
708 compresseddeltalen += e[1]
708 compresseddeltalen += e[1]
709 r = (clen, compresseddeltalen)
709 r = (clen, compresseddeltalen)
710 chaininfocache[rev] = r
710 chaininfocache[rev] = r
711 return r
711 return r
712
712
713 def _deltachain(self, rev, stoprev=None):
713 def _deltachain(self, rev, stoprev=None):
714 """Obtain the delta chain for a revision.
714 """Obtain the delta chain for a revision.
715
715
716 ``stoprev`` specifies a revision to stop at. If not specified, we
716 ``stoprev`` specifies a revision to stop at. If not specified, we
717 stop at the base of the chain.
717 stop at the base of the chain.
718
718
719 Returns a 2-tuple of (chain, stopped) where ``chain`` is a list of
719 Returns a 2-tuple of (chain, stopped) where ``chain`` is a list of
720 revs in ascending order and ``stopped`` is a bool indicating whether
720 revs in ascending order and ``stopped`` is a bool indicating whether
721 ``stoprev`` was hit.
721 ``stoprev`` was hit.
722 """
722 """
723 # Try C implementation.
723 # Try C implementation.
724 try:
724 try:
725 return self.index.deltachain(rev, stoprev, self._generaldelta)
725 return self.index.deltachain(rev, stoprev, self._generaldelta)
726 except AttributeError:
726 except AttributeError:
727 pass
727 pass
728
728
729 chain = []
729 chain = []
730
730
731 # Alias to prevent attribute lookup in tight loop.
731 # Alias to prevent attribute lookup in tight loop.
732 index = self.index
732 index = self.index
733 generaldelta = self._generaldelta
733 generaldelta = self._generaldelta
734
734
735 iterrev = rev
735 iterrev = rev
736 e = index[iterrev]
736 e = index[iterrev]
737 while iterrev != e[3] and iterrev != stoprev:
737 while iterrev != e[3] and iterrev != stoprev:
738 chain.append(iterrev)
738 chain.append(iterrev)
739 if generaldelta:
739 if generaldelta:
740 iterrev = e[3]
740 iterrev = e[3]
741 else:
741 else:
742 iterrev -= 1
742 iterrev -= 1
743 e = index[iterrev]
743 e = index[iterrev]
744
744
745 if iterrev == stoprev:
745 if iterrev == stoprev:
746 stopped = True
746 stopped = True
747 else:
747 else:
748 chain.append(iterrev)
748 chain.append(iterrev)
749 stopped = False
749 stopped = False
750
750
751 chain.reverse()
751 chain.reverse()
752 return chain, stopped
752 return chain, stopped
753
753
754 def ancestors(self, revs, stoprev=0, inclusive=False):
754 def ancestors(self, revs, stoprev=0, inclusive=False):
755 """Generate the ancestors of 'revs' in reverse topological order.
755 """Generate the ancestors of 'revs' in reverse topological order.
756 Does not generate revs lower than stoprev.
756 Does not generate revs lower than stoprev.
757
757
758 See the documentation for ancestor.lazyancestors for more details."""
758 See the documentation for ancestor.lazyancestors for more details."""
759
759
760 # first, make sure start revisions aren't filtered
760 # first, make sure start revisions aren't filtered
761 revs = list(revs)
761 revs = list(revs)
762 checkrev = self.node
762 checkrev = self.node
763 for r in revs:
763 for r in revs:
764 checkrev(r)
764 checkrev(r)
765 # and we're sure ancestors aren't filtered as well
765 # and we're sure ancestors aren't filtered as well
766 if util.safehasattr(parsers, 'rustlazyancestors'):
766 if util.safehasattr(parsers, 'rustlazyancestors'):
767 return ancestor.rustlazyancestors(
767 return ancestor.rustlazyancestors(
768 self.index, revs,
768 self.index, revs,
769 stoprev=stoprev, inclusive=inclusive)
769 stoprev=stoprev, inclusive=inclusive)
770 return ancestor.lazyancestors(self._uncheckedparentrevs, revs,
770 return ancestor.lazyancestors(self._uncheckedparentrevs, revs,
771 stoprev=stoprev, inclusive=inclusive)
771 stoprev=stoprev, inclusive=inclusive)
772
772
773 def descendants(self, revs):
773 def descendants(self, revs):
774 return dagop.descendantrevs(revs, self.revs, self.parentrevs)
774 return dagop.descendantrevs(revs, self.revs, self.parentrevs)
775
775
776 def findcommonmissing(self, common=None, heads=None):
776 def findcommonmissing(self, common=None, heads=None):
777 """Return a tuple of the ancestors of common and the ancestors of heads
777 """Return a tuple of the ancestors of common and the ancestors of heads
778 that are not ancestors of common. In revset terminology, we return the
778 that are not ancestors of common. In revset terminology, we return the
779 tuple:
779 tuple:
780
780
781 ::common, (::heads) - (::common)
781 ::common, (::heads) - (::common)
782
782
783 The list is sorted by revision number, meaning it is
783 The list is sorted by revision number, meaning it is
784 topologically sorted.
784 topologically sorted.
785
785
786 'heads' and 'common' are both lists of node IDs. If heads is
786 'heads' and 'common' are both lists of node IDs. If heads is
787 not supplied, uses all of the revlog's heads. If common is not
787 not supplied, uses all of the revlog's heads. If common is not
788 supplied, uses nullid."""
788 supplied, uses nullid."""
789 if common is None:
789 if common is None:
790 common = [nullid]
790 common = [nullid]
791 if heads is None:
791 if heads is None:
792 heads = self.heads()
792 heads = self.heads()
793
793
794 common = [self.rev(n) for n in common]
794 common = [self.rev(n) for n in common]
795 heads = [self.rev(n) for n in heads]
795 heads = [self.rev(n) for n in heads]
796
796
797 # we want the ancestors, but inclusive
797 # we want the ancestors, but inclusive
798 class lazyset(object):
798 class lazyset(object):
799 def __init__(self, lazyvalues):
799 def __init__(self, lazyvalues):
800 self.addedvalues = set()
800 self.addedvalues = set()
801 self.lazyvalues = lazyvalues
801 self.lazyvalues = lazyvalues
802
802
803 def __contains__(self, value):
803 def __contains__(self, value):
804 return value in self.addedvalues or value in self.lazyvalues
804 return value in self.addedvalues or value in self.lazyvalues
805
805
806 def __iter__(self):
806 def __iter__(self):
807 added = self.addedvalues
807 added = self.addedvalues
808 for r in added:
808 for r in added:
809 yield r
809 yield r
810 for r in self.lazyvalues:
810 for r in self.lazyvalues:
811 if not r in added:
811 if not r in added:
812 yield r
812 yield r
813
813
814 def add(self, value):
814 def add(self, value):
815 self.addedvalues.add(value)
815 self.addedvalues.add(value)
816
816
817 def update(self, values):
817 def update(self, values):
818 self.addedvalues.update(values)
818 self.addedvalues.update(values)
819
819
820 has = lazyset(self.ancestors(common))
820 has = lazyset(self.ancestors(common))
821 has.add(nullrev)
821 has.add(nullrev)
822 has.update(common)
822 has.update(common)
823
823
824 # take all ancestors from heads that aren't in has
824 # take all ancestors from heads that aren't in has
825 missing = set()
825 missing = set()
826 visit = collections.deque(r for r in heads if r not in has)
826 visit = collections.deque(r for r in heads if r not in has)
827 while visit:
827 while visit:
828 r = visit.popleft()
828 r = visit.popleft()
829 if r in missing:
829 if r in missing:
830 continue
830 continue
831 else:
831 else:
832 missing.add(r)
832 missing.add(r)
833 for p in self.parentrevs(r):
833 for p in self.parentrevs(r):
834 if p not in has:
834 if p not in has:
835 visit.append(p)
835 visit.append(p)
836 missing = list(missing)
836 missing = list(missing)
837 missing.sort()
837 missing.sort()
838 return has, [self.node(miss) for miss in missing]
838 return has, [self.node(miss) for miss in missing]
839
839
840 def incrementalmissingrevs(self, common=None):
840 def incrementalmissingrevs(self, common=None):
841 """Return an object that can be used to incrementally compute the
841 """Return an object that can be used to incrementally compute the
842 revision numbers of the ancestors of arbitrary sets that are not
842 revision numbers of the ancestors of arbitrary sets that are not
843 ancestors of common. This is an ancestor.incrementalmissingancestors
843 ancestors of common. This is an ancestor.incrementalmissingancestors
844 object.
844 object.
845
845
846 'common' is a list of revision numbers. If common is not supplied, uses
846 'common' is a list of revision numbers. If common is not supplied, uses
847 nullrev.
847 nullrev.
848 """
848 """
849 if common is None:
849 if common is None:
850 common = [nullrev]
850 common = [nullrev]
851
851
852 return ancestor.incrementalmissingancestors(self.parentrevs, common)
852 return ancestor.incrementalmissingancestors(self.parentrevs, common)
853
853
854 def findmissingrevs(self, common=None, heads=None):
854 def findmissingrevs(self, common=None, heads=None):
855 """Return the revision numbers of the ancestors of heads that
855 """Return the revision numbers of the ancestors of heads that
856 are not ancestors of common.
856 are not ancestors of common.
857
857
858 More specifically, return a list of revision numbers corresponding to
858 More specifically, return a list of revision numbers corresponding to
859 nodes N such that every N satisfies the following constraints:
859 nodes N such that every N satisfies the following constraints:
860
860
861 1. N is an ancestor of some node in 'heads'
861 1. N is an ancestor of some node in 'heads'
862 2. N is not an ancestor of any node in 'common'
862 2. N is not an ancestor of any node in 'common'
863
863
864 The list is sorted by revision number, meaning it is
864 The list is sorted by revision number, meaning it is
865 topologically sorted.
865 topologically sorted.
866
866
867 'heads' and 'common' are both lists of revision numbers. If heads is
867 'heads' and 'common' are both lists of revision numbers. If heads is
868 not supplied, uses all of the revlog's heads. If common is not
868 not supplied, uses all of the revlog's heads. If common is not
869 supplied, uses nullid."""
869 supplied, uses nullid."""
870 if common is None:
870 if common is None:
871 common = [nullrev]
871 common = [nullrev]
872 if heads is None:
872 if heads is None:
873 heads = self.headrevs()
873 heads = self.headrevs()
874
874
875 inc = self.incrementalmissingrevs(common=common)
875 inc = self.incrementalmissingrevs(common=common)
876 return inc.missingancestors(heads)
876 return inc.missingancestors(heads)
877
877
878 def findmissing(self, common=None, heads=None):
878 def findmissing(self, common=None, heads=None):
879 """Return the ancestors of heads that are not ancestors of common.
879 """Return the ancestors of heads that are not ancestors of common.
880
880
881 More specifically, return a list of nodes N such that every N
881 More specifically, return a list of nodes N such that every N
882 satisfies the following constraints:
882 satisfies the following constraints:
883
883
884 1. N is an ancestor of some node in 'heads'
884 1. N is an ancestor of some node in 'heads'
885 2. N is not an ancestor of any node in 'common'
885 2. N is not an ancestor of any node in 'common'
886
886
887 The list is sorted by revision number, meaning it is
887 The list is sorted by revision number, meaning it is
888 topologically sorted.
888 topologically sorted.
889
889
890 'heads' and 'common' are both lists of node IDs. If heads is
890 'heads' and 'common' are both lists of node IDs. If heads is
891 not supplied, uses all of the revlog's heads. If common is not
891 not supplied, uses all of the revlog's heads. If common is not
892 supplied, uses nullid."""
892 supplied, uses nullid."""
893 if common is None:
893 if common is None:
894 common = [nullid]
894 common = [nullid]
895 if heads is None:
895 if heads is None:
896 heads = self.heads()
896 heads = self.heads()
897
897
898 common = [self.rev(n) for n in common]
898 common = [self.rev(n) for n in common]
899 heads = [self.rev(n) for n in heads]
899 heads = [self.rev(n) for n in heads]
900
900
901 inc = self.incrementalmissingrevs(common=common)
901 inc = self.incrementalmissingrevs(common=common)
902 return [self.node(r) for r in inc.missingancestors(heads)]
902 return [self.node(r) for r in inc.missingancestors(heads)]
903
903
904 def nodesbetween(self, roots=None, heads=None):
904 def nodesbetween(self, roots=None, heads=None):
905 """Return a topological path from 'roots' to 'heads'.
905 """Return a topological path from 'roots' to 'heads'.
906
906
907 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
907 Return a tuple (nodes, outroots, outheads) where 'nodes' is a
908 topologically sorted list of all nodes N that satisfy both of
908 topologically sorted list of all nodes N that satisfy both of
909 these constraints:
909 these constraints:
910
910
911 1. N is a descendant of some node in 'roots'
911 1. N is a descendant of some node in 'roots'
912 2. N is an ancestor of some node in 'heads'
912 2. N is an ancestor of some node in 'heads'
913
913
914 Every node is considered to be both a descendant and an ancestor
914 Every node is considered to be both a descendant and an ancestor
915 of itself, so every reachable node in 'roots' and 'heads' will be
915 of itself, so every reachable node in 'roots' and 'heads' will be
916 included in 'nodes'.
916 included in 'nodes'.
917
917
918 'outroots' is the list of reachable nodes in 'roots', i.e., the
918 'outroots' is the list of reachable nodes in 'roots', i.e., the
919 subset of 'roots' that is returned in 'nodes'. Likewise,
919 subset of 'roots' that is returned in 'nodes'. Likewise,
920 'outheads' is the subset of 'heads' that is also in 'nodes'.
920 'outheads' is the subset of 'heads' that is also in 'nodes'.
921
921
922 'roots' and 'heads' are both lists of node IDs. If 'roots' is
922 'roots' and 'heads' are both lists of node IDs. If 'roots' is
923 unspecified, uses nullid as the only root. If 'heads' is
923 unspecified, uses nullid as the only root. If 'heads' is
924 unspecified, uses list of all of the revlog's heads."""
924 unspecified, uses list of all of the revlog's heads."""
925 nonodes = ([], [], [])
925 nonodes = ([], [], [])
926 if roots is not None:
926 if roots is not None:
927 roots = list(roots)
927 roots = list(roots)
928 if not roots:
928 if not roots:
929 return nonodes
929 return nonodes
930 lowestrev = min([self.rev(n) for n in roots])
930 lowestrev = min([self.rev(n) for n in roots])
931 else:
931 else:
932 roots = [nullid] # Everybody's a descendant of nullid
932 roots = [nullid] # Everybody's a descendant of nullid
933 lowestrev = nullrev
933 lowestrev = nullrev
934 if (lowestrev == nullrev) and (heads is None):
934 if (lowestrev == nullrev) and (heads is None):
935 # We want _all_ the nodes!
935 # We want _all_ the nodes!
936 return ([self.node(r) for r in self], [nullid], list(self.heads()))
936 return ([self.node(r) for r in self], [nullid], list(self.heads()))
937 if heads is None:
937 if heads is None:
938 # All nodes are ancestors, so the latest ancestor is the last
938 # All nodes are ancestors, so the latest ancestor is the last
939 # node.
939 # node.
940 highestrev = len(self) - 1
940 highestrev = len(self) - 1
941 # Set ancestors to None to signal that every node is an ancestor.
941 # Set ancestors to None to signal that every node is an ancestor.
942 ancestors = None
942 ancestors = None
943 # Set heads to an empty dictionary for later discovery of heads
943 # Set heads to an empty dictionary for later discovery of heads
944 heads = {}
944 heads = {}
945 else:
945 else:
946 heads = list(heads)
946 heads = list(heads)
947 if not heads:
947 if not heads:
948 return nonodes
948 return nonodes
949 ancestors = set()
949 ancestors = set()
950 # Turn heads into a dictionary so we can remove 'fake' heads.
950 # Turn heads into a dictionary so we can remove 'fake' heads.
951 # Also, later we will be using it to filter out the heads we can't
951 # Also, later we will be using it to filter out the heads we can't
952 # find from roots.
952 # find from roots.
953 heads = dict.fromkeys(heads, False)
953 heads = dict.fromkeys(heads, False)
954 # Start at the top and keep marking parents until we're done.
954 # Start at the top and keep marking parents until we're done.
955 nodestotag = set(heads)
955 nodestotag = set(heads)
956 # Remember where the top was so we can use it as a limit later.
956 # Remember where the top was so we can use it as a limit later.
957 highestrev = max([self.rev(n) for n in nodestotag])
957 highestrev = max([self.rev(n) for n in nodestotag])
958 while nodestotag:
958 while nodestotag:
959 # grab a node to tag
959 # grab a node to tag
960 n = nodestotag.pop()
960 n = nodestotag.pop()
961 # Never tag nullid
961 # Never tag nullid
962 if n == nullid:
962 if n == nullid:
963 continue
963 continue
964 # A node's revision number represents its place in a
964 # A node's revision number represents its place in a
965 # topologically sorted list of nodes.
965 # topologically sorted list of nodes.
966 r = self.rev(n)
966 r = self.rev(n)
967 if r >= lowestrev:
967 if r >= lowestrev:
968 if n not in ancestors:
968 if n not in ancestors:
969 # If we are possibly a descendant of one of the roots
969 # If we are possibly a descendant of one of the roots
970 # and we haven't already been marked as an ancestor
970 # and we haven't already been marked as an ancestor
971 ancestors.add(n) # Mark as ancestor
971 ancestors.add(n) # Mark as ancestor
972 # Add non-nullid parents to list of nodes to tag.
972 # Add non-nullid parents to list of nodes to tag.
973 nodestotag.update([p for p in self.parents(n) if
973 nodestotag.update([p for p in self.parents(n) if
974 p != nullid])
974 p != nullid])
975 elif n in heads: # We've seen it before, is it a fake head?
975 elif n in heads: # We've seen it before, is it a fake head?
976 # So it is, real heads should not be the ancestors of
976 # So it is, real heads should not be the ancestors of
977 # any other heads.
977 # any other heads.
978 heads.pop(n)
978 heads.pop(n)
979 if not ancestors:
979 if not ancestors:
980 return nonodes
980 return nonodes
981 # Now that we have our set of ancestors, we want to remove any
981 # Now that we have our set of ancestors, we want to remove any
982 # roots that are not ancestors.
982 # roots that are not ancestors.
983
983
984 # If one of the roots was nullid, everything is included anyway.
984 # If one of the roots was nullid, everything is included anyway.
985 if lowestrev > nullrev:
985 if lowestrev > nullrev:
986 # But, since we weren't, let's recompute the lowest rev to not
986 # But, since we weren't, let's recompute the lowest rev to not
987 # include roots that aren't ancestors.
987 # include roots that aren't ancestors.
988
988
989 # Filter out roots that aren't ancestors of heads
989 # Filter out roots that aren't ancestors of heads
990 roots = [root for root in roots if root in ancestors]
990 roots = [root for root in roots if root in ancestors]
991 # Recompute the lowest revision
991 # Recompute the lowest revision
992 if roots:
992 if roots:
993 lowestrev = min([self.rev(root) for root in roots])
993 lowestrev = min([self.rev(root) for root in roots])
994 else:
994 else:
995 # No more roots? Return empty list
995 # No more roots? Return empty list
996 return nonodes
996 return nonodes
997 else:
997 else:
998 # We are descending from nullid, and don't need to care about
998 # We are descending from nullid, and don't need to care about
999 # any other roots.
999 # any other roots.
1000 lowestrev = nullrev
1000 lowestrev = nullrev
1001 roots = [nullid]
1001 roots = [nullid]
1002 # Transform our roots list into a set.
1002 # Transform our roots list into a set.
1003 descendants = set(roots)
1003 descendants = set(roots)
1004 # Also, keep the original roots so we can filter out roots that aren't
1004 # Also, keep the original roots so we can filter out roots that aren't
1005 # 'real' roots (i.e. are descended from other roots).
1005 # 'real' roots (i.e. are descended from other roots).
1006 roots = descendants.copy()
1006 roots = descendants.copy()
1007 # Our topologically sorted list of output nodes.
1007 # Our topologically sorted list of output nodes.
1008 orderedout = []
1008 orderedout = []
1009 # Don't start at nullid since we don't want nullid in our output list,
1009 # Don't start at nullid since we don't want nullid in our output list,
1010 # and if nullid shows up in descendants, empty parents will look like
1010 # and if nullid shows up in descendants, empty parents will look like
1011 # they're descendants.
1011 # they're descendants.
1012 for r in self.revs(start=max(lowestrev, 0), stop=highestrev + 1):
1012 for r in self.revs(start=max(lowestrev, 0), stop=highestrev + 1):
1013 n = self.node(r)
1013 n = self.node(r)
1014 isdescendant = False
1014 isdescendant = False
1015 if lowestrev == nullrev: # Everybody is a descendant of nullid
1015 if lowestrev == nullrev: # Everybody is a descendant of nullid
1016 isdescendant = True
1016 isdescendant = True
1017 elif n in descendants:
1017 elif n in descendants:
1018 # n is already a descendant
1018 # n is already a descendant
1019 isdescendant = True
1019 isdescendant = True
1020 # This check only needs to be done here because all the roots
1020 # This check only needs to be done here because all the roots
1021 # will start being marked is descendants before the loop.
1021 # will start being marked is descendants before the loop.
1022 if n in roots:
1022 if n in roots:
1023 # If n was a root, check if it's a 'real' root.
1023 # If n was a root, check if it's a 'real' root.
1024 p = tuple(self.parents(n))
1024 p = tuple(self.parents(n))
1025 # If any of its parents are descendants, it's not a root.
1025 # If any of its parents are descendants, it's not a root.
1026 if (p[0] in descendants) or (p[1] in descendants):
1026 if (p[0] in descendants) or (p[1] in descendants):
1027 roots.remove(n)
1027 roots.remove(n)
1028 else:
1028 else:
1029 p = tuple(self.parents(n))
1029 p = tuple(self.parents(n))
1030 # A node is a descendant if either of its parents are
1030 # A node is a descendant if either of its parents are
1031 # descendants. (We seeded the dependents list with the roots
1031 # descendants. (We seeded the dependents list with the roots
1032 # up there, remember?)
1032 # up there, remember?)
1033 if (p[0] in descendants) or (p[1] in descendants):
1033 if (p[0] in descendants) or (p[1] in descendants):
1034 descendants.add(n)
1034 descendants.add(n)
1035 isdescendant = True
1035 isdescendant = True
1036 if isdescendant and ((ancestors is None) or (n in ancestors)):
1036 if isdescendant and ((ancestors is None) or (n in ancestors)):
1037 # Only include nodes that are both descendants and ancestors.
1037 # Only include nodes that are both descendants and ancestors.
1038 orderedout.append(n)
1038 orderedout.append(n)
1039 if (ancestors is not None) and (n in heads):
1039 if (ancestors is not None) and (n in heads):
1040 # We're trying to figure out which heads are reachable
1040 # We're trying to figure out which heads are reachable
1041 # from roots.
1041 # from roots.
1042 # Mark this head as having been reached
1042 # Mark this head as having been reached
1043 heads[n] = True
1043 heads[n] = True
1044 elif ancestors is None:
1044 elif ancestors is None:
1045 # Otherwise, we're trying to discover the heads.
1045 # Otherwise, we're trying to discover the heads.
1046 # Assume this is a head because if it isn't, the next step
1046 # Assume this is a head because if it isn't, the next step
1047 # will eventually remove it.
1047 # will eventually remove it.
1048 heads[n] = True
1048 heads[n] = True
1049 # But, obviously its parents aren't.
1049 # But, obviously its parents aren't.
1050 for p in self.parents(n):
1050 for p in self.parents(n):
1051 heads.pop(p, None)
1051 heads.pop(p, None)
1052 heads = [head for head, flag in heads.iteritems() if flag]
1052 heads = [head for head, flag in heads.iteritems() if flag]
1053 roots = list(roots)
1053 roots = list(roots)
1054 assert orderedout
1054 assert orderedout
1055 assert roots
1055 assert roots
1056 assert heads
1056 assert heads
1057 return (orderedout, roots, heads)
1057 return (orderedout, roots, heads)
1058
1058
1059 def headrevs(self):
1059 def headrevs(self):
1060 try:
1060 try:
1061 return self.index.headrevs()
1061 return self.index.headrevs()
1062 except AttributeError:
1062 except AttributeError:
1063 return self._headrevs()
1063 return self._headrevs()
1064
1064
1065 def computephases(self, roots):
1065 def computephases(self, roots):
1066 return self.index.computephasesmapsets(roots)
1066 return self.index.computephasesmapsets(roots)
1067
1067
1068 def _headrevs(self):
1068 def _headrevs(self):
1069 count = len(self)
1069 count = len(self)
1070 if not count:
1070 if not count:
1071 return [nullrev]
1071 return [nullrev]
1072 # we won't iter over filtered rev so nobody is a head at start
1072 # we won't iter over filtered rev so nobody is a head at start
1073 ishead = [0] * (count + 1)
1073 ishead = [0] * (count + 1)
1074 index = self.index
1074 index = self.index
1075 for r in self:
1075 for r in self:
1076 ishead[r] = 1 # I may be an head
1076 ishead[r] = 1 # I may be an head
1077 e = index[r]
1077 e = index[r]
1078 ishead[e[5]] = ishead[e[6]] = 0 # my parent are not
1078 ishead[e[5]] = ishead[e[6]] = 0 # my parent are not
1079 return [r for r, val in enumerate(ishead) if val]
1079 return [r for r, val in enumerate(ishead) if val]
1080
1080
1081 def heads(self, start=None, stop=None):
1081 def heads(self, start=None, stop=None):
1082 """return the list of all nodes that have no children
1082 """return the list of all nodes that have no children
1083
1083
1084 if start is specified, only heads that are descendants of
1084 if start is specified, only heads that are descendants of
1085 start will be returned
1085 start will be returned
1086 if stop is specified, it will consider all the revs from stop
1086 if stop is specified, it will consider all the revs from stop
1087 as if they had no children
1087 as if they had no children
1088 """
1088 """
1089 if start is None and stop is None:
1089 if start is None and stop is None:
1090 if not len(self):
1090 if not len(self):
1091 return [nullid]
1091 return [nullid]
1092 return [self.node(r) for r in self.headrevs()]
1092 return [self.node(r) for r in self.headrevs()]
1093
1093
1094 if start is None:
1094 if start is None:
1095 start = nullrev
1095 start = nullrev
1096 else:
1096 else:
1097 start = self.rev(start)
1097 start = self.rev(start)
1098
1098
1099 stoprevs = set(self.rev(n) for n in stop or [])
1099 stoprevs = set(self.rev(n) for n in stop or [])
1100
1100
1101 revs = dagop.headrevssubset(self.revs, self.parentrevs, startrev=start,
1101 revs = dagop.headrevssubset(self.revs, self.parentrevs, startrev=start,
1102 stoprevs=stoprevs)
1102 stoprevs=stoprevs)
1103
1103
1104 return [self.node(rev) for rev in revs]
1104 return [self.node(rev) for rev in revs]
1105
1105
1106 def children(self, node):
1106 def children(self, node):
1107 """find the children of a given node"""
1107 """find the children of a given node"""
1108 c = []
1108 c = []
1109 p = self.rev(node)
1109 p = self.rev(node)
1110 for r in self.revs(start=p + 1):
1110 for r in self.revs(start=p + 1):
1111 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
1111 prevs = [pr for pr in self.parentrevs(r) if pr != nullrev]
1112 if prevs:
1112 if prevs:
1113 for pr in prevs:
1113 for pr in prevs:
1114 if pr == p:
1114 if pr == p:
1115 c.append(self.node(r))
1115 c.append(self.node(r))
1116 elif p == nullrev:
1116 elif p == nullrev:
1117 c.append(self.node(r))
1117 c.append(self.node(r))
1118 return c
1118 return c
1119
1119
1120 def commonancestorsheads(self, a, b):
1120 def commonancestorsheads(self, a, b):
1121 """calculate all the heads of the common ancestors of nodes a and b"""
1121 """calculate all the heads of the common ancestors of nodes a and b"""
1122 a, b = self.rev(a), self.rev(b)
1122 a, b = self.rev(a), self.rev(b)
1123 ancs = self._commonancestorsheads(a, b)
1123 ancs = self._commonancestorsheads(a, b)
1124 return pycompat.maplist(self.node, ancs)
1124 return pycompat.maplist(self.node, ancs)
1125
1125
1126 def _commonancestorsheads(self, *revs):
1126 def _commonancestorsheads(self, *revs):
1127 """calculate all the heads of the common ancestors of revs"""
1127 """calculate all the heads of the common ancestors of revs"""
1128 try:
1128 try:
1129 ancs = self.index.commonancestorsheads(*revs)
1129 ancs = self.index.commonancestorsheads(*revs)
1130 except (AttributeError, OverflowError): # C implementation failed
1130 except (AttributeError, OverflowError): # C implementation failed
1131 ancs = ancestor.commonancestorsheads(self.parentrevs, *revs)
1131 ancs = ancestor.commonancestorsheads(self.parentrevs, *revs)
1132 return ancs
1132 return ancs
1133
1133
1134 def isancestor(self, a, b):
1134 def isancestor(self, a, b):
1135 """return True if node a is an ancestor of node b
1135 """return True if node a is an ancestor of node b
1136
1136
1137 A revision is considered an ancestor of itself."""
1137 A revision is considered an ancestor of itself."""
1138 a, b = self.rev(a), self.rev(b)
1138 a, b = self.rev(a), self.rev(b)
1139 return self.isancestorrev(a, b)
1139 return self.isancestorrev(a, b)
1140
1140
1141 def isancestorrev(self, a, b):
1141 def isancestorrev(self, a, b):
1142 """return True if revision a is an ancestor of revision b
1142 """return True if revision a is an ancestor of revision b
1143
1143
1144 A revision is considered an ancestor of itself.
1144 A revision is considered an ancestor of itself.
1145
1145
1146 The implementation of this is trivial but the use of
1146 The implementation of this is trivial but the use of
1147 commonancestorsheads is not."""
1147 commonancestorsheads is not."""
1148 if a == nullrev:
1148 if a == nullrev:
1149 return True
1149 return True
1150 elif a == b:
1150 elif a == b:
1151 return True
1151 return True
1152 elif a > b:
1152 elif a > b:
1153 return False
1153 return False
1154 return a in self._commonancestorsheads(a, b)
1154 return a in self._commonancestorsheads(a, b)
1155
1155
1156 def ancestor(self, a, b):
1156 def ancestor(self, a, b):
1157 """calculate the "best" common ancestor of nodes a and b"""
1157 """calculate the "best" common ancestor of nodes a and b"""
1158
1158
1159 a, b = self.rev(a), self.rev(b)
1159 a, b = self.rev(a), self.rev(b)
1160 try:
1160 try:
1161 ancs = self.index.ancestors(a, b)
1161 ancs = self.index.ancestors(a, b)
1162 except (AttributeError, OverflowError):
1162 except (AttributeError, OverflowError):
1163 ancs = ancestor.ancestors(self.parentrevs, a, b)
1163 ancs = ancestor.ancestors(self.parentrevs, a, b)
1164 if ancs:
1164 if ancs:
1165 # choose a consistent winner when there's a tie
1165 # choose a consistent winner when there's a tie
1166 return min(map(self.node, ancs))
1166 return min(map(self.node, ancs))
1167 return nullid
1167 return nullid
1168
1168
1169 def _match(self, id):
1169 def _match(self, id):
1170 if isinstance(id, int):
1170 if isinstance(id, int):
1171 # rev
1171 # rev
1172 return self.node(id)
1172 return self.node(id)
1173 if len(id) == 20:
1173 if len(id) == 20:
1174 # possibly a binary node
1174 # possibly a binary node
1175 # odds of a binary node being all hex in ASCII are 1 in 10**25
1175 # odds of a binary node being all hex in ASCII are 1 in 10**25
1176 try:
1176 try:
1177 node = id
1177 node = id
1178 self.rev(node) # quick search the index
1178 self.rev(node) # quick search the index
1179 return node
1179 return node
1180 except error.LookupError:
1180 except error.LookupError:
1181 pass # may be partial hex id
1181 pass # may be partial hex id
1182 try:
1182 try:
1183 # str(rev)
1183 # str(rev)
1184 rev = int(id)
1184 rev = int(id)
1185 if "%d" % rev != id:
1185 if "%d" % rev != id:
1186 raise ValueError
1186 raise ValueError
1187 if rev < 0:
1187 if rev < 0:
1188 rev = len(self) + rev
1188 rev = len(self) + rev
1189 if rev < 0 or rev >= len(self):
1189 if rev < 0 or rev >= len(self):
1190 raise ValueError
1190 raise ValueError
1191 return self.node(rev)
1191 return self.node(rev)
1192 except (ValueError, OverflowError):
1192 except (ValueError, OverflowError):
1193 pass
1193 pass
1194 if len(id) == 40:
1194 if len(id) == 40:
1195 try:
1195 try:
1196 # a full hex nodeid?
1196 # a full hex nodeid?
1197 node = bin(id)
1197 node = bin(id)
1198 self.rev(node)
1198 self.rev(node)
1199 return node
1199 return node
1200 except (TypeError, error.LookupError):
1200 except (TypeError, error.LookupError):
1201 pass
1201 pass
1202
1202
1203 def _partialmatch(self, id):
1203 def _partialmatch(self, id):
1204 # we don't care wdirfilenodeids as they should be always full hash
1204 # we don't care wdirfilenodeids as they should be always full hash
1205 maybewdir = wdirhex.startswith(id)
1205 maybewdir = wdirhex.startswith(id)
1206 try:
1206 try:
1207 partial = self.index.partialmatch(id)
1207 partial = self.index.partialmatch(id)
1208 if partial and self.hasnode(partial):
1208 if partial and self.hasnode(partial):
1209 if maybewdir:
1209 if maybewdir:
1210 # single 'ff...' match in radix tree, ambiguous with wdir
1210 # single 'ff...' match in radix tree, ambiguous with wdir
1211 raise error.RevlogError
1211 raise error.RevlogError
1212 return partial
1212 return partial
1213 if maybewdir:
1213 if maybewdir:
1214 # no 'ff...' match in radix tree, wdir identified
1214 # no 'ff...' match in radix tree, wdir identified
1215 raise error.WdirUnsupported
1215 raise error.WdirUnsupported
1216 return None
1216 return None
1217 except error.RevlogError:
1217 except error.RevlogError:
1218 # parsers.c radix tree lookup gave multiple matches
1218 # parsers.c radix tree lookup gave multiple matches
1219 # fast path: for unfiltered changelog, radix tree is accurate
1219 # fast path: for unfiltered changelog, radix tree is accurate
1220 if not getattr(self, 'filteredrevs', None):
1220 if not getattr(self, 'filteredrevs', None):
1221 raise error.AmbiguousPrefixLookupError(
1221 raise error.AmbiguousPrefixLookupError(
1222 id, self.indexfile, _('ambiguous identifier'))
1222 id, self.indexfile, _('ambiguous identifier'))
1223 # fall through to slow path that filters hidden revisions
1223 # fall through to slow path that filters hidden revisions
1224 except (AttributeError, ValueError):
1224 except (AttributeError, ValueError):
1225 # we are pure python, or key was too short to search radix tree
1225 # we are pure python, or key was too short to search radix tree
1226 pass
1226 pass
1227
1227
1228 if id in self._pcache:
1228 if id in self._pcache:
1229 return self._pcache[id]
1229 return self._pcache[id]
1230
1230
1231 if len(id) <= 40:
1231 if len(id) <= 40:
1232 try:
1232 try:
1233 # hex(node)[:...]
1233 # hex(node)[:...]
1234 l = len(id) // 2 # grab an even number of digits
1234 l = len(id) // 2 # grab an even number of digits
1235 prefix = bin(id[:l * 2])
1235 prefix = bin(id[:l * 2])
1236 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
1236 nl = [e[7] for e in self.index if e[7].startswith(prefix)]
1237 nl = [n for n in nl if hex(n).startswith(id) and
1237 nl = [n for n in nl if hex(n).startswith(id) and
1238 self.hasnode(n)]
1238 self.hasnode(n)]
1239 if nullhex.startswith(id):
1239 if nullhex.startswith(id):
1240 nl.append(nullid)
1240 nl.append(nullid)
1241 if len(nl) > 0:
1241 if len(nl) > 0:
1242 if len(nl) == 1 and not maybewdir:
1242 if len(nl) == 1 and not maybewdir:
1243 self._pcache[id] = nl[0]
1243 self._pcache[id] = nl[0]
1244 return nl[0]
1244 return nl[0]
1245 raise error.AmbiguousPrefixLookupError(
1245 raise error.AmbiguousPrefixLookupError(
1246 id, self.indexfile, _('ambiguous identifier'))
1246 id, self.indexfile, _('ambiguous identifier'))
1247 if maybewdir:
1247 if maybewdir:
1248 raise error.WdirUnsupported
1248 raise error.WdirUnsupported
1249 return None
1249 return None
1250 except TypeError:
1250 except TypeError:
1251 pass
1251 pass
1252
1252
1253 def lookup(self, id):
1253 def lookup(self, id):
1254 """locate a node based on:
1254 """locate a node based on:
1255 - revision number or str(revision number)
1255 - revision number or str(revision number)
1256 - nodeid or subset of hex nodeid
1256 - nodeid or subset of hex nodeid
1257 """
1257 """
1258 n = self._match(id)
1258 n = self._match(id)
1259 if n is not None:
1259 if n is not None:
1260 return n
1260 return n
1261 n = self._partialmatch(id)
1261 n = self._partialmatch(id)
1262 if n:
1262 if n:
1263 return n
1263 return n
1264
1264
1265 raise error.LookupError(id, self.indexfile, _('no match found'))
1265 raise error.LookupError(id, self.indexfile, _('no match found'))
1266
1266
1267 def shortest(self, node, minlength=1):
1267 def shortest(self, node, minlength=1):
1268 """Find the shortest unambiguous prefix that matches node."""
1268 """Find the shortest unambiguous prefix that matches node."""
1269 def isvalid(prefix):
1269 def isvalid(prefix):
1270 try:
1270 try:
1271 node = self._partialmatch(prefix)
1271 node = self._partialmatch(prefix)
1272 except error.AmbiguousPrefixLookupError:
1272 except error.AmbiguousPrefixLookupError:
1273 return False
1273 return False
1274 except error.WdirUnsupported:
1274 except error.WdirUnsupported:
1275 # single 'ff...' match
1275 # single 'ff...' match
1276 return True
1276 return True
1277 if node is None:
1277 if node is None:
1278 raise error.LookupError(node, self.indexfile, _('no node'))
1278 raise error.LookupError(node, self.indexfile, _('no node'))
1279 return True
1279 return True
1280
1280
1281 def maybewdir(prefix):
1281 def maybewdir(prefix):
1282 return all(c == 'f' for c in prefix)
1282 return all(c == 'f' for c in prefix)
1283
1283
1284 hexnode = hex(node)
1284 hexnode = hex(node)
1285
1285
1286 def disambiguate(hexnode, minlength):
1286 def disambiguate(hexnode, minlength):
1287 """Disambiguate against wdirid."""
1287 """Disambiguate against wdirid."""
1288 for length in range(minlength, 41):
1288 for length in range(minlength, 41):
1289 prefix = hexnode[:length]
1289 prefix = hexnode[:length]
1290 if not maybewdir(prefix):
1290 if not maybewdir(prefix):
1291 return prefix
1291 return prefix
1292
1292
1293 if not getattr(self, 'filteredrevs', None):
1293 if not getattr(self, 'filteredrevs', None):
1294 try:
1294 try:
1295 length = max(self.index.shortest(node), minlength)
1295 length = max(self.index.shortest(node), minlength)
1296 return disambiguate(hexnode, length)
1296 return disambiguate(hexnode, length)
1297 except error.RevlogError:
1297 except error.RevlogError:
1298 if node != wdirid:
1298 if node != wdirid:
1299 raise error.LookupError(node, self.indexfile, _('no node'))
1299 raise error.LookupError(node, self.indexfile, _('no node'))
1300 except AttributeError:
1300 except AttributeError:
1301 # Fall through to pure code
1301 # Fall through to pure code
1302 pass
1302 pass
1303
1303
1304 if node == wdirid:
1304 if node == wdirid:
1305 for length in range(minlength, 41):
1305 for length in range(minlength, 41):
1306 prefix = hexnode[:length]
1306 prefix = hexnode[:length]
1307 if isvalid(prefix):
1307 if isvalid(prefix):
1308 return prefix
1308 return prefix
1309
1309
1310 for length in range(minlength, 41):
1310 for length in range(minlength, 41):
1311 prefix = hexnode[:length]
1311 prefix = hexnode[:length]
1312 if isvalid(prefix):
1312 if isvalid(prefix):
1313 return disambiguate(hexnode, length)
1313 return disambiguate(hexnode, length)
1314
1314
1315 def cmp(self, node, text):
1315 def cmp(self, node, text):
1316 """compare text with a given file revision
1316 """compare text with a given file revision
1317
1317
1318 returns True if text is different than what is stored.
1318 returns True if text is different than what is stored.
1319 """
1319 """
1320 p1, p2 = self.parents(node)
1320 p1, p2 = self.parents(node)
1321 return storageutil.hashrevisionsha1(text, p1, p2) != node
1321 return storageutil.hashrevisionsha1(text, p1, p2) != node
1322
1322
1323 def _cachesegment(self, offset, data):
1323 def _cachesegment(self, offset, data):
1324 """Add a segment to the revlog cache.
1324 """Add a segment to the revlog cache.
1325
1325
1326 Accepts an absolute offset and the data that is at that location.
1326 Accepts an absolute offset and the data that is at that location.
1327 """
1327 """
1328 o, d = self._chunkcache
1328 o, d = self._chunkcache
1329 # try to add to existing cache
1329 # try to add to existing cache
1330 if o + len(d) == offset and len(d) + len(data) < _chunksize:
1330 if o + len(d) == offset and len(d) + len(data) < _chunksize:
1331 self._chunkcache = o, d + data
1331 self._chunkcache = o, d + data
1332 else:
1332 else:
1333 self._chunkcache = offset, data
1333 self._chunkcache = offset, data
1334
1334
1335 def _readsegment(self, offset, length, df=None):
1335 def _readsegment(self, offset, length, df=None):
1336 """Load a segment of raw data from the revlog.
1336 """Load a segment of raw data from the revlog.
1337
1337
1338 Accepts an absolute offset, length to read, and an optional existing
1338 Accepts an absolute offset, length to read, and an optional existing
1339 file handle to read from.
1339 file handle to read from.
1340
1340
1341 If an existing file handle is passed, it will be seeked and the
1341 If an existing file handle is passed, it will be seeked and the
1342 original seek position will NOT be restored.
1342 original seek position will NOT be restored.
1343
1343
1344 Returns a str or buffer of raw byte data.
1344 Returns a str or buffer of raw byte data.
1345
1346 Raises if the requested number of bytes could not be read.
1345 """
1347 """
1346 # Cache data both forward and backward around the requested
1348 # Cache data both forward and backward around the requested
1347 # data, in a fixed size window. This helps speed up operations
1349 # data, in a fixed size window. This helps speed up operations
1348 # involving reading the revlog backwards.
1350 # involving reading the revlog backwards.
1349 cachesize = self._chunkcachesize
1351 cachesize = self._chunkcachesize
1350 realoffset = offset & ~(cachesize - 1)
1352 realoffset = offset & ~(cachesize - 1)
1351 reallength = (((offset + length + cachesize) & ~(cachesize - 1))
1353 reallength = (((offset + length + cachesize) & ~(cachesize - 1))
1352 - realoffset)
1354 - realoffset)
1353 with self._datareadfp(df) as df:
1355 with self._datareadfp(df) as df:
1354 df.seek(realoffset)
1356 df.seek(realoffset)
1355 d = df.read(reallength)
1357 d = df.read(reallength)
1358
1356 self._cachesegment(realoffset, d)
1359 self._cachesegment(realoffset, d)
1357 if offset != realoffset or reallength != length:
1360 if offset != realoffset or reallength != length:
1358 return util.buffer(d, offset - realoffset, length)
1361 startoffset = offset - realoffset
1362 if len(d) - startoffset < length:
1363 raise error.RevlogError(
1364 _('partial read of revlog %s; expected %d bytes from '
1365 'offset %d, got %d') %
1366 (self.indexfile if self._inline else self.datafile,
1367 length, realoffset, len(d) - startoffset))
1368
1369 return util.buffer(d, startoffset, length)
1370
1371 if len(d) < length:
1372 raise error.RevlogError(
1373 _('partial read of revlog %s; expected %d bytes from offset '
1374 '%d, got %d') %
1375 (self.indexfile if self._inline else self.datafile,
1376 length, offset, len(d)))
1377
1359 return d
1378 return d
1360
1379
1361 def _getsegment(self, offset, length, df=None):
1380 def _getsegment(self, offset, length, df=None):
1362 """Obtain a segment of raw data from the revlog.
1381 """Obtain a segment of raw data from the revlog.
1363
1382
1364 Accepts an absolute offset, length of bytes to obtain, and an
1383 Accepts an absolute offset, length of bytes to obtain, and an
1365 optional file handle to the already-opened revlog. If the file
1384 optional file handle to the already-opened revlog. If the file
1366 handle is used, it's original seek position will not be preserved.
1385 handle is used, it's original seek position will not be preserved.
1367
1386
1368 Requests for data may be returned from a cache.
1387 Requests for data may be returned from a cache.
1369
1388
1370 Returns a str or a buffer instance of raw byte data.
1389 Returns a str or a buffer instance of raw byte data.
1371 """
1390 """
1372 o, d = self._chunkcache
1391 o, d = self._chunkcache
1373 l = len(d)
1392 l = len(d)
1374
1393
1375 # is it in the cache?
1394 # is it in the cache?
1376 cachestart = offset - o
1395 cachestart = offset - o
1377 cacheend = cachestart + length
1396 cacheend = cachestart + length
1378 if cachestart >= 0 and cacheend <= l:
1397 if cachestart >= 0 and cacheend <= l:
1379 if cachestart == 0 and cacheend == l:
1398 if cachestart == 0 and cacheend == l:
1380 return d # avoid a copy
1399 return d # avoid a copy
1381 return util.buffer(d, cachestart, cacheend - cachestart)
1400 return util.buffer(d, cachestart, cacheend - cachestart)
1382
1401
1383 return self._readsegment(offset, length, df=df)
1402 return self._readsegment(offset, length, df=df)
1384
1403
1385 def _getsegmentforrevs(self, startrev, endrev, df=None):
1404 def _getsegmentforrevs(self, startrev, endrev, df=None):
1386 """Obtain a segment of raw data corresponding to a range of revisions.
1405 """Obtain a segment of raw data corresponding to a range of revisions.
1387
1406
1388 Accepts the start and end revisions and an optional already-open
1407 Accepts the start and end revisions and an optional already-open
1389 file handle to be used for reading. If the file handle is read, its
1408 file handle to be used for reading. If the file handle is read, its
1390 seek position will not be preserved.
1409 seek position will not be preserved.
1391
1410
1392 Requests for data may be satisfied by a cache.
1411 Requests for data may be satisfied by a cache.
1393
1412
1394 Returns a 2-tuple of (offset, data) for the requested range of
1413 Returns a 2-tuple of (offset, data) for the requested range of
1395 revisions. Offset is the integer offset from the beginning of the
1414 revisions. Offset is the integer offset from the beginning of the
1396 revlog and data is a str or buffer of the raw byte data.
1415 revlog and data is a str or buffer of the raw byte data.
1397
1416
1398 Callers will need to call ``self.start(rev)`` and ``self.length(rev)``
1417 Callers will need to call ``self.start(rev)`` and ``self.length(rev)``
1399 to determine where each revision's data begins and ends.
1418 to determine where each revision's data begins and ends.
1400 """
1419 """
1401 # Inlined self.start(startrev) & self.end(endrev) for perf reasons
1420 # Inlined self.start(startrev) & self.end(endrev) for perf reasons
1402 # (functions are expensive).
1421 # (functions are expensive).
1403 index = self.index
1422 index = self.index
1404 istart = index[startrev]
1423 istart = index[startrev]
1405 start = int(istart[0] >> 16)
1424 start = int(istart[0] >> 16)
1406 if startrev == endrev:
1425 if startrev == endrev:
1407 end = start + istart[1]
1426 end = start + istart[1]
1408 else:
1427 else:
1409 iend = index[endrev]
1428 iend = index[endrev]
1410 end = int(iend[0] >> 16) + iend[1]
1429 end = int(iend[0] >> 16) + iend[1]
1411
1430
1412 if self._inline:
1431 if self._inline:
1413 start += (startrev + 1) * self._io.size
1432 start += (startrev + 1) * self._io.size
1414 end += (endrev + 1) * self._io.size
1433 end += (endrev + 1) * self._io.size
1415 length = end - start
1434 length = end - start
1416
1435
1417 return start, self._getsegment(start, length, df=df)
1436 return start, self._getsegment(start, length, df=df)
1418
1437
1419 def _chunk(self, rev, df=None):
1438 def _chunk(self, rev, df=None):
1420 """Obtain a single decompressed chunk for a revision.
1439 """Obtain a single decompressed chunk for a revision.
1421
1440
1422 Accepts an integer revision and an optional already-open file handle
1441 Accepts an integer revision and an optional already-open file handle
1423 to be used for reading. If used, the seek position of the file will not
1442 to be used for reading. If used, the seek position of the file will not
1424 be preserved.
1443 be preserved.
1425
1444
1426 Returns a str holding uncompressed data for the requested revision.
1445 Returns a str holding uncompressed data for the requested revision.
1427 """
1446 """
1428 return self.decompress(self._getsegmentforrevs(rev, rev, df=df)[1])
1447 return self.decompress(self._getsegmentforrevs(rev, rev, df=df)[1])
1429
1448
1430 def _chunks(self, revs, df=None, targetsize=None):
1449 def _chunks(self, revs, df=None, targetsize=None):
1431 """Obtain decompressed chunks for the specified revisions.
1450 """Obtain decompressed chunks for the specified revisions.
1432
1451
1433 Accepts an iterable of numeric revisions that are assumed to be in
1452 Accepts an iterable of numeric revisions that are assumed to be in
1434 ascending order. Also accepts an optional already-open file handle
1453 ascending order. Also accepts an optional already-open file handle
1435 to be used for reading. If used, the seek position of the file will
1454 to be used for reading. If used, the seek position of the file will
1436 not be preserved.
1455 not be preserved.
1437
1456
1438 This function is similar to calling ``self._chunk()`` multiple times,
1457 This function is similar to calling ``self._chunk()`` multiple times,
1439 but is faster.
1458 but is faster.
1440
1459
1441 Returns a list with decompressed data for each requested revision.
1460 Returns a list with decompressed data for each requested revision.
1442 """
1461 """
1443 if not revs:
1462 if not revs:
1444 return []
1463 return []
1445 start = self.start
1464 start = self.start
1446 length = self.length
1465 length = self.length
1447 inline = self._inline
1466 inline = self._inline
1448 iosize = self._io.size
1467 iosize = self._io.size
1449 buffer = util.buffer
1468 buffer = util.buffer
1450
1469
1451 l = []
1470 l = []
1452 ladd = l.append
1471 ladd = l.append
1453
1472
1454 if not self._withsparseread:
1473 if not self._withsparseread:
1455 slicedchunks = (revs,)
1474 slicedchunks = (revs,)
1456 else:
1475 else:
1457 slicedchunks = deltautil.slicechunk(self, revs,
1476 slicedchunks = deltautil.slicechunk(self, revs,
1458 targetsize=targetsize)
1477 targetsize=targetsize)
1459
1478
1460 for revschunk in slicedchunks:
1479 for revschunk in slicedchunks:
1461 firstrev = revschunk[0]
1480 firstrev = revschunk[0]
1462 # Skip trailing revisions with empty diff
1481 # Skip trailing revisions with empty diff
1463 for lastrev in revschunk[::-1]:
1482 for lastrev in revschunk[::-1]:
1464 if length(lastrev) != 0:
1483 if length(lastrev) != 0:
1465 break
1484 break
1466
1485
1467 try:
1486 try:
1468 offset, data = self._getsegmentforrevs(firstrev, lastrev, df=df)
1487 offset, data = self._getsegmentforrevs(firstrev, lastrev, df=df)
1469 except OverflowError:
1488 except OverflowError:
1470 # issue4215 - we can't cache a run of chunks greater than
1489 # issue4215 - we can't cache a run of chunks greater than
1471 # 2G on Windows
1490 # 2G on Windows
1472 return [self._chunk(rev, df=df) for rev in revschunk]
1491 return [self._chunk(rev, df=df) for rev in revschunk]
1473
1492
1474 decomp = self.decompress
1493 decomp = self.decompress
1475 for rev in revschunk:
1494 for rev in revschunk:
1476 chunkstart = start(rev)
1495 chunkstart = start(rev)
1477 if inline:
1496 if inline:
1478 chunkstart += (rev + 1) * iosize
1497 chunkstart += (rev + 1) * iosize
1479 chunklength = length(rev)
1498 chunklength = length(rev)
1480 ladd(decomp(buffer(data, chunkstart - offset, chunklength)))
1499 ladd(decomp(buffer(data, chunkstart - offset, chunklength)))
1481
1500
1482 return l
1501 return l
1483
1502
1484 def _chunkclear(self):
1503 def _chunkclear(self):
1485 """Clear the raw chunk cache."""
1504 """Clear the raw chunk cache."""
1486 self._chunkcache = (0, '')
1505 self._chunkcache = (0, '')
1487
1506
1488 def deltaparent(self, rev):
1507 def deltaparent(self, rev):
1489 """return deltaparent of the given revision"""
1508 """return deltaparent of the given revision"""
1490 base = self.index[rev][3]
1509 base = self.index[rev][3]
1491 if base == rev:
1510 if base == rev:
1492 return nullrev
1511 return nullrev
1493 elif self._generaldelta:
1512 elif self._generaldelta:
1494 return base
1513 return base
1495 else:
1514 else:
1496 return rev - 1
1515 return rev - 1
1497
1516
1498 def issnapshot(self, rev):
1517 def issnapshot(self, rev):
1499 """tells whether rev is a snapshot
1518 """tells whether rev is a snapshot
1500 """
1519 """
1501 if rev == nullrev:
1520 if rev == nullrev:
1502 return True
1521 return True
1503 deltap = self.deltaparent(rev)
1522 deltap = self.deltaparent(rev)
1504 if deltap == nullrev:
1523 if deltap == nullrev:
1505 return True
1524 return True
1506 p1, p2 = self.parentrevs(rev)
1525 p1, p2 = self.parentrevs(rev)
1507 if deltap in (p1, p2):
1526 if deltap in (p1, p2):
1508 return False
1527 return False
1509 return self.issnapshot(deltap)
1528 return self.issnapshot(deltap)
1510
1529
1511 def snapshotdepth(self, rev):
1530 def snapshotdepth(self, rev):
1512 """number of snapshot in the chain before this one"""
1531 """number of snapshot in the chain before this one"""
1513 if not self.issnapshot(rev):
1532 if not self.issnapshot(rev):
1514 raise error.ProgrammingError('revision %d not a snapshot')
1533 raise error.ProgrammingError('revision %d not a snapshot')
1515 return len(self._deltachain(rev)[0]) - 1
1534 return len(self._deltachain(rev)[0]) - 1
1516
1535
1517 def revdiff(self, rev1, rev2):
1536 def revdiff(self, rev1, rev2):
1518 """return or calculate a delta between two revisions
1537 """return or calculate a delta between two revisions
1519
1538
1520 The delta calculated is in binary form and is intended to be written to
1539 The delta calculated is in binary form and is intended to be written to
1521 revlog data directly. So this function needs raw revision data.
1540 revlog data directly. So this function needs raw revision data.
1522 """
1541 """
1523 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
1542 if rev1 != nullrev and self.deltaparent(rev2) == rev1:
1524 return bytes(self._chunk(rev2))
1543 return bytes(self._chunk(rev2))
1525
1544
1526 return mdiff.textdiff(self.revision(rev1, raw=True),
1545 return mdiff.textdiff(self.revision(rev1, raw=True),
1527 self.revision(rev2, raw=True))
1546 self.revision(rev2, raw=True))
1528
1547
1529 def revision(self, nodeorrev, _df=None, raw=False):
1548 def revision(self, nodeorrev, _df=None, raw=False):
1530 """return an uncompressed revision of a given node or revision
1549 """return an uncompressed revision of a given node or revision
1531 number.
1550 number.
1532
1551
1533 _df - an existing file handle to read from. (internal-only)
1552 _df - an existing file handle to read from. (internal-only)
1534 raw - an optional argument specifying if the revision data is to be
1553 raw - an optional argument specifying if the revision data is to be
1535 treated as raw data when applying flag transforms. 'raw' should be set
1554 treated as raw data when applying flag transforms. 'raw' should be set
1536 to True when generating changegroups or in debug commands.
1555 to True when generating changegroups or in debug commands.
1537 """
1556 """
1538 if isinstance(nodeorrev, int):
1557 if isinstance(nodeorrev, int):
1539 rev = nodeorrev
1558 rev = nodeorrev
1540 node = self.node(rev)
1559 node = self.node(rev)
1541 else:
1560 else:
1542 node = nodeorrev
1561 node = nodeorrev
1543 rev = None
1562 rev = None
1544
1563
1545 cachedrev = None
1564 cachedrev = None
1546 flags = None
1565 flags = None
1547 rawtext = None
1566 rawtext = None
1548 if node == nullid:
1567 if node == nullid:
1549 return ""
1568 return ""
1550 if self._revisioncache:
1569 if self._revisioncache:
1551 if self._revisioncache[0] == node:
1570 if self._revisioncache[0] == node:
1552 # _cache only stores rawtext
1571 # _cache only stores rawtext
1553 if raw:
1572 if raw:
1554 return self._revisioncache[2]
1573 return self._revisioncache[2]
1555 # duplicated, but good for perf
1574 # duplicated, but good for perf
1556 if rev is None:
1575 if rev is None:
1557 rev = self.rev(node)
1576 rev = self.rev(node)
1558 if flags is None:
1577 if flags is None:
1559 flags = self.flags(rev)
1578 flags = self.flags(rev)
1560 # no extra flags set, no flag processor runs, text = rawtext
1579 # no extra flags set, no flag processor runs, text = rawtext
1561 if flags == REVIDX_DEFAULT_FLAGS:
1580 if flags == REVIDX_DEFAULT_FLAGS:
1562 return self._revisioncache[2]
1581 return self._revisioncache[2]
1563 # rawtext is reusable. need to run flag processor
1582 # rawtext is reusable. need to run flag processor
1564 rawtext = self._revisioncache[2]
1583 rawtext = self._revisioncache[2]
1565
1584
1566 cachedrev = self._revisioncache[1]
1585 cachedrev = self._revisioncache[1]
1567
1586
1568 # look up what we need to read
1587 # look up what we need to read
1569 if rawtext is None:
1588 if rawtext is None:
1570 if rev is None:
1589 if rev is None:
1571 rev = self.rev(node)
1590 rev = self.rev(node)
1572
1591
1573 chain, stopped = self._deltachain(rev, stoprev=cachedrev)
1592 chain, stopped = self._deltachain(rev, stoprev=cachedrev)
1574 if stopped:
1593 if stopped:
1575 rawtext = self._revisioncache[2]
1594 rawtext = self._revisioncache[2]
1576
1595
1577 # drop cache to save memory
1596 # drop cache to save memory
1578 self._revisioncache = None
1597 self._revisioncache = None
1579
1598
1580 targetsize = None
1599 targetsize = None
1581 rawsize = self.index[rev][2]
1600 rawsize = self.index[rev][2]
1582 if 0 <= rawsize:
1601 if 0 <= rawsize:
1583 targetsize = 4 * rawsize
1602 targetsize = 4 * rawsize
1584
1603
1585 bins = self._chunks(chain, df=_df, targetsize=targetsize)
1604 bins = self._chunks(chain, df=_df, targetsize=targetsize)
1586 if rawtext is None:
1605 if rawtext is None:
1587 rawtext = bytes(bins[0])
1606 rawtext = bytes(bins[0])
1588 bins = bins[1:]
1607 bins = bins[1:]
1589
1608
1590 rawtext = mdiff.patches(rawtext, bins)
1609 rawtext = mdiff.patches(rawtext, bins)
1591 self._revisioncache = (node, rev, rawtext)
1610 self._revisioncache = (node, rev, rawtext)
1592
1611
1593 if flags is None:
1612 if flags is None:
1594 if rev is None:
1613 if rev is None:
1595 rev = self.rev(node)
1614 rev = self.rev(node)
1596 flags = self.flags(rev)
1615 flags = self.flags(rev)
1597
1616
1598 text, validatehash = self._processflags(rawtext, flags, 'read', raw=raw)
1617 text, validatehash = self._processflags(rawtext, flags, 'read', raw=raw)
1599 if validatehash:
1618 if validatehash:
1600 self.checkhash(text, node, rev=rev)
1619 self.checkhash(text, node, rev=rev)
1601
1620
1602 return text
1621 return text
1603
1622
1604 def hash(self, text, p1, p2):
1623 def hash(self, text, p1, p2):
1605 """Compute a node hash.
1624 """Compute a node hash.
1606
1625
1607 Available as a function so that subclasses can replace the hash
1626 Available as a function so that subclasses can replace the hash
1608 as needed.
1627 as needed.
1609 """
1628 """
1610 return storageutil.hashrevisionsha1(text, p1, p2)
1629 return storageutil.hashrevisionsha1(text, p1, p2)
1611
1630
1612 def _processflags(self, text, flags, operation, raw=False):
1631 def _processflags(self, text, flags, operation, raw=False):
1613 """Inspect revision data flags and applies transforms defined by
1632 """Inspect revision data flags and applies transforms defined by
1614 registered flag processors.
1633 registered flag processors.
1615
1634
1616 ``text`` - the revision data to process
1635 ``text`` - the revision data to process
1617 ``flags`` - the revision flags
1636 ``flags`` - the revision flags
1618 ``operation`` - the operation being performed (read or write)
1637 ``operation`` - the operation being performed (read or write)
1619 ``raw`` - an optional argument describing if the raw transform should be
1638 ``raw`` - an optional argument describing if the raw transform should be
1620 applied.
1639 applied.
1621
1640
1622 This method processes the flags in the order (or reverse order if
1641 This method processes the flags in the order (or reverse order if
1623 ``operation`` is 'write') defined by REVIDX_FLAGS_ORDER, applying the
1642 ``operation`` is 'write') defined by REVIDX_FLAGS_ORDER, applying the
1624 flag processors registered for present flags. The order of flags defined
1643 flag processors registered for present flags. The order of flags defined
1625 in REVIDX_FLAGS_ORDER needs to be stable to allow non-commutativity.
1644 in REVIDX_FLAGS_ORDER needs to be stable to allow non-commutativity.
1626
1645
1627 Returns a 2-tuple of ``(text, validatehash)`` where ``text`` is the
1646 Returns a 2-tuple of ``(text, validatehash)`` where ``text`` is the
1628 processed text and ``validatehash`` is a bool indicating whether the
1647 processed text and ``validatehash`` is a bool indicating whether the
1629 returned text should be checked for hash integrity.
1648 returned text should be checked for hash integrity.
1630
1649
1631 Note: If the ``raw`` argument is set, it has precedence over the
1650 Note: If the ``raw`` argument is set, it has precedence over the
1632 operation and will only update the value of ``validatehash``.
1651 operation and will only update the value of ``validatehash``.
1633 """
1652 """
1634 # fast path: no flag processors will run
1653 # fast path: no flag processors will run
1635 if flags == 0:
1654 if flags == 0:
1636 return text, True
1655 return text, True
1637 if not operation in ('read', 'write'):
1656 if not operation in ('read', 'write'):
1638 raise error.ProgrammingError(_("invalid '%s' operation") %
1657 raise error.ProgrammingError(_("invalid '%s' operation") %
1639 operation)
1658 operation)
1640 # Check all flags are known.
1659 # Check all flags are known.
1641 if flags & ~REVIDX_KNOWN_FLAGS:
1660 if flags & ~REVIDX_KNOWN_FLAGS:
1642 raise error.RevlogError(_("incompatible revision flag '%#x'") %
1661 raise error.RevlogError(_("incompatible revision flag '%#x'") %
1643 (flags & ~REVIDX_KNOWN_FLAGS))
1662 (flags & ~REVIDX_KNOWN_FLAGS))
1644 validatehash = True
1663 validatehash = True
1645 # Depending on the operation (read or write), the order might be
1664 # Depending on the operation (read or write), the order might be
1646 # reversed due to non-commutative transforms.
1665 # reversed due to non-commutative transforms.
1647 orderedflags = REVIDX_FLAGS_ORDER
1666 orderedflags = REVIDX_FLAGS_ORDER
1648 if operation == 'write':
1667 if operation == 'write':
1649 orderedflags = reversed(orderedflags)
1668 orderedflags = reversed(orderedflags)
1650
1669
1651 for flag in orderedflags:
1670 for flag in orderedflags:
1652 # If a flagprocessor has been registered for a known flag, apply the
1671 # If a flagprocessor has been registered for a known flag, apply the
1653 # related operation transform and update result tuple.
1672 # related operation transform and update result tuple.
1654 if flag & flags:
1673 if flag & flags:
1655 vhash = True
1674 vhash = True
1656
1675
1657 if flag not in self._flagprocessors:
1676 if flag not in self._flagprocessors:
1658 message = _("missing processor for flag '%#x'") % (flag)
1677 message = _("missing processor for flag '%#x'") % (flag)
1659 raise error.RevlogError(message)
1678 raise error.RevlogError(message)
1660
1679
1661 processor = self._flagprocessors[flag]
1680 processor = self._flagprocessors[flag]
1662 if processor is not None:
1681 if processor is not None:
1663 readtransform, writetransform, rawtransform = processor
1682 readtransform, writetransform, rawtransform = processor
1664
1683
1665 if raw:
1684 if raw:
1666 vhash = rawtransform(self, text)
1685 vhash = rawtransform(self, text)
1667 elif operation == 'read':
1686 elif operation == 'read':
1668 text, vhash = readtransform(self, text)
1687 text, vhash = readtransform(self, text)
1669 else: # write operation
1688 else: # write operation
1670 text, vhash = writetransform(self, text)
1689 text, vhash = writetransform(self, text)
1671 validatehash = validatehash and vhash
1690 validatehash = validatehash and vhash
1672
1691
1673 return text, validatehash
1692 return text, validatehash
1674
1693
1675 def checkhash(self, text, node, p1=None, p2=None, rev=None):
1694 def checkhash(self, text, node, p1=None, p2=None, rev=None):
1676 """Check node hash integrity.
1695 """Check node hash integrity.
1677
1696
1678 Available as a function so that subclasses can extend hash mismatch
1697 Available as a function so that subclasses can extend hash mismatch
1679 behaviors as needed.
1698 behaviors as needed.
1680 """
1699 """
1681 try:
1700 try:
1682 if p1 is None and p2 is None:
1701 if p1 is None and p2 is None:
1683 p1, p2 = self.parents(node)
1702 p1, p2 = self.parents(node)
1684 if node != self.hash(text, p1, p2):
1703 if node != self.hash(text, p1, p2):
1685 # Clear the revision cache on hash failure. The revision cache
1704 # Clear the revision cache on hash failure. The revision cache
1686 # only stores the raw revision and clearing the cache does have
1705 # only stores the raw revision and clearing the cache does have
1687 # the side-effect that we won't have a cache hit when the raw
1706 # the side-effect that we won't have a cache hit when the raw
1688 # revision data is accessed. But this case should be rare and
1707 # revision data is accessed. But this case should be rare and
1689 # it is extra work to teach the cache about the hash
1708 # it is extra work to teach the cache about the hash
1690 # verification state.
1709 # verification state.
1691 if self._revisioncache and self._revisioncache[0] == node:
1710 if self._revisioncache and self._revisioncache[0] == node:
1692 self._revisioncache = None
1711 self._revisioncache = None
1693
1712
1694 revornode = rev
1713 revornode = rev
1695 if revornode is None:
1714 if revornode is None:
1696 revornode = templatefilters.short(hex(node))
1715 revornode = templatefilters.short(hex(node))
1697 raise error.RevlogError(_("integrity check failed on %s:%s")
1716 raise error.RevlogError(_("integrity check failed on %s:%s")
1698 % (self.indexfile, pycompat.bytestr(revornode)))
1717 % (self.indexfile, pycompat.bytestr(revornode)))
1699 except error.RevlogError:
1718 except error.RevlogError:
1700 if self._censorable and storageutil.iscensoredtext(text):
1719 if self._censorable and storageutil.iscensoredtext(text):
1701 raise error.CensoredNodeError(self.indexfile, node, text)
1720 raise error.CensoredNodeError(self.indexfile, node, text)
1702 raise
1721 raise
1703
1722
1704 def _enforceinlinesize(self, tr, fp=None):
1723 def _enforceinlinesize(self, tr, fp=None):
1705 """Check if the revlog is too big for inline and convert if so.
1724 """Check if the revlog is too big for inline and convert if so.
1706
1725
1707 This should be called after revisions are added to the revlog. If the
1726 This should be called after revisions are added to the revlog. If the
1708 revlog has grown too large to be an inline revlog, it will convert it
1727 revlog has grown too large to be an inline revlog, it will convert it
1709 to use multiple index and data files.
1728 to use multiple index and data files.
1710 """
1729 """
1711 tiprev = len(self) - 1
1730 tiprev = len(self) - 1
1712 if (not self._inline or
1731 if (not self._inline or
1713 (self.start(tiprev) + self.length(tiprev)) < _maxinline):
1732 (self.start(tiprev) + self.length(tiprev)) < _maxinline):
1714 return
1733 return
1715
1734
1716 trinfo = tr.find(self.indexfile)
1735 trinfo = tr.find(self.indexfile)
1717 if trinfo is None:
1736 if trinfo is None:
1718 raise error.RevlogError(_("%s not found in the transaction")
1737 raise error.RevlogError(_("%s not found in the transaction")
1719 % self.indexfile)
1738 % self.indexfile)
1720
1739
1721 trindex = trinfo[2]
1740 trindex = trinfo[2]
1722 if trindex is not None:
1741 if trindex is not None:
1723 dataoff = self.start(trindex)
1742 dataoff = self.start(trindex)
1724 else:
1743 else:
1725 # revlog was stripped at start of transaction, use all leftover data
1744 # revlog was stripped at start of transaction, use all leftover data
1726 trindex = len(self) - 1
1745 trindex = len(self) - 1
1727 dataoff = self.end(tiprev)
1746 dataoff = self.end(tiprev)
1728
1747
1729 tr.add(self.datafile, dataoff)
1748 tr.add(self.datafile, dataoff)
1730
1749
1731 if fp:
1750 if fp:
1732 fp.flush()
1751 fp.flush()
1733 fp.close()
1752 fp.close()
1734
1753
1735 with self._indexfp('r') as ifh, self._datafp('w') as dfh:
1754 with self._indexfp('r') as ifh, self._datafp('w') as dfh:
1736 for r in self:
1755 for r in self:
1737 dfh.write(self._getsegmentforrevs(r, r, df=ifh)[1])
1756 dfh.write(self._getsegmentforrevs(r, r, df=ifh)[1])
1738
1757
1739 with self._indexfp('w') as fp:
1758 with self._indexfp('w') as fp:
1740 self.version &= ~FLAG_INLINE_DATA
1759 self.version &= ~FLAG_INLINE_DATA
1741 self._inline = False
1760 self._inline = False
1742 io = self._io
1761 io = self._io
1743 for i in self:
1762 for i in self:
1744 e = io.packentry(self.index[i], self.node, self.version, i)
1763 e = io.packentry(self.index[i], self.node, self.version, i)
1745 fp.write(e)
1764 fp.write(e)
1746
1765
1747 # the temp file replace the real index when we exit the context
1766 # the temp file replace the real index when we exit the context
1748 # manager
1767 # manager
1749
1768
1750 tr.replace(self.indexfile, trindex * self._io.size)
1769 tr.replace(self.indexfile, trindex * self._io.size)
1751 self._chunkclear()
1770 self._chunkclear()
1752
1771
1753 def _nodeduplicatecallback(self, transaction, node):
1772 def _nodeduplicatecallback(self, transaction, node):
1754 """called when trying to add a node already stored.
1773 """called when trying to add a node already stored.
1755 """
1774 """
1756
1775
1757 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1776 def addrevision(self, text, transaction, link, p1, p2, cachedelta=None,
1758 node=None, flags=REVIDX_DEFAULT_FLAGS, deltacomputer=None):
1777 node=None, flags=REVIDX_DEFAULT_FLAGS, deltacomputer=None):
1759 """add a revision to the log
1778 """add a revision to the log
1760
1779
1761 text - the revision data to add
1780 text - the revision data to add
1762 transaction - the transaction object used for rollback
1781 transaction - the transaction object used for rollback
1763 link - the linkrev data to add
1782 link - the linkrev data to add
1764 p1, p2 - the parent nodeids of the revision
1783 p1, p2 - the parent nodeids of the revision
1765 cachedelta - an optional precomputed delta
1784 cachedelta - an optional precomputed delta
1766 node - nodeid of revision; typically node is not specified, and it is
1785 node - nodeid of revision; typically node is not specified, and it is
1767 computed by default as hash(text, p1, p2), however subclasses might
1786 computed by default as hash(text, p1, p2), however subclasses might
1768 use different hashing method (and override checkhash() in such case)
1787 use different hashing method (and override checkhash() in such case)
1769 flags - the known flags to set on the revision
1788 flags - the known flags to set on the revision
1770 deltacomputer - an optional deltacomputer instance shared between
1789 deltacomputer - an optional deltacomputer instance shared between
1771 multiple calls
1790 multiple calls
1772 """
1791 """
1773 if link == nullrev:
1792 if link == nullrev:
1774 raise error.RevlogError(_("attempted to add linkrev -1 to %s")
1793 raise error.RevlogError(_("attempted to add linkrev -1 to %s")
1775 % self.indexfile)
1794 % self.indexfile)
1776
1795
1777 if flags:
1796 if flags:
1778 node = node or self.hash(text, p1, p2)
1797 node = node or self.hash(text, p1, p2)
1779
1798
1780 rawtext, validatehash = self._processflags(text, flags, 'write')
1799 rawtext, validatehash = self._processflags(text, flags, 'write')
1781
1800
1782 # If the flag processor modifies the revision data, ignore any provided
1801 # If the flag processor modifies the revision data, ignore any provided
1783 # cachedelta.
1802 # cachedelta.
1784 if rawtext != text:
1803 if rawtext != text:
1785 cachedelta = None
1804 cachedelta = None
1786
1805
1787 if len(rawtext) > _maxentrysize:
1806 if len(rawtext) > _maxentrysize:
1788 raise error.RevlogError(
1807 raise error.RevlogError(
1789 _("%s: size of %d bytes exceeds maximum revlog storage of 2GiB")
1808 _("%s: size of %d bytes exceeds maximum revlog storage of 2GiB")
1790 % (self.indexfile, len(rawtext)))
1809 % (self.indexfile, len(rawtext)))
1791
1810
1792 node = node or self.hash(rawtext, p1, p2)
1811 node = node or self.hash(rawtext, p1, p2)
1793 if node in self.nodemap:
1812 if node in self.nodemap:
1794 return node
1813 return node
1795
1814
1796 if validatehash:
1815 if validatehash:
1797 self.checkhash(rawtext, node, p1=p1, p2=p2)
1816 self.checkhash(rawtext, node, p1=p1, p2=p2)
1798
1817
1799 return self.addrawrevision(rawtext, transaction, link, p1, p2, node,
1818 return self.addrawrevision(rawtext, transaction, link, p1, p2, node,
1800 flags, cachedelta=cachedelta,
1819 flags, cachedelta=cachedelta,
1801 deltacomputer=deltacomputer)
1820 deltacomputer=deltacomputer)
1802
1821
1803 def addrawrevision(self, rawtext, transaction, link, p1, p2, node, flags,
1822 def addrawrevision(self, rawtext, transaction, link, p1, p2, node, flags,
1804 cachedelta=None, deltacomputer=None):
1823 cachedelta=None, deltacomputer=None):
1805 """add a raw revision with known flags, node and parents
1824 """add a raw revision with known flags, node and parents
1806 useful when reusing a revision not stored in this revlog (ex: received
1825 useful when reusing a revision not stored in this revlog (ex: received
1807 over wire, or read from an external bundle).
1826 over wire, or read from an external bundle).
1808 """
1827 """
1809 dfh = None
1828 dfh = None
1810 if not self._inline:
1829 if not self._inline:
1811 dfh = self._datafp("a+")
1830 dfh = self._datafp("a+")
1812 ifh = self._indexfp("a+")
1831 ifh = self._indexfp("a+")
1813 try:
1832 try:
1814 return self._addrevision(node, rawtext, transaction, link, p1, p2,
1833 return self._addrevision(node, rawtext, transaction, link, p1, p2,
1815 flags, cachedelta, ifh, dfh,
1834 flags, cachedelta, ifh, dfh,
1816 deltacomputer=deltacomputer)
1835 deltacomputer=deltacomputer)
1817 finally:
1836 finally:
1818 if dfh:
1837 if dfh:
1819 dfh.close()
1838 dfh.close()
1820 ifh.close()
1839 ifh.close()
1821
1840
1822 def compress(self, data):
1841 def compress(self, data):
1823 """Generate a possibly-compressed representation of data."""
1842 """Generate a possibly-compressed representation of data."""
1824 if not data:
1843 if not data:
1825 return '', data
1844 return '', data
1826
1845
1827 compressed = self._compressor.compress(data)
1846 compressed = self._compressor.compress(data)
1828
1847
1829 if compressed:
1848 if compressed:
1830 # The revlog compressor added the header in the returned data.
1849 # The revlog compressor added the header in the returned data.
1831 return '', compressed
1850 return '', compressed
1832
1851
1833 if data[0:1] == '\0':
1852 if data[0:1] == '\0':
1834 return '', data
1853 return '', data
1835 return 'u', data
1854 return 'u', data
1836
1855
1837 def decompress(self, data):
1856 def decompress(self, data):
1838 """Decompress a revlog chunk.
1857 """Decompress a revlog chunk.
1839
1858
1840 The chunk is expected to begin with a header identifying the
1859 The chunk is expected to begin with a header identifying the
1841 format type so it can be routed to an appropriate decompressor.
1860 format type so it can be routed to an appropriate decompressor.
1842 """
1861 """
1843 if not data:
1862 if not data:
1844 return data
1863 return data
1845
1864
1846 # Revlogs are read much more frequently than they are written and many
1865 # Revlogs are read much more frequently than they are written and many
1847 # chunks only take microseconds to decompress, so performance is
1866 # chunks only take microseconds to decompress, so performance is
1848 # important here.
1867 # important here.
1849 #
1868 #
1850 # We can make a few assumptions about revlogs:
1869 # We can make a few assumptions about revlogs:
1851 #
1870 #
1852 # 1) the majority of chunks will be compressed (as opposed to inline
1871 # 1) the majority of chunks will be compressed (as opposed to inline
1853 # raw data).
1872 # raw data).
1854 # 2) decompressing *any* data will likely by at least 10x slower than
1873 # 2) decompressing *any* data will likely by at least 10x slower than
1855 # returning raw inline data.
1874 # returning raw inline data.
1856 # 3) we want to prioritize common and officially supported compression
1875 # 3) we want to prioritize common and officially supported compression
1857 # engines
1876 # engines
1858 #
1877 #
1859 # It follows that we want to optimize for "decompress compressed data
1878 # It follows that we want to optimize for "decompress compressed data
1860 # when encoded with common and officially supported compression engines"
1879 # when encoded with common and officially supported compression engines"
1861 # case over "raw data" and "data encoded by less common or non-official
1880 # case over "raw data" and "data encoded by less common or non-official
1862 # compression engines." That is why we have the inline lookup first
1881 # compression engines." That is why we have the inline lookup first
1863 # followed by the compengines lookup.
1882 # followed by the compengines lookup.
1864 #
1883 #
1865 # According to `hg perfrevlogchunks`, this is ~0.5% faster for zlib
1884 # According to `hg perfrevlogchunks`, this is ~0.5% faster for zlib
1866 # compressed chunks. And this matters for changelog and manifest reads.
1885 # compressed chunks. And this matters for changelog and manifest reads.
1867 t = data[0:1]
1886 t = data[0:1]
1868
1887
1869 if t == 'x':
1888 if t == 'x':
1870 try:
1889 try:
1871 return _zlibdecompress(data)
1890 return _zlibdecompress(data)
1872 except zlib.error as e:
1891 except zlib.error as e:
1873 raise error.RevlogError(_('revlog decompress error: %s') %
1892 raise error.RevlogError(_('revlog decompress error: %s') %
1874 stringutil.forcebytestr(e))
1893 stringutil.forcebytestr(e))
1875 # '\0' is more common than 'u' so it goes first.
1894 # '\0' is more common than 'u' so it goes first.
1876 elif t == '\0':
1895 elif t == '\0':
1877 return data
1896 return data
1878 elif t == 'u':
1897 elif t == 'u':
1879 return util.buffer(data, 1)
1898 return util.buffer(data, 1)
1880
1899
1881 try:
1900 try:
1882 compressor = self._decompressors[t]
1901 compressor = self._decompressors[t]
1883 except KeyError:
1902 except KeyError:
1884 try:
1903 try:
1885 engine = util.compengines.forrevlogheader(t)
1904 engine = util.compengines.forrevlogheader(t)
1886 compressor = engine.revlogcompressor()
1905 compressor = engine.revlogcompressor()
1887 self._decompressors[t] = compressor
1906 self._decompressors[t] = compressor
1888 except KeyError:
1907 except KeyError:
1889 raise error.RevlogError(_('unknown compression type %r') % t)
1908 raise error.RevlogError(_('unknown compression type %r') % t)
1890
1909
1891 return compressor.decompress(data)
1910 return compressor.decompress(data)
1892
1911
1893 def _addrevision(self, node, rawtext, transaction, link, p1, p2, flags,
1912 def _addrevision(self, node, rawtext, transaction, link, p1, p2, flags,
1894 cachedelta, ifh, dfh, alwayscache=False,
1913 cachedelta, ifh, dfh, alwayscache=False,
1895 deltacomputer=None):
1914 deltacomputer=None):
1896 """internal function to add revisions to the log
1915 """internal function to add revisions to the log
1897
1916
1898 see addrevision for argument descriptions.
1917 see addrevision for argument descriptions.
1899
1918
1900 note: "addrevision" takes non-raw text, "_addrevision" takes raw text.
1919 note: "addrevision" takes non-raw text, "_addrevision" takes raw text.
1901
1920
1902 if "deltacomputer" is not provided or None, a defaultdeltacomputer will
1921 if "deltacomputer" is not provided or None, a defaultdeltacomputer will
1903 be used.
1922 be used.
1904
1923
1905 invariants:
1924 invariants:
1906 - rawtext is optional (can be None); if not set, cachedelta must be set.
1925 - rawtext is optional (can be None); if not set, cachedelta must be set.
1907 if both are set, they must correspond to each other.
1926 if both are set, they must correspond to each other.
1908 """
1927 """
1909 if node == nullid:
1928 if node == nullid:
1910 raise error.RevlogError(_("%s: attempt to add null revision") %
1929 raise error.RevlogError(_("%s: attempt to add null revision") %
1911 self.indexfile)
1930 self.indexfile)
1912 if node == wdirid or node in wdirfilenodeids:
1931 if node == wdirid or node in wdirfilenodeids:
1913 raise error.RevlogError(_("%s: attempt to add wdir revision") %
1932 raise error.RevlogError(_("%s: attempt to add wdir revision") %
1914 self.indexfile)
1933 self.indexfile)
1915
1934
1916 if self._inline:
1935 if self._inline:
1917 fh = ifh
1936 fh = ifh
1918 else:
1937 else:
1919 fh = dfh
1938 fh = dfh
1920
1939
1921 btext = [rawtext]
1940 btext = [rawtext]
1922
1941
1923 curr = len(self)
1942 curr = len(self)
1924 prev = curr - 1
1943 prev = curr - 1
1925 offset = self.end(prev)
1944 offset = self.end(prev)
1926 p1r, p2r = self.rev(p1), self.rev(p2)
1945 p1r, p2r = self.rev(p1), self.rev(p2)
1927
1946
1928 # full versions are inserted when the needed deltas
1947 # full versions are inserted when the needed deltas
1929 # become comparable to the uncompressed text
1948 # become comparable to the uncompressed text
1930 if rawtext is None:
1949 if rawtext is None:
1931 # need rawtext size, before changed by flag processors, which is
1950 # need rawtext size, before changed by flag processors, which is
1932 # the non-raw size. use revlog explicitly to avoid filelog's extra
1951 # the non-raw size. use revlog explicitly to avoid filelog's extra
1933 # logic that might remove metadata size.
1952 # logic that might remove metadata size.
1934 textlen = mdiff.patchedsize(revlog.size(self, cachedelta[0]),
1953 textlen = mdiff.patchedsize(revlog.size(self, cachedelta[0]),
1935 cachedelta[1])
1954 cachedelta[1])
1936 else:
1955 else:
1937 textlen = len(rawtext)
1956 textlen = len(rawtext)
1938
1957
1939 if deltacomputer is None:
1958 if deltacomputer is None:
1940 deltacomputer = deltautil.deltacomputer(self)
1959 deltacomputer = deltautil.deltacomputer(self)
1941
1960
1942 revinfo = _revisioninfo(node, p1, p2, btext, textlen, cachedelta, flags)
1961 revinfo = _revisioninfo(node, p1, p2, btext, textlen, cachedelta, flags)
1943
1962
1944 deltainfo = deltacomputer.finddeltainfo(revinfo, fh)
1963 deltainfo = deltacomputer.finddeltainfo(revinfo, fh)
1945
1964
1946 e = (offset_type(offset, flags), deltainfo.deltalen, textlen,
1965 e = (offset_type(offset, flags), deltainfo.deltalen, textlen,
1947 deltainfo.base, link, p1r, p2r, node)
1966 deltainfo.base, link, p1r, p2r, node)
1948 self.index.append(e)
1967 self.index.append(e)
1949 self.nodemap[node] = curr
1968 self.nodemap[node] = curr
1950
1969
1951 # Reset the pure node cache start lookup offset to account for new
1970 # Reset the pure node cache start lookup offset to account for new
1952 # revision.
1971 # revision.
1953 if self._nodepos is not None:
1972 if self._nodepos is not None:
1954 self._nodepos = curr
1973 self._nodepos = curr
1955
1974
1956 entry = self._io.packentry(e, self.node, self.version, curr)
1975 entry = self._io.packentry(e, self.node, self.version, curr)
1957 self._writeentry(transaction, ifh, dfh, entry, deltainfo.data,
1976 self._writeentry(transaction, ifh, dfh, entry, deltainfo.data,
1958 link, offset)
1977 link, offset)
1959
1978
1960 rawtext = btext[0]
1979 rawtext = btext[0]
1961
1980
1962 if alwayscache and rawtext is None:
1981 if alwayscache and rawtext is None:
1963 rawtext = deltacomputer.buildtext(revinfo, fh)
1982 rawtext = deltacomputer.buildtext(revinfo, fh)
1964
1983
1965 if type(rawtext) == bytes: # only accept immutable objects
1984 if type(rawtext) == bytes: # only accept immutable objects
1966 self._revisioncache = (node, curr, rawtext)
1985 self._revisioncache = (node, curr, rawtext)
1967 self._chainbasecache[curr] = deltainfo.chainbase
1986 self._chainbasecache[curr] = deltainfo.chainbase
1968 return node
1987 return node
1969
1988
1970 def _writeentry(self, transaction, ifh, dfh, entry, data, link, offset):
1989 def _writeentry(self, transaction, ifh, dfh, entry, data, link, offset):
1971 # Files opened in a+ mode have inconsistent behavior on various
1990 # Files opened in a+ mode have inconsistent behavior on various
1972 # platforms. Windows requires that a file positioning call be made
1991 # platforms. Windows requires that a file positioning call be made
1973 # when the file handle transitions between reads and writes. See
1992 # when the file handle transitions between reads and writes. See
1974 # 3686fa2b8eee and the mixedfilemodewrapper in windows.py. On other
1993 # 3686fa2b8eee and the mixedfilemodewrapper in windows.py. On other
1975 # platforms, Python or the platform itself can be buggy. Some versions
1994 # platforms, Python or the platform itself can be buggy. Some versions
1976 # of Solaris have been observed to not append at the end of the file
1995 # of Solaris have been observed to not append at the end of the file
1977 # if the file was seeked to before the end. See issue4943 for more.
1996 # if the file was seeked to before the end. See issue4943 for more.
1978 #
1997 #
1979 # We work around this issue by inserting a seek() before writing.
1998 # We work around this issue by inserting a seek() before writing.
1980 # Note: This is likely not necessary on Python 3.
1999 # Note: This is likely not necessary on Python 3.
1981 ifh.seek(0, os.SEEK_END)
2000 ifh.seek(0, os.SEEK_END)
1982 if dfh:
2001 if dfh:
1983 dfh.seek(0, os.SEEK_END)
2002 dfh.seek(0, os.SEEK_END)
1984
2003
1985 curr = len(self) - 1
2004 curr = len(self) - 1
1986 if not self._inline:
2005 if not self._inline:
1987 transaction.add(self.datafile, offset)
2006 transaction.add(self.datafile, offset)
1988 transaction.add(self.indexfile, curr * len(entry))
2007 transaction.add(self.indexfile, curr * len(entry))
1989 if data[0]:
2008 if data[0]:
1990 dfh.write(data[0])
2009 dfh.write(data[0])
1991 dfh.write(data[1])
2010 dfh.write(data[1])
1992 ifh.write(entry)
2011 ifh.write(entry)
1993 else:
2012 else:
1994 offset += curr * self._io.size
2013 offset += curr * self._io.size
1995 transaction.add(self.indexfile, offset, curr)
2014 transaction.add(self.indexfile, offset, curr)
1996 ifh.write(entry)
2015 ifh.write(entry)
1997 ifh.write(data[0])
2016 ifh.write(data[0])
1998 ifh.write(data[1])
2017 ifh.write(data[1])
1999 self._enforceinlinesize(transaction, ifh)
2018 self._enforceinlinesize(transaction, ifh)
2000
2019
2001 def addgroup(self, deltas, linkmapper, transaction, addrevisioncb=None):
2020 def addgroup(self, deltas, linkmapper, transaction, addrevisioncb=None):
2002 """
2021 """
2003 add a delta group
2022 add a delta group
2004
2023
2005 given a set of deltas, add them to the revision log. the
2024 given a set of deltas, add them to the revision log. the
2006 first delta is against its parent, which should be in our
2025 first delta is against its parent, which should be in our
2007 log, the rest are against the previous delta.
2026 log, the rest are against the previous delta.
2008
2027
2009 If ``addrevisioncb`` is defined, it will be called with arguments of
2028 If ``addrevisioncb`` is defined, it will be called with arguments of
2010 this revlog and the node that was added.
2029 this revlog and the node that was added.
2011 """
2030 """
2012
2031
2013 nodes = []
2032 nodes = []
2014
2033
2015 r = len(self)
2034 r = len(self)
2016 end = 0
2035 end = 0
2017 if r:
2036 if r:
2018 end = self.end(r - 1)
2037 end = self.end(r - 1)
2019 ifh = self._indexfp("a+")
2038 ifh = self._indexfp("a+")
2020 isize = r * self._io.size
2039 isize = r * self._io.size
2021 if self._inline:
2040 if self._inline:
2022 transaction.add(self.indexfile, end + isize, r)
2041 transaction.add(self.indexfile, end + isize, r)
2023 dfh = None
2042 dfh = None
2024 else:
2043 else:
2025 transaction.add(self.indexfile, isize, r)
2044 transaction.add(self.indexfile, isize, r)
2026 transaction.add(self.datafile, end)
2045 transaction.add(self.datafile, end)
2027 dfh = self._datafp("a+")
2046 dfh = self._datafp("a+")
2028 def flush():
2047 def flush():
2029 if dfh:
2048 if dfh:
2030 dfh.flush()
2049 dfh.flush()
2031 ifh.flush()
2050 ifh.flush()
2032 try:
2051 try:
2033 deltacomputer = deltautil.deltacomputer(self)
2052 deltacomputer = deltautil.deltacomputer(self)
2034 # loop through our set of deltas
2053 # loop through our set of deltas
2035 for data in deltas:
2054 for data in deltas:
2036 node, p1, p2, linknode, deltabase, delta, flags = data
2055 node, p1, p2, linknode, deltabase, delta, flags = data
2037 link = linkmapper(linknode)
2056 link = linkmapper(linknode)
2038 flags = flags or REVIDX_DEFAULT_FLAGS
2057 flags = flags or REVIDX_DEFAULT_FLAGS
2039
2058
2040 nodes.append(node)
2059 nodes.append(node)
2041
2060
2042 if node in self.nodemap:
2061 if node in self.nodemap:
2043 self._nodeduplicatecallback(transaction, node)
2062 self._nodeduplicatecallback(transaction, node)
2044 # this can happen if two branches make the same change
2063 # this can happen if two branches make the same change
2045 continue
2064 continue
2046
2065
2047 for p in (p1, p2):
2066 for p in (p1, p2):
2048 if p not in self.nodemap:
2067 if p not in self.nodemap:
2049 raise error.LookupError(p, self.indexfile,
2068 raise error.LookupError(p, self.indexfile,
2050 _('unknown parent'))
2069 _('unknown parent'))
2051
2070
2052 if deltabase not in self.nodemap:
2071 if deltabase not in self.nodemap:
2053 raise error.LookupError(deltabase, self.indexfile,
2072 raise error.LookupError(deltabase, self.indexfile,
2054 _('unknown delta base'))
2073 _('unknown delta base'))
2055
2074
2056 baserev = self.rev(deltabase)
2075 baserev = self.rev(deltabase)
2057
2076
2058 if baserev != nullrev and self.iscensored(baserev):
2077 if baserev != nullrev and self.iscensored(baserev):
2059 # if base is censored, delta must be full replacement in a
2078 # if base is censored, delta must be full replacement in a
2060 # single patch operation
2079 # single patch operation
2061 hlen = struct.calcsize(">lll")
2080 hlen = struct.calcsize(">lll")
2062 oldlen = self.rawsize(baserev)
2081 oldlen = self.rawsize(baserev)
2063 newlen = len(delta) - hlen
2082 newlen = len(delta) - hlen
2064 if delta[:hlen] != mdiff.replacediffheader(oldlen, newlen):
2083 if delta[:hlen] != mdiff.replacediffheader(oldlen, newlen):
2065 raise error.CensoredBaseError(self.indexfile,
2084 raise error.CensoredBaseError(self.indexfile,
2066 self.node(baserev))
2085 self.node(baserev))
2067
2086
2068 if not flags and self._peek_iscensored(baserev, delta, flush):
2087 if not flags and self._peek_iscensored(baserev, delta, flush):
2069 flags |= REVIDX_ISCENSORED
2088 flags |= REVIDX_ISCENSORED
2070
2089
2071 # We assume consumers of addrevisioncb will want to retrieve
2090 # We assume consumers of addrevisioncb will want to retrieve
2072 # the added revision, which will require a call to
2091 # the added revision, which will require a call to
2073 # revision(). revision() will fast path if there is a cache
2092 # revision(). revision() will fast path if there is a cache
2074 # hit. So, we tell _addrevision() to always cache in this case.
2093 # hit. So, we tell _addrevision() to always cache in this case.
2075 # We're only using addgroup() in the context of changegroup
2094 # We're only using addgroup() in the context of changegroup
2076 # generation so the revision data can always be handled as raw
2095 # generation so the revision data can always be handled as raw
2077 # by the flagprocessor.
2096 # by the flagprocessor.
2078 self._addrevision(node, None, transaction, link,
2097 self._addrevision(node, None, transaction, link,
2079 p1, p2, flags, (baserev, delta),
2098 p1, p2, flags, (baserev, delta),
2080 ifh, dfh,
2099 ifh, dfh,
2081 alwayscache=bool(addrevisioncb),
2100 alwayscache=bool(addrevisioncb),
2082 deltacomputer=deltacomputer)
2101 deltacomputer=deltacomputer)
2083
2102
2084 if addrevisioncb:
2103 if addrevisioncb:
2085 addrevisioncb(self, node)
2104 addrevisioncb(self, node)
2086
2105
2087 if not dfh and not self._inline:
2106 if not dfh and not self._inline:
2088 # addrevision switched from inline to conventional
2107 # addrevision switched from inline to conventional
2089 # reopen the index
2108 # reopen the index
2090 ifh.close()
2109 ifh.close()
2091 dfh = self._datafp("a+")
2110 dfh = self._datafp("a+")
2092 ifh = self._indexfp("a+")
2111 ifh = self._indexfp("a+")
2093 finally:
2112 finally:
2094 if dfh:
2113 if dfh:
2095 dfh.close()
2114 dfh.close()
2096 ifh.close()
2115 ifh.close()
2097
2116
2098 return nodes
2117 return nodes
2099
2118
2100 def iscensored(self, rev):
2119 def iscensored(self, rev):
2101 """Check if a file revision is censored."""
2120 """Check if a file revision is censored."""
2102 if not self._censorable:
2121 if not self._censorable:
2103 return False
2122 return False
2104
2123
2105 return self.flags(rev) & REVIDX_ISCENSORED
2124 return self.flags(rev) & REVIDX_ISCENSORED
2106
2125
2107 def _peek_iscensored(self, baserev, delta, flush):
2126 def _peek_iscensored(self, baserev, delta, flush):
2108 """Quickly check if a delta produces a censored revision."""
2127 """Quickly check if a delta produces a censored revision."""
2109 if not self._censorable:
2128 if not self._censorable:
2110 return False
2129 return False
2111
2130
2112 return storageutil.deltaiscensored(delta, baserev, self.rawsize)
2131 return storageutil.deltaiscensored(delta, baserev, self.rawsize)
2113
2132
2114 def getstrippoint(self, minlink):
2133 def getstrippoint(self, minlink):
2115 """find the minimum rev that must be stripped to strip the linkrev
2134 """find the minimum rev that must be stripped to strip the linkrev
2116
2135
2117 Returns a tuple containing the minimum rev and a set of all revs that
2136 Returns a tuple containing the minimum rev and a set of all revs that
2118 have linkrevs that will be broken by this strip.
2137 have linkrevs that will be broken by this strip.
2119 """
2138 """
2120 return storageutil.resolvestripinfo(minlink, len(self) - 1,
2139 return storageutil.resolvestripinfo(minlink, len(self) - 1,
2121 self.headrevs(),
2140 self.headrevs(),
2122 self.linkrev, self.parentrevs)
2141 self.linkrev, self.parentrevs)
2123
2142
2124 def strip(self, minlink, transaction):
2143 def strip(self, minlink, transaction):
2125 """truncate the revlog on the first revision with a linkrev >= minlink
2144 """truncate the revlog on the first revision with a linkrev >= minlink
2126
2145
2127 This function is called when we're stripping revision minlink and
2146 This function is called when we're stripping revision minlink and
2128 its descendants from the repository.
2147 its descendants from the repository.
2129
2148
2130 We have to remove all revisions with linkrev >= minlink, because
2149 We have to remove all revisions with linkrev >= minlink, because
2131 the equivalent changelog revisions will be renumbered after the
2150 the equivalent changelog revisions will be renumbered after the
2132 strip.
2151 strip.
2133
2152
2134 So we truncate the revlog on the first of these revisions, and
2153 So we truncate the revlog on the first of these revisions, and
2135 trust that the caller has saved the revisions that shouldn't be
2154 trust that the caller has saved the revisions that shouldn't be
2136 removed and that it'll re-add them after this truncation.
2155 removed and that it'll re-add them after this truncation.
2137 """
2156 """
2138 if len(self) == 0:
2157 if len(self) == 0:
2139 return
2158 return
2140
2159
2141 rev, _ = self.getstrippoint(minlink)
2160 rev, _ = self.getstrippoint(minlink)
2142 if rev == len(self):
2161 if rev == len(self):
2143 return
2162 return
2144
2163
2145 # first truncate the files on disk
2164 # first truncate the files on disk
2146 end = self.start(rev)
2165 end = self.start(rev)
2147 if not self._inline:
2166 if not self._inline:
2148 transaction.add(self.datafile, end)
2167 transaction.add(self.datafile, end)
2149 end = rev * self._io.size
2168 end = rev * self._io.size
2150 else:
2169 else:
2151 end += rev * self._io.size
2170 end += rev * self._io.size
2152
2171
2153 transaction.add(self.indexfile, end)
2172 transaction.add(self.indexfile, end)
2154
2173
2155 # then reset internal state in memory to forget those revisions
2174 # then reset internal state in memory to forget those revisions
2156 self._revisioncache = None
2175 self._revisioncache = None
2157 self._chaininfocache = {}
2176 self._chaininfocache = {}
2158 self._chunkclear()
2177 self._chunkclear()
2159 for x in pycompat.xrange(rev, len(self)):
2178 for x in pycompat.xrange(rev, len(self)):
2160 del self.nodemap[self.node(x)]
2179 del self.nodemap[self.node(x)]
2161
2180
2162 del self.index[rev:-1]
2181 del self.index[rev:-1]
2163 self._nodepos = None
2182 self._nodepos = None
2164
2183
2165 def checksize(self):
2184 def checksize(self):
2166 expected = 0
2185 expected = 0
2167 if len(self):
2186 if len(self):
2168 expected = max(0, self.end(len(self) - 1))
2187 expected = max(0, self.end(len(self) - 1))
2169
2188
2170 try:
2189 try:
2171 with self._datafp() as f:
2190 with self._datafp() as f:
2172 f.seek(0, 2)
2191 f.seek(0, 2)
2173 actual = f.tell()
2192 actual = f.tell()
2174 dd = actual - expected
2193 dd = actual - expected
2175 except IOError as inst:
2194 except IOError as inst:
2176 if inst.errno != errno.ENOENT:
2195 if inst.errno != errno.ENOENT:
2177 raise
2196 raise
2178 dd = 0
2197 dd = 0
2179
2198
2180 try:
2199 try:
2181 f = self.opener(self.indexfile)
2200 f = self.opener(self.indexfile)
2182 f.seek(0, 2)
2201 f.seek(0, 2)
2183 actual = f.tell()
2202 actual = f.tell()
2184 f.close()
2203 f.close()
2185 s = self._io.size
2204 s = self._io.size
2186 i = max(0, actual // s)
2205 i = max(0, actual // s)
2187 di = actual - (i * s)
2206 di = actual - (i * s)
2188 if self._inline:
2207 if self._inline:
2189 databytes = 0
2208 databytes = 0
2190 for r in self:
2209 for r in self:
2191 databytes += max(0, self.length(r))
2210 databytes += max(0, self.length(r))
2192 dd = 0
2211 dd = 0
2193 di = actual - len(self) * s - databytes
2212 di = actual - len(self) * s - databytes
2194 except IOError as inst:
2213 except IOError as inst:
2195 if inst.errno != errno.ENOENT:
2214 if inst.errno != errno.ENOENT:
2196 raise
2215 raise
2197 di = 0
2216 di = 0
2198
2217
2199 return (dd, di)
2218 return (dd, di)
2200
2219
2201 def files(self):
2220 def files(self):
2202 res = [self.indexfile]
2221 res = [self.indexfile]
2203 if not self._inline:
2222 if not self._inline:
2204 res.append(self.datafile)
2223 res.append(self.datafile)
2205 return res
2224 return res
2206
2225
2207 def emitrevisions(self, nodes, nodesorder=None, revisiondata=False,
2226 def emitrevisions(self, nodes, nodesorder=None, revisiondata=False,
2208 assumehaveparentrevisions=False,
2227 assumehaveparentrevisions=False,
2209 deltamode=repository.CG_DELTAMODE_STD):
2228 deltamode=repository.CG_DELTAMODE_STD):
2210 if nodesorder not in ('nodes', 'storage', 'linear', None):
2229 if nodesorder not in ('nodes', 'storage', 'linear', None):
2211 raise error.ProgrammingError('unhandled value for nodesorder: %s' %
2230 raise error.ProgrammingError('unhandled value for nodesorder: %s' %
2212 nodesorder)
2231 nodesorder)
2213
2232
2214 if nodesorder is None and not self._generaldelta:
2233 if nodesorder is None and not self._generaldelta:
2215 nodesorder = 'storage'
2234 nodesorder = 'storage'
2216
2235
2217 if (not self._storedeltachains and
2236 if (not self._storedeltachains and
2218 deltamode != repository.CG_DELTAMODE_PREV):
2237 deltamode != repository.CG_DELTAMODE_PREV):
2219 deltamode = repository.CG_DELTAMODE_FULL
2238 deltamode = repository.CG_DELTAMODE_FULL
2220
2239
2221 return storageutil.emitrevisions(
2240 return storageutil.emitrevisions(
2222 self, nodes, nodesorder, revlogrevisiondelta,
2241 self, nodes, nodesorder, revlogrevisiondelta,
2223 deltaparentfn=self.deltaparent,
2242 deltaparentfn=self.deltaparent,
2224 candeltafn=self.candelta,
2243 candeltafn=self.candelta,
2225 rawsizefn=self.rawsize,
2244 rawsizefn=self.rawsize,
2226 revdifffn=self.revdiff,
2245 revdifffn=self.revdiff,
2227 flagsfn=self.flags,
2246 flagsfn=self.flags,
2228 deltamode=deltamode,
2247 deltamode=deltamode,
2229 revisiondata=revisiondata,
2248 revisiondata=revisiondata,
2230 assumehaveparentrevisions=assumehaveparentrevisions)
2249 assumehaveparentrevisions=assumehaveparentrevisions)
2231
2250
2232 DELTAREUSEALWAYS = 'always'
2251 DELTAREUSEALWAYS = 'always'
2233 DELTAREUSESAMEREVS = 'samerevs'
2252 DELTAREUSESAMEREVS = 'samerevs'
2234 DELTAREUSENEVER = 'never'
2253 DELTAREUSENEVER = 'never'
2235
2254
2236 DELTAREUSEFULLADD = 'fulladd'
2255 DELTAREUSEFULLADD = 'fulladd'
2237
2256
2238 DELTAREUSEALL = {'always', 'samerevs', 'never', 'fulladd'}
2257 DELTAREUSEALL = {'always', 'samerevs', 'never', 'fulladd'}
2239
2258
2240 def clone(self, tr, destrevlog, addrevisioncb=None,
2259 def clone(self, tr, destrevlog, addrevisioncb=None,
2241 deltareuse=DELTAREUSESAMEREVS, deltabothparents=None):
2260 deltareuse=DELTAREUSESAMEREVS, deltabothparents=None):
2242 """Copy this revlog to another, possibly with format changes.
2261 """Copy this revlog to another, possibly with format changes.
2243
2262
2244 The destination revlog will contain the same revisions and nodes.
2263 The destination revlog will contain the same revisions and nodes.
2245 However, it may not be bit-for-bit identical due to e.g. delta encoding
2264 However, it may not be bit-for-bit identical due to e.g. delta encoding
2246 differences.
2265 differences.
2247
2266
2248 The ``deltareuse`` argument control how deltas from the existing revlog
2267 The ``deltareuse`` argument control how deltas from the existing revlog
2249 are preserved in the destination revlog. The argument can have the
2268 are preserved in the destination revlog. The argument can have the
2250 following values:
2269 following values:
2251
2270
2252 DELTAREUSEALWAYS
2271 DELTAREUSEALWAYS
2253 Deltas will always be reused (if possible), even if the destination
2272 Deltas will always be reused (if possible), even if the destination
2254 revlog would not select the same revisions for the delta. This is the
2273 revlog would not select the same revisions for the delta. This is the
2255 fastest mode of operation.
2274 fastest mode of operation.
2256 DELTAREUSESAMEREVS
2275 DELTAREUSESAMEREVS
2257 Deltas will be reused if the destination revlog would pick the same
2276 Deltas will be reused if the destination revlog would pick the same
2258 revisions for the delta. This mode strikes a balance between speed
2277 revisions for the delta. This mode strikes a balance between speed
2259 and optimization.
2278 and optimization.
2260 DELTAREUSENEVER
2279 DELTAREUSENEVER
2261 Deltas will never be reused. This is the slowest mode of execution.
2280 Deltas will never be reused. This is the slowest mode of execution.
2262 This mode can be used to recompute deltas (e.g. if the diff/delta
2281 This mode can be used to recompute deltas (e.g. if the diff/delta
2263 algorithm changes).
2282 algorithm changes).
2264
2283
2265 Delta computation can be slow, so the choice of delta reuse policy can
2284 Delta computation can be slow, so the choice of delta reuse policy can
2266 significantly affect run time.
2285 significantly affect run time.
2267
2286
2268 The default policy (``DELTAREUSESAMEREVS``) strikes a balance between
2287 The default policy (``DELTAREUSESAMEREVS``) strikes a balance between
2269 two extremes. Deltas will be reused if they are appropriate. But if the
2288 two extremes. Deltas will be reused if they are appropriate. But if the
2270 delta could choose a better revision, it will do so. This means if you
2289 delta could choose a better revision, it will do so. This means if you
2271 are converting a non-generaldelta revlog to a generaldelta revlog,
2290 are converting a non-generaldelta revlog to a generaldelta revlog,
2272 deltas will be recomputed if the delta's parent isn't a parent of the
2291 deltas will be recomputed if the delta's parent isn't a parent of the
2273 revision.
2292 revision.
2274
2293
2275 In addition to the delta policy, the ``deltabothparents`` argument
2294 In addition to the delta policy, the ``deltabothparents`` argument
2276 controls whether to compute deltas against both parents for merges.
2295 controls whether to compute deltas against both parents for merges.
2277 By default, the current default is used.
2296 By default, the current default is used.
2278 """
2297 """
2279 if deltareuse not in self.DELTAREUSEALL:
2298 if deltareuse not in self.DELTAREUSEALL:
2280 raise ValueError(_('value for deltareuse invalid: %s') % deltareuse)
2299 raise ValueError(_('value for deltareuse invalid: %s') % deltareuse)
2281
2300
2282 if len(destrevlog):
2301 if len(destrevlog):
2283 raise ValueError(_('destination revlog is not empty'))
2302 raise ValueError(_('destination revlog is not empty'))
2284
2303
2285 if getattr(self, 'filteredrevs', None):
2304 if getattr(self, 'filteredrevs', None):
2286 raise ValueError(_('source revlog has filtered revisions'))
2305 raise ValueError(_('source revlog has filtered revisions'))
2287 if getattr(destrevlog, 'filteredrevs', None):
2306 if getattr(destrevlog, 'filteredrevs', None):
2288 raise ValueError(_('destination revlog has filtered revisions'))
2307 raise ValueError(_('destination revlog has filtered revisions'))
2289
2308
2290 # lazydeltabase controls whether to reuse a cached delta, if possible.
2309 # lazydeltabase controls whether to reuse a cached delta, if possible.
2291 oldlazydeltabase = destrevlog._lazydeltabase
2310 oldlazydeltabase = destrevlog._lazydeltabase
2292 oldamd = destrevlog._deltabothparents
2311 oldamd = destrevlog._deltabothparents
2293
2312
2294 try:
2313 try:
2295 if deltareuse == self.DELTAREUSEALWAYS:
2314 if deltareuse == self.DELTAREUSEALWAYS:
2296 destrevlog._lazydeltabase = True
2315 destrevlog._lazydeltabase = True
2297 elif deltareuse == self.DELTAREUSESAMEREVS:
2316 elif deltareuse == self.DELTAREUSESAMEREVS:
2298 destrevlog._lazydeltabase = False
2317 destrevlog._lazydeltabase = False
2299
2318
2300 destrevlog._deltabothparents = deltabothparents or oldamd
2319 destrevlog._deltabothparents = deltabothparents or oldamd
2301
2320
2302 populatecachedelta = deltareuse in (self.DELTAREUSEALWAYS,
2321 populatecachedelta = deltareuse in (self.DELTAREUSEALWAYS,
2303 self.DELTAREUSESAMEREVS)
2322 self.DELTAREUSESAMEREVS)
2304
2323
2305 deltacomputer = deltautil.deltacomputer(destrevlog)
2324 deltacomputer = deltautil.deltacomputer(destrevlog)
2306 index = self.index
2325 index = self.index
2307 for rev in self:
2326 for rev in self:
2308 entry = index[rev]
2327 entry = index[rev]
2309
2328
2310 # Some classes override linkrev to take filtered revs into
2329 # Some classes override linkrev to take filtered revs into
2311 # account. Use raw entry from index.
2330 # account. Use raw entry from index.
2312 flags = entry[0] & 0xffff
2331 flags = entry[0] & 0xffff
2313 linkrev = entry[4]
2332 linkrev = entry[4]
2314 p1 = index[entry[5]][7]
2333 p1 = index[entry[5]][7]
2315 p2 = index[entry[6]][7]
2334 p2 = index[entry[6]][7]
2316 node = entry[7]
2335 node = entry[7]
2317
2336
2318 # (Possibly) reuse the delta from the revlog if allowed and
2337 # (Possibly) reuse the delta from the revlog if allowed and
2319 # the revlog chunk is a delta.
2338 # the revlog chunk is a delta.
2320 cachedelta = None
2339 cachedelta = None
2321 rawtext = None
2340 rawtext = None
2322 if populatecachedelta:
2341 if populatecachedelta:
2323 dp = self.deltaparent(rev)
2342 dp = self.deltaparent(rev)
2324 if dp != nullrev:
2343 if dp != nullrev:
2325 cachedelta = (dp, bytes(self._chunk(rev)))
2344 cachedelta = (dp, bytes(self._chunk(rev)))
2326
2345
2327 if not cachedelta:
2346 if not cachedelta:
2328 rawtext = self.revision(rev, raw=True)
2347 rawtext = self.revision(rev, raw=True)
2329
2348
2330
2349
2331 if deltareuse == self.DELTAREUSEFULLADD:
2350 if deltareuse == self.DELTAREUSEFULLADD:
2332 destrevlog.addrevision(rawtext, tr, linkrev, p1, p2,
2351 destrevlog.addrevision(rawtext, tr, linkrev, p1, p2,
2333 cachedelta=cachedelta,
2352 cachedelta=cachedelta,
2334 node=node, flags=flags,
2353 node=node, flags=flags,
2335 deltacomputer=deltacomputer)
2354 deltacomputer=deltacomputer)
2336 else:
2355 else:
2337 ifh = destrevlog.opener(destrevlog.indexfile, 'a+',
2356 ifh = destrevlog.opener(destrevlog.indexfile, 'a+',
2338 checkambig=False)
2357 checkambig=False)
2339 dfh = None
2358 dfh = None
2340 if not destrevlog._inline:
2359 if not destrevlog._inline:
2341 dfh = destrevlog.opener(destrevlog.datafile, 'a+')
2360 dfh = destrevlog.opener(destrevlog.datafile, 'a+')
2342 try:
2361 try:
2343 destrevlog._addrevision(node, rawtext, tr, linkrev, p1,
2362 destrevlog._addrevision(node, rawtext, tr, linkrev, p1,
2344 p2, flags, cachedelta, ifh, dfh,
2363 p2, flags, cachedelta, ifh, dfh,
2345 deltacomputer=deltacomputer)
2364 deltacomputer=deltacomputer)
2346 finally:
2365 finally:
2347 if dfh:
2366 if dfh:
2348 dfh.close()
2367 dfh.close()
2349 ifh.close()
2368 ifh.close()
2350
2369
2351 if addrevisioncb:
2370 if addrevisioncb:
2352 addrevisioncb(self, rev, node)
2371 addrevisioncb(self, rev, node)
2353 finally:
2372 finally:
2354 destrevlog._lazydeltabase = oldlazydeltabase
2373 destrevlog._lazydeltabase = oldlazydeltabase
2355 destrevlog._deltabothparents = oldamd
2374 destrevlog._deltabothparents = oldamd
2356
2375
2357 def censorrevision(self, tr, censornode, tombstone=b''):
2376 def censorrevision(self, tr, censornode, tombstone=b''):
2358 if (self.version & 0xFFFF) == REVLOGV0:
2377 if (self.version & 0xFFFF) == REVLOGV0:
2359 raise error.RevlogError(_('cannot censor with version %d revlogs') %
2378 raise error.RevlogError(_('cannot censor with version %d revlogs') %
2360 self.version)
2379 self.version)
2361
2380
2362 censorrev = self.rev(censornode)
2381 censorrev = self.rev(censornode)
2363 tombstone = storageutil.packmeta({b'censored': tombstone}, b'')
2382 tombstone = storageutil.packmeta({b'censored': tombstone}, b'')
2364
2383
2365 if len(tombstone) > self.rawsize(censorrev):
2384 if len(tombstone) > self.rawsize(censorrev):
2366 raise error.Abort(_('censor tombstone must be no longer than '
2385 raise error.Abort(_('censor tombstone must be no longer than '
2367 'censored data'))
2386 'censored data'))
2368
2387
2369 # Rewriting the revlog in place is hard. Our strategy for censoring is
2388 # Rewriting the revlog in place is hard. Our strategy for censoring is
2370 # to create a new revlog, copy all revisions to it, then replace the
2389 # to create a new revlog, copy all revisions to it, then replace the
2371 # revlogs on transaction close.
2390 # revlogs on transaction close.
2372
2391
2373 newindexfile = self.indexfile + b'.tmpcensored'
2392 newindexfile = self.indexfile + b'.tmpcensored'
2374 newdatafile = self.datafile + b'.tmpcensored'
2393 newdatafile = self.datafile + b'.tmpcensored'
2375
2394
2376 # This is a bit dangerous. We could easily have a mismatch of state.
2395 # This is a bit dangerous. We could easily have a mismatch of state.
2377 newrl = revlog(self.opener, newindexfile, newdatafile,
2396 newrl = revlog(self.opener, newindexfile, newdatafile,
2378 censorable=True)
2397 censorable=True)
2379 newrl.version = self.version
2398 newrl.version = self.version
2380 newrl._generaldelta = self._generaldelta
2399 newrl._generaldelta = self._generaldelta
2381 newrl._io = self._io
2400 newrl._io = self._io
2382
2401
2383 for rev in self.revs():
2402 for rev in self.revs():
2384 node = self.node(rev)
2403 node = self.node(rev)
2385 p1, p2 = self.parents(node)
2404 p1, p2 = self.parents(node)
2386
2405
2387 if rev == censorrev:
2406 if rev == censorrev:
2388 newrl.addrawrevision(tombstone, tr, self.linkrev(censorrev),
2407 newrl.addrawrevision(tombstone, tr, self.linkrev(censorrev),
2389 p1, p2, censornode, REVIDX_ISCENSORED)
2408 p1, p2, censornode, REVIDX_ISCENSORED)
2390
2409
2391 if newrl.deltaparent(rev) != nullrev:
2410 if newrl.deltaparent(rev) != nullrev:
2392 raise error.Abort(_('censored revision stored as delta; '
2411 raise error.Abort(_('censored revision stored as delta; '
2393 'cannot censor'),
2412 'cannot censor'),
2394 hint=_('censoring of revlogs is not '
2413 hint=_('censoring of revlogs is not '
2395 'fully implemented; please report '
2414 'fully implemented; please report '
2396 'this bug'))
2415 'this bug'))
2397 continue
2416 continue
2398
2417
2399 if self.iscensored(rev):
2418 if self.iscensored(rev):
2400 if self.deltaparent(rev) != nullrev:
2419 if self.deltaparent(rev) != nullrev:
2401 raise error.Abort(_('cannot censor due to censored '
2420 raise error.Abort(_('cannot censor due to censored '
2402 'revision having delta stored'))
2421 'revision having delta stored'))
2403 rawtext = self._chunk(rev)
2422 rawtext = self._chunk(rev)
2404 else:
2423 else:
2405 rawtext = self.revision(rev, raw=True)
2424 rawtext = self.revision(rev, raw=True)
2406
2425
2407 newrl.addrawrevision(rawtext, tr, self.linkrev(rev), p1, p2, node,
2426 newrl.addrawrevision(rawtext, tr, self.linkrev(rev), p1, p2, node,
2408 self.flags(rev))
2427 self.flags(rev))
2409
2428
2410 tr.addbackup(self.indexfile, location='store')
2429 tr.addbackup(self.indexfile, location='store')
2411 if not self._inline:
2430 if not self._inline:
2412 tr.addbackup(self.datafile, location='store')
2431 tr.addbackup(self.datafile, location='store')
2413
2432
2414 self.opener.rename(newrl.indexfile, self.indexfile)
2433 self.opener.rename(newrl.indexfile, self.indexfile)
2415 if not self._inline:
2434 if not self._inline:
2416 self.opener.rename(newrl.datafile, self.datafile)
2435 self.opener.rename(newrl.datafile, self.datafile)
2417
2436
2418 self.clearcaches()
2437 self.clearcaches()
2419 self._loadindex(self.version, None)
2438 self._loadindex(self.version, None)
2420
2439
2421 def verifyintegrity(self, state):
2440 def verifyintegrity(self, state):
2422 """Verifies the integrity of the revlog.
2441 """Verifies the integrity of the revlog.
2423
2442
2424 Yields ``revlogproblem`` instances describing problems that are
2443 Yields ``revlogproblem`` instances describing problems that are
2425 found.
2444 found.
2426 """
2445 """
2427 dd, di = self.checksize()
2446 dd, di = self.checksize()
2428 if dd:
2447 if dd:
2429 yield revlogproblem(error=_('data length off by %d bytes') % dd)
2448 yield revlogproblem(error=_('data length off by %d bytes') % dd)
2430 if di:
2449 if di:
2431 yield revlogproblem(error=_('index contains %d extra bytes') % di)
2450 yield revlogproblem(error=_('index contains %d extra bytes') % di)
2432
2451
2433 version = self.version & 0xFFFF
2452 version = self.version & 0xFFFF
2434
2453
2435 # The verifier tells us what version revlog we should be.
2454 # The verifier tells us what version revlog we should be.
2436 if version != state['expectedversion']:
2455 if version != state['expectedversion']:
2437 yield revlogproblem(
2456 yield revlogproblem(
2438 warning=_("warning: '%s' uses revlog format %d; expected %d") %
2457 warning=_("warning: '%s' uses revlog format %d; expected %d") %
2439 (self.indexfile, version, state['expectedversion']))
2458 (self.indexfile, version, state['expectedversion']))
2440
2459
2441 state['skipread'] = set()
2460 state['skipread'] = set()
2442
2461
2443 for rev in self:
2462 for rev in self:
2444 node = self.node(rev)
2463 node = self.node(rev)
2445
2464
2446 # Verify contents. 4 cases to care about:
2465 # Verify contents. 4 cases to care about:
2447 #
2466 #
2448 # common: the most common case
2467 # common: the most common case
2449 # rename: with a rename
2468 # rename: with a rename
2450 # meta: file content starts with b'\1\n', the metadata
2469 # meta: file content starts with b'\1\n', the metadata
2451 # header defined in filelog.py, but without a rename
2470 # header defined in filelog.py, but without a rename
2452 # ext: content stored externally
2471 # ext: content stored externally
2453 #
2472 #
2454 # More formally, their differences are shown below:
2473 # More formally, their differences are shown below:
2455 #
2474 #
2456 # | common | rename | meta | ext
2475 # | common | rename | meta | ext
2457 # -------------------------------------------------------
2476 # -------------------------------------------------------
2458 # flags() | 0 | 0 | 0 | not 0
2477 # flags() | 0 | 0 | 0 | not 0
2459 # renamed() | False | True | False | ?
2478 # renamed() | False | True | False | ?
2460 # rawtext[0:2]=='\1\n'| False | True | True | ?
2479 # rawtext[0:2]=='\1\n'| False | True | True | ?
2461 #
2480 #
2462 # "rawtext" means the raw text stored in revlog data, which
2481 # "rawtext" means the raw text stored in revlog data, which
2463 # could be retrieved by "revision(rev, raw=True)". "text"
2482 # could be retrieved by "revision(rev, raw=True)". "text"
2464 # mentioned below is "revision(rev, raw=False)".
2483 # mentioned below is "revision(rev, raw=False)".
2465 #
2484 #
2466 # There are 3 different lengths stored physically:
2485 # There are 3 different lengths stored physically:
2467 # 1. L1: rawsize, stored in revlog index
2486 # 1. L1: rawsize, stored in revlog index
2468 # 2. L2: len(rawtext), stored in revlog data
2487 # 2. L2: len(rawtext), stored in revlog data
2469 # 3. L3: len(text), stored in revlog data if flags==0, or
2488 # 3. L3: len(text), stored in revlog data if flags==0, or
2470 # possibly somewhere else if flags!=0
2489 # possibly somewhere else if flags!=0
2471 #
2490 #
2472 # L1 should be equal to L2. L3 could be different from them.
2491 # L1 should be equal to L2. L3 could be different from them.
2473 # "text" may or may not affect commit hash depending on flag
2492 # "text" may or may not affect commit hash depending on flag
2474 # processors (see revlog.addflagprocessor).
2493 # processors (see revlog.addflagprocessor).
2475 #
2494 #
2476 # | common | rename | meta | ext
2495 # | common | rename | meta | ext
2477 # -------------------------------------------------
2496 # -------------------------------------------------
2478 # rawsize() | L1 | L1 | L1 | L1
2497 # rawsize() | L1 | L1 | L1 | L1
2479 # size() | L1 | L2-LM | L1(*) | L1 (?)
2498 # size() | L1 | L2-LM | L1(*) | L1 (?)
2480 # len(rawtext) | L2 | L2 | L2 | L2
2499 # len(rawtext) | L2 | L2 | L2 | L2
2481 # len(text) | L2 | L2 | L2 | L3
2500 # len(text) | L2 | L2 | L2 | L3
2482 # len(read()) | L2 | L2-LM | L2-LM | L3 (?)
2501 # len(read()) | L2 | L2-LM | L2-LM | L3 (?)
2483 #
2502 #
2484 # LM: length of metadata, depending on rawtext
2503 # LM: length of metadata, depending on rawtext
2485 # (*): not ideal, see comment in filelog.size
2504 # (*): not ideal, see comment in filelog.size
2486 # (?): could be "- len(meta)" if the resolved content has
2505 # (?): could be "- len(meta)" if the resolved content has
2487 # rename metadata
2506 # rename metadata
2488 #
2507 #
2489 # Checks needed to be done:
2508 # Checks needed to be done:
2490 # 1. length check: L1 == L2, in all cases.
2509 # 1. length check: L1 == L2, in all cases.
2491 # 2. hash check: depending on flag processor, we may need to
2510 # 2. hash check: depending on flag processor, we may need to
2492 # use either "text" (external), or "rawtext" (in revlog).
2511 # use either "text" (external), or "rawtext" (in revlog).
2493
2512
2494 try:
2513 try:
2495 skipflags = state.get('skipflags', 0)
2514 skipflags = state.get('skipflags', 0)
2496 if skipflags:
2515 if skipflags:
2497 skipflags &= self.flags(rev)
2516 skipflags &= self.flags(rev)
2498
2517
2499 if skipflags:
2518 if skipflags:
2500 state['skipread'].add(node)
2519 state['skipread'].add(node)
2501 else:
2520 else:
2502 # Side-effect: read content and verify hash.
2521 # Side-effect: read content and verify hash.
2503 self.revision(node)
2522 self.revision(node)
2504
2523
2505 l1 = self.rawsize(rev)
2524 l1 = self.rawsize(rev)
2506 l2 = len(self.revision(node, raw=True))
2525 l2 = len(self.revision(node, raw=True))
2507
2526
2508 if l1 != l2:
2527 if l1 != l2:
2509 yield revlogproblem(
2528 yield revlogproblem(
2510 error=_('unpacked size is %d, %d expected') % (l2, l1),
2529 error=_('unpacked size is %d, %d expected') % (l2, l1),
2511 node=node)
2530 node=node)
2512
2531
2513 except error.CensoredNodeError:
2532 except error.CensoredNodeError:
2514 if state['erroroncensored']:
2533 if state['erroroncensored']:
2515 yield revlogproblem(error=_('censored file data'),
2534 yield revlogproblem(error=_('censored file data'),
2516 node=node)
2535 node=node)
2517 state['skipread'].add(node)
2536 state['skipread'].add(node)
2518 except Exception as e:
2537 except Exception as e:
2519 yield revlogproblem(
2538 yield revlogproblem(
2520 error=_('unpacking %s: %s') % (short(node),
2539 error=_('unpacking %s: %s') % (short(node),
2521 stringutil.forcebytestr(e)),
2540 stringutil.forcebytestr(e)),
2522 node=node)
2541 node=node)
2523 state['skipread'].add(node)
2542 state['skipread'].add(node)
2524
2543
2525 def storageinfo(self, exclusivefiles=False, sharedfiles=False,
2544 def storageinfo(self, exclusivefiles=False, sharedfiles=False,
2526 revisionscount=False, trackedsize=False,
2545 revisionscount=False, trackedsize=False,
2527 storedsize=False):
2546 storedsize=False):
2528 d = {}
2547 d = {}
2529
2548
2530 if exclusivefiles:
2549 if exclusivefiles:
2531 d['exclusivefiles'] = [(self.opener, self.indexfile)]
2550 d['exclusivefiles'] = [(self.opener, self.indexfile)]
2532 if not self._inline:
2551 if not self._inline:
2533 d['exclusivefiles'].append((self.opener, self.datafile))
2552 d['exclusivefiles'].append((self.opener, self.datafile))
2534
2553
2535 if sharedfiles:
2554 if sharedfiles:
2536 d['sharedfiles'] = []
2555 d['sharedfiles'] = []
2537
2556
2538 if revisionscount:
2557 if revisionscount:
2539 d['revisionscount'] = len(self)
2558 d['revisionscount'] = len(self)
2540
2559
2541 if trackedsize:
2560 if trackedsize:
2542 d['trackedsize'] = sum(map(self.rawsize, iter(self)))
2561 d['trackedsize'] = sum(map(self.rawsize, iter(self)))
2543
2562
2544 if storedsize:
2563 if storedsize:
2545 d['storedsize'] = sum(self.opener.stat(path).st_size
2564 d['storedsize'] = sum(self.opener.stat(path).st_size
2546 for path in self.files())
2565 for path in self.files())
2547
2566
2548 return d
2567 return d
General Comments 0
You need to be logged in to leave comments. Login now