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