Mercurial > hg > nginx
annotate src/http/modules/ngx_http_degradation_module.c @ 8518:f9b1a4e52597 quic
QUIC: changed c->quic->pto_count type to ngx_uint_t.
This field is served as a simple counter for PTO backoff.
author | Sergey Kandaurov <pluknet@nginx.com> |
---|---|
date | Wed, 19 Aug 2020 15:58:03 +0300 |
parents | 67653855682e |
children |
rev | line source |
---|---|
3326 | 1 |
2 /* | |
3 * Copyright (C) Igor Sysoev | |
4412 | 4 * Copyright (C) Nginx, Inc. |
3326 | 5 */ |
6 | |
7 | |
8 #include <ngx_config.h> | |
9 #include <ngx_core.h> | |
10 #include <ngx_http.h> | |
11 | |
12 | |
13 typedef struct { | |
14 size_t sbrk_size; | |
15 } ngx_http_degradation_main_conf_t; | |
16 | |
17 | |
18 typedef struct { | |
19 ngx_uint_t degrade; | |
20 } ngx_http_degradation_loc_conf_t; | |
21 | |
22 | |
23 static ngx_conf_enum_t ngx_http_degrade[] = { | |
24 { ngx_string("204"), 204 }, | |
25 { ngx_string("444"), 444 }, | |
26 { ngx_null_string, 0 } | |
27 }; | |
28 | |
29 | |
30 static void *ngx_http_degradation_create_main_conf(ngx_conf_t *cf); | |
31 static void *ngx_http_degradation_create_loc_conf(ngx_conf_t *cf); | |
32 static char *ngx_http_degradation_merge_loc_conf(ngx_conf_t *cf, void *parent, | |
33 void *child); | |
34 static char *ngx_http_degradation(ngx_conf_t *cf, ngx_command_t *cmd, | |
35 void *conf); | |
36 static ngx_int_t ngx_http_degradation_init(ngx_conf_t *cf); | |
37 | |
38 | |
39 static ngx_command_t ngx_http_degradation_commands[] = { | |
40 | |
41 { ngx_string("degradation"), | |
42 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE1, | |
43 ngx_http_degradation, | |
44 NGX_HTTP_MAIN_CONF_OFFSET, | |
45 0, | |
46 NULL }, | |
47 | |
48 { ngx_string("degrade"), | |
49 NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1, | |
50 ngx_conf_set_enum_slot, | |
51 NGX_HTTP_LOC_CONF_OFFSET, | |
52 offsetof(ngx_http_degradation_loc_conf_t, degrade), | |
53 &ngx_http_degrade }, | |
54 | |
55 ngx_null_command | |
56 }; | |
57 | |
58 | |
59 static ngx_http_module_t ngx_http_degradation_module_ctx = { | |
60 NULL, /* preconfiguration */ | |
61 ngx_http_degradation_init, /* postconfiguration */ | |
62 | |
63 ngx_http_degradation_create_main_conf, /* create main configuration */ | |
64 NULL, /* init main configuration */ | |
65 | |
66 NULL, /* create server configuration */ | |
67 NULL, /* merge server configuration */ | |
68 | |
69 ngx_http_degradation_create_loc_conf, /* create location configuration */ | |
70 ngx_http_degradation_merge_loc_conf /* merge location configuration */ | |
71 }; | |
72 | |
73 | |
74 ngx_module_t ngx_http_degradation_module = { | |
75 NGX_MODULE_V1, | |
76 &ngx_http_degradation_module_ctx, /* module context */ | |
77 ngx_http_degradation_commands, /* module directives */ | |
78 NGX_HTTP_MODULE, /* module type */ | |
79 NULL, /* init master */ | |
80 NULL, /* init module */ | |
81 NULL, /* init process */ | |
82 NULL, /* init thread */ | |
83 NULL, /* exit thread */ | |
84 NULL, /* exit process */ | |
85 NULL, /* exit master */ | |
86 NGX_MODULE_V1_PADDING | |
87 }; | |
88 | |
89 | |
90 static ngx_int_t | |
91 ngx_http_degradation_handler(ngx_http_request_t *r) | |
92 { | |
3772 | 93 ngx_http_degradation_loc_conf_t *dlcf; |
3326 | 94 |
95 dlcf = ngx_http_get_module_loc_conf(r, ngx_http_degradation_module); | |
96 | |
3772 | 97 if (dlcf->degrade && ngx_http_degraded(r)) { |
98 return dlcf->degrade; | |
3326 | 99 } |
100 | |
3772 | 101 return NGX_DECLINED; |
102 } | |
103 | |
104 | |
105 ngx_uint_t | |
106 ngx_http_degraded(ngx_http_request_t *r) | |
107 { | |
108 time_t now; | |
109 ngx_uint_t log; | |
110 static size_t sbrk_size; | |
111 static time_t sbrk_time; | |
112 ngx_http_degradation_main_conf_t *dmcf; | |
113 | |
3326 | 114 dmcf = ngx_http_get_module_main_conf(r, ngx_http_degradation_module); |
115 | |
116 if (dmcf->sbrk_size) { | |
117 | |
3772 | 118 log = 0; |
3326 | 119 now = ngx_time(); |
120 | |
121 /* lock mutex */ | |
122 | |
123 if (now != sbrk_time) { | |
124 | |
125 /* | |
126 * ELF/i386 is loaded at 0x08000000, 128M | |
127 * ELF/amd64 is loaded at 0x00400000, 4M | |
128 * | |
4572
67653855682e
Fixed spelling in multiline C comments.
Ruslan Ermilov <ru@nginx.com>
parents:
4412
diff
changeset
|
129 * use a function address to subtract the loading address |
3326 | 130 */ |
131 | |
132 sbrk_size = (size_t) sbrk(0) - ((uintptr_t) ngx_palloc & ~0x3FFFFF); | |
133 sbrk_time = now; | |
3772 | 134 log = 1; |
3326 | 135 } |
136 | |
137 /* unlock mutex */ | |
138 | |
139 if (sbrk_size >= dmcf->sbrk_size) { | |
3772 | 140 if (log) { |
141 ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0, | |
142 "degradation sbrk:%uzM", | |
143 sbrk_size / (1024 * 1024)); | |
144 } | |
145 | |
146 return 1; | |
3326 | 147 } |
148 } | |
149 | |
3772 | 150 return 0; |
3326 | 151 } |
152 | |
153 | |
154 static void * | |
155 ngx_http_degradation_create_main_conf(ngx_conf_t *cf) | |
156 { | |
157 ngx_http_degradation_main_conf_t *dmcf; | |
158 | |
159 dmcf = ngx_pcalloc(cf->pool, sizeof(ngx_http_degradation_main_conf_t)); | |
160 if (dmcf == NULL) { | |
161 return NULL; | |
162 } | |
163 | |
164 return dmcf; | |
165 } | |
166 | |
167 | |
168 static void * | |
169 ngx_http_degradation_create_loc_conf(ngx_conf_t *cf) | |
170 { | |
171 ngx_http_degradation_loc_conf_t *conf; | |
172 | |
173 conf = ngx_palloc(cf->pool, sizeof(ngx_http_degradation_loc_conf_t)); | |
174 if (conf == NULL) { | |
175 return NULL; | |
176 } | |
177 | |
178 conf->degrade = NGX_CONF_UNSET_UINT; | |
179 | |
180 return conf; | |
181 } | |
182 | |
183 | |
184 static char * | |
185 ngx_http_degradation_merge_loc_conf(ngx_conf_t *cf, void *parent, void *child) | |
186 { | |
187 ngx_http_degradation_loc_conf_t *prev = parent; | |
188 ngx_http_degradation_loc_conf_t *conf = child; | |
189 | |
190 ngx_conf_merge_uint_value(conf->degrade, prev->degrade, 0); | |
191 | |
192 return NGX_CONF_OK; | |
193 } | |
194 | |
195 | |
196 static char * | |
197 ngx_http_degradation(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
198 { | |
199 ngx_http_degradation_main_conf_t *dmcf = conf; | |
200 | |
201 ngx_str_t *value, s; | |
202 | |
203 value = cf->args->elts; | |
204 | |
205 if (ngx_strncmp(value[1].data, "sbrk=", 5) == 0) { | |
206 | |
207 s.len = value[1].len - 5; | |
208 s.data = value[1].data + 5; | |
209 | |
210 dmcf->sbrk_size = ngx_parse_size(&s); | |
211 if (dmcf->sbrk_size == (size_t) NGX_ERROR) { | |
212 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
213 "invalid sbrk size \"%V\"", &value[1]); | |
214 return NGX_CONF_ERROR; | |
215 } | |
3389 | 216 |
217 return NGX_CONF_OK; | |
3326 | 218 } |
219 | |
3389 | 220 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
221 "invalid parameter \"%V\"", &value[1]); | |
222 | |
223 return NGX_CONF_ERROR; | |
3326 | 224 } |
225 | |
226 | |
227 static ngx_int_t | |
228 ngx_http_degradation_init(ngx_conf_t *cf) | |
229 { | |
230 ngx_http_handler_pt *h; | |
231 ngx_http_core_main_conf_t *cmcf; | |
232 | |
233 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module); | |
234 | |
235 h = ngx_array_push(&cmcf->phases[NGX_HTTP_PREACCESS_PHASE].handlers); | |
236 if (h == NULL) { | |
237 return NGX_ERROR; | |
238 } | |
239 | |
240 *h = ngx_http_degradation_handler; | |
241 | |
242 return NGX_OK; | |
243 } |