Use libnet_smbconf_reg_open_path() in libnet_smbconf_drop().
[ira/wip.git] / source3 / libnet / libnet_conf.c
index a8a8e01538ee5e23f5552948b756ea6fd5fb886a..ca5b0c408fb9addc1258adc17ea4fa9783da443c 100644 (file)
@@ -1,7 +1,7 @@
 /*
  *  Unix SMB/CIFS implementation.
  *  libnet smbconf registry Support
- *  Copyright (C) Michael Adam 2007
+ *  Copyright (C) Michael Adam 2007-2008
  *  Copyright (C) Guenther Deschner 2007
  *
  *  This program is free software; you can redistribute it and/or modify
@@ -19,6 +19,7 @@
  */
 
 #include "includes.h"
+#include "libnet/libnet.h"
 
 /**********************************************************************
  *
 
  **********************************************************************/
 
-/*
+/**
+ * add a string to a talloced array of strings.
+ */
+static WERROR libnet_smbconf_add_string_to_array(TALLOC_CTX *mem_ctx,
+                                               char ***array,
+                                               uint32_t count,
+                                               const char *string)
+{
+       char **new_array = NULL;
+
+       if ((array == NULL) || (string == NULL)) {
+               return WERR_INVALID_PARAM;
+       }
+
+       new_array = TALLOC_REALLOC_ARRAY(mem_ctx, *array, char *, count + 1);
+       if (new_array == NULL) {
+               return WERR_NOMEM;
+       }
+
+       new_array[count] = talloc_strdup(new_array, string);
+
+       *array = new_array;
+
+       return WERR_OK;
+}
+
+/**
  * Open a subkey of KEY_SMBCONF (i.e a service)
  */
-WERROR libnet_smbconf_reg_open_path(TALLOC_CTX *ctx,
-                                   const char *subkeyname,
-                                   uint32 desired_access,
-                                   struct registry_key **key)
+static WERROR libnet_smbconf_reg_open_path(TALLOC_CTX *mem_ctx,
+                                          const char *path,
+                                          uint32 desired_access,
+                                          struct registry_key **key)
 {
        WERROR werr = WERR_OK;
-       char *path = NULL;
        NT_USER_TOKEN *token;
 
-       if (!(token = registry_create_admin_token(ctx))) {
-               DEBUG(1, ("Error creating admin token\n"));
+       if (path == NULL) {
+               DEBUG(1, ("Error: NULL path string given\n"));
+               werr = WERR_INVALID_PARAM;
                goto done;
        }
 
-       if (subkeyname == NULL) {
-               path = talloc_strdup(ctx, KEY_SMBCONF);
-       } else {
-               path = talloc_asprintf(ctx, "%s\\%s", KEY_SMBCONF, subkeyname);
+       token = registry_create_admin_token(mem_ctx);
+       if (token == NULL) {
+               DEBUG(1, ("Error creating admin token\n"));
+               /* what is the appropriate error code here? */
+               werr = WERR_CAN_NOT_COMPLETE;
+               goto done;
        }
 
-       werr = reg_open_path(ctx, path, desired_access,
-                            token, key);
+       werr = reg_open_path(mem_ctx, path, desired_access, token, key);
 
        if (!W_ERROR_IS_OK(werr)) {
                DEBUG(1, ("Error opening registry path '%s': %s\n",
@@ -59,37 +87,45 @@ WERROR libnet_smbconf_reg_open_path(TALLOC_CTX *ctx,
        }
 
 done:
-       TALLOC_FREE(path);
        return werr;
 }
 
-/*
- * open the base key KEY_SMBCONF
+/**
+ * Open a subkey of KEY_SMBCONF (i.e a service)
  */
-WERROR libnet_smbconf_reg_open_basepath(TALLOC_CTX *ctx, uint32 desired_access,
-                                       struct registry_key **key)
+static WERROR libnet_smbconf_reg_open_service_key(TALLOC_CTX *ctx,
+                                                 const char *servicename,
+                                                 uint32 desired_access,
+                                                 struct registry_key **key)
 {
-       return libnet_smbconf_reg_open_path(ctx, NULL, desired_access, key);
+       WERROR werr = WERR_OK;
+       char *path = NULL;
+       NT_USER_TOKEN *token;
+
+       if (servicename == NULL) {
+               DEBUG(3, ("Error: NULL servicename given.\n"));
+               werr = WERR_INVALID_PARAM;
+               goto done;
+       }
+
+       path = talloc_asprintf(ctx, "%s\\%s", KEY_SMBCONF, servicename);
+
+       werr = libnet_smbconf_reg_open_path(ctx, path, desired_access, key);
+
+done:
+       TALLOC_FREE(path);
+       return werr;
 }
 
 /*
- * check if a subkey of KEY_SMBCONF of a given name exists
+ * open the base key KEY_SMBCONF
  */
-bool libnet_smbconf_key_exists(const char *subkeyname)
+static WERROR libnet_smbconf_reg_open_basekey(TALLOC_CTX *ctx,
+                                             uint32 desired_access,
+                                             struct registry_key **key)
 {
-       bool ret = false;
-       WERROR werr = WERR_OK;
-       TALLOC_CTX *mem_ctx = talloc_stackframe();
-       struct registry_key *key = NULL;
-
-       werr = libnet_smbconf_reg_open_path(mem_ctx, subkeyname, REG_KEY_READ,
-                                           &key);
-       if (W_ERROR_IS_OK(werr)) {
-               ret = true;
-       }
-
-       TALLOC_FREE(mem_ctx);
-       return ret;
+       return libnet_smbconf_reg_open_path(ctx, KEY_SMBCONF, desired_access,
+                                           key);
 }
 
 static bool libnet_smbconf_value_exists(struct registry_key *key,
@@ -112,9 +148,9 @@ static bool libnet_smbconf_value_exists(struct registry_key *key,
 /*
  * create a subkey of KEY_SMBCONF
  */
-WERROR libnet_smbconf_reg_createkey_internal(TALLOC_CTX *ctx,
-                                            const char * subkeyname,
-                                            struct registry_key **newkey)
+static WERROR libnet_smbconf_reg_createkey_internal(TALLOC_CTX *ctx,
+                                                   const char * subkeyname,
+                                                   struct registry_key **newkey)
 {
        WERROR werr = WERR_OK;
        struct registry_key *create_parent = NULL;
@@ -129,8 +165,8 @@ WERROR libnet_smbconf_reg_createkey_internal(TALLOC_CTX *ctx,
                goto done;
        }
 
-       werr = libnet_smbconf_reg_open_basepath(create_ctx, REG_KEY_WRITE,
-                                               &create_parent);
+       werr = libnet_smbconf_reg_open_basekey(create_ctx, REG_KEY_WRITE,
+                                              &create_parent);
        if (!W_ERROR_IS_OK(werr)) {
                goto done;
        }
@@ -138,12 +174,12 @@ WERROR libnet_smbconf_reg_createkey_internal(TALLOC_CTX *ctx,
        werr = reg_createkey(ctx, create_parent, subkeyname,
                             REG_KEY_WRITE, newkey, &action);
        if (W_ERROR_IS_OK(werr) && (action != REG_CREATED_NEW_KEY)) {
-               d_fprintf(stderr, "Key '%s' already exists.\n", subkeyname);
+               DEBUG(10, ("Key '%s' already exists.\n", subkeyname));
                werr = WERR_ALREADY_EXISTS;
        }
        if (!W_ERROR_IS_OK(werr)) {
-               d_fprintf(stderr, "Error creating key %s: %s\n",
-                        subkeyname, dos_errstr(werr));
+               DEBUG(5, ("Error creating key %s: %s\n",
+                        subkeyname, dos_errstr(werr)));
        }
 
 done:
@@ -154,7 +190,7 @@ done:
 /*
  * add a value to a key.
  */
-WERROR libnet_smbconf_reg_setvalue_internal(struct registry_key *key,
+static WERROR libnet_smbconf_reg_setvalue_internal(struct registry_key *key,
                                                   const char *valname,
                                                   const char *valstr)
 {
@@ -169,11 +205,11 @@ WERROR libnet_smbconf_reg_setvalue_internal(struct registry_key *key,
                                                  &canon_valstr))
        {
                if (canon_valname == NULL) {
-                       d_fprintf(stderr, "invalid parameter '%s' given\n",
-                                 valname);
+                       DEBUG(5, ("invalid parameter '%s' given\n",
+                                 valname));
                } else {
-                       d_fprintf(stderr, "invalid value '%s' given for "
-                                 "parameter '%s'\n", valstr, valname);
+                       DEBUG(5, ("invalid value '%s' given for "
+                                 "parameter '%s'\n", valstr, valname));
                }
                werr = WERR_INVALID_PARAM;
                goto done;
@@ -186,16 +222,16 @@ WERROR libnet_smbconf_reg_setvalue_internal(struct registry_key *key,
        val.v.sz.len = strlen(canon_valstr) + 1;
 
        if (registry_smbconf_valname_forbidden(canon_valname)) {
-               d_fprintf(stderr, "Parameter '%s' not allowed in registry.\n",
-                         canon_valname);
+               DEBUG(5, ("Parameter '%s' not allowed in registry.\n",
+                         canon_valname));
                werr = WERR_INVALID_PARAM;
                goto done;
        }
 
        subkeyname = strrchr_m(key->key->name, '\\');
        if ((subkeyname == NULL) || (*(subkeyname +1) == '\0')) {
-               d_fprintf(stderr, "Invalid registry key '%s' given as "
-                         "smbconf section.\n", key->key->name);
+               DEBUG(5, ("Invalid registry key '%s' given as "
+                         "smbconf section.\n", key->key->name));
                werr = WERR_INVALID_PARAM;
                goto done;
        }
@@ -203,19 +239,18 @@ WERROR libnet_smbconf_reg_setvalue_internal(struct registry_key *key,
        if (!strequal(subkeyname, GLOBAL_NAME) &&
            lp_parameter_is_global(valname))
        {
-               d_fprintf(stderr, "Global paramter '%s' not allowed in "
+               DEBUG(5, ("Global paramter '%s' not allowed in "
                          "service definition ('%s').\n", canon_valname,
-                         subkeyname);
+                         subkeyname));
                werr = WERR_INVALID_PARAM;
                goto done;
        }
 
        werr = reg_setvalue(key, canon_valname, &val);
        if (!W_ERROR_IS_OK(werr)) {
-               d_fprintf(stderr,
-                         "Error adding value '%s' to "
+               DEBUG(5, ("Error adding value '%s' to "
                          "key '%s': %s\n",
-                         canon_valname, key->key->name, dos_errstr(werr));
+                         canon_valname, key->key->name, dos_errstr(werr)));
        }
 
 done:
@@ -229,8 +264,8 @@ done:
  * which are ar stored as REG_SZ values, so the incomplete
  * handling should be ok.
  */
-char *libnet_smbconf_format_registry_value(TALLOC_CTX *mem_ctx,
-                                          struct registry_value *value)
+static char *libnet_smbconf_format_registry_value(TALLOC_CTX *mem_ctx,
+                                                 struct registry_value *value)
 {
        char *result = NULL;
 
@@ -266,6 +301,80 @@ char *libnet_smbconf_format_registry_value(TALLOC_CTX *mem_ctx,
        return result;
 }
 
+/**
+ * Get the values of a key as a list of value names
+ * and a list of value strings (ordered)
+ */
+static WERROR libnet_smbconf_reg_get_values(TALLOC_CTX *mem_ctx,
+                                           struct registry_key *key,
+                                           uint32_t *num_values,
+                                           char ***value_names,
+                                           char ***value_strings)
+{
+       TALLOC_CTX *tmp_ctx = NULL;
+       WERROR werr = WERR_OK;
+       uint32_t count;
+       struct registry_value *valvalue = NULL;
+       char *valname = NULL;
+       char **tmp_valnames = NULL;
+       char **tmp_valstrings = NULL;
+
+       if ((num_values == NULL) || (value_names == NULL) ||
+           (value_strings == NULL))
+       {
+               werr = WERR_INVALID_PARAM;
+               goto done;
+       }
+
+       tmp_ctx = talloc_new(mem_ctx);
+       if (tmp_ctx == NULL) {
+               werr = WERR_NOMEM;
+               goto done;
+       }
+
+       for (count = 0;
+            W_ERROR_IS_OK(werr = reg_enumvalue(tmp_ctx, key, count, &valname,
+                                               &valvalue));
+            count++)
+       {
+               char *valstring;
+
+               werr = libnet_smbconf_add_string_to_array(tmp_ctx,
+                                                         &tmp_valnames,
+                                                         count, valname);
+               if (!W_ERROR_IS_OK(werr)) {
+                       goto done;
+               }
+
+               valstring = libnet_smbconf_format_registry_value(tmp_ctx,
+                                                                valvalue);
+               werr = libnet_smbconf_add_string_to_array(tmp_ctx,
+                                                         &tmp_valstrings,
+                                                         count,
+                                                         valstring);
+               if (!W_ERROR_IS_OK(werr)) {
+                       goto done;
+               }
+       }
+       if (!W_ERROR_EQUAL(WERR_NO_MORE_ITEMS, werr)) {
+               goto done;
+       }
+
+       werr = WERR_OK;
+
+       *num_values = count;
+       if (count > 0) {
+               *value_names = talloc_move(mem_ctx, &tmp_valnames);
+               *value_strings = talloc_move(mem_ctx, &tmp_valstrings);
+       } else {
+               *value_names = NULL;
+               *value_strings = NULL;
+       }
+
+done:
+       TALLOC_FREE(tmp_ctx);
+       return werr;
+}
 
 /**********************************************************************
  *
@@ -286,12 +395,6 @@ WERROR libnet_smbconf_drop(void)
        TALLOC_CTX* mem_ctx = talloc_stackframe();
        enum winreg_CreateAction action;
 
-       if (!(token = registry_create_admin_token(mem_ctx))) {
-               /* what is the appropriate error code here? */
-               werr = WERR_CAN_NOT_COMPLETE;
-               goto done;
-       }
-
        path = talloc_strdup(mem_ctx, KEY_SMBCONF);
        if (path == NULL) {
                werr = WERR_NOMEM;
@@ -299,7 +402,8 @@ WERROR libnet_smbconf_drop(void)
        }
        p = strrchr(path, '\\');
        *p = '\0';
-       werr = reg_open_path(mem_ctx, path, REG_KEY_WRITE, token, &parent_key);
+       werr = libnet_smbconf_reg_open_path(mem_ctx, path, REG_KEY_WRITE,
+                                           &parent_key);
 
        if (!W_ERROR_IS_OK(werr)) {
                goto done;
@@ -319,6 +423,212 @@ done:
        return werr;
 }
 
+/**
+ * Get the whole configuration as lists of strings with counts:
+ *
+ *  num_shares   : number of shares
+ *  share_names  : list of length num_shares of share names
+ *  num_params   : list of length num_shares of parameter counts for each share
+ *  param_names  : list of lists of parameter names for each share
+ *  param_values : list of lists of parameter values for each share
+ */
+WERROR libnet_smbconf_get_config(TALLOC_CTX *mem_ctx, uint32_t *num_shares,
+                                char ***share_names, uint32_t **num_params,
+                                char ****param_names, char ****param_values)
+{
+       WERROR werr = WERR_OK;
+       TALLOC_CTX *tmp_ctx = NULL;
+       uint32_t tmp_num_shares;
+       char **tmp_share_names;
+       uint32_t *tmp_num_params;
+       char ***tmp_param_names;
+       char ***tmp_param_values;
+       uint32_t count;
+
+       if ((num_shares == NULL) || (share_names == NULL) ||
+           (num_params == NULL) || (param_names == NULL) ||
+           (param_values == NULL))
+       {
+               werr = WERR_INVALID_PARAM;
+               goto done;
+       }
+
+       tmp_ctx = talloc_new(mem_ctx);
+       if (tmp_ctx == NULL) {
+               werr = WERR_NOMEM;
+               goto done;
+       }
+
+       werr = libnet_smbconf_get_share_names(tmp_ctx, &tmp_num_shares,
+                                             &tmp_share_names);
+       if (!W_ERROR_IS_OK(werr)) {
+               goto done;
+       }
+
+       tmp_num_params   = TALLOC_ARRAY(tmp_ctx, uint32_t, tmp_num_shares);
+       tmp_param_names  = TALLOC_ARRAY(tmp_ctx, char **, tmp_num_shares);
+       tmp_param_values = TALLOC_ARRAY(tmp_ctx, char **, tmp_num_shares);
+
+       if ((tmp_num_params == NULL) || (tmp_param_names == NULL) ||
+           (tmp_param_values == NULL))
+       {
+               werr = WERR_NOMEM;
+               goto done;
+       }
+
+       for (count = 0; count < tmp_num_shares; count++) {
+               werr = libnet_smbconf_getshare(mem_ctx, tmp_share_names[count],
+                                              &tmp_num_params[count],
+                                              &tmp_param_names[count],
+                                              &tmp_param_values[count]);
+               if (!W_ERROR_IS_OK(werr)) {
+                       goto done;
+               }
+       }
+
+       werr = WERR_OK;
+
+       *num_shares = tmp_num_shares;
+       if (tmp_num_shares > 0) {
+               *share_names = talloc_move(mem_ctx, &tmp_share_names);
+               *num_params = talloc_move(mem_ctx, &tmp_num_params);
+               *param_names = talloc_move(mem_ctx, &tmp_param_names);
+               *param_values = talloc_move(mem_ctx, &tmp_param_values);
+       } else {
+               *share_names = NULL;
+               *num_params = NULL;
+               *param_names = NULL;
+               *param_values = NULL;
+       }
+
+done:
+       TALLOC_FREE(tmp_ctx);
+       return werr;
+}
+
+
+/**
+ * get the list of share names defined in the configuration.
+ */
+WERROR libnet_smbconf_get_share_names(TALLOC_CTX *mem_ctx, uint32_t *num_shares,
+                                     char ***share_names)
+{
+       uint32_t count;
+       uint32_t added_count = 0;
+       TALLOC_CTX *tmp_ctx = NULL;
+       WERROR werr = WERR_OK;
+       struct registry_key *key = NULL;
+       char *subkey_name = NULL;
+       char **tmp_share_names = NULL;
+
+       if ((num_shares == NULL) || (share_names == NULL)) {
+               werr = WERR_INVALID_PARAM;
+               goto done;
+       }
+
+       tmp_ctx = talloc_new(mem_ctx);
+       if (tmp_ctx == NULL) {
+               werr = WERR_NOMEM;
+               goto done;
+       }
+
+       /* make sure "global" is always listed first */
+       if (libnet_smbconf_share_exists(GLOBAL_NAME)) {
+               werr = libnet_smbconf_add_string_to_array(tmp_ctx,
+                                                         &tmp_share_names,
+                                                         0, GLOBAL_NAME);
+               if (!W_ERROR_IS_OK(werr)) {
+                       goto done;
+               }
+               added_count++;
+       }
+
+       werr = libnet_smbconf_reg_open_basekey(tmp_ctx,
+                                              SEC_RIGHTS_ENUM_SUBKEYS,
+                                              &key);
+       if (!W_ERROR_IS_OK(werr)) {
+               goto done;
+       }
+
+       for (count = 0;
+            W_ERROR_IS_OK(werr = reg_enumkey(tmp_ctx, key, count,
+                                             &subkey_name, NULL));
+            count++)
+       {
+               if (strequal(subkey_name, GLOBAL_NAME)) {
+                       continue;
+               }
+
+               werr = libnet_smbconf_add_string_to_array(tmp_ctx,
+                                                         &tmp_share_names,
+                                                         added_count,
+                                                         subkey_name);
+               if (!W_ERROR_IS_OK(werr)) {
+                       goto done;
+               }
+               added_count++;
+       }
+       if (!W_ERROR_EQUAL(WERR_NO_MORE_ITEMS, werr)) {
+               goto done;
+       }
+       werr = WERR_OK;
+
+       *num_shares = added_count;
+       if (added_count > 0) {
+               *share_names = talloc_move(mem_ctx, &tmp_share_names);
+       } else {
+               *share_names = NULL;
+       }
+
+done:
+       TALLOC_FREE(tmp_ctx);
+       return werr;
+}
+
+/**
+ * check if a share/service of a given name exists
+ */
+bool libnet_smbconf_share_exists(const char *servicename)
+{
+       bool ret = false;
+       WERROR werr = WERR_OK;
+       TALLOC_CTX *mem_ctx = talloc_stackframe();
+       struct registry_key *key = NULL;
+
+       werr = libnet_smbconf_reg_open_service_key(mem_ctx, servicename,
+                                                  REG_KEY_READ, &key);
+       if (W_ERROR_IS_OK(werr)) {
+               ret = true;
+       }
+
+       TALLOC_FREE(mem_ctx);
+       return ret;
+}
+
+/**
+ * get a definition of a share (service) from configuration.
+ */
+WERROR libnet_smbconf_getshare(TALLOC_CTX *mem_ctx, const char *servicename,
+                              uint32_t *num_params, char ***param_names,
+                              char ***param_values)
+{
+       WERROR werr = WERR_OK;
+       struct registry_key *key = NULL;
+
+       werr = libnet_smbconf_reg_open_service_key(mem_ctx, servicename,
+                                                  REG_KEY_READ, &key);
+       if (!W_ERROR_IS_OK(werr)) {
+               goto done;
+       }
+
+       werr = libnet_smbconf_reg_get_values(mem_ctx, key, num_params,
+                                            param_names, param_values);
+
+done:
+       TALLOC_FREE(key);
+       return werr;
+}
+
 /**
  * delete a service from configuration
  */
@@ -328,7 +638,7 @@ WERROR libnet_smbconf_delshare(const char *servicename)
        struct registry_key *key = NULL;
        TALLOC_CTX *ctx = talloc_stackframe();
 
-       werr = libnet_smbconf_reg_open_basepath(ctx, REG_KEY_WRITE, &key);
+       werr = libnet_smbconf_reg_open_basekey(ctx, REG_KEY_WRITE, &key);
        if (!W_ERROR_IS_OK(werr)) {
                goto done;
        }
@@ -351,12 +661,12 @@ WERROR libnet_smbconf_setparm(const char *service,
        struct registry_key *key = NULL;
        TALLOC_CTX *mem_ctx = talloc_stackframe();
 
-       if (!libnet_smbconf_key_exists(service)) {
+       if (!libnet_smbconf_share_exists(service)) {
                werr = libnet_smbconf_reg_createkey_internal(mem_ctx, service,
                                                             &key);
        } else {
-               werr = libnet_smbconf_reg_open_path(mem_ctx, service,
-                                                   REG_KEY_WRITE, &key);
+               werr = libnet_smbconf_reg_open_service_key(mem_ctx, service,
+                                                          REG_KEY_WRITE, &key);
        }
        if (!W_ERROR_IS_OK(werr)) {
                goto done;
@@ -386,13 +696,13 @@ WERROR libnet_smbconf_getparm(TALLOC_CTX *mem_ctx,
                goto done;
        }
 
-       if (!libnet_smbconf_key_exists(service)) {
+       if (!libnet_smbconf_share_exists(service)) {
                werr = WERR_NO_SUCH_SERVICE;
                goto done;
        }
 
-       werr = libnet_smbconf_reg_open_path(mem_ctx, service, REG_KEY_READ,
-                                           &key);
+       werr = libnet_smbconf_reg_open_service_key(mem_ctx, service,
+                                                  REG_KEY_READ, &key);
        if (!W_ERROR_IS_OK(werr)) {
                goto done;
        }
@@ -429,11 +739,12 @@ WERROR libnet_smbconf_delparm(const char *service,
        WERROR werr = WERR_OK;
        TALLOC_CTX *mem_ctx = talloc_stackframe();
 
-       if (!libnet_smbconf_key_exists(service)) {
+       if (!libnet_smbconf_share_exists(service)) {
                return WERR_NO_SUCH_SERVICE;
        }
 
-       werr = libnet_smbconf_reg_open_path(mem_ctx, service, REG_KEY_ALL, &key);
+       werr = libnet_smbconf_reg_open_service_key(mem_ctx, service,
+                                                  REG_KEY_ALL, &key);
        if (!W_ERROR_IS_OK(werr)) {
                goto done;
        }