Mercurial > hg > nginx
diff src/stream/ngx_stream_variables.c @ 6670:c6372a40c2a7
Stream: the $protocol variable.
The variable keeps protocol used by the client, "TCP" or "UDP".
author | Vladimir Homutov <vl@nginx.com> |
---|---|
date | Fri, 26 Aug 2016 15:33:07 +0300 |
parents | 164a0824ce20 |
children | 38143d1abdec |
line wrap: on
line diff
--- a/src/stream/ngx_stream_variables.c +++ b/src/stream/ngx_stream_variables.c @@ -40,6 +40,8 @@ static ngx_int_t ngx_stream_variable_tim ngx_stream_variable_value_t *v, uintptr_t data); static ngx_int_t ngx_stream_variable_time_local(ngx_stream_session_t *s, ngx_stream_variable_value_t *v, uintptr_t data); +static ngx_int_t ngx_stream_variable_protocol(ngx_stream_session_t *s, + ngx_stream_variable_value_t *v, uintptr_t data); static ngx_stream_variable_t ngx_stream_core_variables[] = { @@ -89,6 +91,9 @@ static ngx_stream_variable_t ngx_stream { ngx_string("time_local"), NULL, ngx_stream_variable_time_local, 0, NGX_STREAM_VAR_NOCACHEABLE, 0 }, + { ngx_string("protocol"), NULL, + ngx_stream_variable_protocol, 0, 0, 0 }, + { ngx_null_string, NULL, NULL, 0, 0, 0 } }; @@ -665,6 +670,20 @@ ngx_stream_variable_time_local(ngx_strea } +static ngx_int_t +ngx_stream_variable_protocol(ngx_stream_session_t *s, + ngx_stream_variable_value_t *v, uintptr_t data) +{ + v->len = 3; + v->valid = 1; + v->no_cacheable = 0; + v->not_found = 0; + v->data = (u_char *) (s->connection->type == SOCK_DGRAM ? "UDP" : "TCP"); + + return NGX_OK; +} + + void * ngx_stream_map_find(ngx_stream_session_t *s, ngx_stream_map_t *map, ngx_str_t *match)