Mercurial > hg
view contrib/fuzz/revlog_corpus.py @ 43859:8766728dbce6
fuzz: add support for fuzzing under either Python 2 or 3
This was more of a hairball than I hoped, but it appears to work. The hg-py3
branch of my oss-fuzz fork on github has the remaining changes to switch us to
Python 3, but we may as well retain Python 2 fuzzing support for at least a
little while.
Differential Revision: https://phab.mercurial-scm.org/D7592
author | Augie Fackler <augie@google.com> |
---|---|
date | Mon, 09 Dec 2019 22:20:35 -0500 |
parents | ba84a1ae4ae5 |
children | 6000f5b25c9b |
line wrap: on
line source
from __future__ import absolute_import 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' ) with zipfile.ZipFile(args.out[0], "w", zipfile.ZIP_STORED) as zf: if os.path.exists(changelog): with open(changelog, 'rb') as f: zf.writestr("00changelog.i", f.read()) if os.path.exists(contributing): with open(contributing, 'rb') as f: zf.writestr("contributing.i", f.read())