pvar: add $challenge.* pseudovariable

Author: razvancrainea

pvar.c

@@ -64,6 +64,8 @@
 #include "parser/parse_pai.h"
 #include "parser/digest/digest.h"
 #include "parser/contact/parse_contact.h"
+#include "parser/parse_authenticate.h"
+#include "lib/digest_auth/digest_auth.h"
 
 #define is_in_str(p, in) (p<in->s+in->len && *p)
 
@@ -1816,6 +1818,66 @@ static int pv_get_authattr(struct sip_msg *msg, pv_param_t *param,
 }
 
 
+static int pv_get_cauthattr(struct sip_msg *msg, pv_param_t *param,
+		pv_value_t *res)
+{
+	struct authenticate_body *auth = NULL;
+	str *s;
+
+	if(msg==NULL)
+		return -1;
+
+	if (msg->first_line.type!=SIP_REPLY) {
+		LM_DBG("{Proxy,WWW}-Authenticate header should be only in replies\n");
+		return pv_get_null(msg, param, res);
+	}
+	switch (msg->REPLY_STATUS) {
+		case 401:
+			parse_www_authenticate_header(msg, NULL, &auth);
+			break;
+		case 407:
+			parse_proxy_authenticate_header(msg, NULL, &auth);
+			break;
+		default:
+			LM_DBG("bad reply code for a challenge!\n");
+			break;
+	}
+	if (!auth) {
+		LM_DBG("no {Proxy,WWW}-Authenticate header\n");
+		return pv_get_null(msg, param, res);
+	}
+
+	switch (param->pvn.u.isname.name.n) {
+		case 1:
+			s = (str *)print_digest_algorithm(auth->algorithm);
+			break;
+		case 2:
+			s = &auth->realm;
+			break;
+		case 3:
+			s = &auth->nonce;
+			break;
+		case 4:
+			s = &auth->opaque;
+			break;
+		case 5:
+			s = &auth->qop;
+			break;
+		case 6:
+			s = &auth->ik;
+			break;
+		case 7:
+			s = &auth->ck;
+			break;
+		default:
+			LM_BUG("unhandled mode %d\n", param->pvn.u.isname.name.n);
+			return pv_get_null(msg, param, res);
+	}
+
+	return pv_get_strval(msg, param, res, s);
+}
+
+
 static inline str *cred_user(struct sip_msg *rq)
 {
 	struct hdr_field* h;
@@ -3955,6 +4017,27 @@ const pv_export_t _pv_names_table[] = {
 	{str_const_init("aU"), /* */
 		PVT_AUTH_USERNAME_WHOLE, pv_get_authattr, 0,
 		0, 0, pv_init_iname, 99},
+	{str_const_init("challenge.algorithm"), /* */
+		PVT_AUTH_USERNAME, pv_get_cauthattr, 0,
+		0, 0, pv_init_iname, 1},
+	{str_const_init("challenge.realm"),
+		PVT_AUTH_REALM, pv_get_cauthattr, 0,
+		0, 0, pv_init_iname, 2},
+	{str_const_init("challenge.nonce"),
+		PVT_AUTH_DURI, pv_get_cauthattr, 0,
+		0, 0, pv_init_iname, 3},
+	{str_const_init("challenge.opaque"),
+		PVT_AUTH_DOMAIN, pv_get_cauthattr, 0,
+		0, 0, pv_init_iname, 4},
+	{str_const_init("challenge.qop"),
+		PVT_AUTH_NONCE, pv_get_cauthattr, 0,
+		0, 0, pv_init_iname, 5},
+	{str_const_init("challenge.ik"), /* */
+		PVT_AUTH_NONCE, pv_get_cauthattr, 0,
+		0, 0, pv_init_iname, 6},
+	{str_const_init("challenge.ck"), /* */
+		PVT_AUTH_RESPONSE, pv_get_cauthattr, 0,
+		0, 0, pv_init_iname, 7},
 	{str_const_init("Au"), /* */
 		PVT_ACC_USERNAME, pv_get_acc_username, 0,
 		0, 0, pv_init_iname, 1},