From 3bce7841290b2ff2840c79d63e210973acfa71f9 Mon Sep 17 00:00:00 2001
From: Daniel-Constantin Mierla <miconda@gmail.com>
Date: Wed, 29 Jun 2016 18:04:19 +0200
Subject: [PATCH] pv: new class of pseudo-variable - $msg(attr)

- return attributes of sip message
	* $msg(len) - sip message lenght
	* $msg(buf) - sip message buffer
	* $msg(body) - sip message body
	* $msg(body_len) - sip message body lenght
	* $msg(hdrs) - sip message headers
	* $msg(fline) - sip message first line
---
 modules/pv/pv.c      |   2 +
 modules/pv/pv_core.c | 102 +++++++++++++++++++++++++++++++++++++++++++
 modules/pv/pv_core.h |   4 ++
 3 files changed, 108 insertions(+)

diff --git a/modules/pv/pv.c b/modules/pv/pv.c
index 1aa53edb391..3e335966c61 100644
--- a/modules/pv/pv.c
+++ b/modules/pv/pv.c
@@ -470,6 +470,8 @@ static pv_export_t mod_pvs[] = {
 		pv_parse_K_name, 0, 0, 0 },
 	{ {"expires", (sizeof("expires")-1)}, PVT_OTHER, pv_get_expires, 0,
 		pv_parse_expires_name, 0, 0, 0 },
+	{ {"msg", (sizeof("msg")-1)}, PVT_OTHER, pv_get_msg_attrs, 0,
+		pv_parse_msg_attrs_name, 0, 0, 0 },
 
 	{ {0, 0}, 0, 0, 0, 0, 0, 0, 0 }
 };
diff --git a/modules/pv/pv_core.c b/modules/pv/pv_core.c
index 5836e232851..6c1ec41f3a9 100644
--- a/modules/pv/pv_core.c
+++ b/modules/pv/pv_core.c
@@ -32,6 +32,7 @@
 #include "../../lib/kcore/cmpapi.h"
 #include "../../tcp_conn.h"
 #include "../../pvapi.h"
+#include "../../trim.h"
 
 #include "../../parser/parse_from.h"
 #include "../../parser/parse_uri.h"
@@ -3112,3 +3113,104 @@ int pv_get_expires(sip_msg_t *msg, pv_param_t *param, pv_value_t *res)
 			return pv_get_null(msg, param, res);
 	}
 }
+
+/**
+ *
+ */
+int pv_parse_msg_attrs_name(pv_spec_p sp, str *in)
+{
+	if(sp==NULL || in==NULL || in->len<=0)
+		return -1;
+
+	switch(in->len)
+	{
+		case 3:
+			if(strncmp(in->s, "len", 3)==0)
+				sp->pvp.pvn.u.isname.name.n = 0;
+			else if(strncmp(in->s, "buf", 3)==0)
+				sp->pvp.pvn.u.isname.name.n = 1;
+			else goto error;
+		break;
+		case 4:
+			if(strncmp(in->s, "body", 4)==0)
+				sp->pvp.pvn.u.isname.name.n = 2;
+			else if(strncmp(in->s, "hdrs", 4)==0)
+				sp->pvp.pvn.u.isname.name.n = 3;
+			else goto error;
+		case 5:
+			if(strncmp(in->s, "fline", 5)==0)
+				sp->pvp.pvn.u.isname.name.n = 4;
+			else goto error;
+		case 8:
+			if(strncmp(in->s, "body_len", 8)==0)
+				sp->pvp.pvn.u.isname.name.n = 5;
+			else goto error;
+		default:
+			goto error;
+	}
+	sp->pvp.pvn.type = PV_NAME_INTSTR;
+	sp->pvp.pvn.u.isname.type = 0;
+
+	return 0;
+
+error:
+	LM_ERR("unknown PV expires key: %.*s\n", in->len, in->s);
+	return -1;
+}
+
+/**
+ *
+ */
+int pv_get_msg_attrs(sip_msg_t *msg, pv_param_t *param, pv_value_t *res)
+{
+	str s;
+	if(msg==NULL)
+		return pv_get_null(msg, param, res);
+
+	if(param==NULL)
+		return pv_get_null(msg, param, res);
+
+	if (parse_headers(msg, HDR_EOH_F, 0) == -1) {
+		LM_ERR("failed to parse headers\n");
+		return pv_get_null(msg, param, res);
+	}
+
+	switch(param->pvn.u.isname.name.n)
+	{
+		case 0: /* length */
+			return pv_get_uintval(msg, param, res, msg->len);
+		case 1: /* buffer */
+			s.s = msg->buf;
+			s.len = msg->len;
+			return pv_get_strval(msg, param, res, &s);
+		case 2: /* body */
+			s.s = get_body(msg);
+			if(s.s == NULL) {
+				LM_DBG("no message body\n");
+				return pv_get_null(msg, param, res);
+			}
+			s.len = msg->buf + msg->len - s.s;
+			return pv_get_strval(msg, param, res, &s);
+		case 3: /* headers */
+			if(msg->unparsed==NULL)
+				return pv_get_null(msg, param, res);
+			s.s = msg->buf + msg->first_line.len;
+			s.len = msg->unparsed - s.s;
+			trim(&s);
+			return pv_get_strval(msg, param, res, &s);
+		case 4: /* first line */
+			s.s = msg->buf;
+			s.len = msg->first_line.len;
+			trim(&s);
+			return pv_get_strval(msg, param, res, &s);
+		case 5: /* body size */
+			s.s = get_body( msg );
+			s.len = 0;
+			if (s.s != NULL)
+				s.len = msg->buf + msg->len - s.s;
+			return pv_get_sintval(msg, param, res, s.len);
+
+		default:
+			return pv_get_null(msg, param, res);
+	}
+}
diff --git a/modules/pv/pv_core.h b/modules/pv/pv_core.h
index 6d6c1a5b147..c994e90b039 100644
--- a/modules/pv/pv_core.h
+++ b/modules/pv/pv_core.h
@@ -330,5 +330,9 @@ int pv_parse_expires_name(pv_spec_p sp, str *in);
 int pv_get_expires(sip_msg_t *msg, pv_param_t *param,
 		pv_value_t *res);
 
+int pv_parse_msg_attrs_name(pv_spec_p sp, str *in);
+
+int pv_get_msg_attrs(sip_msg_t *msg, pv_param_t *param,
+		pv_value_t *res);
 #endif