show upstream through brower

Author: xiaokai-wang

src/ngx_stream_upsync_module.c

@@ -3685,50 +3685,51 @@ ngx_stream_upsync_show_upstream(ngx_stream_upstream_srv_conf_t *uscf, ngx_buf_t
 static void
 ngx_stream_upsync_show(ngx_stream_session_t *s)
 {
-    ngx_buf_t                             *b;
-    ngx_uint_t                             i;
-    ngx_stream_upstream_srv_conf_t       **uscfp = NULL;
-    ngx_stream_upstream_main_conf_t       *umcf;
+    ngx_buf_t                           *b_header, *b_body;
+    ngx_uint_t                           i;
+    ngx_stream_upstream_srv_conf_t     **uscfp = NULL;
+    ngx_stream_upstream_main_conf_t     *umcf;
 
     umcf = ngx_stream_cycle_get_module_main_conf(ngx_cycle, 
                                                  ngx_stream_upstream_module);
 
     uscfp = umcf->upstreams.elts;
 
-    b = ngx_create_temp_buf(s->connection->pool, 
-                            NGX_PAGE_SIZE * NGX_PAGE_NUMBER);
-    if (b == NULL) {
+    b_header = ngx_create_temp_buf(s->connection->pool, NGX_PAGE_SIZE);
+    b_body = ngx_create_temp_buf(s->connection->pool, 
+                                 NGX_PAGE_SIZE * NGX_PAGE_NUMBER);
+    if (b_header == NULL || b_body == NULL) {
         ngx_log_error(NGX_LOG_ERR, s->connection->log, 0, 
                       "alloc upsync_show buf failed");
 
         ngx_stream_close_connection(s->connection);
         return;
     }
 
-    //HTTP Header
-    b->last = ngx_snprintf(b->last, b->end - b->last,
-                           "HTTP/1.0 200 OK\r\n");
-    b->last = ngx_snprintf(b->last, b->end - b->last,
-                           "Server: nginx\r\n");
-    b->last = ngx_snprintf(b->last, b->end - b->last,
-                           "Content-Type: text/plain\r\n");
-    b->last = ngx_snprintf(b->last, b->end - b->last,
-                           "Connection: close\r\n\r\n");
-
-    ngx_stream_upsync_show_send(s, b); //send header
-
-    b->pos = b->last;
     if (umcf->upstreams.nelts == 0) {
-        b->last = ngx_snprintf(b->last, b->end - b->last,
-                               "No upstreams defined\n");
+        b_body->last = ngx_snprintf(b_body->last, b_body->end - b_body->last,
+                                    "No upstreams defined\n");
     }
 
     for (i = 0; i < umcf->upstreams.nelts; i++) {
-        ngx_stream_upsync_show_upstream(uscfp[i], b);
-        b->last = ngx_snprintf(b->last, b->end - b->last, "\n");
+        ngx_stream_upsync_show_upstream(uscfp[i], b_body);
+        b_body->last = ngx_snprintf(b_body->last, b_body->end - b_body->last, "\n");
     }
 
-    ngx_stream_upsync_show_send(s, b); //send body
+    //HTTP Header
+    b_header->last = ngx_snprintf(b_header->last, b_header->end - b_header->last,
+                                  "HTTP/1.0 200 OK\r\n");
+    b_header->last = ngx_snprintf(b_header->last, b_header->end - b_header->last,
+                                  "Server: nginx\r\n");
+    b_header->last = ngx_snprintf(b_header->last, b_header->end - b_header->last,
+                                  "Content-Type: text/plain\r\n");
+    b_header->last = ngx_snprintf(b_header->last, b_header->end - b_header->last,
+                                  "Content-Length: %d\r\n", b_body->last - b_body->pos);
+    b_header->last = ngx_snprintf(b_header->last, b_header->end - b_header->last,
+                                  "Connection: close\r\n\r\n");
+
+    ngx_stream_upsync_show_send(s, b_header); //send header
+    ngx_stream_upsync_show_send(s, b_body); //send body
 
     ngx_stream_close_connection(s->connection);