Browse Source

kemix: added kemi functions to return user-agent header body

- shortcut to improve tests for security in scripts
Daniel-Constantin Mierla 6 năm trước cách đây
mục cha
commit
067fbd44e1
1 tập tin đã thay đổi với 60 bổ sung0 xóa
  1. 60 0
      src/modules/kemix/kemix_mod.c

+ 60 - 0
src/modules/kemix/kemix_mod.c

@@ -272,6 +272,51 @@ static sr_kemi_xval_t* ki_kx_get_rhoste(sip_msg_t *msg)
 	return ki_kx_get_ruri_attr(msg, 2, SR_KEMI_XVAL_NULL_EMPTY);
 }
 
+/**
+ *
+ */
+static sr_kemi_xval_t* ki_kx_get_ua_mode(sip_msg_t *msg, int rmode)
+{
+	memset(&_sr_kemi_kx_xval, 0, sizeof(sr_kemi_xval_t));
+	if(msg==NULL) {
+		sr_kemi_xval_null(&_sr_kemi_kx_xval, rmode);
+		return &_sr_kemi_kx_xval;
+	}
+	if(msg->user_agent==NULL && ((parse_headers(msg, HDR_USERAGENT_F, 0)==-1)
+			|| (msg->user_agent==NULL))) {
+		sr_kemi_xval_null(&_sr_kemi_kx_xval, rmode);
+		return &_sr_kemi_kx_xval;
+	}
+
+	_sr_kemi_kx_xval.vtype = SR_KEMIP_STR;
+	_sr_kemi_kx_xval.v.s = msg->user_agent->body;
+	return &_sr_kemi_kx_xval;
+}
+
+/**
+ *
+ */
+static sr_kemi_xval_t* ki_kx_get_ua(sip_msg_t *msg)
+{
+	return ki_kx_get_ua_mode(msg, SR_KEMI_XVAL_NULL_NONE);
+}
+
+/**
+ *
+ */
+static sr_kemi_xval_t* ki_kx_gete_ua(sip_msg_t *msg)
+{
+	return ki_kx_get_ua_mode(msg, SR_KEMI_XVAL_NULL_EMPTY);
+}
+
+/**
+ *
+ */
+static sr_kemi_xval_t* ki_kx_getw_ua(sip_msg_t *msg)
+{
+	return ki_kx_get_ua_mode(msg, SR_KEMI_XVAL_NULL_PRINT);
+}
+
 /**
  *
  */
@@ -327,6 +372,21 @@ static sr_kemi_t sr_kemi_kx_exports[] = {
 		{ SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE,
 			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
 	},
+	{ str_init("kx"), str_init("get_ua"),
+		SR_KEMIP_XVAL, ki_kx_get_ua,
+		{ SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE,
+			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+	},
+	{ str_init("kx"), str_init("gete_ua"),
+		SR_KEMIP_XVAL, ki_kx_gete_ua,
+		{ SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE,
+			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+	},
+	{ str_init("kx"), str_init("getw_ua"),
+		SR_KEMIP_XVAL, ki_kx_getw_ua,
+		{ SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE,
+			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
+	},
 
 	{ {0, 0}, {0, 0}, 0, NULL, { 0, 0, 0, 0, 0, 0 } }
 };