2 Unix SMB/CIFS implementation.
6 Copyright (C) 2005,2006 Tim Potter <tpot@samba.org>
7 Copyright (C) 2006 Simo Sorce <idra@samba.org>
8 Copyright (C) 2007 Jelmer Vernooij <jelmer@samba.org>
10 ** NOTE! The following LGPL license applies to the ldb
11 ** library. This does NOT imply that all of Samba is released
14 This library is free software; you can redistribute it and/or
15 modify it under the terms of the GNU Lesser General Public
16 License as published by the Free Software Foundation; either
17 version 3 of the License, or (at your option) any later version.
19 This library is distributed in the hope that it will be useful,
20 but WITHOUT ANY WARRANTY; without even the implied warranty of
21 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22 Lesser General Public License for more details.
24 You should have received a copy of the GNU Lesser General Public
25 License along with this library; if not, see <http://www.gnu.org/licenses/>.
36 #include "ldb_errors.h"
37 #include "ldb_private.h"
39 typedef struct ldb_message ldb_msg;
40 typedef struct ldb_context ldb;
41 typedef struct ldb_dn ldb_dn;
42 typedef struct ldb_ldif ldb_ldif;
43 typedef struct ldb_message_element ldb_msg_element;
44 typedef int ldb_error;
50 %include "exception.i"
53 %constant int SCOPE_DEFAULT = LDB_SCOPE_DEFAULT;
54 %constant int SCOPE_BASE = LDB_SCOPE_BASE;
55 %constant int SCOPE_ONELEVEL = LDB_SCOPE_ONELEVEL;
56 %constant int SCOPE_SUBTREE = LDB_SCOPE_SUBTREE;
58 %constant int CHANGETYPE_NONE = LDB_CHANGETYPE_NONE;
59 %constant int CHANGETYPE_ADD = LDB_CHANGETYPE_ADD;
60 %constant int CHANGETYPE_DELETE = LDB_CHANGETYPE_DELETE;
61 %constant int CHANGETYPE_MODIFY = LDB_CHANGETYPE_MODIFY;
64 * Wrap struct ldb_context
67 /* The ldb functions will crash if a NULL ldb context is passed so
68 catch this before it happens. */
70 %typemap(check,noblock=1) struct ldb_context* {
72 SWIG_exception(SWIG_ValueError,
73 "ldb context must be non-NULL");
76 %typemap(check,noblock=1) ldb_msg * {
78 SWIG_exception(SWIG_ValueError,
79 "Message can not be None");
86 %typemap(in,noblock=1) struct ldb_val *INPUT (struct ldb_val temp) {
88 if (!PyString_Check($input)) {
89 PyErr_SetString(PyExc_TypeError, "string arg expected");
92 $1->length = PyString_Size($input);
93 $1->data = PyString_AsString($input);
96 %typemap(out,noblock=1) struct ldb_val {
97 $result = PyString_FromStringAndSize((const char *)$1.data, $1.length);
101 * Wrap struct ldb_result
104 %typemap(in,noblock=1,numinputs=0) struct ldb_result **OUT (struct ldb_result *temp_ldb_result) {
105 $1 = &temp_ldb_result;
109 %typemap(argout,noblock=1) struct ldb_result ** (int i) {
110 $result = PyList_New((*$1)->count);
111 for (i = 0; i < (*$1)->count; i++) {
112 PyList_SetItem($result, i,
113 SWIG_NewPointerObj((*$1)->msgs[i], SWIGTYPE_p_ldb_message, 0)
118 %typemap(in,noblock=1,numinputs=1) const char * const *attrs {
119 if ($input == Py_None) {
121 } else if (PySequence_Check($input)) {
123 $1 = talloc_array(NULL, char *, PySequence_Size($input)+1);
124 for(i = 0; i < PySequence_Size($input); i++)
125 $1[i] = PyString_AsString(PySequence_GetItem($input, i));
128 SWIG_exception(SWIG_TypeError, "expected sequence");
132 %typemap(freearg,noblock=1) const char * const *attrs {
137 %types(struct ldb_result *);
143 %rename(__str__) ldb_dn::get_linearized;
144 %rename(__cmp__) ldb_dn::compare;
145 %rename(__len__) ldb_dn::get_comp_num;
147 typedef struct ldb_dn {
149 ldb_dn(ldb *ldb_ctx, const char *str)
151 ldb_dn *ret = ldb_dn_new(ldb_ctx, ldb_ctx, str);
152 /* ldb_dn_new() doesn't accept NULL as memory context, so
153 we do it this way... */
154 talloc_steal(NULL, ret);
157 SWIG_exception(SWIG_ValueError,
158 "unable to parse dn string");
162 ~ldb_dn() { talloc_free($self); }
164 const char *get_casefold();
165 const char *get_linearized();
166 ldb_dn *parent() { return ldb_dn_get_parent(NULL, $self); }
167 int compare(ldb_dn *other);
171 bool check_special(const char *name);
173 bool add_child(ldb_dn *child);
174 bool add_base(ldb_dn *base);
175 const char *canonical_str() {
176 return ldb_dn_canonical_string($self, $self);
178 const char *canonical_ex_str() {
179 return ldb_dn_canonical_ex_string($self, $self);
182 ldb_dn *__add__(ldb_dn *other)
184 ldb_dn *ret = ldb_dn_copy(NULL, $self);
185 ldb_dn_add_child(ret, other);
189 /* FIXME: implement __getslice__ */
196 int ldb_dn_from_pyobject(TALLOC_CTX *mem_ctx, PyObject *object,
197 struct ldb_context *ldb_ctx, ldb_dn **dn)
201 if (ldb_ctx != NULL && PyString_Check(object)) {
202 *dn = ldb_dn_new(mem_ctx, ldb_ctx, PyString_AsString(object));
205 ret = SWIG_ConvertPtr(object, (void **)&odn, SWIGTYPE_p_ldb_dn,
206 SWIG_POINTER_EXCEPTION);
207 *dn = ldb_dn_copy(mem_ctx, odn);
211 ldb_msg_element *ldb_msg_element_from_pyobject(TALLOC_CTX *mem_ctx,
212 PyObject *set_obj, int flags,
213 const char *attr_name)
215 struct ldb_message_element *me = talloc(mem_ctx, struct ldb_message_element);
216 me->name = attr_name;
218 if (PyString_Check(set_obj)) {
220 me->values = talloc_array(me, struct ldb_val, me->num_values);
221 me->values[0].length = PyString_Size(set_obj);
222 me->values[0].data = (uint8_t *)talloc_strdup(me->values,
223 PyString_AsString(set_obj));
224 } else if (PySequence_Check(set_obj)) {
226 me->num_values = PySequence_Size(set_obj);
227 me->values = talloc_array(me, struct ldb_val, me->num_values);
228 for (i = 0; i < me->num_values; i++) {
229 PyObject *obj = PySequence_GetItem(set_obj, i);
230 me->values[i].length = PyString_Size(obj);
231 me->values[i].data = (uint8_t *)PyString_AsString(obj);
241 PyObject *ldb_msg_element_to_set(ldb_msg_element *me)
246 /* Python << 2.5 doesn't have PySet_New and PySet_Add. */
247 result = PyList_New(me->num_values);
249 for (i = 0; i < me->num_values; i++) {
250 PyList_SetItem(result, i,
251 PyString_FromStringAndSize((const char *)me->values[i].data,
252 me->values[i].length));
261 /* ldb_message_element */
262 %rename(__cmp__) ldb_message_element::compare;
263 %rename(MessageElement) ldb_msg_element;
264 typedef struct ldb_message_element {
267 PyObject *__iter__(void)
269 return PyObject_GetIter(ldb_msg_element_to_set($self));
272 PyObject *__set__(void)
274 return ldb_msg_element_to_set($self);
277 ldb_msg_element(PyObject *set_obj, int flags=0, const char *name = NULL)
279 return ldb_msg_element_from_pyobject(NULL, set_obj, flags, name);
284 return $self->num_values;
290 if (i < 0 || i >= $self->num_values)
293 return PyString_FromStringAndSize(
294 (const char *)$self->values[i].data,
295 $self->values[i].length);
298 ~ldb_msg_element() { talloc_free($self); }
299 int compare(ldb_msg_element *);
302 def __getitem__(self, i):
305 raise KeyError("no such value")
308 def __eq__(self, other):
309 if (isinstance(other, str) and
310 len(set(self)) == 1 and
311 set(self).pop() == other):
313 return self.__cmp__(other) == 0
320 %rename(Message) ldb_message;
322 %rename(__delitem__) ldb_message::remove_attr;
323 %typemap(out) ldb_msg_element * {
325 PyErr_SetString(PyExc_KeyError, "no such element");
327 $result = SWIG_NewPointerObj($1, SWIGTYPE_p_ldb_message_element, 0);
329 %rename(__getitem__) ldb_message::find_element;
330 //%typemap(out) ldb_msg_element *;
334 PyObject *ldb_msg_list_elements(ldb_msg *msg)
337 PyObject *obj = PyList_New(msg->num_elements);
338 for (i = 0; i < msg->num_elements; i++)
339 PyList_SetItem(obj, i, PyString_FromString(msg->elements[i].name));
346 typedef struct ldb_message {
350 ldb_msg(ldb_dn *dn = NULL) {
351 ldb_msg *ret = ldb_msg_new(NULL);
352 ret->dn = talloc_reference(ret, dn);
355 ~ldb_msg() { talloc_free($self); }
356 ldb_msg_element *find_element(const char *name);
359 void __setitem__(const char *attr_name, ldb_msg_element *val)
361 struct ldb_message_element *el;
363 ldb_msg_remove_attr($self, attr_name);
365 el = talloc($self, struct ldb_message_element);
366 el->name = talloc_strdup(el, attr_name);
367 el->num_values = val->num_values;
368 el->values = talloc_reference(el, val->values);
370 ldb_msg_add($self, el, val->flags);
373 void __setitem__(const char *attr_name, PyObject *val)
375 struct ldb_message_element *el = ldb_msg_element_from_pyobject(NULL,
377 talloc_steal($self, el);
378 ldb_msg_remove_attr($self, attr_name);
379 ldb_msg_add($self, el, el->flags);
382 unsigned int __len__() { return $self->num_elements; }
386 return ldb_msg_list_elements($self);
389 PyObject *__iter__(void)
391 return PyObject_GetIter(ldb_msg_list_elements($self));
394 void remove_attr(const char *name);
398 /* FIXME: Convert ldb_result to 3-tuple:
399 (msgs, refs, controls)
402 typedef struct ldb_ldif ldb_ldif;
406 static void py_ldb_debug(void *context, enum ldb_debug_level level, const char *fmt, va_list ap)
409 PyObject *fn = context;
411 vasprintf(&text, fmt, ap);
412 PyObject_CallFunction(fn, (char *)"(i,s)", level, text);
417 %typemap(in,numinputs=1,noblock=1) (void (*debug)(void *context, enum ldb_debug_level level, const char *fmt, va_list ap), void *context) {
419 /* FIXME: Should be decreased somewhere as well. Perhaps register a
420 destructor and tie it to the ldb context ? */
427 static PyObject *ldb_ldif_to_pyobject(ldb_ldif *ldif)
432 return Py_BuildValue((char *)"(iO)", ldif->changetype,
433 SWIG_NewPointerObj(ldif->msg, SWIGTYPE_p_ldb_message, 0));
443 PyObject *PyExc_LdbError;
447 LdbError = _ldb.LdbError
451 PyExc_LdbError = PyErr_NewException((char *)"_ldb.LdbError", NULL, NULL);
452 PyDict_SetItemString(d, "LdbError", PyExc_LdbError);
455 %ignore _LDB_ERRORS_H_;
457 %include "include/ldb_errors.h"
464 %typemap(out,noblock=1) ldb_error {
465 if ($1 != LDB_SUCCESS) {
466 PyErr_SetObject(PyExc_LdbError, Py_BuildValue((char *)"(i,s)", $1, ldb_strerror($1)));
472 %rename(Ldb) ldb_context;
474 %typemap(in,noblock=1) struct ldb_dn * {
475 if (ldb_dn_from_pyobject(NULL, $input, arg1, &$1) != 0) {
480 %typemap(freearg,noblock=1) struct ldb_dn * {
484 /* Top-level ldb operations */
485 typedef struct ldb_context {
487 ldb(void) { return ldb_init(NULL); }
489 ldb_error connect(const char *url, unsigned int flags = 0,
490 const char *options[] = NULL);
492 ~ldb() { talloc_free($self); }
493 ldb_error search(ldb_dn *base = NULL,
494 enum ldb_scope scope = LDB_SCOPE_DEFAULT,
495 const char *expression = NULL,
496 const char * const *attrs = NULL,
497 struct ldb_result **OUT);
498 ldb_error delete(ldb_dn *dn);
499 ldb_error rename(ldb_dn *olddn, ldb_dn *newdn);
500 ldb_error add(ldb_msg *add_msg);
501 ldb_error add(PyObject *py_msg)
504 int dict_pos, msg_pos;
505 PyObject *key, *value;
506 ldb_msg_element *msgel;
509 if (PyDict_Check(py_msg)) {
510 msg = ldb_msg_new(NULL);
511 msg->elements = talloc_zero_array(msg, struct ldb_message_element, PyDict_Size(py_msg));
512 msg_pos = dict_pos = 0;
513 while (PyDict_Next(py_msg, &dict_pos, &key, &value)) {
514 if (!strcmp(PyString_AsString(key), "dn")) {
515 if (ldb_dn_from_pyobject(msg, value, $self, &msg->dn) != 0) {
516 return LDB_ERR_OTHER;
519 msgel = ldb_msg_element_from_pyobject(msg->elements, value, 0, PyString_AsString(key));
521 SWIG_exception(SWIG_TypeError, "unable to import element");
522 return LDB_ERR_OTHER;
524 memcpy(&msg->elements[msg_pos], msgel, sizeof(*msgel));
529 if (msg->dn == NULL) {
530 SWIG_exception(SWIG_TypeError, "no dn set");
531 return LDB_ERR_OTHER;
534 msg->num_elements = msg_pos;
536 if (SWIG_ConvertPtr(py_msg, (void **)&msg, SWIGTYPE_p_ldb_message, 0) != 0)
537 return LDB_ERR_OTHER;
540 ret = ldb_add($self, msg);
546 return LDB_ERR_OTHER;
548 ldb_error modify(ldb_msg *message);
549 ldb_dn *get_config_basedn();
550 ldb_dn *get_root_basedn();
551 ldb_dn *get_schema_basedn();
552 ldb_dn *get_default_basedn();
553 const char *errstring();
554 void set_create_perms(unsigned int perms);
555 void set_modules_dir(const char *path);
556 ldb_error set_debug(void (*debug)(void *context, enum ldb_debug_level level,
557 const char *fmt, va_list ap),
559 ldb_error set_opaque(const char *name, void *value);
560 void *get_opaque(const char *name);
561 ldb_error transaction_start();
562 ldb_error transaction_commit();
563 ldb_error transaction_cancel();
566 %typemap(in,numinputs=0,noblock=1) struct ldb_result **result_as_bool (struct ldb_result *tmp) { $1 = &tmp; }
567 %typemap(argout,noblock=1) struct ldb_result **result_as_bool { $result = ((*$1)->count > 0)?Py_True:Py_False; }
568 %typemap(freearg,noblock=1) struct ldb_result **result_as_bool { talloc_free(*$1); }
569 ldb_error __contains__(ldb_dn *dn, struct ldb_result **result_as_bool)
571 return ldb_search($self, dn, LDB_SCOPE_BASE, NULL, NULL,
575 PyObject *parse_ldif(const char *s)
577 PyObject *list = PyList_New(0);
578 struct ldb_ldif *ldif;
579 while ((ldif = ldb_ldif_read_string($self, &s)) != NULL) {
580 PyList_Append(list, ldb_ldif_to_pyobject(ldif));
582 return PyObject_GetIter(list);
588 def __init__(self, url=None, flags=0, options=None):
589 _ldb.Ldb_swiginit(self,_ldb.new_Ldb())
591 self.connect(url, flags, options)
596 %typemap(in,noblock=1) struct ldb_dn *;
597 %typemap(freearg,noblock=1) struct ldb_dn *;
599 %nodefault ldb_message;
600 %nodefault ldb_context;
603 %rename(valid_attr_name) ldb_valid_attr_name;
604 int ldb_valid_attr_name(const char *s);
606 typedef unsigned long time_t;
609 static char *timestring(time_t t)
611 char *tresult = ldb_timestring(NULL, t);
612 char *result = strdup(tresult);
613 talloc_free(tresult);
618 %rename(string_to_time) ldb_string_to_time;
619 time_t ldb_string_to_time(const char *s);
621 %typemap(in,noblock=1) const struct ldb_module_ops * {
622 $1 = talloc_zero(talloc_autofree_context(), struct ldb_module_ops);
624 $1->name = (char *)PyObject_GetAttrString($input, (char *)"name");
627 %rename(register_module) ldb_register_module;
628 ldb_error ldb_register_module(const struct ldb_module_ops *);