2 Unix SMB/CIFS implementation.
4 Python interface to ldb - utility functions.
6 Copyright (C) 2007-2010 Jelmer Vernooij <jelmer@samba.org>
8 ** NOTE! The following LGPL license applies to the ldb
9 ** library. This does NOT imply that all of Samba is released
12 This library is free software; you can redistribute it and/or
13 modify it under the terms of the GNU Lesser General Public
14 License as published by the Free Software Foundation; either
15 version 3 of the License, or (at your option) any later version.
17 This library is distributed in the hope that it will be useful,
18 but WITHOUT ANY WARRANTY; without even the implied warranty of
19 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20 Lesser General Public License for more details.
22 You should have received a copy of the GNU Lesser General Public
23 License along with this library; if not, see <http://www.gnu.org/licenses/>.
26 #include "lib/replace/system/python.h"
30 static PyObject
*ldb_module
= NULL
;
33 * Find out PyTypeObject in ldb module for a given typename
35 static PyTypeObject
* PyLdb_GetPyType(const char *typename
)
37 PyTypeObject
*type
= NULL
;
40 if (ldb_module
== NULL
) {
41 ldb_module
= PyImport_ImportModule("ldb");
42 if (ldb_module
== NULL
) {
47 type
= (PyTypeObject
*)PyObject_GetAttrString(ldb_module
, typename
);
51 PyErr_Format(PyExc_NameError
,
52 "Unable to find type %s in ldb module",
57 ok
= PyType_Check(type
);
59 PyErr_Format(PyExc_TypeError
,
60 "Expected type ldb.%s, not %s",
61 typename
, Py_TYPE(type
)->tp_name
);
69 bool pyldb_check_type(PyObject
*obj
, const char *typename
)
72 PyTypeObject
*type
= PyLdb_GetPyType(typename
);
74 ok
= PyObject_TypeCheck(obj
, type
);
81 * Obtain a ldb DN from a Python object.
83 * @param mem_ctx Memory context
84 * @param object Python object
85 * @param ldb_ctx LDB context
86 * @return Whether or not the conversion succeeded
88 bool pyldb_Object_AsDn(TALLOC_CTX
*mem_ctx
, PyObject
*object
,
89 struct ldb_context
*ldb_ctx
, struct ldb_dn
**dn
)
92 PyTypeObject
*PyLdb_Dn_Type
;
95 if (ldb_ctx
!= NULL
&& (PyUnicode_Check(object
))) {
96 const char *odn_str
= NULL
;
98 odn_str
= PyUnicode_AsUTF8(object
);
99 if (odn_str
== NULL
) {
103 odn
= ldb_dn_new(mem_ctx
, ldb_ctx
, odn_str
);
113 if (ldb_ctx
!= NULL
&& PyBytes_Check(object
)) {
114 const char *odn_str
= NULL
;
116 odn_str
= PyBytes_AsString(object
);
117 if (odn_str
== NULL
) {
121 odn
= ldb_dn_new(mem_ctx
, ldb_ctx
, odn_str
);
131 PyLdb_Dn_Type
= PyLdb_GetPyType("Dn");
132 if (PyLdb_Dn_Type
== NULL
) {
136 is_dn
= PyObject_TypeCheck(object
, PyLdb_Dn_Type
);
137 Py_DECREF(PyLdb_Dn_Type
);
139 *dn
= pyldb_Dn_AS_DN(object
);
143 PyErr_SetString(PyExc_TypeError
, "Expected DN");
147 PyObject
*pyldb_Dn_FromDn(struct ldb_dn
*dn
)
149 TALLOC_CTX
*mem_ctx
= NULL
;
150 struct ldb_dn
*dn_ref
= NULL
;
151 PyLdbDnObject
*py_ret
;
152 PyTypeObject
*PyLdb_Dn_Type
;
158 mem_ctx
= talloc_new(NULL
);
159 if (mem_ctx
== NULL
) {
160 return PyErr_NoMemory();
163 dn_ref
= talloc_reference(mem_ctx
, dn
);
164 if (dn_ref
== NULL
) {
165 talloc_free(mem_ctx
);
166 return PyErr_NoMemory();
169 PyLdb_Dn_Type
= PyLdb_GetPyType("Dn");
170 if (PyLdb_Dn_Type
== NULL
) {
171 talloc_free(mem_ctx
);
175 py_ret
= (PyLdbDnObject
*)PyLdb_Dn_Type
->tp_alloc(PyLdb_Dn_Type
, 0);
176 Py_DECREF(PyLdb_Dn_Type
);
177 if (py_ret
== NULL
) {
178 talloc_free(mem_ctx
);
182 py_ret
->mem_ctx
= mem_ctx
;
184 return (PyObject
*)py_ret
;