0
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
|
4 */
|
|
5
|
|
6
|
|
7 #include <ngx_config.h>
|
|
8 #include <ngx_core.h>
|
|
9 #include <ngx_channel.h>
|
|
10
|
|
11
|
|
12 ngx_int_t ngx_write_channel(ngx_socket_t s, ngx_channel_t *ch, size_t size,
|
|
13 ngx_log_t *log)
|
|
14 {
|
|
15 ssize_t n;
|
|
16 ngx_err_t err;
|
|
17 struct iovec iov[1];
|
|
18 struct msghdr msg;
|
|
19
|
18
|
20 #if (NGX_HAVE_MSGHDR_MSG_CONTROL)
|
0
|
21
|
|
22 union {
|
|
23 struct cmsghdr cm;
|
|
24 char space[CMSG_SPACE(sizeof(int))];
|
|
25 } cmsg;
|
|
26
|
|
27 if (ch->fd == -1) {
|
|
28 msg.msg_control = NULL;
|
|
29 msg.msg_controllen = 0;
|
|
30
|
|
31 } else {
|
|
32 msg.msg_control = (caddr_t) &cmsg;
|
|
33 msg.msg_controllen = sizeof(cmsg);
|
|
34
|
|
35 cmsg.cm.cmsg_len = sizeof(cmsg);
|
|
36 cmsg.cm.cmsg_level = SOL_SOCKET;
|
|
37 cmsg.cm.cmsg_type = SCM_RIGHTS;
|
|
38 *(int *) CMSG_DATA(&cmsg.cm) = ch->fd;
|
|
39 }
|
|
40
|
32
|
41 msg.msg_flags = 0;
|
|
42
|
0
|
43 #else
|
|
44
|
|
45 if (ch->fd == -1) {
|
|
46 msg.msg_accrights = NULL;
|
|
47 msg.msg_accrightslen = 0;
|
|
48
|
|
49 } else {
|
|
50 msg.msg_accrights = (caddr_t) &ch->fd;
|
|
51 msg.msg_accrightslen = sizeof(int);
|
|
52 }
|
|
53
|
|
54 #endif
|
|
55
|
|
56 iov[0].iov_base = (char *) ch;
|
|
57 iov[0].iov_len = size;
|
|
58
|
|
59 msg.msg_name = NULL;
|
|
60 msg.msg_namelen = 0;
|
|
61 msg.msg_iov = iov;
|
|
62 msg.msg_iovlen = 1;
|
|
63
|
|
64 n = sendmsg(s, &msg, 0);
|
|
65
|
|
66 if (n == -1) {
|
|
67 err = ngx_errno;
|
|
68 if (err == NGX_EAGAIN) {
|
|
69 return NGX_AGAIN;
|
|
70 }
|
|
71
|
|
72 ngx_log_error(NGX_LOG_ALERT, log, err, "sendmsg() failed");
|
|
73 return NGX_ERROR;
|
|
74 }
|
|
75
|
|
76 return NGX_OK;
|
|
77 }
|
|
78
|
|
79
|
|
80 ngx_int_t ngx_read_channel(ngx_socket_t s, ngx_channel_t *ch, size_t size,
|
|
81 ngx_log_t *log)
|
|
82 {
|
|
83 ssize_t n;
|
|
84 ngx_err_t err;
|
|
85 struct iovec iov[1];
|
|
86 struct msghdr msg;
|
|
87
|
18
|
88 #if (NGX_HAVE_MSGHDR_MSG_CONTROL)
|
0
|
89 union {
|
|
90 struct cmsghdr cm;
|
|
91 char space[CMSG_SPACE(sizeof(int))];
|
|
92 } cmsg;
|
|
93 #else
|
|
94 int fd;
|
|
95 #endif
|
|
96
|
|
97 iov[0].iov_base = (char *) ch;
|
|
98 iov[0].iov_len = size;
|
|
99
|
|
100 msg.msg_name = NULL;
|
|
101 msg.msg_namelen = 0;
|
|
102 msg.msg_iov = iov;
|
|
103 msg.msg_iovlen = 1;
|
|
104
|
18
|
105 #if (NGX_HAVE_MSGHDR_MSG_CONTROL)
|
0
|
106 msg.msg_control = (caddr_t) &cmsg;
|
|
107 msg.msg_controllen = sizeof(cmsg);
|
|
108 #else
|
|
109 msg.msg_accrights = (caddr_t) &fd;
|
|
110 msg.msg_accrightslen = sizeof(int);
|
|
111 #endif
|
|
112
|
|
113 n = recvmsg(s, &msg, 0);
|
|
114
|
|
115 if (n == -1) {
|
|
116 err = ngx_errno;
|
|
117 if (err == NGX_EAGAIN) {
|
|
118 return NGX_AGAIN;
|
|
119 }
|
|
120
|
|
121 ngx_log_error(NGX_LOG_ALERT, log, err, "recvmsg() failed");
|
|
122 return NGX_ERROR;
|
|
123 }
|
|
124
|
10
|
125 if (n == 0) {
|
|
126 ngx_log_debug0(NGX_LOG_DEBUG_CORE, log, 0, "recvmsg() returned zero");
|
|
127 return NGX_ERROR;
|
|
128 }
|
|
129
|
0
|
130 if ((size_t) n < sizeof(ngx_channel_t)) {
|
|
131 ngx_log_error(NGX_LOG_ALERT, log, 0,
|
10
|
132 "recvmsg() returned not enough data: %uz", n);
|
0
|
133 return NGX_ERROR;
|
|
134 }
|
|
135
|
18
|
136 #if (NGX_HAVE_MSGHDR_MSG_CONTROL)
|
0
|
137
|
|
138 if (ch->command == NGX_CMD_OPEN_CHANNEL) {
|
|
139
|
58
|
140 if (cmsg.cm.cmsg_len < (socklen_t) sizeof(cmsg)) {
|
0
|
141 ngx_log_error(NGX_LOG_ALERT, log, 0,
|
|
142 "recvmsg() returned too small ancillary data");
|
|
143 return NGX_ERROR;
|
|
144 }
|
|
145
|
|
146 if (cmsg.cm.cmsg_level != SOL_SOCKET || cmsg.cm.cmsg_type != SCM_RIGHTS)
|
|
147 {
|
|
148 ngx_log_error(NGX_LOG_ALERT, log, 0,
|
|
149 "recvmsg() returned invalid ancillary data "
|
|
150 "level %d or type %d",
|
|
151 cmsg.cm.cmsg_level, cmsg.cm.cmsg_type);
|
|
152 return NGX_ERROR;
|
|
153 }
|
|
154
|
|
155 ch->fd = *(int *) CMSG_DATA(&cmsg.cm);
|
|
156 }
|
|
157
|
|
158 if (msg.msg_flags & (MSG_TRUNC|MSG_CTRUNC)) {
|
|
159 ngx_log_error(NGX_LOG_ALERT, log, 0,
|
|
160 "recvmsg() truncated data");
|
|
161 }
|
|
162
|
|
163 #else
|
|
164
|
|
165 if (ch->command == NGX_CMD_OPEN_CHANNEL) {
|
|
166 if (msg.msg_accrightslen != sizeof(int)) {
|
|
167 ngx_log_error(NGX_LOG_ALERT, log, 0,
|
|
168 "recvmsg() returned no ancillary data");
|
|
169 return NGX_ERROR;
|
|
170 }
|
|
171
|
|
172 ch->fd = fd;
|
|
173 }
|
|
174
|
|
175 #endif
|
|
176
|
|
177 return n;
|
|
178 }
|
|
179
|
|
180
|
|
181 ngx_int_t ngx_add_channel_event(ngx_cycle_t *cycle, ngx_fd_t fd,
|
|
182 ngx_int_t event, ngx_event_handler_pt handler)
|
|
183 {
|
|
184 ngx_event_t *ev, *rev, *wev;
|
|
185 ngx_connection_t *c;
|
|
186
|
|
187 c = &cycle->connections[fd];
|
|
188 rev = &cycle->read_events[fd];
|
|
189 wev = &cycle->write_events[fd];
|
|
190
|
|
191 ngx_memzero(c, sizeof(ngx_connection_t));
|
|
192 ngx_memzero(rev, sizeof(ngx_event_t));
|
|
193 ngx_memzero(wev, sizeof(ngx_event_t));
|
|
194
|
|
195 c->fd = fd;
|
|
196 c->pool = cycle->pool;
|
|
197
|
|
198 c->read = rev;
|
|
199 c->write = wev;
|
|
200
|
|
201 c->log = cycle->log;
|
|
202 rev->log = cycle->log;
|
|
203 wev->log = cycle->log;
|
|
204 rev->index = NGX_INVALID_INDEX;
|
|
205 wev->index = NGX_INVALID_INDEX;
|
|
206 rev->data = c;
|
|
207 wev->data = c;
|
|
208
|
|
209 #if (NGX_THREADS)
|
|
210 rev->lock = &c->lock;
|
|
211 wev->lock = &c->lock;
|
|
212 rev->own_lock = &c->lock;
|
|
213 wev->own_lock = &c->lock;
|
|
214 #endif
|
|
215
|
|
216 ev = (event == NGX_READ_EVENT) ? rev : wev;
|
|
217
|
58
|
218 ev->handler = handler;
|
0
|
219
|
|
220 if (ngx_add_conn && (ngx_event_flags & NGX_USE_EPOLL_EVENT) == 0) {
|
|
221 if (ngx_add_conn(c) == NGX_ERROR) {
|
|
222 return NGX_ERROR;
|
|
223 }
|
|
224
|
|
225 } else {
|
|
226 if (ngx_add_event(ev, event, 0) == NGX_ERROR) {
|
|
227 return NGX_ERROR;
|
|
228 }
|
|
229 }
|
|
230
|
|
231 return NGX_OK;
|
|
232 }
|
|
233
|
|
234
|
|
235 void ngx_close_channel(ngx_fd_t *fd, ngx_log_t *log)
|
|
236 {
|
|
237 if (close(fd[0]) == -1) {
|
14
|
238 ngx_log_error(NGX_LOG_ALERT, log, ngx_errno, "close() channel failed");
|
0
|
239 }
|
|
240
|
|
241 if (close(fd[1]) == -1) {
|
14
|
242 ngx_log_error(NGX_LOG_ALERT, log, ngx_errno, "close() channel failed");
|
0
|
243 }
|
|
244 }
|