Commit 6da5172e authored by Kirill Smelkov's avatar Kirill Smelkov

bigfile/py: Teach storeblk() how to correctly propagate traceback on error

Previously we were limited to printing traceback starting down from just
storeblk() via explicit PyErr_PrintEx() - because pybuf was attached to
memory which could go away right after return from C function - so we
had to destroy that object for sure, not letting any traceback to hold a
reference to it.

This turned out to be too limiting and not showing full context where
errors happen.

So do the following trick: before returning, reattach pybuf to empty
region at NULL, and this way we don't need to worry about pybuf pointing
to memory which can go away -> thus instead of printing exception locally
- just return it the usual way it is done with C api in Python.

NOTE In contrast to PyMemoryViewObject, PyBufferObject definition is not
public, so to support Python2 - had to copy its definition to PY2 compat
header.

NOTE2 loadblk() is not touched - the loading is done from sighandler
context, which simulates as if it work in separate python thread, so it
is leaved as is for now.
parent d53271b9
...@@ -288,9 +288,9 @@ PyFunc(pyfileh_dirty_writeout, ...@@ -288,9 +288,9 @@ PyFunc(pyfileh_dirty_writeout,
err = fileh_dirty_writeout(pyfileh, flags); err = fileh_dirty_writeout(pyfileh, flags);
if (err) { if (err) {
// TODO check if pyerr already occured - just re-raise if (!PyErr_Occurred())
// XXX non-informative // XXX not very informative
PyErr_SetString(PyExc_RuntimeError, "fileh_dirty_writeout fail"); PyErr_SetString(PyExc_RuntimeError, "fileh_dirty_writeout fail");
return NULL; return NULL;
} }
...@@ -588,10 +588,11 @@ err: ...@@ -588,10 +588,11 @@ err:
static int pybigfile_storeblk(BigFile *file, blk_t blk, const void *buf) static int pybigfile_storeblk(BigFile *file, blk_t blk, const void *buf)
{ {
PyBigFile *pyfile = upcast(PyBigFile *, file); PyBigFile *pyfile = upcast(PyBigFile *, file);
PyObject *pybuf = NULL; PyObject *pybuf;
PyObject *storeret = NULL; PyObject *storeret = NULL;
PyGILState_STATE gstate; PyGILState_STATE gstate;
int err;
// XXX so far storeblk() is called only from py/user context (vs SIGSEGV // XXX so far storeblk() is called only from py/user context (vs SIGSEGV
// context) - no need to save/restore interpreter state. // context) - no need to save/restore interpreter state.
...@@ -607,37 +608,49 @@ static int pybigfile_storeblk(BigFile *file, blk_t blk, const void *buf) ...@@ -607,37 +608,49 @@ static int pybigfile_storeblk(BigFile *file, blk_t blk, const void *buf)
pybuf = PyMemoryView_FromMemory((void *)buf, file->blksize, PyBUF_READ); pybuf = PyMemoryView_FromMemory((void *)buf, file->blksize, PyBUF_READ);
#endif #endif
if (!pybuf) if (!pybuf)
goto err; goto out;
/* NOTE K = unsigned long long */ /* NOTE K = unsigned long long */
BUILD_ASSERT(sizeof(blk) == sizeof(unsigned long long)); BUILD_ASSERT(sizeof(blk) == sizeof(unsigned long long));
storeret = PyObject_CallMethod(pyfile, "storeblk", "KO", blk, pybuf); storeret = PyObject_CallMethod(pyfile, "storeblk", "KO", blk, pybuf);
if (!storeret)
goto err;
out:
/* we need to know only whether storeret != NULL, decref it now */ /* we need to know only whether storeret != NULL, decref it now */
Py_XDECREF(storeret); Py_XDECREF(storeret);
/* verify pybuf is not held - its memory can go away right after return /* repoint pybuf to empty region - the original memory attached to it can
* (if e.g. dirty page was not mapped in any vma) */ * go away right after we return (if e.g. dirty page was not mapped in any
if (pybuf) * vma), but we need pybuf to stay not corrupt - for printing full
BUG_ON(pybuf->ob_refcnt != 1); * traceback in case of storeblk() error. */
#if BIGFILE_USE_OLD_BUFFER
PyBufferObject *pybufo = (PyBufferObject *)pybuf;
pybufo->b_ptr = NULL;
pybufo->b_size = 0;
pybufo->b_offset = 0;
pybufo->b_hash = -1;
Py_CLEAR(pybufo->b_base);
#else
PyMemoryViewObject *pybufm = (PyMemoryViewObject *)pybuf;
pybufm->view.buf = NULL;
pybufm->view.len = 0;
Py_CLEAR(pybufm->view.obj);
#endif
Py_XDECREF(pybuf); /* verify that we actually tweaked pybuf ok */
Py_buffer pybuf_view;
err = PyObject_GetBuffer(pybuf, &pybuf_view, PyBUF_SIMPLE);
BUG_ON(err);
BUG_ON(pybuf_view.buf != NULL);
BUG_ON(pybuf_view.len != 0);
PyBuffer_Release(&pybuf_view);
/* done with pybuf */
Py_DECREF(pybuf);
out:
PyGILState_Release(gstate); PyGILState_Release(gstate);
/* storeblk() job done */ /* storeblk() job done */
return storeret ? 0 : -1; return storeret ? 0 : -1;
err:
/* error happened - dump traceback and return */
// XXX do we need this in storeblk? (but can't return with pybuf->ob_refcnt
// != 1) and on errors it is not so.
PyErr_PrintEx(0);
goto out;
} }
...@@ -814,6 +827,11 @@ _init_bigfile(void) ...@@ -814,6 +827,11 @@ _init_bigfile(void)
PyObject *m; PyObject *m;
int err; int err;
/* verify we copied struct PyBufferObject definition ok */
#if PY_MAJOR_VERSION < 3
BUG_ON(sizeof(PyBufferObject) != PyBuffer_Type.tp_basicsize);
#endif
/* setup virtmem gil hooks for python */ /* setup virtmem gil hooks for python */
virt_lock_hookgil(&py_virt_gil_hooks); virt_lock_hookgil(&py_virt_gil_hooks);
......
...@@ -37,4 +37,18 @@ PyMemoryView_FromMemory(char *mem, Py_ssize_t size, int flags) ...@@ -37,4 +37,18 @@ PyMemoryView_FromMemory(char *mem, Py_ssize_t size, int flags)
#endif #endif
/* structure of PyBufferObject (Objects/bufferobject.c) */
#if PY_MAJOR_VERSION < 3
typedef struct {
PyObject_HEAD
PyObject *b_base;
void *b_ptr;
Py_ssize_t b_size;
Py_ssize_t b_offset;
int b_readonly;
long b_hash;
} PyBufferObject;
#endif
#endif #endif
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment