2008-12-19 04:22:07 +03:00
/*
Unix SMB / CIFS implementation .
2009-04-23 03:21:47 +04:00
Python interface to ldb .
2008-12-19 04:22:07 +03:00
Copyright ( C ) 2005 , 2006 Tim Potter < tpot @ samba . org >
Copyright ( C ) 2006 Simo Sorce < idra @ samba . org >
2010-05-07 04:15:28 +04:00
Copyright ( C ) 2007 - 2010 Jelmer Vernooij < jelmer @ samba . org >
Copyright ( C ) 2009 - 2010 Matthias Dieter Wallnöfer
2011-06-22 07:49:37 +04:00
Copyright ( C ) 2009 - 2011 Andrew Tridgell
Copyright ( C ) 2009 - 2011 Andrew Bartlett
2008-12-19 04:22:07 +03:00
2010-12-29 20:56:13 +03:00
* * NOTE ! The following LGPL license applies to the ldb
* * library . This does NOT imply that all of Samba is released
* * under the LGPL
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
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/>.
*/
2010-12-12 23:40:03 +03:00
# include <Python.h>
2010-12-30 20:10:49 +03:00
# include <pytalloc.h>
2010-12-12 23:40:03 +03:00
# include "ldb_private.h"
2008-12-19 04:22:07 +03:00
# include "pyldb.h"
2010-12-29 20:09:09 +03:00
void initldb ( void ) ;
2010-12-30 21:39:14 +03:00
static PyObject * PyLdbMessage_FromMessage ( struct ldb_message * msg ) ;
static PyObject * PyExc_LdbError ;
2011-02-07 09:50:36 +03:00
staticforward PyTypeObject PyLdbControl ;
staticforward PyTypeObject PyLdbResult ;
2010-12-30 21:39:14 +03:00
staticforward PyTypeObject PyLdbMessage ;
staticforward PyTypeObject PyLdbModule ;
staticforward PyTypeObject PyLdbDn ;
staticforward PyTypeObject PyLdb ;
staticforward PyTypeObject PyLdbMessageElement ;
staticforward PyTypeObject PyLdbTree ;
static PyObject * PyLdb_FromLdbContext ( struct ldb_context * ldb_ctx ) ;
static PyObject * PyLdbModule_FromModule ( struct ldb_module * mod ) ;
static struct ldb_message_element * PyObject_AsMessageElement (
TALLOC_CTX * mem_ctx ,
PyObject * set_obj ,
2011-04-10 21:48:07 +04:00
unsigned int flags ,
2010-12-30 21:39:14 +03:00
const char * attr_name ) ;
2010-12-29 20:09:09 +03:00
2008-12-23 13:25:06 +03:00
/* There's no Py_ssize_t in 2.4, apparently */
# if PY_MAJOR_VERSION == 2 && PY_MINOR_VERSION < 5
typedef int Py_ssize_t ;
typedef inquiry lenfunc ;
2008-12-23 14:14:25 +03:00
typedef intargfunc ssizeargfunc ;
2008-12-23 13:25:06 +03:00
# endif
2009-01-08 14:20:20 +03:00
# ifndef Py_RETURN_NONE
# define Py_RETURN_NONE return Py_INCREF(Py_None), Py_None
# endif
2011-01-03 00:34:26 +03:00
# define SIGN(a) (((a) == 0)?0:((a) < 0?-1:1))
2011-02-07 09:50:36 +03:00
static PyObject * py_ldb_control_str ( PyLdbControlObject * self )
{
if ( self - > data ! = NULL ) {
char * control = ldb_control_to_string ( self - > mem_ctx , self - > data ) ;
if ( control = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
return PyString_FromString ( control ) ;
} else {
return PyString_FromFormat ( " ldb control " ) ;
}
}
static void py_ldb_control_dealloc ( PyLdbControlObject * self )
{
if ( self - > mem_ctx ! = NULL ) {
talloc_free ( self - > mem_ctx ) ;
}
2011-02-26 21:35:27 +03:00
self - > data = NULL ;
2011-02-07 09:50:36 +03:00
self - > ob_type - > tp_free ( self ) ;
}
static PyObject * py_ldb_control_get_oid ( PyLdbControlObject * self )
{
return PyString_FromString ( self - > data - > oid ) ;
}
static PyObject * py_ldb_control_get_critical ( PyLdbControlObject * self )
{
return PyBool_FromLong ( self - > data - > critical ) ;
}
static PyObject * py_ldb_control_set_critical ( PyLdbControlObject * self , PyObject * value , void * closure )
{
if ( PyObject_IsTrue ( value ) ) {
self - > data - > critical = true ;
} else {
self - > data - > critical = false ;
}
return 0 ;
}
static PyObject * py_ldb_control_new ( PyTypeObject * type , PyObject * args , PyObject * kwargs )
{
char * data = NULL ;
const char * const kwnames [ ] = { " ldb " , " data " , NULL } ;
struct ldb_control * parsed_controls ;
PyLdbControlObject * ret ;
PyObject * py_ldb ;
TALLOC_CTX * mem_ctx ;
struct ldb_context * ldb_ctx ;
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " Os " ,
discard_const_p ( char * , kwnames ) ,
& py_ldb , & data ) )
return NULL ;
mem_ctx = talloc_new ( NULL ) ;
if ( mem_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
ldb_ctx = PyLdb_AsLdbContext ( py_ldb ) ;
parsed_controls = ldb_parse_control_from_string ( ldb_ctx , mem_ctx , data ) ;
if ( ! parsed_controls ) {
talloc_free ( mem_ctx ) ;
PyErr_SetString ( PyExc_ValueError , " unable to parse control string " ) ;
return NULL ;
}
ret = PyObject_New ( PyLdbControlObject , type ) ;
if ( ret = = NULL ) {
PyErr_NoMemory ( ) ;
talloc_free ( mem_ctx ) ;
return NULL ;
}
ret - > mem_ctx = mem_ctx ;
2011-02-26 21:35:27 +03:00
ret - > data = talloc_move ( mem_ctx , & parsed_controls ) ;
2011-02-07 09:50:36 +03:00
if ( ret - > data = = NULL ) {
Py_DECREF ( ret ) ;
PyErr_NoMemory ( ) ;
talloc_free ( mem_ctx ) ;
return NULL ;
}
return ( PyObject * ) ret ;
}
static PyGetSetDef py_ldb_control_getset [ ] = {
{ discard_const_p ( char , " oid " ) , ( getter ) py_ldb_control_get_oid , NULL , NULL } ,
{ discard_const_p ( char , " critical " ) , ( getter ) py_ldb_control_get_critical , ( setter ) py_ldb_control_set_critical , NULL } ,
{ NULL }
} ;
static PyTypeObject PyLdbControl = {
. tp_name = " ldb.control " ,
. tp_dealloc = ( destructor ) py_ldb_control_dealloc ,
. tp_getattro = PyObject_GenericGetAttr ,
. tp_basicsize = sizeof ( PyLdbControlObject ) ,
. tp_getset = py_ldb_control_getset ,
. tp_doc = " LDB control. " ,
. tp_str = ( reprfunc ) py_ldb_control_str ,
. tp_new = py_ldb_control_new ,
. tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE ,
} ;
2009-07-18 18:11:21 +04:00
static void PyErr_SetLdbError ( PyObject * error , int ret , struct ldb_context * ldb_ctx )
{
if ( ret = = LDB_ERR_PYTHON_EXCEPTION )
return ; /* Python exception should already be set, just keep that */
PyErr_SetObject ( error ,
2010-11-15 06:30:54 +03:00
Py_BuildValue ( discard_const_p ( char , " (i,s) " ) , ret ,
ldb_ctx = = NULL ? ldb_strerror ( ret ) : ldb_errstring ( ldb_ctx ) ) ) ;
2009-07-18 18:11:21 +04:00
}
2011-01-11 08:45:39 +03:00
static PyObject * PyObject_FromLdbValue ( struct ldb_val * val )
2008-12-19 04:22:07 +03:00
{
2011-01-11 08:45:39 +03:00
return PyString_FromStringAndSize ( ( const char * ) val - > data , val - > length ) ;
2008-12-19 04:22:07 +03:00
}
2011-02-07 09:50:36 +03:00
/**
* Create a Python object from a ldb_result .
*
* @ param result LDB result to convert
* @ return Python object with converted result ( a list object )
*/
static PyObject * PyLdbControl_FromControl ( struct ldb_control * control )
{
TALLOC_CTX * ctl_ctx = talloc_new ( NULL ) ;
PyLdbControlObject * ctrl ;
if ( ctl_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
ctrl = ( PyLdbControlObject * ) PyLdbControl . tp_alloc ( & PyLdbControl , 0 ) ;
if ( ctrl = = NULL ) {
2011-02-26 21:35:27 +03:00
talloc_free ( ctl_ctx ) ;
2011-02-07 09:50:36 +03:00
PyErr_NoMemory ( ) ;
return NULL ;
}
ctrl - > mem_ctx = ctl_ctx ;
ctrl - > data = talloc_steal ( ctrl - > mem_ctx , control ) ;
if ( ctrl - > data = = NULL ) {
Py_DECREF ( ctrl ) ;
PyErr_NoMemory ( ) ;
return NULL ;
}
return ( PyObject * ) ctrl ;
}
2009-03-21 00:52:57 +03:00
/**
* Create a Python object from a ldb_result .
*
* @ param result LDB result to convert
* @ return Python object with converted result ( a list object )
*/
2008-12-19 04:22:07 +03:00
static PyObject * PyLdbResult_FromResult ( struct ldb_result * result )
{
2011-02-07 09:50:36 +03:00
PyLdbResultObject * ret ;
PyObject * list , * controls , * referals ;
2010-11-06 19:48:39 +03:00
Py_ssize_t i ;
2011-02-07 09:50:36 +03:00
2008-12-21 00:21:39 +03:00
if ( result = = NULL ) {
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2011-02-07 09:50:36 +03:00
}
ret = ( PyLdbResultObject * ) PyLdbResult . tp_alloc ( & PyLdbResult , 0 ) ;
if ( ret = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
list = PyList_New ( result - > count ) ;
if ( list = = NULL ) {
PyErr_NoMemory ( ) ;
Py_DECREF ( ret ) ;
return NULL ;
}
2008-12-19 04:22:07 +03:00
for ( i = 0 ; i < result - > count ; i + + ) {
2011-02-07 09:50:36 +03:00
PyList_SetItem ( list , i , PyLdbMessage_FromMessage ( result - > msgs [ i ] ) ) ;
2008-12-19 04:22:07 +03:00
}
2011-02-07 09:50:36 +03:00
ret - > mem_ctx = talloc_new ( NULL ) ;
if ( ret - > mem_ctx = = NULL ) {
Py_DECREF ( list ) ;
Py_DECREF ( ret ) ;
PyErr_NoMemory ( ) ;
return NULL ;
}
ret - > msgs = list ;
if ( result - > controls ) {
controls = PyList_New ( 1 ) ;
if ( controls = = NULL ) {
Py_DECREF ( ret ) ;
PyErr_NoMemory ( ) ;
return NULL ;
}
for ( i = 0 ; result - > controls [ i ] ; i + + ) {
PyObject * ctrl = ( PyObject * ) PyLdbControl_FromControl ( result - > controls [ i ] ) ;
if ( ctrl = = NULL ) {
Py_DECREF ( ret ) ;
Py_DECREF ( controls ) ;
PyErr_NoMemory ( ) ;
return NULL ;
}
PyList_SetItem ( controls , i , ctrl ) ;
}
} else {
/*
* No controls so we keep an empty list
*/
controls = PyList_New ( 0 ) ;
if ( controls = = NULL ) {
Py_DECREF ( ret ) ;
PyErr_NoMemory ( ) ;
return NULL ;
}
}
ret - > controls = controls ;
i = 0 ;
while ( result - > refs & & result - > refs [ i ] ) {
i + + ;
}
referals = PyList_New ( i ) ;
if ( referals = = NULL ) {
Py_DECREF ( ret ) ;
PyErr_NoMemory ( ) ;
return NULL ;
}
for ( i = 0 ; result - > refs & & result - > refs [ i ] ; i + + ) {
PyList_SetItem ( referals , i , PyString_FromString ( result - > refs [ i ] ) ) ;
}
ret - > referals = referals ;
return ( PyObject * ) ret ;
2008-12-19 04:22:07 +03:00
}
2009-03-21 00:52:57 +03:00
/**
* Create a LDB Result from a Python object .
* If conversion fails , NULL will be returned and a Python exception set .
*
* @ param mem_ctx Memory context in which to allocate the LDB Result
* @ param obj Python object to convert
* @ return a ldb_result , or NULL if the conversion failed
*/
static struct ldb_result * PyLdbResult_AsResult ( TALLOC_CTX * mem_ctx ,
2010-11-15 06:30:54 +03:00
PyObject * obj )
2008-12-19 04:22:07 +03:00
{
2008-12-19 19:08:35 +03:00
struct ldb_result * res ;
2010-11-06 19:48:39 +03:00
Py_ssize_t i ;
2009-05-31 18:19:11 +04:00
2008-12-19 19:08:35 +03:00
if ( obj = = Py_None )
return NULL ;
res = talloc_zero ( mem_ctx , struct ldb_result ) ;
res - > count = PyList_Size ( obj ) ;
res - > msgs = talloc_array ( res , struct ldb_message * , res - > count ) ;
for ( i = 0 ; i < res - > count ; i + + ) {
PyObject * item = PyList_GetItem ( obj , i ) ;
res - > msgs [ i ] = PyLdbMessage_AsMessage ( item ) ;
}
return res ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_dn_validate ( PyLdbDnObject * self )
{
2008-12-23 07:07:29 +03:00
return PyBool_FromLong ( ldb_dn_validate ( self - > dn ) ) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_dn_is_valid ( PyLdbDnObject * self )
{
2008-12-23 07:07:29 +03:00
return PyBool_FromLong ( ldb_dn_is_valid ( self - > dn ) ) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_dn_is_special ( PyLdbDnObject * self )
{
2008-12-23 07:07:29 +03:00
return PyBool_FromLong ( ldb_dn_is_special ( self - > dn ) ) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_dn_is_null ( PyLdbDnObject * self )
{
2008-12-23 07:07:29 +03:00
return PyBool_FromLong ( ldb_dn_is_null ( self - > dn ) ) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_dn_get_casefold ( PyLdbDnObject * self )
{
2008-12-23 07:07:29 +03:00
return PyString_FromString ( ldb_dn_get_casefold ( self - > dn ) ) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_dn_get_linearized ( PyLdbDnObject * self )
{
2008-12-23 07:07:29 +03:00
return PyString_FromString ( ldb_dn_get_linearized ( self - > dn ) ) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_dn_canonical_str ( PyLdbDnObject * self )
{
2008-12-23 07:07:29 +03:00
return PyString_FromString ( ldb_dn_canonical_string ( self - > dn , self - > dn ) ) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_dn_canonical_ex_str ( PyLdbDnObject * self )
{
2008-12-23 07:07:29 +03:00
return PyString_FromString ( ldb_dn_canonical_ex_string ( self - > dn , self - > dn ) ) ;
2008-12-19 04:22:07 +03:00
}
2011-06-22 07:49:37 +04:00
static PyObject * py_ldb_dn_extended_str ( PyLdbDnObject * self , PyObject * args , PyObject * kwargs )
{
const char * const kwnames [ ] = { " mode " , NULL } ;
int mode = 1 ;
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " |i " ,
discard_const_p ( char * , kwnames ) ,
& mode ) )
return NULL ;
return PyString_FromString ( ldb_dn_get_extended_linearized ( self - > dn , self - > dn , mode ) ) ;
}
2011-06-22 08:44:12 +04:00
static PyObject * py_ldb_dn_get_extended_component ( PyLdbDnObject * self , PyObject * args )
{
char * name ;
const struct ldb_val * val ;
if ( ! PyArg_ParseTuple ( args , " s " , & name ) )
return NULL ;
val = ldb_dn_get_extended_component ( self - > dn , name ) ;
if ( val = = NULL ) {
Py_RETURN_NONE ;
}
return PyString_FromStringAndSize ( ( const char * ) val - > data , val - > length ) ;
}
static PyObject * py_ldb_dn_set_extended_component ( PyLdbDnObject * self , PyObject * args )
{
char * name ;
PyObject * value ;
int err ;
if ( ! PyArg_ParseTuple ( args , " sO " , & name , & value ) )
return NULL ;
if ( value = = Py_None ) {
err = ldb_dn_set_extended_component ( self - > dn , name , NULL ) ;
} else {
struct ldb_val val ;
if ( ! PyString_Check ( value ) ) {
PyErr_SetString ( PyExc_TypeError , " Expected a string argument " ) ;
return NULL ;
}
val . data = ( uint8_t * ) PyString_AsString ( value ) ;
val . length = PyString_Size ( value ) ;
err = ldb_dn_set_extended_component ( self - > dn , name , & val ) ;
}
if ( err ! = LDB_SUCCESS ) {
PyErr_SetString ( PyExc_TypeError , " Failed to set extended component " ) ;
return NULL ;
}
Py_RETURN_NONE ;
}
2008-12-19 04:22:07 +03:00
static PyObject * py_ldb_dn_repr ( PyLdbDnObject * self )
{
2008-12-23 07:07:29 +03:00
return PyString_FromFormat ( " Dn(%s) " , PyObject_REPR(PyString_FromString(ldb_dn_get_linearized(self->dn)))) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_dn_check_special ( PyLdbDnObject * self , PyObject * args )
{
char * name ;
if ( ! PyArg_ParseTuple ( args , " s " , & name ) )
return NULL ;
2008-12-23 07:07:29 +03:00
return ldb_dn_check_special ( self - > dn , name ) ? Py_True : Py_False ;
2008-12-19 04:22:07 +03:00
}
2008-12-19 19:08:35 +03:00
2008-12-19 04:22:07 +03:00
static int py_ldb_dn_compare ( PyLdbDnObject * dn1 , PyLdbDnObject * dn2 )
{
2009-08-21 11:50:04 +04:00
int ret ;
ret = ldb_dn_compare ( dn1 - > dn , dn2 - > dn ) ;
if ( ret < 0 ) ret = - 1 ;
if ( ret > 0 ) ret = 1 ;
return ret ;
2008-12-19 04:22:07 +03:00
}
2008-12-19 16:41:44 +03:00
static PyObject * py_ldb_dn_get_parent ( PyLdbDnObject * self )
{
struct ldb_dn * dn = PyLdbDn_AsDn ( ( PyObject * ) self ) ;
2009-06-17 20:25:21 +04:00
struct ldb_dn * parent ;
2009-06-17 22:32:35 +04:00
PyLdbDnObject * py_ret ;
TALLOC_CTX * mem_ctx = talloc_new ( NULL ) ;
2009-06-17 20:25:21 +04:00
2009-06-17 22:32:35 +04:00
parent = ldb_dn_get_parent ( mem_ctx , dn ) ;
if ( parent = = NULL ) {
talloc_free ( mem_ctx ) ;
Py_RETURN_NONE ;
}
2009-06-17 20:25:21 +04:00
2009-06-17 22:32:35 +04:00
py_ret = ( PyLdbDnObject * ) PyLdbDn . tp_alloc ( & PyLdbDn , 0 ) ;
if ( py_ret = = NULL ) {
PyErr_NoMemory ( ) ;
talloc_free ( mem_ctx ) ;
return NULL ;
}
py_ret - > mem_ctx = mem_ctx ;
py_ret - > dn = parent ;
return ( PyObject * ) py_ret ;
2008-12-19 16:41:44 +03:00
}
2008-12-19 19:08:35 +03:00
# define dn_ldb_ctx(dn) ((struct ldb_context *)dn)
2008-12-19 16:41:44 +03:00
static PyObject * py_ldb_dn_add_child ( PyLdbDnObject * self , PyObject * args )
{
PyObject * py_other ;
2008-12-19 19:08:35 +03:00
struct ldb_dn * dn , * other ;
2008-12-21 00:21:39 +03:00
if ( ! PyArg_ParseTuple ( args , " O " , & py_other ) )
2008-12-19 16:41:44 +03:00
return NULL ;
2008-12-19 19:08:35 +03:00
dn = PyLdbDn_AsDn ( ( PyObject * ) self ) ;
if ( ! PyObject_AsDn ( NULL , py_other , dn_ldb_ctx ( dn ) , & other ) )
return NULL ;
2008-12-19 16:41:44 +03:00
2008-12-19 19:08:35 +03:00
return ldb_dn_add_child ( dn , other ) ? Py_True : Py_False ;
2008-12-19 16:41:44 +03:00
}
static PyObject * py_ldb_dn_add_base ( PyLdbDnObject * self , PyObject * args )
{
PyObject * py_other ;
2008-12-19 19:08:35 +03:00
struct ldb_dn * other , * dn ;
2008-12-21 00:21:39 +03:00
if ( ! PyArg_ParseTuple ( args , " O " , & py_other ) )
2008-12-19 16:41:44 +03:00
return NULL ;
2008-12-19 19:08:35 +03:00
dn = PyLdbDn_AsDn ( ( PyObject * ) self ) ;
2008-12-19 16:41:44 +03:00
2008-12-19 19:08:35 +03:00
if ( ! PyObject_AsDn ( NULL , py_other , dn_ldb_ctx ( dn ) , & other ) )
return NULL ;
return ldb_dn_add_base ( dn , other ) ? Py_True : Py_False ;
2008-12-19 16:41:44 +03:00
}
2011-07-14 07:17:49 +04:00
static PyObject * py_ldb_dn_is_child_of ( PyLdbDnObject * self , PyObject * args )
2011-05-24 01:30:15 +04:00
{
PyObject * py_base ;
struct ldb_dn * dn , * base ;
if ( ! PyArg_ParseTuple ( args , " O " , & py_base ) )
return NULL ;
dn = PyLdbDn_AsDn ( ( PyObject * ) self ) ;
if ( ! PyObject_AsDn ( NULL , py_base , dn_ldb_ctx ( dn ) , & base ) )
return NULL ;
2011-07-14 07:17:49 +04:00
return PyBool_FromLong ( ldb_dn_compare_base ( base , dn ) = = 0 ) ;
2011-05-24 01:30:15 +04:00
}
2008-12-19 04:22:07 +03:00
static PyMethodDef py_ldb_dn_methods [ ] = {
{ " validate " , ( PyCFunction ) py_ldb_dn_validate , METH_NOARGS ,
" S.validate() -> bool \n "
" Validate DN is correct. " } ,
{ " is_valid " , ( PyCFunction ) py_ldb_dn_is_valid , METH_NOARGS ,
" S.is_valid() -> bool \n " } ,
{ " is_special " , ( PyCFunction ) py_ldb_dn_is_special , METH_NOARGS ,
" S.is_special() -> bool \n "
" Check whether this is a special LDB DN. " } ,
{ " is_null " , ( PyCFunction ) py_ldb_dn_is_null , METH_NOARGS ,
" Check whether this is a null DN. " } ,
{ " get_casefold " , ( PyCFunction ) py_ldb_dn_get_casefold , METH_NOARGS ,
NULL } ,
{ " get_linearized " , ( PyCFunction ) py_ldb_dn_get_linearized , METH_NOARGS ,
NULL } ,
{ " canonical_str " , ( PyCFunction ) py_ldb_dn_canonical_str , METH_NOARGS ,
" S.canonical_str() -> string \n "
" Canonical version of this DN (like a posix path). " } ,
2011-07-14 07:17:49 +04:00
{ " is_child_of " , ( PyCFunction ) py_ldb_dn_is_child_of , METH_VARARGS ,
" S.is_child_of(basedn) -> int \n Returns True if this DN is a child of basedn \n " } ,
2008-12-19 04:22:07 +03:00
{ " canonical_ex_str " , ( PyCFunction ) py_ldb_dn_canonical_ex_str , METH_NOARGS ,
" S.canonical_ex_str() -> string \n "
" Canonical version of this DN (like a posix path, with terminating newline). " } ,
2011-06-22 07:49:37 +04:00
{ " extended_str " , ( PyCFunction ) py_ldb_dn_extended_str , METH_VARARGS | METH_KEYWORDS ,
" S.extended_str(mode=1) -> string \n "
" Extended version of this DN " } ,
2008-12-19 16:41:44 +03:00
{ " parent " , ( PyCFunction ) py_ldb_dn_get_parent , METH_NOARGS ,
" S.parent() -> dn \n "
" Get the parent for this DN. " } ,
{ " add_child " , ( PyCFunction ) py_ldb_dn_add_child , METH_VARARGS ,
" S.add_child(dn) -> None \n "
" Add a child DN to this DN. " } ,
{ " add_base " , ( PyCFunction ) py_ldb_dn_add_base , METH_VARARGS ,
2008-12-21 00:21:39 +03:00
" S.add_base(dn) -> None \n "
2008-12-19 16:41:44 +03:00
" Add a base DN to this DN. " } ,
2008-12-19 19:08:35 +03:00
{ " check_special " , ( PyCFunction ) py_ldb_dn_check_special , METH_VARARGS ,
2010-12-30 05:43:23 +03:00
" S.check_special(name) -> bool \n \n "
" Check if name is a special DN name " } ,
2011-06-22 08:44:12 +04:00
{ " get_extended_component " , ( PyCFunction ) py_ldb_dn_get_extended_component , METH_VARARGS ,
" S.get_extended_component(name) -> string \n \n "
" returns a DN extended component as a binary string " } ,
{ " set_extended_component " , ( PyCFunction ) py_ldb_dn_set_extended_component , METH_VARARGS ,
" S.set_extended_component(name, value) -> string \n \n "
" set a DN extended component as a binary string " } ,
2008-12-19 04:22:07 +03:00
{ NULL }
} ;
2008-12-19 16:41:44 +03:00
static Py_ssize_t py_ldb_dn_len ( PyLdbDnObject * self )
{
return ldb_dn_get_comp_num ( PyLdbDn_AsDn ( ( PyObject * ) self ) ) ;
}
static PyObject * py_ldb_dn_concat ( PyLdbDnObject * self , PyObject * py_other )
{
struct ldb_dn * dn = PyLdbDn_AsDn ( ( PyObject * ) self ) ,
2008-12-19 19:08:35 +03:00
* other ;
2009-06-17 22:23:54 +04:00
PyLdbDnObject * py_ret ;
2008-12-19 19:08:35 +03:00
if ( ! PyObject_AsDn ( NULL , py_other , NULL , & other ) )
return NULL ;
2009-06-17 22:23:54 +04:00
py_ret = ( PyLdbDnObject * ) PyLdbDn . tp_alloc ( & PyLdbDn , 0 ) ;
if ( py_ret = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
py_ret - > mem_ctx = talloc_new ( NULL ) ;
py_ret - > dn = ldb_dn_copy ( py_ret - > mem_ctx , dn ) ;
ldb_dn_add_child ( py_ret - > dn , other ) ;
return ( PyObject * ) py_ret ;
2008-12-19 16:41:44 +03:00
}
static PySequenceMethods py_ldb_dn_seq = {
. sq_length = ( lenfunc ) py_ldb_dn_len ,
. sq_concat = ( binaryfunc ) py_ldb_dn_concat ,
} ;
static PyObject * py_ldb_dn_new ( PyTypeObject * type , PyObject * args , PyObject * kwargs )
{
2008-12-19 19:08:35 +03:00
struct ldb_dn * ret ;
char * str ;
PyObject * py_ldb ;
struct ldb_context * ldb_ctx ;
2009-06-17 22:17:35 +04:00
TALLOC_CTX * mem_ctx ;
2008-12-23 07:07:29 +03:00
PyLdbDnObject * py_ret ;
2009-02-05 13:04:28 +03:00
const char * const kwnames [ ] = { " ldb " , " dn " , NULL } ;
2008-12-19 19:08:35 +03:00
2009-02-05 13:04:28 +03:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " Os " ,
discard_const_p ( char * , kwnames ) ,
& py_ldb , & str ) )
2008-12-19 19:08:35 +03:00
return NULL ;
ldb_ctx = PyLdb_AsLdbContext ( py_ldb ) ;
2009-05-31 18:19:11 +04:00
2009-06-17 22:17:35 +04:00
mem_ctx = talloc_new ( NULL ) ;
if ( mem_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
ret = ldb_dn_new ( mem_ctx , ldb_ctx , str ) ;
2011-03-04 12:44:22 +03:00
if ( ! ldb_dn_validate ( ret ) ) {
2009-06-17 22:17:35 +04:00
talloc_free ( mem_ctx ) ;
2008-12-19 16:41:44 +03:00
PyErr_SetString ( PyExc_ValueError , " unable to parse dn string " ) ;
return NULL ;
}
2008-12-23 07:07:29 +03:00
py_ret = ( PyLdbDnObject * ) type - > tp_alloc ( type , 0 ) ;
if ( ret = = NULL ) {
2009-06-17 22:17:35 +04:00
talloc_free ( mem_ctx ) ;
2008-12-23 07:07:29 +03:00
PyErr_NoMemory ( ) ;
return NULL ;
}
2009-06-17 22:17:35 +04:00
py_ret - > mem_ctx = mem_ctx ;
2008-12-23 07:07:29 +03:00
py_ret - > dn = ret ;
return ( PyObject * ) py_ret ;
2008-12-19 16:41:44 +03:00
}
2008-12-23 07:07:29 +03:00
static void py_ldb_dn_dealloc ( PyLdbDnObject * self )
{
talloc_free ( self - > mem_ctx ) ;
2010-12-29 17:58:12 +03:00
PyObject_Del ( self ) ;
2008-12-19 19:08:35 +03:00
}
2010-12-30 21:39:14 +03:00
static PyTypeObject PyLdbDn = {
2010-04-04 03:48:35 +04:00
. tp_name = " ldb.Dn " ,
2008-12-19 04:22:07 +03:00
. tp_methods = py_ldb_dn_methods ,
. tp_str = ( reprfunc ) py_ldb_dn_get_linearized ,
. tp_repr = ( reprfunc ) py_ldb_dn_repr ,
. tp_compare = ( cmpfunc ) py_ldb_dn_compare ,
2008-12-19 16:41:44 +03:00
. tp_as_sequence = & py_ldb_dn_seq ,
2008-12-19 04:22:07 +03:00
. tp_doc = " A LDB distinguished name. " ,
2008-12-19 16:41:44 +03:00
. tp_new = py_ldb_dn_new ,
2008-12-23 07:07:29 +03:00
. tp_dealloc = ( destructor ) py_ldb_dn_dealloc ,
2011-01-17 08:23:23 +03:00
. tp_basicsize = sizeof ( PyLdbDnObject ) ,
2008-12-21 05:08:14 +03:00
. tp_flags = Py_TPFLAGS_DEFAULT ,
2008-12-19 04:22:07 +03:00
} ;
/* Debug */
2008-12-19 19:08:35 +03:00
static void py_ldb_debug ( void * context , enum ldb_debug_level level , const char * fmt , va_list ap ) PRINTF_ATTRIBUTE ( 3 , 0 ) ;
2008-12-19 04:22:07 +03:00
static void py_ldb_debug ( void * context , enum ldb_debug_level level , const char * fmt , va_list ap )
{
2009-02-05 13:04:28 +03:00
PyObject * fn = ( PyObject * ) context ;
PyObject_CallFunction ( fn , discard_const_p ( char , " (i,O) " ) , level , PyString_FromFormatV ( fmt , ap ) ) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_set_debug ( PyLdbObject * self , PyObject * args )
{
PyObject * cb ;
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
if ( ! PyArg_ParseTuple ( args , " O " , & cb ) )
return NULL ;
Py_INCREF ( cb ) ;
/* FIXME: Where do we DECREF cb ? */
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ldb_set_debug ( self - > ldb_ctx , py_ldb_debug , cb ) , PyLdb_AsLdbContext ( self ) ) ;
2009-05-31 18:19:11 +04:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_set_create_perms ( PyTypeObject * self , PyObject * args )
{
unsigned int perms ;
if ( ! PyArg_ParseTuple ( args , " I " , & perms ) )
return NULL ;
ldb_set_create_perms ( PyLdb_AsLdbContext ( self ) , perms ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_set_modules_dir ( PyTypeObject * self , PyObject * args )
{
char * modules_dir ;
if ( ! PyArg_ParseTuple ( args , " s " , & modules_dir ) )
return NULL ;
ldb_set_modules_dir ( PyLdb_AsLdbContext ( self ) , modules_dir ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_transaction_start ( PyLdbObject * self )
{
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ldb_transaction_start ( PyLdb_AsLdbContext ( self ) ) , PyLdb_AsLdbContext ( self ) ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_transaction_commit ( PyLdbObject * self )
{
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ldb_transaction_commit ( PyLdb_AsLdbContext ( self ) ) , PyLdb_AsLdbContext ( self ) ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
2009-11-26 07:32:06 +03:00
static PyObject * py_ldb_transaction_prepare_commit ( PyLdbObject * self )
{
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ldb_transaction_prepare_commit ( PyLdb_AsLdbContext ( self ) ) , PyLdb_AsLdbContext ( self ) ) ;
Py_RETURN_NONE ;
}
2008-12-19 04:22:07 +03:00
static PyObject * py_ldb_transaction_cancel ( PyLdbObject * self )
{
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ldb_transaction_cancel ( PyLdb_AsLdbContext ( self ) ) , PyLdb_AsLdbContext ( self ) ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_setup_wellknown_attributes ( PyLdbObject * self )
{
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ldb_setup_wellknown_attributes ( PyLdb_AsLdbContext ( self ) ) , PyLdb_AsLdbContext ( self ) ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_repr ( PyLdbObject * self )
{
return PyString_FromFormat ( " <ldb connection> " ) ;
}
static PyObject * py_ldb_get_root_basedn ( PyLdbObject * self )
{
struct ldb_dn * dn = ldb_get_root_basedn ( PyLdb_AsLdbContext ( self ) ) ;
if ( dn = = NULL )
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
return PyLdbDn_FromDn ( dn ) ;
}
static PyObject * py_ldb_get_schema_basedn ( PyLdbObject * self )
{
struct ldb_dn * dn = ldb_get_schema_basedn ( PyLdb_AsLdbContext ( self ) ) ;
if ( dn = = NULL )
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
return PyLdbDn_FromDn ( dn ) ;
}
static PyObject * py_ldb_get_config_basedn ( PyLdbObject * self )
{
struct ldb_dn * dn = ldb_get_config_basedn ( PyLdb_AsLdbContext ( self ) ) ;
if ( dn = = NULL )
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
return PyLdbDn_FromDn ( dn ) ;
}
static PyObject * py_ldb_get_default_basedn ( PyLdbObject * self )
{
struct ldb_dn * dn = ldb_get_default_basedn ( PyLdb_AsLdbContext ( self ) ) ;
if ( dn = = NULL )
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
return PyLdbDn_FromDn ( dn ) ;
}
2009-03-20 02:33:43 +03:00
static const char * * PyList_AsStringList ( TALLOC_CTX * mem_ctx , PyObject * list ,
2010-11-15 06:30:54 +03:00
const char * paramname )
2008-12-19 16:41:44 +03:00
{
const char * * ret ;
2010-11-06 19:48:39 +03:00
Py_ssize_t i ;
2008-12-19 16:41:44 +03:00
if ( ! PyList_Check ( list ) ) {
2009-03-20 02:33:43 +03:00
PyErr_Format ( PyExc_TypeError , " %s is not a list " , paramname ) ;
2008-12-19 16:41:44 +03:00
return NULL ;
}
2008-12-19 19:08:35 +03:00
ret = talloc_array ( NULL , const char * , PyList_Size ( list ) + 1 ) ;
2010-12-30 22:11:59 +03:00
if ( ret = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2008-12-19 16:41:44 +03:00
for ( i = 0 ; i < PyList_Size ( list ) ; i + + ) {
PyObject * item = PyList_GetItem ( list , i ) ;
if ( ! PyString_Check ( item ) ) {
2009-03-20 02:33:43 +03:00
PyErr_Format ( PyExc_TypeError , " %s should be strings " , paramname ) ;
2008-12-19 16:41:44 +03:00
return NULL ;
}
2009-08-04 01:58:16 +04:00
ret [ i ] = talloc_strndup ( ret , PyString_AsString ( item ) ,
2010-11-15 06:30:54 +03:00
PyString_Size ( item ) ) ;
2008-12-19 16:41:44 +03:00
}
2008-12-19 19:08:35 +03:00
ret [ i ] = NULL ;
2008-12-19 16:41:44 +03:00
return ret ;
}
2008-12-21 01:00:23 +03:00
static int py_ldb_init ( PyLdbObject * self , PyObject * args , PyObject * kwargs )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
const char * const kwnames [ ] = { " url " , " flags " , " options " , NULL } ;
2008-12-19 16:41:44 +03:00
char * url = NULL ;
PyObject * py_options = Py_None ;
const char * * options ;
2011-04-10 21:48:07 +04:00
unsigned int flags = 0 ;
2008-12-19 16:41:44 +03:00
int ret ;
2008-12-21 01:00:23 +03:00
struct ldb_context * ldb ;
2008-12-19 16:41:44 +03:00
2011-04-10 21:48:07 +04:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " |zIO:Ldb.__init__ " ,
2009-02-05 13:04:28 +03:00
discard_const_p ( char * , kwnames ) ,
& url , & flags , & py_options ) )
2008-12-21 01:00:23 +03:00
return - 1 ;
2008-12-19 04:22:07 +03:00
2008-12-21 01:00:23 +03:00
ldb = PyLdb_AsLdbContext ( self ) ;
2008-12-19 04:22:07 +03:00
2008-12-19 16:41:44 +03:00
if ( py_options = = Py_None ) {
options = NULL ;
} else {
2009-03-20 02:33:43 +03:00
options = PyList_AsStringList ( ldb , py_options , " options " ) ;
2008-12-19 16:41:44 +03:00
if ( options = = NULL )
2008-12-21 01:00:23 +03:00
return - 1 ;
2008-12-19 16:41:44 +03:00
}
2009-05-31 18:19:11 +04:00
2008-12-19 16:41:44 +03:00
if ( url ! = NULL ) {
ret = ldb_connect ( ldb , url , flags , options ) ;
2008-12-21 01:00:23 +03:00
if ( ret ! = LDB_SUCCESS ) {
2009-04-23 03:21:47 +04:00
PyErr_SetLdbError ( PyExc_LdbError , ret , ldb ) ;
2008-12-21 01:00:23 +03:00
return - 1 ;
}
2008-12-19 16:41:44 +03:00
}
talloc_free ( options ) ;
2008-12-21 01:00:23 +03:00
return 0 ;
}
2008-12-19 16:41:44 +03:00
2008-12-21 01:00:23 +03:00
static PyObject * py_ldb_new ( PyTypeObject * type , PyObject * args , PyObject * kwargs )
{
2008-12-23 07:07:29 +03:00
PyLdbObject * ret ;
2008-12-21 01:00:23 +03:00
struct ldb_context * ldb ;
2009-06-17 22:43:25 +04:00
ret = ( PyLdbObject * ) type - > tp_alloc ( type , 0 ) ;
if ( ret = = NULL ) {
2008-12-21 01:00:23 +03:00
PyErr_NoMemory ( ) ;
return NULL ;
}
2009-06-17 22:43:25 +04:00
ret - > mem_ctx = talloc_new ( NULL ) ;
ldb = ldb_init ( ret - > mem_ctx , NULL ) ;
2008-12-21 01:00:23 +03:00
2009-06-17 22:43:25 +04:00
if ( ldb = = NULL ) {
2008-12-23 07:07:29 +03:00
PyErr_NoMemory ( ) ;
return NULL ;
}
2009-06-17 22:43:25 +04:00
2008-12-23 07:07:29 +03:00
ret - > ldb_ctx = ldb ;
return ( PyObject * ) ret ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_connect ( PyLdbObject * self , PyObject * args , PyObject * kwargs )
{
char * url ;
2011-04-10 21:48:07 +04:00
unsigned int flags = 0 ;
2008-12-19 04:22:07 +03:00
PyObject * py_options = Py_None ;
int ret ;
const char * * options ;
2009-02-05 13:04:28 +03:00
const char * const kwnames [ ] = { " url " , " flags " , " options " , NULL } ;
2011-04-10 21:48:07 +04:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " |zIO " ,
2009-02-05 13:04:28 +03:00
discard_const_p ( char * , kwnames ) ,
& url , & flags , & py_options ) )
2008-12-19 04:22:07 +03:00
return NULL ;
if ( py_options = = Py_None ) {
options = NULL ;
} else {
2009-03-20 02:33:43 +03:00
options = PyList_AsStringList ( NULL , py_options , " options " ) ;
2008-12-19 16:41:44 +03:00
if ( options = = NULL )
2008-12-19 04:22:07 +03:00
return NULL ;
}
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
ret = ldb_connect ( PyLdb_AsLdbContext ( self ) , url , flags , options ) ;
talloc_free ( options ) ;
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , PyLdb_AsLdbContext ( self ) ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
2011-06-14 10:39:49 +04:00
static PyObject * py_ldb_modify ( PyLdbObject * self , PyObject * args , PyObject * kwargs )
2008-12-19 04:22:07 +03:00
{
PyObject * py_msg ;
2009-11-20 14:22:38 +03:00
PyObject * py_controls = Py_None ;
struct ldb_context * ldb_ctx ;
struct ldb_request * req ;
struct ldb_control * * parsed_controls ;
struct ldb_message * msg ;
2008-12-19 04:22:07 +03:00
int ret ;
2010-06-20 20:06:54 +04:00
TALLOC_CTX * mem_ctx ;
2011-06-14 10:39:49 +04:00
bool validate = true ;
const char * const kwnames [ ] = { " message " , " controls " , " validate " , NULL } ;
2010-06-20 20:06:54 +04:00
2011-06-14 10:39:49 +04:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " O|Ob " ,
discard_const_p ( char * , kwnames ) ,
& py_msg , & py_controls , & validate ) )
2008-12-19 04:22:07 +03:00
return NULL ;
2010-06-20 20:06:54 +04:00
mem_ctx = talloc_new ( NULL ) ;
if ( mem_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2009-11-20 14:22:38 +03:00
ldb_ctx = PyLdb_AsLdbContext ( self ) ;
if ( py_controls = = Py_None ) {
parsed_controls = NULL ;
} else {
2010-06-20 20:06:54 +04:00
const char * * controls = PyList_AsStringList ( mem_ctx , py_controls , " controls " ) ;
parsed_controls = ldb_parse_control_strings ( ldb_ctx , mem_ctx , controls ) ;
2009-11-20 14:22:38 +03:00
talloc_free ( controls ) ;
}
2008-12-19 04:22:07 +03:00
if ( ! PyLdbMessage_Check ( py_msg ) ) {
PyErr_SetString ( PyExc_TypeError , " Expected Ldb Message " ) ;
2010-06-20 20:06:54 +04:00
talloc_free ( mem_ctx ) ;
2008-12-19 04:22:07 +03:00
return NULL ;
}
2009-11-20 14:22:38 +03:00
msg = PyLdbMessage_AsMessage ( py_msg ) ;
2011-06-14 10:39:49 +04:00
if ( validate ) {
ret = ldb_msg_sanity_check ( ldb_ctx , msg ) ;
if ( ret ! = LDB_SUCCESS ) {
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
talloc_free ( mem_ctx ) ;
return NULL ;
}
2010-06-20 20:06:54 +04:00
}
2009-11-20 14:22:38 +03:00
2010-06-20 20:06:54 +04:00
ret = ldb_build_mod_req ( & req , ldb_ctx , mem_ctx , msg , parsed_controls ,
NULL , ldb_op_default_callback , NULL ) ;
2009-11-20 14:22:38 +03:00
if ( ret ! = LDB_SUCCESS ) {
PyErr_SetString ( PyExc_TypeError , " failed to build request " ) ;
2010-06-20 20:06:54 +04:00
talloc_free ( mem_ctx ) ;
2009-11-20 14:22:38 +03:00
return NULL ;
}
2008-12-19 04:22:07 +03:00
2010-12-21 16:05:18 +03:00
/* do request and autostart a transaction */
2009-11-20 14:22:38 +03:00
/* Then let's LDB handle the message error in case of pb as they are meaningful */
2010-06-20 20:06:54 +04:00
ret = ldb_transaction_start ( ldb_ctx ) ;
if ( ret ! = LDB_SUCCESS ) {
talloc_free ( mem_ctx ) ;
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
2010-12-21 16:05:18 +03:00
}
2009-11-20 14:22:38 +03:00
2010-12-21 16:05:18 +03:00
ret = ldb_request ( ldb_ctx , req ) ;
if ( ret = = LDB_SUCCESS ) {
ret = ldb_wait ( req - > handle , LDB_WAIT_ALL ) ;
}
2009-11-20 14:22:38 +03:00
if ( ret = = LDB_SUCCESS ) {
2010-12-21 16:05:18 +03:00
ret = ldb_transaction_commit ( ldb_ctx ) ;
} else {
2009-11-20 14:22:38 +03:00
ldb_transaction_cancel ( ldb_ctx ) ;
if ( ldb_ctx - > err_string = = NULL ) {
/* no error string was setup by the backend */
ldb_asprintf_errstring ( ldb_ctx , " %s (%d) " , ldb_strerror ( ret ) , ret ) ;
}
}
2010-06-20 20:06:54 +04:00
talloc_free ( mem_ctx ) ;
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
2009-09-23 00:51:25 +04:00
2010-11-18 23:09:01 +03:00
/**
* Obtain a ldb message from a Python Dictionary object .
*
* @ param mem_ctx Memory context
* @ param py_obj Python Dictionary object
* @ param ldb_ctx LDB context
* @ param mod_flags Flags to be set on every message element
* @ return ldb_message on success or NULL on failure
*/
static struct ldb_message * PyDict_AsMessage ( TALLOC_CTX * mem_ctx ,
PyObject * py_obj ,
struct ldb_context * ldb_ctx ,
unsigned int mod_flags )
{
struct ldb_message * msg ;
unsigned int msg_pos = 0 ;
Py_ssize_t dict_pos = 0 ;
PyObject * key , * value ;
struct ldb_message_element * msg_el ;
PyObject * dn_value = PyDict_GetItemString ( py_obj , " dn " ) ;
msg = ldb_msg_new ( mem_ctx ) ;
msg - > elements = talloc_zero_array ( msg , struct ldb_message_element , PyDict_Size ( py_obj ) ) ;
if ( dn_value ) {
if ( ! PyObject_AsDn ( msg , dn_value , ldb_ctx , & msg - > dn ) ) {
PyErr_SetString ( PyExc_TypeError , " unable to import dn object " ) ;
return NULL ;
}
if ( msg - > dn = = NULL ) {
PyErr_SetString ( PyExc_TypeError , " dn set but not found " ) ;
return NULL ;
}
} else {
PyErr_SetString ( PyExc_TypeError , " no dn set " ) ;
return NULL ;
}
while ( PyDict_Next ( py_obj , & dict_pos , & key , & value ) ) {
char * key_str = PyString_AsString ( key ) ;
if ( strcmp ( key_str , " dn " ) ! = 0 ) {
msg_el = PyObject_AsMessageElement ( msg - > elements , value ,
mod_flags , key_str ) ;
if ( msg_el = = NULL ) {
PyErr_SetString ( PyExc_TypeError , " unable to import element " ) ;
return NULL ;
}
memcpy ( & msg - > elements [ msg_pos ] , msg_el , sizeof ( * msg_el ) ) ;
msg_pos + + ;
}
}
msg - > num_elements = msg_pos ;
return msg ;
}
2011-06-14 10:41:35 +04:00
static PyObject * py_ldb_add ( PyLdbObject * self , PyObject * args , PyObject * kwargs )
2008-12-19 04:22:07 +03:00
{
2010-11-18 23:09:01 +03:00
PyObject * py_obj ;
2008-12-19 04:22:07 +03:00
int ret ;
2009-09-23 00:51:25 +04:00
struct ldb_context * ldb_ctx ;
struct ldb_request * req ;
2010-11-18 23:09:01 +03:00
struct ldb_message * msg = NULL ;
2009-09-23 00:51:25 +04:00
PyObject * py_controls = Py_None ;
2009-06-17 22:43:25 +04:00
TALLOC_CTX * mem_ctx ;
2009-09-23 00:51:25 +04:00
struct ldb_control * * parsed_controls ;
2011-06-14 10:41:35 +04:00
const char * const kwnames [ ] = { " message " , " controls " , NULL } ;
2008-12-19 04:22:07 +03:00
2011-06-14 10:41:35 +04:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " O|O " ,
discard_const_p ( char * , kwnames ) ,
& py_obj , & py_controls ) )
2008-12-19 04:22:07 +03:00
return NULL ;
2009-06-17 22:43:25 +04:00
mem_ctx = talloc_new ( NULL ) ;
2010-06-19 00:08:58 +04:00
if ( mem_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
ldb_ctx = PyLdb_AsLdbContext ( self ) ;
2009-09-23 00:51:25 +04:00
if ( py_controls = = Py_None ) {
parsed_controls = NULL ;
} else {
2010-06-19 00:08:58 +04:00
const char * * controls = PyList_AsStringList ( mem_ctx , py_controls , " controls " ) ;
parsed_controls = ldb_parse_control_strings ( ldb_ctx , mem_ctx , controls ) ;
2009-09-23 00:51:25 +04:00
talloc_free ( controls ) ;
}
2008-12-19 04:22:07 +03:00
2010-11-18 23:09:01 +03:00
if ( PyLdbMessage_Check ( py_obj ) ) {
msg = PyLdbMessage_AsMessage ( py_obj ) ;
} else if ( PyDict_Check ( py_obj ) ) {
msg = PyDict_AsMessage ( mem_ctx , py_obj , ldb_ctx , LDB_FLAG_MOD_ADD ) ;
2008-12-21 00:21:39 +03:00
} else {
2010-11-18 23:09:01 +03:00
PyErr_SetString ( PyExc_TypeError ,
" Dictionary or LdbMessage object expected! " ) ;
2008-12-21 00:21:39 +03:00
}
2010-11-18 23:09:01 +03:00
if ( ! msg ) {
/* we should have a PyErr already set */
talloc_free ( mem_ctx ) ;
return NULL ;
}
2009-09-23 00:51:25 +04:00
ret = ldb_msg_sanity_check ( ldb_ctx , msg ) ;
2010-12-21 16:05:18 +03:00
if ( ret ! = LDB_SUCCESS ) {
2010-06-20 20:06:54 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
2009-09-23 00:51:25 +04:00
talloc_free ( mem_ctx ) ;
return NULL ;
2010-12-21 16:05:18 +03:00
}
2009-09-23 00:51:25 +04:00
2010-12-21 16:05:18 +03:00
ret = ldb_build_add_req ( & req , ldb_ctx , mem_ctx , msg , parsed_controls ,
2010-06-19 00:08:58 +04:00
NULL , ldb_op_default_callback , NULL ) ;
2010-12-21 16:05:18 +03:00
if ( ret ! = LDB_SUCCESS ) {
2009-09-23 00:51:25 +04:00
PyErr_SetString ( PyExc_TypeError , " failed to build request " ) ;
talloc_free ( mem_ctx ) ;
return NULL ;
}
2009-05-31 18:19:11 +04:00
2009-09-23 00:51:25 +04:00
/* do request and autostart a transaction */
/* Then let's LDB handle the message error in case of pb as they are meaningful */
2010-12-21 16:05:18 +03:00
ret = ldb_transaction_start ( ldb_ctx ) ;
if ( ret ! = LDB_SUCCESS ) {
2009-09-23 00:51:25 +04:00
talloc_free ( mem_ctx ) ;
2010-06-19 00:08:58 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
2010-12-21 16:05:18 +03:00
}
2009-09-23 00:51:25 +04:00
2010-12-21 16:05:18 +03:00
ret = ldb_request ( ldb_ctx , req ) ;
if ( ret = = LDB_SUCCESS ) {
ret = ldb_wait ( req - > handle , LDB_WAIT_ALL ) ;
}
2009-09-23 00:51:25 +04:00
if ( ret = = LDB_SUCCESS ) {
2010-12-21 16:05:18 +03:00
ret = ldb_transaction_commit ( ldb_ctx ) ;
} else {
ldb_transaction_cancel ( ldb_ctx ) ;
2009-09-23 00:51:25 +04:00
if ( ldb_ctx - > err_string = = NULL ) {
/* no error string was setup by the backend */
ldb_asprintf_errstring ( ldb_ctx , " %s (%d) " , ldb_strerror ( ret ) , ret ) ;
}
}
2010-06-19 00:08:58 +04:00
2009-06-17 22:43:25 +04:00
talloc_free ( mem_ctx ) ;
2010-06-19 00:08:58 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
2011-06-14 10:41:35 +04:00
static PyObject * py_ldb_delete ( PyLdbObject * self , PyObject * args , PyObject * kwargs )
2008-12-19 04:22:07 +03:00
{
PyObject * py_dn ;
struct ldb_dn * dn ;
int ret ;
2010-06-19 00:04:07 +04:00
struct ldb_context * ldb_ctx ;
struct ldb_request * req ;
PyObject * py_controls = Py_None ;
2009-11-15 16:23:32 +03:00
TALLOC_CTX * mem_ctx ;
2010-06-19 00:04:07 +04:00
struct ldb_control * * parsed_controls ;
2011-06-14 10:41:35 +04:00
const char * const kwnames [ ] = { " dn " , " controls " , NULL } ;
2010-06-19 00:04:07 +04:00
2011-06-14 10:41:35 +04:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " O|O " ,
discard_const_p ( char * , kwnames ) ,
& py_dn , & py_controls ) )
2008-12-19 04:22:07 +03:00
return NULL ;
2009-11-15 16:23:32 +03:00
mem_ctx = talloc_new ( NULL ) ;
if ( mem_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2010-06-19 00:04:07 +04:00
ldb_ctx = PyLdb_AsLdbContext ( self ) ;
if ( py_controls = = Py_None ) {
parsed_controls = NULL ;
} else {
const char * * controls = PyList_AsStringList ( mem_ctx , py_controls , " controls " ) ;
parsed_controls = ldb_parse_control_strings ( ldb_ctx , mem_ctx , controls ) ;
talloc_free ( controls ) ;
}
if ( ! PyObject_AsDn ( mem_ctx , py_dn , ldb_ctx , & dn ) ) {
talloc_free ( mem_ctx ) ;
return NULL ;
}
ret = ldb_build_del_req ( & req , ldb_ctx , mem_ctx , dn , parsed_controls ,
NULL , ldb_op_default_callback , NULL ) ;
if ( ret ! = LDB_SUCCESS ) {
PyErr_SetString ( PyExc_TypeError , " failed to build request " ) ;
2009-11-15 16:23:32 +03:00
talloc_free ( mem_ctx ) ;
2008-12-19 04:22:07 +03:00
return NULL ;
2009-11-15 16:23:32 +03:00
}
2008-12-19 04:22:07 +03:00
2010-06-19 00:04:07 +04:00
/* do request and autostart a transaction */
/* Then let's LDB handle the message error in case of pb as they are meaningful */
ret = ldb_transaction_start ( ldb_ctx ) ;
if ( ret ! = LDB_SUCCESS ) {
talloc_free ( mem_ctx ) ;
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
}
ret = ldb_request ( ldb_ctx , req ) ;
if ( ret = = LDB_SUCCESS ) {
ret = ldb_wait ( req - > handle , LDB_WAIT_ALL ) ;
}
if ( ret = = LDB_SUCCESS ) {
ret = ldb_transaction_commit ( ldb_ctx ) ;
} else {
ldb_transaction_cancel ( ldb_ctx ) ;
if ( ldb_ctx - > err_string = = NULL ) {
/* no error string was setup by the backend */
ldb_asprintf_errstring ( ldb_ctx , " %s (%d) " , ldb_strerror ( ret ) , ret ) ;
}
}
2009-11-15 16:23:32 +03:00
talloc_free ( mem_ctx ) ;
2010-06-19 00:04:07 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
2011-06-14 10:41:35 +04:00
static PyObject * py_ldb_rename ( PyLdbObject * self , PyObject * args , PyObject * kwargs )
2008-12-19 04:22:07 +03:00
{
PyObject * py_dn1 , * py_dn2 ;
struct ldb_dn * dn1 , * dn2 ;
int ret ;
2008-12-21 06:36:16 +03:00
struct ldb_context * ldb ;
2009-06-17 22:43:25 +04:00
TALLOC_CTX * mem_ctx ;
2010-09-05 02:57:16 +04:00
PyObject * py_controls = Py_None ;
struct ldb_control * * parsed_controls ;
struct ldb_context * ldb_ctx ;
struct ldb_request * req ;
2011-06-14 10:41:35 +04:00
const char * const kwnames [ ] = { " dn1 " , " dn2 " , " controls " , NULL } ;
2010-06-20 20:06:54 +04:00
2010-09-05 02:57:16 +04:00
ldb_ctx = PyLdb_AsLdbContext ( self ) ;
2011-06-14 10:41:35 +04:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " OO|O " ,
discard_const_p ( char * , kwnames ) ,
& py_dn1 , & py_dn2 , & py_controls ) )
2008-12-19 04:22:07 +03:00
return NULL ;
2010-09-05 02:57:16 +04:00
2009-06-17 22:43:25 +04:00
mem_ctx = talloc_new ( NULL ) ;
if ( mem_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2008-12-21 06:36:16 +03:00
ldb = PyLdb_AsLdbContext ( self ) ;
2010-06-20 20:06:54 +04:00
2010-09-05 02:57:16 +04:00
if ( py_controls = = Py_None ) {
parsed_controls = NULL ;
} else {
const char * * controls = PyList_AsStringList ( mem_ctx , py_controls , " controls " ) ;
parsed_controls = ldb_parse_control_strings ( ldb_ctx , mem_ctx , controls ) ;
talloc_free ( controls ) ;
}
2009-06-17 22:43:25 +04:00
if ( ! PyObject_AsDn ( mem_ctx , py_dn1 , ldb , & dn1 ) ) {
talloc_free ( mem_ctx ) ;
2008-12-19 19:08:35 +03:00
return NULL ;
2009-06-17 22:43:25 +04:00
}
2008-12-19 19:08:35 +03:00
2009-06-17 22:43:25 +04:00
if ( ! PyObject_AsDn ( mem_ctx , py_dn2 , ldb , & dn2 ) ) {
talloc_free ( mem_ctx ) ;
2008-12-19 19:08:35 +03:00
return NULL ;
2009-06-17 22:43:25 +04:00
}
2008-12-19 04:22:07 +03:00
2010-09-05 02:57:16 +04:00
ret = ldb_build_rename_req ( & req , ldb_ctx , mem_ctx , dn1 , dn2 , parsed_controls ,
NULL , ldb_op_default_callback , NULL ) ;
if ( ret ! = LDB_SUCCESS ) {
PyErr_SetString ( PyExc_TypeError , " failed to build request " ) ;
talloc_free ( mem_ctx ) ;
return NULL ;
}
/* do request and autostart a transaction */
/* Then let's LDB handle the message error in case of pb as they are meaningful */
ret = ldb_transaction_start ( ldb_ctx ) ;
if ( ret ! = LDB_SUCCESS ) {
talloc_free ( mem_ctx ) ;
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
}
ret = ldb_request ( ldb_ctx , req ) ;
if ( ret = = LDB_SUCCESS ) {
ret = ldb_wait ( req - > handle , LDB_WAIT_ALL ) ;
}
if ( ret = = LDB_SUCCESS ) {
ret = ldb_transaction_commit ( ldb_ctx ) ;
} else {
ldb_transaction_cancel ( ldb_ctx ) ;
if ( ldb_ctx - > err_string = = NULL ) {
/* no error string was setup by the backend */
ldb_asprintf_errstring ( ldb_ctx , " %s (%d) " , ldb_strerror ( ret ) , ret ) ;
}
}
2009-06-17 22:43:25 +04:00
talloc_free ( mem_ctx ) ;
2010-09-05 02:57:16 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_schema_attribute_remove ( PyLdbObject * self , PyObject * args )
{
char * name ;
if ( ! PyArg_ParseTuple ( args , " s " , & name ) )
return NULL ;
ldb_schema_attribute_remove ( PyLdb_AsLdbContext ( self ) , name ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_schema_attribute_add ( PyLdbObject * self , PyObject * args )
{
char * attribute , * syntax ;
unsigned int flags ;
int ret ;
if ( ! PyArg_ParseTuple ( args , " sIs " , & attribute , & flags , & syntax ) )
return NULL ;
ret = ldb_schema_attribute_add ( PyLdb_AsLdbContext ( self ) , attribute , flags , syntax ) ;
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , PyLdb_AsLdbContext ( self ) ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * ldb_ldif_to_pyobject ( struct ldb_ldif * ldif )
{
if ( ldif = = NULL ) {
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
} else {
/* We don't want this attached to the 'ldb' any more */
2009-02-05 13:04:28 +03:00
return Py_BuildValue ( discard_const_p ( char , " (iO) " ) ,
ldif - > changetype ,
PyLdbMessage_FromMessage ( ldif - > msg ) ) ;
2008-12-19 04:22:07 +03:00
}
}
2010-11-15 05:20:31 +03:00
static PyObject * py_ldb_write_ldif ( PyLdbObject * self , PyObject * args )
2009-08-26 09:59:00 +04:00
{
int changetype ;
PyObject * py_msg ;
struct ldb_ldif ldif ;
PyObject * ret ;
char * string ;
TALLOC_CTX * mem_ctx ;
if ( ! PyArg_ParseTuple ( args , " Oi " , & py_msg , & changetype ) )
return NULL ;
if ( ! PyLdbMessage_Check ( py_msg ) ) {
PyErr_SetString ( PyExc_TypeError , " Expected Ldb Message for msg " ) ;
return NULL ;
}
ldif . msg = PyLdbMessage_AsMessage ( py_msg ) ;
ldif . changetype = changetype ;
mem_ctx = talloc_new ( NULL ) ;
string = ldb_ldif_write_string ( PyLdb_AsLdbContext ( self ) , mem_ctx , & ldif ) ;
if ( ! string ) {
PyErr_SetString ( PyExc_KeyError , " Failed to generate LDIF " ) ;
return NULL ;
}
ret = PyString_FromString ( string ) ;
talloc_free ( mem_ctx ) ;
return ret ;
}
2008-12-19 04:22:07 +03:00
static PyObject * py_ldb_parse_ldif ( PyLdbObject * self , PyObject * args )
{
PyObject * list ;
struct ldb_ldif * ldif ;
const char * s ;
2009-07-06 03:31:38 +04:00
TALLOC_CTX * mem_ctx ;
2008-12-21 00:21:39 +03:00
if ( ! PyArg_ParseTuple ( args , " s " , & s ) )
2008-12-19 04:22:07 +03:00
return NULL ;
2009-07-06 03:31:38 +04:00
mem_ctx = talloc_new ( NULL ) ;
if ( ! mem_ctx ) {
Py_RETURN_NONE ;
}
2008-12-19 04:22:07 +03:00
list = PyList_New ( 0 ) ;
2009-07-06 03:31:38 +04:00
while ( s & & * s ! = ' \0 ' ) {
ldif = ldb_ldif_read_string ( self - > ldb_ctx , & s ) ;
talloc_steal ( mem_ctx , ldif ) ;
if ( ldif ) {
PyList_Append ( list , ldb_ldif_to_pyobject ( ldif ) ) ;
} else {
PyErr_SetString ( PyExc_ValueError , " unable to parse ldif string " ) ;
talloc_free ( mem_ctx ) ;
return NULL ;
}
2008-12-19 04:22:07 +03:00
}
2009-07-06 03:31:38 +04:00
talloc_free ( mem_ctx ) ; /* The pyobject already has a reference to the things it needs */
2008-12-19 04:22:07 +03:00
return PyObject_GetIter ( list ) ;
}
2009-08-24 14:11:43 +04:00
static PyObject * py_ldb_msg_diff ( PyLdbObject * self , PyObject * args )
{
2010-07-16 14:40:50 +04:00
int ldb_ret ;
2009-08-24 14:11:43 +04:00
PyObject * py_msg_old ;
PyObject * py_msg_new ;
struct ldb_message * diff ;
2010-07-16 14:40:50 +04:00
struct ldb_context * ldb ;
2009-08-24 14:11:43 +04:00
PyObject * py_ret ;
if ( ! PyArg_ParseTuple ( args , " OO " , & py_msg_old , & py_msg_new ) )
return NULL ;
if ( ! PyLdbMessage_Check ( py_msg_old ) ) {
PyErr_SetString ( PyExc_TypeError , " Expected Ldb Message for old message " ) ;
return NULL ;
}
if ( ! PyLdbMessage_Check ( py_msg_new ) ) {
PyErr_SetString ( PyExc_TypeError , " Expected Ldb Message for new message " ) ;
return NULL ;
}
2010-07-16 14:40:50 +04:00
ldb = PyLdb_AsLdbContext ( self ) ;
ldb_ret = ldb_msg_difference ( ldb , ldb ,
PyLdbMessage_AsMessage ( py_msg_old ) ,
PyLdbMessage_AsMessage ( py_msg_new ) ,
& diff ) ;
if ( ldb_ret ! = LDB_SUCCESS ) {
2009-10-01 16:02:59 +04:00
PyErr_SetString ( PyExc_RuntimeError , " Failed to generate the Ldb Message diff " ) ;
2009-10-01 15:59:02 +04:00
return NULL ;
}
2009-08-24 14:11:43 +04:00
py_ret = PyLdbMessage_FromMessage ( diff ) ;
2010-07-16 14:40:50 +04:00
talloc_unlink ( ldb , diff ) ;
2009-08-24 14:11:43 +04:00
return py_ret ;
}
2008-12-19 04:22:07 +03:00
static PyObject * py_ldb_schema_format_value ( PyLdbObject * self , PyObject * args )
{
const struct ldb_schema_attribute * a ;
struct ldb_val old_val ;
struct ldb_val new_val ;
TALLOC_CTX * mem_ctx ;
PyObject * ret ;
char * element_name ;
PyObject * val ;
if ( ! PyArg_ParseTuple ( args , " sO " , & element_name , & val ) )
return NULL ;
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
mem_ctx = talloc_new ( NULL ) ;
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
old_val . data = ( uint8_t * ) PyString_AsString ( val ) ;
old_val . length = PyString_Size ( val ) ;
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
a = ldb_schema_attribute_by_name ( PyLdb_AsLdbContext ( self ) , element_name ) ;
if ( a = = NULL ) {
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
if ( a - > syntax - > ldif_write_fn ( PyLdb_AsLdbContext ( self ) , mem_ctx , & old_val , & new_val ) ! = 0 ) {
talloc_free ( mem_ctx ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
ret = PyString_FromStringAndSize ( ( const char * ) new_val . data , new_val . length ) ;
talloc_free ( mem_ctx ) ;
return ret ;
}
2008-12-19 16:41:44 +03:00
static PyObject * py_ldb_search ( PyLdbObject * self , PyObject * args , PyObject * kwargs )
{
PyObject * py_base = Py_None ;
2009-10-27 21:07:53 +03:00
int scope = LDB_SCOPE_DEFAULT ;
2008-12-19 16:41:44 +03:00
char * expr = NULL ;
PyObject * py_attrs = Py_None ;
PyObject * py_controls = Py_None ;
2009-02-05 13:04:28 +03:00
const char * const kwnames [ ] = { " base " , " scope " , " expression " , " attrs " , " controls " , NULL } ;
2008-12-19 16:41:44 +03:00
int ret ;
struct ldb_result * res ;
struct ldb_request * req ;
const char * * attrs ;
struct ldb_context * ldb_ctx ;
struct ldb_control * * parsed_controls ;
struct ldb_dn * base ;
2009-06-17 20:26:40 +04:00
PyObject * py_ret ;
2010-06-20 20:20:00 +04:00
TALLOC_CTX * mem_ctx ;
2008-12-19 16:41:44 +03:00
2009-10-27 21:11:15 +03:00
/* type "int" rather than "enum" for "scope" is intentional */
2009-02-05 13:04:28 +03:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " |OizOO " ,
discard_const_p ( char * , kwnames ) ,
& py_base , & scope , & expr , & py_attrs , & py_controls ) )
2008-12-19 16:41:44 +03:00
return NULL ;
2010-06-20 20:20:00 +04:00
mem_ctx = talloc_new ( NULL ) ;
if ( mem_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2008-12-19 16:41:44 +03:00
ldb_ctx = PyLdb_AsLdbContext ( self ) ;
if ( py_attrs = = Py_None ) {
attrs = NULL ;
} else {
2010-06-20 20:20:00 +04:00
attrs = PyList_AsStringList ( mem_ctx , py_attrs , " attrs " ) ;
if ( attrs = = NULL ) {
talloc_free ( mem_ctx ) ;
2008-12-19 16:41:44 +03:00
return NULL ;
2010-06-20 20:20:00 +04:00
}
2008-12-19 16:41:44 +03:00
}
if ( py_base = = Py_None ) {
base = ldb_get_default_basedn ( ldb_ctx ) ;
} else {
2009-06-17 20:36:16 +04:00
if ( ! PyObject_AsDn ( ldb_ctx , py_base , ldb_ctx , & base ) ) {
talloc_free ( attrs ) ;
2008-12-19 16:41:44 +03:00
return NULL ;
2009-06-17 20:36:16 +04:00
}
2008-12-19 16:41:44 +03:00
}
if ( py_controls = = Py_None ) {
parsed_controls = NULL ;
} else {
2010-06-20 20:20:00 +04:00
const char * * controls = PyList_AsStringList ( mem_ctx , py_controls , " controls " ) ;
parsed_controls = ldb_parse_control_strings ( ldb_ctx , mem_ctx , controls ) ;
2008-12-19 16:41:44 +03:00
talloc_free ( controls ) ;
}
2010-06-20 20:20:00 +04:00
res = talloc_zero ( mem_ctx , struct ldb_result ) ;
2008-12-19 19:08:35 +03:00
if ( res = = NULL ) {
2008-12-19 16:41:44 +03:00
PyErr_NoMemory ( ) ;
2010-06-20 20:20:00 +04:00
talloc_free ( mem_ctx ) ;
2008-12-19 16:41:44 +03:00
return NULL ;
}
2010-06-20 20:20:00 +04:00
ret = ldb_build_search_req ( & req , ldb_ctx , mem_ctx ,
2008-12-19 16:41:44 +03:00
base ,
scope ,
expr ,
attrs ,
parsed_controls ,
res ,
ldb_search_default_callback ,
NULL ) ;
if ( ret ! = LDB_SUCCESS ) {
2010-06-20 20:20:00 +04:00
talloc_free ( mem_ctx ) ;
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
2008-12-19 16:41:44 +03:00
return NULL ;
}
2010-10-04 00:43:33 +04:00
talloc_steal ( req , attrs ) ;
2008-12-19 16:41:44 +03:00
ret = ldb_request ( ldb_ctx , req ) ;
2009-05-31 18:19:11 +04:00
2008-12-19 16:41:44 +03:00
if ( ret = = LDB_SUCCESS ) {
ret = ldb_wait ( req - > handle , LDB_WAIT_ALL ) ;
}
2009-01-07 08:29:23 +03:00
if ( ret ! = LDB_SUCCESS ) {
2010-06-20 20:20:00 +04:00
talloc_free ( mem_ctx ) ;
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb_ctx ) ;
2009-01-07 08:29:23 +03:00
return NULL ;
}
2009-06-17 20:26:40 +04:00
py_ret = PyLdbResult_FromResult ( res ) ;
2010-06-20 20:20:00 +04:00
talloc_free ( mem_ctx ) ;
2009-06-17 20:26:40 +04:00
return py_ret ;
2008-12-19 16:41:44 +03:00
}
static PyObject * py_ldb_get_opaque ( PyLdbObject * self , PyObject * args )
{
char * name ;
void * data ;
2008-12-19 19:08:35 +03:00
if ( ! PyArg_ParseTuple ( args , " s " , & name ) )
2008-12-19 16:41:44 +03:00
return NULL ;
data = ldb_get_opaque ( PyLdb_AsLdbContext ( self ) , name ) ;
2008-12-21 06:36:16 +03:00
if ( data = = NULL )
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-21 06:36:16 +03:00
2008-12-19 16:41:44 +03:00
/* FIXME: More interpretation */
2008-12-21 06:36:16 +03:00
return Py_True ;
2008-12-19 16:41:44 +03:00
}
static PyObject * py_ldb_set_opaque ( PyLdbObject * self , PyObject * args )
{
char * name ;
PyObject * data ;
2008-12-19 19:08:35 +03:00
if ( ! PyArg_ParseTuple ( args , " sO " , & name , & data ) )
2008-12-19 16:41:44 +03:00
return NULL ;
/* FIXME: More interpretation */
ldb_set_opaque ( PyLdb_AsLdbContext ( self ) , name , data ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 16:41:44 +03:00
}
2008-12-21 06:36:16 +03:00
static PyObject * py_ldb_modules ( PyLdbObject * self )
{
struct ldb_context * ldb = PyLdb_AsLdbContext ( self ) ;
PyObject * ret = PyList_New ( 0 ) ;
struct ldb_module * mod ;
for ( mod = ldb - > modules ; mod ; mod = mod - > next ) {
PyList_Append ( ret , PyLdbModule_FromModule ( mod ) ) ;
}
return ret ;
}
2010-06-21 10:55:18 +04:00
static PyObject * py_ldb_sequence_number ( PyLdbObject * self , PyObject * args )
{
struct ldb_context * ldb = PyLdb_AsLdbContext ( self ) ;
int type , ret ;
uint64_t value ;
if ( ! PyArg_ParseTuple ( args , " i " , & type ) )
return NULL ;
/* FIXME: More interpretation */
ret = ldb_sequence_number ( ldb , type , & value ) ;
if ( ret ! = LDB_SUCCESS ) {
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , ldb ) ;
return NULL ;
}
return PyLong_FromLongLong ( value ) ;
}
2008-12-19 04:22:07 +03:00
static PyMethodDef py_ldb_methods [ ] = {
{ " set_debug " , ( PyCFunction ) py_ldb_set_debug , METH_VARARGS ,
" S.set_debug(callback) -> None \n "
" Set callback for LDB debug messages. \n "
" The callback should accept a debug level and debug text. " } ,
{ " set_create_perms " , ( PyCFunction ) py_ldb_set_create_perms , METH_VARARGS ,
" S.set_create_perms(mode) -> None \n "
" Set mode to use when creating new LDB files. " } ,
{ " set_modules_dir " , ( PyCFunction ) py_ldb_set_modules_dir , METH_VARARGS ,
" S.set_modules_dir(path) -> None \n "
" Set path LDB should search for modules " } ,
{ " transaction_start " , ( PyCFunction ) py_ldb_transaction_start , METH_NOARGS ,
" S.transaction_start() -> None \n "
" Start a new transaction. " } ,
2009-11-26 07:32:06 +03:00
{ " transaction_prepare_commit " , ( PyCFunction ) py_ldb_transaction_prepare_commit , METH_NOARGS ,
" S.transaction_prepare_commit() -> None \n "
" prepare to commit a new transaction (2-stage commit). " } ,
2008-12-19 04:22:07 +03:00
{ " transaction_commit " , ( PyCFunction ) py_ldb_transaction_commit , METH_NOARGS ,
" S.transaction_commit() -> None \n "
" commit a new transaction. " } ,
{ " transaction_cancel " , ( PyCFunction ) py_ldb_transaction_cancel , METH_NOARGS ,
" S.transaction_cancel() -> None \n "
" cancel a new transaction. " } ,
{ " setup_wellknown_attributes " , ( PyCFunction ) py_ldb_setup_wellknown_attributes , METH_NOARGS ,
NULL } ,
{ " get_root_basedn " , ( PyCFunction ) py_ldb_get_root_basedn , METH_NOARGS ,
NULL } ,
{ " get_schema_basedn " , ( PyCFunction ) py_ldb_get_schema_basedn , METH_NOARGS ,
NULL } ,
{ " get_default_basedn " , ( PyCFunction ) py_ldb_get_default_basedn , METH_NOARGS ,
NULL } ,
{ " get_config_basedn " , ( PyCFunction ) py_ldb_get_config_basedn , METH_NOARGS ,
NULL } ,
{ " connect " , ( PyCFunction ) py_ldb_connect , METH_VARARGS | METH_KEYWORDS ,
2008-12-19 16:41:44 +03:00
" S.connect(url, flags=0, options=None) -> None \n "
2008-12-19 04:22:07 +03:00
" Connect to a LDB URL. " } ,
2011-06-14 10:39:49 +04:00
{ " modify " , ( PyCFunction ) py_ldb_modify , METH_VARARGS | METH_KEYWORDS ,
" S.modify(message, controls=None, validate=False) -> None \n "
2008-12-19 04:22:07 +03:00
" Modify an entry. " } ,
2011-06-14 10:41:35 +04:00
{ " add " , ( PyCFunction ) py_ldb_add , METH_VARARGS | METH_KEYWORDS ,
" S.add(message, controls=None) -> None \n "
2008-12-19 04:22:07 +03:00
" Add an entry. " } ,
2011-06-14 10:41:35 +04:00
{ " delete " , ( PyCFunction ) py_ldb_delete , METH_VARARGS | METH_KEYWORDS ,
" S.delete(dn, controls=None) -> None \n "
2008-12-19 04:22:07 +03:00
" Remove an entry. " } ,
2011-06-14 10:41:35 +04:00
{ " rename " , ( PyCFunction ) py_ldb_rename , METH_VARARGS | METH_KEYWORDS ,
" S.rename(old_dn, new_dn, controls=None) -> None \n "
2008-12-19 04:22:07 +03:00
" Rename an entry. " } ,
2008-12-19 16:41:44 +03:00
{ " search " , ( PyCFunction ) py_ldb_search , METH_VARARGS | METH_KEYWORDS ,
2008-12-21 01:00:23 +03:00
" S.search(base=None, scope=None, expression=None, attrs=None, controls=None) -> msgs \n "
2008-12-19 16:41:44 +03:00
" Search in a database. \n "
" \n "
" :param base: Optional base DN to search \n "
" :param scope: Search scope (SCOPE_BASE, SCOPE_ONELEVEL or SCOPE_SUBTREE) \n "
" :param expression: Optional search expression \n "
" :param attrs: Attributes to return (defaults to all) \n "
" :param controls: Optional list of controls \n "
" :return: Iterator over Message objects \n "
} ,
2008-12-19 04:22:07 +03:00
{ " schema_attribute_remove " , ( PyCFunction ) py_ldb_schema_attribute_remove , METH_VARARGS ,
NULL } ,
{ " schema_attribute_add " , ( PyCFunction ) py_ldb_schema_attribute_add , METH_VARARGS ,
NULL } ,
{ " schema_format_value " , ( PyCFunction ) py_ldb_schema_format_value , METH_VARARGS ,
NULL } ,
{ " parse_ldif " , ( PyCFunction ) py_ldb_parse_ldif , METH_VARARGS ,
" S.parse_ldif(ldif) -> iter(messages) \n "
2008-12-21 00:21:39 +03:00
" Parse a string formatted using LDIF. " } ,
2009-08-26 09:59:00 +04:00
{ " write_ldif " , ( PyCFunction ) py_ldb_write_ldif , METH_VARARGS ,
" S.write_ldif(message, changetype) -> ldif \n "
" Print the message as a string formatted using LDIF. " } ,
2009-08-24 14:11:43 +04:00
{ " msg_diff " , ( PyCFunction ) py_ldb_msg_diff , METH_VARARGS ,
" S.msg_diff(Message) -> Message \n "
" Return an LDB Message of the difference between two Message objects. " } ,
2008-12-21 06:36:16 +03:00
{ " get_opaque " , ( PyCFunction ) py_ldb_get_opaque , METH_VARARGS ,
2008-12-21 00:21:39 +03:00
" S.get_opaque(name) -> value \n "
" Get an opaque value set on this LDB connection. \n "
" :note: The returned value may not be useful in Python. "
2008-12-19 16:41:44 +03:00
} ,
2008-12-21 06:36:16 +03:00
{ " set_opaque " , ( PyCFunction ) py_ldb_set_opaque , METH_VARARGS ,
2008-12-19 16:41:44 +03:00
" S.set_opaque(name, value) -> None \n "
" Set an opaque value on this LDB connection. \n "
2008-12-21 00:21:39 +03:00
" :note: Passing incorrect values may cause crashes. " } ,
2008-12-21 06:36:16 +03:00
{ " modules " , ( PyCFunction ) py_ldb_modules , METH_NOARGS ,
" S.modules() -> list \n "
" Return the list of modules on this LDB connection " } ,
2010-06-21 10:55:18 +04:00
{ " sequence_number " , ( PyCFunction ) py_ldb_sequence_number , METH_VARARGS ,
" S.sequence_number(type) -> value \n "
" Return the value of the sequence according to the requested type " } ,
2008-12-19 04:22:07 +03:00
{ NULL } ,
} ;
2010-12-30 21:39:14 +03:00
static PyObject * PyLdbModule_FromModule ( struct ldb_module * mod )
2008-12-19 04:22:07 +03:00
{
2008-12-23 07:07:29 +03:00
PyLdbModuleObject * ret ;
ret = ( PyLdbModuleObject * ) PyLdbModule . tp_alloc ( & PyLdbModule , 0 ) ;
if ( ret = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
ret - > mem_ctx = talloc_new ( NULL ) ;
ret - > mod = talloc_reference ( ret - > mem_ctx , mod ) ;
return ( PyObject * ) ret ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_get_firstmodule ( PyLdbObject * self , void * closure )
{
return PyLdbModule_FromModule ( PyLdb_AsLdbContext ( self ) - > modules ) ;
}
static PyGetSetDef py_ldb_getset [ ] = {
2009-02-05 13:04:28 +03:00
{ discard_const_p ( char , " firstmodule " ) , ( getter ) py_ldb_get_firstmodule , NULL , NULL } ,
2008-12-19 04:22:07 +03:00
{ NULL }
} ;
2008-12-19 16:41:44 +03:00
static int py_ldb_contains ( PyLdbObject * self , PyObject * obj )
{
struct ldb_context * ldb_ctx = PyLdb_AsLdbContext ( self ) ;
2008-12-19 19:08:35 +03:00
struct ldb_dn * dn ;
2008-12-19 16:41:44 +03:00
struct ldb_result * result ;
2010-11-06 19:49:18 +03:00
unsigned int count ;
2008-12-19 16:41:44 +03:00
int ret ;
2008-12-19 19:08:35 +03:00
2010-11-09 01:21:57 +03:00
if ( ! PyObject_AsDn ( ldb_ctx , obj , ldb_ctx , & dn ) ) {
2008-12-19 19:08:35 +03:00
return - 1 ;
2010-11-09 01:21:57 +03:00
}
2008-12-19 19:08:35 +03:00
2010-11-06 19:49:18 +03:00
ret = ldb_search ( ldb_ctx , ldb_ctx , & result , dn , LDB_SCOPE_BASE , NULL ,
NULL ) ;
2008-12-19 19:08:35 +03:00
if ( ret ! = LDB_SUCCESS ) {
2009-04-23 03:21:47 +04:00
PyErr_SetLdbError ( PyExc_LdbError , ret , ldb_ctx ) ;
2008-12-19 19:08:35 +03:00
return - 1 ;
}
2008-12-19 16:41:44 +03:00
count = result - > count ;
talloc_free ( result ) ;
2010-11-09 01:21:57 +03:00
if ( count > 1 ) {
PyErr_Format ( PyExc_RuntimeError ,
" Searching for [%s] dn gave %u results! " ,
ldb_dn_get_linearized ( dn ) ,
count ) ;
return - 1 ;
2010-11-06 19:49:18 +03:00
}
2010-11-09 01:21:57 +03:00
return count ;
2008-12-19 16:41:44 +03:00
}
static PySequenceMethods py_ldb_seq = {
. sq_contains = ( objobjproc ) py_ldb_contains ,
} ;
2009-12-20 20:31:27 +03:00
static PyObject * PyLdb_FromLdbContext ( struct ldb_context * ldb_ctx )
2008-12-19 19:08:35 +03:00
{
2008-12-23 07:07:29 +03:00
PyLdbObject * ret ;
ret = ( PyLdbObject * ) PyLdb . tp_alloc ( & PyLdb , 0 ) ;
if ( ret = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
ret - > mem_ctx = talloc_new ( NULL ) ;
ret - > ldb_ctx = talloc_reference ( ret - > mem_ctx , ldb_ctx ) ;
return ( PyObject * ) ret ;
}
static void py_ldb_dealloc ( PyLdbObject * self )
{
talloc_free ( self - > mem_ctx ) ;
2011-01-01 08:55:54 +03:00
self - > ob_type - > tp_free ( self ) ;
2008-12-19 19:08:35 +03:00
}
2010-12-30 21:39:14 +03:00
static PyTypeObject PyLdb = {
2010-04-04 03:48:35 +04:00
. tp_name = " ldb.Ldb " ,
2008-12-19 04:22:07 +03:00
. tp_methods = py_ldb_methods ,
. tp_repr = ( reprfunc ) py_ldb_repr ,
. tp_new = py_ldb_new ,
2008-12-21 01:00:23 +03:00
. tp_init = ( initproc ) py_ldb_init ,
2008-12-23 07:07:29 +03:00
. tp_dealloc = ( destructor ) py_ldb_dealloc ,
2008-12-19 04:22:07 +03:00
. tp_getset = py_ldb_getset ,
2008-12-21 01:00:23 +03:00
. tp_getattro = PyObject_GenericGetAttr ,
2008-12-19 04:22:07 +03:00
. tp_basicsize = sizeof ( PyLdbObject ) ,
. tp_doc = " Connection to a LDB database. " ,
2008-12-19 16:41:44 +03:00
. tp_as_sequence = & py_ldb_seq ,
2008-12-19 19:08:35 +03:00
. tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE ,
2008-12-19 04:22:07 +03:00
} ;
2011-02-07 09:50:36 +03:00
static void py_ldb_result_dealloc ( PyLdbResultObject * self )
{
talloc_free ( self - > mem_ctx ) ;
Py_DECREF ( self - > msgs ) ;
Py_DECREF ( self - > referals ) ;
Py_DECREF ( self - > controls ) ;
self - > ob_type - > tp_free ( self ) ;
}
static PyObject * py_ldb_result_get_msgs ( PyLdbResultObject * self , void * closure )
{
Py_INCREF ( self - > msgs ) ;
return self - > msgs ;
}
static PyObject * py_ldb_result_get_controls ( PyLdbResultObject * self , void * closure )
{
Py_INCREF ( self - > controls ) ;
return self - > controls ;
}
static PyObject * py_ldb_result_get_referals ( PyLdbResultObject * self , void * closure )
{
Py_INCREF ( self - > referals ) ;
return self - > referals ;
}
static PyObject * py_ldb_result_get_count ( PyLdbResultObject * self , void * closure )
{
Py_ssize_t size ;
if ( self - > msgs = = NULL ) {
PyErr_SetString ( PyExc_AttributeError , " Count attribute is meaningless in this context " ) ;
return NULL ;
}
size = PyList_Size ( self - > msgs ) ;
return PyInt_FromLong ( size ) ;
}
static PyGetSetDef py_ldb_result_getset [ ] = {
{ discard_const_p ( char , " controls " ) , ( getter ) py_ldb_result_get_controls , NULL , NULL } ,
{ discard_const_p ( char , " msgs " ) , ( getter ) py_ldb_result_get_msgs , NULL , NULL } ,
{ discard_const_p ( char , " referals " ) , ( getter ) py_ldb_result_get_referals , NULL , NULL } ,
{ discard_const_p ( char , " count " ) , ( getter ) py_ldb_result_get_count , NULL , NULL } ,
{ NULL }
} ;
static PyObject * py_ldb_result_iter ( PyLdbResultObject * self )
{
return PyObject_GetIter ( self - > msgs ) ;
}
static Py_ssize_t py_ldb_result_len ( PyLdbResultObject * self )
{
return PySequence_Size ( self - > msgs ) ;
}
static PyObject * py_ldb_result_find ( PyLdbResultObject * self , Py_ssize_t idx )
{
return PySequence_GetItem ( self - > msgs , idx ) ;
}
static PySequenceMethods py_ldb_result_seq = {
. sq_length = ( lenfunc ) py_ldb_result_len ,
. sq_item = ( ssizeargfunc ) py_ldb_result_find ,
} ;
static PyObject * py_ldb_result_repr ( PyLdbObject * self )
{
return PyString_FromFormat ( " <ldb result> " ) ;
}
static PyTypeObject PyLdbResult = {
. tp_name = " ldb.Result " ,
. tp_repr = ( reprfunc ) py_ldb_result_repr ,
. tp_dealloc = ( destructor ) py_ldb_result_dealloc ,
. tp_iter = ( getiterfunc ) py_ldb_result_iter ,
. tp_getset = py_ldb_result_getset ,
. tp_getattro = PyObject_GenericGetAttr ,
. tp_basicsize = sizeof ( PyLdbResultObject ) ,
. tp_as_sequence = & py_ldb_result_seq ,
. tp_doc = " LDB result. " ,
. tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE ,
} ;
2008-12-19 04:22:07 +03:00
static PyObject * py_ldb_module_repr ( PyLdbModuleObject * self )
{
return PyString_FromFormat ( " <ldb module '%s'> " , PyLdbModule_AsModule ( self ) - > ops - > name ) ;
}
static PyObject * py_ldb_module_str ( PyLdbModuleObject * self )
{
return PyString_FromString ( PyLdbModule_AsModule ( self ) - > ops - > name ) ;
}
static PyObject * py_ldb_module_start_transaction ( PyLdbModuleObject * self )
{
PyLdbModule_AsModule ( self ) - > ops - > start_transaction ( PyLdbModule_AsModule ( self ) ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_module_end_transaction ( PyLdbModuleObject * self )
{
PyLdbModule_AsModule ( self ) - > ops - > end_transaction ( PyLdbModule_AsModule ( self ) ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_module_del_transaction ( PyLdbModuleObject * self )
{
PyLdbModule_AsModule ( self ) - > ops - > del_transaction ( PyLdbModule_AsModule ( self ) ) ;
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_module_search ( PyLdbModuleObject * self , PyObject * args , PyObject * kwargs )
{
2009-06-17 20:26:40 +04:00
PyObject * py_base , * py_tree , * py_attrs , * py_ret ;
2009-10-27 21:07:53 +03:00
int ret , scope ;
2008-12-19 04:22:07 +03:00
struct ldb_request * req ;
2009-02-05 13:04:28 +03:00
const char * const kwnames [ ] = { " base " , " scope " , " tree " , " attrs " , NULL } ;
2008-12-19 19:08:35 +03:00
struct ldb_module * mod ;
2009-06-17 20:36:16 +04:00
const char * const * attrs ;
2009-02-05 13:04:28 +03:00
2009-10-27 21:11:15 +03:00
/* type "int" rather than "enum" for "scope" is intentional */
2009-02-05 13:04:28 +03:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " OiOO " ,
discard_const_p ( char * , kwnames ) ,
& py_base , & scope , & py_tree , & py_attrs ) )
2008-12-19 04:22:07 +03:00
return NULL ;
2008-12-23 07:07:29 +03:00
mod = self - > mod ;
2008-12-19 19:08:35 +03:00
2009-06-17 20:36:16 +04:00
if ( py_attrs = = Py_None ) {
attrs = NULL ;
} else {
attrs = PyList_AsStringList ( NULL , py_attrs , " attrs " ) ;
if ( attrs = = NULL )
return NULL ;
}
2008-12-21 06:36:16 +03:00
ret = ldb_build_search_req ( & req , mod - > ldb , NULL , PyLdbDn_AsDn ( py_base ) ,
2009-06-17 20:36:16 +04:00
scope , NULL /* expr */ , attrs ,
2008-12-21 06:36:16 +03:00
NULL /* controls */ , NULL , NULL , NULL ) ;
2009-06-17 20:36:16 +04:00
talloc_steal ( req , attrs ) ;
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , mod - > ldb ) ;
2008-12-21 06:36:16 +03:00
2009-06-17 20:26:40 +04:00
req - > op . search . res = NULL ;
2008-12-19 19:08:35 +03:00
ret = mod - > ops - > search ( mod , req ) ;
2008-12-19 04:22:07 +03:00
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , mod - > ldb ) ;
2008-12-19 04:22:07 +03:00
2009-06-17 20:26:40 +04:00
py_ret = PyLdbResult_FromResult ( req - > op . search . res ) ;
talloc_free ( req ) ;
return py_ret ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_module_add ( PyLdbModuleObject * self , PyObject * args )
{
struct ldb_request * req ;
PyObject * py_message ;
int ret ;
2008-12-21 06:36:16 +03:00
struct ldb_module * mod ;
2008-12-19 04:22:07 +03:00
if ( ! PyArg_ParseTuple ( args , " O " , & py_message ) )
return NULL ;
req = talloc_zero ( NULL , struct ldb_request ) ;
req - > operation = LDB_ADD ;
req - > op . add . message = PyLdbMessage_AsMessage ( py_message ) ;
2008-12-21 06:36:16 +03:00
mod = PyLdbModule_AsModule ( self ) ;
ret = mod - > ops - > add ( mod , req ) ;
2008-12-19 04:22:07 +03:00
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , mod - > ldb ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_module_modify ( PyLdbModuleObject * self , PyObject * args )
{
int ret ;
struct ldb_request * req ;
PyObject * py_message ;
2008-12-21 06:36:16 +03:00
struct ldb_module * mod ;
2008-12-19 04:22:07 +03:00
if ( ! PyArg_ParseTuple ( args , " O " , & py_message ) )
return NULL ;
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
req = talloc_zero ( NULL , struct ldb_request ) ;
req - > operation = LDB_MODIFY ;
req - > op . mod . message = PyLdbMessage_AsMessage ( py_message ) ;
2009-05-31 18:19:11 +04:00
2008-12-21 06:36:16 +03:00
mod = PyLdbModule_AsModule ( self ) ;
ret = mod - > ops - > modify ( mod , req ) ;
2008-12-19 04:22:07 +03:00
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , mod - > ldb ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_module_delete ( PyLdbModuleObject * self , PyObject * args )
{
int ret ;
struct ldb_request * req ;
PyObject * py_dn ;
if ( ! PyArg_ParseTuple ( args , " O " , & py_dn ) )
return NULL ;
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
req = talloc_zero ( NULL , struct ldb_request ) ;
req - > operation = LDB_DELETE ;
req - > op . del . dn = PyLdbDn_AsDn ( py_dn ) ;
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
ret = PyLdbModule_AsModule ( self ) - > ops - > del ( PyLdbModule_AsModule ( self ) , req ) ;
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , NULL ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_ldb_module_rename ( PyLdbModuleObject * self , PyObject * args )
{
int ret ;
struct ldb_request * req ;
PyObject * py_dn1 , * py_dn2 ;
if ( ! PyArg_ParseTuple ( args , " OO " , & py_dn1 , & py_dn2 ) )
return NULL ;
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
req = talloc_zero ( NULL , struct ldb_request ) ;
req - > operation = LDB_RENAME ;
req - > op . rename . olddn = PyLdbDn_AsDn ( py_dn1 ) ;
req - > op . rename . newdn = PyLdbDn_AsDn ( py_dn2 ) ;
2009-05-31 18:19:11 +04:00
2008-12-19 04:22:07 +03:00
ret = PyLdbModule_AsModule ( self ) - > ops - > rename ( PyLdbModule_AsModule ( self ) , req ) ;
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , NULL ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyMethodDef py_ldb_module_methods [ ] = {
{ " search " , ( PyCFunction ) py_ldb_module_search , METH_VARARGS | METH_KEYWORDS , NULL } ,
{ " add " , ( PyCFunction ) py_ldb_module_add , METH_VARARGS , NULL } ,
{ " modify " , ( PyCFunction ) py_ldb_module_modify , METH_VARARGS , NULL } ,
{ " rename " , ( PyCFunction ) py_ldb_module_rename , METH_VARARGS , NULL } ,
{ " delete " , ( PyCFunction ) py_ldb_module_delete , METH_VARARGS , NULL } ,
{ " start_transaction " , ( PyCFunction ) py_ldb_module_start_transaction , METH_NOARGS , NULL } ,
{ " end_transaction " , ( PyCFunction ) py_ldb_module_end_transaction , METH_NOARGS , NULL } ,
{ " del_transaction " , ( PyCFunction ) py_ldb_module_del_transaction , METH_NOARGS , NULL } ,
{ NULL } ,
} ;
2008-12-23 07:07:29 +03:00
static void py_ldb_module_dealloc ( PyLdbModuleObject * self )
{
talloc_free ( self - > mem_ctx ) ;
2010-12-29 17:58:12 +03:00
PyObject_Del ( self ) ;
2008-12-23 07:07:29 +03:00
}
2010-12-30 21:39:14 +03:00
static PyTypeObject PyLdbModule = {
2010-04-04 03:48:35 +04:00
. tp_name = " ldb.LdbModule " ,
2008-12-19 04:22:07 +03:00
. tp_methods = py_ldb_module_methods ,
. tp_repr = ( reprfunc ) py_ldb_module_repr ,
. tp_str = ( reprfunc ) py_ldb_module_str ,
2008-12-23 07:07:29 +03:00
. tp_basicsize = sizeof ( PyLdbModuleObject ) ,
. tp_dealloc = ( destructor ) py_ldb_module_dealloc ,
2008-12-21 05:08:14 +03:00
. tp_flags = Py_TPFLAGS_DEFAULT ,
2008-12-19 04:22:07 +03:00
} ;
2009-03-21 03:00:18 +03:00
/**
* Create a ldb_message_element from a Python object .
*
* This will accept any sequence objects that contains strings , or
* a string object .
*
* A reference to set_obj will be borrowed .
*
* @ param mem_ctx Memory context
* @ param set_obj Python object to convert
* @ param flags ldb_message_element flags to set
* @ param attr_name Name of the attribute
* @ return New ldb_message_element , allocated as child of mem_ctx
*/
2010-12-30 21:39:14 +03:00
static struct ldb_message_element * PyObject_AsMessageElement (
TALLOC_CTX * mem_ctx ,
2010-11-06 18:29:06 +03:00
PyObject * set_obj ,
2011-04-10 21:48:07 +04:00
unsigned int flags ,
2010-11-06 18:29:06 +03:00
const char * attr_name )
2008-12-19 04:22:07 +03:00
{
2008-12-21 00:21:39 +03:00
struct ldb_message_element * me ;
2008-12-19 19:08:35 +03:00
2010-07-15 08:00:48 +04:00
if ( PyLdbMessageElement_Check ( set_obj ) ) {
PyLdbMessageElementObject * set_obj_as_me = ( PyLdbMessageElementObject * ) set_obj ;
2010-12-30 22:11:59 +03:00
/* We have to talloc_reference() the memory context, not the pointer
* which may not actually be it ' s own context */
2010-07-15 08:00:48 +04:00
if ( talloc_reference ( mem_ctx , set_obj_as_me - > mem_ctx ) ) {
return PyLdbMessageElement_AsMessageElement ( set_obj ) ;
}
return NULL ;
}
2008-12-19 19:08:35 +03:00
me = talloc ( mem_ctx , struct ldb_message_element ) ;
2010-12-30 21:59:01 +03:00
if ( me = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2008-12-19 04:22:07 +03:00
2009-08-04 01:58:16 +04:00
me - > name = talloc_strdup ( me , attr_name ) ;
2008-12-21 00:21:39 +03:00
me - > flags = flags ;
if ( PyString_Check ( set_obj ) ) {
me - > num_values = 1 ;
me - > values = talloc_array ( me , struct ldb_val , me - > num_values ) ;
me - > values [ 0 ] . length = PyString_Size ( set_obj ) ;
2009-08-04 01:58:16 +04:00
me - > values [ 0 ] . data = talloc_memdup ( me ,
2010-02-11 15:37:44 +03:00
( uint8_t * ) PyString_AsString ( set_obj ) , me - > values [ 0 ] . length + 1 ) ;
2008-12-21 00:21:39 +03:00
} else if ( PySequence_Check ( set_obj ) ) {
2010-11-06 19:48:39 +03:00
Py_ssize_t i ;
2008-12-21 00:21:39 +03:00
me - > num_values = PySequence_Size ( set_obj ) ;
me - > values = talloc_array ( me , struct ldb_val , me - > num_values ) ;
for ( i = 0 ; i < me - > num_values ; i + + ) {
PyObject * obj = PySequence_GetItem ( set_obj , i ) ;
2010-08-22 08:50:22 +04:00
if ( ! PyString_Check ( obj ) ) {
PyErr_Format ( PyExc_TypeError ,
2010-11-06 20:03:22 +03:00
" Expected string as element %zd in list " , i ) ;
2010-08-22 08:50:22 +04:00
talloc_free ( me ) ;
return NULL ;
}
2008-12-23 07:34:21 +03:00
2008-12-21 00:21:39 +03:00
me - > values [ i ] . length = PyString_Size ( obj ) ;
2009-08-04 01:58:16 +04:00
me - > values [ i ] . data = talloc_memdup ( me ,
2010-02-11 15:37:44 +03:00
( uint8_t * ) PyString_AsString ( obj ) , me - > values [ i ] . length + 1 ) ;
2008-12-21 00:21:39 +03:00
}
} else {
talloc_free ( me ) ;
me = NULL ;
}
return me ;
2008-12-19 04:22:07 +03:00
}
2010-11-06 19:48:39 +03:00
static PyObject * ldb_msg_element_to_set ( struct ldb_context * ldb_ctx ,
struct ldb_message_element * me )
2008-12-19 04:22:07 +03:00
{
2010-11-06 19:48:39 +03:00
Py_ssize_t i ;
2008-12-21 00:21:39 +03:00
PyObject * result ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
/* Python << 2.5 doesn't have PySet_New and PySet_Add. */
result = PyList_New ( me - > num_values ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
for ( i = 0 ; i < me - > num_values ; i + + ) {
PyList_SetItem ( result , i ,
2011-01-11 08:45:39 +03:00
PyObject_FromLdbValue ( & me - > values [ i ] ) ) ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return result ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static PyObject * py_ldb_msg_element_get ( PyLdbMessageElementObject * self , PyObject * args )
2008-12-19 04:22:07 +03:00
{
2010-11-06 19:50:25 +03:00
unsigned int i ;
if ( ! PyArg_ParseTuple ( args , " I " , & i ) )
2008-12-19 04:22:07 +03:00
return NULL ;
2010-11-06 19:50:25 +03:00
if ( i > = PyLdbMessageElement_AsMessageElement ( self ) - > num_values )
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
2011-01-11 08:45:39 +03:00
return PyObject_FromLdbValue ( & ( PyLdbMessageElement_AsMessageElement ( self ) - > values [ i ] ) ) ;
2008-12-19 04:22:07 +03:00
}
2009-08-26 09:01:12 +04:00
static PyObject * py_ldb_msg_element_flags ( PyLdbMessageElementObject * self , PyObject * args )
{
2010-12-30 22:11:59 +03:00
struct ldb_message_element * el = PyLdbMessageElement_AsMessageElement ( self ) ;
2009-08-26 09:01:12 +04:00
return PyInt_FromLong ( el - > flags ) ;
}
static PyObject * py_ldb_msg_element_set_flags ( PyLdbMessageElementObject * self , PyObject * args )
{
2011-04-10 21:48:07 +04:00
unsigned int flags ;
2009-08-26 09:01:12 +04:00
struct ldb_message_element * el ;
2011-04-10 21:48:07 +04:00
if ( ! PyArg_ParseTuple ( args , " I " , & flags ) )
2009-08-26 09:01:12 +04:00
return NULL ;
el = PyLdbMessageElement_AsMessageElement ( self ) ;
el - > flags = flags ;
Py_RETURN_NONE ;
}
2008-12-19 04:22:07 +03:00
static PyMethodDef py_ldb_msg_element_methods [ ] = {
{ " get " , ( PyCFunction ) py_ldb_msg_element_get , METH_VARARGS , NULL } ,
2009-08-26 09:01:12 +04:00
{ " set_flags " , ( PyCFunction ) py_ldb_msg_element_set_flags , METH_VARARGS , NULL } ,
{ " flags " , ( PyCFunction ) py_ldb_msg_element_flags , METH_NOARGS , NULL } ,
2008-12-19 04:22:07 +03:00
{ NULL } ,
} ;
2008-12-19 16:41:44 +03:00
static Py_ssize_t py_ldb_msg_element_len ( PyLdbMessageElementObject * self )
{
return PyLdbMessageElement_AsMessageElement ( self ) - > num_values ;
}
2008-12-19 19:08:35 +03:00
static PyObject * py_ldb_msg_element_find ( PyLdbMessageElementObject * self , Py_ssize_t idx )
{
struct ldb_message_element * el = PyLdbMessageElement_AsMessageElement ( self ) ;
if ( idx < 0 | | idx > = el - > num_values ) {
PyErr_SetString ( PyExc_IndexError , " Out of range " ) ;
return NULL ;
}
return PyString_FromStringAndSize ( ( char * ) el - > values [ idx ] . data , el - > values [ idx ] . length ) ;
}
2008-12-19 16:41:44 +03:00
static PySequenceMethods py_ldb_msg_element_seq = {
. sq_length = ( lenfunc ) py_ldb_msg_element_len ,
2008-12-19 19:08:35 +03:00
. sq_item = ( ssizeargfunc ) py_ldb_msg_element_find ,
2008-12-19 16:41:44 +03:00
} ;
static int py_ldb_msg_element_cmp ( PyLdbMessageElementObject * self , PyLdbMessageElementObject * other )
{
2011-01-03 00:34:26 +03:00
int ret = ldb_msg_element_compare ( PyLdbMessageElement_AsMessageElement ( self ) ,
PyLdbMessageElement_AsMessageElement ( other ) ) ;
return SIGN ( ret ) ;
2008-12-19 16:41:44 +03:00
}
2008-12-19 19:08:35 +03:00
static PyObject * py_ldb_msg_element_iter ( PyLdbMessageElementObject * self )
2008-12-19 16:41:44 +03:00
{
return PyObject_GetIter ( ldb_msg_element_to_set ( NULL , PyLdbMessageElement_AsMessageElement ( self ) ) ) ;
}
2010-12-30 21:39:14 +03:00
static PyObject * PyLdbMessageElement_FromMessageElement ( struct ldb_message_element * el , TALLOC_CTX * mem_ctx )
2008-12-19 19:08:35 +03:00
{
2008-12-23 07:07:29 +03:00
PyLdbMessageElementObject * ret ;
2010-12-30 05:43:23 +03:00
ret = PyObject_New ( PyLdbMessageElementObject , & PyLdbMessageElement ) ;
2008-12-23 07:07:29 +03:00
if ( ret = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2008-12-23 07:34:21 +03:00
ret - > mem_ctx = talloc_new ( NULL ) ;
if ( talloc_reference ( ret - > mem_ctx , mem_ctx ) = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2008-12-23 07:07:29 +03:00
ret - > el = el ;
return ( PyObject * ) ret ;
2008-12-19 19:08:35 +03:00
}
static PyObject * py_ldb_msg_element_new ( PyTypeObject * type , PyObject * args , PyObject * kwargs )
{
PyObject * py_elements = NULL ;
struct ldb_message_element * el ;
2011-04-10 21:48:07 +04:00
unsigned int flags = 0 ;
2008-12-19 19:08:35 +03:00
char * name = NULL ;
2009-02-05 13:04:28 +03:00
const char * const kwnames [ ] = { " elements " , " flags " , " name " , NULL } ;
2008-12-23 07:07:29 +03:00
PyLdbMessageElementObject * ret ;
2009-06-17 22:43:25 +04:00
TALLOC_CTX * mem_ctx ;
2009-02-05 13:04:28 +03:00
2011-04-10 21:48:07 +04:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " |OIs " ,
2009-02-05 13:04:28 +03:00
discard_const_p ( char * , kwnames ) ,
& py_elements , & flags , & name ) )
2008-12-19 19:08:35 +03:00
return NULL ;
2009-06-17 22:43:25 +04:00
mem_ctx = talloc_new ( NULL ) ;
if ( mem_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
el = talloc_zero ( mem_ctx , struct ldb_message_element ) ;
2010-12-30 21:59:01 +03:00
if ( el = = NULL ) {
PyErr_NoMemory ( ) ;
talloc_free ( mem_ctx ) ;
return NULL ;
}
2008-12-19 19:08:35 +03:00
if ( py_elements ! = NULL ) {
2010-11-06 19:48:39 +03:00
Py_ssize_t i ;
2008-12-23 07:34:21 +03:00
if ( PyString_Check ( py_elements ) ) {
2008-12-21 00:21:39 +03:00
el - > num_values = 1 ;
el - > values = talloc_array ( el , struct ldb_val , 1 ) ;
2010-12-30 21:59:01 +03:00
if ( el - > values = = NULL ) {
talloc_free ( mem_ctx ) ;
PyErr_NoMemory ( ) ;
return NULL ;
}
2008-12-21 00:21:39 +03:00
el - > values [ 0 ] . length = PyString_Size ( py_elements ) ;
2010-12-30 21:59:01 +03:00
el - > values [ 0 ] . data = talloc_memdup ( el - > values ,
2010-02-11 15:37:44 +03:00
( uint8_t * ) PyString_AsString ( py_elements ) , el - > values [ 0 ] . length + 1 ) ;
2008-12-23 07:34:21 +03:00
} else if ( PySequence_Check ( py_elements ) ) {
2008-12-21 00:21:39 +03:00
el - > num_values = PySequence_Size ( py_elements ) ;
el - > values = talloc_array ( el , struct ldb_val , el - > num_values ) ;
2010-12-30 21:59:01 +03:00
if ( el - > values = = NULL ) {
talloc_free ( mem_ctx ) ;
PyErr_NoMemory ( ) ;
return NULL ;
}
2008-12-21 00:21:39 +03:00
for ( i = 0 ; i < el - > num_values ; i + + ) {
PyObject * item = PySequence_GetItem ( py_elements , i ) ;
2010-12-30 21:59:01 +03:00
if ( item = = NULL ) {
talloc_free ( mem_ctx ) ;
return NULL ;
}
2009-08-05 05:34:08 +04:00
if ( ! PyString_Check ( item ) ) {
PyErr_Format ( PyExc_TypeError ,
2010-11-06 20:03:22 +03:00
" Expected string as element %zd in list " , i ) ;
2009-08-05 05:34:08 +04:00
talloc_free ( mem_ctx ) ;
return NULL ;
}
2009-08-05 04:35:45 +04:00
el - > values [ i ] . length = PyString_Size ( item ) ;
2011-01-03 03:43:51 +03:00
el - > values [ i ] . data = talloc_memdup ( el ,
2010-02-11 15:37:44 +03:00
( uint8_t * ) PyString_AsString ( item ) , el - > values [ i ] . length + 1 ) ;
2008-12-21 00:21:39 +03:00
}
2008-12-23 07:34:21 +03:00
} else {
PyErr_SetString ( PyExc_TypeError ,
" Expected string or list " ) ;
2009-06-17 22:43:25 +04:00
talloc_free ( mem_ctx ) ;
2008-12-23 07:34:21 +03:00
return NULL ;
2008-12-19 19:08:35 +03:00
}
}
el - > flags = flags ;
el - > name = talloc_strdup ( el , name ) ;
2010-12-30 21:59:01 +03:00
ret = PyObject_New ( PyLdbMessageElementObject , type ) ;
2008-12-23 07:07:29 +03:00
if ( ret = = NULL ) {
2009-06-17 22:43:25 +04:00
talloc_free ( mem_ctx ) ;
2008-12-23 07:07:29 +03:00
return NULL ;
}
2009-06-17 22:43:25 +04:00
ret - > mem_ctx = mem_ctx ;
ret - > el = el ;
2008-12-23 07:07:29 +03:00
return ( PyObject * ) ret ;
2008-12-19 19:08:35 +03:00
}
static PyObject * py_ldb_msg_element_repr ( PyLdbMessageElementObject * self )
{
2008-12-21 00:21:39 +03:00
char * element_str = NULL ;
2010-11-06 19:48:39 +03:00
Py_ssize_t i ;
2008-12-21 00:21:39 +03:00
struct ldb_message_element * el = PyLdbMessageElement_AsMessageElement ( self ) ;
PyObject * ret ;
for ( i = 0 ; i < el - > num_values ; i + + ) {
PyObject * o = py_ldb_msg_element_find ( self , i ) ;
if ( element_str = = NULL )
element_str = talloc_strdup ( NULL , PyObject_REPR ( o ) ) ;
else
element_str = talloc_asprintf_append ( element_str , " ,%s " , PyObject_REPR ( o ) ) ;
}
2010-04-07 22:40:06 +04:00
if ( element_str ! = NULL ) {
ret = PyString_FromFormat ( " MessageElement([%s]) " , element_str ) ;
talloc_free ( element_str ) ;
} else {
ret = PyString_FromString ( " MessageElement([]) " ) ;
}
2008-12-21 00:21:39 +03:00
return ret ;
2008-12-19 19:08:35 +03:00
}
2008-12-21 09:34:27 +03:00
static PyObject * py_ldb_msg_element_str ( PyLdbMessageElementObject * self )
{
struct ldb_message_element * el = PyLdbMessageElement_AsMessageElement ( self ) ;
if ( el - > num_values = = 1 )
return PyString_FromStringAndSize ( ( char * ) el - > values [ 0 ] . data , el - > values [ 0 ] . length ) ;
2010-12-21 16:05:18 +03:00
else
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-21 09:34:27 +03:00
}
2008-12-23 07:07:29 +03:00
static void py_ldb_msg_element_dealloc ( PyLdbMessageElementObject * self )
{
talloc_free ( self - > mem_ctx ) ;
2010-12-29 17:58:12 +03:00
PyObject_Del ( self ) ;
2008-12-23 07:07:29 +03:00
}
2010-12-21 16:05:18 +03:00
static PyTypeObject PyLdbMessageElement = {
2010-04-04 03:48:35 +04:00
. tp_name = " ldb.MessageElement " ,
2008-12-19 04:22:07 +03:00
. tp_basicsize = sizeof ( PyLdbMessageElementObject ) ,
2008-12-23 07:07:29 +03:00
. tp_dealloc = ( destructor ) py_ldb_msg_element_dealloc ,
2008-12-19 19:08:35 +03:00
. tp_repr = ( reprfunc ) py_ldb_msg_element_repr ,
2008-12-21 09:34:27 +03:00
. tp_str = ( reprfunc ) py_ldb_msg_element_str ,
2008-12-19 04:22:07 +03:00
. tp_methods = py_ldb_msg_element_methods ,
2008-12-19 16:41:44 +03:00
. tp_compare = ( cmpfunc ) py_ldb_msg_element_cmp ,
2008-12-19 19:08:35 +03:00
. tp_iter = ( getiterfunc ) py_ldb_msg_element_iter ,
2008-12-19 16:41:44 +03:00
. tp_as_sequence = & py_ldb_msg_element_seq ,
2008-12-19 19:08:35 +03:00
. tp_new = py_ldb_msg_element_new ,
2008-12-21 05:08:14 +03:00
. tp_flags = Py_TPFLAGS_DEFAULT ,
2008-12-19 04:22:07 +03:00
} ;
2010-11-18 23:11:30 +03:00
static PyObject * py_ldb_msg_from_dict ( PyTypeObject * type , PyObject * args )
{
PyObject * py_ldb ;
PyObject * py_dict ;
PyObject * py_ret ;
struct ldb_message * msg ;
struct ldb_context * ldb_ctx ;
unsigned int mod_flags = LDB_FLAG_MOD_REPLACE ;
if ( ! PyArg_ParseTuple ( args , " O!O!|I " ,
& PyLdb , & py_ldb , & PyDict_Type , & py_dict ,
& mod_flags ) ) {
return NULL ;
}
/* mask only flags we are going to use */
mod_flags = LDB_FLAG_MOD_TYPE ( mod_flags ) ;
if ( ! mod_flags ) {
PyErr_SetString ( PyExc_ValueError ,
" FLAG_MOD_ADD, FLAG_MOD_REPLACE or FLAG_MOD_DELETE "
" expected as mod_flag value " ) ;
return NULL ;
}
ldb_ctx = PyLdb_AsLdbContext ( py_ldb ) ;
msg = PyDict_AsMessage ( ldb_ctx , py_dict , ldb_ctx , mod_flags ) ;
if ( ! msg ) {
return NULL ;
}
py_ret = PyLdbMessage_FromMessage ( msg ) ;
talloc_unlink ( ldb_ctx , msg ) ;
return py_ret ;
}
2008-12-19 04:22:07 +03:00
static PyObject * py_ldb_msg_remove_attr ( PyLdbMessageObject * self , PyObject * args )
{
char * name ;
if ( ! PyArg_ParseTuple ( args , " s " , & name ) )
return NULL ;
2008-12-23 07:07:29 +03:00
ldb_msg_remove_attr ( self - > msg , name ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
2008-12-19 16:41:44 +03:00
static PyObject * py_ldb_msg_keys ( PyLdbMessageObject * self )
{
struct ldb_message * msg = PyLdbMessage_AsMessage ( self ) ;
2010-11-06 19:48:39 +03:00
Py_ssize_t i , j = 0 ;
2008-12-19 16:41:44 +03:00
PyObject * obj = PyList_New ( msg - > num_elements + ( msg - > dn ! = NULL ? 1 : 0 ) ) ;
if ( msg - > dn ! = NULL ) {
PyList_SetItem ( obj , j , PyString_FromString ( " dn " ) ) ;
j + + ;
}
for ( i = 0 ; i < msg - > num_elements ; i + + ) {
PyList_SetItem ( obj , j , PyString_FromString ( msg - > elements [ i ] . name ) ) ;
j + + ;
}
return obj ;
}
2008-12-21 00:21:39 +03:00
static PyObject * py_ldb_msg_getitem_helper ( PyLdbMessageObject * self , PyObject * py_name )
2008-12-19 04:22:07 +03:00
{
struct ldb_message_element * el ;
2009-09-14 19:03:30 +04:00
char * name ;
2008-12-23 07:07:29 +03:00
struct ldb_message * msg = PyLdbMessage_AsMessage ( self ) ;
2009-09-14 19:03:30 +04:00
if ( ! PyString_Check ( py_name ) ) {
PyErr_SetNone ( PyExc_TypeError ) ;
return NULL ;
}
name = PyString_AsString ( py_name ) ;
2008-12-19 19:08:35 +03:00
if ( ! strcmp ( name , " dn " ) )
2008-12-23 07:07:29 +03:00
return PyLdbDn_FromDn ( msg - > dn ) ;
el = ldb_msg_find_element ( msg , name ) ;
2008-12-21 00:21:39 +03:00
if ( el = = NULL ) {
2008-12-19 04:22:07 +03:00
return NULL ;
2008-12-21 00:21:39 +03:00
}
2010-07-15 08:00:48 +04:00
return ( PyObject * ) PyLdbMessageElement_FromMessageElement ( el , msg - > elements ) ;
2008-12-19 04:22:07 +03:00
}
2008-12-21 00:21:39 +03:00
static PyObject * py_ldb_msg_getitem ( PyLdbMessageObject * self , PyObject * py_name )
{
PyObject * ret = py_ldb_msg_getitem_helper ( self , py_name ) ;
if ( ret = = NULL ) {
PyErr_SetString ( PyExc_KeyError , " No such element " ) ;
return NULL ;
}
return ret ;
}
2008-12-19 19:08:35 +03:00
static PyObject * py_ldb_msg_get ( PyLdbMessageObject * self , PyObject * args )
{
2011-06-11 16:57:02 +04:00
PyObject * name , * ret , * retobj ;
retobj = NULL ;
if ( ! PyArg_ParseTuple ( args , " O|O " , & name , & retobj ) )
2008-12-19 19:08:35 +03:00
return NULL ;
2008-12-21 00:21:39 +03:00
ret = py_ldb_msg_getitem_helper ( self , name ) ;
2009-09-14 19:03:30 +04:00
if ( ret = = NULL ) {
if ( PyErr_Occurred ( ) )
return NULL ;
2011-06-11 16:57:02 +04:00
if ( retobj ! = NULL ) {
return retobj ;
} else {
Py_RETURN_NONE ;
}
2009-09-14 19:03:30 +04:00
}
2008-12-19 19:08:35 +03:00
return ret ;
}
static PyObject * py_ldb_msg_items ( PyLdbMessageObject * self )
{
struct ldb_message * msg = PyLdbMessage_AsMessage ( self ) ;
2010-11-06 19:48:39 +03:00
Py_ssize_t i , j = 0 ;
2008-12-19 19:08:35 +03:00
PyObject * l = PyList_New ( msg - > num_elements + ( msg - > dn = = NULL ? 0 : 1 ) ) ;
if ( msg - > dn ! = NULL ) {
PyList_SetItem ( l , 0 , Py_BuildValue ( " (sO) " , " dn " , PyLdbDn_FromDn ( msg - > dn ) ) ) ;
j + + ;
}
for ( i = 0 ; i < msg - > num_elements ; i + + , j + + ) {
2010-12-30 21:59:01 +03:00
PyObject * py_el = PyLdbMessageElement_FromMessageElement ( & msg - > elements [ i ] , msg - > elements ) ;
PyObject * value = Py_BuildValue ( " (sO) " , msg - > elements [ i ] . name , py_el ) ;
PyList_SetItem ( l , j , value ) ;
2008-12-19 19:08:35 +03:00
}
return l ;
}
2010-12-21 16:05:18 +03:00
static PyObject * py_ldb_msg_elements ( PyLdbMessageObject * self )
{
struct ldb_message * msg = PyLdbMessage_AsMessage ( self ) ;
Py_ssize_t i = 0 ;
PyObject * l = PyList_New ( msg - > num_elements ) ;
for ( i = 0 ; i < msg - > num_elements ; i + + ) {
PyList_SetItem ( l , i , PyLdbMessageElement_FromMessageElement ( & msg - > elements [ i ] , msg - > elements ) ) ;
}
return l ;
}
static PyObject * py_ldb_msg_add ( PyLdbMessageObject * self , PyObject * args )
{
struct ldb_message * msg = PyLdbMessage_AsMessage ( self ) ;
2010-12-30 22:11:59 +03:00
PyLdbMessageElementObject * py_element ;
2010-12-22 11:56:01 +03:00
int ret ;
2010-12-21 16:05:18 +03:00
struct ldb_message_element * el ;
if ( ! PyArg_ParseTuple ( args , " O! " , & PyLdbMessageElement , & py_element ) )
return NULL ;
2011-01-03 03:43:51 +03:00
el = talloc_reference ( msg , py_element - > el ) ;
2010-12-21 16:05:18 +03:00
if ( el = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
ret = ldb_msg_add ( msg , el , el - > flags ) ;
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , NULL ) ;
Py_RETURN_NONE ;
}
static PyMethodDef py_ldb_msg_methods [ ] = {
2010-11-18 23:11:30 +03:00
{ " from_dict " , ( PyCFunction ) py_ldb_msg_from_dict , METH_CLASS | METH_VARARGS ,
2010-12-21 16:05:18 +03:00
" Message.from_dict(ldb, dict, mod_flag=FLAG_MOD_REPLACE) -> ldb.Message \n "
2010-11-18 23:11:30 +03:00
" Class method to create ldb.Message object from Dictionary. \n "
2010-12-21 16:05:18 +03:00
" mod_flag is one of FLAG_MOD_ADD, FLAG_MOD_REPLACE or FLAG_MOD_DELETE. " } ,
{ " keys " , ( PyCFunction ) py_ldb_msg_keys , METH_NOARGS ,
" S.keys() -> list \n \n "
" Return sequence of all attribute names. " } ,
{ " remove " , ( PyCFunction ) py_ldb_msg_remove_attr , METH_VARARGS ,
" S.remove(name) \n \n "
" Remove all entries for attributes with the specified name. " } ,
2008-12-19 19:08:35 +03:00
{ " get " , ( PyCFunction ) py_ldb_msg_get , METH_VARARGS , NULL } ,
{ " items " , ( PyCFunction ) py_ldb_msg_items , METH_NOARGS , NULL } ,
2010-12-21 16:05:18 +03:00
{ " elements " , ( PyCFunction ) py_ldb_msg_elements , METH_NOARGS , NULL } ,
{ " add " , ( PyCFunction ) py_ldb_msg_add , METH_VARARGS ,
" S.append(element) \n \n "
" Add an element to this message. " } ,
2008-12-19 19:08:35 +03:00
{ NULL } ,
} ;
2008-12-21 06:36:16 +03:00
static PyObject * py_ldb_msg_iter ( PyLdbMessageObject * self )
{
PyObject * list , * iter ;
list = py_ldb_msg_keys ( self ) ;
iter = PyObject_GetIter ( list ) ;
Py_DECREF ( list ) ;
return iter ;
}
2008-12-19 19:08:35 +03:00
static int py_ldb_msg_setitem ( PyLdbMessageObject * self , PyObject * name , PyObject * value )
2008-12-19 04:22:07 +03:00
{
2009-08-04 01:58:16 +04:00
char * attr_name ;
if ( ! PyString_Check ( name ) ) {
PyErr_SetNone ( PyExc_TypeError ) ;
return - 1 ;
}
2010-12-21 16:05:18 +03:00
2009-08-04 01:58:16 +04:00
attr_name = PyString_AsString ( name ) ;
2008-12-19 04:22:07 +03:00
if ( value = = NULL ) {
2009-08-04 01:58:16 +04:00
/* delitem */
2008-12-23 07:07:29 +03:00
ldb_msg_remove_attr ( self - > msg , attr_name ) ;
2008-12-19 04:22:07 +03:00
} else {
2009-08-04 01:58:16 +04:00
struct ldb_message_element * el = PyObject_AsMessageElement ( self - > msg ,
2010-07-15 08:01:56 +04:00
value , 0 , attr_name ) ;
2008-12-19 19:08:35 +03:00
if ( el = = NULL )
return - 1 ;
ldb_msg_remove_attr ( PyLdbMessage_AsMessage ( self ) , attr_name ) ;
ldb_msg_add ( PyLdbMessage_AsMessage ( self ) , el , el - > flags ) ;
2008-12-19 04:22:07 +03:00
}
2008-12-19 19:08:35 +03:00
return 0 ;
2008-12-19 04:22:07 +03:00
}
2008-12-19 16:41:44 +03:00
static Py_ssize_t py_ldb_msg_length ( PyLdbMessageObject * self )
{
return PyLdbMessage_AsMessage ( self ) - > num_elements ;
}
2008-12-19 04:22:07 +03:00
static PyMappingMethods py_ldb_msg_mapping = {
2008-12-19 16:41:44 +03:00
. mp_length = ( lenfunc ) py_ldb_msg_length ,
2008-12-19 04:22:07 +03:00
. mp_subscript = ( binaryfunc ) py_ldb_msg_getitem ,
. mp_ass_subscript = ( objobjargproc ) py_ldb_msg_setitem ,
} ;
static PyObject * py_ldb_msg_new ( PyTypeObject * type , PyObject * args , PyObject * kwargs )
{
2009-02-05 13:04:28 +03:00
const char * const kwnames [ ] = { " dn " , NULL } ;
2008-12-19 19:08:35 +03:00
struct ldb_message * ret ;
2009-08-03 20:15:16 +04:00
TALLOC_CTX * mem_ctx ;
2008-12-19 19:08:35 +03:00
PyObject * pydn = NULL ;
2008-12-23 07:07:29 +03:00
PyLdbMessageObject * py_ret ;
2009-02-05 13:04:28 +03:00
if ( ! PyArg_ParseTupleAndKeywords ( args , kwargs , " |O " ,
discard_const_p ( char * , kwnames ) ,
& pydn ) )
2008-12-19 04:22:07 +03:00
return NULL ;
2009-08-03 20:15:16 +04:00
mem_ctx = talloc_new ( NULL ) ;
if ( mem_ctx = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
ret = ldb_msg_new ( mem_ctx ) ;
2008-12-19 19:08:35 +03:00
if ( ret = = NULL ) {
2009-08-03 20:15:16 +04:00
talloc_free ( mem_ctx ) ;
2008-12-19 19:08:35 +03:00
PyErr_NoMemory ( ) ;
return NULL ;
}
2009-06-17 21:01:06 +04:00
if ( pydn ! = NULL ) {
2009-06-17 22:17:35 +04:00
struct ldb_dn * dn ;
if ( ! PyObject_AsDn ( NULL , pydn , NULL , & dn ) ) {
2009-08-03 20:15:16 +04:00
talloc_free ( mem_ctx ) ;
2008-12-19 19:08:35 +03:00
return NULL ;
2009-06-17 21:01:06 +04:00
}
2009-06-17 22:17:35 +04:00
ret - > dn = talloc_reference ( ret , dn ) ;
2009-06-17 21:01:06 +04:00
}
2008-12-19 19:08:35 +03:00
2008-12-23 07:07:29 +03:00
py_ret = ( PyLdbMessageObject * ) type - > tp_alloc ( type , 0 ) ;
if ( py_ret = = NULL ) {
PyErr_NoMemory ( ) ;
2009-08-03 20:15:16 +04:00
talloc_free ( mem_ctx ) ;
2008-12-23 07:07:29 +03:00
return NULL ;
}
2009-08-03 20:15:16 +04:00
py_ret - > mem_ctx = mem_ctx ;
py_ret - > msg = ret ;
2008-12-23 07:07:29 +03:00
return ( PyObject * ) py_ret ;
2008-12-19 04:22:07 +03:00
}
2010-12-30 21:39:14 +03:00
static PyObject * PyLdbMessage_FromMessage ( struct ldb_message * msg )
2008-12-19 04:22:07 +03:00
{
2008-12-23 07:07:29 +03:00
PyLdbMessageObject * ret ;
ret = ( PyLdbMessageObject * ) PyLdbMessage . tp_alloc ( & PyLdbMessage , 0 ) ;
if ( ret = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
ret - > mem_ctx = talloc_new ( NULL ) ;
ret - > msg = talloc_reference ( ret - > mem_ctx , msg ) ;
return ( PyObject * ) ret ;
2008-12-19 04:22:07 +03:00
}
2008-12-19 19:08:35 +03:00
static PyObject * py_ldb_msg_get_dn ( PyLdbMessageObject * self , void * closure )
{
2009-06-17 22:17:35 +04:00
struct ldb_message * msg = PyLdbMessage_AsMessage ( self ) ;
return PyLdbDn_FromDn ( msg - > dn ) ;
2008-12-19 19:08:35 +03:00
}
static int py_ldb_msg_set_dn ( PyLdbMessageObject * self , PyObject * value , void * closure )
{
2009-06-17 22:17:35 +04:00
struct ldb_message * msg = PyLdbMessage_AsMessage ( self ) ;
2009-08-03 20:15:16 +04:00
if ( ! PyLdbDn_Check ( value ) ) {
PyErr_SetNone ( PyExc_TypeError ) ;
return - 1 ;
}
2009-06-17 22:17:35 +04:00
msg - > dn = talloc_reference ( msg , PyLdbDn_AsDn ( value ) ) ;
2008-12-19 19:08:35 +03:00
return 0 ;
}
static PyGetSetDef py_ldb_msg_getset [ ] = {
2009-02-05 13:04:28 +03:00
{ discard_const_p ( char , " dn " ) , ( getter ) py_ldb_msg_get_dn , ( setter ) py_ldb_msg_set_dn , NULL } ,
2008-12-19 19:08:35 +03:00
{ NULL }
} ;
2008-12-21 00:21:39 +03:00
static PyObject * py_ldb_msg_repr ( PyLdbMessageObject * self )
{
PyObject * dict = PyDict_New ( ) , * ret ;
if ( PyDict_Update ( dict , ( PyObject * ) self ) ! = 0 )
return NULL ;
ret = PyString_FromFormat ( " Message(%s) " , PyObject_REPR ( dict ) ) ;
Py_DECREF ( dict ) ;
return ret ;
}
2008-12-23 07:07:29 +03:00
static void py_ldb_msg_dealloc ( PyLdbMessageObject * self )
{
talloc_free ( self - > mem_ctx ) ;
2010-12-29 17:58:12 +03:00
PyObject_Del ( self ) ;
2008-12-23 07:07:29 +03:00
}
2010-05-07 04:15:28 +04:00
static int py_ldb_msg_compare ( PyLdbMessageObject * py_msg1 ,
PyLdbMessageObject * py_msg2 )
{
struct ldb_message * msg1 = PyLdbMessage_AsMessage ( py_msg1 ) ,
* msg2 = PyLdbMessage_AsMessage ( py_msg2 ) ;
unsigned int i ;
int ret ;
2010-06-07 23:26:33 +04:00
if ( ( msg1 - > dn ! = NULL ) | | ( msg2 - > dn ! = NULL ) ) {
ret = ldb_dn_compare ( msg1 - > dn , msg2 - > dn ) ;
if ( ret ! = 0 ) {
2011-01-03 00:34:26 +03:00
return SIGN ( ret ) ;
2010-06-07 23:26:33 +04:00
}
2010-05-07 04:15:28 +04:00
}
ret = msg1 - > num_elements - msg2 - > num_elements ;
if ( ret ! = 0 ) {
2011-01-03 00:34:26 +03:00
return SIGN ( ret ) ;
2010-05-07 04:15:28 +04:00
}
for ( i = 0 ; i < msg1 - > num_elements ; i + + ) {
ret = ldb_msg_element_compare_name ( & msg1 - > elements [ i ] ,
& msg2 - > elements [ i ] ) ;
if ( ret ! = 0 ) {
2011-01-03 00:34:26 +03:00
return SIGN ( ret ) ;
2010-05-07 04:15:28 +04:00
}
ret = ldb_msg_element_compare ( & msg1 - > elements [ i ] ,
& msg2 - > elements [ i ] ) ;
if ( ret ! = 0 ) {
2011-01-03 00:34:26 +03:00
return SIGN ( ret ) ;
2010-05-07 04:15:28 +04:00
}
}
return 0 ;
}
2010-12-30 21:39:14 +03:00
static PyTypeObject PyLdbMessage = {
2010-04-04 03:48:35 +04:00
. tp_name = " ldb.Message " ,
2008-12-19 04:22:07 +03:00
. tp_methods = py_ldb_msg_methods ,
2008-12-19 19:08:35 +03:00
. tp_getset = py_ldb_msg_getset ,
2008-12-19 04:22:07 +03:00
. tp_as_mapping = & py_ldb_msg_mapping ,
. tp_basicsize = sizeof ( PyLdbMessageObject ) ,
2008-12-23 07:07:29 +03:00
. tp_dealloc = ( destructor ) py_ldb_msg_dealloc ,
2008-12-19 04:22:07 +03:00
. tp_new = py_ldb_msg_new ,
2008-12-21 00:21:39 +03:00
. tp_repr = ( reprfunc ) py_ldb_msg_repr ,
2008-12-21 05:08:14 +03:00
. tp_flags = Py_TPFLAGS_DEFAULT ,
2008-12-21 06:36:16 +03:00
. tp_iter = ( getiterfunc ) py_ldb_msg_iter ,
2010-05-07 04:15:28 +04:00
. tp_compare = ( cmpfunc ) py_ldb_msg_compare ,
2008-12-19 04:22:07 +03:00
} ;
2010-12-30 21:39:14 +03:00
static PyObject * PyLdbTree_FromTree ( struct ldb_parse_tree * tree )
2008-12-19 19:08:35 +03:00
{
2008-12-23 07:07:29 +03:00
PyLdbTreeObject * ret ;
ret = ( PyLdbTreeObject * ) PyLdbTree . tp_alloc ( & PyLdbTree , 0 ) ;
if ( ret = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2009-05-31 18:19:11 +04:00
2008-12-23 07:07:29 +03:00
ret - > mem_ctx = talloc_new ( NULL ) ;
ret - > tree = talloc_reference ( ret - > mem_ctx , tree ) ;
return ( PyObject * ) ret ;
}
static void py_ldb_tree_dealloc ( PyLdbTreeObject * self )
{
talloc_free ( self - > mem_ctx ) ;
2010-12-29 17:58:12 +03:00
PyObject_Del ( self ) ;
2008-12-19 19:08:35 +03:00
}
2010-12-30 21:39:14 +03:00
static PyTypeObject PyLdbTree = {
2010-04-04 03:48:35 +04:00
. tp_name = " ldb.Tree " ,
2008-12-19 19:08:35 +03:00
. tp_basicsize = sizeof ( PyLdbTreeObject ) ,
2008-12-23 07:07:29 +03:00
. tp_dealloc = ( destructor ) py_ldb_tree_dealloc ,
2008-12-21 05:08:14 +03:00
. tp_flags = Py_TPFLAGS_DEFAULT ,
2008-12-19 19:08:35 +03:00
} ;
2008-12-19 04:22:07 +03:00
/* Ldb_module */
2008-12-23 07:07:29 +03:00
static int py_module_search ( struct ldb_module * mod , struct ldb_request * req )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result , * py_base , * py_attrs , * py_tree ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
py_base = PyLdbDn_FromDn ( req - > op . search . base ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_base = = NULL )
return LDB_ERR_OPERATIONS_ERROR ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
py_tree = PyLdbTree_FromTree ( req - > op . search . tree ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_tree = = NULL )
return LDB_ERR_OPERATIONS_ERROR ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( req - > op . search . attrs = = NULL ) {
py_attrs = Py_None ;
} else {
int i , len ;
for ( len = 0 ; req - > op . search . attrs [ len ] ; len + + ) ;
py_attrs = PyList_New ( len ) ;
for ( i = 0 ; i < len ; i + + )
PyList_SetItem ( py_attrs , i , PyString_FromString ( req - > op . search . attrs [ i ] ) ) ;
}
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " search " ) ,
discard_const_p ( char , " OiOO " ) ,
py_base , req - > op . search . scope , py_tree , py_attrs ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_attrs ) ;
Py_DECREF ( py_tree ) ;
Py_DECREF ( py_base ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_result = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-19 19:08:35 +03:00
req - > op . search . res = PyLdbResult_AsResult ( NULL , py_result ) ;
2008-12-19 04:22:07 +03:00
if ( req - > op . search . res = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_result ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_SUCCESS ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static int py_module_add ( struct ldb_module * mod , struct ldb_request * req )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result , * py_msg ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_msg = PyLdbMessage_FromMessage ( discard_const_p ( struct ldb_message , req - > op . add . message ) ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_msg = = NULL ) {
return LDB_ERR_OPERATIONS_ERROR ;
}
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " add " ) ,
discard_const_p ( char , " O " ) ,
py_msg ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_msg ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_result = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_result ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_SUCCESS ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static int py_module_modify ( struct ldb_module * mod , struct ldb_request * req )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result , * py_msg ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_msg = PyLdbMessage_FromMessage ( discard_const_p ( struct ldb_message , req - > op . mod . message ) ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_msg = = NULL ) {
return LDB_ERR_OPERATIONS_ERROR ;
}
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " modify " ) ,
discard_const_p ( char , " O " ) ,
py_msg ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_msg ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_result = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_result ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_SUCCESS ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static int py_module_del ( struct ldb_module * mod , struct ldb_request * req )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result , * py_dn ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
py_dn = PyLdbDn_FromDn ( req - > op . del . dn ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_dn = = NULL )
return LDB_ERR_OPERATIONS_ERROR ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " delete " ) ,
discard_const_p ( char , " O " ) ,
py_dn ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_result = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_result ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_SUCCESS ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static int py_module_rename ( struct ldb_module * mod , struct ldb_request * req )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result , * py_olddn , * py_newdn ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
py_olddn = PyLdbDn_FromDn ( req - > op . rename . olddn ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_olddn = = NULL )
return LDB_ERR_OPERATIONS_ERROR ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
py_newdn = PyLdbDn_FromDn ( req - > op . rename . newdn ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_newdn = = NULL )
return LDB_ERR_OPERATIONS_ERROR ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " rename " ) ,
discard_const_p ( char , " OO " ) ,
py_olddn , py_newdn ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_olddn ) ;
Py_DECREF ( py_newdn ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_result = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_result ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_SUCCESS ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static int py_module_request ( struct ldb_module * mod , struct ldb_request * req )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " request " ) ,
discard_const_p ( char , " " ) ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_ERR_OPERATIONS_ERROR ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static int py_module_extended ( struct ldb_module * mod , struct ldb_request * req )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " extended " ) ,
discard_const_p ( char , " " ) ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_ERR_OPERATIONS_ERROR ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static int py_module_start_transaction ( struct ldb_module * mod )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " start_transaction " ) ,
discard_const_p ( char , " " ) ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_result = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_result ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_SUCCESS ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static int py_module_end_transaction ( struct ldb_module * mod )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " end_transaction " ) ,
discard_const_p ( char , " " ) ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_result = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_result ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_SUCCESS ;
2008-12-19 04:22:07 +03:00
}
2008-12-23 07:07:29 +03:00
static int py_module_del_transaction ( struct ldb_module * mod )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_ldb = ( PyObject * ) mod - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallMethod ( py_ldb , discard_const_p ( char , " del_transaction " ) ,
discard_const_p ( char , " " ) ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_result = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
Py_DECREF ( py_result ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return LDB_SUCCESS ;
2008-12-19 04:22:07 +03:00
}
2008-12-19 19:08:35 +03:00
static int py_module_destructor ( struct ldb_module * mod )
2008-12-19 04:22:07 +03:00
{
2008-12-21 00:21:39 +03:00
Py_DECREF ( ( PyObject * ) mod - > private_data ) ;
return 0 ;
2008-12-19 04:22:07 +03:00
}
2009-02-05 13:04:28 +03:00
static int py_module_init ( struct ldb_module * mod )
2008-12-19 04:22:07 +03:00
{
2009-02-05 13:04:28 +03:00
PyObject * py_class = ( PyObject * ) mod - > ops - > private_data ;
2008-12-21 00:21:39 +03:00
PyObject * py_result , * py_next , * py_ldb ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
py_ldb = PyLdb_FromLdbContext ( mod - > ldb ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_ldb = = NULL )
return LDB_ERR_OPERATIONS_ERROR ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
py_next = PyLdbModule_FromModule ( mod - > next ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_next = = NULL )
return LDB_ERR_OPERATIONS_ERROR ;
2008-12-19 04:22:07 +03:00
2009-02-05 13:04:28 +03:00
py_result = PyObject_CallFunction ( py_class , discard_const_p ( char , " OO " ) ,
py_ldb , py_next ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
if ( py_result = = NULL ) {
2008-12-21 06:36:16 +03:00
return LDB_ERR_PYTHON_EXCEPTION ;
2008-12-21 00:21:39 +03:00
}
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
mod - > private_data = py_result ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
talloc_set_destructor ( mod , py_module_destructor ) ;
2008-12-19 04:22:07 +03:00
2008-12-21 00:21:39 +03:00
return ldb_next_init ( mod ) ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_register_module ( PyObject * module , PyObject * args )
{
int ret ;
struct ldb_module_ops * ops ;
PyObject * input ;
if ( ! PyArg_ParseTuple ( args , " O " , & input ) )
return NULL ;
2008-12-21 00:21:39 +03:00
ops = talloc_zero ( talloc_autofree_context ( ) , struct ldb_module_ops ) ;
2008-12-19 04:22:07 +03:00
if ( ops = = NULL ) {
PyErr_NoMemory ( ) ;
return NULL ;
}
2009-02-05 13:04:28 +03:00
ops - > name = talloc_strdup ( ops , PyString_AsString ( PyObject_GetAttrString ( input , discard_const_p ( char , " name " ) ) ) ) ;
2008-12-21 00:21:39 +03:00
Py_INCREF ( input ) ;
ops - > private_data = input ;
ops - > init_context = py_module_init ;
ops - > search = py_module_search ;
ops - > add = py_module_add ;
ops - > modify = py_module_modify ;
ops - > del = py_module_del ;
ops - > rename = py_module_rename ;
ops - > request = py_module_request ;
ops - > extended = py_module_extended ;
ops - > start_transaction = py_module_start_transaction ;
ops - > end_transaction = py_module_end_transaction ;
ops - > del_transaction = py_module_del_transaction ;
2008-12-19 04:22:07 +03:00
ret = ldb_register_module ( ops ) ;
2009-04-23 03:21:47 +04:00
PyErr_LDB_ERROR_IS_ERR_RAISE ( PyExc_LdbError , ret , NULL ) ;
2008-12-19 04:22:07 +03:00
2009-01-06 06:13:57 +03:00
Py_RETURN_NONE ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_timestring ( PyObject * module , PyObject * args )
{
2010-11-06 18:29:27 +03:00
/* most times "time_t" is a signed integer type with 32 or 64 bit:
* http : //stackoverflow.com/questions/471248/what-is-ultimately-a-time-t-typedef-to */
long int t_val ;
2008-12-19 04:22:07 +03:00
char * tresult ;
PyObject * ret ;
2010-11-06 18:29:27 +03:00
if ( ! PyArg_ParseTuple ( args , " l " , & t_val ) )
2008-12-19 04:22:07 +03:00
return NULL ;
2010-11-06 18:29:27 +03:00
tresult = ldb_timestring ( NULL , ( time_t ) t_val ) ;
2008-12-21 00:21:39 +03:00
ret = PyString_FromString ( tresult ) ;
talloc_free ( tresult ) ;
return ret ;
2008-12-19 04:22:07 +03:00
}
static PyObject * py_string_to_time ( PyObject * module , PyObject * args )
{
char * str ;
if ( ! PyArg_ParseTuple ( args , " s " , & str ) )
return NULL ;
return PyInt_FromLong ( ldb_string_to_time ( str ) ) ;
}
static PyObject * py_valid_attr_name ( PyObject * self , PyObject * args )
{
char * name ;
if ( ! PyArg_ParseTuple ( args , " s " , & name ) )
return NULL ;
return PyBool_FromLong ( ldb_valid_attr_name ( name ) ) ;
}
2011-07-28 11:03:06 +04:00
/*
encode a string using RFC2254 rules
*/
static PyObject * py_binary_encode ( PyObject * self , PyObject * args )
{
char * str , * encoded ;
Py_ssize_t size ;
struct ldb_val val ;
PyObject * ret ;
if ( ! PyArg_ParseTuple ( args , " s# " , & str , & size ) )
return NULL ;
val . data = ( uint8_t * ) str ;
val . length = size ;
encoded = ldb_binary_encode ( NULL , val ) ;
if ( encoded = = NULL ) {
PyErr_SetString ( PyExc_TypeError , " unable to encode binary string " ) ;
return NULL ;
}
ret = PyString_FromString ( encoded ) ;
talloc_free ( encoded ) ;
return ret ;
}
/*
decode a string using RFC2254 rules
*/
static PyObject * py_binary_decode ( PyObject * self , PyObject * args )
{
char * str ;
struct ldb_val val ;
PyObject * ret ;
if ( ! PyArg_ParseTuple ( args , " s " , & str ) )
return NULL ;
val = ldb_binary_decode ( NULL , str ) ;
if ( val . data = = NULL ) {
PyErr_SetString ( PyExc_TypeError , " unable to decode binary string " ) ;
return NULL ;
}
ret = Py_BuildValue ( " s# " , val . data , val . length ) ;
talloc_free ( val . data ) ;
return ret ;
}
2008-12-19 04:22:07 +03:00
static PyMethodDef py_ldb_global_methods [ ] = {
{ " register_module " , py_register_module , METH_VARARGS ,
" S.register_module(module) -> None \n "
" Register a LDB module. " } ,
{ " timestring " , py_timestring , METH_VARARGS ,
" S.timestring(int) -> string \n "
" Generate a LDAP time string from a UNIX timestamp " } ,
{ " string_to_time " , py_string_to_time , METH_VARARGS ,
" S.string_to_time(string) -> int \n "
" Parse a LDAP time string into a UNIX timestamp. " } ,
{ " valid_attr_name " , py_valid_attr_name , METH_VARARGS ,
" S.valid_attr_name(name) -> bool \n "
" Check whether the supplied name is a valid attribute name. " } ,
{ " open " , ( PyCFunction ) py_ldb_new , METH_VARARGS | METH_KEYWORDS ,
NULL } ,
2011-07-28 11:03:06 +04:00
{ " binary_encode " , py_binary_encode , METH_VARARGS ,
" S.binary_encode(string) -> string \n "
" Perform a RFC2254 binary encoding on a string " } ,
{ " binary_decode " , py_binary_decode , METH_VARARGS ,
" S.binary_decode(string) -> string \n "
" Perform a RFC2254 binary decode on a string " } ,
2008-12-19 04:22:07 +03:00
{ NULL }
} ;
void initldb ( void )
{
PyObject * m ;
if ( PyType_Ready ( & PyLdbDn ) < 0 )
return ;
if ( PyType_Ready ( & PyLdbMessage ) < 0 )
return ;
if ( PyType_Ready ( & PyLdbMessageElement ) < 0 )
return ;
if ( PyType_Ready ( & PyLdb ) < 0 )
return ;
if ( PyType_Ready ( & PyLdbModule ) < 0 )
return ;
2008-12-19 19:08:35 +03:00
if ( PyType_Ready ( & PyLdbTree ) < 0 )
return ;
2011-02-07 09:50:36 +03:00
if ( PyType_Ready ( & PyLdbResult ) < 0 )
return ;
if ( PyType_Ready ( & PyLdbControl ) < 0 )
return ;
2008-12-19 04:22:07 +03:00
m = Py_InitModule3 ( " ldb " , py_ldb_global_methods ,
" An interface to LDB, a LDAP-like API that can either to talk an embedded database (TDB-based) or a standards-compliant LDAP server. " ) ;
if ( m = = NULL )
return ;
2010-06-21 10:55:18 +04:00
PyModule_AddObject ( m , " SEQ_HIGHEST_SEQ " , PyInt_FromLong ( LDB_SEQ_HIGHEST_SEQ ) ) ;
PyModule_AddObject ( m , " SEQ_HIGHEST_TIMESTAMP " , PyInt_FromLong ( LDB_SEQ_HIGHEST_TIMESTAMP ) ) ;
PyModule_AddObject ( m , " SEQ_NEXT " , PyInt_FromLong ( LDB_SEQ_NEXT ) ) ;
2008-12-19 04:22:07 +03:00
PyModule_AddObject ( m , " SCOPE_DEFAULT " , PyInt_FromLong ( LDB_SCOPE_DEFAULT ) ) ;
PyModule_AddObject ( m , " SCOPE_BASE " , PyInt_FromLong ( LDB_SCOPE_BASE ) ) ;
PyModule_AddObject ( m , " SCOPE_ONELEVEL " , PyInt_FromLong ( LDB_SCOPE_ONELEVEL ) ) ;
PyModule_AddObject ( m , " SCOPE_SUBTREE " , PyInt_FromLong ( LDB_SCOPE_SUBTREE ) ) ;
PyModule_AddObject ( m , " CHANGETYPE_NONE " , PyInt_FromLong ( LDB_CHANGETYPE_NONE ) ) ;
PyModule_AddObject ( m , " CHANGETYPE_ADD " , PyInt_FromLong ( LDB_CHANGETYPE_ADD ) ) ;
PyModule_AddObject ( m , " CHANGETYPE_DELETE " , PyInt_FromLong ( LDB_CHANGETYPE_DELETE ) ) ;
PyModule_AddObject ( m , " CHANGETYPE_MODIFY " , PyInt_FromLong ( LDB_CHANGETYPE_MODIFY ) ) ;
2009-08-26 09:01:12 +04:00
PyModule_AddObject ( m , " FLAG_MOD_ADD " , PyInt_FromLong ( LDB_FLAG_MOD_ADD ) ) ;
PyModule_AddObject ( m , " FLAG_MOD_REPLACE " , PyInt_FromLong ( LDB_FLAG_MOD_REPLACE ) ) ;
PyModule_AddObject ( m , " FLAG_MOD_DELETE " , PyInt_FromLong ( LDB_FLAG_MOD_DELETE ) ) ;
2008-12-21 01:00:23 +03:00
PyModule_AddObject ( m , " SUCCESS " , PyInt_FromLong ( LDB_SUCCESS ) ) ;
PyModule_AddObject ( m , " ERR_OPERATIONS_ERROR " , PyInt_FromLong ( LDB_ERR_OPERATIONS_ERROR ) ) ;
PyModule_AddObject ( m , " ERR_PROTOCOL_ERROR " , PyInt_FromLong ( LDB_ERR_PROTOCOL_ERROR ) ) ;
PyModule_AddObject ( m , " ERR_TIME_LIMIT_EXCEEDED " , PyInt_FromLong ( LDB_ERR_TIME_LIMIT_EXCEEDED ) ) ;
PyModule_AddObject ( m , " ERR_SIZE_LIMIT_EXCEEDED " , PyInt_FromLong ( LDB_ERR_SIZE_LIMIT_EXCEEDED ) ) ;
PyModule_AddObject ( m , " ERR_COMPARE_FALSE " , PyInt_FromLong ( LDB_ERR_COMPARE_FALSE ) ) ;
PyModule_AddObject ( m , " ERR_COMPARE_TRUE " , PyInt_FromLong ( LDB_ERR_COMPARE_TRUE ) ) ;
PyModule_AddObject ( m , " ERR_AUTH_METHOD_NOT_SUPPORTED " , PyInt_FromLong ( LDB_ERR_AUTH_METHOD_NOT_SUPPORTED ) ) ;
PyModule_AddObject ( m , " ERR_STRONG_AUTH_REQUIRED " , PyInt_FromLong ( LDB_ERR_STRONG_AUTH_REQUIRED ) ) ;
PyModule_AddObject ( m , " ERR_REFERRAL " , PyInt_FromLong ( LDB_ERR_REFERRAL ) ) ;
PyModule_AddObject ( m , " ERR_ADMIN_LIMIT_EXCEEDED " , PyInt_FromLong ( LDB_ERR_ADMIN_LIMIT_EXCEEDED ) ) ;
PyModule_AddObject ( m , " ERR_UNSUPPORTED_CRITICAL_EXTENSION " , PyInt_FromLong ( LDB_ERR_UNSUPPORTED_CRITICAL_EXTENSION ) ) ;
PyModule_AddObject ( m , " ERR_CONFIDENTIALITY_REQUIRED " , PyInt_FromLong ( LDB_ERR_CONFIDENTIALITY_REQUIRED ) ) ;
PyModule_AddObject ( m , " ERR_SASL_BIND_IN_PROGRESS " , PyInt_FromLong ( LDB_ERR_SASL_BIND_IN_PROGRESS ) ) ;
PyModule_AddObject ( m , " ERR_NO_SUCH_ATTRIBUTE " , PyInt_FromLong ( LDB_ERR_NO_SUCH_ATTRIBUTE ) ) ;
PyModule_AddObject ( m , " ERR_UNDEFINED_ATTRIBUTE_TYPE " , PyInt_FromLong ( LDB_ERR_UNDEFINED_ATTRIBUTE_TYPE ) ) ;
PyModule_AddObject ( m , " ERR_INAPPROPRIATE_MATCHING " , PyInt_FromLong ( LDB_ERR_INAPPROPRIATE_MATCHING ) ) ;
PyModule_AddObject ( m , " ERR_CONSTRAINT_VIOLATION " , PyInt_FromLong ( LDB_ERR_CONSTRAINT_VIOLATION ) ) ;
PyModule_AddObject ( m , " ERR_ATTRIBUTE_OR_VALUE_EXISTS " , PyInt_FromLong ( LDB_ERR_ATTRIBUTE_OR_VALUE_EXISTS ) ) ;
PyModule_AddObject ( m , " ERR_INVALID_ATTRIBUTE_SYNTAX " , PyInt_FromLong ( LDB_ERR_INVALID_ATTRIBUTE_SYNTAX ) ) ;
PyModule_AddObject ( m , " ERR_NO_SUCH_OBJECT " , PyInt_FromLong ( LDB_ERR_NO_SUCH_OBJECT ) ) ;
PyModule_AddObject ( m , " ERR_ALIAS_PROBLEM " , PyInt_FromLong ( LDB_ERR_ALIAS_PROBLEM ) ) ;
PyModule_AddObject ( m , " ERR_INVALID_DN_SYNTAX " , PyInt_FromLong ( LDB_ERR_INVALID_DN_SYNTAX ) ) ;
PyModule_AddObject ( m , " ERR_ALIAS_DEREFERINCING_PROBLEM " , PyInt_FromLong ( LDB_ERR_ALIAS_DEREFERENCING_PROBLEM ) ) ;
PyModule_AddObject ( m , " ERR_INAPPROPRIATE_AUTHENTICATION " , PyInt_FromLong ( LDB_ERR_INAPPROPRIATE_AUTHENTICATION ) ) ;
PyModule_AddObject ( m , " ERR_INVALID_CREDENTIALS " , PyInt_FromLong ( LDB_ERR_INVALID_CREDENTIALS ) ) ;
PyModule_AddObject ( m , " ERR_INSUFFICIENT_ACCESS_RIGHTS " , PyInt_FromLong ( LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS ) ) ;
PyModule_AddObject ( m , " ERR_BUSY " , PyInt_FromLong ( LDB_ERR_BUSY ) ) ;
PyModule_AddObject ( m , " ERR_UNAVAILABLE " , PyInt_FromLong ( LDB_ERR_UNAVAILABLE ) ) ;
PyModule_AddObject ( m , " ERR_UNWILLING_TO_PERFORM " , PyInt_FromLong ( LDB_ERR_UNWILLING_TO_PERFORM ) ) ;
PyModule_AddObject ( m , " ERR_LOOP_DETECT " , PyInt_FromLong ( LDB_ERR_LOOP_DETECT ) ) ;
PyModule_AddObject ( m , " ERR_NAMING_VIOLATION " , PyInt_FromLong ( LDB_ERR_NAMING_VIOLATION ) ) ;
PyModule_AddObject ( m , " ERR_OBJECT_CLASS_VIOLATION " , PyInt_FromLong ( LDB_ERR_OBJECT_CLASS_VIOLATION ) ) ;
PyModule_AddObject ( m , " ERR_NOT_ALLOWED_ON_NON_LEAF " , PyInt_FromLong ( LDB_ERR_NOT_ALLOWED_ON_NON_LEAF ) ) ;
PyModule_AddObject ( m , " ERR_NOT_ALLOWED_ON_RDN " , PyInt_FromLong ( LDB_ERR_NOT_ALLOWED_ON_RDN ) ) ;
2008-12-21 09:34:27 +03:00
PyModule_AddObject ( m , " ERR_ENTRY_ALREADY_EXISTS " , PyInt_FromLong ( LDB_ERR_ENTRY_ALREADY_EXISTS ) ) ;
2008-12-21 01:00:23 +03:00
PyModule_AddObject ( m , " ERR_OBJECT_CLASS_MODS_PROHIBITED " , PyInt_FromLong ( LDB_ERR_OBJECT_CLASS_MODS_PROHIBITED ) ) ;
PyModule_AddObject ( m , " ERR_AFFECTS_MULTIPLE_DSAS " , PyInt_FromLong ( LDB_ERR_AFFECTS_MULTIPLE_DSAS ) ) ;
PyModule_AddObject ( m , " ERR_OTHER " , PyInt_FromLong ( LDB_ERR_OTHER ) ) ;
2010-04-08 22:28:11 +04:00
PyModule_AddObject ( m , " FLG_RDONLY " , PyInt_FromLong ( LDB_FLG_RDONLY ) ) ;
PyModule_AddObject ( m , " FLG_NOSYNC " , PyInt_FromLong ( LDB_FLG_NOSYNC ) ) ;
PyModule_AddObject ( m , " FLG_RECONNECT " , PyInt_FromLong ( LDB_FLG_RECONNECT ) ) ;
PyModule_AddObject ( m , " FLG_NOMMAP " , PyInt_FromLong ( LDB_FLG_NOMMAP ) ) ;
2009-08-15 17:18:46 +04:00
2008-12-19 04:22:07 +03:00
PyModule_AddObject ( m , " __docformat__ " , PyString_FromString ( " restructuredText " ) ) ;
2009-02-05 13:04:28 +03:00
PyExc_LdbError = PyErr_NewException ( discard_const_p ( char , " _ldb.LdbError " ) , NULL , NULL ) ;
2008-12-21 00:21:39 +03:00
PyModule_AddObject ( m , " LdbError " , PyExc_LdbError ) ;
2008-12-19 04:22:07 +03:00
2008-12-19 16:41:44 +03:00
Py_INCREF ( & PyLdb ) ;
Py_INCREF ( & PyLdbDn ) ;
Py_INCREF ( & PyLdbModule ) ;
Py_INCREF ( & PyLdbMessage ) ;
Py_INCREF ( & PyLdbMessageElement ) ;
2008-12-19 19:08:35 +03:00
Py_INCREF ( & PyLdbTree ) ;
2011-02-07 09:50:36 +03:00
Py_INCREF ( & PyLdbResult ) ;
Py_INCREF ( & PyLdbControl ) ;
2008-12-19 04:22:07 +03:00
2008-12-19 16:41:44 +03:00
PyModule_AddObject ( m , " Ldb " , ( PyObject * ) & PyLdb ) ;
PyModule_AddObject ( m , " Dn " , ( PyObject * ) & PyLdbDn ) ;
PyModule_AddObject ( m , " Message " , ( PyObject * ) & PyLdbMessage ) ;
PyModule_AddObject ( m , " MessageElement " , ( PyObject * ) & PyLdbMessageElement ) ;
PyModule_AddObject ( m , " Module " , ( PyObject * ) & PyLdbModule ) ;
2008-12-19 19:08:35 +03:00
PyModule_AddObject ( m , " Tree " , ( PyObject * ) & PyLdbTree ) ;
2011-02-07 09:50:36 +03:00
PyModule_AddObject ( m , " Control " , ( PyObject * ) & PyLdbControl ) ;
2010-10-05 02:36:21 +04:00
PyModule_AddObject ( m , " __version__ " , PyString_FromString ( PACKAGE_VERSION ) ) ;
2011-06-22 06:34:32 +04:00
# define ADD_LDB_STRING(val) PyModule_AddObject(m, #val, PyString_FromString(val))
ADD_LDB_STRING ( LDB_SYNTAX_DN ) ;
ADD_LDB_STRING ( LDB_SYNTAX_DN ) ;
ADD_LDB_STRING ( LDB_SYNTAX_DIRECTORY_STRING ) ;
ADD_LDB_STRING ( LDB_SYNTAX_INTEGER ) ;
ADD_LDB_STRING ( LDB_SYNTAX_BOOLEAN ) ;
ADD_LDB_STRING ( LDB_SYNTAX_OCTET_STRING ) ;
ADD_LDB_STRING ( LDB_SYNTAX_UTC_TIME ) ;
2008-12-19 04:22:07 +03:00
}