source4: Use wrapper for string to integer conversion
[nivanova/samba-autobuild/.git] / source4 / lib / registry / ldb.c
index b2b1e745f344c150d0762b72a0e9892c9758c33f..ec293c46149b6c2bea8688d3e532c4e9e1a3b606 100644 (file)
@@ -20,8 +20,8 @@
 
 #include "includes.h"
 #include "registry.h"
-#include "lib/ldb/include/ldb.h"
-#include "lib/ldb/include/ldb_errors.h"
+#include <ldb.h>
+#include <ldb_errors.h>
 #include "ldb_wrap.h"
 #include "librpc/gen_ndr/winreg.h"
 #include "param/param.h"
@@ -65,7 +65,7 @@ static void reg_ldb_unpack_value(TALLOC_CTX *mem_ctx,
                        /* The data should be provided as UTF16 string */
                        convert_string_talloc(mem_ctx, CH_UTF8, CH_UTF16,
                                              val->data, val->length,
-                                             (void **)&data->data, &data->length, false);
+                                             (void **)&data->data, &data->length);
                } else {
                        data->data = NULL;
                        data->length = 0;
@@ -75,8 +75,16 @@ static void reg_ldb_unpack_value(TALLOC_CTX *mem_ctx,
        case REG_DWORD:
        case REG_DWORD_BIG_ENDIAN:
                if (val != NULL) {
+                       int error = 0;
                        /* The data is a plain DWORD */
-                       uint32_t tmp = strtoul((char *)val->data, NULL, 0);
+                       uint32_t tmp;
+
+                       tmp = strtoul_err((char *)val->data, NULL, 0, &error);
+                       if (error != 0) {
+                               data->data = NULL;
+                               data->length = 0;
+                               break;
+                       }
                        data->data = talloc_size(mem_ctx, sizeof(uint32_t));
                        if (data->data != NULL) {
                                SIVAL(data->data, 0, tmp);
@@ -90,8 +98,16 @@ static void reg_ldb_unpack_value(TALLOC_CTX *mem_ctx,
 
        case REG_QWORD:
                if (val != NULL) {
+                       int error = 0;
                        /* The data is a plain QWORD */
-                       uint64_t tmp = strtoull((char *)val->data, NULL, 0);
+                       uint64_t tmp;
+
+                       tmp = strtoull_err((char *)val->data, NULL, 0, &error);
+                       if (error != 0) {
+                               data->data = NULL;
+                               data->length = 0;
+                               break;
+                       }
                        data->data = talloc_size(mem_ctx, sizeof(uint64_t));
                        if (data->data != NULL) {
                                SBVAL(data->data, 0, tmp);
@@ -159,8 +175,7 @@ static struct ldb_message *reg_ldb_pack_value(struct ldb_context *ctx,
                        /* The data is provided as UTF16 string */
                        ret2 = convert_string_talloc(mem_ctx, CH_UTF16, CH_UTF8,
                                                     (void *)data.data, data.length,
-                                                    (void **)&val->data, &val->length,
-                                                    false);
+                                                    (void **)&val->data, &val->length);
                        if (ret2) {
                                ret = ldb_msg_add_value(msg, "data", val, NULL);
                        } else {
@@ -201,7 +216,7 @@ static struct ldb_message *reg_ldb_pack_value(struct ldb_context *ctx,
                                char *conv_str;
 
                                conv_str = talloc_asprintf(msg, "0x%16.16llx",
-                                                          BVAL(data.data, 0));
+                                                          (unsigned long long)BVAL(data.data, 0));
                                if (conv_str == NULL) {
                                        talloc_free(msg);
                                        return NULL;
@@ -277,44 +292,46 @@ static struct ldb_dn *reg_path_to_ldb(TALLOC_CTX *mem_ctx,
                                      const char *path, const char *add)
 {
        struct ldb_dn *ret;
-       char *mypath = talloc_strdup(mem_ctx, path);
+       char *mypath;
        char *begin;
        struct ldb_key_data *kd = talloc_get_type(from, struct ldb_key_data);
        struct ldb_context *ldb = kd->ldb;
 
+       mypath = talloc_strdup(mem_ctx, path);
+       if (mypath == NULL) {
+               return NULL;
+       }
+
        ret = ldb_dn_new(mem_ctx, ldb, add);
        if (!ldb_dn_validate(ret)) {
                talloc_free(ret);
                return NULL;
        }
 
-       while (mypath) {
-               char *keyname;
-
-               begin = strrchr(mypath, '\\');
+       if (!ldb_dn_add_base(ret, kd->dn)) {
+               talloc_free(ret);
+               return NULL;
+       }
 
-               if (begin) keyname = begin + 1;
-               else keyname = mypath;
+       while (mypath[0] != '\0') {
+               begin = strchr(mypath, '\\');
+               if (begin != NULL) {
+                       *begin = '\0';
+               }
 
-               if (keyname[0] != '\0') {
-                       if (!ldb_dn_add_base_fmt(ret, "key=%s",
-                                                reg_ldb_escape(mem_ctx,
-                                                               keyname)))
-                       {
-                               talloc_free(ret);
-                               return NULL;
-                       }
+               if (!ldb_dn_add_child_fmt(ret, "key=%s",
+                                         reg_ldb_escape(mem_ctx, mypath))) {
+                       talloc_free(ret);
+                       return NULL;
                }
 
-               if(begin) {
-                       *begin = '\0';
+               if (begin != NULL) {
+                       mypath = begin + 1;
                } else {
                        break;
                }
        }
 
-       ldb_dn_add_base(ret, kd->dn);
-
        return ret;
 }
 
@@ -407,7 +424,8 @@ static WERROR ldb_get_default_value(TALLOC_CTX *mem_ctx,
        struct ldb_result *res;
        int ret;
 
-       ret = ldb_search(c, mem_ctx, &res, kd->dn, LDB_SCOPE_BASE, attrs, "(dn=*)");
+       ret = ldb_search(c, mem_ctx, &res, kd->dn, LDB_SCOPE_BASE, attrs,
+                        NULL);
 
        if (ret != LDB_SUCCESS) {
                DEBUG(0, ("Error getting default value for '%s': %s\n",
@@ -415,8 +433,10 @@ static WERROR ldb_get_default_value(TALLOC_CTX *mem_ctx,
                return WERR_FOOBAR;
        }
 
-       if (res->count == 0 || res->msgs[0]->num_elements == 0)
-               return WERR_BADFILE;
+       if (res->count == 0 || res->msgs[0]->num_elements == 0) {
+               talloc_free(res);
+               return WERR_FILE_NOT_FOUND;
+       }
 
        if ((data_type != NULL) && (data != NULL)) {
                reg_ldb_unpack_value(mem_ctx, res->msgs[0], name, data_type,
@@ -464,10 +484,6 @@ static WERROR ldb_get_value(TALLOC_CTX *mem_ctx, struct hive_key *k,
        const char *res_name;
        uint32_t idx;
 
-       if (name == NULL) {
-               return WERR_INVALID_PARAM;
-       }
-
        /* the default value was requested, give it back */
        if (name[0] == '\0') {
                return ldb_get_default_value(mem_ctx, k, NULL, data_type, data);
@@ -488,45 +504,45 @@ static WERROR ldb_get_value(TALLOC_CTX *mem_ctx, struct hive_key *k,
                }
        }
 
-       return WERR_BADFILE;
+       return WERR_FILE_NOT_FOUND;
 }
 
 static WERROR ldb_open_key(TALLOC_CTX *mem_ctx, const struct hive_key *h,
                           const char *name, struct hive_key **key)
 {
        struct ldb_result *res;
-       struct ldb_dn *ldap_path;
+       struct ldb_dn *ldb_path;
        int ret;
        struct ldb_key_data *newkd;
        struct ldb_key_data *kd = talloc_get_type(h, struct ldb_key_data);
        struct ldb_context *c = kd->ldb;
 
-       if (name == NULL) {
-               return WERR_INVALID_PARAM;
-       }
-
-       ldap_path = reg_path_to_ldb(mem_ctx, h, name, NULL);
-       W_ERROR_HAVE_NO_MEMORY(ldap_path);
+       ldb_path = reg_path_to_ldb(mem_ctx, h, name, NULL);
+       W_ERROR_HAVE_NO_MEMORY(ldb_path);
 
-       ret = ldb_search(c, mem_ctx, &res, ldap_path, LDB_SCOPE_BASE, NULL, "(key=*)");
+       ret = ldb_search(c, mem_ctx, &res, ldb_path, LDB_SCOPE_BASE, NULL,
+                        NULL);
 
        if (ret != LDB_SUCCESS) {
                DEBUG(3, ("Error opening key '%s': %s\n",
-                       ldb_dn_get_linearized(ldap_path), ldb_errstring(c)));
+                       ldb_dn_get_linearized(ldb_path), ldb_errstring(c)));
                return WERR_FOOBAR;
        } else if (res->count == 0) {
                DEBUG(3, ("Key '%s' not found\n",
-                       ldb_dn_get_linearized(ldap_path)));
+                       ldb_dn_get_linearized(ldb_path)));
                talloc_free(res);
-               return WERR_BADFILE;
+               return WERR_FILE_NOT_FOUND;
        }
 
        newkd = talloc_zero(mem_ctx, struct ldb_key_data);
+       W_ERROR_HAVE_NO_MEMORY(newkd);
        newkd->key.ops = &reg_backend_ldb;
        newkd->ldb = talloc_reference(newkd, kd->ldb);
-       newkd->dn = ldb_dn_copy(mem_ctx, res->msgs[0]->dn);
+       newkd->dn = ldb_dn_copy(newkd, res->msgs[0]->dn);
        newkd->classname = talloc_steal(newkd,
-                                       ldb_msg_find_attr_as_string(res->msgs[0], "classname", NULL);
+                                       ldb_msg_find_attr_as_string(res->msgs[0], "classname", NULL));
+
+       talloc_free(res);
 
        *key = (struct hive_key *)newkd;
 
@@ -545,7 +561,7 @@ WERROR reg_open_ldb_file(TALLOC_CTX *parent_ctx, const char *location,
        struct ldb_message *attrs_msg;
 
        if (location == NULL)
-               return WERR_INVALID_PARAM;
+               return WERR_INVALID_PARAMETER;
 
        wrap = ldb_wrap_connect(parent_ctx, ev_ctx, lp_ctx,
                                location, session_info, credentials, 0);
@@ -583,26 +599,28 @@ static WERROR ldb_add_key(TALLOC_CTX *mem_ctx, const struct hive_key *parent,
                          struct hive_key **newkey)
 {
        struct ldb_key_data *parentkd = discard_const_p(struct ldb_key_data, parent);
+       struct ldb_dn *ldb_path;
        struct ldb_message *msg;
        struct ldb_key_data *newkd;
        int ret;
 
-       if (name == NULL) {
-               return WERR_INVALID_PARAM;
-       }
+       ldb_path = reg_path_to_ldb(mem_ctx, parent, name, NULL);
+       W_ERROR_HAVE_NO_MEMORY(ldb_path);
 
        msg = ldb_msg_new(mem_ctx);
        W_ERROR_HAVE_NO_MEMORY(msg);
 
-       msg->dn = reg_path_to_ldb(msg, parent, name, NULL);
-       W_ERROR_HAVE_NO_MEMORY(msg->dn);
+       msg->dn = ldb_path;
 
-       ldb_msg_add_string(msg, "key", talloc_strdup(mem_ctx, name));
-       if (classname != NULL)
-               ldb_msg_add_string(msg, "classname",
-                                  talloc_strdup(mem_ctx, classname));
+       ldb_msg_add_string(msg, "key", name);
+       if (classname != NULL) {
+               ldb_msg_add_string(msg, "classname", classname);
+       }
 
        ret = ldb_add(parentkd->ldb, msg);
+
+       talloc_free(msg);
+
        if (ret == LDB_ERR_ENTRY_ALREADY_EXISTS) {
                return WERR_ALREADY_EXISTS;
        }
@@ -612,13 +630,13 @@ static WERROR ldb_add_key(TALLOC_CTX *mem_ctx, const struct hive_key *parent,
                return WERR_FOOBAR;
        }
 
-       DEBUG(2, ("key added: %s\n", ldb_dn_get_linearized(msg->dn)));
+       DEBUG(2, ("key added: %s\n", ldb_dn_get_linearized(ldb_path)));
 
        newkd = talloc_zero(mem_ctx, struct ldb_key_data);
        W_ERROR_HAVE_NO_MEMORY(newkd);
        newkd->ldb = talloc_reference(newkd, parentkd->ldb);
        newkd->key.ops = &reg_backend_ldb;
-       newkd->dn = talloc_steal(newkd, msg->dn);
+       newkd->dn = talloc_steal(newkd, ldb_path);
        newkd->classname = talloc_steal(newkd, classname);
 
        *newkey = (struct hive_key *)newkd;
@@ -638,21 +656,29 @@ static WERROR ldb_del_value(TALLOC_CTX *mem_ctx, struct hive_key *key,
        struct ldb_message *msg;
        struct ldb_dn *childdn;
 
-       if (child == NULL) {
-               return WERR_INVALID_PARAM;
-       }
-
        if (child[0] == '\0') {
                /* default value */
                msg = talloc_zero(mem_ctx, struct ldb_message);
                W_ERROR_HAVE_NO_MEMORY(msg);
                msg->dn = ldb_dn_copy(msg, kd->dn);
                W_ERROR_HAVE_NO_MEMORY(msg->dn);
-               ldb_msg_add_empty(msg, "data", LDB_FLAG_MOD_DELETE, NULL);
-               ldb_msg_add_empty(msg, "type", LDB_FLAG_MOD_DELETE, NULL);
+               ret = ldb_msg_add_empty(msg, "data", LDB_FLAG_MOD_DELETE, NULL);
+               if (ret != LDB_SUCCESS) {
+                       return WERR_FOOBAR;
+               }
+               ret = ldb_msg_add_empty(msg, "type", LDB_FLAG_MOD_DELETE,
+                                       NULL);
+               if (ret != LDB_SUCCESS) {
+                       return WERR_FOOBAR;
+               }
 
                ret = ldb_modify(kd->ldb, msg);
-               if (ret != LDB_SUCCESS) {
+
+               talloc_free(msg);
+
+               if (ret == LDB_ERR_NO_SUCH_ATTRIBUTE) {
+                       return WERR_FILE_NOT_FOUND;
+               } else if (ret != LDB_SUCCESS) {
                        DEBUG(1, ("ldb_del_value: %s\n", ldb_errstring(kd->ldb)));
                        return WERR_FOOBAR;
                }
@@ -671,7 +697,7 @@ static WERROR ldb_del_value(TALLOC_CTX *mem_ctx, struct hive_key *key,
                talloc_free(childdn);
 
                if (ret == LDB_ERR_NO_SUCH_OBJECT) {
-                       return WERR_BADFILE;
+                       return WERR_FILE_NOT_FOUND;
                } else if (ret != LDB_SUCCESS) {
                        DEBUG(1, ("ldb_del_value: %s\n", ldb_errstring(kd->ldb)));
                        return WERR_FOOBAR;
@@ -691,43 +717,39 @@ static WERROR ldb_del_key(TALLOC_CTX *mem_ctx, const struct hive_key *key,
        unsigned int i;
        int ret;
        struct ldb_key_data *parentkd = talloc_get_type(key, struct ldb_key_data);
-       struct ldb_dn *ldap_path;
+       struct ldb_dn *ldb_path;
        struct ldb_context *c = parentkd->ldb;
        struct ldb_result *res_keys;
        struct ldb_result *res_vals;
        WERROR werr;
        struct hive_key *hk;
 
-       if (name == NULL) {
-               return WERR_INVALID_PARAM;
-       }
-
        /* Verify key exists by opening it */
        werr = ldb_open_key(mem_ctx, key, name, &hk);
        if (!W_ERROR_IS_OK(werr)) {
                return werr;
        }
 
-       ldap_path = reg_path_to_ldb(mem_ctx, key, name, NULL);
-       W_ERROR_HAVE_NO_MEMORY(ldap_path);
+       ldb_path = reg_path_to_ldb(mem_ctx, key, name, NULL);
+       W_ERROR_HAVE_NO_MEMORY(ldb_path);
 
        /* Search for subkeys */
-       ret = ldb_search(c, mem_ctx, &res_keys, ldap_path, LDB_SCOPE_ONELEVEL,
+       ret = ldb_search(c, mem_ctx, &res_keys, ldb_path, LDB_SCOPE_ONELEVEL,
                         NULL, "(key=*)");
 
        if (ret != LDB_SUCCESS) {
                DEBUG(0, ("Error getting subkeys for '%s': %s\n",
-                     ldb_dn_get_linearized(ldap_path), ldb_errstring(c)));
+                     ldb_dn_get_linearized(ldb_path), ldb_errstring(c)));
                return WERR_FOOBAR;
        }
 
        /* Search for values */
-       ret = ldb_search(c, mem_ctx, &res_vals, ldap_path, LDB_SCOPE_ONELEVEL,
+       ret = ldb_search(c, mem_ctx, &res_vals, ldb_path, LDB_SCOPE_ONELEVEL,
                         NULL, "(value=*)");
 
        if (ret != LDB_SUCCESS) {
                DEBUG(0, ("Error getting values for '%s': %s\n",
-                     ldb_dn_get_linearized(ldap_path), ldb_errstring(c)));
+                     ldb_dn_get_linearized(ldb_path), ldb_errstring(c)));
                return WERR_FOOBAR;
        }
 
@@ -767,9 +789,11 @@ static WERROR ldb_del_key(TALLOC_CTX *mem_ctx, const struct hive_key *key,
                        }
                }
        }
+       talloc_free(res_keys);
+       talloc_free(res_vals);
 
        /* Delete the key itself */
-       ret = ldb_delete(c, ldap_path);
+       ret = ldb_delete(c, ldb_path);
 
        if (ret != LDB_SUCCESS)
        {
@@ -805,10 +829,6 @@ static WERROR ldb_set_value(struct hive_key *parent,
        int ret;
        TALLOC_CTX *mem_ctx = talloc_init("ldb_set_value");
 
-       if (name == NULL) {
-               return WERR_INVALID_PARAM;
-       }
-
        msg = reg_ldb_pack_value(kd->ldb, mem_ctx, name, type, data);
        W_ERROR_HAVE_NO_MEMORY(msg);
 
@@ -836,7 +856,7 @@ static WERROR ldb_set_value(struct hive_key *parent,
        if (ret == LDB_ERR_NO_SUCH_OBJECT) {
                i = 0;
                while (i < msg->num_elements) {
-                       if (msg->elements[i].flags == LDB_FLAG_MOD_DELETE) {
+                       if (LDB_FLAG_MOD_TYPE(msg->elements[i].flags) == LDB_FLAG_MOD_DELETE) {
                                ldb_msg_remove_element(msg, &msg->elements[i]);
                        } else {
                                ++i;
@@ -849,6 +869,8 @@ static WERROR ldb_set_value(struct hive_key *parent,
                ret = LDB_SUCCESS;
        }
 
+       talloc_free(msg);
+
        if (ret != LDB_SUCCESS) {
                DEBUG(1, ("ldb_set_value: %s\n", ldb_errstring(kd->ldb)));
                talloc_free(mem_ctx);
@@ -900,7 +922,7 @@ static WERROR ldb_get_key_info(TALLOC_CTX *mem_ctx,
         * remain { NULL, 0 }. */
        werr = ldb_get_default_value(mem_ctx, key, NULL, &default_value_type,
                                     &default_value);
-       if ((!W_ERROR_IS_OK(werr)) && (!W_ERROR_EQUAL(werr, WERR_BADFILE))) {
+       if ((!W_ERROR_IS_OK(werr)) && (!W_ERROR_EQUAL(werr, WERR_FILE_NOT_FOUND))) {
                return werr;
        }