mirror of
https://github.com/nginx/nginx.git
synced 2025-06-12 21:52:41 +08:00
343 lines
8.3 KiB
C
343 lines
8.3 KiB
C
|
|
/*
|
|
* Copyright (C) Igor Sysoev
|
|
* Copyright (C) Nginx, Inc.
|
|
*/
|
|
|
|
|
|
#include <ngx_config.h>
|
|
#include <ngx_core.h>
|
|
#include <ngx_event.h>
|
|
|
|
|
|
static ngx_chain_t *ngx_udp_output_chain_to_iovec(ngx_iovec_t *vec,
|
|
ngx_chain_t *in, ngx_log_t *log);
|
|
static ssize_t ngx_sendmsg(ngx_connection_t *c, ngx_iovec_t *vec);
|
|
|
|
|
|
ngx_chain_t *
|
|
ngx_udp_unix_sendmsg_chain(ngx_connection_t *c, ngx_chain_t *in, off_t limit)
|
|
{
|
|
ssize_t n;
|
|
off_t send;
|
|
ngx_chain_t *cl;
|
|
ngx_event_t *wev;
|
|
ngx_iovec_t vec;
|
|
struct iovec iovs[NGX_IOVS_PREALLOCATE];
|
|
|
|
wev = c->write;
|
|
|
|
if (!wev->ready) {
|
|
return in;
|
|
}
|
|
|
|
#if (NGX_HAVE_KQUEUE)
|
|
|
|
if ((ngx_event_flags & NGX_USE_KQUEUE_EVENT) && wev->pending_eof) {
|
|
(void) ngx_connection_error(c, wev->kq_errno,
|
|
"kevent() reported about an closed connection");
|
|
wev->error = 1;
|
|
return NGX_CHAIN_ERROR;
|
|
}
|
|
|
|
#endif
|
|
|
|
/* the maximum limit size is the maximum size_t value - the page size */
|
|
|
|
if (limit == 0 || limit > (off_t) (NGX_MAX_SIZE_T_VALUE - ngx_pagesize)) {
|
|
limit = NGX_MAX_SIZE_T_VALUE - ngx_pagesize;
|
|
}
|
|
|
|
send = 0;
|
|
|
|
vec.iovs = iovs;
|
|
vec.nalloc = NGX_IOVS_PREALLOCATE;
|
|
|
|
for ( ;; ) {
|
|
|
|
/* create the iovec and coalesce the neighbouring bufs */
|
|
|
|
cl = ngx_udp_output_chain_to_iovec(&vec, in, c->log);
|
|
|
|
if (cl == NGX_CHAIN_ERROR) {
|
|
return NGX_CHAIN_ERROR;
|
|
}
|
|
|
|
if (cl && cl->buf->in_file) {
|
|
ngx_log_error(NGX_LOG_ALERT, c->log, 0,
|
|
"file buf in sendmsg "
|
|
"t:%d r:%d f:%d %p %p-%p %p %O-%O",
|
|
cl->buf->temporary,
|
|
cl->buf->recycled,
|
|
cl->buf->in_file,
|
|
cl->buf->start,
|
|
cl->buf->pos,
|
|
cl->buf->last,
|
|
cl->buf->file,
|
|
cl->buf->file_pos,
|
|
cl->buf->file_last);
|
|
|
|
ngx_debug_point();
|
|
|
|
return NGX_CHAIN_ERROR;
|
|
}
|
|
|
|
if (cl == in) {
|
|
return in;
|
|
}
|
|
|
|
send += vec.size;
|
|
|
|
n = ngx_sendmsg(c, &vec);
|
|
|
|
if (n == NGX_ERROR) {
|
|
return NGX_CHAIN_ERROR;
|
|
}
|
|
|
|
if (n == NGX_AGAIN) {
|
|
wev->ready = 0;
|
|
return in;
|
|
}
|
|
|
|
c->sent += n;
|
|
|
|
in = ngx_chain_update_sent(in, n);
|
|
|
|
if (send >= limit || in == NULL) {
|
|
return in;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
static ngx_chain_t *
|
|
ngx_udp_output_chain_to_iovec(ngx_iovec_t *vec, ngx_chain_t *in, ngx_log_t *log)
|
|
{
|
|
size_t total, size;
|
|
u_char *prev;
|
|
ngx_uint_t n, flush;
|
|
ngx_chain_t *cl;
|
|
struct iovec *iov;
|
|
|
|
cl = in;
|
|
iov = NULL;
|
|
prev = NULL;
|
|
total = 0;
|
|
n = 0;
|
|
flush = 0;
|
|
|
|
for ( /* void */ ; in && !flush; in = in->next) {
|
|
|
|
if (in->buf->flush || in->buf->last_buf) {
|
|
flush = 1;
|
|
}
|
|
|
|
if (ngx_buf_special(in->buf)) {
|
|
continue;
|
|
}
|
|
|
|
if (in->buf->in_file) {
|
|
break;
|
|
}
|
|
|
|
if (!ngx_buf_in_memory(in->buf)) {
|
|
ngx_log_error(NGX_LOG_ALERT, log, 0,
|
|
"bad buf in output chain "
|
|
"t:%d r:%d f:%d %p %p-%p %p %O-%O",
|
|
in->buf->temporary,
|
|
in->buf->recycled,
|
|
in->buf->in_file,
|
|
in->buf->start,
|
|
in->buf->pos,
|
|
in->buf->last,
|
|
in->buf->file,
|
|
in->buf->file_pos,
|
|
in->buf->file_last);
|
|
|
|
ngx_debug_point();
|
|
|
|
return NGX_CHAIN_ERROR;
|
|
}
|
|
|
|
size = in->buf->last - in->buf->pos;
|
|
|
|
if (prev == in->buf->pos) {
|
|
iov->iov_len += size;
|
|
|
|
} else {
|
|
if (n == vec->nalloc) {
|
|
ngx_log_error(NGX_LOG_ALERT, log, 0,
|
|
"too many parts in a datagram");
|
|
return NGX_CHAIN_ERROR;
|
|
}
|
|
|
|
iov = &vec->iovs[n++];
|
|
|
|
iov->iov_base = (void *) in->buf->pos;
|
|
iov->iov_len = size;
|
|
}
|
|
|
|
prev = in->buf->pos + size;
|
|
total += size;
|
|
}
|
|
|
|
if (!flush) {
|
|
#if (NGX_SUPPRESS_WARN)
|
|
vec->size = 0;
|
|
vec->count = 0;
|
|
#endif
|
|
return cl;
|
|
}
|
|
|
|
/* zero-sized datagram; pretend to have at least 1 iov */
|
|
if (n == 0) {
|
|
iov = &vec->iovs[n++];
|
|
iov->iov_base = NULL;
|
|
iov->iov_len = 0;
|
|
}
|
|
|
|
vec->count = n;
|
|
vec->size = total;
|
|
|
|
return in;
|
|
}
|
|
|
|
|
|
static ssize_t
|
|
ngx_sendmsg(ngx_connection_t *c, ngx_iovec_t *vec)
|
|
{
|
|
ssize_t n;
|
|
ngx_err_t err;
|
|
struct msghdr msg;
|
|
|
|
#if (NGX_HAVE_MSGHDR_MSG_CONTROL)
|
|
|
|
#if (NGX_HAVE_IP_SENDSRCADDR)
|
|
u_char msg_control[CMSG_SPACE(sizeof(struct in_addr))];
|
|
#elif (NGX_HAVE_IP_PKTINFO)
|
|
u_char msg_control[CMSG_SPACE(sizeof(struct in_pktinfo))];
|
|
#endif
|
|
|
|
#if (NGX_HAVE_INET6 && NGX_HAVE_IPV6_RECVPKTINFO)
|
|
u_char msg_control6[CMSG_SPACE(sizeof(struct in6_pktinfo))];
|
|
#endif
|
|
|
|
#endif
|
|
|
|
ngx_memzero(&msg, sizeof(struct msghdr));
|
|
|
|
if (c->socklen) {
|
|
msg.msg_name = c->sockaddr;
|
|
msg.msg_namelen = c->socklen;
|
|
}
|
|
|
|
msg.msg_iov = vec->iovs;
|
|
msg.msg_iovlen = vec->count;
|
|
|
|
#if (NGX_HAVE_MSGHDR_MSG_CONTROL)
|
|
|
|
if (c->listening && c->listening->wildcard && c->local_sockaddr) {
|
|
|
|
#if (NGX_HAVE_IP_SENDSRCADDR)
|
|
|
|
if (c->local_sockaddr->sa_family == AF_INET) {
|
|
struct cmsghdr *cmsg;
|
|
struct in_addr *addr;
|
|
struct sockaddr_in *sin;
|
|
|
|
msg.msg_control = &msg_control;
|
|
msg.msg_controllen = sizeof(msg_control);
|
|
|
|
cmsg = CMSG_FIRSTHDR(&msg);
|
|
cmsg->cmsg_level = IPPROTO_IP;
|
|
cmsg->cmsg_type = IP_SENDSRCADDR;
|
|
cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_addr));
|
|
|
|
sin = (struct sockaddr_in *) c->local_sockaddr;
|
|
|
|
addr = (struct in_addr *) CMSG_DATA(cmsg);
|
|
*addr = sin->sin_addr;
|
|
}
|
|
|
|
#elif (NGX_HAVE_IP_PKTINFO)
|
|
|
|
if (c->local_sockaddr->sa_family == AF_INET) {
|
|
struct cmsghdr *cmsg;
|
|
struct in_pktinfo *pkt;
|
|
struct sockaddr_in *sin;
|
|
|
|
msg.msg_control = &msg_control;
|
|
msg.msg_controllen = sizeof(msg_control);
|
|
|
|
cmsg = CMSG_FIRSTHDR(&msg);
|
|
cmsg->cmsg_level = IPPROTO_IP;
|
|
cmsg->cmsg_type = IP_PKTINFO;
|
|
cmsg->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
|
|
|
|
sin = (struct sockaddr_in *) c->local_sockaddr;
|
|
|
|
pkt = (struct in_pktinfo *) CMSG_DATA(cmsg);
|
|
ngx_memzero(pkt, sizeof(struct in_pktinfo));
|
|
pkt->ipi_spec_dst = sin->sin_addr;
|
|
}
|
|
|
|
#endif
|
|
|
|
#if (NGX_HAVE_INET6 && NGX_HAVE_IPV6_RECVPKTINFO)
|
|
|
|
if (c->local_sockaddr->sa_family == AF_INET6) {
|
|
struct cmsghdr *cmsg;
|
|
struct in6_pktinfo *pkt6;
|
|
struct sockaddr_in6 *sin6;
|
|
|
|
msg.msg_control = &msg_control6;
|
|
msg.msg_controllen = sizeof(msg_control6);
|
|
|
|
cmsg = CMSG_FIRSTHDR(&msg);
|
|
cmsg->cmsg_level = IPPROTO_IPV6;
|
|
cmsg->cmsg_type = IPV6_PKTINFO;
|
|
cmsg->cmsg_len = CMSG_LEN(sizeof(struct in6_pktinfo));
|
|
|
|
sin6 = (struct sockaddr_in6 *) c->local_sockaddr;
|
|
|
|
pkt6 = (struct in6_pktinfo *) CMSG_DATA(cmsg);
|
|
ngx_memzero(pkt6, sizeof(struct in6_pktinfo));
|
|
pkt6->ipi6_addr = sin6->sin6_addr;
|
|
}
|
|
|
|
#endif
|
|
}
|
|
|
|
#endif
|
|
|
|
eintr:
|
|
|
|
n = sendmsg(c->fd, &msg, 0);
|
|
|
|
ngx_log_debug2(NGX_LOG_DEBUG_EVENT, c->log, 0,
|
|
"sendmsg: %z of %uz", n, vec->size);
|
|
|
|
if (n == -1) {
|
|
err = ngx_errno;
|
|
|
|
switch (err) {
|
|
case NGX_EAGAIN:
|
|
ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
|
|
"sendmsg() not ready");
|
|
return NGX_AGAIN;
|
|
|
|
case NGX_EINTR:
|
|
ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
|
|
"sendmsg() was interrupted");
|
|
goto eintr;
|
|
|
|
default:
|
|
c->write->error = 1;
|
|
ngx_connection_error(c, err, "sendmsg() failed");
|
|
return NGX_ERROR;
|
|
}
|
|
}
|
|
|
|
return n;
|
|
}
|