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