nginx: src/os/unix/ngx_udp_sendmsg

annotate src/os/unix/ngx_udp_sendmsg_chain.c @ 8116:3108d4d668e4

Style.

author	BullerDu <yefei.dyf@alibaba-inc.com>
date	Fri, 16 Dec 2022 01:15:15 +0400
parents	0f6cc8f73744
children	af5adec171b4

rev	line source
6692 56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	1
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	2 /*
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	3 * Copyright (C) Igor Sysoev
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	4 * Copyright (C) Nginx, Inc.
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	5 */
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	6
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	7
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	8 #include <ngx_config.h>
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	9 #include <ngx_core.h>
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	10 #include <ngx_event.h>
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	11
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	12
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	13 static ngx_chain_t ngx_udp_output_chain_to_iovec(ngx_iovec_t vec,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	14 ngx_chain_t in, ngx_log_t log);
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	15 static ssize_t ngx_sendmsg_vec(ngx_connection_t c, ngx_iovec_t vec);
6692 56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	16
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	17
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	18 ngx_chain_t *
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	19 ngx_udp_unix_sendmsg_chain(ngx_connection_t c, ngx_chain_t in, off_t limit)
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	20 {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	21 ssize_t n;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	22 off_t send;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	23 ngx_chain_t *cl;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	24 ngx_event_t *wev;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	25 ngx_iovec_t vec;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	26 struct iovec iovs[NGX_IOVS_PREALLOCATE];
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	27
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	28 wev = c->write;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	29
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	30 if (!wev->ready) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	31 return in;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	32 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	33
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	34 #if (NGX_HAVE_KQUEUE)
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	35
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	36 if ((ngx_event_flags & NGX_USE_KQUEUE_EVENT) && wev->pending_eof) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	37 (void) ngx_connection_error(c, wev->kq_errno,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	38 "kevent() reported about an closed connection");
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	39 wev->error = 1;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	40 return NGX_CHAIN_ERROR;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	41 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	42
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	43 #endif
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	44
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	45 /* the maximum limit size is the maximum size_t value - the page size */
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	46
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	47 if (limit == 0 \|\| limit > (off_t) (NGX_MAX_SIZE_T_VALUE - ngx_pagesize)) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	48 limit = NGX_MAX_SIZE_T_VALUE - ngx_pagesize;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	49 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	50
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	51 send = 0;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	52
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	53 vec.iovs = iovs;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	54 vec.nalloc = NGX_IOVS_PREALLOCATE;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	55
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	56 for ( ;; ) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	57
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	58 /* create the iovec and coalesce the neighbouring bufs */
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	59
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	60 cl = ngx_udp_output_chain_to_iovec(&vec, in, c->log);
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	61
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	62 if (cl == NGX_CHAIN_ERROR) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	63 return NGX_CHAIN_ERROR;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	64 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	65
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	66 if (cl && cl->buf->in_file) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	67 ngx_log_error(NGX_LOG_ALERT, c->log, 0,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	68 "file buf in sendmsg "
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	69 "t:%d r:%d f:%d %p %p-%p %p %O-%O",
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	70 cl->buf->temporary,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	71 cl->buf->recycled,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	72 cl->buf->in_file,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	73 cl->buf->start,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	74 cl->buf->pos,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	75 cl->buf->last,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	76 cl->buf->file,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	77 cl->buf->file_pos,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	78 cl->buf->file_last);
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	79
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	80 ngx_debug_point();
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	81
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	82 return NGX_CHAIN_ERROR;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	83 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	84
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	85 if (cl == in) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	86 return in;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	87 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	88
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	89 send += vec.size;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	90
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	91 n = ngx_sendmsg_vec(c, &vec);
6692 56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	92
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	93 if (n == NGX_ERROR) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	94 return NGX_CHAIN_ERROR;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	95 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	96
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	97 if (n == NGX_AGAIN) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	98 wev->ready = 0;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	99 return in;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	100 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	101
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	102 c->sent += n;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	103
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	104 in = ngx_chain_update_sent(in, n);
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	105
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	106 if (send >= limit \|\| in == NULL) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	107 return in;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	108 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	109 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	110 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	111
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	112
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	113 static ngx_chain_t *
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	114 ngx_udp_output_chain_to_iovec(ngx_iovec_t vec, ngx_chain_t in, ngx_log_t *log)
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	115 {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	116 size_t total, size;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	117 u_char *prev;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	118 ngx_uint_t n, flush;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	119 ngx_chain_t *cl;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	120 struct iovec *iov;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	121
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	122 cl = in;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	123 iov = NULL;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	124 prev = NULL;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	125 total = 0;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	126 n = 0;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	127 flush = 0;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	128
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	129 for ( /* void */ ; in && !flush; in = in->next) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	130
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	131 if (in->buf->flush \|\| in->buf->last_buf) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	132 flush = 1;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	133 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	134
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	135 if (ngx_buf_special(in->buf)) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	136 continue;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	137 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	138
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	139 if (in->buf->in_file) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	140 break;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	141 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	142
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	143 if (!ngx_buf_in_memory(in->buf)) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	144 ngx_log_error(NGX_LOG_ALERT, log, 0,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	145 "bad buf in output chain "
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	146 "t:%d r:%d f:%d %p %p-%p %p %O-%O",
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	147 in->buf->temporary,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	148 in->buf->recycled,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	149 in->buf->in_file,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	150 in->buf->start,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	151 in->buf->pos,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	152 in->buf->last,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	153 in->buf->file,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	154 in->buf->file_pos,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	155 in->buf->file_last);
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	156
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	157 ngx_debug_point();
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	158
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	159 return NGX_CHAIN_ERROR;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	160 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	161
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	162 size = in->buf->last - in->buf->pos;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	163
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	164 if (prev == in->buf->pos) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	165 iov->iov_len += size;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	166
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	167 } else {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	168 if (n == vec->nalloc) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	169 ngx_log_error(NGX_LOG_ALERT, log, 0,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	170 "too many parts in a datagram");
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	171 return NGX_CHAIN_ERROR;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	172 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	173
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	174 iov = &vec->iovs[n++];
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	175
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	176 iov->iov_base = (void *) in->buf->pos;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	177 iov->iov_len = size;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	178 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	179
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	180 prev = in->buf->pos + size;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	181 total += size;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	182 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	183
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	184 if (!flush) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	185 #if (NGX_SUPPRESS_WARN)
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	186 vec->size = 0;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	187 vec->count = 0;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	188 #endif
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	189 return cl;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	190 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	191
7665 d127837c714f Stream: fixed processing of zero length UDP packets (ticket #1982). Vladimir Homutov <vl@nginx.com> parents: 7019 diff changeset	192 /* zero-sized datagram; pretend to have at least 1 iov */
d127837c714f Stream: fixed processing of zero length UDP packets (ticket #1982). Vladimir Homutov <vl@nginx.com> parents: 7019 diff changeset	193 if (n == 0) {
d127837c714f Stream: fixed processing of zero length UDP packets (ticket #1982). Vladimir Homutov <vl@nginx.com> parents: 7019 diff changeset	194 iov = &vec->iovs[n++];
d127837c714f Stream: fixed processing of zero length UDP packets (ticket #1982). Vladimir Homutov <vl@nginx.com> parents: 7019 diff changeset	195 iov->iov_base = NULL;
d127837c714f Stream: fixed processing of zero length UDP packets (ticket #1982). Vladimir Homutov <vl@nginx.com> parents: 7019 diff changeset	196 iov->iov_len = 0;
d127837c714f Stream: fixed processing of zero length UDP packets (ticket #1982). Vladimir Homutov <vl@nginx.com> parents: 7019 diff changeset	197 }
d127837c714f Stream: fixed processing of zero length UDP packets (ticket #1982). Vladimir Homutov <vl@nginx.com> parents: 7019 diff changeset	198
6692 56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	199 vec->count = n;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	200 vec->size = total;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	201
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	202 return in;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	203 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	204
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	205
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	206 static ssize_t
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	207 ngx_sendmsg_vec(ngx_connection_t c, ngx_iovec_t vec)
6692 56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	208 {
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	209 struct msghdr msg;
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	210
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	211 #if (NGX_HAVE_ADDRINFO_CMSG)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	212 struct cmsghdr *cmsg;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	213 u_char msg_control[CMSG_SPACE(sizeof(ngx_addrinfo_t))];
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	214 #endif
dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	215
6692 56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	216 ngx_memzero(&msg, sizeof(struct msghdr));
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	217
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	218 if (c->socklen) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	219 msg.msg_name = c->sockaddr;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	220 msg.msg_namelen = c->socklen;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	221 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	222
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	223 msg.msg_iov = vec->iovs;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	224 msg.msg_iovlen = vec->count;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	225
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	226 #if (NGX_HAVE_ADDRINFO_CMSG)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	227 if (c->listening && c->listening->wildcard && c->local_sockaddr) {
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	228
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	229 msg.msg_control = msg_control;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	230 msg.msg_controllen = sizeof(msg_control);
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	231 ngx_memzero(msg_control, sizeof(msg_control));
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	232
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	233 cmsg = CMSG_FIRSTHDR(&msg);
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	234
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	235 msg.msg_controllen = ngx_set_srcaddr_cmsg(cmsg, c->local_sockaddr);
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	236 }
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	237 #endif
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	238
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	239 return ngx_sendmsg(c, &msg, 0);
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	240 }
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	241
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	242
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	243 #if (NGX_HAVE_ADDRINFO_CMSG)
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	244
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	245 size_t
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	246 ngx_set_srcaddr_cmsg(struct cmsghdr cmsg, struct sockaddr local_sockaddr)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	247 {
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	248 size_t len;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	249 #if (NGX_HAVE_IP_SENDSRCADDR)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	250 struct in_addr *addr;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	251 struct sockaddr_in *sin;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	252 #elif (NGX_HAVE_IP_PKTINFO)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	253 struct in_pktinfo *pkt;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	254 struct sockaddr_in *sin;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	255 #endif
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	256
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	257 #if (NGX_HAVE_INET6 && NGX_HAVE_IPV6_RECVPKTINFO)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	258 struct in6_pktinfo *pkt6;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	259 struct sockaddr_in6 *sin6;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	260 #endif
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	261
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	262
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	263 #if (NGX_HAVE_IP_SENDSRCADDR) \|\| (NGX_HAVE_IP_PKTINFO)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	264
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	265 if (local_sockaddr->sa_family == AF_INET) {
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	266
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	267 cmsg->cmsg_level = IPPROTO_IP;
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	268
dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	269 #if (NGX_HAVE_IP_SENDSRCADDR)
dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	270
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	271 cmsg->cmsg_type = IP_SENDSRCADDR;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	272 cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_addr));
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	273 len = CMSG_SPACE(sizeof(struct in_addr));
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	274
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	275 sin = (struct sockaddr_in *) local_sockaddr;
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	276
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	277 addr = (struct in_addr *) CMSG_DATA(cmsg);
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	278 *addr = sin->sin_addr;
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	279
dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	280 #elif (NGX_HAVE_IP_PKTINFO)
dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	281
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	282 cmsg->cmsg_type = IP_PKTINFO;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	283 cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	284 len = CMSG_SPACE(sizeof(struct in_pktinfo));
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	285
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	286 sin = (struct sockaddr_in *) local_sockaddr;
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	287
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	288 pkt = (struct in_pktinfo *) CMSG_DATA(cmsg);
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	289 ngx_memzero(pkt, sizeof(struct in_pktinfo));
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	290 pkt->ipi_spec_dst = sin->sin_addr;
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	291
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	292 #endif
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	293 return len;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	294 }
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	295
dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	296 #endif
dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	297
dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	298 #if (NGX_HAVE_INET6 && NGX_HAVE_IPV6_RECVPKTINFO)
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	299 if (local_sockaddr->sa_family == AF_INET6) {
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	300
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	301 cmsg->cmsg_level = IPPROTO_IPV6;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	302 cmsg->cmsg_type = IPV6_PKTINFO;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	303 cmsg->cmsg_len = CMSG_LEN(sizeof(struct in6_pktinfo));
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	304 len = CMSG_SPACE(sizeof(struct in6_pktinfo));
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	305
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	306 sin6 = (struct sockaddr_in6 *) local_sockaddr;
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	307
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	308 pkt6 = (struct in6_pktinfo *) CMSG_DATA(cmsg);
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	309 ngx_memzero(pkt6, sizeof(struct in6_pktinfo));
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	310 pkt6->ipi6_addr = sin6->sin6_addr;
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	311
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	312 return len;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	313 }
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	314 #endif
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	315
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	316 return 0;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	317 }
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	318
8003 0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	319
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	320 ngx_int_t
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	321 ngx_get_srcaddr_cmsg(struct cmsghdr cmsg, struct sockaddr local_sockaddr)
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	322 {
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	323
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	324 #if (NGX_HAVE_IP_RECVDSTADDR)
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	325 struct in_addr *addr;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	326 struct sockaddr_in *sin;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	327 #elif (NGX_HAVE_IP_PKTINFO)
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	328 struct in_pktinfo *pkt;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	329 struct sockaddr_in *sin;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	330 #endif
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	331
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	332 #if (NGX_HAVE_INET6 && NGX_HAVE_IPV6_RECVPKTINFO)
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	333 struct in6_pktinfo *pkt6;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	334 struct sockaddr_in6 *sin6;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	335 #endif
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	336
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	337
8116 3108d4d668e4 Style. BullerDu <yefei.dyf@alibaba-inc.com> parents: 8003 diff changeset	338 #if (NGX_HAVE_IP_RECVDSTADDR)
8003 0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	339
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	340 if (cmsg->cmsg_level == IPPROTO_IP
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	341 && cmsg->cmsg_type == IP_RECVDSTADDR
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	342 && local_sockaddr->sa_family == AF_INET)
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	343 {
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	344 addr = (struct in_addr *) CMSG_DATA(cmsg);
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	345 sin = (struct sockaddr_in *) local_sockaddr;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	346 sin->sin_addr = *addr;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	347
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	348 return NGX_OK;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	349 }
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	350
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	351 #elif (NGX_HAVE_IP_PKTINFO)
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	352
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	353 if (cmsg->cmsg_level == IPPROTO_IP
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	354 && cmsg->cmsg_type == IP_PKTINFO
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	355 && local_sockaddr->sa_family == AF_INET)
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	356 {
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	357 pkt = (struct in_pktinfo *) CMSG_DATA(cmsg);
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	358 sin = (struct sockaddr_in *) local_sockaddr;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	359 sin->sin_addr = pkt->ipi_addr;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	360
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	361 return NGX_OK;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	362 }
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	363
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	364 #endif
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	365
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	366 #if (NGX_HAVE_INET6 && NGX_HAVE_IPV6_RECVPKTINFO)
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	367
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	368 if (cmsg->cmsg_level == IPPROTO_IPV6
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	369 && cmsg->cmsg_type == IPV6_PKTINFO
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	370 && local_sockaddr->sa_family == AF_INET6)
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	371 {
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	372 pkt6 = (struct in6_pktinfo *) CMSG_DATA(cmsg);
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	373 sin6 = (struct sockaddr_in6 *) local_sockaddr;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	374 sin6->sin6_addr = pkt6->ipi6_addr;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	375
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	376 return NGX_OK;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	377 }
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	378
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	379 #endif
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	380
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	381 return NGX_DECLINED;
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	382 }
0f6cc8f73744 Core: added function for local source address cmsg. Vladimir Homutov <vl@nginx.com> parents: 8002 diff changeset	383
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	384 #endif
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	385
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	386
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	387 ssize_t
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	388 ngx_sendmsg(ngx_connection_t c, struct msghdr msg, int flags)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	389 {
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	390 ssize_t n;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	391 ngx_err_t err;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	392 #if (NGX_DEBUG)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	393 size_t size;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	394 ngx_uint_t i;
6980 dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	395 #endif
dbb0c854e308 Set UDP datagram source address (ticket #1239). Roman Arutyunyan <arut@nginx.com> parents: 6692 diff changeset	396
6692 56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	397 eintr:
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	398
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	399 n = sendmsg(c->fd, msg, flags);
6692 56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	400
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	401 if (n == -1) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	402 err = ngx_errno;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	403
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	404 switch (err) {
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	405 case NGX_EAGAIN:
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	406 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	407 "sendmsg() not ready");
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	408 return NGX_AGAIN;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	409
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	410 case NGX_EINTR:
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	411 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	412 "sendmsg() was interrupted");
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	413 goto eintr;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	414
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	415 default:
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	416 c->write->error = 1;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	417 ngx_connection_error(c, err, "sendmsg() failed");
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	418 return NGX_ERROR;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	419 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	420 }
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	421
8002 cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	422 #if (NGX_DEBUG)
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	423 for (i = 0, size = 0; i < (size_t) msg->msg_iovlen; i++) {
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	424 size += msg->msg_iov[i].iov_len;
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	425 }
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	426
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	427 ngx_log_debug2(NGX_LOG_DEBUG_EVENT, c->log, 0,
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	428 "sendmsg: %z of %uz", n, size);
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	429 #endif
cfe1284e5d1d Core: made the ngx_sendmsg() function non-static. Vladimir Homutov <vl@nginx.com> parents: 7665 diff changeset	430
6692 56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	431 return n;
56fc55e32f23 Stream: filters. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	432 }

Mercurial > hg > nginx

annotate src/os/unix/ngx_udp_sendmsg_chain.c @ 8116:3108d4d668e4