Changeset 1954

Timestamp:

May 13, 2008 9:38:50 AM (16 years ago)

Author:

bennylp

Message:

Ticket #533: Support for parsing Retry-After SIP header

Location:

pjproject/trunk/pjsip

Files:

• include/pjsip/sip_msg.h (modified) (1 diff)
• src/pjsip/sip_msg.c (modified) (3 diffs)
• src/pjsip/sip_parser.c (modified) (1 diff)

pjproject/trunk/pjsip/include/pjsip/sip_msg.h

@@ -1650 +1650 @@
  * Retry-After header.
  */
-typedef pjsip_generic_int_hdr pjsip_retry_after_hdr;
+typedef struct pjsip_retry_after_hdr
+{
+    /** Standard header field. */
+    PJSIP_DECL_HDR_MEMBER(struct pjsip_retry_after_hdr); 
+    pj_int32_t  ivalue;         /**< Retry-After value      */
+    pjsip_param param;          /**< Optional parameters    */
+    pj_str_t    comment;        /**< Optional comments.     */
+} pjsip_retry_after_hdr;
+
 
 /**

pjproject/trunk/pjsip/src/pjsip/sip_msg.c

@@ -1645 +1645 @@
  * Retry-After header.
  */
+static int pjsip_retry_after_hdr_print(pjsip_retry_after_hdr *r, 
+                                       char *buf, pj_size_t size );
+static pjsip_retry_after_hdr* pjsip_retry_after_hdr_clone(pj_pool_t *pool, 
+                                                          const pjsip_retry_after_hdr *r);
+static pjsip_retry_after_hdr* 
+pjsip_retry_after_hdr_shallow_clone(pj_pool_t *pool, 
+                                    const pjsip_retry_after_hdr *r );
+
+static pjsip_hdr_vptr retry_after_hdr_vptr = 
+{
+    (pjsip_hdr_clone_fptr) &pjsip_retry_after_hdr_clone,
+    (pjsip_hdr_clone_fptr) &pjsip_retry_after_hdr_shallow_clone,
+    (pjsip_hdr_print_fptr) &pjsip_retry_after_hdr_print,
+};
+
+
 PJ_DEF(pjsip_retry_after_hdr*) pjsip_retry_after_hdr_init( pj_pool_t *pool,
                                                            void *mem,
@@ -1653 +1669 @@
     PJ_UNUSED_ARG(pool);
 
-    init_hdr(hdr, PJSIP_H_RETRY_AFTER, &generic_int_hdr_vptr);
+    init_hdr(hdr, PJSIP_H_RETRY_AFTER, &retry_after_hdr_vptr);
     hdr->ivalue = value;
+    pj_list_init(&hdr->param);
     return hdr;
 }
@@ -1665 +1682 @@
 }
 
+
+static int pjsip_retry_after_hdr_print(pjsip_retry_after_hdr *hdr, 
+                                       char *buf, pj_size_t size)
+{
+    char *p = buf;
+    char *endbuf = buf + size;
+    const pj_str_t *hname = &hdr->name;
+    const pjsip_parser_const_t *pc = pjsip_parser_const();
+    int printed;
+    
+    if ((pj_ssize_t)size < hdr->name.slen + 2+11)
+        return -1;
+
+    pj_memcpy(p, hdr->name.ptr, hdr->name.slen);
+    p += hname->slen;
+    *p++ = ':';
+    *p++ = ' ';
+
+    p += pj_utoa(hdr->ivalue, p);
+
+    if (hdr->comment.slen) {
+        pj_bool_t enclosed;
+
+        if (endbuf-p < hdr->comment.slen + 3)
+            return -1;
+
+        enclosed = (*hdr->comment.ptr == '(');
+        if (!enclosed)
+            *p++ = '(';
+        pj_memcpy(p, hdr->comment.ptr, hdr->comment.slen);
+        p += hdr->comment.slen;
+        if (!enclosed)
+            *p++ = ')';
+
+        if (!pj_list_empty(&hdr->param))
+            *p++ = ' ';
+    }
+
+    printed = pjsip_param_print_on(&hdr->param, p, endbuf-p,
+                                   &pc->pjsip_TOKEN_SPEC,
+                                   &pc->pjsip_TOKEN_SPEC, 
+                                   ';');
+    if (printed < 0)
+        return printed;
+
+    p += printed;
+
+    return p - buf;
+}
+
+static pjsip_retry_after_hdr* pjsip_retry_after_hdr_clone(pj_pool_t *pool, 
+                                                          const pjsip_retry_after_hdr *rhs)
+{
+    pjsip_retry_after_hdr *hdr = pjsip_retry_after_hdr_create(pool, rhs->ivalue);
+    pj_strdup(pool, &hdr->comment, &rhs->comment);
+    pjsip_param_clone(pool, &hdr->param, &rhs->param);
+    return hdr;
+}
+
+static pjsip_retry_after_hdr* 
+pjsip_retry_after_hdr_shallow_clone(pj_pool_t *pool, 
+                                    const pjsip_retry_after_hdr *rhs)
+{
+    pjsip_retry_after_hdr *hdr = PJ_POOL_ALLOC_T(pool, pjsip_retry_after_hdr);
+    pj_memcpy(hdr, rhs, sizeof(*hdr));
+    pjsip_param_shallow_clone(pool, &hdr->param, &rhs->param);
+    return hdr;
+}
 
 ///////////////////////////////////////////////////////////////////////////////

pjproject/trunk/pjsip/src/pjsip/sip_parser.c

@@ -1826 +1826 @@
 {
     pjsip_retry_after_hdr *hdr;
+    pj_scanner *scanner = ctx->scanner;
+    pj_str_t tmp;
+
     hdr = pjsip_retry_after_hdr_create(ctx->pool, 0);
-    parse_generic_int_hdr(hdr, ctx->scanner);
+    
+    pj_scan_get(scanner, &pconst.pjsip_DIGIT_SPEC, &tmp);
+    hdr->ivalue = pj_strtoul(&tmp);
+
+    while (!pj_scan_is_eof(scanner) && *scanner->curptr!='\r' &&
+           *scanner->curptr=='\n')
+    {
+        if (*scanner->curptr=='(') {
+            pj_scan_get_quote(scanner, '(', ')', &hdr->comment);
+        } else if (*scanner->curptr==';') {
+            pjsip_param *prm = PJ_POOL_ALLOC_T(ctx->pool, pjsip_param);
+            int_parse_param(scanner, ctx->pool, &prm->name, &prm->value, 0);
+            pj_list_push_back(&hdr->param, prm);
+        }
+    }
+
+    parse_hdr_end(scanner);
     return (pjsip_hdr*)hdr;
 }