1
2 /*
3 * Copyright (C) Igor Sysoev
4 * Copyright (C) NGINX, Inc.
5 */
6
7 #include <nxt_main.h>
8
9
10 nxt_conn_io_t nxt_unix_conn_io = {
11 .connect = nxt_conn_io_connect,
12 .accept = nxt_conn_io_accept,
13
14 .read = nxt_conn_io_read,
15 .recvbuf = nxt_conn_io_recvbuf,
16 .recv = nxt_conn_io_recv,
17
18 .write = nxt_conn_io_write,
19 .sendbuf = nxt_conn_io_sendbuf,
20
21 #if (NXT_HAVE_LINUX_SENDFILE)
22 .old_sendbuf = nxt_linux_event_conn_io_sendfile,
23 #elif (NXT_HAVE_FREEBSD_SENDFILE)
24 .old_sendbuf = nxt_freebsd_event_conn_io_sendfile,
25 #elif (NXT_HAVE_MACOSX_SENDFILE)
26 .old_sendbuf = nxt_macosx_event_conn_io_sendfile,
27 #elif (NXT_HAVE_SOLARIS_SENDFILEV)
28 .old_sendbuf = nxt_solaris_event_conn_io_sendfilev,
29 #elif (NXT_HAVE_AIX_SEND_FILE)
30 .old_sendbuf = nxt_aix_event_conn_io_send_file,
31 #elif (NXT_HAVE_HPUX_SENDFILE)
32 .old_sendbuf = nxt_hpux_event_conn_io_sendfile,
33 #else
34 .old_sendbuf = nxt_event_conn_io_sendbuf,
35 #endif
36
37 .writev = nxt_event_conn_io_writev,
38 .send = nxt_event_conn_io_send,
39 };
40
41
42 nxt_conn_t *
nxt_conn_create(nxt_mp_t * mp,nxt_task_t * task)43 nxt_conn_create(nxt_mp_t *mp, nxt_task_t *task)
44 {
45 nxt_conn_t *c;
46 nxt_thread_t *thr;
47
48 c = nxt_mp_zget(mp, sizeof(nxt_conn_t));
49 if (nxt_slow_path(c == NULL)) {
50 return NULL;
51 }
52
53 c->mem_pool = mp;
54
55 c->socket.fd = -1;
56
57 c->socket.log = &c->log;
58 c->log = *task->log;
59
60 /* The while loop skips possible uint32_t overflow. */
61
62 while (c->log.ident == 0) {
63 c->log.ident = nxt_task_next_ident();
64 }
65
66 thr = nxt_thread();
67 thr->engine->connections++;
68
69 c->task.thread = thr;
70 c->task.log = &c->log;
71 c->task.ident = c->log.ident;
72 c->socket.task = &c->task;
73 c->read_timer.task = &c->task;
74 c->write_timer.task = &c->task;
75
76 c->io = thr->engine->event.io;
77 c->max_chunk = NXT_INT32_T_MAX;
78 c->sendfile = NXT_CONN_SENDFILE_UNSET;
79
80 c->socket.read_work_queue = &thr->engine->fast_work_queue;
81 c->socket.write_work_queue = &thr->engine->fast_work_queue;
82
83 nxt_conn_timer_init(&c->read_timer, c, c->socket.read_work_queue);
84 nxt_conn_timer_init(&c->write_timer, c, c->socket.write_work_queue);
85
86 nxt_log_debug(&c->log, "connections: %uD", thr->engine->connections);
87
88 return c;
89 }
90
91
92 void
nxt_conn_free(nxt_task_t * task,nxt_conn_t * c)93 nxt_conn_free(nxt_task_t *task, nxt_conn_t *c)
94 {
95 nxt_mp_t *mp;
96
97 task->thread->engine->connections--;
98
99 mp = c->mem_pool;
100 nxt_mp_release(mp);
101 }
102
103
104 void
nxt_conn_timer(nxt_event_engine_t * engine,nxt_conn_t * c,const nxt_conn_state_t * state,nxt_timer_t * timer)105 nxt_conn_timer(nxt_event_engine_t *engine, nxt_conn_t *c,
106 const nxt_conn_state_t *state, nxt_timer_t *timer)
107 {
108 nxt_msec_t value;
109
110 if (state->timer_value != NULL) {
111 value = state->timer_value(c, state->timer_data);
112
113 if (value != 0) {
114 timer->handler = state->timer_handler;
115 nxt_timer_add(engine, timer, value);
116 }
117 }
118 }
119
120
121 void
nxt_conn_work_queue_set(nxt_conn_t * c,nxt_work_queue_t * wq)122 nxt_conn_work_queue_set(nxt_conn_t *c, nxt_work_queue_t *wq)
123 {
124 c->read_work_queue = wq;
125 c->write_work_queue = wq;
126 c->read_timer.work_queue = wq;
127 c->write_timer.work_queue = wq;
128 }
129
130
131 nxt_sockaddr_t *
nxt_conn_local_addr(nxt_task_t * task,nxt_conn_t * c)132 nxt_conn_local_addr(nxt_task_t *task, nxt_conn_t *c)
133 {
134 int ret;
135 size_t size, length;
136 socklen_t socklen;
137 nxt_sockaddr_t *sa;
138
139 if (c->local != NULL) {
140 return c->local;
141 }
142
143 /* AF_UNIX should not get in here. */
144
145 switch (c->remote->u.sockaddr.sa_family) {
146 #if (NXT_INET6)
147 case AF_INET6:
148 socklen = sizeof(struct sockaddr_in6);
149 length = NXT_INET6_ADDR_STR_LEN;
150 size = offsetof(nxt_sockaddr_t, u) + socklen + length;
151 break;
152 #endif
153 case AF_INET:
154 default:
155 socklen = sizeof(struct sockaddr_in);
156 length = NXT_INET_ADDR_STR_LEN;
157 size = offsetof(nxt_sockaddr_t, u) + socklen + length;
158 break;
159 }
160
161 sa = nxt_mp_get(c->mem_pool, size);
162 if (nxt_slow_path(sa == NULL)) {
163 return NULL;
164 }
165
166 sa->socklen = socklen;
167 sa->length = length;
168
169 ret = getsockname(c->socket.fd, &sa->u.sockaddr, &socklen);
170 if (nxt_slow_path(ret != 0)) {
171 nxt_alert(task, "getsockname(%d) failed", c->socket.fd);
172 return NULL;
173 }
174
175 c->local = sa;
176
177 nxt_sockaddr_text(sa);
178
179 /*
180 * TODO: here we can adjust the end of non-freeable block
181 * in c->mem_pool to the end of actual sockaddr length.
182 */
183
184 return sa;
185 }
186