1 2 /* 3 * Copyright (C) Igor Sysoev 4 * Copyright (C) NGINX, Inc. 5 */ 6 7 #include <nxt_main.h> 8 9 10 void 11 nxt_conn_wait(nxt_conn_t *c) 12 { 13 nxt_event_engine_t *engine; 14 const nxt_conn_state_t *state; 15 16 nxt_debug(c->socket.task, "conn wait fd:%d rdy:%d", 17 c->socket.fd, c->socket.read_ready); 18 19 engine = c->socket.task->thread->engine; 20 state = c->read_state; 21 22 if (c->socket.read_ready) { 23 nxt_work_queue_add(&engine->fast_work_queue, state->ready_handler, 24 c->socket.task, c, c->socket.data); 25 return; 26 } 27 28 c->socket.read_handler = state->ready_handler; 29 c->socket.error_handler = state->error_handler; 30 31 nxt_conn_timer(engine, c, state, &c->read_timer); 32 33 nxt_fd_event_enable_read(engine, &c->socket); 34 } 35 36 37 void 38 nxt_conn_io_read(nxt_task_t *task, void *obj, void *data) 39 { 40 ssize_t n; 41 nxt_conn_t *c; 42 nxt_work_queue_t *wq; 43 nxt_event_engine_t *engine; 44 nxt_work_handler_t handler; 45 const nxt_conn_state_t *state; 46 47 c = obj; 48 49 nxt_debug(task, "conn read fd:%d rdy:%d cl:%d", 50 c->socket.fd, c->socket.read_ready, c->socket.closed); 51 52 engine = task->thread->engine; 53 54 state = c->read_state; 55 56 if (c->socket.read_ready) { 57 58 if (state->io_read_handler == NULL) { 59 n = c->io->recvbuf(c, c->read); 60 61 } else { 62 n = state->io_read_handler(c); 63 /* The state can be changed by io_read_handler. */ 64 state = c->read_state; 65 } 66 67 if (n > 0) { 68 c->nbytes = n; 69 70 nxt_recvbuf_update(c->read, n); 71 72 nxt_fd_event_block_read(engine, &c->socket); 73 74 if (state->timer_autoreset) { 75 nxt_timer_disable(engine, &c->read_timer); 76 } 77 78 wq = c->read_work_queue; 79 handler = state->ready_handler; 80 81 nxt_work_queue_add(wq, handler, task, c, data); 82 83 return; 84 } 85 86 if (n != NXT_AGAIN) { 87 nxt_fd_event_block_read(engine, &c->socket); 88 nxt_timer_disable(engine, &c->read_timer); 89 90 wq = &engine->fast_work_queue; 91 92 handler = (n == 0) ? state->close_handler : state->error_handler; 93 94 nxt_work_queue_add(wq, handler, task, c, data); 95 96 return; 97 } 98 } 99 100 /* 101 * Here c->io->read() is assigned instead of direct nxt_conn_io_read() 102 * because the function can be called by nxt_kqueue_conn_io_read(). 103 */ 104 c->socket.read_handler = c->io->read; 105 c->socket.error_handler = state->error_handler; 106 107 if (nxt_fd_event_is_disabled(c->socket.read)) { 108 nxt_fd_event_enable_read(engine, &c->socket); 109 } 110 111 if (state->timer_autoreset || c->read_timer.state == NXT_TIMER_DISABLED) { 112 nxt_conn_timer(engine, c, state, &c->read_timer); 113 } 114 } 115 116 117 ssize_t 118 nxt_conn_io_recvbuf(nxt_conn_t *c, nxt_buf_t *b) 119 { 120 ssize_t n; 121 nxt_err_t err; 122 nxt_uint_t niov; 123 struct iovec iov[NXT_IOBUF_MAX]; 124 nxt_recvbuf_coalesce_t rb; 125 126 rb.buf = b; 127 rb.iobuf = iov; 128 rb.nmax = NXT_IOBUF_MAX; 129 rb.size = 0; 130 131 niov = nxt_recvbuf_mem_coalesce(&rb); 132 133 if (niov == 1) { 134 /* Disposal of surplus kernel iovec copy-in operation. */ 135 return nxt_conn_io_recv(c, iov->iov_base, iov->iov_len, 0); 136 } 137 138 for ( ;; ) { 139 n = readv(c->socket.fd, iov, niov); 140 141 err = (n == -1) ? nxt_socket_errno : 0; 142 143 nxt_debug(c->socket.task, "readv(%d, %ui): %z", c->socket.fd, niov, n); 144 145 if (n > 0) { 146 if ((size_t) n < rb.size) { 147 c->socket.read_ready = 0; 148 } 149 150 return n; 151 } 152 153 if (n == 0) { 154 c->socket.closed = 1; 155 c->socket.read_ready = 0; 156 return n; 157 } 158 159 /* n == -1 */ 160 161 switch (err) { 162 163 case NXT_EAGAIN: 164 nxt_debug(c->socket.task, "readv() %E", err); 165 c->socket.read_ready = 0; 166 return NXT_AGAIN; 167 168 case NXT_EINTR: 169 nxt_debug(c->socket.task, "readv() %E", err); 170 continue; 171 172 default: 173 c->socket.error = err; 174 nxt_log(c->socket.task, nxt_socket_error_level(err), 175 "readv(%d, %ui) failed %E", c->socket.fd, niov, err); 176 177 return NXT_ERROR; 178 } 179 } 180 } 181 182 183 ssize_t 184 nxt_conn_io_recv(nxt_conn_t *c, void *buf, size_t size, nxt_uint_t flags) 185 { 186 ssize_t n; 187 nxt_err_t err; 188 189 for ( ;; ) { 190 n = recv(c->socket.fd, buf, size, flags); 191 192 err = (n == -1) ? nxt_socket_errno : 0; 193 194 nxt_debug(c->socket.task, "recv(%d, %p, %uz, 0x%ui): %z", 195 c->socket.fd, buf, size, flags, n); 196 197 if (n > 0) { 198 if ((size_t) n < size) { 199 c->socket.read_ready = 0; 200 } 201 202 return n; 203 } 204 205 if (n == 0) { 206 c->socket.closed = 1; 207 c->socket.read_ready = 0; 208 209 return n; 210 } 211 212 /* n == -1 */ 213 214 switch (err) { 215 216 case NXT_EAGAIN: 217 nxt_debug(c->socket.task, "recv() %E", err); 218 c->socket.read_ready = 0; 219 220 return NXT_AGAIN; 221 222 case NXT_EINTR: 223 nxt_debug(c->socket.task, "recv() %E", err); 224 continue; 225 226 default: 227 c->socket.error = err; 228 nxt_log(c->socket.task, nxt_socket_error_level(err), 229 "recv(%d, %p, %uz, %ui) failed %E", 230 c->socket.fd, buf, size, flags, err); 231 232 return NXT_ERROR; 233 } 234 } 235 } 236