Merge "rtp_engine/res_rtp_asterisk: Fix RTP struct reentrancy crashes."
[asterisk/asterisk.git] / res / res_pjsip_xpidf_body_generator.c
index fba6152..9240465 100644 (file)
 static void *xpidf_allocate_body(void *data)
 {
        struct ast_sip_exten_state_data *state_data = data;
+       char *local = ast_strdupa(state_data->local);
        pjxpidf_pres *pres;
        pj_str_t name;
 
-       pres = pjxpidf_create(state_data->pool, pj_cstr(&name, state_data->local));
+       pres = pjxpidf_create(state_data->pool, pj_cstr(&name, ast_strip_quoted(local, "<", ">")));
        return pres;
 }
 
@@ -96,8 +97,7 @@ static int xpidf_generate_body_content(void *body, void *data)
        return 0;
 }
 
-#define MAX_STRING_GROWTHS 3
-#define XML_PROLOG 39
+#define MAX_STRING_GROWTHS 5
 
 static void xpidf_to_string(void *body, struct ast_str **str)
 {
@@ -106,14 +106,13 @@ static void xpidf_to_string(void *body, struct ast_str **str)
        int size;
 
        do {
-               size = pjxpidf_print(pres, ast_str_buffer(*str), ast_str_size(*str));
-               if (size == XML_PROLOG) {
+               size = pjxpidf_print(pres, ast_str_buffer(*str), ast_str_size(*str) - 1);
+               if (size <= AST_PJSIP_XML_PROLOG_LEN) {
                        ast_str_make_space(str, ast_str_size(*str) * 2);
                        ++growths;
                }
-       } while (size == XML_PROLOG && growths < MAX_STRING_GROWTHS);
-
-       if (size == XML_PROLOG) {
+       } while (size <= AST_PJSIP_XML_PROLOG_LEN && growths < MAX_STRING_GROWTHS);
+       if (size <= AST_PJSIP_XML_PROLOG_LEN) {
                ast_log(LOG_WARNING, "XPIDF body text too large\n");
                return;
        }
@@ -125,6 +124,7 @@ static void xpidf_to_string(void *body, struct ast_str **str)
 static struct ast_sip_pubsub_body_generator xpidf_body_generator = {
        .type = "application",
        .subtype = "xpidf+xml",
+       .body_type = AST_SIP_EXTEN_STATE_DATA,
        .allocate_body = xpidf_allocate_body,
        .generate_body_content = xpidf_generate_body_content,
        .to_string = xpidf_to_string,
@@ -134,6 +134,7 @@ static struct ast_sip_pubsub_body_generator xpidf_body_generator = {
 static struct ast_sip_pubsub_body_generator cpim_pidf_body_generator = {
        .type = "application",
        .subtype = "cpim-pidf+xml",
+       .body_type = AST_SIP_EXTEN_STATE_DATA,
        .allocate_body = xpidf_allocate_body,
        .generate_body_content = xpidf_generate_body_content,
        .to_string = xpidf_to_string,
@@ -148,6 +149,8 @@ static void unregister_all(void)
 
 static int load_module(void)
 {
+       CHECK_PJSIP_PUBSUB_MODULE_LOADED();
+
        if (ast_sip_pubsub_register_body_generator(&xpidf_body_generator)) {
                goto fail;
        }
@@ -170,7 +173,8 @@ static int unload_module(void)
 }
 
 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "PJSIP Extension State PIDF Provider",
-               .load = load_module,
-               .unload = unload_module,
-               .load_pri = AST_MODPRI_CHANNEL_DEPEND,
+       .support_level = AST_MODULE_SUPPORT_CORE,
+       .load = load_module,
+       .unload = unload_module,
+       .load_pri = AST_MODPRI_CHANNEL_DEPEND,
 );