5 #include <sys/socket.h>
7 #include <netinet/tcp.h>
10 static memcached_return
set_hostinfo(memcached_server_st
*server
)
14 if ((h
= gethostbyname(server
->hostname
)) == NULL
)
16 return MEMCACHED_HOST_LOCKUP_FAILURE
;
19 server
->servAddr
.sin_family
= h
->h_addrtype
;
20 memcpy((char *) &server
->servAddr
.sin_addr
.s_addr
, h
->h_addr_list
[0], h
->h_length
);
21 server
->servAddr
.sin_port
= htons(server
->port
);
23 return MEMCACHED_SUCCESS
;
26 static memcached_return
unix_socket_connect(memcached_st
*ptr
, unsigned int server_key
)
28 struct sockaddr_un servAddr
;
31 if (ptr
->hosts
[server_key
].fd
== -1)
33 if ((ptr
->hosts
[server_key
].fd
= socket(AF_UNIX
, SOCK_STREAM
, 0)) < 0)
35 ptr
->cached_errno
= errno
;
36 return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE
;
39 memset(&servAddr
, 0, sizeof (struct sockaddr_un
));
40 servAddr
.sun_family
= AF_UNIX
;
41 strcpy(servAddr
.sun_path
, ptr
->hosts
[server_key
].hostname
); /* Copy filename */
43 addrlen
= strlen(servAddr
.sun_path
) + sizeof(servAddr
.sun_family
);
46 if (connect(ptr
->hosts
[server_key
].fd
,
47 (struct sockaddr
*)&servAddr
,
48 sizeof(servAddr
)) < 0)
51 /* We are spinning waiting on connect */
56 case EISCONN
: /* We were spinning waiting on connect */
59 ptr
->cached_errno
= errno
;
60 return MEMCACHED_ERRNO
;
65 return MEMCACHED_SUCCESS
;
68 static memcached_return
udp_connect(memcached_st
*ptr
, unsigned int server_key
)
70 if (ptr
->hosts
[server_key
].fd
== -1)
72 /* Old connection junk still is in the structure */
73 WATCHPOINT_ASSERT(ptr
->hosts
[server_key
].stack_responses
== 0);
76 If we have not allocated the hosts object.
77 Or if the cache has not been set.
79 if (ptr
->hosts
[server_key
].sockaddr_inited
== MEMCACHED_NOT_ALLOCATED
||
80 (!(ptr
->flags
& MEM_USE_CACHE_LOOKUPS
)))
84 rc
= set_hostinfo(&ptr
->hosts
[server_key
]);
85 if (rc
!= MEMCACHED_SUCCESS
)
88 ptr
->hosts
[server_key
].sockaddr_inited
= MEMCACHED_ALLOCATED
;
91 /* Create the socket */
92 if ((ptr
->hosts
[server_key
].fd
= socket(AF_INET
, SOCK_DGRAM
, 0)) < 0)
94 ptr
->cached_errno
= errno
;
95 return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE
;
99 return MEMCACHED_SUCCESS
;
102 static memcached_return
tcp_connect(memcached_st
*ptr
, unsigned int server_key
)
104 if (ptr
->hosts
[server_key
].fd
== -1)
106 /* Old connection junk still is in the structure */
107 WATCHPOINT_ASSERT(ptr
->hosts
[server_key
].stack_responses
== 0);
109 if (ptr
->hosts
[server_key
].servAddr
.sin_family
== 0)
113 rc
= set_hostinfo(&ptr
->hosts
[server_key
]);
114 if (rc
!= MEMCACHED_SUCCESS
)
118 /* Create the socket */
119 if ((ptr
->hosts
[server_key
].fd
= socket(AF_INET
, SOCK_STREAM
, 0)) < 0)
121 ptr
->cached_errno
= errno
;
122 return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE
;
125 /* For the moment, not getting a nonblocking mode will note be fatal */
126 if (ptr
->flags
& MEM_NO_BLOCK
)
130 flags
= fcntl(ptr
->hosts
[server_key
].fd
, F_GETFL
, 0);
132 (void)fcntl(ptr
->hosts
[server_key
].fd
, F_SETFL
, flags
| O_NONBLOCK
);
135 if (ptr
->flags
& MEM_TCP_NODELAY
)
139 setsockopt(ptr
->hosts
[server_key
].fd
, IPPROTO_TCP
, TCP_NODELAY
,
140 &flag
, (socklen_t
)sizeof(int));
145 setsockopt(ptr
->hosts
[server_key
].fd
, SOL_SOCKET
, SO_SNDBUF
,
146 &ptr
->send_size
, (socklen_t
)sizeof(int));
151 setsockopt(ptr
->hosts
[server_key
].fd
, SOL_SOCKET
, SO_SNDBUF
,
152 &ptr
->recv_size
, (socklen_t
)sizeof(int));
155 /* connect to server */
157 if (connect(ptr
->hosts
[server_key
].fd
,
158 (struct sockaddr
*)&ptr
->hosts
[server_key
].servAddr
,
159 sizeof(struct sockaddr
)) < 0)
162 /* We are spinning waiting on connect */
167 case EISCONN
: /* We were spinning waiting on connect */
170 ptr
->cached_errno
= errno
;
171 return MEMCACHED_ERRNO
;
175 WATCHPOINT_ASSERT(ptr
->hosts
[server_key
].stack_responses
== 0);
178 return MEMCACHED_SUCCESS
;
182 memcached_return
memcached_connect(memcached_st
*ptr
, unsigned int server_key
)
184 memcached_return rc
= MEMCACHED_NO_SERVERS
;
185 LIBMEMCACHED_MEMCACHED_CONNECT_START();
187 if (ptr
->connected
== ptr
->number_of_hosts
&& ptr
->number_of_hosts
)
188 return MEMCACHED_SUCCESS
;
190 if (ptr
->hosts
== NULL
|| ptr
->number_of_hosts
== 0)
191 return MEMCACHED_NO_SERVERS
;
193 /* We need to clean up the multi startup piece */
196 rc
= tcp_connect(ptr
, server_key
);
197 switch (ptr
->hosts
[server_key
].type
)
199 case MEMCACHED_CONNECTION_UNKNOWN
:
200 WATCHPOINT_ASSERT(0);
201 rc
= MEMCACHED_NOT_SUPPORTED
;
203 case MEMCACHED_CONNECTION_UDP
:
204 rc
= udp_connect(ptr
, server_key
);
206 case MEMCACHED_CONNECTION_TCP
:
207 rc
= tcp_connect(ptr
, server_key
);
209 case MEMCACHED_CONNECTION_UNIX_SOCKET
:
210 rc
= unix_socket_connect(ptr
, server_key
);
218 for (x
= 0; x
< ptr
->number_of_hosts
; x
++)
220 memcached_return possible_rc
;
222 possible_rc
= MEMCACHED_NOT_SUPPORTED
; /* Remove warning */
224 switch (ptr
->hosts
[x
].type
)
226 case MEMCACHED_CONNECTION_UNKNOWN
:
227 WATCHPOINT_ASSERT(0);
228 possible_rc
= MEMCACHED_NOT_SUPPORTED
;
230 case MEMCACHED_CONNECTION_UDP
:
231 possible_rc
= udp_connect(ptr
, x
);
233 case MEMCACHED_CONNECTION_TCP
:
234 possible_rc
= tcp_connect(ptr
, x
);
236 case MEMCACHED_CONNECTION_UNIX_SOCKET
:
237 possible_rc
= unix_socket_connect(ptr
, x
);
240 rc
= MEMCACHED_SUCCESS
;
242 if (possible_rc
!= MEMCACHED_SUCCESS
)
243 rc
= MEMCACHED_SOME_ERRORS
;
246 LIBMEMCACHED_MEMCACHED_CONNECT_END();