mirror of
https://github.com/samba-team/samba.git
synced 2024-12-25 23:21:54 +03:00
702e85e48f
With the changes to make samba python code Py2/Py3 compatible there now are many instances where string content is decoded. Decoded string variables in Py2 are returned as the unicode type. Many Py2 c-module functions that take string arguments only check for the string type. However now it's quite possibe the content formally passed as a string argument is now passed as unicode after being decoded, such arguments are rejected and code can fail subtly. This only affects places where the type is directly checked e.g. via PyStr_Check etc. arguments that are parsed by ParseTuple* functions generally already accept both string and unicode (if 's', 'z', 's*' format specifiers are used) Signed-off-by: Noel Power <noel.power@suse.com> Reviewed-by: Alexander Bokovoy <ab@samba.org>
922 lines
22 KiB
C
922 lines
22 KiB
C
/*
|
|
Unix SMB/CIFS implementation.
|
|
Python bindings for tevent
|
|
|
|
Copyright (C) Jelmer Vernooij 2010
|
|
|
|
** NOTE! The following LGPL license applies to the tevent
|
|
** library. This does NOT imply that all of Samba is released
|
|
** under the LGPL
|
|
|
|
This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 3 of the License, or (at your option) any later version.
|
|
|
|
This library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with this library; if not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <Python.h>
|
|
#include "replace.h"
|
|
#include <tevent.h>
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
#define PyStr_Check PyUnicode_Check
|
|
#define PyStr_FromString PyUnicode_FromString
|
|
#define PyStr_AsUTF8 PyUnicode_AsUTF8
|
|
#define PyInt_FromLong PyLong_FromLong
|
|
#else
|
|
#define PyStr_Check PyString_Check
|
|
#define PyStr_FromString PyString_FromString
|
|
#define PyStr_AsUTF8 PyString_AsString
|
|
#endif
|
|
|
|
void init_tevent(void);
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
struct tevent_context *ev;
|
|
} TeventContext_Object;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
struct tevent_queue *queue;
|
|
} TeventQueue_Object;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
struct tevent_req *req;
|
|
} TeventReq_Object;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
struct tevent_signal *signal;
|
|
} TeventSignal_Object;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
struct tevent_timer *timer;
|
|
PyObject *callback;
|
|
} TeventTimer_Object;
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
struct tevent_fd *fd;
|
|
} TeventFd_Object;
|
|
|
|
static PyTypeObject TeventContext_Type;
|
|
static PyTypeObject TeventReq_Type;
|
|
static PyTypeObject TeventQueue_Type;
|
|
static PyTypeObject TeventSignal_Type;
|
|
static PyTypeObject TeventTimer_Type;
|
|
static PyTypeObject TeventFd_Type;
|
|
|
|
static int py_context_init(struct tevent_context *ev)
|
|
{
|
|
/* FIXME */
|
|
return 0;
|
|
}
|
|
|
|
static struct tevent_fd *py_add_fd(struct tevent_context *ev,
|
|
TALLOC_CTX *mem_ctx,
|
|
int fd, uint16_t flags,
|
|
tevent_fd_handler_t handler,
|
|
void *private_data,
|
|
const char *handler_name,
|
|
const char *location)
|
|
{
|
|
/* FIXME */
|
|
return NULL;
|
|
}
|
|
|
|
static void py_set_fd_close_fn(struct tevent_fd *fde,
|
|
tevent_fd_close_fn_t close_fn)
|
|
{
|
|
/* FIXME */
|
|
}
|
|
|
|
static uint16_t py_get_fd_flags(struct tevent_fd *fde)
|
|
{
|
|
/* FIXME */
|
|
return 0;
|
|
}
|
|
|
|
static void py_set_fd_flags(struct tevent_fd *fde, uint16_t flags)
|
|
{
|
|
/* FIXME */
|
|
}
|
|
|
|
/* timed_event functions */
|
|
static struct tevent_timer *py_add_timer(struct tevent_context *ev,
|
|
TALLOC_CTX *mem_ctx,
|
|
struct timeval next_event,
|
|
tevent_timer_handler_t handler,
|
|
void *private_data,
|
|
const char *handler_name,
|
|
const char *location)
|
|
{
|
|
/* FIXME */
|
|
return NULL;
|
|
}
|
|
|
|
/* immediate event functions */
|
|
static void py_schedule_immediate(struct tevent_immediate *im,
|
|
struct tevent_context *ev,
|
|
tevent_immediate_handler_t handler,
|
|
void *private_data,
|
|
const char *handler_name,
|
|
const char *location)
|
|
{
|
|
/* FIXME */
|
|
}
|
|
|
|
/* signal functions */
|
|
static struct tevent_signal *py_add_signal(struct tevent_context *ev,
|
|
TALLOC_CTX *mem_ctx,
|
|
int signum, int sa_flags,
|
|
tevent_signal_handler_t handler,
|
|
void *private_data,
|
|
const char *handler_name,
|
|
const char *location)
|
|
{
|
|
/* FIXME */
|
|
return NULL;
|
|
}
|
|
|
|
/* loop functions */
|
|
static int py_loop_once(struct tevent_context *ev, const char *location)
|
|
{
|
|
/* FIXME */
|
|
return 0;
|
|
}
|
|
|
|
static int py_loop_wait(struct tevent_context *ev, const char *location)
|
|
{
|
|
/* FIXME */
|
|
return 0;
|
|
}
|
|
|
|
const static struct tevent_ops py_tevent_ops = {
|
|
.context_init = py_context_init,
|
|
.add_fd = py_add_fd,
|
|
.set_fd_close_fn = py_set_fd_close_fn,
|
|
.get_fd_flags = py_get_fd_flags,
|
|
.set_fd_flags = py_set_fd_flags,
|
|
.add_timer = py_add_timer,
|
|
.schedule_immediate = py_schedule_immediate,
|
|
.add_signal = py_add_signal,
|
|
.loop_wait = py_loop_wait,
|
|
.loop_once = py_loop_once,
|
|
};
|
|
|
|
static PyObject *py_register_backend(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *name, *py_backend;
|
|
|
|
if (!PyArg_ParseTuple(args, "O", &py_backend))
|
|
return NULL;
|
|
|
|
name = PyObject_GetAttrString(py_backend, "name");
|
|
if (name == NULL) {
|
|
PyErr_SetNone(PyExc_AttributeError);
|
|
return NULL;
|
|
}
|
|
|
|
if (!(PyStr_Check(name) || PyUnicode_Check(name))) {
|
|
PyErr_SetNone(PyExc_TypeError);
|
|
Py_DECREF(name);
|
|
return NULL;
|
|
}
|
|
|
|
if (!tevent_register_backend(PyStr_AsUTF8(name), &py_tevent_ops)) { /* FIXME: What to do with backend */
|
|
PyErr_SetNone(PyExc_RuntimeError);
|
|
Py_DECREF(name);
|
|
return NULL;
|
|
}
|
|
|
|
Py_DECREF(name);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_context_reinitialise(TeventContext_Object *self)
|
|
{
|
|
int ret = tevent_re_initialise(self->ev);
|
|
if (ret != 0) {
|
|
PyErr_SetNone(PyExc_RuntimeError);
|
|
return NULL;
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_queue_stop(TeventQueue_Object *self)
|
|
{
|
|
tevent_queue_stop(self->queue);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_queue_start(TeventQueue_Object *self)
|
|
{
|
|
tevent_queue_start(self->queue);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static void py_queue_trigger(struct tevent_req *req, void *private_data)
|
|
{
|
|
PyObject *callback = private_data, *ret;
|
|
|
|
ret = PyObject_CallFunction(callback, discard_const_p(char, ""));
|
|
Py_XDECREF(ret);
|
|
}
|
|
|
|
static PyObject *py_tevent_queue_add(TeventQueue_Object *self, PyObject *args)
|
|
{
|
|
TeventContext_Object *py_ev;
|
|
TeventReq_Object *py_req;
|
|
PyObject *trigger;
|
|
bool ret;
|
|
|
|
if (!PyArg_ParseTuple(args, "O!O!O",
|
|
&TeventContext_Type, &py_ev,
|
|
&TeventReq_Type, &py_req,
|
|
&trigger))
|
|
return NULL;
|
|
|
|
Py_INCREF(trigger);
|
|
|
|
ret = tevent_queue_add(self->queue, py_ev->ev, py_req->req,
|
|
py_queue_trigger, trigger);
|
|
if (!ret) {
|
|
PyErr_SetString(PyExc_RuntimeError, "queue add failed");
|
|
Py_DECREF(trigger);
|
|
return NULL;
|
|
}
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyMethodDef py_tevent_queue_methods[] = {
|
|
{ "stop", (PyCFunction)py_tevent_queue_stop, METH_NOARGS,
|
|
"S.stop()" },
|
|
{ "start", (PyCFunction)py_tevent_queue_start, METH_NOARGS,
|
|
"S.start()" },
|
|
{ "add", (PyCFunction)py_tevent_queue_add, METH_VARARGS,
|
|
"S.add(ctx, req, trigger, baton)" },
|
|
{ NULL },
|
|
};
|
|
|
|
static PyObject *py_tevent_context_wakeup_send(PyObject *self, PyObject *args)
|
|
{
|
|
/* FIXME */
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_context_loop_wait(TeventContext_Object *self)
|
|
{
|
|
if (tevent_loop_wait(self->ev) != 0) {
|
|
PyErr_SetNone(PyExc_RuntimeError);
|
|
return NULL;
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_context_loop_once(TeventContext_Object *self)
|
|
{
|
|
if (tevent_loop_once(self->ev) != 0) {
|
|
PyErr_SetNone(PyExc_RuntimeError);
|
|
return NULL;
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static void py_tevent_signal_handler(struct tevent_context *ev,
|
|
struct tevent_signal *se,
|
|
int signum,
|
|
int count,
|
|
void *siginfo,
|
|
void *private_data)
|
|
{
|
|
PyObject *callback = (PyObject *)private_data, *ret;
|
|
|
|
ret = PyObject_CallFunction(callback, discard_const_p(char, "ii"), signum, count);
|
|
Py_XDECREF(ret);
|
|
}
|
|
|
|
static void py_tevent_signal_dealloc(TeventSignal_Object *self)
|
|
{
|
|
talloc_free(self->signal);
|
|
PyObject_Del(self);
|
|
}
|
|
|
|
static PyTypeObject TeventSignal_Type = {
|
|
.tp_name = "tevent.Signal",
|
|
.tp_basicsize = sizeof(TeventSignal_Object),
|
|
.tp_dealloc = (destructor)py_tevent_signal_dealloc,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
};
|
|
|
|
static PyObject *py_tevent_context_add_signal(TeventContext_Object *self, PyObject *args)
|
|
{
|
|
int signum, sa_flags;
|
|
PyObject *handler;
|
|
struct tevent_signal *sig;
|
|
TeventSignal_Object *ret;
|
|
|
|
if (!PyArg_ParseTuple(args, "iiO", &signum, &sa_flags, &handler))
|
|
return NULL;
|
|
|
|
Py_INCREF(handler);
|
|
sig = tevent_add_signal(self->ev, NULL, signum, sa_flags,
|
|
py_tevent_signal_handler, handler);
|
|
|
|
ret = PyObject_New(TeventSignal_Object, &TeventSignal_Type);
|
|
if (ret == NULL) {
|
|
PyErr_NoMemory();
|
|
talloc_free(sig);
|
|
return NULL;
|
|
}
|
|
|
|
ret->signal = sig;
|
|
|
|
return (PyObject *)ret;
|
|
}
|
|
|
|
static void py_timer_handler(struct tevent_context *ev,
|
|
struct tevent_timer *te,
|
|
struct timeval current_time,
|
|
void *private_data)
|
|
{
|
|
TeventTimer_Object *self = private_data;
|
|
PyObject *ret;
|
|
|
|
ret = PyObject_CallFunction(self->callback, discard_const_p(char, "l"), te);
|
|
if (ret == NULL) {
|
|
/* No Python stack to propagate exception to; just print traceback */
|
|
PyErr_PrintEx(0);
|
|
}
|
|
Py_XDECREF(ret);
|
|
}
|
|
|
|
static void py_tevent_timer_dealloc(TeventTimer_Object *self)
|
|
{
|
|
if (self->timer) {
|
|
talloc_free(self->timer);
|
|
}
|
|
Py_DECREF(self->callback);
|
|
PyObject_Del(self);
|
|
}
|
|
|
|
static int py_tevent_timer_traverse(TeventTimer_Object *self, visitproc visit, void *arg)
|
|
{
|
|
Py_VISIT(self->callback);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject* py_tevent_timer_get_active(TeventTimer_Object *self) {
|
|
return PyBool_FromLong(self->timer != NULL);
|
|
}
|
|
|
|
struct PyGetSetDef py_tevent_timer_getset[] = {
|
|
{
|
|
.name = discard_const_p(char, "active"),
|
|
.get = (getter)py_tevent_timer_get_active,
|
|
.doc = discard_const_p(char, "true if the timer is scheduled to run"),
|
|
},
|
|
{NULL},
|
|
};
|
|
|
|
static PyTypeObject TeventTimer_Type = {
|
|
.tp_name = "tevent.Timer",
|
|
.tp_basicsize = sizeof(TeventTimer_Object),
|
|
.tp_dealloc = (destructor)py_tevent_timer_dealloc,
|
|
.tp_traverse = (traverseproc)py_tevent_timer_traverse,
|
|
.tp_getset = py_tevent_timer_getset,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
};
|
|
|
|
struct TeventTimer_Object_ref {
|
|
TeventTimer_Object *obj;
|
|
};
|
|
|
|
static int TeventTimer_Object_ref_destructor(struct TeventTimer_Object_ref *ref)
|
|
{
|
|
ref->obj->timer = NULL;
|
|
Py_DECREF(ref->obj);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *py_tevent_context_add_timer_internal(TeventContext_Object *self,
|
|
struct timeval next_event,
|
|
PyObject *callback)
|
|
{
|
|
/* Ownership notes:
|
|
*
|
|
* There are 5 pieces in play; two tevent contexts and 3 Python objects:
|
|
* - The tevent timer
|
|
* - The tevent context
|
|
* - The Python context -- "self"
|
|
* - The Python timer (TeventTimer_Object) -- "ret"
|
|
* - The Python callback function -- "callback"
|
|
*
|
|
* We only use the Python context for getting the tevent context,
|
|
* afterwards it can be destroyed.
|
|
*
|
|
* The tevent context owns the tevent timer.
|
|
*
|
|
* The tevent timer holds a reference to the Python timer, so the Python
|
|
* timer must always outlive the tevent timer.
|
|
* The Python timer has a pointer to the tevent timer; a destructor is
|
|
* used to set this to NULL when the tevent timer is deallocated.
|
|
*
|
|
* The tevent timer can be deallocated in these cases:
|
|
* 1) when the context is destroyed
|
|
* 2) after the event fires
|
|
* Posssibly, API might be added to cancel (free the tevent timer).
|
|
*
|
|
* The Python timer holds a reference to the callback.
|
|
*/
|
|
TeventTimer_Object *ret;
|
|
struct TeventTimer_Object_ref *ref;
|
|
|
|
ret = PyObject_New(TeventTimer_Object, &TeventTimer_Type);
|
|
if (ret == NULL) {
|
|
PyErr_NoMemory();
|
|
return NULL;
|
|
}
|
|
Py_INCREF(callback);
|
|
ret->callback = callback;
|
|
ret->timer = tevent_add_timer(self->ev, NULL, next_event, py_timer_handler,
|
|
ret);
|
|
if (ret->timer == NULL) {
|
|
Py_DECREF(ret);
|
|
PyErr_SetString(PyExc_RuntimeError, "Could not initialize timer");
|
|
return NULL;
|
|
}
|
|
ref = talloc(ret->timer, struct TeventTimer_Object_ref);
|
|
if (ref == NULL) {
|
|
talloc_free(ret->timer);
|
|
Py_DECREF(ret);
|
|
PyErr_SetString(PyExc_RuntimeError, "Could not initialize timer");
|
|
return NULL;
|
|
}
|
|
Py_INCREF(ret);
|
|
ref->obj = ret;
|
|
|
|
talloc_set_destructor(ref, TeventTimer_Object_ref_destructor);
|
|
|
|
return (PyObject *)ret;
|
|
}
|
|
|
|
static PyObject *py_tevent_context_add_timer(TeventContext_Object *self, PyObject *args)
|
|
{
|
|
struct timeval next_event;
|
|
PyObject *callback;
|
|
if (!PyArg_ParseTuple(args, "lO", &next_event, &callback))
|
|
return NULL;
|
|
|
|
return py_tevent_context_add_timer_internal(self, next_event, callback);
|
|
}
|
|
|
|
static PyObject *py_tevent_context_add_timer_offset(TeventContext_Object *self, PyObject *args)
|
|
{
|
|
struct timeval next_event;
|
|
double offset;
|
|
int seconds;
|
|
PyObject *callback;
|
|
if (!PyArg_ParseTuple(args, "dO", &offset, &callback))
|
|
return NULL;
|
|
|
|
seconds = offset;
|
|
offset -= seconds;
|
|
next_event = tevent_timeval_current_ofs(seconds, (int)(offset*1000000));
|
|
return py_tevent_context_add_timer_internal(self, next_event, callback);
|
|
}
|
|
|
|
static void py_fd_handler(struct tevent_context *ev,
|
|
struct tevent_fd *fde,
|
|
uint16_t flags,
|
|
void *private_data)
|
|
{
|
|
PyObject *callback = private_data, *ret;
|
|
|
|
ret = PyObject_CallFunction(callback, discard_const_p(char, "i"), flags);
|
|
Py_XDECREF(ret);
|
|
}
|
|
|
|
static void py_tevent_fp_dealloc(TeventFd_Object *self)
|
|
{
|
|
talloc_free(self->fd);
|
|
PyObject_Del(self);
|
|
}
|
|
|
|
static PyTypeObject TeventFd_Type = {
|
|
.tp_name = "tevent.Fd",
|
|
.tp_basicsize = sizeof(TeventFd_Object),
|
|
.tp_dealloc = (destructor)py_tevent_fp_dealloc,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
};
|
|
|
|
static PyObject *py_tevent_context_add_fd(TeventContext_Object *self, PyObject *args)
|
|
{
|
|
int fd, flags;
|
|
PyObject *handler;
|
|
struct tevent_fd *tfd;
|
|
TeventFd_Object *ret;
|
|
|
|
if (!PyArg_ParseTuple(args, "iiO", &fd, &flags, &handler))
|
|
return NULL;
|
|
|
|
tfd = tevent_add_fd(self->ev, NULL, fd, flags, py_fd_handler, handler);
|
|
if (tfd == NULL) {
|
|
PyErr_SetNone(PyExc_RuntimeError);
|
|
return NULL;
|
|
}
|
|
|
|
ret = PyObject_New(TeventFd_Object, &TeventFd_Type);
|
|
if (ret == NULL) {
|
|
talloc_free(tfd);
|
|
return NULL;
|
|
}
|
|
ret->fd = tfd;
|
|
|
|
return (PyObject *)ret;
|
|
}
|
|
|
|
static PyMethodDef py_tevent_context_methods[] = {
|
|
{ "reinitialise", (PyCFunction)py_tevent_context_reinitialise, METH_NOARGS,
|
|
"S.reinitialise()" },
|
|
{ "wakeup_send", (PyCFunction)py_tevent_context_wakeup_send,
|
|
METH_VARARGS, "S.wakeup_send(wakeup_time) -> req" },
|
|
{ "loop_wait", (PyCFunction)py_tevent_context_loop_wait,
|
|
METH_NOARGS, "S.loop_wait()" },
|
|
{ "loop_once", (PyCFunction)py_tevent_context_loop_once,
|
|
METH_NOARGS, "S.loop_once()" },
|
|
{ "add_signal", (PyCFunction)py_tevent_context_add_signal,
|
|
METH_VARARGS, "S.add_signal(signum, sa_flags, handler) -> signal" },
|
|
{ "add_timer", (PyCFunction)py_tevent_context_add_timer,
|
|
METH_VARARGS, "S.add_timer(next_event, handler) -> timer" },
|
|
{ "add_timer_offset", (PyCFunction)py_tevent_context_add_timer_offset,
|
|
METH_VARARGS, "S.add_timer(offset_seconds, handler) -> timer" },
|
|
{ "add_fd", (PyCFunction)py_tevent_context_add_fd,
|
|
METH_VARARGS, "S.add_fd(fd, flags, handler) -> fd" },
|
|
{ NULL },
|
|
};
|
|
|
|
static PyObject *py_tevent_req_wakeup_recv(PyObject *self)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_req_received(PyObject *self)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_req_is_error(PyObject *self)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_req_poll(PyObject *self)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_req_is_in_progress(PyObject *self)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyGetSetDef py_tevent_req_getsetters[] = {
|
|
{
|
|
.name = discard_const_p(char, "in_progress"),
|
|
.get = (getter)py_tevent_req_is_in_progress,
|
|
.doc = discard_const_p(char, "Whether the request is in progress"),
|
|
},
|
|
{ NULL }
|
|
};
|
|
|
|
static PyObject *py_tevent_req_post(PyObject *self, PyObject *args)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_req_set_error(PyObject *self, PyObject *args)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_req_done(PyObject *self)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_req_notify_callback(PyObject *self)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_req_set_endtime(PyObject *self, PyObject *args)
|
|
{
|
|
/* FIXME */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_tevent_req_cancel(TeventReq_Object *self)
|
|
{
|
|
if (!tevent_req_cancel(self->req)) {
|
|
PyErr_SetNone(PyExc_RuntimeError);
|
|
return NULL;
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyMethodDef py_tevent_req_methods[] = {
|
|
{ "wakeup_recv", (PyCFunction)py_tevent_req_wakeup_recv, METH_NOARGS,
|
|
"Wakeup received" },
|
|
{ "received", (PyCFunction)py_tevent_req_received, METH_NOARGS,
|
|
"Receive finished" },
|
|
{ "is_error", (PyCFunction)py_tevent_req_is_error, METH_NOARGS,
|
|
"is_error() -> (error, state)" },
|
|
{ "poll", (PyCFunction)py_tevent_req_poll, METH_VARARGS,
|
|
"poll(ctx)" },
|
|
{ "post", (PyCFunction)py_tevent_req_post, METH_VARARGS,
|
|
"post(ctx) -> req" },
|
|
{ "set_error", (PyCFunction)py_tevent_req_set_error, METH_VARARGS,
|
|
"set_error(error)" },
|
|
{ "done", (PyCFunction)py_tevent_req_done, METH_NOARGS,
|
|
"done()" },
|
|
{ "notify_callback", (PyCFunction)py_tevent_req_notify_callback,
|
|
METH_NOARGS, "notify_callback()" },
|
|
{ "set_endtime", (PyCFunction)py_tevent_req_set_endtime,
|
|
METH_VARARGS, "set_endtime(ctx, endtime)" },
|
|
{ "cancel", (PyCFunction)py_tevent_req_cancel,
|
|
METH_NOARGS, "cancel()" },
|
|
{ NULL }
|
|
};
|
|
|
|
static void py_tevent_req_dealloc(TeventReq_Object *self)
|
|
{
|
|
talloc_free(self->req);
|
|
PyObject_DEL(self);
|
|
}
|
|
|
|
static PyTypeObject TeventReq_Type = {
|
|
.tp_name = "tevent.Request",
|
|
.tp_basicsize = sizeof(TeventReq_Object),
|
|
.tp_methods = py_tevent_req_methods,
|
|
.tp_dealloc = (destructor)py_tevent_req_dealloc,
|
|
.tp_getset = py_tevent_req_getsetters,
|
|
/* FIXME: .tp_new = py_tevent_req_new, */
|
|
};
|
|
|
|
static PyObject *py_tevent_queue_get_length(TeventQueue_Object *self)
|
|
{
|
|
return PyInt_FromLong(tevent_queue_length(self->queue));
|
|
}
|
|
|
|
static PyGetSetDef py_tevent_queue_getsetters[] = {
|
|
{
|
|
.name = discard_const_p(char, "length"),
|
|
.get = (getter)py_tevent_queue_get_length,
|
|
.doc = discard_const_p(char, "The number of elements in the queue."),
|
|
},
|
|
{ NULL },
|
|
};
|
|
|
|
static void py_tevent_queue_dealloc(TeventQueue_Object *self)
|
|
{
|
|
talloc_free(self->queue);
|
|
PyObject_Del(self);
|
|
}
|
|
|
|
static PyTypeObject TeventQueue_Type = {
|
|
.tp_name = "tevent.Queue",
|
|
.tp_basicsize = sizeof(TeventQueue_Object),
|
|
.tp_dealloc = (destructor)py_tevent_queue_dealloc,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
.tp_getset = py_tevent_queue_getsetters,
|
|
.tp_methods = py_tevent_queue_methods,
|
|
};
|
|
|
|
static PyObject *py_tevent_context_signal_support(PyObject *_self)
|
|
{
|
|
TeventContext_Object *self = (TeventContext_Object *)_self;
|
|
return PyBool_FromLong(tevent_signal_support(self->ev));
|
|
}
|
|
|
|
static PyGetSetDef py_tevent_context_getsetters[] = {
|
|
{
|
|
.name = discard_const_p(char, "signal_support"),
|
|
.get = (getter)py_tevent_context_signal_support,
|
|
.doc = discard_const_p(char, "if this platform and tevent context support signal handling"),
|
|
},
|
|
{ NULL }
|
|
};
|
|
|
|
static void py_tevent_context_dealloc(TeventContext_Object *self)
|
|
{
|
|
talloc_free(self->ev);
|
|
PyObject_Del(self);
|
|
}
|
|
|
|
static PyObject *py_tevent_context_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
|
|
{
|
|
const char * const kwnames[] = { "name", NULL };
|
|
char *name = NULL;
|
|
struct tevent_context *ev;
|
|
TeventContext_Object *ret;
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|s", discard_const_p(char *, kwnames), &name))
|
|
return NULL;
|
|
|
|
if (name == NULL) {
|
|
ev = tevent_context_init(NULL);
|
|
} else {
|
|
ev = tevent_context_init_byname(NULL, name);
|
|
}
|
|
|
|
if (ev == NULL) {
|
|
PyErr_SetNone(PyExc_RuntimeError);
|
|
return NULL;
|
|
}
|
|
|
|
ret = PyObject_New(TeventContext_Object, type);
|
|
if (ret == NULL) {
|
|
PyErr_NoMemory();
|
|
talloc_free(ev);
|
|
return NULL;
|
|
}
|
|
|
|
ret->ev = ev;
|
|
return (PyObject *)ret;
|
|
}
|
|
|
|
static PyTypeObject TeventContext_Type = {
|
|
.tp_name = "tevent.Context",
|
|
.tp_new = py_tevent_context_new,
|
|
.tp_basicsize = sizeof(TeventContext_Object),
|
|
.tp_dealloc = (destructor)py_tevent_context_dealloc,
|
|
.tp_methods = py_tevent_context_methods,
|
|
.tp_getset = py_tevent_context_getsetters,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
};
|
|
|
|
static PyObject *py_set_default_backend(PyObject *self, PyObject *args)
|
|
{
|
|
char *backend_name;
|
|
if (!PyArg_ParseTuple(args, "s", &backend_name))
|
|
return NULL;
|
|
|
|
tevent_set_default_backend(backend_name);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *py_backend_list(PyObject *self)
|
|
{
|
|
PyObject *ret = NULL;
|
|
PyObject *string = NULL;
|
|
int i, result;
|
|
const char **backends = NULL;
|
|
|
|
ret = PyList_New(0);
|
|
if (ret == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
backends = tevent_backend_list(NULL);
|
|
if (backends == NULL) {
|
|
PyErr_SetNone(PyExc_RuntimeError);
|
|
goto err;
|
|
}
|
|
for (i = 0; backends[i]; i++) {
|
|
string = PyStr_FromString(backends[i]);
|
|
if (!string) {
|
|
goto err;
|
|
}
|
|
result = PyList_Append(ret, string);
|
|
if (result) {
|
|
goto err;
|
|
}
|
|
Py_DECREF(string);
|
|
string = NULL;
|
|
}
|
|
|
|
talloc_free(backends);
|
|
|
|
return ret;
|
|
|
|
err:
|
|
Py_XDECREF(ret);
|
|
Py_XDECREF(string);
|
|
talloc_free(backends);
|
|
return NULL;
|
|
}
|
|
|
|
static PyMethodDef tevent_methods[] = {
|
|
{ "register_backend", (PyCFunction)py_register_backend, METH_VARARGS,
|
|
"register_backend(backend)" },
|
|
{ "set_default_backend", (PyCFunction)py_set_default_backend,
|
|
METH_VARARGS, "set_default_backend(backend)" },
|
|
{ "backend_list", (PyCFunction)py_backend_list,
|
|
METH_NOARGS, "backend_list() -> list" },
|
|
{ NULL },
|
|
};
|
|
|
|
#define MODULE_DOC PyDoc_STR("Python wrapping of talloc-maintained objects.")
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
static struct PyModuleDef moduledef = {
|
|
PyModuleDef_HEAD_INIT,
|
|
.m_name = "_tevent",
|
|
.m_doc = MODULE_DOC,
|
|
.m_size = -1,
|
|
.m_methods = tevent_methods,
|
|
};
|
|
#endif
|
|
|
|
PyObject * module_init(void);
|
|
PyObject * module_init(void)
|
|
{
|
|
PyObject *m;
|
|
|
|
if (PyType_Ready(&TeventContext_Type) < 0)
|
|
return NULL;
|
|
|
|
if (PyType_Ready(&TeventQueue_Type) < 0)
|
|
return NULL;
|
|
|
|
if (PyType_Ready(&TeventReq_Type) < 0)
|
|
return NULL;
|
|
|
|
if (PyType_Ready(&TeventSignal_Type) < 0)
|
|
return NULL;
|
|
|
|
if (PyType_Ready(&TeventTimer_Type) < 0)
|
|
return NULL;
|
|
|
|
if (PyType_Ready(&TeventFd_Type) < 0)
|
|
return NULL;
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
m = PyModule_Create(&moduledef);
|
|
#else
|
|
m = Py_InitModule3("_tevent", tevent_methods, MODULE_DOC);
|
|
#endif
|
|
if (m == NULL)
|
|
return NULL;
|
|
|
|
Py_INCREF(&TeventContext_Type);
|
|
PyModule_AddObject(m, "Context", (PyObject *)&TeventContext_Type);
|
|
|
|
Py_INCREF(&TeventQueue_Type);
|
|
PyModule_AddObject(m, "Queue", (PyObject *)&TeventQueue_Type);
|
|
|
|
Py_INCREF(&TeventReq_Type);
|
|
PyModule_AddObject(m, "Request", (PyObject *)&TeventReq_Type);
|
|
|
|
Py_INCREF(&TeventSignal_Type);
|
|
PyModule_AddObject(m, "Signal", (PyObject *)&TeventSignal_Type);
|
|
|
|
Py_INCREF(&TeventTimer_Type);
|
|
PyModule_AddObject(m, "Timer", (PyObject *)&TeventTimer_Type);
|
|
|
|
Py_INCREF(&TeventFd_Type);
|
|
PyModule_AddObject(m, "Fd", (PyObject *)&TeventFd_Type);
|
|
|
|
PyModule_AddStringConstant(m, "__version__", PACKAGE_VERSION);
|
|
|
|
return m;
|
|
}
|
|
|
|
#if PY_MAJOR_VERSION >= 3
|
|
PyMODINIT_FUNC PyInit__tevent(void);
|
|
PyMODINIT_FUNC PyInit__tevent(void)
|
|
{
|
|
return module_init();
|
|
}
|
|
#else
|
|
void init_tevent(void);
|
|
void init_tevent(void)
|
|
{
|
|
module_init();
|
|
}
|
|
#endif
|