X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=lib%2Fmemcached_connect.c;h=381053950b16540dbc5d05925ebcc87031d369ab;hb=113160884d25c401dc352cfab093b5b7b1a4fde0;hp=1be786e1c6149d4480fdef280ee2d115ec774285;hpb=1fc4b1ae18810551553837b4c67c6075f1ad5941;p=awesomized%2Flibmemcached diff --git a/lib/memcached_connect.c b/lib/memcached_connect.c index 1be786e1..38105395 100644 --- a/lib/memcached_connect.c +++ b/lib/memcached_connect.c @@ -1,87 +1,51 @@ -#include - -memcached_return memcached_server_add(memcached_st *ptr, char *hostname, unsigned int port) -{ - if (!port) - port= MEMCACHED_DEFAULT_PORT; - - if (!hostname) - hostname= "localhost"; - - - ptr->hosts= (memcached_host_st *)realloc(ptr->hosts, sizeof(memcached_host_st) * (size_t)ptr->number_of_hosts); - ptr->hosts[ptr->number_of_hosts].hostname= - (char *)malloc(sizeof(char) * strlen(hostname)); - memcpy(ptr->hosts[ptr->number_of_hosts].hostname, hostname, strlen(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) { unsigned int x; struct sockaddr_in localAddr, servAddr; struct hostent *h; - memcached_host_st *host_ptr; - 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) + if (ptr->hosts[x].fd == -1) { - fprintf(stderr, "unknown host '%s'\n", ptr->hosts[x].hostname); - return MEMCACHED_HOST_LOCKUP_FAILURE; - } + 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) - { - fprintf(stderr, "cannot open socket"); - 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) - { - fprintf(stderr, "cannot bind port TCP %u\n", ptr->hosts[x].port); - 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) - { - fprintf(stderr, "cannot connect to host '%s' (%u) (error: %s)\n", ptr->hosts[x].hostname, - ptr->hosts[x].port, - strerror(errno)); - 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++; } } - - ptr->connected= 1; + LIBMEMCACHED_MEMCACHED_CONNECT_END(); return MEMCACHED_SUCCESS; }