├── README.md ├── config └── ngx_http_upconf_module.c /README.md: -------------------------------------------------------------------------------- 1 | Name 2 | ==== 3 | 4 | nginx-upconf-module - Nginx C module, provides an HTTP interface for adding, removing, and modifying back-end servers dynamically and without having to reload the configuration, seems like Nginx-Plus ngx_http_upstream_conf_module. 5 | 6 | It may not always be convenient to modify configuration files and restart NGINX. For example, if you are experiencing large amounts of traffic and high load, restarting NGINX and reloading the configuration at that point further increases load on the system and can temporarily degrade performance. 7 | 8 | The module can be more smoothly expansion and constriction, and will not influence the performance. 9 | 10 | Table of Contents 11 | ================= 12 | 13 | * [Name](#name) 14 | * [Status](#status) 15 | * [Synopsis](#synopsis) 16 | * [Description](#description) 17 | * [Directives](#functions) 18 | * [upstream_conf](#upstream_conf) 19 | * [upconf_dump_path](#upconf_dump_path) 20 | * [Compatibility](#compatibility) 21 | * [Installation](#installation) 22 | * [Author](#author) 23 | * [Copyright and License](#copyright-and-license) 24 | * [Dependency](#dependency) 25 | 26 | Status 27 | ====== 28 | 29 | This module is still under active development. 30 | 31 | Synopsis 32 | ======== 33 | 34 | ```nginx 35 | http { 36 | upstream test { 37 | server 127.0.0.1:8089 weight=1, fail_timeout=10, max_fails=3; 38 | } 39 | 40 | upstream bar { 41 | server 127.0.0.1:8090 weight=1, fail_timeout=10, max_fails=3; 42 | } 43 | 44 | server { 45 | listen 8080; 46 | 47 | location = /proxy_test { 48 | proxy_pass http://test; 49 | } 50 | 51 | location = /bar { 52 | proxy_pass http://bar; 53 | } 54 | 55 | location = /upstream_conf { 56 | upstream_conf; 57 | upconf_dump_path /usr/local/nginx/conf/servers; 58 | } 59 | 60 | } 61 | } 62 | ``` 63 | 64 | Description 65 | ====== 66 | 67 | On-The-Fly Configuration Using an HTTP-based API: 68 | 69 | * timely 70 | 71 | effective in a minute. 72 | 73 | * performance 74 | 75 | http api, like a request to nginx, updating ip router nginx needn't reload, so affecting nginx performance is little. 76 | 77 | * availability 78 | 79 | support dumping servers to local, so you can reload nginx anytime, will keep service high available. 80 | 81 | * health_check 82 | 83 | nginx-upconf-module support adding or deleting servers health check, needing nginx_upstream_check_module. Recommending nginx-upsync-module + nginx_upstream_check_module. 84 | 85 | Diretives 86 | ====== 87 | 88 | upstream_conf 89 | ----------- 90 | ``` 91 | syntax: upstream_conf 92 | ``` 93 | default: none 94 | 95 | context: location 96 | 97 | description: just like the [Description](http://nginx.org/en/docs/http/ngx_http_upstream_conf_module.html#upstream_conf) 98 | 99 | [Back to TOC](#table-of-contents) 100 | 101 | upconf_dump_path 102 | ----------- 103 | `syntax: upconf_dump_path $path` 104 | 105 | default: /tmp 106 | 107 | context: location 108 | 109 | description: dump the upstream backends to the $path. 110 | 111 | [Back to TOC](#table-of-contents) 112 | 113 | Compatibility 114 | ============= 115 | 116 | The module was developed base on nginx1.8.0. 117 | 118 | Compatible with Nginx-1.8.x. 119 | 120 | Compatible with Nginx-1.9.x. 121 | 122 | [Back to TOC](#table-of-contents) 123 | 124 | Installation 125 | ============ 126 | 127 | This module can be used independently, can be download[Github](https://github.com/xiaokai-wang/nginx-upconf-module.git). 128 | 129 | Grab the nginx source code from [nginx.org](http://nginx.org/), for example, the version 1.8.0 (see nginx compatibility), and then build the source with this module: 130 | 131 | ```bash 132 | wget 'http://nginx.org/download/nginx-1.8.0.tar.gz' 133 | tar -xzvf nginx-1.8.0.tar.gz 134 | cd nginx-1.8.0/ 135 | ``` 136 | 137 | ```bash 138 | ./configure --add-module=/path/to/ngx_sync_msg_module --add-module=/path/to/nginx-upconf-module 139 | make 140 | make install 141 | ``` 142 | 143 | [Back to TOC](#table-of-contents) 144 | 145 | Author 146 | ====== 147 | 148 | Xiaokai Wang (王晓开) , Weibo Inc. 149 | 150 | [Back to TOC](#table-of-contents) 151 | 152 | Copyright and License 153 | ===================== 154 | 155 | This README template copy from agentzh. 156 | 157 | This module is licensed under the BSD license. 158 | 159 | Copyright (C) 2014 by Xiaokai Wang 160 | 161 | All rights reserved. 162 | 163 | Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: 164 | 165 | * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. 166 | 167 | * Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. 168 | 169 | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 170 | 171 | [Back to TOC](#table-of-contents) 172 | 173 | dependency 174 | ======== 175 | * ngx_sync_msg_module: https://github.com/yzprofile/ngx_sync_msg_module 176 | 177 | [back to toc](#table-of-contents) 178 | 179 | 180 | -------------------------------------------------------------------------------- /config: -------------------------------------------------------------------------------- 1 | have=NGX_HTTP_UPCONF . auto/have 2 | ngx_addon_name=ngx_http_upconf_module 3 | HTTP_MODULES="$HTTP_MODULES ngx_http_upconf_module" 4 | NGX_ADDON_SRCS="$NGX_ADDON_SRCS $ngx_addon_dir/src/ngx_http_upconf_module.c" 5 | -------------------------------------------------------------------------------- /ngx_http_upconf_module.c: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright (C) 2015 Weibo Group Holding Limited 3 | * Copyright (C) 2015 Xiaokai Wang (xiaokai.wang@live.com) 4 | */ 5 | 6 | 7 | #include 8 | #include 9 | #include 10 | 11 | 12 | #define NGX_DELAY_DELETE 75 * 1000 13 | 14 | #define NGX_HTTP_UPCONF_OP_LIST 0 15 | #define NGX_HTTP_UPCONF_OP_ADD 1 16 | #define NGX_HTTP_UPCONF_OP_REMOVE 2 17 | #define NGX_HTTP_UPCONF_OP_BACKUP 4 18 | #define NGX_HTTP_UPCONF_OP_PARAM 8 19 | 20 | 21 | #define NGX_HTTP_UPCONF_WEIGHT 1 22 | #define NGX_HTTP_UPCONF_MAX_FAILS 2 23 | #define NGX_HTTP_UPCONF_FAIL_TIMEOUT 4 24 | #define NGX_HTTP_UPCONF_UP 8 25 | #define NGX_HTTP_UPCONF_DOWN 16 26 | 27 | 28 | static const ngx_str_t ngx_http_upconf_params[] = { 29 | ngx_string("arg_upstream"), 30 | ngx_string("arg_verbose"), 31 | ngx_string("arg_add"), 32 | ngx_string("arg_remove"), 33 | ngx_string("arg_backup"), 34 | ngx_string("arg_server"), 35 | ngx_string("arg_weight"), 36 | ngx_string("arg_max_fails"), 37 | ngx_string("arg_fail_timeout"), 38 | ngx_string("arg_up"), 39 | ngx_string("arg_down") 40 | }; 41 | 42 | 43 | typedef struct { 44 | ngx_int_t op; 45 | ngx_int_t op_params; 46 | ngx_int_t verbose; 47 | ngx_int_t status; 48 | 49 | ngx_str_t ip_port; 50 | ngx_str_t upstream; 51 | 52 | time_t fail_timeout; 53 | ngx_int_t weight; 54 | ngx_uint_t max_fails; 55 | 56 | unsigned down:1; 57 | unsigned backup:1; 58 | } ngx_http_server_t; 59 | 60 | 61 | typedef struct { 62 | ngx_str_t upconf_dump_path; 63 | 64 | ngx_uint_t upstream_conf; 65 | } ngx_http_upconf_loc_conf_t; 66 | 67 | 68 | typedef struct { 69 | ngx_event_t delay_delete_ev; 70 | 71 | time_t start_sec; 72 | ngx_msec_t start_msec; 73 | 74 | void *data; 75 | } ngx_delay_event_t; 76 | 77 | 78 | #if (NGX_HTTP_UPSTREAM_CHECK) 79 | 80 | extern ngx_uint_t ngx_http_upstream_check_add_dynamic_peer(ngx_pool_t *pool, 81 | ngx_http_upstream_srv_conf_t *us, ngx_addr_t *peer_addr); 82 | extern void ngx_http_upstream_check_delete_dynamic_peer(ngx_str_t *name, 83 | ngx_addr_t *peer_addr); 84 | 85 | #endif 86 | 87 | 88 | static char *ngx_http_upconf_set(ngx_conf_t *cf, ngx_command_t *cmd, 89 | void *conf); 90 | static ngx_int_t ngx_http_upconf_handler(ngx_http_request_t *r); 91 | static void ngx_http_upconf_init(ngx_http_request_t *r); 92 | static ngx_int_t ngx_http_upconf_build_op(ngx_http_request_t *r, 93 | ngx_http_server_t *server); 94 | static ngx_int_t ngx_http_upconf_add_peer(ngx_http_request_t *r, 95 | ngx_http_server_t *server, ngx_http_upstream_srv_conf_t *uscf); 96 | static ngx_int_t ngx_http_upconf_update_peer(ngx_http_request_t *r, 97 | ngx_http_server_t *server, ngx_http_upstream_srv_conf_t *uscf); 98 | static ngx_int_t ngx_http_upconf_remove_peer(ngx_http_request_t *r, 99 | ngx_http_server_t *server, ngx_http_upstream_srv_conf_t *uscf); 100 | 101 | static ngx_int_t ngx_http_upconf_check_server( 102 | ngx_http_upstream_srv_conf_t *uscf, ngx_str_t *ip_port); 103 | static void *ngx_http_upconf_lookup_upstream(ngx_http_request_t *r, 104 | ngx_http_server_t *server); 105 | static void *ngx_http_upconf_server(ngx_http_request_t *r, 106 | ngx_http_server_t *server); 107 | 108 | static void *ngx_http_upconf_create_loc_conf(ngx_conf_t *cf); 109 | static char *ngx_http_upconf_merge_loc_conf(ngx_conf_t *cf, void *parent, 110 | void *child) 111 | 112 | static ngx_int_t ngx_http_upconf_init_process(ngx_cycle_t *cycle); 113 | static ngx_int_t ngx_http_upconf_init_peers(ngx_cycle_t *cycle, 114 | ngx_http_upstream_srv_conf_t *uscf); 115 | static ngx_int_t ngx_http_upconf_parse_dump_file( 116 | ngx_http_upstream_srv_conf_t *uscf, u_char * server_file); 117 | 118 | void ngx_http_upconf_finalize_request(ngx_http_request_t *r, 119 | ngx_http_server_t *server); 120 | 121 | 122 | static ngx_command_t ngx_http_upconf_commands[] = { 123 | 124 | { ngx_string("upstream_conf"), 125 | NGX_HTTP_LOC_CONF|NGX_CONF_FLAG, 126 | ngx_http_upconf_set, 127 | NGX_HTTP_LOC_CONF_OFFSET, 128 | offsetof(ngx_http_upconf_loc_conf_t, upstream_conf), 129 | NULL }, 130 | 131 | { ngx_string("upconf_dump_path"), 132 | NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1, 133 | ngx_conf_set_str_slot, 134 | NGX_HTTP_LOC_CONF_OFFSET, 135 | offsetof(ngx_http_upconf_loc_conf_t, upconf_dump_path), 136 | NULL }, 137 | 138 | ngx_null_command 139 | }; 140 | 141 | 142 | static ngx_http_module_t ngx_http_upconf_module_ctx = { 143 | NULL, /* preconfiguration */ 144 | NULL, /* postconfiguration */ 145 | 146 | NULL, /* create main configuration */ 147 | NULL, /* init main configuration */ 148 | 149 | NULL, /* create server configuration */ 150 | NULL, /* merge server configuration */ 151 | 152 | ngx_http_upconf_create_loc_conf, /* create location configuration */ 153 | ngx_http_upconf_merge_loc_conf /* merge location configuration */ 154 | }; 155 | 156 | 157 | ngx_module_t ngx_http_upconf_module = { 158 | NGX_MODULE_V1, 159 | &ngx_http_upconf_module_ctx, /* module context */ 160 | ngx_http_upconf_commands, /* module directives */ 161 | NGX_HTTP_MODULE, /* module type */ 162 | NULL, /* init master */ 163 | NULL, /* init module */ 164 | ngx_http_upconf_init_process, /* init process */ 165 | NULL, /* init thread */ 166 | NULL, /* exit thread */ 167 | NULL, /* exit process */ 168 | NULL, /* exit master */ 169 | NGX_MODULE_V1_PADDING 170 | }; 171 | 172 | 173 | static char * 174 | ngx_http_upconf_set(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) 175 | { 176 | char *ret; 177 | 178 | ngx_http_core_loc_conf_t *clcf; 179 | ngx_http_upconf_loc_conf_t *uclcf = conf; 180 | 181 | ret = ngx_conf_set_flag_slot(cf, cmd, conf); 182 | if(ret != NGX_CONF_OK) { 183 | return ret; 184 | } 185 | 186 | if(uclcf->upstream_conf){ 187 | clcf = ngx_http_conf_get_module_loc_conf(cf, ngx_http_core_module); 188 | clcf->handler = ngx_http_upconf_handler; 189 | } 190 | 191 | return NGX_CONF_OK; 192 | } 193 | 194 | 195 | static ngx_int_t 196 | ngx_http_upconf_handler(ngx_http_request_t *r) 197 | { 198 | ngx_int_t rc; 199 | 200 | if (!(r->method & NGX_HTTP_GET)) { 201 | return NGX_HTTP_NOT_ALLOWED; 202 | } 203 | 204 | rc = ngx_http_read_client_request_body(r, ngx_http_upconf_init); 205 | if (rc >= NGX_HTTP_SPECIAL_RESPONSE) { 206 | return rc; 207 | } 208 | 209 | return NGX_DONE; 210 | } 211 | 212 | 213 | static void 214 | ngx_http_upconf_init(ngx_http_request_t *r) 215 | { 216 | ngx_http_server_t server; 217 | ngx_http_upstream_srv_conf_t *uscf; 218 | 219 | if (ngx_http_upconf_build_op(r, &server) != NGX_OK) { 220 | ngx_log_error(NGX_LOG_ERROR, r->connection->log, 0, 221 | "upconf_init: args invalid \"%V\"", &r->args); 222 | 223 | ngx_http_upconf_finalize_request(r, &server); 224 | 225 | return; 226 | } 227 | 228 | ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, 229 | "upconf_init: parse args succeed"); 230 | 231 | if ((uscf = ngx_http_upconf_lookup_upstream(r, &server)) == NULL) { 232 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 233 | "upconf_init: upstream \"%V\" not exist", 234 | &server->upstream); 235 | return NGX_ERROR; 236 | } 237 | 238 | server.status = ngx_http_upconf_op(r, &server, uscf); 239 | 240 | ngx_http_upconf_finalize_request(r, &server); 241 | 242 | return; 243 | } 244 | 245 | 246 | ngx_int_t 247 | ngx_http_upconf_build_op(ngx_http_request_t *r, ngx_http_server_t *server) 248 | { 249 | size_t args_size; 250 | u_char *low; 251 | ngx_str_t *args; 252 | ngx_uint_t i; 253 | ngx_uint_t key; 254 | ngx_http_variable_value_t *var; 255 | 256 | ngx_memzero(server, sizeof(ngx_http_server_t)); 257 | 258 | /* default setting for op */ 259 | server->op = NGX_HTTP_UPCONF_OP_LIST; 260 | server->status = NGX_HTTP_OK; 261 | ngx_str_null(&server->upstream); 262 | server->weight = 1; 263 | server->max_fails = 1; 264 | server->fail_timeout = 10; 265 | 266 | args = (ngx_str_t *)&ngx_http_upconf_params; 267 | args_size = sizeof(ngx_http_upconf_params) / sizeof(ngx_str_t); 268 | for (i=0; i < args_size; i++) { 269 | 270 | low = ngx_pnalloc(r->pool, args[i].len); 271 | if (low == NULL) { 272 | server->status = NGX_HTTP_INTERNAL_SERVER_ERROR; 273 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 274 | "upconf_build_op: failed to allocate memory"); 275 | return NGX_ERROR; 276 | } 277 | 278 | key = ngx_hash_strlow(low, args[i].data, args[i].len); 279 | var = ngx_http_get_variable(r, &args[i], key); 280 | 281 | if (!var->not_found) { 282 | if (ngx_strcmp("arg_upstream", args[i].data) == 0) { 283 | server->upstream.data = var->data; 284 | server->upstream.len = var->len; 285 | 286 | } else if (ngx_strcmp("arg_verbose", args[i].data) == 0) { 287 | server->verbose = 1; 288 | 289 | } else if (ngx_strcmp("arg_add", args[i].data) == 0) { 290 | server->op |= NGX_HTTP_UPCONF_OP_ADD; 291 | 292 | } else if (ngx_strcmp("arg_remove", args[i].data) == 0) { 293 | server->op |= NGX_HTTP_UPCONF_OP_REMOVE; 294 | 295 | } else if (ngx_strcmp("arg_backup", args[i].data) == 0) { 296 | server->backup = 1; 297 | 298 | } else if (ngx_strcmp("arg_server", args[i].data) == 0) { 299 | server->ip_port.data = var->data; 300 | server->ip_port.len = var->len; 301 | 302 | } else if (ngx_strcmp("arg_weight", args[i].data) == 0) { 303 | server->weight = ngx_atoi(var->data, var->len); 304 | if (server->weight == NGX_ERROR) { 305 | server->status = NGX_HTTP_BAD_REQUEST; 306 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 307 | "upconf_build_op: weight is invalid"); 308 | return NGX_ERROR; 309 | } 310 | 311 | server->op |= NGX_HTTP_UPCONF_OP_PARAM; 312 | server->op_param |= NGX_HTTP_UPCONF_WEIGHT; 313 | server->verbose = 1; 314 | 315 | } else if (ngx_strcmp("arg_max_fails", args[i].data) == 0) { 316 | op->max_fails = ngx_atoi(var->data, var->len); 317 | if (op->max_fails == NGX_ERROR) { 318 | op->status = NGX_HTTP_BAD_REQUEST; 319 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 320 | "upconf_build_op: max_fails is invalid"); 321 | return NGX_ERROR; 322 | } 323 | 324 | server->op |= NGX_HTTP_UPCONF_OP_PARAM; 325 | server->op_param |= NGX_HTTP_UPCONF_MAX_FAILS; 326 | server->verbose = 1; 327 | 328 | } else if (ngx_strcmp("arg_fail_timeout", args[i].data) == 0) { 329 | op->fail_timeout = ngx_atoi(var->data, var->len); 330 | if (op->fail_timeout == NGX_ERROR) { 331 | op->status = NGX_HTTP_BAD_REQUEST; 332 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 333 | "upconf_build_op: fail_timeout is invalid"); 334 | return NGX_ERROR; 335 | } 336 | 337 | server->op |= NGX_HTTP_UPCONF_OP_PARAM; 338 | server->op_param |= NGX_HTTP_UPCONF_FAIL_TIMEOUT; 339 | server->verbose = 1; 340 | 341 | } else if (ngx_strcmp("arg_up", args[i].data) == 0) { 342 | server->up = 1; 343 | server->op |= NGX_HTTP_UPCONF_OP_PARAM; 344 | server->op_param |= NGX_HTTP_UPCONF_UP; 345 | server->verbose = 1; 346 | 347 | } else if (ngx_strcmp("arg_down", args[i].data) == 0) { 348 | server->down = 1; 349 | server->op |= NGX_HTTP_UPCONF_OP_PARAM; 350 | server->op_param |= NGX_HTTP_UPCONF_DOWN; 351 | server->verbose = 1; 352 | 353 | } 354 | } 355 | } 356 | 357 | /* can not add and remove at once */ 358 | if ((server->op & NGX_HTTP_UPCONF_OP_ADD) && 359 | (server->op & NGX_HTTP_UPCONF_OP_REMOVE)) 360 | { 361 | server->status = NGX_HTTP_BAD_REQUEST; 362 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 363 | "upconf_build_op: add and remove at once are not allowed"); 364 | return NGX_ERROR; 365 | } 366 | 367 | if (server->op & NGX_HTTP_UPCONF_OP_ADD) { 368 | server->op = NGX_HTTP_UPCONF_OP_ADD; 369 | 370 | } else if (server->op & NGX_HTTP_UPCONF_OP_REMOVE) { 371 | server->op = NGX_HTTP_UPCONF_OP_REMOVE; 372 | } 373 | 374 | /* can not up and down at once */ 375 | if (server->up && server->down) { 376 | server->status = NGX_HTTP_BAD_REQUEST; 377 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 378 | "upconf_build_op: down and up at once are not allowed"); 379 | return NGX_ERROR; 380 | } 381 | 382 | return NGX_OK; 383 | } 384 | 385 | 386 | ngx_int_t 387 | ngx_http_upconf_op(ngx_http_request_t *r, ngx_http_server_t *server, 388 | ngx_http_upstream_srv_conf_t *uscf) 389 | { 390 | ngx_int_t rc; 391 | 392 | switch (server->op) { 393 | 394 | case NGX_HTTP_UPCONF_OP_ADD: 395 | rc = ngx_http_upconf_add_peer(r, server, uscf); 396 | break; 397 | 398 | case NGX_HTTP_UPCONF_OP_REMOVE: 399 | rc = ngx_http_upconf_remove_peer(r, server, uscf); 400 | break; 401 | 402 | case NGX_HTTP_UPCONF_OP_PARAM: 403 | rc = ngx_http_upconf_update_peer(r, server, uscf); 404 | break; 405 | 406 | case NGX_HTTP_UPCONF_OP_LIST: 407 | default: 408 | rc = NGX_OK; 409 | break; 410 | } 411 | 412 | return rc; 413 | } 414 | 415 | 416 | static ngx_int_t 417 | ngx_http_upconf_add_peer(ngx_http_request_t *r, ngx_http_server_t *server, 418 | ngx_http_upstream_srv_conf_t *uscf) 419 | { 420 | ngx_uint_t n, w; 421 | ngx_http_upstream_server_t *upstream_server; 422 | ngx_http_upstream_rr_peers_t *peers=NULL, tmp_peers=NULL; 423 | ngx_http_upstream_srv_conf_t *uscf; 424 | 425 | if (ngx_http_upconf_check_server(uscf, &server->ip_port) != -1) { 426 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 427 | "upconf_add_peer: server \"%V\" exist", &server->ip_port); 428 | return NGX_ERROR; 429 | } 430 | 431 | if ((upstream_server = ngx_http_upconf_server(r, server)) != NULL) { 432 | return NGX_ERROR; 433 | } 434 | 435 | if (uscf->peer.data != NULL) { 436 | tmp_peers = (ngx_http_upstream_rr_peers_t *)uscf->peer.data; 437 | 438 | } else { 439 | return NGX_ERROR; 440 | } 441 | 442 | n = tmp_peers->number + 1; 443 | peers = ngx_calloc(sizeof(ngx_http_upstream_rr_peers_t) 444 | + sizeof(ngx_http_upstream_rr_peer_t) * (n - 1), 445 | r->connection->log); 446 | if (peers == NULL) { 447 | goto invalid; 448 | } 449 | ngx_memcpy(peers, tmp_peers, sizeof(ngx_http_upstream_rr_peers_t) + 450 | + sizeof(ngx_http_upstream_rr_peer_t) * (tmp_peers->number)); 451 | 452 | peers->peer[n - 1].sockaddr = upstream_server->addrs->sockaddr; 453 | peers->peer[n - 1].socklen = upstream_server->addrs->socklen; 454 | peers->peer[n - 1].name = upstream_server->addrs->name; 455 | peers->peer[n - 1].max_fails = upstream_server->max_fails; 456 | peers->peer[n - 1].fail_timeout = upstream_server->fail_timeout; 457 | peers->peer[n - 1].down = upstream_server->down; 458 | peers->peer[n - 1].weight = upstream_server->weight; 459 | peers->peer[n - 1].effective_weight = upstream_server->weight; 460 | peers->peer[n - 1].current_weight = 0; 461 | 462 | #if (NGX_HTTP_UPSTREAM_CHECK) 463 | ngx_uint_t index; 464 | index = ngx_http_upstream_check_add_dynamic_peer(cycle->pool, 465 | uscf, server->addrs); 466 | peers->peer[m].check_index = index; 467 | #endif 468 | w = tmp_peers->total_weight + server->weight; 469 | 470 | peers->single = (n == 1); 471 | peers->number = n; 472 | peers->weighted = (w != n); 473 | peers->total_weight = w; 474 | 475 | uscf->peer.data = peers; 476 | peers->next = tmp_peers->next; 477 | 478 | if (r->start_sec != 0) { 479 | ngx_http_upconf_event_init(tmp_peers, upconf_server, NGX_ADD); 480 | 481 | } else { 482 | ngx_free(tmp_peers); 483 | } 484 | 485 | return NGX_OK; 486 | 487 | invalid: 488 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 489 | "upconf_add_peer: add failed \"%V\"", &uscf->host); 490 | 491 | if (peers != NULL) { 492 | ngx_free(peers); 493 | } 494 | peers = NULL; 495 | 496 | uscf->peer.data = tmp_peers; 497 | 498 | return NGX_OK; 499 | } 500 | 501 | 502 | static ngx_int_t 503 | ngx_http_upconf_update_peer(ngx_http_request_t *r, ngx_http_server_t *server, 504 | ngx_http_upstream_srv_conf_t *uscf) 505 | { 506 | ngx_uint_t n, w, index; 507 | ngx_http_upstream_server_t *upstream_server; 508 | ngx_http_upstream_rr_peers_t *peers=NULL; 509 | ngx_http_upstream_srv_conf_t *uscf; 510 | 511 | if ((index = ngx_http_upconf_check_server(uscf, &server->ip_port)) == -1) { 512 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 513 | "upconf_update_peer: server \"%V\" not exist", 514 | &server->ip_port); 515 | return NGX_ERROR; 516 | } 517 | 518 | if (uscf->peer.data != NULL) { 519 | peers = (ngx_http_upstream_rr_peers_t *)uscf->peer.data; 520 | 521 | } else { 522 | return NGX_ERROR; 523 | } 524 | 525 | n = peers->number; 526 | w = peers->total_weight - server->weight; 527 | 528 | peers->weighted = (w != n); 529 | peers->total_weight = w; 530 | peers->peer[n].max_fails = server->max_fails; 531 | peers->peer[n].fail_timeout = server->fail_timeout; 532 | peers->peer[n].down = server->down; 533 | peers->peer[n].weight = server->weight; 534 | 535 | return NGX_OK; 536 | } 537 | 538 | 539 | static ngx_int_t 540 | ngx_http_upconf_remove_peer(ngx_http_request_t *r, ngx_http_server_t *server, 541 | ngx_http_upstream_srv_conf_t *uscf) 542 | { 543 | ngx_uint_t i, n, w, index; 544 | ngx_http_upstream_server_t *upstream_server; 545 | ngx_http_upstream_rr_peers_t *peers=NULL, tmp_peers=NULL; 546 | ngx_http_upstream_srv_conf_t *uscf; 547 | 548 | if ((index = ngx_http_upconf_check_server(uscf, &server->ip_port)) == -1) { 549 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 550 | "upconf_remove_peer: server \"%V\" not exist", 551 | &server->ip_port); 552 | return NGX_ERROR; 553 | } 554 | 555 | if (uscf->peer.data != NULL) { 556 | tmp_peers = (ngx_http_upstream_rr_peers_t *)uscf->peer.data; 557 | 558 | } else { 559 | return NGX_ERROR; 560 | } 561 | 562 | n = tmp_peers->number - 1; 563 | w = tmp_peers->total_weight - server->weight; 564 | 565 | peers = ngx_calloc(sizeof(ngx_http_upconf_rr_peers_t) 566 | + sizeof(ngx_http_upconf_rr_peer_t) * (n - 1), 567 | r->connection->log); 568 | 569 | if (peers == NULL) { 570 | goto invalid; 571 | } 572 | 573 | peers->single = (n == 1); 574 | peers->number = n; 575 | peers->weighted = (w != n); 576 | peers->total_weight = w; 577 | peers->name = &uscf->host; 578 | 579 | n = 0; 580 | for (i = 0; i < tmp_peers->number; i++) { 581 | if (index == i) { 582 | 583 | #if (NGX_HTTP_UPSTREAM_CHECK) 584 | 585 | ngx_addr_t addr; 586 | addr.sockaddr = tmp_peers->peer[i].sockaddr; 587 | addr.socklen = tmp_peers->peer[i].socklen; 588 | addr.name = tmp_peers->peer[i].name; 589 | ngx_http_upstream_check_delete_dynamic_peer(tmp_peers->name, &addr); 590 | 591 | #endif 592 | 593 | tmp_peers->peer[i].down = NGX_MAX_VALUE; 594 | continue; 595 | } 596 | 597 | peers->peer[n].sockaddr = tmp_peers->peer[i].sockaddr; 598 | peers->peer[n].socklen = tmp_peers->peer[i].socklen; 599 | peers->peer[n].name.len = tmp_peers->peer[i].name.len; 600 | peers->peer[n].name.data = tmp_peers->peer[i].name.data; 601 | peers->peer[n].max_fails = tmp_peers->peer[i].max_fails; 602 | peers->peer[n].fail_timeout = tmp_peers->peer[i].fail_timeout; 603 | peers->peer[n].down = tmp_peers->peer[i].down; 604 | peers->peer[n].weight = tmp_peers->peer[i].weight; 605 | peers->peer[n].effective_weight = tmp_peers->peer[i].effective_weight; 606 | peers->peer[n].current_weight = tmp_peers->peer[i].current_weight; 607 | 608 | #if (NGX_HTTP_UPSTREAM_CHECK) 609 | peers->peer[n].check_index = tmp_peers->peer[i].check_index; 610 | #endif 611 | 612 | n++; 613 | } 614 | 615 | uscf->peer.data = peers; 616 | 617 | /* backup servers */ 618 | peers->next = tmp_backup; 619 | 620 | ngx_http_upconf_event_init(tmp_peers, upstream_host, NGX_DEL); 621 | 622 | return NGX_OK; 623 | 624 | invalid: 625 | ngx_log_error(NGX_LOG_EMERG, cycle->log, 0, 626 | "upstream_remove_server remove failed \"%V\"", 627 | &server->ip_port); 628 | 629 | if (peers != NULL) { 630 | ngx_free(peers); 631 | } 632 | 633 | uscf->peer.data = tmp_peers; 634 | 635 | return NGX_ERROR; 636 | } 637 | 638 | 639 | static ngx_int_t 640 | ngx_http_upconf_check_server(ngx_http_upstream_srv_conf_t *uscf, 641 | ngx_str_t *ip_port) 642 | { 643 | ngx_uint_t i; 644 | ngx_http_upstream_rr_peers_t *peers = NULL; 645 | 646 | if (uscf->peer.data != NULL) { 647 | peers = (ngx_http_upstream_rr_peers_t *)uscf->peer.data; 648 | 649 | } else { 650 | return -1; 651 | } 652 | 653 | for (i = 0; i < peers->number; j++) { 654 | 655 | if (ngx_memcmp(peers->peer[i].name.data, 656 | ip_port->data, peers->peer[i].name.len) 657 | == 0) 658 | { 659 | return i; 660 | } 661 | } 662 | 663 | return -1; 664 | } 665 | 666 | 667 | static void 668 | ngx_http_upconf_event_init(ngx_http_upstream_rr_peers_t *tmp_peers, 669 | ngx_http_upconf_host_t *upstream_host, ngx_flag_t flag) 670 | { 671 | ngx_time_t *tp; 672 | ngx_delay_event_t *delay_event; 673 | 674 | delay_event = ngx_calloc(sizeof(*delay_event), ngx_cycle->log); 675 | if (delay_event == NULL) { 676 | ngx_log_error(NGX_LOG_ERR, ngx_cycle->log, 0, 677 | "dynamic_update_upconf_event_init: calloc failed"); 678 | } 679 | 680 | tp = ngx_timeofday(); 681 | delay_event->start_sec = tp->sec; 682 | delay_event->start_msec = tp->msec; 683 | 684 | if (flag == NGX_ADD) { 685 | delay_event->delay_delete_ev.handler = ngx_http_upconf_add_delay_del; 686 | delay_event->delay_delete_ev.log = ngx_cycle->log; 687 | delay_event->delay_delete_ev.data = delay_event; 688 | delay_event->delay_delete_ev.timer_set = 0; 689 | 690 | } else { 691 | 692 | delay_event->delay_delete_ev.handler = ngx_http_upconf_remove_delay_del; 693 | delay_event->delay_delete_ev.log = ngx_cycle->log; 694 | delay_event->delay_delete_ev.data = delay_event; 695 | delay_event->delay_delete_ev.timer_set = 0; 696 | 697 | } 698 | 699 | delay_event->data = tmp_peers; 700 | ngx_add_timer(&delay_event->delay_delete_ev, NGX_DELAY_DELETE); 701 | 702 | return; 703 | } 704 | 705 | 706 | static void 707 | ngx_http_upconf_add_delay_del(ngx_event_t *event) 708 | { 709 | ngx_uint_t i; 710 | ngx_connection_t *c; 711 | ngx_delay_event_t *delay_event; 712 | ngx_http_request_t *r=NULL; 713 | ngx_http_log_ctx_t *ctx=NULL; 714 | ngx_http_upconf_rr_peers_t *tmp_peers=NULL; 715 | 716 | delay_event = event->data; 717 | if (delay_event == NULL) { 718 | return; 719 | } 720 | tmp_peers = delay_event->data; 721 | 722 | c = ngx_cycle->connections; 723 | for (i = 0; i < ngx_cycle->connection_n; i++) { 724 | 725 | if (c[i].fd == (ngx_socket_t) -1) { 726 | continue; 727 | } else { 728 | 729 | if (c[i].log->data != NULL) { 730 | ctx = c[i].log->data; 731 | r = ctx->current_request; 732 | } 733 | } 734 | 735 | if (r) { 736 | if (r->start_sec < delay_event->start_sec) { 737 | ngx_add_timer(&delay_event->delay_delete_ev, NGX_DELAY_DELETE); 738 | return; 739 | } 740 | 741 | if (r->start_sec == delay_event->start_sec) { 742 | 743 | if (r->start_msec <= delay_event->start_msec) { 744 | ngx_add_timer(&delay_event->delay_delete_ev, NGX_DELAY_DELETE); 745 | return; 746 | } 747 | } 748 | } 749 | } 750 | 751 | if (tmp_peers != NULL) { 752 | 753 | ngx_free(tmp_peers); 754 | tmp_peers = NULL; 755 | } 756 | 757 | ngx_free(delay_event); 758 | delay_event = NULL; 759 | 760 | return; 761 | } 762 | 763 | 764 | static void 765 | ngx_http_upconf_remove_delay_del(ngx_event_t *event) 766 | { 767 | ngx_uint_t i; 768 | ngx_connection_t *c; 769 | ngx_delay_event_t *delay_event; 770 | ngx_http_request_t *r=NULL; 771 | ngx_http_log_ctx_t *ctx=NULL; 772 | ngx_http_upconf_rr_peers_t *tmp_peers=NULL; 773 | 774 | u_char *namep = NULL; 775 | struct sockaddr *saddr = NULL; 776 | 777 | delay_event = event->data; 778 | if (delay_event == NULL) { 779 | return; 780 | } 781 | tmp_peers = delay_event->data; 782 | 783 | c = ngx_cycle->connections; 784 | for (i = 0; i < ngx_cycle->connection_n; i++) { 785 | 786 | if (c[i].fd == (ngx_socket_t) -1) { 787 | continue; 788 | } else { 789 | 790 | if (c[i].log->data != NULL) { 791 | ctx = c[i].log->data; 792 | r = ctx->request; 793 | } 794 | } 795 | 796 | if (r) { 797 | if (r->start_sec < delay_event->start_sec) { 798 | ngx_add_timer(&delay_event->delay_delete_ev, NGX_DELAY_DELETE); 799 | return; 800 | } 801 | 802 | if (r->start_sec == delay_event->start_sec) { 803 | 804 | if (r->start_msec <= delay_event->start_msec) { 805 | ngx_add_timer(&delay_event->delay_delete_ev, NGX_DELAY_DELETE); 806 | return; 807 | } 808 | } 809 | } 810 | } 811 | 812 | if (tmp_peers != NULL) { 813 | for (i = 0; i < tmp_peers->number; i++) { 814 | 815 | if (tmp_peers->peer[i].check_index != NGX_MAX_VALUE) { 816 | continue; 817 | } 818 | 819 | saddr = tmp_peers->peer[i].sockaddr; 820 | if (saddr != NULL) { 821 | ngx_free(saddr); 822 | 823 | } else { 824 | continue; 825 | } 826 | 827 | namep = tmp_peers->peer[i].name.data; 828 | if (namep != NULL) { 829 | ngx_free(namep); 830 | } 831 | 832 | saddr = NULL, namep = NULL; 833 | } 834 | } 835 | 836 | if (tmp_peers != NULL) { 837 | ngx_free(tmp_peers); 838 | tmp_peers = NULL; 839 | } 840 | 841 | ngx_free(delay_event); 842 | delay_event = NULL; 843 | 844 | return; 845 | } 846 | 847 | 848 | static void * 849 | ngx_http_upconf_server(ngx_http_request_t *r, ngx_http_server_t *server) 850 | { 851 | u_char *port, *p, *last, *pp; 852 | ngx_int_t n; 853 | ngx_addr_t *addrs; 854 | ngx_http_upstream_server_t *upstream_server; 855 | 856 | struct sockaddr_in *sin; 857 | 858 | upstream_server = ngx_pcalloc(r->pool, sizeof(ngx_http_upstream_server_t)); 859 | 860 | p = server->ip_port.data; 861 | last = p + ngx_strlen(p); 862 | 863 | port = ngx_strlchr(p, last, ':'); 864 | if (port == NULL) { 865 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 866 | "upconf_server: has no port in %s", p); 867 | return NULL; 868 | } 869 | 870 | n = ngx_atoi(port + 1, last - port - 1); 871 | if (n < 1 || n > 65535) { 872 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 873 | "upconf_server: invalid port in %s", p); 874 | return NULL; 875 | } 876 | 877 | sin = ngx_calloc(sizeof(struct sockaddr_in), cycle->log); 878 | if (sin == NULL) { 879 | goto invalid; 880 | } 881 | 882 | sin->sin_family = AF_INET; 883 | sin->sin_port = htons((in_port_t) n); 884 | sin->sin_addr.s_addr = ngx_inet_addr(p, port - p); 885 | 886 | if (sin->sin_addr.s_addr == INADDR_NONE) { 887 | ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, 888 | "upconf_server: invalid ip in %s", p); 889 | ngx_free(sin); 890 | return NULL; 891 | } 892 | 893 | addrs = ngx_calloc(sizeof(ngx_addr_t), cycle->log); 894 | if (addrs == NULL) { 895 | return NULL; 896 | } 897 | 898 | addrs->sockaddr = (struct sockaddr *) sin; 899 | addrs->socklen = sizeof(struct sockaddr_in); 900 | 901 | pp = ngx_calloc(last - p, cycle->log); 902 | if (pp == NULL) { 903 | goto invalid; 904 | } 905 | addrs->name.len = ngx_sprintf(pp, "%s", p) - pp; 906 | addrs->name.data = pp; 907 | 908 | ngx_memzero(upstream_server, sizeof(ngx_http_upstream_server_t)); 909 | 910 | upstream_server->addrs = addrs; 911 | upstream_server->naddrs = 1; 912 | upstream_server->down = server->down; 913 | upstream_server->backup = server->backup; 914 | upstream_server->weight = server->weight; 915 | upstream_server->max_fails = server->max_fails; 916 | upstream_server->fail_timeout = server->fail_timeout; 917 | 918 | return upstream_server; 919 | 920 | invalid: 921 | 922 | if (server->addrs->sockaddr != NULL) { 923 | ngx_free(server->addrs->sockaddr); 924 | } 925 | if (server->addrs->name.data != NULL) { 926 | ngx_free(server->addrs->name.data); 927 | 928 | ngx_free(upstream_server->addrs); 929 | 930 | return NULL; 931 | } 932 | 933 | 934 | static void * 935 | ngx_http_upconf_lookup_upstream(ngx_http_request_t *r, 936 | ngx_http_server_t *server) 937 | { 938 | ngx_http_upstream_srv_conf_t **uscfp, *uscf; 939 | ngx_http_upstream_main_conf_t *umcf; 940 | 941 | umcf = ngx_http_get_module_main_conf(r, ngx_http_upstream_module); 942 | 943 | uscfp = umcf->upstreams.elts; 944 | for (i = 0; i < umcf->upstreams.nelts; i++) { 945 | 946 | if (uscfp[i]->host.len == upstream_name->len 947 | && ngx_strncasecmp(uscfp[i]->host.data, upstream_name->data, 948 | uscfp[i]->host.len) == 0) 949 | { 950 | return uscfp[i]; 951 | } 952 | } 953 | 954 | return NULL; 955 | } 956 | 957 | 958 | static void * 959 | ngx_http_upconf_create_loc_conf(ngx_conf_t *cf) 960 | { 961 | ngx_http_upconf_loc_conf_t *conf; 962 | 963 | conf = ngx_pcalloc(cf->pool, sizeof(*conf)); 964 | if (conf == NULL) { 965 | return NULL; 966 | } 967 | 968 | conf->upstream_conf = NGX_CONF_UNSET_UINT; 969 | 970 | return conf; 971 | } 972 | 973 | 974 | static char * 975 | ngx_http_upconf_merge_loc_conf(ngx_conf_t *cf, void *parent, void *child) 976 | { 977 | ngx_http_upconf_loc_conf_t *prev = parent; 978 | ngx_http_upconf_loc_conf_t *conf = child; 979 | 980 | ngx_conf_merge_uint_value(conf->upstream_conf, prev->upstream_conf, 0); 981 | 982 | ngx_conf_merge_str_value(conf->upconf_dump_path, prev->upconf_dump_path, "/tmp"); 983 | 984 | return NGX_CONF_OK; 985 | } 986 | 987 | 988 | static ngx_int_t 989 | ngx_http_upconf_init_process(ngx_cycle_t *cycle) 990 | { 991 | u_char *server_file; 992 | ngx_uint_t i; 993 | ngx_http_upstream_srv_conf_t **uscfp; 994 | ngx_http_upstream_main_conf_t *umcf; 995 | 996 | umcf = ngx_http_cycle_get_module_main_conf(cycle, ngx_http_upstream_module); 997 | 998 | uscfp = umcf->upstreams.elts; 999 | for (i = 0; i < umcf->upstreams.nelts; i++) { 1000 | 1001 | ngx_http_upconf_init_peers(cycle, uscfp[i]); 1002 | 1003 | ngx_http_upconf_parse_dump_file(uscfp[i], server_file); 1004 | 1005 | } 1006 | 1007 | return NGX_OK; 1008 | } 1009 | 1010 | 1011 | static ngx_int_t 1012 | ngx_http_upconf_init_peers(ngx_cycle_t *cycle, 1013 | ngx_http_upstream_srv_conf_t *uscf) 1014 | { 1015 | ngx_uint_t i, n, len; 1016 | ngx_http_upstream_rr_peers_t *peers = NULL, *tmp_peers = NULL; 1017 | 1018 | u_char *namep = NULL; 1019 | struct sockaddr *saddr = NULL; 1020 | len = sizeof(struct sockaddr); 1021 | 1022 | ngx_queue_init(&upconf_server->add_ev); 1023 | ngx_queue_init(&upconf_server->delete_ev); 1024 | 1025 | if (uscf->peer.data != NULL) { 1026 | tmp_peers = (ngx_http_upstream_rr_peers_t *)uscf->peer.data; 1027 | } 1028 | 1029 | if (tmp_peers) { 1030 | n = tmp_peers->number; 1031 | 1032 | peers = ngx_calloc(sizeof(ngx_http_upstream_rr_peers_t) 1033 | + sizeof(ngx_http_upstream_rr_peer_t) * (n - 1), 1034 | cycle->log); 1035 | if (peers == NULL) { 1036 | goto invalid; 1037 | } 1038 | 1039 | peers->single = tmp_peers->single; 1040 | peers->number = tmp_peers->number; 1041 | peers->weighted = tmp_peers->weighted; 1042 | peers->total_weight = tmp_peers->total_weight; 1043 | peers->name = tmp_peers->name; 1044 | 1045 | n = 0; 1046 | for (i = 0; i < tmp_peers->number; i++) { 1047 | 1048 | if ((saddr = ngx_calloc(len, cycle->log)) == NULL) { 1049 | goto invalid; 1050 | } 1051 | ngx_memcpy(saddr, tmp_peers->peer[i].sockaddr, len); 1052 | peers->peer[i].sockaddr = saddr; 1053 | 1054 | peers->peer[i].socklen = tmp_peers->peer[i].socklen; 1055 | peers->peer[i].name.len = tmp_peers->peer[i].name.len; 1056 | 1057 | if ((namep = ngx_calloc(tmp_peers->peer[i].name.len, 1058 | cycle->log)) == NULL) { 1059 | goto invalid; 1060 | } 1061 | ngx_memcpy(namep, tmp_peers->peer[i].name.data, 1062 | tmp_peers->peer[i].name.len); 1063 | peers->peer[i].name.data = namep; 1064 | 1065 | peers->peer[i].max_fails = tmp_peers->peer[i].max_fails; 1066 | peers->peer[i].fail_timeout = tmp_peers->peer[i].fail_timeout; 1067 | peers->peer[i].down = tmp_peers->peer[i].down; 1068 | peers->peer[i].weight = tmp_peers->peer[i].weight; 1069 | peers->peer[i].effective_weight = tmp_peers->peer[i].effective_weight; 1070 | peers->peer[i].current_weight = tmp_peers->peer[i].current_weight; 1071 | 1072 | #if (NGX_HTTP_UPSTREAM_CHECK) 1073 | peers->peer[i].check_index = tmp_peers->peer[i].check_index; 1074 | #endif 1075 | } 1076 | 1077 | uscf->peer.data = peers; 1078 | peers->next = tmp_peers->next; 1079 | 1080 | ngx_pfree(cycle->pool, tmp_peers); 1081 | } 1082 | 1083 | return NGX_OK; 1084 | 1085 | invalid: 1086 | ngx_log_error(NGX_LOG_ERR, cycle->log, 0, 1087 | "upconf_init_peers: copy failed \"%V\"", &uscf->host); 1088 | 1089 | if (peers != NULL) { 1090 | ngx_free(peers); 1091 | } 1092 | uscf->peer.data = tmp_peers; 1093 | 1094 | return NGX_ERROR; 1095 | } 1096 | 1097 | 1098 | static ngx_int_t 1099 | ngx_http_upconf_parse_dump_file(ngx_http_upstream_srv_conf_t *uscf, 1100 | u_char * server_file) 1101 | { 1102 | char *prev = NULL, *delimiter = NULL, read_line[1024]; 1103 | ngx_int_t max_fails; 1104 | ngx_str_t s; 1105 | ngx_pool_t *pool; 1106 | ngx_connection_t connection; 1107 | ngx_http_server_t server; 1108 | ngx_http_request_t r; 1109 | 1110 | pool = ngx_create_pool(NGX_DEFAULT_POOL_SIZE, ngx_cycle->log); 1111 | if (pool == NULL) { 1112 | ngx_log_error(NGX_LOG_ERR, ngx_cycle->log, 0, 1113 | "upconf_parse_dump_file: alloc pool failed"); 1114 | return NGX_ERROR; 1115 | } 1116 | 1117 | r.pool = pool; 1118 | connection.log = ngx_cycle->log; 1119 | r.connection = &connection; 1120 | r.start_sec = 0; 1121 | 1122 | FILE *fp = ngx_fopen((char *)server_file, "r"); 1123 | if (fp == NULL) { 1124 | ngx_log_error(NGX_LOG_ERR, ngx_cycle->log, 0, 1125 | "upconf_parse_dump_file: open dump file \"%s\" failed", 1126 | server_file); 1127 | ngx_destroy_pool(pool); 1128 | 1129 | return NGX_ERROR; 1130 | } 1131 | 1132 | while (ngx_fgets(read_line, 1024, fp) != NULL) { 1133 | 1134 | prev = read_line; 1135 | while(*prev != ';' && *prev != '\0') { 1136 | 1137 | if (ngx_strncmp(prev, "server", 6) == 0) { 1138 | prev += 7; 1139 | delimiter = ngx_strchr(prev, ' '); 1140 | if (delimiter == NULL) { 1141 | delimiter = ngx_strchr(prev, ';'); 1142 | } 1143 | if (delimiter == NULL) { 1144 | ngx_log_error(NGX_LOG_ERR, ngx_cycle->log, 0, 1145 | "server format error: \"%s\" ", read_line); 1146 | break; 1147 | } 1148 | 1149 | server->ip_port.len = delimiter - prev; 1150 | server->ip_port.data = ngx_pcalloc(pool, server->ip_port.len); 1151 | if (server->ip_port.data == NULL) { 1152 | ngx_log_error(NGX_LOG_ERR, pool->log, 0, 1153 | "upconf_init_peers: memory alloc error"); 1154 | return NGX_ERROR; 1155 | } 1156 | ngx_memmove(server->ip_port.data, prev, server->ip_port.len); 1157 | 1158 | server->upstream = uscf->host; 1159 | 1160 | /* default value, server attribute */ 1161 | server->weight = 1; 1162 | server->max_fails = 2; 1163 | server_conf->fail_timeout = 10; 1164 | 1165 | server->down = 0; 1166 | server_conf->backup = 0; 1167 | 1168 | prev = delimiter; 1169 | delimiter = NULL; 1170 | 1171 | continue; 1172 | } 1173 | 1174 | if (ngx_strncmp(prev, "weight=", 7) == 0) { 1175 | prev += 7; 1176 | delimiter = ngx_strchr(prev, ' '); 1177 | if (delimiter == NULL) { 1178 | delimiter = ngx_strchr(prev, ';'); 1179 | } 1180 | 1181 | if (delimiter == NULL) { 1182 | continue; 1183 | } 1184 | 1185 | server->weight = ngx_atoi((u_char *)prev, 1186 | (size_t)(delimiter - prev)); 1187 | 1188 | if (server->weight < 0) { 1189 | server->weight = 1; 1190 | } 1191 | prev = delimiter; 1192 | delimiter = NULL; 1193 | 1194 | continue; 1195 | } 1196 | 1197 | if (ngx_strncmp(prev, "max_fails=", 10) == 0) { 1198 | prev += 10; 1199 | delimiter = ngx_strchr(prev, ' '); 1200 | if (delimiter == NULL) { 1201 | delimiter = ngx_strchr(prev, ';'); 1202 | } 1203 | 1204 | if (delimiter == NULL) { 1205 | continue; 1206 | } 1207 | 1208 | max_fails = ngx_atoi((u_char *)prev, 1209 | (size_t)(delimiter - prev)); 1210 | 1211 | if (max_fails < 0) { 1212 | server->max_fails = 2; 1213 | 1214 | } else { 1215 | server->max_fails = max_fails; 1216 | } 1217 | prev = delimiter; 1218 | delimiter = NULL; 1219 | 1220 | continue; 1221 | } 1222 | 1223 | if (ngx_strncmp(prev, "fail_timeout=", 13) == 0) { 1224 | prev += 13; 1225 | delimiter = ngx_strchr(prev, ' '); 1226 | if (delimiter == NULL) { 1227 | delimiter = ngx_strchr(prev, ';'); 1228 | } 1229 | 1230 | if (delimiter == NULL) { 1231 | continue; 1232 | } 1233 | 1234 | s.data = (u_char *)prev; 1235 | s.len = delimiter - prev; 1236 | server->fail_timeout = ngx_parse_time(&s, 1); 1237 | 1238 | if (server->fail_timeout < 0) { 1239 | server->fail_timeout = 10; 1240 | } 1241 | prev = delimiter; 1242 | delimiter = NULL; 1243 | 1244 | continue; 1245 | } 1246 | 1247 | prev++; 1248 | } 1249 | 1250 | ngx_http_upconf_add_peer(&r, &server, uscf); 1251 | } 1252 | ngx_fclose(fp); 1253 | 1254 | ngx_destroy_pool(pool); 1255 | 1256 | return NGX_OK; 1257 | } 1258 | 1259 | 1260 | void 1261 | ngx_http_upconf_finalize_request(ngx_http_request_t *r, 1262 | ngx_http_server_t *server) 1263 | { 1264 | ngx_int_t len; 1265 | ngx_buf_t *b = NULL; 1266 | 1267 | ngx_log_debug0(NGX_LOG_DEBUG, r->connection->log, 0, 1268 | "upconf_finalize_request: upstream finalize_success"); 1269 | 1270 | len = ngx_strlen(add_succeed); 1271 | 1272 | b = ngx_create_temp_buf(r->pool, len); 1273 | ngx_sprintf(b->pos, "%s", add_succeed); 1274 | b->last = b->pos + len; 1275 | 1276 | b->last_buf = 1; 1277 | ngx_chain_t out; 1278 | out.buf = b; 1279 | out.next = NULL; 1280 | 1281 | r->headers_out.status = NGX_HTTP_OK; 1282 | r->headers_out.content_length_n = len; 1283 | ngx_str_set(&r->headers_out.content_type, "text/plain"); 1284 | 1285 | r->connection->buffered |= NGX_HTTP_WRITE_BUFFERED; 1286 | ngx_int_t ret = ngx_http_send_header(r); 1287 | ret = ngx_http_output_filter(r, &out); 1288 | 1289 | ngx_http_finalize_request(r, ret); 1290 | return; 1291 | } 1292 | --------------------------------------------------------------------------------