]> git.saurik.com Git - redis.git/blame - src/anet.c
non blocking loading of the DB / AOF with informations and ETA in INFO output
[redis.git] / src / anet.c
CommitLineData
ed9b544e 1/* anet.c -- Basic TCP socket stuff made a bit less boring
2 *
12d090d2 3 * Copyright (c) 2006-2010, Salvatore Sanfilippo <antirez at gmail dot com>
ed9b544e 4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions are met:
8 *
9 * * Redistributions of source code must retain the above copyright notice,
10 * this list of conditions and the following disclaimer.
11 * * Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * * Neither the name of Redis nor the names of its contributors may be used
15 * to endorse or promote products derived from this software without
16 * specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
22 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28 * POSSIBILITY OF SUCH DAMAGE.
29 */
30
23d4709d 31#include "fmacros.h"
32
ed9b544e 33#include <sys/types.h>
34#include <sys/socket.h>
c61e6925 35#include <sys/un.h>
ed9b544e 36#include <netinet/in.h>
37#include <netinet/tcp.h>
38#include <arpa/inet.h>
39#include <unistd.h>
40#include <fcntl.h>
41#include <string.h>
42#include <netdb.h>
43#include <errno.h>
44#include <stdarg.h>
45#include <stdio.h>
46
47#include "anet.h"
48
49static void anetSetError(char *err, const char *fmt, ...)
50{
51 va_list ap;
52
53 if (!err) return;
54 va_start(ap, fmt);
55 vsnprintf(err, ANET_ERR_LEN, fmt, ap);
56 va_end(ap);
57}
58
59int anetNonBlock(char *err, int fd)
60{
61 int flags;
62
63 /* Set the socket nonblocking.
64 * Note that fcntl(2) for F_GETFL and F_SETFL can't be
65 * interrupted by a signal. */
66 if ((flags = fcntl(fd, F_GETFL)) == -1) {
67 anetSetError(err, "fcntl(F_GETFL): %s\n", strerror(errno));
68 return ANET_ERR;
69 }
70 if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1) {
71 anetSetError(err, "fcntl(F_SETFL,O_NONBLOCK): %s\n", strerror(errno));
72 return ANET_ERR;
73 }
74 return ANET_OK;
75}
76
77int anetTcpNoDelay(char *err, int fd)
78{
79 int yes = 1;
80 if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &yes, sizeof(yes)) == -1)
81 {
82 anetSetError(err, "setsockopt TCP_NODELAY: %s\n", strerror(errno));
83 return ANET_ERR;
84 }
85 return ANET_OK;
86}
87
88int anetSetSendBuffer(char *err, int fd, int buffsize)
89{
90 if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &buffsize, sizeof(buffsize)) == -1)
91 {
92 anetSetError(err, "setsockopt SO_SNDBUF: %s\n", strerror(errno));
93 return ANET_ERR;
94 }
95 return ANET_OK;
96}
97
98int anetTcpKeepAlive(char *err, int fd)
99{
100 int yes = 1;
101 if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &yes, sizeof(yes)) == -1) {
102 anetSetError(err, "setsockopt SO_KEEPALIVE: %s\n", strerror(errno));
103 return ANET_ERR;
104 }
105 return ANET_OK;
106}
107
108int anetResolve(char *err, char *host, char *ipbuf)
109{
110 struct sockaddr_in sa;
111
112 sa.sin_family = AF_INET;
113 if (inet_aton(host, &sa.sin_addr) == 0) {
114 struct hostent *he;
115
116 he = gethostbyname(host);
117 if (he == NULL) {
118 anetSetError(err, "can't resolve: %s\n", host);
119 return ANET_ERR;
120 }
121 memcpy(&sa.sin_addr, he->h_addr, sizeof(struct in_addr));
122 }
123 strcpy(ipbuf,inet_ntoa(sa.sin_addr));
124 return ANET_OK;
125}
126
704bd093
PN
127static int anetCreateSocket(char *err, int domain) {
128 int s, on = 1;
129 if ((s = socket(domain, SOCK_STREAM, 0)) == -1) {
130 anetSetError(err, "creating socket: %s\n", strerror(errno));
131 return ANET_ERR;
132 }
133
134 /* Make sure connection-intensive things like the redis benckmark
135 * will be able to close/open sockets a zillion of times */
136 if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1) {
137 anetSetError(err, "setsockopt SO_REUSEADDR: %s\n", strerror(errno));
138 return ANET_ERR;
139 }
140 return s;
141}
142
ed9b544e 143#define ANET_CONNECT_NONE 0
144#define ANET_CONNECT_NONBLOCK 1
145static int anetTcpGenericConnect(char *err, char *addr, int port, int flags)
146{
704bd093 147 int s;
ed9b544e 148 struct sockaddr_in sa;
149
704bd093 150 if ((s = anetCreateSocket(err,AF_INET)) == ANET_ERR)
ed9b544e 151 return ANET_ERR;
ed9b544e 152
153 sa.sin_family = AF_INET;
154 sa.sin_port = htons(port);
155 if (inet_aton(addr, &sa.sin_addr) == 0) {
156 struct hostent *he;
157
158 he = gethostbyname(addr);
159 if (he == NULL) {
160 anetSetError(err, "can't resolve: %s\n", addr);
161 close(s);
162 return ANET_ERR;
163 }
164 memcpy(&sa.sin_addr, he->h_addr, sizeof(struct in_addr));
165 }
166 if (flags & ANET_CONNECT_NONBLOCK) {
167 if (anetNonBlock(err,s) != ANET_OK)
168 return ANET_ERR;
169 }
170 if (connect(s, (struct sockaddr*)&sa, sizeof(sa)) == -1) {
171 if (errno == EINPROGRESS &&
172 flags & ANET_CONNECT_NONBLOCK)
173 return s;
174
175 anetSetError(err, "connect: %s\n", strerror(errno));
176 close(s);
177 return ANET_ERR;
178 }
179 return s;
180}
181
182int anetTcpConnect(char *err, char *addr, int port)
183{
184 return anetTcpGenericConnect(err,addr,port,ANET_CONNECT_NONE);
185}
186
187int anetTcpNonBlockConnect(char *err, char *addr, int port)
188{
189 return anetTcpGenericConnect(err,addr,port,ANET_CONNECT_NONBLOCK);
190}
191
c61e6925
PN
192int anetUnixGenericConnect(char *err, char *path, int flags)
193{
194 int s;
195 struct sockaddr_un sa;
196
704bd093 197 if ((s = anetCreateSocket(err,AF_LOCAL)) == ANET_ERR)
c61e6925 198 return ANET_ERR;
704bd093 199
c61e6925
PN
200 sa.sun_family = AF_LOCAL;
201 strncpy(sa.sun_path,path,sizeof(sa.sun_path)-1);
202 if (flags & ANET_CONNECT_NONBLOCK) {
203 if (anetNonBlock(err,s) != ANET_OK)
204 return ANET_ERR;
205 }
206 if (connect(s,(struct sockaddr*)&sa,sizeof(sa)) == -1) {
207 if (errno == EINPROGRESS &&
208 flags & ANET_CONNECT_NONBLOCK)
209 return s;
210
211 anetSetError(err, "connect: %s\n", strerror(errno));
212 close(s);
213 return ANET_ERR;
214 }
215 return s;
216}
217
218int anetUnixConnect(char *err, char *path)
219{
220 return anetUnixGenericConnect(err,path,ANET_CONNECT_NONE);
221}
222
223int anetUnixNonBlockConnect(char *err, char *path)
224{
225 return anetUnixGenericConnect(err,path,ANET_CONNECT_NONBLOCK);
226}
227
ed9b544e 228/* Like read(2) but make sure 'count' is read before to return
229 * (unless error or EOF condition is encountered) */
a4d1ba9a 230int anetRead(int fd, char *buf, int count)
ed9b544e 231{
232 int nread, totlen = 0;
233 while(totlen != count) {
234 nread = read(fd,buf,count-totlen);
235 if (nread == 0) return totlen;
236 if (nread == -1) return -1;
237 totlen += nread;
238 buf += nread;
239 }
240 return totlen;
241}
242
243/* Like write(2) but make sure 'count' is read before to return
244 * (unless error is encountered) */
a4d1ba9a 245int anetWrite(int fd, char *buf, int count)
ed9b544e 246{
247 int nwritten, totlen = 0;
248 while(totlen != count) {
249 nwritten = write(fd,buf,count-totlen);
250 if (nwritten == 0) return totlen;
251 if (nwritten == -1) return -1;
252 totlen += nwritten;
253 buf += nwritten;
254 }
255 return totlen;
256}
257
704bd093
PN
258static int anetListen(char *err, int s, struct sockaddr *sa, socklen_t len) {
259 if (bind(s,sa,len) == -1) {
260 anetSetError(err, "bind: %s\n", strerror(errno));
261 close(s);
ed9b544e 262 return ANET_ERR;
263 }
704bd093
PN
264 if (listen(s, 511) == -1) { /* the magic 511 constant is from nginx */
265 anetSetError(err, "listen: %s\n", strerror(errno));
ed9b544e 266 close(s);
267 return ANET_ERR;
268 }
704bd093
PN
269 return ANET_OK;
270}
271
272int anetTcpServer(char *err, int port, char *bindaddr)
273{
274 int s;
275 struct sockaddr_in sa;
276
277 if ((s = anetCreateSocket(err,AF_INET)) == ANET_ERR)
278 return ANET_ERR;
279
ed9b544e 280 memset(&sa,0,sizeof(sa));
281 sa.sin_family = AF_INET;
282 sa.sin_port = htons(port);
283 sa.sin_addr.s_addr = htonl(INADDR_ANY);
704bd093
PN
284 if (bindaddr && inet_aton(bindaddr, &sa.sin_addr) == 0) {
285 anetSetError(err, "Invalid bind address\n");
ed9b544e 286 close(s);
287 return ANET_ERR;
288 }
704bd093 289 if (anetListen(err,s,(struct sockaddr*)&sa,sizeof(sa)) == ANET_ERR)
ed9b544e 290 return ANET_ERR;
ed9b544e 291 return s;
292}
293
c61e6925
PN
294int anetUnixServer(char *err, char *path)
295{
296 int s;
297 struct sockaddr_un sa;
298
704bd093 299 if ((s = anetCreateSocket(err,AF_LOCAL)) == ANET_ERR)
c61e6925 300 return ANET_ERR;
704bd093 301
c61e6925
PN
302 memset(&sa,0,sizeof(sa));
303 sa.sun_family = AF_LOCAL;
304 strncpy(sa.sun_path,path,sizeof(sa.sun_path)-1);
704bd093 305 if (anetListen(err,s,(struct sockaddr*)&sa,sizeof(sa)) == ANET_ERR)
c61e6925 306 return ANET_ERR;
c61e6925
PN
307 return s;
308}
309
ab17b909 310static int anetGenericAccept(char *err, int s, struct sockaddr *sa, socklen_t *len) {
ed9b544e 311 int fd;
ed9b544e 312 while(1) {
ab17b909 313 fd = accept(s,sa,len);
ed9b544e 314 if (fd == -1) {
315 if (errno == EINTR)
316 continue;
317 else {
318 anetSetError(err, "accept: %s\n", strerror(errno));
319 return ANET_ERR;
320 }
321 }
322 break;
323 }
ab17b909
PN
324 return fd;
325}
326
327int anetTcpAccept(char *err, int s, char *ip, int *port) {
328 int fd;
329 struct sockaddr_in sa;
330 socklen_t salen = sizeof(sa);
331 if ((fd = anetGenericAccept(err,s,(struct sockaddr*)&sa,&salen)) == ANET_ERR)
332 return ANET_ERR;
333
ed9b544e 334 if (ip) strcpy(ip,inet_ntoa(sa.sin_addr));
335 if (port) *port = ntohs(sa.sin_port);
336 return fd;
337}
ab17b909 338
4fe83b55 339int anetUnixAccept(char *err, int s) {
ab17b909
PN
340 int fd;
341 struct sockaddr_un sa;
342 socklen_t salen = sizeof(sa);
343 if ((fd = anetGenericAccept(err,s,(struct sockaddr*)&sa,&salen)) == ANET_ERR)
344 return ANET_ERR;
345
ab17b909
PN
346 return fd;
347}