kemix: added kemi functions to return user-agent header body
authorDaniel-Constantin Mierla <miconda@gmail.com>
Wed, 5 Jun 2019 09:40:12 +0000 (11:40 +0200)
committerDaniel-Constantin Mierla <miconda@gmail.com>
Wed, 5 Jun 2019 09:51:24 +0000 (11:51 +0200)
- shortcut to improve tests for security in scripts

src/modules/kemix/kemix_mod.c

index ff67544..4f69098 100644 (file)
@@ -275,6 +275,51 @@ static sr_kemi_xval_t* ki_kx_get_rhoste(sip_msg_t *msg)
 /**
  *
  */
+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);
+}
+
+/**
+ *
+ */
 /* clang-format off */
 static sr_kemi_t sr_kemi_kx_exports[] = {
        { str_init("kx"), str_init("get_ruri"),
@@ -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 } }
 };