From 4e29c23e8b310d111503dee44b6de3aa8f57d2ce Mon Sep 17 00:00:00 2001
From: Daniel-Constantin Mierla <miconda@gmail.com>
Date: Fri, 28 May 2021 14:55:42 +0200
Subject: [PATCH] mtree: number of matched records for tree type=1 stored in
 pv_count

---
 src/modules/mtree/mtree.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/modules/mtree/mtree.c b/src/modules/mtree/mtree.c
index fe4dba1a205..f9b755d4d87 100644
--- a/src/modules/mtree/mtree.c
+++ b/src/modules/mtree/mtree.c
@@ -44,6 +44,7 @@ extern pv_spec_t pv_value;
 extern pv_spec_t pv_values;
 extern pv_spec_t pv_dstid;
 extern pv_spec_t pv_weight;
+extern pv_spec_t pv_count;
 extern int _mt_tree_type;
 extern int _mt_ignore_duplicates;
 extern int _mt_allow_duplicates;
@@ -599,6 +600,13 @@ int mt_match_prefix(struct sip_msg *msg, m_tree_t *it,
 			add_avp(dstid_name_type, dstid_avp_name, avp_value);
 		}
 	}
+	memset(&val, 0, sizeof(pv_value_t));
+	val.flags = PV_VAL_INT;
+	val.ri = n;
+	if(pv_value.setf(msg, &pv_count.pvp, (int)EQ_T, &val)<0) {
+		LM_ERR("setting PV count failed\n");
+		return -2;
+	}
 
 	return 0;
 }