Răsfoiți Sursa

textops: remove hf with regex functions exported to kemi framework

Daniel-Constantin Mierla 8 ani în urmă
părinte
comite
09ce805064
1 a modificat fișierele cu 56 adăugiri și 2 ștergeri
  1. 56 2
      src/modules/textops/textops.c

+ 56 - 2
src/modules/textops/textops.c

@@ -1098,7 +1098,25 @@ static int remove_hf_re_f(struct sip_msg* msg, char* key, char* foo)
 	return remove_hf_re(msg, (regex_t*)key);
 	return remove_hf_re(msg, (regex_t*)key);
 }
 }
 
 
-static int remove_hf_exp_re(sip_msg_t* msg, regex_t *mre, regex_t *sre)
+static int ki_remove_hf_re(sip_msg_t *msg, str *ematch)
+{
+	regex_t mre;
+	int ret;
+
+	memset(&mre, 0, sizeof(regex_t));
+	if (regcomp(&mre, ematch->s, REG_EXTENDED|REG_ICASE|REG_NEWLINE)!=0) {
+		LM_ERR("failed to compile regex: %.*s\n", ematch->len, ematch->s);
+		return -1;
+	}
+
+	ret = remove_hf_re(msg, &mre);
+
+	regfree(&mre);
+
+	return ret;
+}
+
+static int remove_hf_exp(sip_msg_t* msg, regex_t *mre, regex_t *sre)
 {
 {
 	struct hdr_field *hf;
 	struct hdr_field *hf;
 	struct lump* l;
 	struct lump* l;
@@ -1143,7 +1161,33 @@ static int remove_hf_exp_re(sip_msg_t* msg, regex_t *mre, regex_t *sre)
 
 
 static int remove_hf_exp_f(struct sip_msg* msg, char* ematch, char* eskip)
 static int remove_hf_exp_f(struct sip_msg* msg, char* ematch, char* eskip)
 {
 {
-	return remove_hf_exp_re(msg, (regex_t*)ematch, (regex_t*)eskip);
+	return remove_hf_exp(msg, (regex_t*)ematch, (regex_t*)eskip);
+}
+
+static int ki_remove_hf_exp(sip_msg_t *msg, str *ematch, str *eskip)
+{
+	regex_t mre;
+	regex_t sre;
+	int ret;
+
+	memset(&mre, 0, sizeof(regex_t));
+	memset(&sre, 0, sizeof(regex_t));
+	if (regcomp(&mre, ematch->s, REG_EXTENDED|REG_ICASE|REG_NEWLINE)!=0) {
+		LM_ERR("failed to compile regex: %.*s\n", ematch->len, ematch->s);
+		return -1;
+	}
+	if (regcomp(&sre, eskip->s, REG_EXTENDED|REG_ICASE|REG_NEWLINE)!=0) {
+		LM_ERR("failed to compile regex: %.*s\n", eskip->len, eskip->s);
+		regfree(&mre);
+		return -1;
+	}
+
+	ret = remove_hf_exp(msg, &mre, &sre);
+
+	regfree(&mre);
+	regfree(&sre);
+
+	return ret;
 }
 }
 
 
 static int is_present_hf_helper_f(struct sip_msg* msg, gparam_t* gp)
 static int is_present_hf_helper_f(struct sip_msg* msg, gparam_t* gp)
@@ -3490,6 +3534,16 @@ static sr_kemi_t sr_kemi_textops_exports[] = {
 		{ SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_STR,
 		{ SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_STR,
 			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
 			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
 	},
 	},
+	{ str_init("textops"), str_init("remove_hf_re"),
+		SR_KEMIP_INT, ki_remove_hf_re,
+		{ SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE,
+			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+	},
+	{ str_init("textops"), str_init("remove_hf_exp"),
+		SR_KEMIP_INT, ki_remove_hf_exp,
+		{ SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_NONE,
+			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+	},
 
 
 	{ {0, 0}, {0, 0}, 0, NULL, { 0, 0, 0, 0, 0, 0 } }
 	{ {0, 0}, {0, 0}, 0, NULL, { 0, 0, 0, 0, 0, 0 } }
 };
 };