static ngx_int_t ngx_http_push_stream_send_response_all_channels_info_detailed(ngx_http_request_t *r, ngx_str_t *prefix) { ngx_int_t rc, content_len = 0; ngx_chain_t *chain, *first = NULL, *last = NULL; ngx_str_t *currenttime, *hostname, *text, *header_response; ngx_queue_t queue_channel_info; ngx_queue_t *cur, *next; ngx_http_push_stream_shm_data_t *data = (ngx_http_push_stream_shm_data_t *) ngx_http_push_stream_shm_zone->data; ngx_slab_pool_t *shpool = (ngx_slab_pool_t *) ngx_http_push_stream_shm_zone->shm.addr; ngx_http_push_stream_content_subtype_t *subtype = ngx_http_push_stream_match_channel_info_format_and_content_type(r, 1); ngx_pool_t *temp_pool = ngx_http_push_stream_get_temp_pool(r); const ngx_str_t *format; const ngx_str_t *head = subtype->format_group_head; const ngx_str_t *tail = subtype->format_group_tail; ngx_queue_init(&queue_channel_info); ngx_shmtx_lock(&shpool->mutex); ngx_http_push_stream_rbtree_walker_channel_info_locked(&data->tree, temp_pool, data->tree.root, &queue_channel_info, prefix); ngx_shmtx_unlock(&shpool->mutex); // format content body cur = ngx_queue_head(&queue_channel_info); while (cur != &queue_channel_info) { next = ngx_queue_next(cur); ngx_http_push_stream_channel_info_t *channel_info = (ngx_http_push_stream_channel_info_t *) cur; if ((chain = ngx_http_push_stream_get_buf(r)) == NULL) { ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "push stream module: unable to allocate memory for response channels info"); return NGX_HTTP_INTERNAL_SERVER_ERROR; } format = (next != &queue_channel_info) ? subtype->format_group_item : subtype->format_group_last_item; if ((text = ngx_http_push_stream_channel_info_formatted(temp_pool, format, &channel_info->id, channel_info->published_messages, channel_info->stored_messages, channel_info->subscribers)) == NULL) { ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "push stream module: unable to allocate memory to format channel info"); return NGX_HTTP_INTERNAL_SERVER_ERROR; } chain->buf->last_buf = 0; chain->buf->memory = 1; chain->buf->pos = text->data; chain->buf->last = text->data + text->len; chain->buf->start = chain->buf->pos; chain->buf->end = chain->buf->last; content_len += text->len; if (first == NULL) { first = chain; } if (last != NULL) { last->next = chain; } last = chain; cur = next; } // get formatted current time currenttime = ngx_http_push_stream_get_formatted_current_time(temp_pool); // get formatted hostname hostname = ngx_http_push_stream_get_formatted_hostname(temp_pool); // format content header if ((header_response = ngx_http_push_stream_create_str(temp_pool, head->len + hostname->len + currenttime->len + NGX_INT_T_LEN)) == NULL) { ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "push stream module: unable to allocate memory for response channels info"); return NGX_HTTP_INTERNAL_SERVER_ERROR; } ngx_sprintf(header_response->data, (char *) head->data, hostname->data, currenttime->data, data->channels, data->broadcast_channels, ngx_time() - data->startup); header_response->len = ngx_strlen(header_response->data); content_len += header_response->len + tail->len; r->headers_out.content_type.len = subtype->content_type->len; r->headers_out.content_type.data = subtype->content_type->data; r->headers_out.content_length_n = content_len; r->headers_out.status = NGX_HTTP_OK; rc = ngx_http_send_header(r); if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) { return rc; } // send content header ngx_http_push_stream_send_response_text(r, header_response->data, header_response->len,0); // send content body if (first != NULL) { ngx_http_push_stream_output_filter(r, first); } // send content footer return ngx_http_push_stream_send_response_text(r, tail->data, tail->len, 1); }
static ngx_int_t ngx_http_push_stream_send_response_channels_info(ngx_http_request_t *r, ngx_queue_t *queue_channel_info) { ngx_int_t rc, content_len = 0; ngx_chain_t *chain, *first = NULL, *last = NULL; ngx_str_t *currenttime, *hostname, *text, *header_response; ngx_queue_t *q; ngx_http_push_stream_main_conf_t *mcf = ngx_http_get_module_main_conf(r, ngx_http_push_stream_module); ngx_http_push_stream_shm_data_t *data = mcf->shm_data; ngx_http_push_stream_content_subtype_t *subtype = ngx_http_push_stream_match_channel_info_format_and_content_type(r, 1); const ngx_str_t *format; const ngx_str_t *head = subtype->format_group_head; const ngx_str_t *tail = subtype->format_group_tail; // format content body for (q = ngx_queue_head(queue_channel_info); q != ngx_queue_sentinel(queue_channel_info); q = ngx_queue_next(q)) { ngx_http_push_stream_channel_info_t *channel_info = ngx_queue_data(q, ngx_http_push_stream_channel_info_t, queue); if ((chain = ngx_http_push_stream_get_buf(r)) == NULL) { ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "push stream module: unable to allocate memory for response channels info"); return NGX_HTTP_INTERNAL_SERVER_ERROR; } format = (q != ngx_queue_last(queue_channel_info)) ? subtype->format_group_item : subtype->format_group_last_item; if ((text = ngx_http_push_stream_channel_info_formatted(r->pool, format, &channel_info->id, channel_info->published_messages, channel_info->stored_messages, channel_info->subscribers)) == NULL) { ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "push stream module: unable to allocate memory to format channel info"); return NGX_HTTP_INTERNAL_SERVER_ERROR; } chain->buf->last_buf = 0; chain->buf->memory = 1; chain->buf->temporary = 0; chain->buf->pos = text->data; chain->buf->last = text->data + text->len; chain->buf->start = chain->buf->pos; chain->buf->end = chain->buf->last; content_len += text->len; if (first == NULL) { first = chain; } if (last != NULL) { last->next = chain; } last = chain; } // get formatted current time currenttime = ngx_http_push_stream_get_formatted_current_time(r->pool); // get formatted hostname hostname = ngx_http_push_stream_get_formatted_hostname(r->pool); // format content header if ((header_response = ngx_http_push_stream_create_str(r->pool, head->len + hostname->len + currenttime->len + NGX_INT_T_LEN)) == NULL) { ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "push stream module: unable to allocate memory for response channels info"); return NGX_HTTP_INTERNAL_SERVER_ERROR; } ngx_sprintf(header_response->data, (char *) head->data, hostname->data, currenttime->data, data->channels, data->wildcard_channels, ngx_time() - data->startup); header_response->len = ngx_strlen(header_response->data); content_len += header_response->len + tail->len; r->headers_out.content_type_len = subtype->content_type->len; r->headers_out.content_type = *subtype->content_type; r->headers_out.content_length_n = content_len; r->headers_out.status = NGX_HTTP_OK; rc = ngx_http_send_header(r); if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) { return rc; } // send content header ngx_http_push_stream_send_response_text(r, header_response->data, header_response->len,0); // send content body if (first != NULL) { ngx_http_push_stream_output_filter(r, first); } // send content footer return ngx_http_push_stream_send_response_text(r, tail->data, tail->len, 1); }