xref: /unit/src/nxt_conn.c (revision 771:f349b2d68e75)
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