mirror of
https://github.com/python/cpython.git
synced 2024-12-06 00:05:32 +08:00
e7951976ad
which shoulda coulda woulda oughta been PyErr_SetString().
252 lines
5.8 KiB
C
252 lines
5.8 KiB
C
/***********************************************************
|
|
Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam,
|
|
The Netherlands.
|
|
|
|
All Rights Reserved
|
|
|
|
Permission to use, copy, modify, and distribute this software and its
|
|
documentation for any purpose and without fee is hereby granted,
|
|
provided that the above copyright notice appear in all copies and that
|
|
both that copyright notice and this permission notice appear in
|
|
supporting documentation, and that the names of Stichting Mathematisch
|
|
Centrum or CWI or Corporation for National Research Initiatives or
|
|
CNRI not be used in advertising or publicity pertaining to
|
|
distribution of the software without specific, written prior
|
|
permission.
|
|
|
|
While CWI is the initial source for this software, a modified version
|
|
is made available by the Corporation for National Research Initiatives
|
|
(CNRI) at the Internet address ftp://ftp.python.org.
|
|
|
|
STICHTING MATHEMATISCH CENTRUM AND CNRI DISCLAIM ALL WARRANTIES WITH
|
|
REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF
|
|
MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH
|
|
CENTRUM OR CNRI BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL
|
|
DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
|
|
PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
|
|
TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
******************************************************************/
|
|
|
|
/* dl module */
|
|
|
|
#include "Python.h"
|
|
|
|
#include <dlfcn.h>
|
|
|
|
#ifndef RTLD_LAZY
|
|
#define RTLD_LAZY 1
|
|
#endif
|
|
|
|
typedef ANY *PyUnivPtr;
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyUnivPtr *dl_handle;
|
|
} dlobject;
|
|
|
|
staticforward PyTypeObject Dltype;
|
|
|
|
static PyObject *Dlerror;
|
|
|
|
static PyObject *
|
|
newdlobject(handle)
|
|
PyUnivPtr *handle;
|
|
{
|
|
dlobject *xp;
|
|
xp = PyObject_NEW(dlobject, &Dltype);
|
|
if (xp == NULL)
|
|
return NULL;
|
|
xp->dl_handle = handle;
|
|
return (PyObject *)xp;
|
|
}
|
|
|
|
static void
|
|
dl_dealloc(xp)
|
|
dlobject *xp;
|
|
{
|
|
if (xp->dl_handle != NULL)
|
|
dlclose(xp->dl_handle);
|
|
PyMem_DEL(xp);
|
|
}
|
|
|
|
static PyObject *
|
|
dl_close(xp, args)
|
|
dlobject *xp;
|
|
PyObject *args;
|
|
{
|
|
if (!PyArg_Parse(args, ""))
|
|
return NULL;
|
|
if (xp->dl_handle != NULL) {
|
|
dlclose(xp->dl_handle);
|
|
xp->dl_handle = NULL;
|
|
}
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static PyObject *
|
|
dl_sym(xp, args)
|
|
dlobject *xp;
|
|
PyObject *args;
|
|
{
|
|
char *name;
|
|
PyUnivPtr *func;
|
|
if (!PyArg_Parse(args, "s", &name))
|
|
return NULL;
|
|
func = dlsym(xp->dl_handle, name);
|
|
if (func == NULL) {
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
return PyInt_FromLong((long)func);
|
|
}
|
|
|
|
static PyObject *
|
|
dl_call(xp, args)
|
|
dlobject *xp;
|
|
PyObject *args; /* (varargs) */
|
|
{
|
|
PyObject *name;
|
|
long (*func)();
|
|
long alist[10];
|
|
long res;
|
|
int i;
|
|
int n = PyTuple_Size(args);
|
|
if (n < 1) {
|
|
PyErr_SetString(PyExc_TypeError, "at least a name is needed");
|
|
return NULL;
|
|
}
|
|
name = PyTuple_GetItem(args, 0);
|
|
if (!PyString_Check(name)) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"function name must be a string");
|
|
return NULL;
|
|
}
|
|
func = dlsym(xp->dl_handle, PyString_AsString(name));
|
|
if (func == NULL) {
|
|
PyErr_SetString(PyExc_ValueError, dlerror());
|
|
return NULL;
|
|
}
|
|
if (n-1 > 10) {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"too many arguments (max 10)");
|
|
return NULL;
|
|
}
|
|
for (i = 1; i < n; i++) {
|
|
PyObject *v = PyTuple_GetItem(args, i);
|
|
if (PyInt_Check(v))
|
|
alist[i-1] = PyInt_AsLong(v);
|
|
else if (PyString_Check(v))
|
|
alist[i-1] = (long)PyString_AsString(v);
|
|
else if (v == Py_None)
|
|
alist[i-1] = (long) ((char *)NULL);
|
|
else {
|
|
PyErr_SetString(PyExc_TypeError,
|
|
"arguments must be int, string or None");
|
|
return NULL;
|
|
}
|
|
}
|
|
for (; i <= 10; i++)
|
|
alist[i-1] = 0;
|
|
res = (*func)(alist[0], alist[1], alist[2], alist[3], alist[4],
|
|
alist[5], alist[6], alist[7], alist[8], alist[9]);
|
|
return PyInt_FromLong(res);
|
|
}
|
|
|
|
static PyMethodDef dlobject_methods[] = {
|
|
{"call", (PyCFunction)dl_call, 1 /* varargs */},
|
|
{"sym", (PyCFunction)dl_sym},
|
|
{"close", (PyCFunction)dl_close},
|
|
{NULL, NULL} /* Sentinel */
|
|
};
|
|
|
|
static PyObject *
|
|
dl_getattr(xp, name)
|
|
dlobject *xp;
|
|
char *name;
|
|
{
|
|
return Py_FindMethod(dlobject_methods, (PyObject *)xp, name);
|
|
}
|
|
|
|
|
|
static PyTypeObject Dltype = {
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
|
0, /*ob_size*/
|
|
"dl", /*tp_name*/
|
|
sizeof(dlobject), /*tp_basicsize*/
|
|
0, /*tp_itemsize*/
|
|
/* methods */
|
|
(destructor)dl_dealloc, /*tp_dealloc*/
|
|
0, /*tp_print*/
|
|
(getattrfunc)dl_getattr,/*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
0, /*tp_compare*/
|
|
0, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
0, /*tp_as_mapping*/
|
|
0, /*tp_hash*/
|
|
};
|
|
|
|
static PyObject *
|
|
dl_open(self, args)
|
|
PyObject *self;
|
|
PyObject *args;
|
|
{
|
|
char *name;
|
|
int mode;
|
|
PyUnivPtr *handle;
|
|
if (PyArg_Parse(args, "z", &name))
|
|
mode = RTLD_LAZY;
|
|
else {
|
|
PyErr_Clear();
|
|
if (!PyArg_Parse(args, "(zi)", &name, &mode))
|
|
return NULL;
|
|
#ifndef RTLD_NOW
|
|
if (mode != RTLD_LAZY) {
|
|
PyErr_SetString(PyExc_ValueError, "mode must be 1");
|
|
return NULL;
|
|
}
|
|
#endif
|
|
}
|
|
handle = dlopen(name, mode);
|
|
if (handle == NULL) {
|
|
PyErr_SetString(Dlerror, dlerror());
|
|
return NULL;
|
|
}
|
|
return newdlobject(handle);
|
|
}
|
|
|
|
static PyMethodDef dl_methods[] = {
|
|
{"open", dl_open},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
void
|
|
initdl()
|
|
{
|
|
PyObject *m, *d, *x;
|
|
|
|
if (sizeof(int) != sizeof(long) ||
|
|
sizeof(long) != sizeof(char *)) {
|
|
PyErr_SetString(PyExc_SystemError,
|
|
"module dl requires sizeof(int) == sizeof(long) == sizeof(char*)");
|
|
return;
|
|
}
|
|
|
|
/* Create the module and add the functions */
|
|
m = Py_InitModule("dl", dl_methods);
|
|
|
|
/* Add some symbolic constants to the module */
|
|
d = PyModule_GetDict(m);
|
|
Dlerror = x = PyErr_NewException("dl.error", NULL, NULL);
|
|
PyDict_SetItemString(d, "error", x);
|
|
x = PyInt_FromLong((long)RTLD_LAZY);
|
|
PyDict_SetItemString(d, "RTLD_LAZY", x);
|
|
#ifdef RTLD_NOW
|
|
x = PyInt_FromLong((long)RTLD_NOW);
|
|
PyDict_SetItemString(d, "RTLD_NOW", x);
|
|
#endif
|
|
}
|