6 static memcached_return
set_hostinfo(memcached_server_st
*server
)
11 char str_port
[NI_MAXSERV
];
13 sprintf(str_port
, "%u", server
->port
);
15 memset(&hints
, 0, sizeof(hints
));
17 // hints.ai_family= AF_INET;
18 if (server
->type
== MEMCACHED_CONNECTION_UDP
)
20 hints
.ai_protocol
= IPPROTO_UDP
;
21 hints
.ai_socktype
= SOCK_DGRAM
;
25 hints
.ai_socktype
= SOCK_STREAM
;
26 hints
.ai_protocol
= IPPROTO_TCP
;
29 e
= getaddrinfo(server
->hostname
, str_port
, &hints
, &ai
);
32 WATCHPOINT_STRING(server
->hostname
);
33 WATCHPOINT_STRING(gai_strerror(e
));
34 return MEMCACHED_HOST_LOOKUP_FAILURE
;
37 if (server
->address_info
)
39 freeaddrinfo(server
->address_info
);
40 server
->address_info
= NULL
;
42 server
->address_info
= ai
;
44 return MEMCACHED_SUCCESS
;
47 static memcached_return
set_socket_options(memcached_server_st
*ptr
)
49 WATCHPOINT_ASSERT(ptr
->fd
!= -1);
51 if (ptr
->type
== MEMCACHED_CONNECTION_UDP
)
52 return MEMCACHED_SUCCESS
;
55 if (ptr
->root
->snd_timeout
)
58 struct timeval waittime
;
61 waittime
.tv_usec
= ptr
->root
->snd_timeout
;
63 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_SNDTIMEO
,
64 &waittime
, (socklen_t
)sizeof(struct timeval
));
65 WATCHPOINT_ASSERT(error
== 0);
70 if (ptr
->root
->rcv_timeout
)
73 struct timeval waittime
;
76 waittime
.tv_usec
= ptr
->root
->rcv_timeout
;
78 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_RCVTIMEO
,
79 &waittime
, (socklen_t
)sizeof(struct timeval
));
80 WATCHPOINT_ASSERT(error
== 0);
89 linger
.l_linger
= MEMCACHED_DEFAULT_TIMEOUT
;
90 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_LINGER
,
91 &linger
, (socklen_t
)sizeof(struct linger
));
92 WATCHPOINT_ASSERT(error
== 0);
95 if (ptr
->root
->flags
& MEM_TCP_NODELAY
)
100 error
= setsockopt(ptr
->fd
, IPPROTO_TCP
, TCP_NODELAY
,
101 &flag
, (socklen_t
)sizeof(int));
102 WATCHPOINT_ASSERT(error
== 0);
105 if (ptr
->root
->send_size
)
109 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_SNDBUF
,
110 &ptr
->root
->send_size
, (socklen_t
)sizeof(int));
111 WATCHPOINT_ASSERT(error
== 0);
114 if (ptr
->root
->recv_size
)
118 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_RCVBUF
,
119 &ptr
->root
->recv_size
, (socklen_t
)sizeof(int));
120 WATCHPOINT_ASSERT(error
== 0);
123 /* For the moment, not getting a nonblocking mode will not be fatal */
124 if (ptr
->root
->flags
& MEM_NO_BLOCK
)
128 flags
= fcntl(ptr
->fd
, F_GETFL
, 0);
129 unlikely (flags
!= -1)
131 (void)fcntl(ptr
->fd
, F_SETFL
, flags
| O_NONBLOCK
);
135 return MEMCACHED_SUCCESS
;
138 static memcached_return
unix_socket_connect(memcached_server_st
*ptr
)
140 struct sockaddr_un servAddr
;
145 if ((ptr
->fd
= socket(AF_UNIX
, SOCK_STREAM
, 0)) < 0)
147 ptr
->cached_errno
= errno
;
148 return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE
;
151 memset(&servAddr
, 0, sizeof (struct sockaddr_un
));
152 servAddr
.sun_family
= AF_UNIX
;
153 strcpy(servAddr
.sun_path
, ptr
->hostname
); /* Copy filename */
155 addrlen
= strlen(servAddr
.sun_path
) + sizeof(servAddr
.sun_family
);
159 (struct sockaddr
*)&servAddr
,
160 sizeof(servAddr
)) < 0)
168 case EISCONN
: /* We were spinning waiting on connect */
171 WATCHPOINT_ERRNO(errno
);
172 ptr
->cached_errno
= errno
;
173 return MEMCACHED_ERRNO
;
178 WATCHPOINT_ASSERT(ptr
->fd
!= -1);
179 return MEMCACHED_SUCCESS
;
182 static memcached_return
network_connect(memcached_server_st
*ptr
)
186 struct addrinfo
*use
;
188 if (ptr
->root
->server_failure_limit
!= 0)
190 if (ptr
->server_failure_counter
>= ptr
->root
->server_failure_limit
)
192 memcached_server_remove(ptr
);
193 return MEMCACHED_FAILURE
;
197 if (ptr
->sockaddr_inited
||
198 (!(ptr
->root
->flags
& MEM_USE_CACHE_LOOKUPS
)))
202 rc
= set_hostinfo(ptr
);
203 if (rc
!= MEMCACHED_SUCCESS
)
205 ptr
->sockaddr_inited
= true;
208 use
= ptr
->address_info
;
209 /* Create the socket */
212 /* Memcache server does not support IPV6 in udp mode, so skip if not ipv4 */
213 if (ptr
->type
== MEMCACHED_CONNECTION_UDP
&& use
->ai_family
!= AF_INET
)
219 if ((ptr
->fd
= socket(use
->ai_family
,
221 use
->ai_protocol
)) < 0)
223 ptr
->cached_errno
= errno
;
224 WATCHPOINT_ERRNO(errno
);
225 return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE
;
228 (void)set_socket_options(ptr
);
231 if (ptr
->root
->connect_timeout
)
233 flags
= fcntl(ptr
->fd
, F_GETFL
, 0);
234 if (flags
!= -1 && !(flags
& O_NONBLOCK
))
235 (void)fcntl(ptr
->fd
, F_SETFL
, flags
| O_NONBLOCK
);
238 /* connect to server */
239 while (ptr
->fd
!= -1 &&
240 connect(ptr
->fd
, use
->ai_addr
, use
->ai_addrlen
) < 0)
242 ptr
->cached_errno
= errno
;
243 if (errno
== EINPROGRESS
|| /* nonblocking mode - first return, */
244 errno
== EALREADY
) /* nonblocking mode - subsequent returns */
246 struct pollfd fds
[1] = { [0].fd
= ptr
->fd
, [0].events
= POLLOUT
};
247 int error
= poll(fds
, 1, ptr
->root
->connect_timeout
);
249 if (error
!= 1 || fds
[0].revents
& POLLERR
)
251 if (fds
[0].revents
& POLLERR
)
254 int len
= sizeof (err
);
255 (void)getsockopt(ptr
->fd
, SOL_SOCKET
, SO_ERROR
, &err
, &len
);
256 ptr
->cached_errno
= errno
;
259 (void)close(ptr
->fd
);
263 else if (errno
== EISCONN
) /* we are connected :-) */
267 else if (errno
!= EINTR
)
269 (void)close(ptr
->fd
);
278 if (ptr
->root
->connect_timeout
&& (flags
& O_NONBLOCK
) == 0)
279 (void)fcntl(ptr
->fd
, F_SETFL
, flags
& ~O_NONBLOCK
);
281 WATCHPOINT_ASSERT(ptr
->cursor_active
== 0);
282 ptr
->server_failure_counter
= 0;
283 return MEMCACHED_SUCCESS
;
291 /* Failed to connect. schedule next retry */
292 if (ptr
->root
->retry_timeout
)
294 struct timeval next_time
;
296 if (gettimeofday(&next_time
, NULL
) == 0)
297 ptr
->next_retry
= next_time
.tv_sec
+ ptr
->root
->retry_timeout
;
299 ptr
->server_failure_counter
+= 1;
300 return MEMCACHED_ERRNO
; /* The last error should be from connect() */
303 ptr
->server_failure_counter
= 0;
304 return MEMCACHED_SUCCESS
; /* The last error should be from connect() */
308 memcached_return
memcached_connect(memcached_server_st
*ptr
)
310 memcached_return rc
= MEMCACHED_NO_SERVERS
;
311 LIBMEMCACHED_MEMCACHED_CONNECT_START();
313 if (ptr
->root
->retry_timeout
)
315 struct timeval next_time
;
317 gettimeofday(&next_time
, NULL
);
318 if (next_time
.tv_sec
< ptr
->next_retry
)
319 return MEMCACHED_TIMEOUT
;
321 /* We need to clean up the multi startup piece */
324 case MEMCACHED_CONNECTION_UNKNOWN
:
325 WATCHPOINT_ASSERT(0);
326 rc
= MEMCACHED_NOT_SUPPORTED
;
328 case MEMCACHED_CONNECTION_UDP
:
329 case MEMCACHED_CONNECTION_TCP
:
330 rc
= network_connect(ptr
);
332 case MEMCACHED_CONNECTION_UNIX_SOCKET
:
333 rc
= unix_socket_connect(ptr
);
336 WATCHPOINT_ASSERT(0);
339 LIBMEMCACHED_MEMCACHED_CONNECT_END();