1 /*
2 Unix SMB/CIFS implementation.
3
4 Python interface to ldb - utility functions.
5
6 Copyright (C) 2007-2010 Jelmer Vernooij <jelmer@samba.org>
7
8 ** NOTE! The following LGPL license applies to the ldb
9 ** library. This does NOT imply that all of Samba is released
10 ** under the LGPL
11
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.
16
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.
21
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/>.
24 */
25
26 #include <Python.h>
27 #include "ldb.h"
28 #include "pyldb.h"
29
30 static PyObject *ldb_module = NULL;
31
32 #if PY_MAJOR_VERSION >= 3
33 #define PyStr_Check PyUnicode_Check
34 #define PyStr_AsUTF8 PyUnicode_AsUTF8
35 #else
36 #define PyStr_Check PyString_Check
37 #define PyStr_AsUTF8 PyString_AsString
38 #endif
39
40 /**
41 * Find out PyTypeObject in ldb module for a given typename
42 */
PyLdb_GetPyType(const char * typename)43 static PyTypeObject * PyLdb_GetPyType(const char *typename)
44 {
45 PyObject *py_obj = NULL;
46
47 if (ldb_module == NULL) {
48 ldb_module = PyImport_ImportModule("ldb");
49 if (ldb_module == NULL) {
50 return NULL;
51 }
52 }
53
54 py_obj = PyObject_GetAttrString(ldb_module, typename);
55
56 return (PyTypeObject*)py_obj;
57 }
58
59 /**
60 * Obtain a ldb DN from a Python object.
61 *
62 * @param mem_ctx Memory context
63 * @param object Python object
64 * @param ldb_ctx LDB context
65 * @return Whether or not the conversion succeeded
66 */
pyldb_Object_AsDn(TALLOC_CTX * mem_ctx,PyObject * object,struct ldb_context * ldb_ctx,struct ldb_dn ** dn)67 bool pyldb_Object_AsDn(TALLOC_CTX *mem_ctx, PyObject *object,
68 struct ldb_context *ldb_ctx, struct ldb_dn **dn)
69 {
70 struct ldb_dn *odn;
71 PyTypeObject *PyLdb_Dn_Type;
72
73 if (ldb_ctx != NULL && (PyStr_Check(object) || PyUnicode_Check(object))) {
74 odn = ldb_dn_new(mem_ctx, ldb_ctx, PyStr_AsUTF8(object));
75 *dn = odn;
76 return true;
77 }
78
79 if (ldb_ctx != NULL && PyBytes_Check(object)) {
80 odn = ldb_dn_new(mem_ctx, ldb_ctx, PyBytes_AsString(object));
81 *dn = odn;
82 return true;
83 }
84
85 PyLdb_Dn_Type = PyLdb_GetPyType("Dn");
86 if (PyLdb_Dn_Type == NULL) {
87 return false;
88 }
89
90 if (PyObject_TypeCheck(object, PyLdb_Dn_Type)) {
91 *dn = pyldb_Dn_AsDn(object);
92 return true;
93 }
94
95 PyErr_SetString(PyExc_TypeError, "Expected DN");
96 return false;
97 }
98
pyldb_Dn_FromDn(struct ldb_dn * dn)99 PyObject *pyldb_Dn_FromDn(struct ldb_dn *dn)
100 {
101 PyLdbDnObject *py_ret;
102 PyTypeObject *PyLdb_Dn_Type;
103
104 if (dn == NULL) {
105 Py_RETURN_NONE;
106 }
107
108 PyLdb_Dn_Type = PyLdb_GetPyType("Dn");
109 if (PyLdb_Dn_Type == NULL) {
110 return NULL;
111 }
112
113 py_ret = (PyLdbDnObject *)PyLdb_Dn_Type->tp_alloc(PyLdb_Dn_Type, 0);
114 if (py_ret == NULL) {
115 PyErr_NoMemory();
116 return NULL;
117 }
118 py_ret->mem_ctx = talloc_new(NULL);
119 py_ret->dn = talloc_reference(py_ret->mem_ctx, dn);
120 return (PyObject *)py_ret;
121 }
122