2 * Copyright (c) 2016 Cisco and/or its affiliates.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at:
7 * http://www.apache.org/licenses/LICENSE-2.0
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
17 #include "../pneum/pneum.h"
18 #include <vppinfra/hash.h>
20 static PyObject *pneum_callback = NULL;
23 wrap_pneum_callback (unsigned char * data, int len)
25 PyGILState_STATE gstate;
26 PyObject *result;//, *arglist;
28 gstate = PyGILState_Ensure();
30 /* Time to call the callback */
31 #if PY_VERSION_HEX >= 0x03000000
32 result = PyObject_CallFunction(pneum_callback, "y#", data, len);
34 result = PyObject_CallFunction(pneum_callback, "s#", data, len);
41 PyGILState_Release(gstate);
45 wrap_connect (PyObject *self, PyObject *args)
47 char * name, * chroot_prefix = NULL;
49 PyObject * temp = NULL;
50 pneum_callback_t cb = NULL;
52 if (!PyArg_ParseTuple(args, "s|Os:wrap_connect",
53 &name, &temp, &chroot_prefix))
58 if (!PyCallable_Check(temp))
60 PyErr_SetString(PyExc_TypeError, "parameter must be callable");
64 Py_XINCREF(temp); /* Add a reference to new callback */
65 Py_XDECREF(pneum_callback); /* Dispose of previous callback */
66 pneum_callback = temp; /* Remember new callback */
67 cb = wrap_pneum_callback;
69 Py_BEGIN_ALLOW_THREADS
70 rv = pneum_connect(name, chroot_prefix, cb);
72 return PyLong_FromLong(rv);
76 wrap_disconnect (PyObject *self, PyObject *args)
79 Py_BEGIN_ALLOW_THREADS
80 rv = pneum_disconnect();
82 return PyLong_FromLong(rv);
85 wrap_write (PyObject *self, PyObject *args)
90 if (!PyArg_ParseTuple(args, "s#", &data, &len))
92 Py_BEGIN_ALLOW_THREADS
93 rv = pneum_write(data, len);
96 return PyLong_FromLong(rv);
100 wrap_read (PyObject *self, PyObject *args)
105 Py_BEGIN_ALLOW_THREADS
106 rv = pneum_read(&data, &len);
109 if (rv != 0) { Py_RETURN_NONE; }
110 #if PY_VERSION_HEX >= 0x03000000
111 PyObject *ret = Py_BuildValue("y#", data, len);
113 PyObject *ret = Py_BuildValue("s#", data, len);
115 if (!ret) { Py_RETURN_NONE; }
122 wrap_msg_table (PyObject *self, PyObject *args)
126 uword *h = pneum_msg_table_get_hash();
127 PyObject *ret = PyList_New(pneum_msg_table_size());
128 if (!ret) goto error;
129 hash_foreach_pair (hp, h,
131 PyObject *item = PyTuple_New(2);
132 if (!item) goto error;
133 rv = PyTuple_SetItem(item, 0, PyLong_FromLong((u32)hp->value[0]));
135 rv = PyTuple_SetItem(item, 1, PyString_FromString((char *)hp->key));
137 PyList_SetItem(ret, i, item);
144 /* TODO: Raise exception */
145 printf("msg_table failed");
149 static PyMethodDef vpp_api_Methods[] = {
150 {"connect", wrap_connect, METH_VARARGS, "Connect to the VPP API."},
151 {"disconnect", wrap_disconnect, METH_VARARGS, "Disconnect from the VPP API."},
152 {"write", wrap_write, METH_VARARGS, "Write data to the VPP API."},
153 {"read", wrap_read, METH_VARARGS, "Read data from the VPP API."},
154 {"msg_table", wrap_msg_table, METH_VARARGS, "Get API dictionary."},
155 {NULL, NULL, 0, NULL} /* Sentinel */
158 #if PY_VERSION_HEX >= 0x03000000
160 PyInit_vpp_api (void)
166 #if PY_VERSION_HEX >= 0x03000000
167 static struct PyModuleDef vpp_api_module = {
168 #if PY_VERSION_HEX >= 0x03020000
169 PyModuleDef_HEAD_INIT,
172 PyObject_HEAD_INIT(NULL)
189 /* Ensure threading is initialised */
190 if (!PyEval_ThreadsInitialized()) {
191 PyEval_InitThreads();
194 #if PY_VERSION_HEX >= 0x03000000
195 return PyModule_Create(&vpp_api_module);
197 Py_InitModule((char *) "vpp_api", vpp_api_Methods);