data_lump_rpl: add support for replacing rpl lump

OpenSIPS · May 13, 2022 · 899c28f · 899c28f
1 parent 197d881
commit 899c28f
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/data_lump_rpl.c b/data_lump_rpl.c
@@ -109,6 +109,34 @@ void free_lump_rpl(struct lump_rpl* lump)
 }
 
 
+struct lump_rpl* get_lump_rpl(struct sip_msg *msg, int flags)
+{
+	struct lump_rpl *lump;
+
+	for (lump = msg->reply_lump; lump; lump = lump->next) {
+		if (lump->flags & flags)
+			return lump;
+	}
+	return NULL;
+}
+
+
+int replace_lump_rpl(struct lump_rpl *lump, char *s, int len, int flags)
+{
+	if ((flags & LUMP_RPL_NODUP) == 0) {
+		LM_ERR("replacing lump should have its own buffer\n");
+		return -1;
+	}
+	if (!((lump->flags)&LUMP_RPL_NOFREE) && ((lump->flags)&LUMP_RPL_NODUP)
+		&& lump->text.s)
+			pkg_free(lump->text.s);
+	lump->flags |= flags;
+	lump->text.s = s;
+	lump->text.len = len;
+	return 0;
+}
+
+
 void unlink_lump_rpl(struct sip_msg * msg, struct lump_rpl* lump)
 {
 	struct lump_rpl *foo,*prev;

diff --git a/data_lump_rpl.h b/data_lump_rpl.h
@@ -54,6 +54,10 @@ struct lump_rpl* add_lump_rpl(struct sip_msg *, char *, int , int );
 
 void free_lump_rpl(struct lump_rpl* );
 
+struct lump_rpl* get_lump_rpl(struct sip_msg *, int);
+
+int replace_lump_rpl(struct lump_rpl *, char *, int , int);
+
 void unlink_lump_rpl(struct sip_msg *, struct lump_rpl* );
 
 void del_nonshm_lump_rpl(  struct lump_rpl ** );