Mercurial > public > mercurial-scm > hg
diff contrib/fuzz/revlog_corpus.py @ 41014:c06f0ef9a5ba
fuzz: new fuzzer for revlog's parse_index2 method
Differential Revision: https://phab.mercurial-scm.org/D5462
author | Augie Fackler <augie@google.com> |
---|---|
date | Wed, 19 Dec 2018 20:26:53 -0500 |
parents | |
children | 2372284d9457 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/contrib/fuzz/revlog_corpus.py Wed Dec 19 20:26:53 2018 -0500 @@ -0,0 +1,28 @@ +from __future__ import absolute_import, print_function + +import argparse +import os +import zipfile + +ap = argparse.ArgumentParser() +ap.add_argument("out", metavar="some.zip", type=str, nargs=1) +args = ap.parse_args() + +reporoot = os.path.normpath(os.path.join(os.path.dirname(__file__), + '..', '..')) +# typically a standalone index +changelog = os.path.join(reporoot, '.hg', 'store', '00changelog.i') +# an inline revlog with only a few revisions +contributing = os.path.join( + reporoot, '.hg', 'store', 'data', 'contrib', 'fuzz', 'mpatch.cc.i') + +print(changelog, os.path.exists(changelog)) +print(contributing, os.path.exists(contributing)) + +with zipfile.ZipFile(args.out[0], "w", zipfile.ZIP_STORED) as zf: + if os.path.exists(changelog): + with open(changelog) as f: + zf.writestr("00changelog.i", f.read()) + if os.path.exists(contributing): + with open(contributing) as f: + zf.writestr("contributing.i", f.read())