- add multiple-result capable versions of match/regmatch/regsub

1 files changed, 129 insertions, 3 deletions

diff --git a/src/format.c b/src/format.c
index 60c3bf6..70d0f19 100644
--- a/src/format.c
+++ b/src/format.c
@@ -2066,7 +2066,7 @@ format_merge(struct plugin_state *state, Slapi_PBlock *pb, Slapi_Entry *e,
 /* Look up the entry's values for the attribute named by the first argument,
  * and use the callback to check if they match the second argument.  If we find
  * exactly one match, store it in the output buffer, otherwise store the text
- * of the default_arg'th argument if given, or return an error if no
+ * of the default_arg'th argument if given, or return everything if no
  * default_arg'th argument was given. */
 static int
 format_match_generic(struct plugin_state *state,
@@ -2156,7 +2156,8 @@ format_match_generic(struct plugin_state *state,
 		break;
 	case 0:
 	default:
-		/* Either no matches, or too many matches to store. */
+		/* Either no matches, or multiple matches, which may be too
+		 * many matches to store. */
 		default_value = NULL;
 		if ((default_arg >= 0) && (argv[default_arg] != NULL)) {
 			default_value = format_get_data(state, pb, e,
@@ -2170,7 +2171,11 @@ format_match_generic(struct plugin_state *state,
 							inref_attr_list,
 							&default_length);
 		}
-		if (default_value != NULL) {
+		if (default_arg < 0) {
+			/* Return all of the matches as a list. */
+			format_add_choice_str(outbuf_choices, outbuf, matches);
+			len = 0;
+		} else if ((default_arg >= 0) && (default_value != NULL)) {
 			if (count == 0) {
 				slapi_log_error(SLAPI_LOG_PLUGIN, plugin_id,
 						"%s: no matching value "
@@ -2282,6 +2287,25 @@ format_match(struct plugin_state *state,
 				    ref_attr_list, inref_attr_list,
 				    "format_match", format_match_cb);
 }
+static int
+format_mmatch(struct plugin_state *state,
+	      Slapi_PBlock *pb, Slapi_Entry *e,
+	      const char *group, const char *set,
+	      const char *args, const char *disallowed,
+	      char *outbuf, int outbuf_len,
+	      struct format_choice **outbuf_choices,
+	      char ***rel_attrs, char ***ref_attrs,
+	      struct format_inref_attr ***inref_attrs,
+	      struct format_ref_attr_list ***ref_attr_list,
+	      struct format_ref_attr_list ***inref_attr_list)
+{
+	return format_match_generic(state, pb, e, group, set, args, 2, -1,
+				    disallowed,
+				    outbuf, outbuf_len, outbuf_choices,
+				    rel_attrs, ref_attrs, inref_attrs,
+				    ref_attr_list, inref_attr_list,
+				    "format_mmatch", format_match_cb);
+}
 
 /* Check for a regex match. */
 static char *
@@ -2323,6 +2347,25 @@ format_regmatch(struct plugin_state *state,
 				    ref_attr_list, inref_attr_list,
 				    "format_regmatch", format_regmatch_cb);
 }
+static int
+format_mregmatch(struct plugin_state *state,
+		 Slapi_PBlock *pb, Slapi_Entry *e,
+		 const char *group, const char *set,
+		 const char *args, const char *disallowed,
+		 char *outbuf, int outbuf_len,
+		 struct format_choice **outbuf_choices,
+		 char ***rel_attrs, char ***ref_attrs,
+		 struct format_inref_attr ***inref_attrs,
+		 struct format_ref_attr_list ***ref_attr_list,
+		 struct format_ref_attr_list ***inref_attr_list)
+{
+	return format_match_generic(state, pb, e, group, set, args, 2, -1,
+				    disallowed,
+				    outbuf, outbuf_len, outbuf_choices,
+				    rel_attrs, ref_attrs, inref_attrs,
+				    ref_attr_list, inref_attr_list,
+				    "format_mregmatch", format_regmatch_cb);
+}
 static char *
 format_regmatchi_cb(const char *pattern, const char *value, char **argv)
 {
@@ -2347,6 +2390,25 @@ format_regmatchi(struct plugin_state *state,
 				    ref_attr_list, inref_attr_list,
 				    "format_regmatchi", format_regmatchi_cb);
 }
+static int
+format_mregmatchi(struct plugin_state *state,
+		  Slapi_PBlock *pb, Slapi_Entry *e,
+		  const char *group, const char *set,
+		  const char *args, const char *disallowed,
+		  char *outbuf, int outbuf_len,
+		  struct format_choice **outbuf_choices,
+		  char ***rel_attrs, char ***ref_attrs,
+		  struct format_inref_attr ***inref_attrs,
+		  struct format_ref_attr_list ***ref_attr_list,
+		  struct format_ref_attr_list ***inref_attr_list)
+{
+	return format_match_generic(state, pb, e, group, set, args, 2, -1,
+				    disallowed,
+				    outbuf, outbuf_len, outbuf_choices,
+				    rel_attrs, ref_attrs, inref_attrs,
+				    ref_attr_list, inref_attr_list,
+				    "format_mregmatchi", format_regmatchi_cb);
+}
 
 /* Check for a regex match and build a custom result from the matching value. */
 static char *
@@ -2481,6 +2543,25 @@ format_regsub(struct plugin_state *state,
 				    ref_attr_list, inref_attr_list,
 				    "format_regsub", format_regsub_cb);
 }
+static int
+format_mregsub(struct plugin_state *state,
+	       Slapi_PBlock *pb, Slapi_Entry *e,
+	       const char *group, const char *set,
+	       const char *args, const char *disallowed,
+	       char *outbuf, int outbuf_len,
+	       struct format_choice **outbuf_choices,
+	       char ***rel_attrs, char ***ref_attrs,
+	       struct format_inref_attr ***inref_attrs,
+	       struct format_ref_attr_list ***ref_attr_list,
+	       struct format_ref_attr_list ***inref_attr_list)
+{
+	return format_match_generic(state, pb, e, group, set, args, 3, -1,
+				    disallowed,
+				    outbuf, outbuf_len, outbuf_choices,
+				    rel_attrs, ref_attrs, inref_attrs,
+				    ref_attr_list, inref_attr_list,
+				    "format_mregsub", format_regsub_cb);
+}
 static char *
 format_regsubi_cb(const char *pattern, const char *value, char **argv)
 {
@@ -2505,6 +2586,25 @@ format_regsubi(struct plugin_state *state,
 				    ref_attr_list, inref_attr_list,
 				    "format_regsubi", format_regsubi_cb);
 }
+static int
+format_mregsubi(struct plugin_state *state,
+	        Slapi_PBlock *pb, Slapi_Entry *e,
+	        const char *group, const char *set,
+	        const char *args, const char *disallowed,
+	        char *outbuf, int outbuf_len,
+	        struct format_choice **outbuf_choices,
+	        char ***rel_attrs, char ***ref_attrs,
+	        struct format_inref_attr ***inref_attrs,
+	        struct format_ref_attr_list ***ref_attr_list,
+	        struct format_ref_attr_list ***inref_attr_list)
+{
+	return format_match_generic(state, pb, e, group, set, args, 3, -1,
+				    disallowed,
+				    outbuf, outbuf_len, outbuf_choices,
+				    rel_attrs, ref_attrs, inref_attrs,
+				    ref_attr_list, inref_attr_list,
+				    "format_mregsubi", format_regsubi_cb);
+}
 
 /* If the attribute given by the first argument is equal to the value given in
  * the second, return the third, else return the fourth.  We used to allow an
@@ -2633,6 +2733,13 @@ format_collect(struct plugin_state *state, Slapi_PBlock *pb, Slapi_Entry *e,
 		format_free_parsed_args(argv);
 		return -EINVAL;
 	}
+	if (outbuf_choices == NULL) {
+		slapi_log_error(SLAPI_LOG_PLUGIN, state->plugin_desc->spd_id,
+				"collect: returns a list, but a list "
+				"would not be appropriate\n");
+		format_free_parsed_args(argv);
+		return -EINVAL;
+	}
 
 	/* Walk the list of the arguments. */
 	choices = NULL;
@@ -2733,6 +2840,13 @@ format_link(struct plugin_state *state, Slapi_PBlock *pb, Slapi_Entry *e,
 		format_free_parsed_args(argv);
 		return -EINVAL;
 	}
+	if (outbuf_choices == NULL) {
+		slapi_log_error(SLAPI_LOG_PLUGIN, state->plugin_desc->spd_id,
+				"link: returns a list, but a list "
+				"would not be appropriate\n");
+		format_free_parsed_args(argv);
+		return -EINVAL;
+	}
 
 	/* Allocate space to store the information. */
 	values = malloc(sizeof(char **) * ((argc + 1) / 3));
@@ -2926,6 +3040,13 @@ format_unique(struct plugin_state *state, Slapi_PBlock *pb, Slapi_Entry *e,
 		value_format = argv[0];
 		default_value = argv[1];
 	}
+	if (outbuf_choices == NULL) {
+		slapi_log_error(SLAPI_LOG_PLUGIN, state->plugin_desc->spd_id,
+				"unique: returns a list, but a list "
+				"would not be appropriate\n");
+		format_free_parsed_args(argv);
+		return -EINVAL;
+	}
 	ret = -ENOENT;
 	values = format_get_data_set(state, pb, e, group, set,
 				     value_format, disallowed,
@@ -3075,6 +3196,11 @@ format_lookup_fn(const char *fnname)
 		{"regmatchi", format_regmatchi},
 		{"regsub", format_regsub},
 		{"regsubi", format_regsubi},
+		{"mmatch", format_mmatch},
+		{"mregmatch", format_mregmatch},
+		{"mregmatchi", format_mregmatchi},
+		{"mregsub", format_mregsub},
+		{"mregsubi", format_mregsubi},
 		{"ifeq", format_ifeq},
 		{"collect", format_collect},
 		{"link", format_link},