common quota parameter parsing function implemented

diff --git a/src/plugins/quota/quota.c b/src/plugins/quota/quota.c
index 29f0b7b64ca9d0060adb5a532c86dce4761f5e58..e15df35ce857896b6033910c8509bea746810183 100644 (file)
--- a/src/plugins/quota/quota.c
+++ b/src/plugins/quota/quota.c
@@ -57,6 +57,16 @@ static const struct quota_backend *quota_internal_backends[] = {
 
 static ARRAY(const struct quota_backend*) quota_backends;
 
+static void hidden_param_handler(struct quota_root *_root, const char *param_value);
+static void ignoreunlim_param_handler(struct quota_root *_root, const char *param_value);
+static void noenforcing_param_handler(struct quota_root *_root, const char *param_value);
+static void ns_param_handler(struct quota_root *_root, const char *param_value);
+
+struct quota_param_parser quota_param_hidden = {.param_name = "hidden", .param_handler = hidden_param_handler};
+struct quota_param_parser quota_param_ignoreunlimited = {.param_name = "ignoreunlimited", .param_handler = ignoreunlim_param_handler};
+struct quota_param_parser quota_param_noenforcing = {.param_name = "noenforcing", .param_handler = noenforcing_param_handler};
+struct quota_param_parser quota_param_ns = {.param_name = "ns=", .param_handler = ns_param_handler};
+
 static enum quota_alloc_result quota_default_test_alloc(
                struct quota_transaction_context *ctx, uoff_t size);
 static void quota_over_flag_check_root(struct quota_root *root);
@@ -1369,3 +1379,73 @@ void quota_recalculate(struct quota_transaction_context *ctx,
 {
        ctx->recalculate = recalculate;
 }
+
+static void hidden_param_handler(struct quota_root *_root, const char *param_value ATTR_UNUSED)
+{
+       _root->hidden = TRUE;
+}
+
+static void ignoreunlim_param_handler(struct quota_root *_root, const char *param_value ATTR_UNUSED)
+{
+       _root->disable_unlimited_tracking = TRUE;
+}
+
+static void noenforcing_param_handler(struct quota_root *_root, const char *param_value ATTR_UNUSED)
+{
+       _root->no_enforcing = TRUE;
+}
+
+static void ns_param_handler(struct quota_root *_root, const char *param_value)
+{
+       _root->ns_prefix = p_strdup(_root->pool, param_value);
+}
+
+int quota_parse_parameters(struct quota_root *root, const char **args, const char **error_r,
+                          const struct quota_param_parser *valid_params, bool fail_on_unknown)
+{
+       const char *tmp_param_name, *tmp_param_val;
+       size_t tmp_param_len;
+
+       while (*args != NULL && (*args)[0] != '\0') {
+               for (; valid_params->param_name != NULL; ++valid_params) {
+                       tmp_param_name = valid_params->param_name;
+                       tmp_param_len = strlen(valid_params->param_name);
+                       if (strncmp(*args, tmp_param_name, tmp_param_len) == 0) {
+                               tmp_param_val = NULL;
+                               *args += tmp_param_len;
+                               if (tmp_param_name[tmp_param_len - 1] == '=') {
+                                       const char *next_colon = strchr(*args, ':');
+                                       tmp_param_val = (next_colon == NULL)?
+                                               t_strdup(*args):
+                                               t_strdup_until(*args, next_colon);
+                                       *args = (next_colon == NULL) ? NULL : next_colon + 1;
+                               }
+                               else if (*args[0] == '\0' ||
+                                        *args[0] == ':') {
+                                       *args = (*args[0] == ':') ? *args + 1 : NULL;
+                                       /* in case parameter is a boolean second parameter
+                                        * string parameter value will be ignored by param_handler
+                                        * we just need some non-NULL value
+                                        * to indicate that argument is to be processed */
+                                       tmp_param_val = "";
+                               }
+                               if (tmp_param_val != NULL) {
+                                       valid_params->param_handler(root, tmp_param_val);
+                                       break;
+                               }
+                       }
+               }
+               if (valid_params->param_name != NULL) {
+                       if (fail_on_unknown) {
+                               *error_r = t_strdup_printf(
+                                       "Unknown parameter for backend %s: %s",
+                                       root->backend.name, *args);
+                               return -1;
+                       }
+                       else {
+                               break;
+                       }
+               }
+       }
+       return 0;
+}

diff --git a/src/plugins/quota/quota.h b/src/plugins/quota/quota.h
index 293ccb41163ab96b0fb92332068089ef1dacbb6d..8663d2b2cb517aa627dc13a3a2f4533bc53de118 100644 (file)
--- a/src/plugins/quota/quota.h
+++ b/src/plugins/quota/quota.h
@@ -19,6 +19,16 @@ struct quota_root;
 struct quota_root_iter;
 struct quota_transaction_context;
 
+struct quota_param_parser {
+       char *param_name;
+       void (* param_handler)(struct quota_root *_root, const char *param_value);
+};
+
+extern struct quota_param_parser quota_param_hidden;
+extern struct quota_param_parser quota_param_ignoreunlimited;
+extern struct quota_param_parser quota_param_noenforcing;
+extern struct quota_param_parser quota_param_ns;
+
 enum quota_recalculate {
        QUOTA_RECALCULATE_DONT = 0,
        /* We may want to recalculate quota because we weren't able to call
@@ -108,4 +118,8 @@ void quota_recalculate(struct quota_transaction_context *ctx,
 /* Execute quota_over_scripts if needed. */
 void quota_over_flag_check_startup(struct quota *quota);
 
+/* Common quota parameters parsing loop */
+int quota_parse_parameters(struct quota_root *root, const char **args, const char **error_r,
+                          const struct quota_param_parser *valid_params, bool fail_on_unknown);
+
 #endif