mirror/nginx
1
0
Fork 0
mirror of https://github.com/nginx/nginx.git synced 2024-12-05 06:19:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Stream: prepared proxy_bind to accept parameters.

Browse Source
This commit is contained in:
Roman Arutyunyan 2016-04-13 15:42:47 +03:00
parent f10bc5a763
commit 77ec993fd7
1 changed files with 48 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/stream/ngx_stream_proxy_module.c
View File

@ -10,6 +10,11 @@
#include <ngx_stream.h> #include <ngx_stream.h>
typedef struct {
ngx_addr_t *addr;
} ngx_stream_upstream_local_t;
typedef struct { typedef struct {
ngx_msec_t connect_timeout; ngx_msec_t connect_timeout;
ngx_msec_t timeout; ngx_msec_t timeout;
@ -21,7 +26,7 @@ typedef struct {
ngx_uint_t next_upstream_tries; ngx_uint_t next_upstream_tries;
ngx_flag_t next_upstream; ngx_flag_t next_upstream;
ngx_flag_t proxy_protocol; ngx_flag_t proxy_protocol;
ngx_addr_t *local; ngx_stream_upstream_local_t *local;
#if (NGX_STREAM_SSL) #if (NGX_STREAM_SSL)
ngx_flag_t ssl_enable; ngx_flag_t ssl_enable;
@ -47,6 +52,8 @@ typedef struct {
static void ngx_stream_proxy_handler(ngx_stream_session_t *s); static void ngx_stream_proxy_handler(ngx_stream_session_t *s);
static ngx_int_t ngx_stream_proxy_set_local(ngx_stream_session_t *s,
ngx_stream_upstream_t *u, ngx_stream_upstream_local_t *local);
static void ngx_stream_proxy_connect(ngx_stream_session_t *s); static void ngx_stream_proxy_connect(ngx_stream_session_t *s);
static void ngx_stream_proxy_init_upstream(ngx_stream_session_t *s); static void ngx_stream_proxy_init_upstream(ngx_stream_session_t *s);
static void ngx_stream_proxy_upstream_handler(ngx_event_t *ev); static void ngx_stream_proxy_upstream_handler(ngx_event_t *ev);
@ -358,7 +365,11 @@ ngx_stream_proxy_handler(ngx_stream_session_t *s)
u->peer.log = c->log; u->peer.log = c->log;
u->peer.log_error = NGX_ERROR_ERR; u->peer.log_error = NGX_ERROR_ERR;
u->peer.local = pscf->local; if (ngx_stream_proxy_set_local(s, u, pscf->local) != NGX_OK) {
ngx_stream_proxy_finalize(s, NGX_ERROR);
return;
}
u->peer.type = c->type; u->peer.type = c->type;
uscf = pscf->upstream; uscf = pscf->upstream;
@ -428,6 +439,24 @@ ngx_stream_proxy_handler(ngx_stream_session_t *s)
} }
static ngx_int_t
ngx_stream_proxy_set_local(ngx_stream_session_t *s, ngx_stream_upstream_t *u,
ngx_stream_upstream_local_t *local)
{
if (local == NULL) {
u->peer.local = NULL;
return NGX_OK;
}
if (local->addr) {
u->peer.local = local->addr;
return NGX_OK;
}
return NGX_OK;
}
static void static void
ngx_stream_proxy_connect(ngx_stream_session_t *s) ngx_stream_proxy_connect(ngx_stream_session_t *s)
{ {
@ -1639,6 +1668,7 @@ ngx_stream_proxy_bind(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
ngx_int_t rc; ngx_int_t rc;
ngx_str_t *value; ngx_str_t *value;
ngx_stream_upstream_local_t *local;
if (pscf->local != NGX_CONF_UNSET_PTR) { if (pscf->local != NGX_CONF_UNSET_PTR) {
return "is duplicate"; return "is duplicate";
@ -1651,17 +1681,24 @@ ngx_stream_proxy_bind(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
return NGX_CONF_OK; return NGX_CONF_OK;
} }
pscf->local = ngx_palloc(cf->pool, sizeof(ngx_addr_t)); local = ngx_palloc(cf->pool, sizeof(ngx_stream_upstream_local_t));
if (pscf->local == NULL) { if (local == NULL) {
return NGX_CONF_ERROR; return NGX_CONF_ERROR;
} }
rc = ngx_parse_addr(cf->pool, pscf->local, value[1].data, value[1].len); pscf->local = local;
local->addr = ngx_palloc(cf->pool, sizeof(ngx_addr_t));
if (local->addr == NULL) {
return NGX_CONF_ERROR;
}
rc = ngx_parse_addr(cf->pool, local->addr, value[1].data, value[1].len);
switch (rc) { switch (rc) {
case NGX_OK: case NGX_OK:
pscf->local->name = value[1]; local->addr->name = value[1];
return NGX_CONF_OK; break;
case NGX_DECLINED: case NGX_DECLINED:
ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
@ -1671,4 +1708,6 @@ ngx_stream_proxy_bind(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
default: default:
return NGX_CONF_ERROR; return NGX_CONF_ERROR;
} }
return NGX_CONF_OK;
} }