Mercurial > hg > nginx-ranges
comparison src/core/ngx_connection.c @ 0:f0b350454894 NGINX_0_1_0
nginx 0.1.0
*) The first public version.
author | Igor Sysoev <http://sysoev.ru> |
---|---|
date | Mon, 04 Oct 2004 00:00:00 +0400 |
parents | |
children | cc9f381affaa |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:f0b350454894 |
---|---|
1 | |
2 /* | |
3 * Copyright (C) Igor Sysoev | |
4 */ | |
5 | |
6 | |
7 #include <ngx_config.h> | |
8 #include <ngx_core.h> | |
9 #include <ngx_event.h> | |
10 | |
11 | |
12 ngx_os_io_t ngx_io; | |
13 | |
14 | |
15 ngx_listening_t *ngx_listening_inet_stream_socket(ngx_conf_t *cf, | |
16 in_addr_t addr, | |
17 in_port_t port) | |
18 { | |
19 size_t len; | |
20 ngx_listening_t *ls; | |
21 struct sockaddr_in *addr_in; | |
22 | |
23 if (!(ls = ngx_array_push(&cf->cycle->listening))) { | |
24 return NULL; | |
25 } | |
26 | |
27 ngx_memzero(ls, sizeof(ngx_listening_t)); | |
28 | |
29 if (!(addr_in = ngx_pcalloc(cf->pool, sizeof(struct sockaddr_in)))) { | |
30 return NULL; | |
31 } | |
32 | |
33 #if (HAVE_SIN_LEN) | |
34 addr_in->sin_len = sizeof(struct sockaddr_in); | |
35 #endif | |
36 addr_in->sin_family = AF_INET; | |
37 addr_in->sin_addr.s_addr = addr; | |
38 addr_in->sin_port = htons(port); | |
39 | |
40 if (!(ls->addr_text.data = ngx_palloc(cf->pool, INET_ADDRSTRLEN + 6))) { | |
41 return NULL; | |
42 } | |
43 | |
44 len = ngx_inet_ntop(AF_INET, &addr, ls->addr_text.data, INET_ADDRSTRLEN); | |
45 ls->addr_text.len = ngx_snprintf((char *) ls->addr_text.data + len, | |
46 6, ":%d", port); | |
47 | |
48 ls->fd = (ngx_socket_t) -1; | |
49 ls->family = AF_INET; | |
50 ls->type = SOCK_STREAM; | |
51 ls->protocol = IPPROTO_IP; | |
52 #if (WIN32) | |
53 ls->flags = WSA_FLAG_OVERLAPPED; | |
54 #endif | |
55 ls->sockaddr = (struct sockaddr *) addr_in; | |
56 ls->socklen = sizeof(struct sockaddr_in); | |
57 ls->addr = offsetof(struct sockaddr_in, sin_addr); | |
58 ls->addr_text_max_len = INET_ADDRSTRLEN; | |
59 | |
60 return ls; | |
61 } | |
62 | |
63 | |
64 ngx_int_t ngx_set_inherited_sockets(ngx_cycle_t *cycle) | |
65 { | |
66 ngx_uint_t i; | |
67 ngx_listening_t *ls; | |
68 struct sockaddr_in *addr_in; | |
69 | |
70 ls = cycle->listening.elts; | |
71 for (i = 0; i < cycle->listening.nelts; i++) { | |
72 | |
73 /* AF_INET only */ | |
74 | |
75 ls[i].sockaddr = ngx_palloc(cycle->pool, sizeof(struct sockaddr_in)); | |
76 if (ls[i].sockaddr == NULL) { | |
77 return NGX_ERROR; | |
78 } | |
79 | |
80 ls[i].socklen = sizeof(struct sockaddr_in); | |
81 if (getsockname(ls[i].fd, ls[i].sockaddr, &ls[i].socklen) == -1) { | |
82 ngx_log_error(NGX_LOG_CRIT, cycle->log, ngx_socket_errno, | |
83 "getsockname() of the inherited " | |
84 "socket #%d failed", ls[i].fd); | |
85 ls[i].ignore = 1; | |
86 continue; | |
87 } | |
88 | |
89 addr_in = (struct sockaddr_in *) ls[i].sockaddr; | |
90 | |
91 if (addr_in->sin_family != AF_INET) { | |
92 ngx_log_error(NGX_LOG_CRIT, cycle->log, ngx_socket_errno, | |
93 "the inherited socket #%d has " | |
94 "unsupported family", ls[i].fd); | |
95 ls[i].ignore = 1; | |
96 continue; | |
97 } | |
98 ls[i].addr_text_max_len = INET_ADDRSTRLEN; | |
99 | |
100 ls[i].addr_text.data = ngx_palloc(cycle->pool, ls[i].addr_text_max_len); | |
101 if (ls[i].addr_text.data == NULL) { | |
102 return NGX_ERROR; | |
103 } | |
104 | |
105 ls[i].family = addr_in->sin_family; | |
106 ls[i].addr_text.len = ngx_sock_ntop(ls[i].family, ls[i].sockaddr, | |
107 ls[i].addr_text.data, | |
108 ls[i].addr_text_max_len); | |
109 if (ls[i].addr_text.len == 0) { | |
110 return NGX_ERROR; | |
111 } | |
112 } | |
113 | |
114 return NGX_OK; | |
115 } | |
116 | |
117 | |
118 ngx_int_t ngx_open_listening_sockets(ngx_cycle_t *cycle) | |
119 { | |
120 ngx_uint_t tries, failed, reuseaddr, i; | |
121 ngx_err_t err; | |
122 ngx_log_t *log; | |
123 ngx_socket_t s; | |
124 ngx_listening_t *ls; | |
125 | |
126 reuseaddr = 1; | |
127 #if (NGX_SUPPRESS_WARN) | |
128 failed = 0; | |
129 #endif | |
130 | |
131 log = cycle->log; | |
132 | |
133 /* TODO: tries configurable */ | |
134 | |
135 for (tries = /* STUB */ 5; tries; tries--) { | |
136 failed = 0; | |
137 | |
138 /* for each listening socket */ | |
139 | |
140 ls = cycle->listening.elts; | |
141 for (i = 0; i < cycle->listening.nelts; i++) { | |
142 | |
143 if (ls[i].ignore) { | |
144 continue; | |
145 } | |
146 | |
147 if (ls[i].fd != -1) { | |
148 continue; | |
149 } | |
150 | |
151 if (ls[i].inherited) { | |
152 | |
153 /* TODO: close on exit */ | |
154 /* TODO: nonblocking */ | |
155 /* TODO: deferred accept */ | |
156 | |
157 continue; | |
158 } | |
159 | |
160 s = ngx_socket(ls[i].family, ls[i].type, ls[i].protocol, | |
161 ls[i].flags); | |
162 | |
163 if (s == -1) { | |
164 ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno, | |
165 ngx_socket_n " %s failed", ls[i].addr_text.data); | |
166 return NGX_ERROR; | |
167 } | |
168 | |
169 #if (WIN32) | |
170 /* | |
171 * Winsock assignes a socket number divisible by 4 | |
172 * so to find a connection we divide a socket number by 4. | |
173 */ | |
174 | |
175 if (s % 4) { | |
176 ngx_log_error(NGX_LOG_EMERG, ls->log, 0, | |
177 ngx_socket_n " created socket %d", s); | |
178 return NGX_ERROR; | |
179 } | |
180 #endif | |
181 | |
182 if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, | |
183 (const void *) &reuseaddr, sizeof(int)) == -1) { | |
184 ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno, | |
185 "setsockopt(SO_REUSEADDR) %s failed", | |
186 ls[i].addr_text.data); | |
187 return NGX_ERROR; | |
188 } | |
189 | |
190 /* TODO: close on exit */ | |
191 | |
192 if (!(ngx_event_flags & NGX_USE_AIO_EVENT)) { | |
193 if (ngx_nonblocking(s) == -1) { | |
194 ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno, | |
195 ngx_nonblocking_n " %s failed", | |
196 ls[i].addr_text.data); | |
197 return NGX_ERROR; | |
198 } | |
199 } | |
200 | |
201 #if 0 | |
202 if (ls[i].nonblocking) { | |
203 if (ngx_nonblocking(s) == -1) { | |
204 ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno, | |
205 ngx_nonblocking_n " %s failed", | |
206 ls[i].addr_text.data); | |
207 return NGX_ERROR; | |
208 } | |
209 } | |
210 #endif | |
211 | |
212 if (bind(s, ls[i].sockaddr, ls[i].socklen) == -1) { | |
213 err = ngx_socket_errno; | |
214 ngx_log_error(NGX_LOG_EMERG, log, err, | |
215 "bind() to %s failed", ls[i].addr_text.data); | |
216 | |
217 if (err != NGX_EADDRINUSE) | |
218 return NGX_ERROR; | |
219 | |
220 if (ngx_close_socket(s) == -1) | |
221 ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno, | |
222 ngx_close_socket_n " %s failed", | |
223 ls[i].addr_text.data); | |
224 | |
225 failed = 1; | |
226 continue; | |
227 } | |
228 | |
229 if (listen(s, ls[i].backlog) == -1) { | |
230 ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno, | |
231 "listen() to %s failed", ls[i].addr_text.data); | |
232 return NGX_ERROR; | |
233 } | |
234 | |
235 /* TODO: deferred accept */ | |
236 | |
237 ls[i].fd = s; | |
238 } | |
239 | |
240 if (!failed) | |
241 break; | |
242 | |
243 /* TODO: delay configurable */ | |
244 | |
245 ngx_log_error(NGX_LOG_NOTICE, log, 0, | |
246 "try again to bind() after 500ms"); | |
247 ngx_msleep(500); | |
248 } | |
249 | |
250 if (failed) { | |
251 ngx_log_error(NGX_LOG_EMERG, log, 0, "still can not bind()"); | |
252 return NGX_ERROR; | |
253 } | |
254 | |
255 return NGX_OK; | |
256 } | |
257 | |
258 | |
259 void ngx_close_listening_sockets(ngx_cycle_t *cycle) | |
260 { | |
261 ngx_uint_t i; | |
262 ngx_socket_t fd; | |
263 ngx_listening_t *ls; | |
264 | |
265 if (ngx_event_flags & NGX_USE_IOCP_EVENT) { | |
266 return; | |
267 } | |
268 | |
269 ngx_accept_mutex_held = 0; | |
270 ngx_accept_mutex = NULL; | |
271 | |
272 ls = cycle->listening.elts; | |
273 for (i = 0; i < cycle->listening.nelts; i++) { | |
274 fd = ls[i].fd; | |
275 | |
276 #if (WIN32) | |
277 /* | |
278 * Winsock assignes a socket number divisible by 4 | |
279 * so to find a connection we divide a socket number by 4. | |
280 */ | |
281 | |
282 fd /= 4; | |
283 #endif | |
284 | |
285 if (ngx_event_flags & NGX_USE_RTSIG_EVENT) { | |
286 if (cycle->connections[fd].read->active) { | |
287 ngx_del_conn(&cycle->connections[fd], NGX_CLOSE_EVENT); | |
288 } | |
289 | |
290 } else { | |
291 if (cycle->read_events[fd].active) { | |
292 ngx_del_event(&cycle->read_events[fd], | |
293 NGX_READ_EVENT, NGX_CLOSE_EVENT); | |
294 } | |
295 } | |
296 | |
297 if (ngx_close_socket(ls[i].fd) == -1) { | |
298 ngx_log_error(NGX_LOG_EMERG, cycle->log, ngx_socket_errno, | |
299 ngx_close_socket_n " %s failed", | |
300 ls[i].addr_text.data); | |
301 } | |
302 | |
303 cycle->connections[fd].fd = (ngx_socket_t) -1; | |
304 } | |
305 } | |
306 | |
307 | |
308 void ngx_close_connection(ngx_connection_t *c) | |
309 { | |
310 ngx_socket_t fd; | |
311 | |
312 if (c->pool == NULL) { | |
313 ngx_log_error(NGX_LOG_ALERT, c->log, 0, "connection already closed"); | |
314 return; | |
315 } | |
316 | |
317 #if (NGX_OPENSSL) | |
318 | |
319 if (c->ssl) { | |
320 if (ngx_ssl_shutdown(c) == NGX_AGAIN) { | |
321 c->read->event_handler = ngx_ssl_close_handler; | |
322 c->write->event_handler = ngx_ssl_close_handler; | |
323 return; | |
324 } | |
325 } | |
326 | |
327 #endif | |
328 | |
329 if (c->read->timer_set) { | |
330 ngx_del_timer(c->read); | |
331 } | |
332 | |
333 if (c->write->timer_set) { | |
334 ngx_del_timer(c->write); | |
335 } | |
336 | |
337 if (ngx_del_conn) { | |
338 ngx_del_conn(c, NGX_CLOSE_EVENT); | |
339 | |
340 } else { | |
341 if (c->read->active || c->read->disabled) { | |
342 ngx_del_event(c->read, NGX_READ_EVENT, NGX_CLOSE_EVENT); | |
343 } | |
344 | |
345 if (c->write->active || c->write->disabled) { | |
346 ngx_del_event(c->write, NGX_WRITE_EVENT, NGX_CLOSE_EVENT); | |
347 } | |
348 } | |
349 | |
350 #if (NGX_THREADS) | |
351 | |
352 /* | |
353 * we have to clean the connection information before the closing | |
354 * because another thread may reopen the same file descriptor | |
355 * before we clean the connection | |
356 */ | |
357 | |
358 if (ngx_mutex_lock(ngx_posted_events_mutex) == NGX_OK) { | |
359 | |
360 if (c->read->prev) { | |
361 ngx_delete_posted_event(c->read); | |
362 } | |
363 | |
364 if (c->write->prev) { | |
365 ngx_delete_posted_event(c->write); | |
366 } | |
367 | |
368 c->read->closed = 1; | |
369 c->write->closed = 1; | |
370 | |
371 if (c->single_connection) { | |
372 ngx_unlock(&c->lock); | |
373 c->read->locked = 0; | |
374 c->write->locked = 0; | |
375 } | |
376 | |
377 ngx_mutex_unlock(ngx_posted_events_mutex); | |
378 } | |
379 | |
380 #else | |
381 | |
382 if (c->read->prev) { | |
383 ngx_delete_posted_event(c->read); | |
384 } | |
385 | |
386 if (c->write->prev) { | |
387 ngx_delete_posted_event(c->write); | |
388 } | |
389 | |
390 c->read->closed = 1; | |
391 c->write->closed = 1; | |
392 | |
393 #endif | |
394 | |
395 fd = c->fd; | |
396 c->fd = (ngx_socket_t) -1; | |
397 c->data = NULL; | |
398 | |
399 ngx_destroy_pool(c->pool); | |
400 | |
401 if (ngx_close_socket(fd) == -1) { | |
402 | |
403 /* we use ngx_cycle->log because c->log was in c->pool */ | |
404 | |
405 ngx_log_error(NGX_LOG_ALERT, ngx_cycle->log, ngx_socket_errno, | |
406 ngx_close_socket_n " failed"); | |
407 } | |
408 } | |
409 | |
410 | |
411 | |
412 ngx_int_t ngx_connection_error(ngx_connection_t *c, ngx_err_t err, char *text) | |
413 { | |
414 ngx_uint_t level; | |
415 | |
416 if (err == NGX_ECONNRESET | |
417 && c->log_error == NGX_ERROR_IGNORE_ECONNRESET) | |
418 { | |
419 return 0; | |
420 } | |
421 | |
422 if (err == NGX_ECONNRESET | |
423 #if !(WIN32) | |
424 || err == NGX_EPIPE | |
425 #endif | |
426 || err == NGX_ENOTCONN | |
427 || err == NGX_ECONNREFUSED | |
428 || err == NGX_EHOSTUNREACH) | |
429 { | |
430 | |
431 switch (c->log_error) { | |
432 | |
433 case NGX_ERROR_IGNORE_ECONNRESET: | |
434 case NGX_ERROR_INFO: | |
435 level = NGX_LOG_INFO; | |
436 break; | |
437 | |
438 case NGX_ERROR_ERR: | |
439 level = NGX_LOG_ERR; | |
440 break; | |
441 | |
442 default: | |
443 level = NGX_LOG_CRIT; | |
444 } | |
445 | |
446 } else { | |
447 level = NGX_LOG_CRIT; | |
448 } | |
449 | |
450 ngx_log_error(level, c->log, err, text); | |
451 | |
452 return NGX_ERROR; | |
453 } |