Mercurial > hg > nginx-quic
annotate src/stream/ngx_stream_variables.c @ 6640:e78eca6bfaf0
HTTP/2: fixed send timer handling.
Checking for return value of c->send_chain() isn't sufficient since there
are data can be left in the SSL buffer. Now the wew->ready flag is used
instead.
In particular, this fixed a connection leak in cases when all streams were
closed, but there's still some data to be sent in the SSL buffer and the
client forgot about the connection.
| author | Valentin Bartenev <vbart@nginx.com> |
|---|---|
| date | Tue, 19 Jul 2016 20:31:09 +0300 |
| parents | eb4293155e87 |
| children | 5e2821c2de46 |
| rev | line source |
|---|---|
| 6607 | 1 |
| 2 /* | |
| 3 * Copyright (C) Igor Sysoev | |
| 4 * Copyright (C) Nginx, Inc. | |
| 5 */ | |
| 6 | |
| 7 | |
| 8 #include <ngx_config.h> | |
| 9 #include <ngx_core.h> | |
| 10 #include <ngx_stream.h> | |
| 11 #include <nginx.h> | |
| 12 | |
| 13 | |
|
6608
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
14 static ngx_int_t ngx_stream_variable_binary_remote_addr( |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
15 ngx_stream_session_t *s, ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
16 static ngx_int_t ngx_stream_variable_remote_addr(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
17 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
18 static ngx_int_t ngx_stream_variable_remote_port(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
19 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
20 static ngx_int_t ngx_stream_variable_server_addr(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
21 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
22 static ngx_int_t ngx_stream_variable_server_port(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
23 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
24 static ngx_int_t ngx_stream_variable_bytes_sent(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
25 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
26 static ngx_int_t ngx_stream_variable_connection(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
27 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
28 |
| 6607 | 29 static ngx_int_t ngx_stream_variable_nginx_version(ngx_stream_session_t *s, |
| 30 ngx_stream_variable_value_t *v, uintptr_t data); | |
|
6608
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
31 static ngx_int_t ngx_stream_variable_hostname(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
32 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
33 static ngx_int_t ngx_stream_variable_pid(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
34 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
35 static ngx_int_t ngx_stream_variable_msec(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
36 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
37 static ngx_int_t ngx_stream_variable_time_iso8601(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
38 ngx_stream_variable_value_t *v, uintptr_t data); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
39 static ngx_int_t ngx_stream_variable_time_local(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
40 ngx_stream_variable_value_t *v, uintptr_t data); |
| 6607 | 41 |
| 42 | |
| 43 static ngx_stream_variable_t ngx_stream_core_variables[] = { | |
| 44 | |
|
6608
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
45 { ngx_string("binary_remote_addr"), NULL, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
46 ngx_stream_variable_binary_remote_addr, 0, 0, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
47 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
48 { ngx_string("remote_addr"), NULL, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
49 ngx_stream_variable_remote_addr, 0, 0, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
50 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
51 { ngx_string("remote_port"), NULL, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
52 ngx_stream_variable_remote_port, 0, 0, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
53 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
54 { ngx_string("server_addr"), NULL, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
55 ngx_stream_variable_server_addr, 0, 0, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
56 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
57 { ngx_string("server_port"), NULL, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
58 ngx_stream_variable_server_port, 0, 0, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
59 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
60 { ngx_string("bytes_sent"), NULL, ngx_stream_variable_bytes_sent, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
61 0, 0, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
62 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
63 { ngx_string("connection"), NULL, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
64 ngx_stream_variable_connection, 0, 0, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
65 |
| 6607 | 66 { ngx_string("nginx_version"), NULL, ngx_stream_variable_nginx_version, |
| 67 0, 0, 0 }, | |
| 68 | |
|
6608
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
69 { ngx_string("hostname"), NULL, ngx_stream_variable_hostname, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
70 0, 0, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
71 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
72 { ngx_string("pid"), NULL, ngx_stream_variable_pid, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
73 0, 0, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
74 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
75 { ngx_string("msec"), NULL, ngx_stream_variable_msec, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
76 0, NGX_STREAM_VAR_NOCACHEABLE, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
77 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
78 { ngx_string("time_iso8601"), NULL, ngx_stream_variable_time_iso8601, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
79 0, NGX_STREAM_VAR_NOCACHEABLE, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
80 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
81 { ngx_string("time_local"), NULL, ngx_stream_variable_time_local, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
82 0, NGX_STREAM_VAR_NOCACHEABLE, 0 }, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
83 |
| 6607 | 84 { ngx_null_string, NULL, NULL, 0, 0, 0 } |
| 85 }; | |
| 86 | |
| 87 | |
| 88 ngx_stream_variable_value_t ngx_stream_variable_null_value = | |
| 89 ngx_stream_variable(""); | |
| 90 ngx_stream_variable_value_t ngx_stream_variable_true_value = | |
| 91 ngx_stream_variable("1"); | |
| 92 | |
| 93 | |
| 94 ngx_stream_variable_t * | |
| 95 ngx_stream_add_variable(ngx_conf_t *cf, ngx_str_t *name, ngx_uint_t flags) | |
| 96 { | |
| 97 ngx_int_t rc; | |
| 98 ngx_uint_t i; | |
| 99 ngx_hash_key_t *key; | |
| 100 ngx_stream_variable_t *v; | |
| 101 ngx_stream_core_main_conf_t *cmcf; | |
| 102 | |
| 103 if (name->len == 0) { | |
| 104 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 105 "invalid variable name \"$\""); | |
| 106 return NULL; | |
| 107 } | |
| 108 | |
| 109 cmcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_core_module); | |
| 110 | |
| 111 key = cmcf->variables_keys->keys.elts; | |
| 112 for (i = 0; i < cmcf->variables_keys->keys.nelts; i++) { | |
| 113 if (name->len != key[i].key.len | |
| 114 || ngx_strncasecmp(name->data, key[i].key.data, name->len) != 0) | |
| 115 { | |
| 116 continue; | |
| 117 } | |
| 118 | |
| 119 v = key[i].value; | |
| 120 | |
| 121 if (!(v->flags & NGX_STREAM_VAR_CHANGEABLE)) { | |
| 122 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 123 "the duplicate \"%V\" variable", name); | |
| 124 return NULL; | |
| 125 } | |
| 126 | |
| 127 return v; | |
| 128 } | |
| 129 | |
| 130 v = ngx_palloc(cf->pool, sizeof(ngx_stream_variable_t)); | |
| 131 if (v == NULL) { | |
| 132 return NULL; | |
| 133 } | |
| 134 | |
| 135 v->name.len = name->len; | |
| 136 v->name.data = ngx_pnalloc(cf->pool, name->len); | |
| 137 if (v->name.data == NULL) { | |
| 138 return NULL; | |
| 139 } | |
| 140 | |
| 141 ngx_strlow(v->name.data, name->data, name->len); | |
| 142 | |
| 143 v->set_handler = NULL; | |
| 144 v->get_handler = NULL; | |
| 145 v->data = 0; | |
| 146 v->flags = flags; | |
| 147 v->index = 0; | |
| 148 | |
| 149 rc = ngx_hash_add_key(cmcf->variables_keys, &v->name, v, 0); | |
| 150 | |
| 151 if (rc == NGX_ERROR) { | |
| 152 return NULL; | |
| 153 } | |
| 154 | |
| 155 if (rc == NGX_BUSY) { | |
| 156 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 157 "conflicting variable name \"%V\"", name); | |
| 158 return NULL; | |
| 159 } | |
| 160 | |
| 161 return v; | |
| 162 } | |
| 163 | |
| 164 | |
| 165 ngx_int_t | |
| 166 ngx_stream_get_variable_index(ngx_conf_t *cf, ngx_str_t *name) | |
| 167 { | |
| 168 ngx_uint_t i; | |
| 169 ngx_stream_variable_t *v; | |
| 170 ngx_stream_core_main_conf_t *cmcf; | |
| 171 | |
| 172 if (name->len == 0) { | |
| 173 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 174 "invalid variable name \"$\""); | |
| 175 return NGX_ERROR; | |
| 176 } | |
| 177 | |
| 178 cmcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_core_module); | |
| 179 | |
| 180 v = cmcf->variables.elts; | |
| 181 | |
| 182 if (v == NULL) { | |
| 183 if (ngx_array_init(&cmcf->variables, cf->pool, 4, | |
| 184 sizeof(ngx_stream_variable_t)) | |
| 185 != NGX_OK) | |
| 186 { | |
| 187 return NGX_ERROR; | |
| 188 } | |
| 189 | |
| 190 } else { | |
| 191 for (i = 0; i < cmcf->variables.nelts; i++) { | |
| 192 if (name->len != v[i].name.len | |
| 193 || ngx_strncasecmp(name->data, v[i].name.data, name->len) != 0) | |
| 194 { | |
| 195 continue; | |
| 196 } | |
| 197 | |
| 198 return i; | |
| 199 } | |
| 200 } | |
| 201 | |
| 202 v = ngx_array_push(&cmcf->variables); | |
| 203 if (v == NULL) { | |
| 204 return NGX_ERROR; | |
| 205 } | |
| 206 | |
| 207 v->name.len = name->len; | |
| 208 v->name.data = ngx_pnalloc(cf->pool, name->len); | |
| 209 if (v->name.data == NULL) { | |
| 210 return NGX_ERROR; | |
| 211 } | |
| 212 | |
| 213 ngx_strlow(v->name.data, name->data, name->len); | |
| 214 | |
| 215 v->set_handler = NULL; | |
| 216 v->get_handler = NULL; | |
| 217 v->data = 0; | |
| 218 v->flags = 0; | |
| 219 v->index = cmcf->variables.nelts - 1; | |
| 220 | |
| 221 return v->index; | |
| 222 } | |
| 223 | |
| 224 | |
| 225 ngx_stream_variable_value_t * | |
| 226 ngx_stream_get_indexed_variable(ngx_stream_session_t *s, ngx_uint_t index) | |
| 227 { | |
| 228 ngx_stream_variable_t *v; | |
| 229 ngx_stream_core_main_conf_t *cmcf; | |
| 230 | |
| 231 cmcf = ngx_stream_get_module_main_conf(s, ngx_stream_core_module); | |
| 232 | |
| 233 if (cmcf->variables.nelts <= index) { | |
| 234 ngx_log_error(NGX_LOG_ALERT, s->connection->log, 0, | |
| 235 "unknown variable index: %ui", index); | |
| 236 return NULL; | |
| 237 } | |
| 238 | |
| 239 if (s->variables[index].not_found || s->variables[index].valid) { | |
| 240 return &s->variables[index]; | |
| 241 } | |
| 242 | |
| 243 v = cmcf->variables.elts; | |
| 244 | |
| 245 if (v[index].get_handler(s, &s->variables[index], v[index].data) | |
| 246 == NGX_OK) | |
| 247 { | |
| 248 if (v[index].flags & NGX_STREAM_VAR_NOCACHEABLE) { | |
| 249 s->variables[index].no_cacheable = 1; | |
| 250 } | |
| 251 | |
| 252 return &s->variables[index]; | |
| 253 } | |
| 254 | |
| 255 s->variables[index].valid = 0; | |
| 256 s->variables[index].not_found = 1; | |
| 257 | |
| 258 return NULL; | |
| 259 } | |
| 260 | |
| 261 | |
| 262 ngx_stream_variable_value_t * | |
| 263 ngx_stream_get_flushed_variable(ngx_stream_session_t *s, ngx_uint_t index) | |
| 264 { | |
| 265 ngx_stream_variable_value_t *v; | |
| 266 | |
| 267 v = &s->variables[index]; | |
| 268 | |
| 269 if (v->valid || v->not_found) { | |
| 270 if (!v->no_cacheable) { | |
| 271 return v; | |
| 272 } | |
| 273 | |
| 274 v->valid = 0; | |
| 275 v->not_found = 0; | |
| 276 } | |
| 277 | |
| 278 return ngx_stream_get_indexed_variable(s, index); | |
| 279 } | |
| 280 | |
| 281 | |
| 282 ngx_stream_variable_value_t * | |
| 283 ngx_stream_get_variable(ngx_stream_session_t *s, ngx_str_t *name, | |
| 284 ngx_uint_t key) | |
| 285 { | |
| 286 ngx_stream_variable_t *v; | |
| 287 ngx_stream_variable_value_t *vv; | |
| 288 ngx_stream_core_main_conf_t *cmcf; | |
| 289 | |
| 290 cmcf = ngx_stream_get_module_main_conf(s, ngx_stream_core_module); | |
| 291 | |
| 292 v = ngx_hash_find(&cmcf->variables_hash, key, name->data, name->len); | |
| 293 | |
| 294 if (v) { | |
| 295 if (v->flags & NGX_STREAM_VAR_INDEXED) { | |
| 296 return ngx_stream_get_flushed_variable(s, v->index); | |
| 297 | |
| 298 } else { | |
| 299 | |
| 300 vv = ngx_palloc(s->connection->pool, | |
| 301 sizeof(ngx_stream_variable_value_t)); | |
| 302 | |
| 303 if (vv && v->get_handler(s, vv, v->data) == NGX_OK) { | |
| 304 return vv; | |
| 305 } | |
| 306 | |
| 307 return NULL; | |
| 308 } | |
| 309 } | |
| 310 | |
| 311 vv = ngx_palloc(s->connection->pool, sizeof(ngx_stream_variable_value_t)); | |
| 312 if (vv == NULL) { | |
| 313 return NULL; | |
| 314 } | |
| 315 | |
| 316 vv->not_found = 1; | |
| 317 | |
| 318 return vv; | |
| 319 } | |
| 320 | |
| 321 | |
| 322 static ngx_int_t | |
|
6608
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
323 ngx_stream_variable_binary_remote_addr(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
324 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
325 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
326 struct sockaddr_in *sin; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
327 #if (NGX_HAVE_INET6) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
328 struct sockaddr_in6 *sin6; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
329 #endif |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
330 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
331 switch (s->connection->sockaddr->sa_family) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
332 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
333 #if (NGX_HAVE_INET6) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
334 case AF_INET6: |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
335 sin6 = (struct sockaddr_in6 *) s->connection->sockaddr; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
336 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
337 v->len = sizeof(struct in6_addr); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
338 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
339 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
340 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
341 v->data = sin6->sin6_addr.s6_addr; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
342 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
343 break; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
344 #endif |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
345 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
346 default: /* AF_INET */ |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
347 sin = (struct sockaddr_in *) s->connection->sockaddr; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
348 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
349 v->len = sizeof(in_addr_t); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
350 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
351 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
352 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
353 v->data = (u_char *) &sin->sin_addr; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
354 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
355 break; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
356 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
357 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
358 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
359 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
360 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
361 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
362 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
363 ngx_stream_variable_remote_addr(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
364 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
365 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
366 v->len = s->connection->addr_text.len; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
367 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
368 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
369 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
370 v->data = s->connection->addr_text.data; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
371 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
372 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
373 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
374 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
375 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
376 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
377 ngx_stream_variable_remote_port(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
378 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
379 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
380 ngx_uint_t port; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
381 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
382 v->len = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
383 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
384 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
385 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
386 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
387 v->data = ngx_pnalloc(s->connection->pool, sizeof("65535") - 1); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
388 if (v->data == NULL) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
389 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
390 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
391 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
392 port = ngx_inet_get_port(s->connection->sockaddr); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
393 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
394 if (port > 0 && port < 65536) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
395 v->len = ngx_sprintf(v->data, "%ui", port) - v->data; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
396 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
397 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
398 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
399 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
400 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
401 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
402 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
403 ngx_stream_variable_server_addr(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
404 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
405 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
406 ngx_str_t str; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
407 u_char addr[NGX_SOCKADDR_STRLEN]; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
408 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
409 str.len = NGX_SOCKADDR_STRLEN; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
410 str.data = addr; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
411 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
412 if (ngx_connection_local_sockaddr(s->connection, &str, 0) != NGX_OK) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
413 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
414 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
415 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
416 str.data = ngx_pnalloc(s->connection->pool, str.len); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
417 if (str.data == NULL) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
418 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
419 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
420 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
421 ngx_memcpy(str.data, addr, str.len); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
422 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
423 v->len = str.len; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
424 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
425 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
426 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
427 v->data = str.data; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
428 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
429 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
430 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
431 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
432 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
433 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
434 ngx_stream_variable_server_port(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
435 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
436 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
437 ngx_uint_t port; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
438 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
439 v->len = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
440 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
441 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
442 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
443 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
444 if (ngx_connection_local_sockaddr(s->connection, NULL, 0) != NGX_OK) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
445 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
446 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
447 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
448 v->data = ngx_pnalloc(s->connection->pool, sizeof("65535") - 1); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
449 if (v->data == NULL) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
450 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
451 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
452 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
453 port = ngx_inet_get_port(s->connection->local_sockaddr); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
454 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
455 if (port > 0 && port < 65536) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
456 v->len = ngx_sprintf(v->data, "%ui", port) - v->data; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
457 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
458 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
459 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
460 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
461 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
462 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
463 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
464 ngx_stream_variable_bytes_sent(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
465 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
466 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
467 u_char *p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
468 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
469 p = ngx_pnalloc(s->connection->pool, NGX_OFF_T_LEN); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
470 if (p == NULL) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
471 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
472 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
473 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
474 v->len = ngx_sprintf(p, "%O", s->connection->sent) - p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
475 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
476 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
477 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
478 v->data = p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
479 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
480 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
481 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
482 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
483 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
484 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
485 ngx_stream_variable_connection(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
486 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
487 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
488 u_char *p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
489 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
490 p = ngx_pnalloc(s->connection->pool, NGX_ATOMIC_T_LEN); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
491 if (p == NULL) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
492 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
493 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
494 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
495 v->len = ngx_sprintf(p, "%uA", s->connection->number) - p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
496 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
497 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
498 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
499 v->data = p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
500 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
501 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
502 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
503 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
504 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
505 static ngx_int_t |
| 6607 | 506 ngx_stream_variable_nginx_version(ngx_stream_session_t *s, |
| 507 ngx_stream_variable_value_t *v, uintptr_t data) | |
| 508 { | |
| 509 v->len = sizeof(NGINX_VERSION) - 1; | |
| 510 v->valid = 1; | |
| 511 v->no_cacheable = 0; | |
| 512 v->not_found = 0; | |
| 513 v->data = (u_char *) NGINX_VERSION; | |
| 514 | |
| 515 return NGX_OK; | |
| 516 } | |
| 517 | |
| 518 | |
|
6608
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
519 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
520 ngx_stream_variable_hostname(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
521 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
522 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
523 v->len = ngx_cycle->hostname.len; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
524 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
525 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
526 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
527 v->data = ngx_cycle->hostname.data; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
528 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
529 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
530 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
531 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
532 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
533 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
534 ngx_stream_variable_pid(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
535 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
536 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
537 u_char *p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
538 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
539 p = ngx_pnalloc(s->connection->pool, NGX_INT64_LEN); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
540 if (p == NULL) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
541 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
542 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
543 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
544 v->len = ngx_sprintf(p, "%P", ngx_pid) - p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
545 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
546 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
547 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
548 v->data = p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
549 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
550 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
551 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
552 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
553 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
554 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
555 ngx_stream_variable_msec(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
556 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
557 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
558 u_char *p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
559 ngx_time_t *tp; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
560 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
561 p = ngx_pnalloc(s->connection->pool, NGX_TIME_T_LEN + 4); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
562 if (p == NULL) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
563 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
564 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
565 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
566 tp = ngx_timeofday(); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
567 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
568 v->len = ngx_sprintf(p, "%T.%03M", tp->sec, tp->msec) - p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
569 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
570 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
571 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
572 v->data = p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
573 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
574 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
575 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
576 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
577 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
578 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
579 ngx_stream_variable_time_iso8601(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
580 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
581 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
582 u_char *p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
583 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
584 p = ngx_pnalloc(s->connection->pool, ngx_cached_http_log_iso8601.len); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
585 if (p == NULL) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
586 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
587 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
588 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
589 ngx_memcpy(p, ngx_cached_http_log_iso8601.data, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
590 ngx_cached_http_log_iso8601.len); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
591 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
592 v->len = ngx_cached_http_log_iso8601.len; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
593 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
594 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
595 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
596 v->data = p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
597 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
598 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
599 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
600 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
601 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
602 static ngx_int_t |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
603 ngx_stream_variable_time_local(ngx_stream_session_t *s, |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
604 ngx_stream_variable_value_t *v, uintptr_t data) |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
605 { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
606 u_char *p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
607 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
608 p = ngx_pnalloc(s->connection->pool, ngx_cached_http_log_time.len); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
609 if (p == NULL) { |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
610 return NGX_ERROR; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
611 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
612 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
613 ngx_memcpy(p, ngx_cached_http_log_time.data, ngx_cached_http_log_time.len); |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
614 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
615 v->len = ngx_cached_http_log_time.len; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
616 v->valid = 1; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
617 v->no_cacheable = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
618 v->not_found = 0; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
619 v->data = p; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
620 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
621 return NGX_OK; |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
622 } |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
623 |
|
eb4293155e87
Stream: core module variables.
Vladimir Homutov <vl@nginx.com>
parents:
6607
diff
changeset
|
624 |
| 6607 | 625 void * |
| 626 ngx_stream_map_find(ngx_stream_session_t *s, ngx_stream_map_t *map, | |
| 627 ngx_str_t *match) | |
| 628 { | |
| 629 void *value; | |
| 630 u_char *low; | |
| 631 size_t len; | |
| 632 ngx_uint_t key; | |
| 633 | |
| 634 len = match->len; | |
| 635 | |
| 636 if (len) { | |
| 637 low = ngx_pnalloc(s->connection->pool, len); | |
| 638 if (low == NULL) { | |
| 639 return NULL; | |
| 640 } | |
| 641 | |
| 642 } else { | |
| 643 low = NULL; | |
| 644 } | |
| 645 | |
| 646 key = ngx_hash_strlow(low, match->data, len); | |
| 647 | |
| 648 value = ngx_hash_find_combined(&map->hash, key, low, len); | |
| 649 if (value) { | |
| 650 return value; | |
| 651 } | |
| 652 | |
| 653 #if (NGX_PCRE) | |
| 654 | |
| 655 if (len && map->nregex) { | |
| 656 ngx_int_t n; | |
| 657 ngx_uint_t i; | |
| 658 ngx_stream_map_regex_t *reg; | |
| 659 | |
| 660 reg = map->regex; | |
| 661 | |
| 662 for (i = 0; i < map->nregex; i++) { | |
| 663 | |
| 664 n = ngx_stream_regex_exec(s, reg[i].regex, match); | |
| 665 | |
| 666 if (n == NGX_OK) { | |
| 667 return reg[i].value; | |
| 668 } | |
| 669 | |
| 670 if (n == NGX_DECLINED) { | |
| 671 continue; | |
| 672 } | |
| 673 | |
| 674 /* NGX_ERROR */ | |
| 675 | |
| 676 return NULL; | |
| 677 } | |
| 678 } | |
| 679 | |
| 680 #endif | |
| 681 | |
| 682 return NULL; | |
| 683 } | |
| 684 | |
| 685 | |
| 686 #if (NGX_PCRE) | |
| 687 | |
| 688 static ngx_int_t | |
| 689 ngx_stream_variable_not_found(ngx_stream_session_t *s, | |
| 690 ngx_stream_variable_value_t *v, uintptr_t data) | |
| 691 { | |
| 692 v->not_found = 1; | |
| 693 return NGX_OK; | |
| 694 } | |
| 695 | |
| 696 | |
| 697 ngx_stream_regex_t * | |
| 698 ngx_stream_regex_compile(ngx_conf_t *cf, ngx_regex_compile_t *rc) | |
| 699 { | |
| 700 u_char *p; | |
| 701 size_t size; | |
| 702 ngx_str_t name; | |
| 703 ngx_uint_t i, n; | |
| 704 ngx_stream_variable_t *v; | |
| 705 ngx_stream_regex_t *re; | |
| 706 ngx_stream_regex_variable_t *rv; | |
| 707 ngx_stream_core_main_conf_t *cmcf; | |
| 708 | |
| 709 rc->pool = cf->pool; | |
| 710 | |
| 711 if (ngx_regex_compile(rc) != NGX_OK) { | |
| 712 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "%V", &rc->err); | |
| 713 return NULL; | |
| 714 } | |
| 715 | |
| 716 re = ngx_pcalloc(cf->pool, sizeof(ngx_stream_regex_t)); | |
| 717 if (re == NULL) { | |
| 718 return NULL; | |
| 719 } | |
| 720 | |
| 721 re->regex = rc->regex; | |
| 722 re->ncaptures = rc->captures; | |
| 723 re->name = rc->pattern; | |
| 724 | |
| 725 cmcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_core_module); | |
| 726 cmcf->ncaptures = ngx_max(cmcf->ncaptures, re->ncaptures); | |
| 727 | |
| 728 n = (ngx_uint_t) rc->named_captures; | |
| 729 | |
| 730 if (n == 0) { | |
| 731 return re; | |
| 732 } | |
| 733 | |
| 734 rv = ngx_palloc(rc->pool, n * sizeof(ngx_stream_regex_variable_t)); | |
| 735 if (rv == NULL) { | |
| 736 return NULL; | |
| 737 } | |
| 738 | |
| 739 re->variables = rv; | |
| 740 re->nvariables = n; | |
| 741 | |
| 742 size = rc->name_size; | |
| 743 p = rc->names; | |
| 744 | |
| 745 for (i = 0; i < n; i++) { | |
| 746 rv[i].capture = 2 * ((p[0] << 8) + p[1]); | |
| 747 | |
| 748 name.data = &p[2]; | |
| 749 name.len = ngx_strlen(name.data); | |
| 750 | |
| 751 v = ngx_stream_add_variable(cf, &name, NGX_STREAM_VAR_CHANGEABLE); | |
| 752 if (v == NULL) { | |
| 753 return NULL; | |
| 754 } | |
| 755 | |
| 756 rv[i].index = ngx_stream_get_variable_index(cf, &name); | |
| 757 if (rv[i].index == NGX_ERROR) { | |
| 758 return NULL; | |
| 759 } | |
| 760 | |
| 761 v->get_handler = ngx_stream_variable_not_found; | |
| 762 | |
| 763 p += size; | |
| 764 } | |
| 765 | |
| 766 return re; | |
| 767 } | |
| 768 | |
| 769 | |
| 770 ngx_int_t | |
| 771 ngx_stream_regex_exec(ngx_stream_session_t *s, ngx_stream_regex_t *re, | |
| 772 ngx_str_t *str) | |
| 773 { | |
| 774 ngx_int_t rc, index; | |
| 775 ngx_uint_t i, n, len; | |
| 776 ngx_stream_variable_value_t *vv; | |
| 777 ngx_stream_core_main_conf_t *cmcf; | |
| 778 | |
| 779 cmcf = ngx_stream_get_module_main_conf(s, ngx_stream_core_module); | |
| 780 | |
| 781 if (re->ncaptures) { | |
| 782 len = cmcf->ncaptures; | |
| 783 | |
| 784 if (s->captures == NULL) { | |
| 785 s->captures = ngx_palloc(s->connection->pool, len * sizeof(int)); | |
| 786 if (s->captures == NULL) { | |
| 787 return NGX_ERROR; | |
| 788 } | |
| 789 } | |
| 790 | |
| 791 } else { | |
| 792 len = 0; | |
| 793 } | |
| 794 | |
| 795 rc = ngx_regex_exec(re->regex, str, s->captures, len); | |
| 796 | |
| 797 if (rc == NGX_REGEX_NO_MATCHED) { | |
| 798 return NGX_DECLINED; | |
| 799 } | |
| 800 | |
| 801 if (rc < 0) { | |
| 802 ngx_log_error(NGX_LOG_ALERT, s->connection->log, 0, | |
| 803 ngx_regex_exec_n " failed: %i on \"%V\" using \"%V\"", | |
| 804 rc, str, &re->name); | |
| 805 return NGX_ERROR; | |
| 806 } | |
| 807 | |
| 808 for (i = 0; i < re->nvariables; i++) { | |
| 809 | |
| 810 n = re->variables[i].capture; | |
| 811 index = re->variables[i].index; | |
| 812 vv = &s->variables[index]; | |
| 813 | |
| 814 vv->len = s->captures[n + 1] - s->captures[n]; | |
| 815 vv->valid = 1; | |
| 816 vv->no_cacheable = 0; | |
| 817 vv->not_found = 0; | |
| 818 vv->data = &str->data[s->captures[n]]; | |
| 819 | |
| 820 #if (NGX_DEBUG) | |
| 821 { | |
| 822 ngx_stream_variable_t *v; | |
| 823 | |
| 824 v = cmcf->variables.elts; | |
| 825 | |
| 826 ngx_log_debug2(NGX_LOG_DEBUG_STREAM, s->connection->log, 0, | |
| 827 "stream regex set $%V to \"%v\"", &v[index].name, vv); | |
| 828 } | |
| 829 #endif | |
| 830 } | |
| 831 | |
| 832 s->ncaptures = rc * 2; | |
| 833 s->captures_data = str->data; | |
| 834 | |
| 835 return NGX_OK; | |
| 836 } | |
| 837 | |
| 838 #endif | |
| 839 | |
| 840 | |
| 841 ngx_int_t | |
| 842 ngx_stream_variables_add_core_vars(ngx_conf_t *cf) | |
| 843 { | |
| 844 ngx_int_t rc; | |
| 845 ngx_stream_variable_t *cv, *v; | |
| 846 ngx_stream_core_main_conf_t *cmcf; | |
| 847 | |
| 848 cmcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_core_module); | |
| 849 | |
| 850 cmcf->variables_keys = ngx_pcalloc(cf->temp_pool, | |
| 851 sizeof(ngx_hash_keys_arrays_t)); | |
| 852 if (cmcf->variables_keys == NULL) { | |
| 853 return NGX_ERROR; | |
| 854 } | |
| 855 | |
| 856 cmcf->variables_keys->pool = cf->pool; | |
| 857 cmcf->variables_keys->temp_pool = cf->pool; | |
| 858 | |
| 859 if (ngx_hash_keys_array_init(cmcf->variables_keys, NGX_HASH_SMALL) | |
| 860 != NGX_OK) | |
| 861 { | |
| 862 return NGX_ERROR; | |
| 863 } | |
| 864 | |
| 865 for (cv = ngx_stream_core_variables; cv->name.len; cv++) { | |
| 866 v = ngx_palloc(cf->pool, sizeof(ngx_stream_variable_t)); | |
| 867 if (v == NULL) { | |
| 868 return NGX_ERROR; | |
| 869 } | |
| 870 | |
| 871 *v = *cv; | |
| 872 | |
| 873 rc = ngx_hash_add_key(cmcf->variables_keys, &v->name, v, | |
| 874 NGX_HASH_READONLY_KEY); | |
| 875 | |
| 876 if (rc == NGX_OK) { | |
| 877 continue; | |
| 878 } | |
| 879 | |
| 880 if (rc == NGX_BUSY) { | |
| 881 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 882 "conflicting variable name \"%V\"", &v->name); | |
| 883 } | |
| 884 | |
| 885 return NGX_ERROR; | |
| 886 } | |
| 887 | |
| 888 return NGX_OK; | |
| 889 } | |
| 890 | |
| 891 | |
| 892 ngx_int_t | |
| 893 ngx_stream_variables_init_vars(ngx_conf_t *cf) | |
| 894 { | |
| 895 ngx_uint_t i, n; | |
| 896 ngx_hash_key_t *key; | |
| 897 ngx_hash_init_t hash; | |
| 898 ngx_stream_variable_t *v, *av; | |
| 899 ngx_stream_core_main_conf_t *cmcf; | |
| 900 | |
| 901 /* set the handlers for the indexed stream variables */ | |
| 902 | |
| 903 cmcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_core_module); | |
| 904 | |
| 905 v = cmcf->variables.elts; | |
| 906 key = cmcf->variables_keys->keys.elts; | |
| 907 | |
| 908 for (i = 0; i < cmcf->variables.nelts; i++) { | |
| 909 | |
| 910 for (n = 0; n < cmcf->variables_keys->keys.nelts; n++) { | |
| 911 | |
| 912 av = key[n].value; | |
| 913 | |
| 914 if (v[i].name.len == key[n].key.len | |
| 915 && ngx_strncmp(v[i].name.data, key[n].key.data, v[i].name.len) | |
| 916 == 0) | |
| 917 { | |
| 918 v[i].get_handler = av->get_handler; | |
| 919 v[i].data = av->data; | |
| 920 | |
| 921 av->flags |= NGX_STREAM_VAR_INDEXED; | |
| 922 v[i].flags = av->flags; | |
| 923 | |
| 924 av->index = i; | |
| 925 | |
| 926 if (av->get_handler == NULL) { | |
| 927 break; | |
| 928 } | |
| 929 | |
| 930 goto next; | |
| 931 } | |
| 932 } | |
| 933 | |
| 934 ngx_log_error(NGX_LOG_EMERG, cf->log, 0, | |
| 935 "unknown \"%V\" variable", &v[i].name); | |
| 936 | |
| 937 return NGX_ERROR; | |
| 938 | |
| 939 next: | |
| 940 continue; | |
| 941 } | |
| 942 | |
| 943 | |
| 944 for (n = 0; n < cmcf->variables_keys->keys.nelts; n++) { | |
| 945 av = key[n].value; | |
| 946 | |
| 947 if (av->flags & NGX_STREAM_VAR_NOHASH) { | |
| 948 key[n].key.data = NULL; | |
| 949 } | |
| 950 } | |
| 951 | |
| 952 | |
| 953 hash.hash = &cmcf->variables_hash; | |
| 954 hash.key = ngx_hash_key; | |
| 955 hash.max_size = cmcf->variables_hash_max_size; | |
| 956 hash.bucket_size = cmcf->variables_hash_bucket_size; | |
| 957 hash.name = "variables_hash"; | |
| 958 hash.pool = cf->pool; | |
| 959 hash.temp_pool = NULL; | |
| 960 | |
| 961 if (ngx_hash_init(&hash, cmcf->variables_keys->keys.elts, | |
| 962 cmcf->variables_keys->keys.nelts) | |
| 963 != NGX_OK) | |
| 964 { | |
| 965 return NGX_ERROR; | |
| 966 } | |
| 967 | |
| 968 cmcf->variables_keys = NULL; | |
| 969 | |
| 970 return NGX_OK; | |
| 971 } |