6 static memcached_return_t
set_hostinfo(memcached_server_st
*server
)
11 char str_port
[NI_MAXSERV
];
13 snprintf(str_port
, NI_MAXSERV
, "%u", (uint32_t)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_t
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);
67 return MEMCACHED_FAILURE
;
72 if (ptr
->root
->rcv_timeout
)
75 struct timeval waittime
;
78 waittime
.tv_usec
= ptr
->root
->rcv_timeout
;
80 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_RCVTIMEO
,
81 &waittime
, (socklen_t
)sizeof(struct timeval
));
82 WATCHPOINT_ASSERT(error
== 0);
84 return MEMCACHED_FAILURE
;
88 if (ptr
->root
->flags
.no_block
)
94 linger
.l_linger
= 0; /* By default on close() just drop the socket */
95 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_LINGER
,
96 &linger
, (socklen_t
)sizeof(struct linger
));
97 WATCHPOINT_ASSERT(error
== 0);
99 return MEMCACHED_FAILURE
;
102 if (ptr
->root
->flags
.tcp_nodelay
)
107 error
= setsockopt(ptr
->fd
, IPPROTO_TCP
, TCP_NODELAY
,
108 &flag
, (socklen_t
)sizeof(int));
109 WATCHPOINT_ASSERT(error
== 0);
111 return MEMCACHED_FAILURE
;
114 if (ptr
->root
->flags
.tcp_keepalive
)
119 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_KEEPALIVE
,
120 &flag
, (socklen_t
)sizeof(int));
121 WATCHPOINT_ASSERT(error
== 0);
123 return MEMCACHED_FAILURE
;
127 if (ptr
->root
->tcp_keepidle
> 0)
131 error
= setsockopt(ptr
->fd
, IPPROTO_TCP
, TCP_KEEPIDLE
,
132 &ptr
->root
->tcp_keepidle
, (socklen_t
)sizeof(int));
133 WATCHPOINT_ASSERT(error
== 0);
135 return MEMCACHED_FAILURE
;
139 if (ptr
->root
->send_size
> 0)
143 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_SNDBUF
,
144 &ptr
->root
->send_size
, (socklen_t
)sizeof(int));
145 WATCHPOINT_ASSERT(error
== 0);
147 return MEMCACHED_FAILURE
;
150 if (ptr
->root
->recv_size
> 0)
154 error
= setsockopt(ptr
->fd
, SOL_SOCKET
, SO_RCVBUF
,
155 &ptr
->root
->recv_size
, (socklen_t
)sizeof(int));
156 WATCHPOINT_ASSERT(error
== 0);
158 return MEMCACHED_FAILURE
;
161 /* libmemcached will always use nonblocking IO to avoid write deadlocks */
165 flags
= fcntl(ptr
->fd
, F_GETFL
, 0);
166 while (flags
== -1 && (errno
== EINTR
|| errno
== EAGAIN
));
168 unlikely (flags
== -1)
170 return MEMCACHED_CONNECTION_FAILURE
;
172 else if ((flags
& O_NONBLOCK
) == 0)
177 rval
= fcntl(ptr
->fd
, F_SETFL
, flags
| O_NONBLOCK
);
178 while (rval
== -1 && (errno
== EINTR
|| errno
== EAGAIN
));
180 unlikely (rval
== -1)
182 return MEMCACHED_CONNECTION_FAILURE
;
186 return MEMCACHED_SUCCESS
;
189 static memcached_return_t
unix_socket_connect(memcached_server_st
*ptr
)
191 struct sockaddr_un servAddr
;
195 if ((ptr
->fd
= socket(AF_UNIX
, SOCK_STREAM
, 0)) < 0)
197 ptr
->cached_errno
= errno
;
198 return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE
;
201 memset(&servAddr
, 0, sizeof (struct sockaddr_un
));
202 servAddr
.sun_family
= AF_UNIX
;
203 strcpy(servAddr
.sun_path
, ptr
->hostname
); /* Copy filename */
207 (struct sockaddr
*)&servAddr
,
208 sizeof(servAddr
)) < 0)
216 case EISCONN
: /* We were spinning waiting on connect */
219 WATCHPOINT_ERRNO(errno
);
220 ptr
->cached_errno
= errno
;
221 return MEMCACHED_ERRNO
;
226 WATCHPOINT_ASSERT(ptr
->fd
!= -1);
227 return MEMCACHED_SUCCESS
;
230 static memcached_return_t
network_connect(memcached_server_st
*ptr
)
234 struct addrinfo
*use
;
236 WATCHPOINT_ASSERT(ptr
->cursor_active
== 0);
238 if (! ptr
->options
.sockaddr_inited
||
239 (!(ptr
->root
->flags
.use_cache_lookups
)))
241 memcached_return_t rc
;
243 rc
= set_hostinfo(ptr
);
244 if (rc
!= MEMCACHED_SUCCESS
)
246 ptr
->options
.sockaddr_inited
= true;
249 use
= ptr
->address_info
;
250 /* Create the socket */
253 /* Memcache server does not support IPV6 in udp mode, so skip if not ipv4 */
254 if (ptr
->type
== MEMCACHED_CONNECTION_UDP
&& use
->ai_family
!= AF_INET
)
260 if ((ptr
->fd
= socket(use
->ai_family
,
262 use
->ai_protocol
)) < 0)
264 ptr
->cached_errno
= errno
;
265 WATCHPOINT_ERRNO(errno
);
266 return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE
;
269 (void)set_socket_options(ptr
);
271 /* connect to server */
272 while (ptr
->fd
!= -1 &&
273 connect(ptr
->fd
, use
->ai_addr
, use
->ai_addrlen
) < 0)
275 ptr
->cached_errno
= errno
;
276 if (errno
== EINPROGRESS
|| /* nonblocking mode - first return, */
277 errno
== EALREADY
) /* nonblocking mode - subsequent returns */
279 struct pollfd fds
[1];
281 fds
[0].events
= POLLOUT
;
282 int error
= poll(fds
, 1, ptr
->root
->connect_timeout
);
284 if (error
!= 1 || fds
[0].revents
& POLLERR
)
286 if (fds
[0].revents
& POLLERR
)
289 socklen_t len
= sizeof (err
);
290 (void)getsockopt(ptr
->fd
, SOL_SOCKET
, SO_ERROR
, &err
, &len
);
291 ptr
->cached_errno
= (err
== 0) ? errno
: err
;
294 (void)close(ptr
->fd
);
298 else if (errno
== EISCONN
) /* we are connected :-) */
302 else if (errno
!= EINTR
)
304 (void)close(ptr
->fd
);
310 #ifdef LIBMEMCACHED_WITH_SASL_SUPPORT
311 if (ptr
->fd
!= -1 && ptr
->root
->sasl
.callbacks
!= NULL
)
313 memcached_return rc
= memcached_sasl_authenticate_connection(ptr
);
314 if (rc
!= MEMCACHED_SUCCESS
)
316 (void)close(ptr
->fd
);
326 ptr
->server_failure_counter
= 0;
327 return MEMCACHED_SUCCESS
;
335 /* Failed to connect. schedule next retry */
336 if (ptr
->root
->retry_timeout
)
338 struct timeval next_time
;
340 if (gettimeofday(&next_time
, NULL
) == 0)
341 ptr
->next_retry
= next_time
.tv_sec
+ ptr
->root
->retry_timeout
;
343 ptr
->server_failure_counter
++;
344 if (ptr
->cached_errno
== 0)
345 return MEMCACHED_TIMEOUT
;
347 return MEMCACHED_ERRNO
; /* The last error should be from connect() */
350 ptr
->server_failure_counter
= 0;
351 return MEMCACHED_SUCCESS
; /* The last error should be from connect() */
355 memcached_return_t
memcached_connect(memcached_server_write_instance_st ptr
)
357 memcached_return_t rc
= MEMCACHED_NO_SERVERS
;
358 LIBMEMCACHED_MEMCACHED_CONNECT_START();
360 /* both retry_timeout and server_failure_limit must be set in order to delay retrying a server on error. */
361 WATCHPOINT_ASSERT(ptr
->root
);
362 if (ptr
->root
->retry_timeout
&& ptr
->root
->server_failure_limit
)
364 struct timeval curr_time
;
366 gettimeofday(&curr_time
, NULL
);
368 /* if we've had too many consecutive errors on this server, mark it dead. */
369 if (ptr
->server_failure_counter
>= ptr
->root
->server_failure_limit
)
371 ptr
->next_retry
= curr_time
.tv_sec
+ ptr
->root
->retry_timeout
;
372 ptr
->server_failure_counter
= 0;
375 if (curr_time
.tv_sec
< ptr
->next_retry
)
377 memcached_st
*root
= (memcached_st
*)ptr
->root
;
378 // @todo fix this by fixing behavior to no longer make use of
380 if (memcached_behavior_get(root
, MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
))
382 run_distribution(root
);
385 root
->last_disconnected_server
= ptr
;
387 return MEMCACHED_SERVER_MARKED_DEAD
;
391 /* We need to clean up the multi startup piece */
394 case MEMCACHED_CONNECTION_UNKNOWN
:
395 WATCHPOINT_ASSERT(0);
396 rc
= MEMCACHED_NOT_SUPPORTED
;
398 case MEMCACHED_CONNECTION_UDP
:
399 case MEMCACHED_CONNECTION_TCP
:
400 rc
= network_connect(ptr
);
402 case MEMCACHED_CONNECTION_UNIX_SOCKET
:
403 rc
= unix_socket_connect(ptr
);
405 case MEMCACHED_CONNECTION_MAX
:
407 WATCHPOINT_ASSERT(0);
410 unlikely ( rc
!= MEMCACHED_SUCCESS
)
412 //@todo create interface around last_discontected_server
413 memcached_st
*root
= (memcached_st
*)ptr
->root
;
414 root
->last_disconnected_server
= ptr
;
417 LIBMEMCACHED_MEMCACHED_CONNECT_END();