mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 09:17:30 +03:00
febe96048e
Summary: Reuse utilities in the fixcorrupt extension to repair changelog. This is better than fixcorrupt because `hg doctor` does not require a repo object. Some messages are updated so they become more consistent with the rest of `hg doctor`. The main motivation is to get changelog fixed early, so other repair logic can check if a commit hash is known by changelog or not. Reviewed By: markbt Differential Revision: D19864418 fbshipit-source-id: 6f95c6c6191d7db2a474a07a5278a857cf41d8e2
141 lines
4.7 KiB
Python
141 lines
4.7 KiB
Python
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2.
|
|
|
|
# fixcorrupt.py
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import time
|
|
|
|
from edenscm.mercurial import encoding, error, progress, registrar, revlog, util
|
|
from edenscm.mercurial.commands.doctor import quickchecklog
|
|
from edenscm.mercurial.i18n import _
|
|
from edenscm.mercurial.node import nullid
|
|
|
|
|
|
testedwith = "ships-with-fb-hgext"
|
|
|
|
cmdtable = {}
|
|
command = registrar.command(cmdtable)
|
|
|
|
|
|
def truncate(ui, svfs, path, size, dryrun=True, backupprefix=""):
|
|
oldsize = svfs.stat(path).st_size
|
|
if oldsize == size:
|
|
return
|
|
if oldsize < size:
|
|
ui.write(
|
|
_("%s: bad truncation request: %s to %s bytes\n") % (path, oldsize, size)
|
|
)
|
|
return
|
|
ui.write(_("truncating %s from %s to %s bytes\n") % (path, oldsize, size))
|
|
if dryrun:
|
|
return
|
|
|
|
svfs.makedirs("truncate-backups")
|
|
with svfs.open(path, "ab+") as f:
|
|
f.seek(size)
|
|
# backup the part being truncated
|
|
backuppart = f.read(oldsize - size)
|
|
if len(backuppart) != oldsize - size:
|
|
raise error.Abort(_("truncate: cannot backup confidently"))
|
|
with svfs.open(
|
|
"truncate-backups/%s%s.backup-byte-%s-to-%s"
|
|
% (backupprefix, svfs.basename(path), size, oldsize),
|
|
"w",
|
|
) as bkf:
|
|
bkf.write(backuppart)
|
|
f.truncate(size)
|
|
|
|
|
|
@command(
|
|
"debugfixcorrupt",
|
|
[("", "no-dryrun", None, _("write changes (destructive)"))],
|
|
_("[OPTION]... [REV [FILE]...]"),
|
|
)
|
|
def fixcorrupt(ui, repo, *args, **opts):
|
|
"""
|
|
Try to fix a corrupted repo by removing bad revisions at the end of
|
|
changelog and manifest. Only works with remotefilelog repos.
|
|
"""
|
|
# the extension only checks manifest and changelog, so it only works with
|
|
# remotefilelog.
|
|
if "remotefilelog" not in repo.requirements and not encoding.environ.get(
|
|
"SKIPREMOTEFILELOGCHECK"
|
|
):
|
|
raise error.Abort(_("only remotefilelog repo is supported"))
|
|
|
|
dryrun = not opts["no_dryrun"]
|
|
|
|
# we may access hidden nodes
|
|
repo = repo.unfiltered()
|
|
manifestlog = repo.manifestlog
|
|
|
|
# only interested in these 2 revlogs
|
|
logs = [("changelog", repo.changelog)]
|
|
if util.safehasattr(manifestlog, "_revlog"):
|
|
logs += [("manifest", manifestlog._revlog)]
|
|
|
|
# ensure they are REVLOGV1 and do not use inline index
|
|
for name, log in logs:
|
|
if (log.version & 0xFFFF) != revlog.REVLOGV1:
|
|
raise error.Abort(
|
|
_("%s: unsupported revlog version %d") % (name, log.version & 0xFFFF)
|
|
)
|
|
if log._inline:
|
|
raise error.Abort(_("%s: inline index is not supported") % (name))
|
|
if repo.svfs.stat(log.indexfile).st_size // 64 != len(log):
|
|
raise error.Abort(_("unexpected index size for %s") % name)
|
|
|
|
# check changelog first, then manifest. manifest revisions with a bad
|
|
# linkrev is also marked broken, even if passes hash check.
|
|
badrevs = {}
|
|
knownbadrevs = set()
|
|
for name, log in logs:
|
|
rev, linkrev = quickchecklog(ui, log, name, knownbadrevs)
|
|
if rev is None:
|
|
continue
|
|
# sanity check
|
|
if rev >= len(log):
|
|
raise error.Abort(_("%s index is corrupted") % name)
|
|
# do not trust 0 being the linkrev
|
|
if linkrev == 0:
|
|
linkrev = rev
|
|
# save the rev numbers
|
|
badrevs[name] = (rev, linkrev)
|
|
knownbadrevs.add(linkrev)
|
|
|
|
if not badrevs:
|
|
ui.write(_("nothing to do\n"))
|
|
return 1
|
|
|
|
# sync broken revisions from manifest to changelog
|
|
if "manifest" in badrevs:
|
|
badlinkrev = badrevs["manifest"][1]
|
|
badrevs["changelog"] = (badlinkrev, badlinkrev)
|
|
|
|
# truncate revlogs
|
|
backupprefix = "%s-" % int(time.time())
|
|
with repo.wlock(), repo.lock():
|
|
repo.destroying()
|
|
for name, log in logs:
|
|
if name not in badrevs:
|
|
continue
|
|
rev, linkrev = badrevs[name]
|
|
if len(log) != rev:
|
|
ui.write(_("%s: will lose %d revisions\n") % (name, len(log) - rev))
|
|
# rev is broken, so log.start(rev) won't work.
|
|
if rev > 0:
|
|
start = log.length(rev - 1) + log.start(rev - 1)
|
|
else:
|
|
start = 0
|
|
truncate(ui, repo.svfs, log.datafile, start, dryrun, backupprefix)
|
|
truncate(ui, repo.svfs, log.indexfile, rev * 64, dryrun, backupprefix)
|
|
if dryrun:
|
|
ui.write(_("re-run with --no-dryrun to fix.\n"))
|
|
else:
|
|
ui.write(_("fix completed. re-run to check more revisions.\n"))
|
|
repo.destroyed()
|