X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=lib%2Fmemcached_connect.c;h=381053950b16540dbc5d05925ebcc87031d369ab;hb=113160884d25c401dc352cfab093b5b7b1a4fde0;hp=105e7d20d82074f89f5689c854249376e6040f92;hpb=a081ca36c7fbf7c73e2a61d4f3c06ab0eff0571e;p=awesomized%2Flibmemcached diff --git a/lib/memcached_connect.c b/lib/memcached_connect.c index 105e7d20..38105395 100644 --- a/lib/memcached_connect.c +++ b/lib/memcached_connect.c @@ -1,41 +1,4 @@ -#include - -memcached_return memcached_server_add(memcached_st *ptr, char *hostname, unsigned int port) -{ - memcached_host_st *new_host_list; - char *new_hostname; - - if (!port) - port= MEMCACHED_DEFAULT_PORT; - - if (!hostname) - hostname= "localhost"; - - - new_host_list= (memcached_host_st *)realloc(ptr->hosts, sizeof(memcached_host_st) * (ptr->number_of_hosts+1)); - if (!new_host_list) - return MEMCACHED_MEMORY_ALLOCATION_FAILURE; - memset(&new_host_list[ptr->number_of_hosts], 0, sizeof(memcached_host_st)); - - if (!new_host_list) - return MEMCACHED_MEMORY_ALLOCATION_FAILURE; - - ptr->hosts= new_host_list; - - new_hostname= - (char *)malloc(sizeof(char) * (strlen(hostname)+1)); - if (!new_hostname) - return MEMCACHED_MEMORY_ALLOCATION_FAILURE; - - memset(new_hostname, 0, strlen(hostname)+1); - memcpy(new_hostname, hostname, strlen(hostname)); - ptr->hosts[ptr->number_of_hosts].hostname= new_hostname; - ptr->hosts[ptr->number_of_hosts].port= port; - ptr->hosts[ptr->number_of_hosts].fd= -1; - ptr->number_of_hosts++; - - return MEMCACHED_SUCCESS; -} +#include "common.h" memcached_return memcached_connect(memcached_st *ptr) { @@ -43,47 +6,46 @@ memcached_return memcached_connect(memcached_st *ptr) struct sockaddr_in localAddr, servAddr; struct hostent *h; - if (ptr->connected) + LIBMEMCACHED_MEMCACHED_CONNECT_START(); + + if (ptr->connected == ptr->number_of_hosts) return MEMCACHED_SUCCESS; if (!ptr->hosts) - { - memcached_return rc; - rc= memcached_server_add(ptr, NULL, 0); - - if (rc != MEMCACHED_SUCCESS) - return rc; - } - + return MEMCACHED_NO_SERVERS; for (x= 0; x < ptr->number_of_hosts; x++) { - if ((h= gethostbyname(ptr->hosts[x].hostname)) == NULL) - return MEMCACHED_HOST_LOCKUP_FAILURE; + if (ptr->hosts[x].fd == -1) + { + if ((h= gethostbyname(ptr->hosts[x].hostname)) == NULL) + return MEMCACHED_HOST_LOCKUP_FAILURE; - servAddr.sin_family= h->h_addrtype; - memcpy((char *) &servAddr.sin_addr.s_addr, h->h_addr_list[0], h->h_length); - servAddr.sin_port = htons(ptr->hosts[x].port); + servAddr.sin_family= h->h_addrtype; + memcpy((char *) &servAddr.sin_addr.s_addr, h->h_addr_list[0], h->h_length); + servAddr.sin_port = htons(ptr->hosts[x].port); - /* Create the socket */ - if ((ptr->hosts[0].fd= socket(AF_INET, SOCK_STREAM, 0)) < 0) - return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE; + /* Create the socket */ + if ((ptr->hosts[0].fd= socket(AF_INET, SOCK_STREAM, 0)) < 0) + return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE; - /* bind any port number */ - localAddr.sin_family = AF_INET; - localAddr.sin_addr.s_addr = htonl(INADDR_ANY); - localAddr.sin_port = htons(0); + /* bind any port number */ + localAddr.sin_family = AF_INET; + localAddr.sin_addr.s_addr = htonl(INADDR_ANY); + localAddr.sin_port = htons(0); - if (bind(ptr->hosts[0].fd, (struct sockaddr *) &localAddr, sizeof(localAddr)) < 0) - return(MEMCACHED_CONNECTION_BIND_FAILURE); + if (bind(ptr->hosts[0].fd, (struct sockaddr *) &localAddr, sizeof(localAddr)) < 0) + return(MEMCACHED_CONNECTION_BIND_FAILURE); - /* connect to server */ - if (connect(ptr->hosts[0].fd, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0) - return MEMCACHED_HOST_LOCKUP_FAILURE; - } + /* connect to server */ + if (connect(ptr->hosts[0].fd, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0) + return MEMCACHED_HOST_LOCKUP_FAILURE; - ptr->connected= 1; + ptr->connected++; + } + } + LIBMEMCACHED_MEMCACHED_CONNECT_END(); return MEMCACHED_SUCCESS; }