r11958: - fixed memory leaks in the ldb_result handling in ldb operations
[abartlet/samba.git/.git] / source4 / ldap_server / ldap_simple_ldb.c
index 7de992f3a485600bc504a0862b3adae00c4a5c32..b9ef085002d1f6b9aaa709e7332c07a0179421a6 100644 (file)
@@ -2,6 +2,7 @@
    Unix SMB/CIFS implementation.
    LDAP server SIMPLE LDB implementation
    Copyright (C) Stefan Metzmacher 2004
+   Copyright (C) Simo Sorce 2004
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
 */
 
 #include "includes.h"
-
-/* TODO: samdb_context is not a pulblic struct */
-struct samdb_context {
-       struct ldb_context *ldb;
-       struct samdb_context **static_ptr;
-};
-
-
-#define ALLOC_CHECK(ptr) do {\
-       if (!(ptr)) {\
+#include "ldap_server/ldap_server.h"
+#include "lib/ldb/include/ldb.h"
+#include "lib/ldb/include/ldb_errors.h"
+#include "auth/auth.h"
+#include "db_wrap.h"
+
+#define VALID_DN_SYNTAX(dn,i) do {\
+       if (!(dn)) {\
                return NT_STATUS_NO_MEMORY;\
+       } else if ((dn)->comp_num < (i)) {\
+               result = LDAP_INVALID_DN_SYNTAX;\
+               errstr = "Invalid DN (" #i " components needed for '" #dn "')";\
+               goto reply;\
        }\
 } while(0)
 
 
 /*
-   fix the DN removing unneded non-significative spaces
-   this function ASSUME the string is talloced
- */
-static char *sldb_fix_dn(const char *dn)
+  map an error code from ldb to ldap
+*/
+static int sldb_map_error(struct ldapsrv_partition *partition, int ldb_ret,
+                         const char **errstr)
 {
-       char *new_dn, *n, *current;
-       int i, j, k;
-
-       /* alloc enough room to host the whole dn as multibyte string */
-       new_dn = talloc(dn, strlen(dn) + 1);
-       if (!new_dn) {
-               DEBUG(0, ("sldb_fix_dn: Out of memory!"));
-               return NULL;
-       }
-
-       i = j = 0;
-       while (dn[i] != '\0') {
-               /* it is legal to check for ascii chars in utf-8 as it is
-                * guaranted to never contain ascii chars (up to 0x7F) as part
-                * of a multibyte sequence */
+       struct ldb_context *samdb = talloc_get_type(partition->private, 
+                                                   struct ldb_context);
+       *errstr = ldb_errstring(samdb);
 
-               new_dn[j] = dn[i];
+       /* its 1:1 for now */
+       return ldb_ret;
+}
 
-               if (dn[i] == ',' || dn[i] == '=') {
-                       /* skip spaces after ',' or '=' */
-                       for (++i; dn[i] == ' '; i++) ;
-                       j++;
-                       continue;
+/*
+  connect to the sam database
+*/
+NTSTATUS sldb_Init(struct ldapsrv_partition *partition, struct ldapsrv_connection *conn) 
+{
+       TALLOC_CTX *mem_ctx = talloc_new(partition);
+       struct ldb_context *ldb;
+       const char *url;
+       url = lp_parm_string(-1, "ldapsrv", "samdb");
+       if (url) {
+
+               ldb = ldb_wrap_connect(mem_ctx, url, 0, NULL);
+               if (ldb == NULL) {
+                       talloc_free(mem_ctx);
+                       return NT_STATUS_INTERNAL_DB_CORRUPTION;
                }
-               if (dn[i] == ' ') {
-                       /* check if there's a ',' after these spaces */
-                       for (k = i; dn[k] == ' '; k++) ;
-                       if (dn[k] == ',') {
-                               /* skip spaces */
-                               i = k;
-                               continue;
-                       } else {
-                               /* fill the dest buffer with the spaces */
-                               for (; dn[i] == ' '; i++, j++) {
-                                       new_dn[j] = ' ';
-                               }
-                               continue;
-                       }
+               if (ldb_set_opaque(ldb, "sessionInfo", conn->session_info)) {
+                       talloc_free(mem_ctx);
+                       return NT_STATUS_NO_MEMORY;
                }
-               i++;
-               j++;
+               talloc_steal(partition, ldb);
+               partition->private = ldb;
+               talloc_free(mem_ctx);
+               return NT_STATUS_OK;
        }
-       new_dn[j] = '\0';
-
-       return new_dn;
+       
+       ldb = samdb_connect(mem_ctx, conn->session_info);
+       if (ldb == NULL) {
+               talloc_free(mem_ctx);
+               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       }
+       talloc_steal(partition, ldb);
+       partition->private = ldb;
+       talloc_free(mem_ctx);
+       return NT_STATUS_OK;
 }
 
+/*
+  Re-connect to the ldb after a bind (this does not handle the bind
+  itself, but just notes the change in credentials)
+*/
+NTSTATUS sldb_Bind(struct ldapsrv_partition *partition, struct ldapsrv_connection *conn) 
+{
+       struct ldb_context *samdb = partition->private;
+       NTSTATUS status;
+       status = sldb_Init(partition, conn);
+       if (NT_STATUS_IS_OK(status)) {
+               /* don't leak the old LDB */
+               talloc_free(samdb);
+       }
+       return status;
+}
 
 static NTSTATUS sldb_Search(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
-                                    struct ldap_SearchRequest *r)
+                           struct ldap_SearchRequest *r)
 {
-       NTSTATUS status;
+       void *local_ctx;
+       struct ldb_dn *basedn;
        struct ldap_Result *done;
        struct ldap_SearchResEntry *ent;
        struct ldapsrv_reply *ent_r, *done_r;
-       int result = 80;
-       struct samdb_context *samdb;
-       struct ldb_message **res;
-       int i, j, y, count;
-       struct ldb_context *ldb;
+       int result = LDAP_SUCCESS;
+       struct ldb_context *samdb;
+       struct ldb_result *res = NULL;
+       int i, j, y, ret;
+       int success_limit = 1;
        enum ldb_scope scope = LDB_SCOPE_DEFAULT;
        const char **attrs = NULL;
-       const char *basedn;
-       const char *errstr;
-
-       samdb = samdb_connect(call);
-       ldb = samdb->ldb;
-       basedn = sldb_fix_dn(r->basedn);
-       if (basedn == NULL) {
-               return NT_STATUS_NO_MEMORY;
-       }
+       const char *errstr = NULL;
+       struct ldb_request lreq;
 
-       DEBUG(10, ("sldb_Search: basedn: [%s]\n", basedn));
-       DEBUG(10, ("sldb_Search: filter: [%s]\n", r->filter));
+       local_ctx = talloc_named(call, 0, "sldb_Search local memory context");
+       NT_STATUS_HAVE_NO_MEMORY(local_ctx);
+
+       samdb = talloc_get_type(partition->private, struct ldb_context);
+
+       basedn = ldb_dn_explode(local_ctx, r->basedn);
+       VALID_DN_SYNTAX(basedn, 0);
+
+       DEBUG(10, ("sldb_Search: basedn: [%s]\n", r->basedn));
+       DEBUG(10, ("sldb_Search: filter: [%s]\n", ldb_filter_from_tree(call, r->tree)));
 
        switch (r->scope) {
                case LDAP_SEARCH_SCOPE_BASE:
                        DEBUG(10,("sldb_Search: scope: [BASE]\n"));
                        scope = LDB_SCOPE_BASE;
+                       success_limit = 0;
                        break;
                case LDAP_SEARCH_SCOPE_SINGLE:
                        DEBUG(10,("sldb_Search: scope: [ONE]\n"));
                        scope = LDB_SCOPE_ONELEVEL;
+                       success_limit = 0;
                        break;
                case LDAP_SEARCH_SCOPE_SUB:
                        DEBUG(10,("sldb_Search: scope: [SUB]\n"));
                        scope = LDB_SCOPE_SUBTREE;
+                       success_limit = 0;
                        break;
        }
 
        if (r->num_attributes >= 1) {
-               attrs = talloc_array_p(samdb, const char *, r->num_attributes+1);
-               ALLOC_CHECK(attrs);
+               attrs = talloc_array(samdb, const char *, r->num_attributes+1);
+               NT_STATUS_HAVE_NO_MEMORY(attrs);
 
                for (i=0; i < r->num_attributes; i++) {
                        DEBUG(10,("sldb_Search: attrs: [%s]\n",r->attributes[i]));
@@ -141,110 +162,125 @@ static NTSTATUS sldb_Search(struct ldapsrv_partition *partition, struct ldapsrv_
                attrs[i] = NULL;
        }
 
-       ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
-       count = ldb_search(ldb, basedn, scope, r->filter, attrs, &res);
+       DEBUG(5,("ldb_request dn=%s filter=%s\n", 
+                r->basedn, ldb_filter_from_tree(call, r->tree)));
 
-       for (i=0; i < count; i++) {
-               ent_r = ldapsrv_init_reply(call, LDAP_TAG_SearchResultEntry);
-               ALLOC_CHECK(ent_r);
+       ZERO_STRUCT(lreq);
+       lreq.operation = LDB_REQ_SEARCH;
+       lreq.op.search.base = basedn;
+       lreq.op.search.scope = scope;
+       lreq.op.search.tree = r->tree;
+       lreq.op.search.attrs = attrs;
 
-               ent = &ent_r->msg.r.SearchResultEntry;
-               ent->dn = talloc_steal(ent_r, res[i]->dn);
-               ent->num_attributes = 0;
-               ent->attributes = NULL;
-               if (res[i]->num_elements == 0) {
-                       goto queue_reply;
-               }
-               ent->num_attributes = res[i]->num_elements;
-               ent->attributes = talloc_array_p(ent_r, struct ldap_attribute, ent->num_attributes);
-               ALLOC_CHECK(ent->attributes);
-               for (j=0; j < ent->num_attributes; j++) {
-                       ent->attributes[j].name = talloc_steal(ent->attributes, res[i]->elements[j].name);
-                       ent->attributes[j].num_values = 0;
-                       ent->attributes[j].values = NULL;
-                       if (r->attributesonly && (res[i]->elements[j].num_values == 0)) {
-                               continue;
+       ret = ldb_request(samdb, &lreq);
+
+       res = talloc_steal(samdb, lreq.op.search.res);
+
+       if (ret == LDB_SUCCESS) {
+               for (i = 0; i < res->count; i++) {
+                       ent_r = ldapsrv_init_reply(call, LDAP_TAG_SearchResultEntry);
+                       NT_STATUS_HAVE_NO_MEMORY(ent_r);
+
+                       ent = &ent_r->msg->r.SearchResultEntry;
+                       ent->dn = ldb_dn_linearize(ent_r, res->msgs[i]->dn);
+                       ent->num_attributes = 0;
+                       ent->attributes = NULL;
+                       if (res->msgs[i]->num_elements == 0) {
+                               goto queue_reply;
                        }
-                       ent->attributes[j].num_values = res[i]->elements[j].num_values;
-                       ent->attributes[j].values = talloc_array_p(ent->attributes,
-                                                       DATA_BLOB, ent->attributes[j].num_values);
-                       ALLOC_CHECK(ent->attributes[j].values);
-                       for (y=0; y < ent->attributes[j].num_values; y++) {
-                               ent->attributes[j].values[y].length = res[i]->elements[j].values[y].length;
-                               ent->attributes[j].values[y].data = talloc_steal(ent->attributes[j].values,
-                                                                       res[i]->elements[j].values[y].data);
+                       ent->num_attributes = res->msgs[i]->num_elements;
+                       ent->attributes = talloc_array(ent_r, struct ldb_message_element, ent->num_attributes);
+                       NT_STATUS_HAVE_NO_MEMORY(ent->attributes);
+                       for (j=0; j < ent->num_attributes; j++) {
+                               ent->attributes[j].name = talloc_steal(ent->attributes, res->msgs[i]->elements[j].name);
+                               ent->attributes[j].num_values = 0;
+                               ent->attributes[j].values = NULL;
+                               if (r->attributesonly && (res->msgs[i]->elements[j].num_values == 0)) {
+                                       continue;
+                               }
+                               ent->attributes[j].num_values = res->msgs[i]->elements[j].num_values;
+                               ent->attributes[j].values = talloc_array(ent->attributes,
+                                                               DATA_BLOB, ent->attributes[j].num_values);
+                               NT_STATUS_HAVE_NO_MEMORY(ent->attributes[j].values);
+                               for (y=0; y < ent->attributes[j].num_values; y++) {
+                                       ent->attributes[j].values[y].length = res->msgs[i]->elements[j].values[y].length;
+                                       ent->attributes[j].values[y].data = talloc_steal(ent->attributes[j].values,
+                                                                               res->msgs[i]->elements[j].values[y].data);
+                               }
                        }
-               }
 queue_reply:
-               status = ldapsrv_queue_reply(call, ent_r);
-               if (!NT_STATUS_IS_OK(status)) {
-                       return status;
+                       ldapsrv_queue_reply(call, ent_r);
                }
        }
 
+reply:
        done_r = ldapsrv_init_reply(call, LDAP_TAG_SearchResultDone);
-       ALLOC_CHECK(done_r);
-
-       if (count > 0) {
-               DEBUG(10,("sldb_Search: results: [%d]\n",count));
-               result = 0;
-               errstr = NULL;
-       } else if (count == 0) {
-               DEBUG(10,("sldb_Search: no results\n"));
-               result = 32;
-               errstr = talloc_strdup(done_r, ldb_errstring(ldb));
-       } else if (count == -1) {
+       NT_STATUS_HAVE_NO_MEMORY(done_r);
+
+       if (ret == LDB_SUCCESS) {
+               if (res->count >= success_limit) {
+                       DEBUG(10,("sldb_Search: results: [%d]\n", res->count));
+                       result = LDAP_SUCCESS;
+                       errstr = NULL;
+               } else if (res->count == 0) {
+                       DEBUG(10,("sldb_Search: no results\n"));
+                       result = LDAP_NO_SUCH_OBJECT;
+                       errstr = ldb_errstring(samdb);
+               }
+       } else {
                DEBUG(10,("sldb_Search: error\n"));
-               result = 1;
-               errstr = talloc_strdup(done_r, ldb_errstring(ldb));
+               result = ret;
+               errstr = ldb_errstring(samdb);
        }
 
-       done = &done_r->msg.r.SearchResultDone;
-       done->resultcode = result;
+       done = &done_r->msg->r.SearchResultDone;
        done->dn = NULL;
-       done->errormessage = errstr;
+       done->resultcode = result;
+       done->errormessage = (errstr?talloc_strdup(done_r, errstr):NULL);
        done->referral = NULL;
 
-       talloc_free(samdb);
+       talloc_free(local_ctx);
 
-       return ldapsrv_queue_reply(call, done_r);
+       ldapsrv_queue_reply(call, done_r);
+       return NT_STATUS_OK;
 }
 
 static NTSTATUS sldb_Add(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
-                                    struct ldap_AddRequest *r)
+                        struct ldap_AddRequest *r)
 {
+       void *local_ctx;
+       struct ldb_dn *dn;
        struct ldap_Result *add_result;
        struct ldapsrv_reply *add_reply;
        int ldb_ret;
-       struct samdb_context *samdb;
-       struct ldb_context *ldb;
-       const char *dn;
-       struct ldb_message *msg;
+       struct ldb_context *samdb;
+       struct ldb_message *msg = NULL;
        int result = LDAP_SUCCESS;
        const char *errstr = NULL;
        int i,j;
 
-       samdb = samdb_connect(call);
-       ldb = samdb->ldb;
-       dn = sldb_fix_dn(r->dn);
-       if (dn == NULL) {
-               return NT_STATUS_NO_MEMORY;
-       }
+       local_ctx = talloc_named(call, 0, "sldb_Add local memory context");
+       NT_STATUS_HAVE_NO_MEMORY(local_ctx);
 
-       DEBUG(10, ("sldb_add: dn: [%s]\n", dn));
+       samdb = talloc_get_type(partition->private, struct ldb_context);
 
-       msg = talloc_p(samdb, struct ldb_message);
-       ALLOC_CHECK(msg);
+       dn = ldb_dn_explode(local_ctx, r->dn);
+       VALID_DN_SYNTAX(dn,1);
 
-       msg->dn = discard_const_p(char, dn);
+       DEBUG(10, ("sldb_add: dn: [%s]\n", r->dn));
+
+       msg = talloc(local_ctx, struct ldb_message);
+       NT_STATUS_HAVE_NO_MEMORY(msg);
+
+       msg->dn = dn;
        msg->private_data = NULL;
        msg->num_elements = 0;
        msg->elements = NULL;
 
        if (r->num_attributes > 0) {
                msg->num_elements = r->num_attributes;
-               msg->elements = talloc_array_p(msg, struct ldb_message_element, msg->num_elements);
-               ALLOC_CHECK(msg->elements);
+               msg->elements = talloc_array(msg, struct ldb_message_element, msg->num_elements);
+               NT_STATUS_HAVE_NO_MEMORY(msg->elements);
 
                for (i=0; i < msg->num_elements; i++) {
                        msg->elements[i].name = discard_const_p(char, r->attributes[i].name);
@@ -254,148 +290,130 @@ static NTSTATUS sldb_Add(struct ldapsrv_partition *partition, struct ldapsrv_cal
                        
                        if (r->attributes[i].num_values > 0) {
                                msg->elements[i].num_values = r->attributes[i].num_values;
-                               msg->elements[i].values = talloc_array_p(msg, struct ldb_val, msg->elements[i].num_values);
-                               ALLOC_CHECK(msg->elements[i].values);
+                               msg->elements[i].values = talloc_array(msg, struct ldb_val, msg->elements[i].num_values);
+                               NT_STATUS_HAVE_NO_MEMORY(msg->elements[i].values);
 
                                for (j=0; j < msg->elements[i].num_values; j++) {
                                        if (!(r->attributes[i].values[j].length > 0)) {
-                                               result = 80;
-                                               goto invalid_input;
+                                               result = LDAP_OTHER;
+                                               errstr = "Empty attribute values are not allowed";
+                                               goto reply;
                                        }
                                        msg->elements[i].values[j].length = r->attributes[i].values[j].length;
                                        msg->elements[i].values[j].data = r->attributes[i].values[j].data;                      
                                }
                        } else {
-                               result = 80;
-                               goto invalid_input;
+                               result = LDAP_OTHER;
+                               errstr = "No attribute values are not allowed";
+                               goto reply;
                        }
                }
        } else {
-               result = 80;
-               goto invalid_input;
+               result = LDAP_OTHER;
+               errstr = "No attributes are not allowed";
+               goto reply;
        }
 
-invalid_input:
-
+reply:
        add_reply = ldapsrv_init_reply(call, LDAP_TAG_AddResponse);
-       ALLOC_CHECK(add_reply);
-
-       add_result = &add_reply->msg.r.AddResponse;
-       add_result->dn = talloc_steal(add_reply, dn);
+       NT_STATUS_HAVE_NO_MEMORY(add_reply);
 
        if (result == LDAP_SUCCESS) {
-               ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
-               ldb_ret = ldb_add(ldb, msg);
-               if (ldb_ret == 0) {
-                       result = LDAP_SUCCESS;
-                       errstr = NULL;
-               } else {
-                       /* currently we have no way to tell if there was an internal ldb error
-                        * or if the object was not found, return the most probable error
-                        */
-                       result = 1;
-                       errstr = talloc_strdup(add_reply, ldb_errstring(ldb));
-               }
-       } else {
-               errstr = talloc_strdup(add_reply,"invalid input data");
+               ldb_ret = ldb_add(samdb, msg);
+               result = sldb_map_error(partition, ldb_ret, &errstr);
        }
 
+       add_result = &add_reply->msg->r.AddResponse;
+       add_result->dn = NULL;
        add_result->resultcode = result;
-       add_result->errormessage = errstr;
+       add_result->errormessage = (errstr?talloc_strdup(add_reply,errstr):NULL);
        add_result->referral = NULL;
 
-       talloc_free(samdb);
+       talloc_free(local_ctx);
 
-       return ldapsrv_queue_reply(call, add_reply);
+       ldapsrv_queue_reply(call, add_reply);
+       return NT_STATUS_OK;
 }
 
 static NTSTATUS sldb_Del(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
                                     struct ldap_DelRequest *r)
 {
+       void *local_ctx;
+       struct ldb_dn *dn;
        struct ldap_Result *del_result;
        struct ldapsrv_reply *del_reply;
        int ldb_ret;
-       struct samdb_context *samdb;
-       struct ldb_context *ldb;
-       const char *dn;
+       struct ldb_context *samdb;
        const char *errstr = NULL;
        int result = LDAP_SUCCESS;
 
-       samdb = samdb_connect(call);
-       ldb = samdb->ldb;
-       dn = sldb_fix_dn(r->dn);
-       if (dn == NULL) {
-               return NT_STATUS_NO_MEMORY;
-       }
+       local_ctx = talloc_named(call, 0, "sldb_Del local memory context");
+       NT_STATUS_HAVE_NO_MEMORY(local_ctx);
 
-       DEBUG(10, ("sldb_Del: dn: [%s]\n", dn));
+       samdb = talloc_get_type(partition->private, struct ldb_context);
 
-       ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
-       ldb_ret = ldb_delete(ldb, dn);
+       dn = ldb_dn_explode(local_ctx, r->dn);
+       VALID_DN_SYNTAX(dn,1);
 
-       errstr = ldb_errstring(ldb);
+       DEBUG(10, ("sldb_Del: dn: [%s]\n", r->dn));
 
+reply:
        del_reply = ldapsrv_init_reply(call, LDAP_TAG_DelResponse);
-       ALLOC_CHECK(del_reply);
+       NT_STATUS_HAVE_NO_MEMORY(del_reply);
 
-       del_result = &del_reply->msg.r.DelResponse;
-       del_result->dn = talloc_steal(del_reply, dn);
-
-       if (ldb_ret == 0) {
-               result = LDAP_SUCCESS;
-               errstr = NULL;
-       } else {
-               /* currently we have no way to tell if there was an internal ldb error
-                * or if the object was not found, return the most probable error
-                */
-               result = LDAP_NO_SUCH_OBJECT;
-               errstr = talloc_strdup(del_reply, ldb_errstring(ldb));
+       if (result == LDAP_SUCCESS) {
+               ldb_ret = ldb_delete(samdb, dn);
+               result = sldb_map_error(partition, ldb_ret, &errstr);
        }
 
+       del_result = &del_reply->msg->r.DelResponse;
+       del_result->dn = NULL;
        del_result->resultcode = result;
-       del_result->errormessage = errstr;
+       del_result->errormessage = (errstr?talloc_strdup(del_reply,errstr):NULL);
        del_result->referral = NULL;
 
-       talloc_free(samdb);
+       talloc_free(local_ctx);
 
-       return ldapsrv_queue_reply(call, del_reply);
+       ldapsrv_queue_reply(call, del_reply);
+       return NT_STATUS_OK;
 }
 
 static NTSTATUS sldb_Modify(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
                                     struct ldap_ModifyRequest *r)
 {
+       void *local_ctx;
+       struct ldb_dn *dn;
        struct ldap_Result *modify_result;
        struct ldapsrv_reply *modify_reply;
        int ldb_ret;
-       struct samdb_context *samdb;
-       struct ldb_context *ldb;
-       const char *dn;
-       struct ldb_message *msg;
+       struct ldb_context *samdb;
+       struct ldb_message *msg = NULL;
        int result = LDAP_SUCCESS;
        const char *errstr = NULL;
        int i,j;
 
-       samdb = samdb_connect(call);
-       ldb = samdb->ldb;
-       dn = sldb_fix_dn(r->dn);
-       if (dn == NULL) {
-               return NT_STATUS_NO_MEMORY;
-       }
+       local_ctx = talloc_named(call, 0, "sldb_Modify local memory context");
+       NT_STATUS_HAVE_NO_MEMORY(local_ctx);
+
+       samdb = talloc_get_type(partition->private, struct ldb_context);
 
-       DEBUG(10, ("sldb_modify: dn: [%s]\n", dn));
+       dn = ldb_dn_explode(local_ctx, r->dn);
+       VALID_DN_SYNTAX(dn, 1);
 
-       msg = talloc_p(samdb, struct ldb_message);
-       ALLOC_CHECK(msg);
+       DEBUG(10, ("sldb_modify: dn: [%s]\n", r->dn));
 
-       msg->dn = discard_const_p(char, dn);
+       msg = talloc(local_ctx, struct ldb_message);
+       NT_STATUS_HAVE_NO_MEMORY(msg);
+
+       msg->dn = dn;
        msg->private_data = NULL;
        msg->num_elements = 0;
        msg->elements = NULL;
 
        if (r->num_mods > 0) {
                msg->num_elements = r->num_mods;
-               msg->elements = talloc_array_p(msg, struct ldb_message_element, r->num_mods);
-               ALLOC_CHECK(msg->elements);
+               msg->elements = talloc_array(msg, struct ldb_message_element, r->num_mods);
+               NT_STATUS_HAVE_NO_MEMORY(msg->elements);
 
                for (i=0; i < msg->num_elements; i++) {
                        msg->elements[i].name = discard_const_p(char, r->mods[i].attrib.name);
@@ -404,8 +422,9 @@ static NTSTATUS sldb_Modify(struct ldapsrv_partition *partition, struct ldapsrv_
 
                        switch (r->mods[i].type) {
                        default:
-                               result = 2;
-                               goto invalid_input;
+                               result = LDAP_PROTOCOL_ERROR;
+                               errstr = "Invalid LDAP_MODIFY_* type";
+                               goto reply;
                        case LDAP_MODIFY_ADD:
                                msg->elements[i].flags = LDB_FLAG_MOD_ADD;
                                break;
@@ -417,135 +436,208 @@ static NTSTATUS sldb_Modify(struct ldapsrv_partition *partition, struct ldapsrv_
                                break;
                        }
 
-                       if (r->mods[i].attrib.num_values > 0) {
-                               msg->elements[i].num_values = r->mods[i].attrib.num_values;
-                               msg->elements[i].values = talloc_array_p(msg, struct ldb_val, msg->elements[i].num_values);
-                               ALLOC_CHECK(msg->elements[i].values);
+                       msg->elements[i].num_values = r->mods[i].attrib.num_values;
+                       if (msg->elements[i].num_values > 0) {
+                               msg->elements[i].values = talloc_array(msg, struct ldb_val, msg->elements[i].num_values);
+                               NT_STATUS_HAVE_NO_MEMORY(msg->elements[i].values);
 
                                for (j=0; j < msg->elements[i].num_values; j++) {
                                        if (!(r->mods[i].attrib.values[j].length > 0)) {
-                                               result = 80;
-                                               goto invalid_input;
+                                               result = LDAP_OTHER;
+                                               errstr = "Empty attribute values are not allowed";
+                                               goto reply;
                                        }
                                        msg->elements[i].values[j].length = r->mods[i].attrib.values[j].length;
                                        msg->elements[i].values[j].data = r->mods[i].attrib.values[j].data;                     
                                }
-                       } else {
-                               /* TODO: test what we should do here 
-                                *
-                                *       LDAP_MODIFY_DELETE is ok to pass here
-                                */
                        }
                }
        } else {
-               result = 80;
-               goto invalid_input;
+               result = LDAP_OTHER;
+               errstr = "No mods are not allowed";
+               goto reply;
        }
 
-invalid_input:
-
+reply:
        modify_reply = ldapsrv_init_reply(call, LDAP_TAG_ModifyResponse);
-       ALLOC_CHECK(modify_reply);
-
-       modify_result = &modify_reply->msg.r.AddResponse;
-       modify_result->dn = talloc_steal(modify_reply, dn);
+       NT_STATUS_HAVE_NO_MEMORY(modify_reply);
 
        if (result == LDAP_SUCCESS) {
-               ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
-               ldb_ret = ldb_modify(ldb, msg);
-               if (ldb_ret == 0) {
-                       result = LDAP_SUCCESS;
-                       errstr = NULL;
-               } else {
-                       /* currently we have no way to tell if there was an internal ldb error
-                        * or if the object was not found, return the most probable error
-                        */
-                       result = 1;
-                       errstr = talloc_strdup(modify_reply, ldb_errstring(ldb));
-               }
-       } else {
-               errstr = talloc_strdup(modify_reply,"invalid input data");
+               ldb_ret = ldb_modify(samdb, msg);
+               result = sldb_map_error(partition, ldb_ret, &errstr);
        }
 
+       modify_result = &modify_reply->msg->r.AddResponse;
+       modify_result->dn = NULL;
        modify_result->resultcode = result;
-       modify_result->errormessage = errstr;
+       modify_result->errormessage = (errstr?talloc_strdup(modify_reply,errstr):NULL);
        modify_result->referral = NULL;
 
-       talloc_free(samdb);
+       talloc_free(local_ctx);
 
-       return ldapsrv_queue_reply(call, modify_reply);
+       ldapsrv_queue_reply(call, modify_reply);
+       return NT_STATUS_OK;
 }
 
 static NTSTATUS sldb_Compare(struct ldapsrv_partition *partition, struct ldapsrv_call *call,
                                     struct ldap_CompareRequest *r)
 {
+       void *local_ctx;
+       struct ldb_dn *dn;
        struct ldap_Result *compare;
        struct ldapsrv_reply *compare_r;
-       int result = 80;
-       struct samdb_context *samdb;
-       struct ldb_message **res;
-       struct ldb_context *ldb;
+       int result = LDAP_SUCCESS;
+       struct ldb_context *samdb;
+       struct ldb_result *res = NULL;
        const char *attrs[1];
-       const char *errstr;
-       const char *dn;
-       const char *filter;
-       int count;
-
-       samdb = samdb_connect(call);
-       ldb = samdb->ldb;
-       dn = sldb_fix_dn(r->dn);
-       if (dn == NULL) {
-               return NT_STATUS_NO_MEMORY;
-       }
+       const char *errstr = NULL;
+       const char *filter = NULL;
+       int ret;
+
+       local_ctx = talloc_named(call, 0, "sldb_Compare local_memory_context");
+       NT_STATUS_HAVE_NO_MEMORY(local_ctx);
+
+       samdb = talloc_get_type(partition->private, struct ldb_context);
+
+       dn = ldb_dn_explode(local_ctx, r->dn);
+       VALID_DN_SYNTAX(dn, 1);
+
+       DEBUG(10, ("sldb_Compare: dn: [%s]\n", r->dn));
+       filter = talloc_asprintf(local_ctx, "(%s=%*s)", r->attribute, 
+                                (int)r->value.length, r->value.data);
+       NT_STATUS_HAVE_NO_MEMORY(filter);
 
-       DEBUG(10, ("sldb_Compare: dn: [%s]\n", dn));
-       filter = talloc_asprintf(samdb, "(%s=%*s)", r->attribute, r->value.length, r->value.data);
-       ALLOC_CHECK(filter);
        DEBUGADD(10, ("sldb_Compare: attribute: [%s]\n", filter));
 
        attrs[0] = NULL;
 
-       ldb_set_alloc(ldb, talloc_ldb_alloc, samdb);
-       count = ldb_search(ldb, dn, LDB_SCOPE_BASE, filter, attrs, &res);
-
+reply:
        compare_r = ldapsrv_init_reply(call, LDAP_TAG_CompareResponse);
-       ALLOC_CHECK(compare_r);
-
-       if (count == 1) {
-               DEBUG(10,("sldb_Compare: matched\n"));
-               result = 0;
-               errstr = NULL;
-       } else if (count == 0) {
-               result = 32;
-               errstr = talloc_strdup(compare_r, ldb_errstring(ldb));
-               DEBUG(10,("sldb_Compare: no results: %s\n", errstr));
-       } else if (count > 1) {
-               result = 80;
-               errstr = talloc_strdup(compare_r, "too many objects match");
-               DEBUG(10,("sldb_Compare: %d results: %s\n", count, errstr));
-       } else if (count == -1) {
-               result = 1;
-               errstr = talloc_strdup(compare_r, ldb_errstring(ldb));
-               DEBUG(10,("sldb_Compare: error: %s\n", errstr));
+       NT_STATUS_HAVE_NO_MEMORY(compare_r);
+
+       if (result == LDAP_SUCCESS) {
+               ret = ldb_search(samdb, dn, LDB_SCOPE_BASE, filter, attrs, &res);
+               talloc_steal(samdb, res);
+               if (ret != LDB_SUCCESS) {
+                       result = LDAP_OTHER;
+                       errstr = ldb_errstring(samdb);
+                       DEBUG(10,("sldb_Compare: error: %s\n", errstr));
+               } else if (res->count == 0) {
+                       DEBUG(10,("sldb_Compare: doesn't matched\n"));
+                       result = LDAP_COMPARE_FALSE;
+                       errstr = NULL;
+               } else if (res->count == 1) {
+                       DEBUG(10,("sldb_Compare: matched\n"));
+                       result = LDAP_COMPARE_TRUE;
+                       errstr = NULL;
+               } else if (res->count > 1) {
+                       result = LDAP_OTHER;
+                       errstr = "too many objects match";
+                       DEBUG(10,("sldb_Compare: %d results: %s\n", res->count, errstr));
+               }
        }
 
-       compare = &compare_r->msg.r.CompareResponse;
-       compare->resultcode = result;
+       compare = &compare_r->msg->r.CompareResponse;
        compare->dn = NULL;
-       compare->errormessage = errstr;
+       compare->resultcode = result;
+       compare->errormessage = (errstr?talloc_strdup(compare_r,errstr):NULL);
        compare->referral = NULL;
 
-       talloc_free(samdb);
+       talloc_free(local_ctx);
+
+       ldapsrv_queue_reply(call, compare_r);
+       return NT_STATUS_OK;
+}
+
+static NTSTATUS sldb_ModifyDN(struct ldapsrv_partition *partition, struct ldapsrv_call *call, struct ldap_ModifyDNRequest *r)
+{
+       void *local_ctx;
+       struct ldb_dn *olddn, *newdn, *newrdn;
+       struct ldb_dn *parentdn = NULL;
+       struct ldap_Result *modifydn;
+       struct ldapsrv_reply *modifydn_r;
+       int ldb_ret;
+       struct ldb_context *samdb;
+       const char *errstr = NULL;
+       int result = LDAP_SUCCESS;
+
+       local_ctx = talloc_named(call, 0, "sldb_ModifyDN local memory context");
+       NT_STATUS_HAVE_NO_MEMORY(local_ctx);
+
+       samdb = talloc_get_type(partition->private, struct ldb_context);
+
+       olddn = ldb_dn_explode(local_ctx, r->dn);
+       VALID_DN_SYNTAX(olddn, 2);
+
+       newrdn = ldb_dn_explode(local_ctx, r->newrdn);
+       VALID_DN_SYNTAX(newrdn, 1);
+
+       DEBUG(10, ("sldb_ModifyDN: olddn: [%s]\n", r->dn));
+       DEBUG(10, ("sldb_ModifyDN: newrdn: [%s]\n", r->newrdn));
+
+       /* we can't handle the rename if we should not remove the old dn */
+       if (!r->deleteolddn) {
+               result = LDAP_UNWILLING_TO_PERFORM;
+               errstr = "Old RDN must be deleted";
+               goto reply;
+       }
+
+       if (newrdn->comp_num > 1) {
+               result = LDAP_NAMING_VIOLATION;
+               errstr = "Error new RDN invalid";
+               goto reply;
+       }
+
+       if (r->newsuperior) {
+               parentdn = ldb_dn_explode(local_ctx, r->newsuperior);
+               VALID_DN_SYNTAX(parentdn, 0);
+               DEBUG(10, ("sldb_ModifyDN: newsuperior: [%s]\n", r->newsuperior));
+               
+               if (parentdn->comp_num < 1) {
+                       result = LDAP_AFFECTS_MULTIPLE_DSAS;
+                       errstr = "Error new Superior DN invalid";
+                       goto reply;
+               }
+       }
+
+       if (!parentdn) {
+               parentdn = ldb_dn_get_parent(local_ctx, olddn);
+               NT_STATUS_HAVE_NO_MEMORY(parentdn);
+       }
+
+       newdn = ldb_dn_make_child(local_ctx, ldb_dn_get_rdn(local_ctx, newrdn), parentdn);
+       NT_STATUS_HAVE_NO_MEMORY(newdn);
+
+reply:
+       modifydn_r = ldapsrv_init_reply(call, LDAP_TAG_ModifyDNResponse);
+       NT_STATUS_HAVE_NO_MEMORY(modifydn_r);
+
+       if (result == LDAP_SUCCESS) {
+               ldb_ret = ldb_rename(samdb, olddn, newdn);
+               result = sldb_map_error(partition, ldb_ret, &errstr);
+       }
+
+       modifydn = &modifydn_r->msg->r.ModifyDNResponse;
+       modifydn->dn = NULL;
+       modifydn->resultcode = result;
+       modifydn->errormessage = (errstr?talloc_strdup(modifydn_r,errstr):NULL);
+       modifydn->referral = NULL;
+
+       talloc_free(local_ctx);
 
-       return ldapsrv_queue_reply(call, compare_r);
+       ldapsrv_queue_reply(call, modifydn_r);
+       return NT_STATUS_OK;
 }
 
 static const struct ldapsrv_partition_ops sldb_ops = {
+       .Init           = sldb_Init,
+       .Bind           = sldb_Bind,
        .Search         = sldb_Search,
        .Add            = sldb_Add,
        .Del            = sldb_Del,
        .Modify         = sldb_Modify,
-       .Compare        = sldb_Compare
+       .Compare        = sldb_Compare,
+       .ModifyDN       = sldb_ModifyDN
 };
 
 const struct ldapsrv_partition_ops *ldapsrv_get_sldb_partition_ops(void)