nginx: src/event/ngx_event

annotate src/event/ngx_event_udp.c @ 7391:27559d4a5151

Stream: fixed possible use of a freed connection. The session handler may result in session termination, thus a connection pool (from which c->udp was allocated) may be destroyed.

author	Vladimir Homutov <vl@nginx.com>
date	Wed, 07 Nov 2018 13:22:14 +0300
parents	d27aa9060c95
children	4698cede59ff

rev	line source
7285 88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	1
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	2 /*
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	3 * Copyright (C) Roman Arutyunyan
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	4 * Copyright (C) Nginx, Inc.
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	5 */
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	6
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	7
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	8 #include <ngx_config.h>
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	9 #include <ngx_core.h>
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	10 #include <ngx_event.h>
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	11
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	12
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	13 #if !(NGX_WIN32)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	14
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	15 struct ngx_udp_connection_s {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	16 ngx_rbtree_node_t node;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	17 ngx_connection_t *connection;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	18 ngx_buf_t *buffer;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	19 };
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	20
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	21
7285 88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	22 static void ngx_close_accepted_udp_connection(ngx_connection_t *c);
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	23 static ssize_t ngx_udp_shared_recv(ngx_connection_t c, u_char buf,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	24 size_t size);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	25 static ngx_int_t ngx_insert_udp_connection(ngx_connection_t *c);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	26 static void ngx_delete_udp_connection(void *data);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	27 static ngx_connection_t ngx_lookup_udp_connection(ngx_listening_t ls,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	28 struct sockaddr *sockaddr, socklen_t socklen,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	29 struct sockaddr *local_sockaddr, socklen_t local_socklen);
7285 88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	30
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	31
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	32 void
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	33 ngx_event_recvmsg(ngx_event_t *ev)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	34 {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	35 ssize_t n;
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	36 ngx_buf_t buf;
7285 88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	37 ngx_log_t *log;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	38 ngx_err_t err;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	39 socklen_t socklen, local_socklen;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	40 ngx_event_t rev, wev;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	41 struct iovec iov[1];
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	42 struct msghdr msg;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	43 ngx_sockaddr_t sa, lsa;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	44 struct sockaddr sockaddr, local_sockaddr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	45 ngx_listening_t *ls;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	46 ngx_event_conf_t *ecf;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	47 ngx_connection_t c, lc;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	48 static u_char buffer[65535];
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	49
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	50 #if (NGX_HAVE_MSGHDR_MSG_CONTROL)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	51
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	52 #if (NGX_HAVE_IP_RECVDSTADDR)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	53 u_char msg_control[CMSG_SPACE(sizeof(struct in_addr))];
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	54 #elif (NGX_HAVE_IP_PKTINFO)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	55 u_char msg_control[CMSG_SPACE(sizeof(struct in_pktinfo))];
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	56 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	57
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	58 #if (NGX_HAVE_INET6 && NGX_HAVE_IPV6_RECVPKTINFO)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	59 u_char msg_control6[CMSG_SPACE(sizeof(struct in6_pktinfo))];
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	60 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	61
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	62 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	63
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	64 if (ev->timedout) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	65 if (ngx_enable_accept_events((ngx_cycle_t *) ngx_cycle) != NGX_OK) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	66 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	67 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	68
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	69 ev->timedout = 0;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	70 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	71
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	72 ecf = ngx_event_get_conf(ngx_cycle->conf_ctx, ngx_event_core_module);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	73
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	74 if (!(ngx_event_flags & NGX_USE_KQUEUE_EVENT)) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	75 ev->available = ecf->multi_accept;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	76 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	77
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	78 lc = ev->data;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	79 ls = lc->listening;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	80 ev->ready = 0;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	81
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	82 ngx_log_debug2(NGX_LOG_DEBUG_EVENT, ev->log, 0,
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	83 "recvmsg on %V, ready: %d", &ls->addr_text, ev->available);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	84
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	85 do {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	86 ngx_memzero(&msg, sizeof(struct msghdr));
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	87
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	88 iov[0].iov_base = (void *) buffer;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	89 iov[0].iov_len = sizeof(buffer);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	90
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	91 msg.msg_name = &sa;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	92 msg.msg_namelen = sizeof(ngx_sockaddr_t);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	93 msg.msg_iov = iov;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	94 msg.msg_iovlen = 1;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	95
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	96 #if (NGX_HAVE_MSGHDR_MSG_CONTROL)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	97
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	98 if (ls->wildcard) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	99
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	100 #if (NGX_HAVE_IP_RECVDSTADDR \|\| NGX_HAVE_IP_PKTINFO)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	101 if (ls->sockaddr->sa_family == AF_INET) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	102 msg.msg_control = &msg_control;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	103 msg.msg_controllen = sizeof(msg_control);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	104 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	105 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	106
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	107 #if (NGX_HAVE_INET6 && NGX_HAVE_IPV6_RECVPKTINFO)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	108 if (ls->sockaddr->sa_family == AF_INET6) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	109 msg.msg_control = &msg_control6;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	110 msg.msg_controllen = sizeof(msg_control6);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	111 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	112 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	113 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	114
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	115 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	116
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	117 n = recvmsg(lc->fd, &msg, 0);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	118
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	119 if (n == -1) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	120 err = ngx_socket_errno;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	121
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	122 if (err == NGX_EAGAIN) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	123 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, ev->log, err,
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	124 "recvmsg() not ready");
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	125 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	126 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	127
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	128 ngx_log_error(NGX_LOG_ALERT, ev->log, err, "recvmsg() failed");
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	129
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	130 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	131 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	132
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	133 #if (NGX_HAVE_MSGHDR_MSG_CONTROL)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	134 if (msg.msg_flags & (MSG_TRUNC\|MSG_CTRUNC)) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	135 ngx_log_error(NGX_LOG_ALERT, ev->log, 0,
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	136 "recvmsg() truncated data");
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	137 continue;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	138 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	139 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	140
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	141 sockaddr = msg.msg_name;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	142 socklen = msg.msg_namelen;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	143
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	144 if (socklen > (socklen_t) sizeof(ngx_sockaddr_t)) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	145 socklen = sizeof(ngx_sockaddr_t);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	146 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	147
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	148 if (socklen == 0) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	149
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	150 /*
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	151 * on Linux recvmsg() returns zero msg_namelen
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	152 * when receiving packets from unbound AF_UNIX sockets
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	153 */
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	154
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	155 socklen = sizeof(struct sockaddr);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	156 ngx_memzero(&sa, sizeof(struct sockaddr));
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	157 sa.sockaddr.sa_family = ls->sockaddr->sa_family;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	158 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	159
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	160 local_sockaddr = ls->sockaddr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	161 local_socklen = ls->socklen;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	162
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	163 #if (NGX_HAVE_MSGHDR_MSG_CONTROL)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	164
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	165 if (ls->wildcard) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	166 struct cmsghdr *cmsg;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	167
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	168 ngx_memcpy(&lsa, local_sockaddr, local_socklen);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	169 local_sockaddr = &lsa.sockaddr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	170
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	171 for (cmsg = CMSG_FIRSTHDR(&msg);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	172 cmsg != NULL;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	173 cmsg = CMSG_NXTHDR(&msg, cmsg))
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	174 {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	175
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	176 #if (NGX_HAVE_IP_RECVDSTADDR)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	177
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	178 if (cmsg->cmsg_level == IPPROTO_IP
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	179 && cmsg->cmsg_type == IP_RECVDSTADDR
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	180 && local_sockaddr->sa_family == AF_INET)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	181 {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	182 struct in_addr *addr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	183 struct sockaddr_in *sin;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	184
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	185 addr = (struct in_addr *) CMSG_DATA(cmsg);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	186 sin = (struct sockaddr_in *) local_sockaddr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	187 sin->sin_addr = *addr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	188
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	189 break;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	190 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	191
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	192 #elif (NGX_HAVE_IP_PKTINFO)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	193
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	194 if (cmsg->cmsg_level == IPPROTO_IP
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	195 && cmsg->cmsg_type == IP_PKTINFO
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	196 && local_sockaddr->sa_family == AF_INET)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	197 {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	198 struct in_pktinfo *pkt;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	199 struct sockaddr_in *sin;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	200
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	201 pkt = (struct in_pktinfo *) CMSG_DATA(cmsg);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	202 sin = (struct sockaddr_in *) local_sockaddr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	203 sin->sin_addr = pkt->ipi_addr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	204
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	205 break;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	206 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	207
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	208 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	209
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	210 #if (NGX_HAVE_INET6 && NGX_HAVE_IPV6_RECVPKTINFO)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	211
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	212 if (cmsg->cmsg_level == IPPROTO_IPV6
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	213 && cmsg->cmsg_type == IPV6_PKTINFO
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	214 && local_sockaddr->sa_family == AF_INET6)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	215 {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	216 struct in6_pktinfo *pkt6;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	217 struct sockaddr_in6 *sin6;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	218
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	219 pkt6 = (struct in6_pktinfo *) CMSG_DATA(cmsg);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	220 sin6 = (struct sockaddr_in6 *) local_sockaddr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	221 sin6->sin6_addr = pkt6->ipi6_addr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	222
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	223 break;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	224 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	225
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	226 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	227
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	228 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	229 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	230
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	231 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	232
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	233 c = ngx_lookup_udp_connection(ls, sockaddr, socklen, local_sockaddr,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	234 local_socklen);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	235
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	236 if (c) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	237
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	238 #if (NGX_DEBUG)
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	239 if (c->log->log_level & NGX_LOG_DEBUG_EVENT) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	240 ngx_log_handler_pt handler;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	241
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	242 handler = c->log->handler;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	243 c->log->handler = NULL;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	244
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	245 ngx_log_debug2(NGX_LOG_DEBUG_EVENT, c->log, 0,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	246 "recvmsg: fd:%d n:%z", c->fd, n);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	247
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	248 c->log->handler = handler;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	249 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	250 #endif
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	251
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	252 ngx_memzero(&buf, sizeof(ngx_buf_t));
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	253
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	254 buf.pos = buffer;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	255 buf.last = buffer + n;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	256
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	257 rev = c->read;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	258
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	259 c->udp->buffer = &buf;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	260 rev->ready = 1;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	261
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	262 rev->handler(rev);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	263
7391 27559d4a5151 Stream: fixed possible use of a freed connection. Vladimir Homutov <vl@nginx.com> parents: 7286 diff changeset	264 if (c->udp) {
27559d4a5151 Stream: fixed possible use of a freed connection. Vladimir Homutov <vl@nginx.com> parents: 7286 diff changeset	265 c->udp->buffer = NULL;
27559d4a5151 Stream: fixed possible use of a freed connection. Vladimir Homutov <vl@nginx.com> parents: 7286 diff changeset	266 }
27559d4a5151 Stream: fixed possible use of a freed connection. Vladimir Homutov <vl@nginx.com> parents: 7286 diff changeset	267
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	268 rev->ready = 0;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	269
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	270 goto next;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	271 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	272
7285 88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	273 #if (NGX_STAT_STUB)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	274 (void) ngx_atomic_fetch_add(ngx_stat_accepted, 1);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	275 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	276
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	277 ngx_accept_disabled = ngx_cycle->connection_n / 8
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	278 - ngx_cycle->free_connection_n;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	279
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	280 c = ngx_get_connection(lc->fd, ev->log);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	281 if (c == NULL) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	282 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	283 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	284
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	285 c->shared = 1;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	286 c->type = SOCK_DGRAM;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	287 c->socklen = socklen;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	288
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	289 #if (NGX_STAT_STUB)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	290 (void) ngx_atomic_fetch_add(ngx_stat_active, 1);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	291 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	292
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	293 c->pool = ngx_create_pool(ls->pool_size, ev->log);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	294 if (c->pool == NULL) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	295 ngx_close_accepted_udp_connection(c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	296 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	297 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	298
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	299 c->sockaddr = ngx_palloc(c->pool, socklen);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	300 if (c->sockaddr == NULL) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	301 ngx_close_accepted_udp_connection(c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	302 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	303 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	304
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	305 ngx_memcpy(c->sockaddr, sockaddr, socklen);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	306
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	307 log = ngx_palloc(c->pool, sizeof(ngx_log_t));
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	308 if (log == NULL) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	309 ngx_close_accepted_udp_connection(c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	310 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	311 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	312
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	313 *log = ls->log;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	314
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	315 c->recv = ngx_udp_shared_recv;
7285 88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	316 c->send = ngx_udp_send;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	317 c->send_chain = ngx_udp_send_chain;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	318
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	319 c->log = log;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	320 c->pool->log = log;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	321 c->listening = ls;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	322
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	323 if (local_sockaddr == &lsa.sockaddr) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	324 local_sockaddr = ngx_palloc(c->pool, local_socklen);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	325 if (local_sockaddr == NULL) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	326 ngx_close_accepted_udp_connection(c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	327 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	328 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	329
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	330 ngx_memcpy(local_sockaddr, &lsa, local_socklen);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	331 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	332
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	333 c->local_sockaddr = local_sockaddr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	334 c->local_socklen = local_socklen;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	335
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	336 c->buffer = ngx_create_temp_buf(c->pool, n);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	337 if (c->buffer == NULL) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	338 ngx_close_accepted_udp_connection(c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	339 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	340 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	341
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	342 c->buffer->last = ngx_cpymem(c->buffer->last, buffer, n);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	343
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	344 rev = c->read;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	345 wev = c->write;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	346
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	347 wev->ready = 1;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	348
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	349 rev->log = log;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	350 wev->log = log;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	351
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	352 /*
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	353 * TODO: MT: - ngx_atomic_fetch_add()
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	354 * or protection by critical section or light mutex
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	355 *
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	356 * TODO: MP: - allocated in a shared memory
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	357 * - ngx_atomic_fetch_add()
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	358 * or protection by critical section or light mutex
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	359 */
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	360
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	361 c->number = ngx_atomic_fetch_add(ngx_connection_counter, 1);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	362
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	363 #if (NGX_STAT_STUB)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	364 (void) ngx_atomic_fetch_add(ngx_stat_handled, 1);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	365 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	366
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	367 if (ls->addr_ntop) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	368 c->addr_text.data = ngx_pnalloc(c->pool, ls->addr_text_max_len);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	369 if (c->addr_text.data == NULL) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	370 ngx_close_accepted_udp_connection(c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	371 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	372 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	373
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	374 c->addr_text.len = ngx_sock_ntop(c->sockaddr, c->socklen,
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	375 c->addr_text.data,
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	376 ls->addr_text_max_len, 0);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	377 if (c->addr_text.len == 0) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	378 ngx_close_accepted_udp_connection(c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	379 return;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	380 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	381 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	382
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	383 #if (NGX_DEBUG)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	384 {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	385 ngx_str_t addr;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	386 u_char text[NGX_SOCKADDR_STRLEN];
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	387
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	388 ngx_debug_accepted_connection(ecf, c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	389
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	390 if (log->log_level & NGX_LOG_DEBUG_EVENT) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	391 addr.data = text;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	392 addr.len = ngx_sock_ntop(c->sockaddr, c->socklen, text,
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	393 NGX_SOCKADDR_STRLEN, 1);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	394
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	395 ngx_log_debug4(NGX_LOG_DEBUG_EVENT, log, 0,
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	396 "*%uA recvmsg: %V fd:%d n:%z",
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	397 c->number, &addr, c->fd, n);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	398 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	399
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	400 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	401 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	402
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	403 if (ngx_insert_udp_connection(c) != NGX_OK) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	404 ngx_close_accepted_udp_connection(c);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	405 return;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	406 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	407
7285 88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	408 log->data = NULL;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	409 log->handler = NULL;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	410
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	411 ls->handler(c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	412
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	413 next:
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	414
7285 88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	415 if (ngx_event_flags & NGX_USE_KQUEUE_EVENT) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	416 ev->available -= n;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	417 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	418
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	419 } while (ev->available);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	420 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	421
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	422
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	423 static void
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	424 ngx_close_accepted_udp_connection(ngx_connection_t *c)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	425 {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	426 ngx_free_connection(c);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	427
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	428 c->fd = (ngx_socket_t) -1;
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	429
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	430 if (c->pool) {
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	431 ngx_destroy_pool(c->pool);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	432 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	433
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	434 #if (NGX_STAT_STUB)
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	435 (void) ngx_atomic_fetch_add(ngx_stat_active, -1);
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	436 #endif
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	437 }
88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	438
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	439
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	440 static ssize_t
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	441 ngx_udp_shared_recv(ngx_connection_t c, u_char buf, size_t size)
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	442 {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	443 ssize_t n;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	444 ngx_buf_t *b;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	445
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	446 if (c->udp == NULL \|\| c->udp->buffer == NULL) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	447 return NGX_AGAIN;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	448 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	449
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	450 b = c->udp->buffer;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	451
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	452 n = ngx_min(b->last - b->pos, (ssize_t) size);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	453
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	454 ngx_memcpy(buf, b->pos, n);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	455
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	456 c->udp->buffer = NULL;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	457 c->read->ready = 0;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	458
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	459 return n;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	460 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	461
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	462
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	463 void
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	464 ngx_udp_rbtree_insert_value(ngx_rbtree_node_t *temp,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	465 ngx_rbtree_node_t node, ngx_rbtree_node_t sentinel)
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	466 {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	467 ngx_int_t rc;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	468 ngx_connection_t c, ct;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	469 ngx_rbtree_node_t **p;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	470 ngx_udp_connection_t udp, udpt;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	471
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	472 for ( ;; ) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	473
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	474 if (node->key < temp->key) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	475
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	476 p = &temp->left;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	477
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	478 } else if (node->key > temp->key) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	479
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	480 p = &temp->right;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	481
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	482 } else { /* node->key == temp->key */
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	483
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	484 udp = (ngx_udp_connection_t *) node;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	485 c = udp->connection;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	486
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	487 udpt = (ngx_udp_connection_t *) temp;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	488 ct = udpt->connection;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	489
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	490 rc = ngx_cmp_sockaddr(c->sockaddr, c->socklen,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	491 ct->sockaddr, ct->socklen, 1);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	492
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	493 if (rc == 0 && c->listening->wildcard) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	494 rc = ngx_cmp_sockaddr(c->local_sockaddr, c->local_socklen,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	495 ct->local_sockaddr, ct->local_socklen, 1);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	496 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	497
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	498 p = (rc < 0) ? &temp->left : &temp->right;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	499 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	500
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	501 if (*p == sentinel) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	502 break;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	503 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	504
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	505 temp = *p;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	506 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	507
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	508 *p = node;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	509 node->parent = temp;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	510 node->left = sentinel;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	511 node->right = sentinel;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	512 ngx_rbt_red(node);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	513 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	514
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	515
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	516 static ngx_int_t
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	517 ngx_insert_udp_connection(ngx_connection_t *c)
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	518 {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	519 uint32_t hash;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	520 ngx_pool_cleanup_t *cln;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	521 ngx_udp_connection_t *udp;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	522
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	523 if (c->udp) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	524 return NGX_OK;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	525 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	526
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	527 udp = ngx_pcalloc(c->pool, sizeof(ngx_udp_connection_t));
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	528 if (udp == NULL) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	529 return NGX_ERROR;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	530 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	531
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	532 udp->connection = c;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	533
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	534 ngx_crc32_init(hash);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	535 ngx_crc32_update(&hash, (u_char *) c->sockaddr, c->socklen);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	536
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	537 if (c->listening->wildcard) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	538 ngx_crc32_update(&hash, (u_char *) c->local_sockaddr, c->local_socklen);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	539 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	540
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	541 ngx_crc32_final(hash);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	542
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	543 udp->node.key = hash;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	544
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	545 cln = ngx_pool_cleanup_add(c->pool, 0);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	546 if (cln == NULL) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	547 return NGX_ERROR;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	548 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	549
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	550 cln->data = c;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	551 cln->handler = ngx_delete_udp_connection;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	552
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	553 ngx_rbtree_insert(&c->listening->rbtree, &udp->node);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	554
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	555 c->udp = udp;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	556
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	557 return NGX_OK;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	558 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	559
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	560
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	561 static void
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	562 ngx_delete_udp_connection(void *data)
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	563 {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	564 ngx_connection_t *c = data;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	565
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	566 ngx_rbtree_delete(&c->listening->rbtree, &c->udp->node);
7391 27559d4a5151 Stream: fixed possible use of a freed connection. Vladimir Homutov <vl@nginx.com> parents: 7286 diff changeset	567
27559d4a5151 Stream: fixed possible use of a freed connection. Vladimir Homutov <vl@nginx.com> parents: 7286 diff changeset	568 c->udp = NULL;
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	569 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	570
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	571
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	572 static ngx_connection_t *
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	573 ngx_lookup_udp_connection(ngx_listening_t ls, struct sockaddr sockaddr,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	574 socklen_t socklen, struct sockaddr *local_sockaddr, socklen_t local_socklen)
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	575 {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	576 uint32_t hash;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	577 ngx_int_t rc;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	578 ngx_connection_t *c;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	579 ngx_rbtree_node_t node, sentinel;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	580 ngx_udp_connection_t *udp;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	581
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	582 #if (NGX_HAVE_UNIX_DOMAIN)
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	583
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	584 if (sockaddr->sa_family == AF_UNIX) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	585 struct sockaddr_un saun = (struct sockaddr_un ) sockaddr;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	586
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	587 if (socklen <= (socklen_t) offsetof(struct sockaddr_un, sun_path)
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	588 \|\| saun->sun_path[0] == '\0')
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	589 {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	590 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, ngx_cycle->log, 0,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	591 "unbound unix socket");
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	592 return NULL;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	593 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	594 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	595
7285 88a624c9b491 Events: moved ngx_recvmsg() to new file src/event/ngx_event_udp.c. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	596 #endif
7286 d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	597
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	598 node = ls->rbtree.root;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	599 sentinel = ls->rbtree.sentinel;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	600
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	601 ngx_crc32_init(hash);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	602 ngx_crc32_update(&hash, (u_char *) sockaddr, socklen);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	603
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	604 if (ls->wildcard) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	605 ngx_crc32_update(&hash, (u_char *) local_sockaddr, local_socklen);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	606 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	607
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	608 ngx_crc32_final(hash);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	609
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	610 while (node != sentinel) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	611
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	612 if (hash < node->key) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	613 node = node->left;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	614 continue;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	615 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	616
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	617 if (hash > node->key) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	618 node = node->right;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	619 continue;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	620 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	621
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	622 /* hash == node->key */
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	623
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	624 udp = (ngx_udp_connection_t *) node;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	625
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	626 c = udp->connection;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	627
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	628 rc = ngx_cmp_sockaddr(sockaddr, socklen,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	629 c->sockaddr, c->socklen, 1);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	630
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	631 if (rc == 0 && ls->wildcard) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	632 rc = ngx_cmp_sockaddr(local_sockaddr, local_socklen,
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	633 c->local_sockaddr, c->local_socklen, 1);
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	634 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	635
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	636 if (rc == 0) {
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	637 return c;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	638 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	639
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	640 node = (rc < 0) ? node->left : node->right;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	641 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	642
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	643 return NULL;
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	644 }
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	645
d27aa9060c95 Stream: udp streams. Roman Arutyunyan <arut@nginx.com> parents: 7285 diff changeset	646 #endif

Mercurial > hg > nginx

annotate src/event/ngx_event_udp.c @ 7391:27559d4a5151