1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-22 13:34:15 +03:00
samba-mirror/lib/ldb/pyldb_util.c
Douglas Bagnall 1d6b472cdc pyldb: rename pyldb_Dn_AsDn() to pyldb_Dn_AS_DN()
Following the python/C convention for checking vs non-checking
convertors.

Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Gary Lockyer <gary@catalyst.net.nz>
2019-07-10 04:32:13 +00:00

143 lines
3.3 KiB
C

/*
Unix SMB/CIFS implementation.
Python interface to ldb - utility functions.
Copyright (C) 2007-2010 Jelmer Vernooij <jelmer@samba.org>
** NOTE! The following LGPL license applies to the ldb
** 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 "ldb.h"
#include "pyldb.h"
static PyObject *ldb_module = NULL;
/**
* Find out PyTypeObject in ldb module for a given typename
*/
static PyTypeObject * PyLdb_GetPyType(const char *typename)
{
PyTypeObject *type = NULL;
bool ok;
if (ldb_module == NULL) {
ldb_module = PyImport_ImportModule("ldb");
if (ldb_module == NULL) {
return NULL;
}
}
type = (PyTypeObject *)PyObject_GetAttrString(ldb_module, typename);
if (type == NULL) {
PyErr_Format(PyExc_NameError,
"Unable to find type %s in ldb module",
typename);
return NULL;
}
ok = PyType_Check(type);
if (! ok) {
PyErr_Format(PyExc_TypeError,
"Expected type ldb.%s, not %s",
typename, Py_TYPE(type)->tp_name);
Py_DECREF(type);
return NULL;
}
return type;
}
bool pyldb_check_type(PyObject *obj, const char *typename)
{
bool ok = false;
PyTypeObject *type = PyLdb_GetPyType(typename);
if (type != NULL) {
ok = PyObject_TypeCheck(obj, type);
Py_DECREF(type);
}
return ok;
}
/**
* Obtain a ldb DN from a Python object.
*
* @param mem_ctx Memory context
* @param object Python object
* @param ldb_ctx LDB context
* @return Whether or not the conversion succeeded
*/
bool pyldb_Object_AsDn(TALLOC_CTX *mem_ctx, PyObject *object,
struct ldb_context *ldb_ctx, struct ldb_dn **dn)
{
struct ldb_dn *odn;
PyTypeObject *PyLdb_Dn_Type;
if (ldb_ctx != NULL && (PyUnicode_Check(object))) {
odn = ldb_dn_new(mem_ctx, ldb_ctx, PyUnicode_AsUTF8(object));
*dn = odn;
return true;
}
if (ldb_ctx != NULL && PyBytes_Check(object)) {
odn = ldb_dn_new(mem_ctx, ldb_ctx, PyBytes_AsString(object));
*dn = odn;
return true;
}
PyLdb_Dn_Type = PyLdb_GetPyType("Dn");
if (PyLdb_Dn_Type == NULL) {
return false;
}
if (PyObject_TypeCheck(object, PyLdb_Dn_Type)) {
*dn = pyldb_Dn_AS_DN(object);
return true;
}
PyErr_SetString(PyExc_TypeError, "Expected DN");
return false;
}
PyObject *pyldb_Dn_FromDn(struct ldb_dn *dn)
{
PyLdbDnObject *py_ret;
PyTypeObject *PyLdb_Dn_Type;
if (dn == NULL) {
Py_RETURN_NONE;
}
PyLdb_Dn_Type = PyLdb_GetPyType("Dn");
if (PyLdb_Dn_Type == NULL) {
return NULL;
}
py_ret = (PyLdbDnObject *)PyLdb_Dn_Type->tp_alloc(PyLdb_Dn_Type, 0);
if (py_ret == NULL) {
PyErr_NoMemory();
return NULL;
}
py_ret->mem_ctx = talloc_new(NULL);
py_ret->dn = talloc_reference(py_ret->mem_ctx, dn);
return (PyObject *)py_ret;
}