mercurial-scm/hg: hgext/censor.py annotate

annotate hgext/censor.py @ 24347:1bcfecbbf569

censor: add censor command to hgext with basic client-side tests The censor command is a core extension which can replace the contents of a historical file revision with a censor "tombstone" which can be exchanged with older clients in place of the real revision data. The command rewrites the filelog by copying revision-by-revision. Care must be taken to expand the fulltext of the children of the censored revision before copying them to the new filelog; they might be stored as deltas against the uncensored revision, and those deltas will be invalidated. For more background on the censorship feature design, see: http://mercurial.selenic.com/wiki/CensorPlan

author	Mike Edgar <adgar@google.com>
date	Sun, 15 Mar 2015 21:52:17 -0400
parents
children	5e111acc1170

rev	line source
24347 1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	1 # Copyright (C) 2015 - Mike Edgar <adgar@google.com>
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	2 #
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	3 # This extension enables removal of file content at a given revision,
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	4 # rewriting the data/metadata of successive revisions to preserve revision log
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	5 # integrity.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	6
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	7 """erase file content at a given revision
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	8
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	9 The censor command instructs Mercurial to erase all content of a file at a given
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	10 revision without updating the changeset hash. This allows existing history to
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	11 remain valid while preventing future clones/pulls from receiving the erased
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	12 data.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	13
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	14 Typical uses for censor are due to security or legal requirements, including::
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	15
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	16 * Passwords, private keys, crytographic material
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	17 * Licensed data/code/libraries for which the license has expired
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	18 * Personally Identifiable Information or other private data
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	19
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	20 Censored file revisions are listed in a tracked file called .hgcensored stored
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	21 in the repository root. The censor command adds an entry to the .hgcensored file
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	22 in the working directory and commits it (much like ``hg tag`` and .hgtags). The
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	23 censored file data is then replaced with a pointer to the new commit, enabling
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	24 verification.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	25
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	26 Censored nodes can interrupt mercurial's typical operation whenever the excised
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	27 data needs to be materialized. Some commands, like ``hg cat``/``hg revert``,
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	28 simply fail when asked to produce censored data. Others, like ``hg verify`` and
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	29 ``hg update``, must be capable of tolerating censored data to continue to
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	30 function in a meaningful way. Such commands only tolerate censored file
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	31 revisions if they are allowed by the policy specified by the "censor.allow"
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	32 config option.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	33 """
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	34
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	35 from mercurial.node import short
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	36 from mercurial import cmdutil, error, filelog, revlog, scmutil, util
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	37 from mercurial.i18n import _
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	38
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	39 cmdtable = {}
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	40 command = cmdutil.command(cmdtable)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	41 testedwith = 'internal'
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	42
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	43 @command('censor',
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	44 [('r', 'rev', '', _('censor file from specified revision'), _('REV')),
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	45 ('t', 'tombstone', '', _('replacement tombstone data'), _('TEXT'))],
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	46 _('-r REV [-t TEXT] [FILE]'))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	47 def censor(ui, repo, path, rev='', tombstone='', **opts):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	48 if not path:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	49 raise util.Abort(_('must specify file path to censor'))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	50 if not rev:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	51 raise util.Abort(_('must specify revision to censor'))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	52
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	53 flog = repo.file(path)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	54 if not len(flog):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	55 raise util.Abort(_('cannot censor file with no history'))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	56
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	57 rev = scmutil.revsingle(repo, rev, rev).rev()
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	58 try:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	59 ctx = repo[rev]
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	60 except KeyError:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	61 raise util.Abort(_('invalid revision identifier %s') % rev)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	62
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	63 try:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	64 fctx = ctx.filectx(path)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	65 except error.LookupError:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	66 raise util.Abort(_('file does not exist at revision %s') % rev)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	67
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	68 fnode = fctx.filenode()
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	69 headctxs = [repo[c] for c in repo.heads()]
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	70 heads = [c for c in headctxs if path in c and c.filenode(path) == fnode]
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	71 if heads:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	72 headlist = ', '.join([short(c.node()) for c in heads])
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	73 raise util.Abort(_('cannot censor file in heads (%s)') % headlist,
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	74 hint=_('clean/delete and commit first'))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	75
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	76 wctx = repo[None]
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	77 wp = wctx.parents()
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	78 if ctx.node() in [p.node() for p in wp]:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	79 raise util.Abort(_('cannot censor working directory'),
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	80 hint=_('clean/delete/update first'))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	81
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	82 flogv = flog.version & 0xFFFF
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	83 if flogv != revlog.REVLOGNG:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	84 raise util.Abort(
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	85 _('censor does not support revlog version %d') % (flogv,))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	86
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	87 tombstone = filelog.packmeta({"censored": tombstone}, "")
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	88
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	89 crev = fctx.filerev()
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	90
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	91 if len(tombstone) > flog.rawsize(crev):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	92 raise util.Abort(_(
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	93 'censor tombstone must be no longer than censored data'))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	94
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	95 # Using two files instead of one makes it easy to rewrite entry-by-entry
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	96 idxread = repo.svfs(flog.indexfile, 'r')
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	97 idxwrite = repo.svfs(flog.indexfile, 'wb', atomictemp=True)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	98 if flog.version & revlog.REVLOGNGINLINEDATA:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	99 dataread, datawrite = idxread, idxwrite
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	100 else:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	101 dataread = repo.svfs(flog.datafile, 'r')
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	102 datawrite = repo.svfs(flog.datafile, 'wb', atomictemp=True)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	103
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	104 # Copy all revlog data up to the entry to be censored.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	105 rio = revlog.revlogio()
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	106 offset = flog.start(crev)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	107
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	108 for chunk in util.filechunkiter(idxread, limit=crev * rio.size):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	109 idxwrite.write(chunk)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	110 for chunk in util.filechunkiter(dataread, limit=offset):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	111 datawrite.write(chunk)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	112
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	113 def rewriteindex(r, newoffs, newdata=None):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	114 """Rewrite the index entry with a new data offset and optional new data.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	115
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	116 The newdata argument, if given, is a tuple of three positive integers:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	117 (new compressed, new uncompressed, added flag bits).
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	118 """
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	119 offlags, comp, uncomp, base, link, p1, p2, nodeid = flog.index[r]
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	120 flags = revlog.gettype(offlags)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	121 if newdata:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	122 comp, uncomp, nflags = newdata
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	123 flags \|= nflags
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	124 offlags = revlog.offset_type(newoffs, flags)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	125 e = (offlags, comp, uncomp, r, link, p1, p2, nodeid)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	126 idxwrite.write(rio.packentry(e, None, flog.version, r))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	127 idxread.seek(rio.size, 1)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	128
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	129 def rewrite(r, offs, data, nflags=revlog.REVIDX_DEFAULT_FLAGS):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	130 """Write the given full text to the filelog with the given data offset.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	131
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	132 Returns:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	133 The integer number of data bytes written, for tracking data offsets.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	134 """
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	135 flag, compdata = flog.compress(data)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	136 newcomp = len(flag) + len(compdata)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	137 rewriteindex(r, offs, (newcomp, len(data), nflags))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	138 datawrite.write(flag)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	139 datawrite.write(compdata)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	140 dataread.seek(flog.length(r), 1)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	141 return newcomp
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	142
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	143 # Rewrite censored revlog entry with (padded) tombstone data.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	144 pad = ' ' * (flog.rawsize(crev) - len(tombstone))
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	145 offset += rewrite(crev, offset, tombstone + pad, revlog.REVIDX_ISCENSORED)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	146
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	147 # Rewrite all following filelog revisions fixing up offsets and deltas.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	148 for srev in xrange(crev + 1, len(flog)):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	149 if crev in flog.parentrevs(srev):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	150 # Immediate children of censored node must be re-added as fulltext.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	151 try:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	152 revdata = flog.revision(srev)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	153 except error.CensoredNodeError, e:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	154 revdata = e.tombstone
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	155 dlen = rewrite(srev, offset, revdata)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	156 else:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	157 # Copy any other revision data verbatim after fixing up the offset.
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	158 rewriteindex(srev, offset)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	159 dlen = flog.length(srev)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	160 for chunk in util.filechunkiter(dataread, limit=dlen):
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	161 datawrite.write(chunk)
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	162 offset += dlen
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	163
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	164 idxread.close()
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	165 idxwrite.close()
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	166 if dataread is not idxread:
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	167 dataread.close()
1bcfecbbf569 censor: add censor command to hgext with basic client-side tests Mike Edgar <adgar@google.com> parents: diff changeset	168 datawrite.close()

Mercurial > public > mercurial-scm > hg

annotate hgext/censor.py @ 24347:1bcfecbbf569