view contrib/fuzz/manifest.cc @ 46017:068307b638f4

statprof: fix off-by-one-line error in output

martinvonz claims they thought that this was intentional, but couldn't remember
the reasoning for it. I can't understand why it would be preferable, and I
didn't see anything in the comments in the file about why this would be useful,
so I'm hopefully not breaking anything by "fixing" it.

### Old output

```
| 100.0%  0.01s  dispatch.py:    run               line 43:  dispatch.run()
| 100.0%  0.01s  dispatch.py:    dispatch          line 115:  status = dispatch(req)
| 100.0%  0.01s  dispatch.py:    _runcatch         line 266:  ret = _runcatch(req) or 0
| 100.0%  0.01s  dispatch.py:    _callcatch        line 442:  return _callcatch(ui, _runc...
| 100.0%  0.01s  scmutil.py:     callcatch         line 451:  return scmutil.callcatch(ui...
| 100.0%  0.01s  dispatch.py:    _runcatchfunc     line 155:  return func()
| 100.0%  0.01s  dispatch.py:    _dispatch         line 432:  return _dispatch(req)
| 100.0%  0.01s  hg.py:          repository        line 1179:  repo = hg.repository(
| 100.0%  0.01s  hg.py:          _peerorrepo       line 221:  peer = _peerorrepo(
| 100.0%  0.01s  util.py:        __getattribute__  line 188:  obj = _peerlookup(path).ins...
| 100.0%  0.01s  localrepo.py:   makelocalrepositoryline 3227:  return makelocalrepository(...
| 100.0%  0.01s  localrepo.py:   __init__          line 683:  return cls(
| 100.0%  0.01s  util.py:        __getattribute__  line 1262:  self._extrafilterid = repov...
| 100.0%  0.01s  <frozen importlib._bootstrap_external>: exec_moduleline          245:  self.__spec__.loader.exec_m...
| 100.0%  0.01s  <frozen importlib._bootstrap_external>: get_codeline       779:
| 100.0%  0.01s  <frozen importlib._bootstrap_external>: path_statsline         868:
| 100.0%  0.01s  <frozen importlib._bootstrap_external>: _path_statline         1012:
```

### New output

```
| 100.0%  0.01s  hg:             <module>          line 43:  dispatch.run()
| 100.0%  0.01s  dispatch.py:    run               line 115:  status = dispatch(req)
| 100.0%  0.01s  dispatch.py:    dispatch          line 266:  ret = _runcatch(req) or 0
| 100.0%  0.01s  dispatch.py:    _runcatch         line 442:  return _callcatch(ui, _runc...
| 100.0%  0.01s  dispatch.py:    _callcatch        line 451:  return scmutil.callcatch(ui...
| 100.0%  0.01s  scmutil.py:     callcatch         line 155:  return func()
| 100.0%  0.01s  dispatch.py:    _runcatchfunc     line 432:  return _dispatch(req)
| 100.0%  0.01s  dispatch.py:    _dispatch         line 1179:  repo = hg.repository(
| 100.0%  0.01s  hg.py:          repository        line 221:  peer = _peerorrepo(
| 100.0%  0.01s  hg.py:          _peerorrepo       line 188:  obj = _peerlookup(path).ins...
| 100.0%  0.01s  localrepo.py:   instance          line 3227:  return makelocalrepository(...
| 100.0%  0.01s  localrepo.py:   makelocalrepositoryline 683:  return cls(
| 100.0%  0.01s  localrepo.py:   __init__          line 1262:  self._extrafilterid = repov...
| 100.0%  0.01s  util.py:        __getattribute__  line 245:  self.__spec__.loader.exec_m...
| 100.0%  0.01s  <frozen importlib._bootstrap_external>: exec_moduleline          779:
| 100.0%  0.01s  <frozen importlib._bootstrap_external>: get_codeline       868:
| 100.0%  0.01s  <frozen importlib._bootstrap_external>: path_statsline         1012:
| 100.0%  0.01s  <frozen importlib._bootstrap_external>: _path_statline         87:
```

Differential Revision: https://phab.mercurial-scm.org/D9510

#include <Python.h>
#include <assert.h>
#include <stdlib.h>
#include <unistd.h>

#include "FuzzedDataProvider.h"
#include "pyutil.h"

#include <string>

extern "C" {

static PYCODETYPE *code;

extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv)
{
	contrib::initpy(*argv[0]);
	code = (PYCODETYPE *)Py_CompileString(R"py(
try:
  lm = parsers.lazymanifest(mdata)
  # iterate the whole thing, which causes the code to fully parse
  # every line in the manifest
  for e, _, _ in lm.iterentries():
      # also exercise __getitem__ et al
      lm[e]
      e in lm
      (e + 'nope') in lm
  lm[b'xyzzy'] = (b'\0' * nlen, 'x')
  # do an insert, text should change
  assert lm.text() != mdata, "insert should change text and didn't: %r %r" % (lm.text(), mdata)
  cloned = lm.filtercopy(lambda x: x != 'xyzzy')
  assert cloned.text() == mdata, 'cloned text should equal mdata'
  cloned.diff(lm)
  del lm[b'xyzzy']
  cloned.diff(lm)
  # should be back to the same
  assert lm.text() == mdata, "delete should have restored text but didn't: %r %r" % (lm.text(), mdata)
except Exception as e:
  pass
  # uncomment this print if you're editing this Python code
  # to debug failures.
  # print e
)py",
	                                      "fuzzer", Py_file_input);
	return 0;
}

int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size)
{
	// Don't allow fuzzer inputs larger than 100k, since we'll just bog
	// down and not accomplish much.
	if (Size > 100000) {
		return 0;
	}
	FuzzedDataProvider provider(Data, Size);
	Py_ssize_t nodelength = provider.ConsumeBool() ? 20 : 32;
	PyObject *nlen = PyLong_FromSsize_t(nodelength);
	PyObject *mtext =
	    PyBytes_FromStringAndSize((const char *)Data, (Py_ssize_t)Size);
	PyObject *locals = PyDict_New();
	PyDict_SetItemString(locals, "mdata", mtext);
	PyDict_SetItemString(locals, "nlen", nlen);
	PyObject *res = PyEval_EvalCode(code, contrib::pyglobals(), locals);
	if (!res) {
		PyErr_Print();
	}
	Py_XDECREF(res);
	Py_DECREF(locals);
	Py_DECREF(mtext);
	return 0; // Non-zero return values are reserved for future use.
}
}