res_resolver_unbound: Fix config documentation.
[asterisk/asterisk.git] / res / res_pjsip_dialog_info_body_generator.c
index d4ad216..5006b9e 100644 (file)
@@ -61,20 +61,20 @@ static void *dialog_info_allocate_body(void *data)
        return ast_sip_presence_xml_create_node(state_data->pool, NULL, "dialog-info");
 }
 
-static struct ast_datastore *dialog_info_xml_state_find_or_create(struct ast_sip_subscription *sub)
+static struct ast_datastore *dialog_info_xml_state_find_or_create(struct ao2_container *datastores)
 {
-       struct ast_datastore *datastore = ast_sip_subscription_get_datastore(sub, "dialog-info+xml");
+       struct ast_datastore *datastore = ast_datastores_find(datastores, "dialog-info+xml");
 
        if (datastore) {
                return datastore;
        }
 
-       datastore = ast_sip_subscription_alloc_datastore(&dialog_info_xml_datastore, "dialog-info+xml");
+       datastore = ast_datastores_alloc_datastore(&dialog_info_xml_datastore, "dialog-info+xml");
        if (!datastore) {
                return NULL;
        }
        datastore->data = ast_calloc(1, sizeof(struct dialog_info_xml_state));
-       if (!datastore->data || ast_sip_subscription_add_datastore(sub, datastore)) {
+       if (!datastore->data || ast_datastores_add(datastores, datastore)) {
                ao2_ref(datastore, -1);
                return NULL;
        }
@@ -82,9 +82,9 @@ static struct ast_datastore *dialog_info_xml_state_find_or_create(struct ast_sip
        return datastore;
 }
 
-static unsigned int dialog_info_xml_get_version(struct ast_sip_subscription *sub, unsigned int *version)
+static unsigned int dialog_info_xml_get_version(struct ao2_container *datastores, unsigned int *version)
 {
-       struct ast_datastore *datastore = dialog_info_xml_state_find_or_create(sub);
+       struct ast_datastore *datastore = dialog_info_xml_state_find_or_create(datastores);
        struct dialog_info_xml_state *state;
 
        if (!datastore) {
@@ -108,11 +108,11 @@ static int dialog_info_generate_body_content(void *body, void *data)
        unsigned int version;
        char version_str[32], sanitized[PJSIP_MAX_URL_SIZE];
 
-       if (!local || !state_data->sub) {
+       if (!local || !state_data->datastores) {
                return -1;
        }
 
-       if (dialog_info_xml_get_version(state_data->sub, &version)) {
+       if (dialog_info_xml_get_version(state_data->datastores, &version)) {
                ast_log(LOG_WARNING, "dialog-info+xml version could not be retrieved from datastore\n");
                return -1;
        }
@@ -133,6 +133,9 @@ static int dialog_info_generate_body_content(void *body, void *data)
 
        dialog = ast_sip_presence_xml_create_node(state_data->pool, dialog_info, "dialog");
        ast_sip_presence_xml_create_attr(state_data->pool, dialog, "id", state_data->exten);
+       if (state_data->exten_state == AST_EXTENSION_RINGING) {
+               ast_sip_presence_xml_create_attr(state_data->pool, dialog, "direction", "recipient");
+       }
 
        state = ast_sip_presence_xml_create_node(state_data->pool, dialog, "state");
        pj_strdup2(state_data->pool, &state->content, statestring);
@@ -154,12 +157,7 @@ static int dialog_info_generate_body_content(void *body, void *data)
 /* The maximum number of times the ast_str() for the body text can grow before we declare an XML body
  * too large to send.
  */
-#define MAX_STRING_GROWTHS 3
-
-/* When having pj_xml_print add the XML prolog to the output body the function will return 39
- * instead of -1 if the rest of the document can not be printed into the body.
- */
-#define XML_PROLOG 39
+#define MAX_STRING_GROWTHS 6
 
 static void dialog_info_to_string(void *body, struct ast_str **str)
 {
@@ -168,14 +166,13 @@ static void dialog_info_to_string(void *body, struct ast_str **str)
        int size;
 
        do {
-               size = pj_xml_print(dialog_info, ast_str_buffer(*str), ast_str_size(*str), PJ_TRUE);
-               if (size == XML_PROLOG) {
+               size = pj_xml_print(dialog_info, ast_str_buffer(*str), ast_str_size(*str) - 1, PJ_TRUE);
+               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, "dialog-info+xml body text too large\n");
                return;
        }
@@ -187,6 +184,7 @@ static void dialog_info_to_string(void *body, struct ast_str **str)
 static struct ast_sip_pubsub_body_generator dialog_info_body_generator = {
        .type = "application",
        .subtype = "dialog-info+xml",
+       .body_type = AST_SIP_EXTEN_STATE_DATA,
        .allocate_body = dialog_info_allocate_body,
        .generate_body_content = dialog_info_generate_body_content,
        .to_string = dialog_info_to_string,
@@ -195,6 +193,8 @@ static struct ast_sip_pubsub_body_generator dialog_info_body_generator = {
 
 static int load_module(void)
 {
+       CHECK_PJSIP_PUBSUB_MODULE_LOADED();
+
        if (ast_sip_pubsub_register_body_generator(&dialog_info_body_generator)) {
                return AST_MODULE_LOAD_DECLINE;
        }
@@ -209,8 +209,8 @@ static int unload_module(void)
 }
 
 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "PJSIP Extension State Dialog Info+XML Provider",
-               .support_level = AST_MODULE_SUPPORT_CORE,
-               .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,
 );