git.m6w6.name Git - awesomized/libmemcached/blob - libmemcached/memcached_connect.c

   1 #include "common.h"
   2 #include <netdb.h>
   3 #include <poll.h>
   4 #include <sys/time.h>
   5
   6 static memcached_return set_hostinfo(memcached_server_st *server)
   7 {
   8   struct addrinfo *ai;
   9   struct addrinfo hints;
  10   int e;
  11   char str_port[NI_MAXSERV];
  12
  13   sprintf(str_port, "%u", server->port);
  14
  15   memset(&hints, 0, sizeof(hints));
  16
  17  // hints.ai_family= AF_INET;
  18   if (server->type == MEMCACHED_CONNECTION_UDP)
  19   {
  20     hints.ai_protocol= IPPROTO_UDP;
  21     hints.ai_socktype= SOCK_DGRAM;
  22   }
  23   else
  24   {
  25     hints.ai_socktype= SOCK_STREAM;
  26     hints.ai_protocol= IPPROTO_TCP;
  27   }
  28
  29   e= getaddrinfo(server->hostname, str_port, &hints, &ai);
  30   if (e != 0)
  31   {
  32     WATCHPOINT_STRING(server->hostname);
  33     WATCHPOINT_STRING(gai_strerror(e));
  34     return MEMCACHED_HOST_LOOKUP_FAILURE;
  35   }
  36
  37   if (server->address_info)
  38   {
  39     freeaddrinfo(server->address_info);
  40     server->address_info= NULL;
  41   }
  42   server->address_info= ai;
  43
  44   return MEMCACHED_SUCCESS;
  45 }
  46
  47 static memcached_return set_socket_options(memcached_server_st *ptr)
  48 {
  49   WATCHPOINT_ASSERT(ptr->fd != -1);
  50
  51   if (ptr->type == MEMCACHED_CONNECTION_UDP)
  52     return MEMCACHED_SUCCESS;
  53
  54 #ifdef HAVE_SNDTIMEO
  55   if (ptr->root->snd_timeout)
  56   {
  57     int error;
  58     struct timeval waittime;
  59
  60     waittime.tv_sec= 0;
  61     waittime.tv_usec= ptr->root->snd_timeout;
  62
  63     error= setsockopt(ptr->fd, SOL_SOCKET, SO_SNDTIMEO,
  64                       &waittime, (socklen_t)sizeof(struct timeval));
  65     WATCHPOINT_ASSERT(error == 0);
  66   }
  67 #endif
  68
  69 #ifdef HAVE_RCVTIMEO
  70   if (ptr->root->rcv_timeout)
  71   {
  72     int error;
  73     struct timeval waittime;
  74
  75     waittime.tv_sec= 0;
  76     waittime.tv_usec= ptr->root->rcv_timeout;
  77
  78     error= setsockopt(ptr->fd, SOL_SOCKET, SO_RCVTIMEO,
  79                       &waittime, (socklen_t)sizeof(struct timeval));
  80     WATCHPOINT_ASSERT(error == 0);
  81   }
  82 #endif
  83
  84   {
  85     int error;
  86     struct linger linger;
  87
  88     linger.l_onoff= 1;
  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);
  93   }
  94
  95   if (ptr->root->flags & MEM_TCP_NODELAY)
  96   {
  97     int flag= 1;
  98     int error;
  99
 100     error= setsockopt(ptr->fd, IPPROTO_TCP, TCP_NODELAY,
 101                       &flag, (socklen_t)sizeof(int));
 102     WATCHPOINT_ASSERT(error == 0);
 103   }
 104
 105   if (ptr->root->send_size)
 106   {
 107     int error;
 108
 109     error= setsockopt(ptr->fd, SOL_SOCKET, SO_SNDBUF,
 110                       &ptr->root->send_size, (socklen_t)sizeof(int));
 111     WATCHPOINT_ASSERT(error == 0);
 112   }
 113
 114   if (ptr->root->recv_size)
 115   {
 116     int error;
 117
 118     error= setsockopt(ptr->fd, SOL_SOCKET, SO_RCVBUF,
 119                       &ptr->root->recv_size, (socklen_t)sizeof(int));
 120     WATCHPOINT_ASSERT(error == 0);
 121   }
 122
 123   /* For the moment, not getting a nonblocking mode will not be fatal */
 124   if (ptr->root->flags & MEM_NO_BLOCK)
 125   {
 126     int flags;
 127
 128     flags= fcntl(ptr->fd, F_GETFL, 0);
 129     unlikely (flags != -1)
 130     {
 131       (void)fcntl(ptr->fd, F_SETFL, flags | O_NONBLOCK);
 132     }
 133   }
 134
 135   return MEMCACHED_SUCCESS;
 136 }
 137
 138 static memcached_return unix_socket_connect(memcached_server_st *ptr)
 139 {
 140   struct sockaddr_un servAddr;
 141   socklen_t addrlen;
 142
 143   if (ptr->fd == -1)
 144   {
 145     if ((ptr->fd= socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
 146     {
 147       ptr->cached_errno= errno;
 148       return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE;
 149     }
 150
 151     memset(&servAddr, 0, sizeof (struct sockaddr_un));
 152     servAddr.sun_family= AF_UNIX;
 153     strcpy(servAddr.sun_path, ptr->hostname); /* Copy filename */
 154
 155     addrlen= strlen(servAddr.sun_path) + sizeof(servAddr.sun_family);
 156
 157 test_connect:
 158     if (connect(ptr->fd,
 159                 (struct sockaddr *)&servAddr,
 160                 sizeof(servAddr)) < 0)
 161     {
 162       switch (errno)
 163       {
 164       case EINPROGRESS:
 165       case EALREADY:
 166       case EINTR:
 167         goto test_connect;
 168       case EISCONN: /* We were spinning waiting on connect */
 169         break;
 170       default:
 171         WATCHPOINT_ERRNO(errno);
 172         ptr->cached_errno= errno;
 173         return MEMCACHED_ERRNO;
 174       }
 175     }
 176   }
 177
 178   WATCHPOINT_ASSERT(ptr->fd != -1);
 179   return MEMCACHED_SUCCESS;
 180 }
 181
 182 static memcached_return network_connect(memcached_server_st *ptr)
 183 {
 184   if (ptr->fd == -1)
 185   {
 186     struct addrinfo *use;
 187
 188     if (ptr->root->server_failure_limit != 0)
 189     {
 190       if (ptr->server_failure_counter >= ptr->root->server_failure_limit)
 191       {
 192           memcached_server_remove(ptr);
 193           return MEMCACHED_FAILURE;
 194       }
 195     }
 196
 197     if (!ptr->sockaddr_inited ||
 198         (!(ptr->root->flags & MEM_USE_CACHE_LOOKUPS)))
 199     {
 200       memcached_return rc;
 201
 202       rc= set_hostinfo(ptr);
 203       if (rc != MEMCACHED_SUCCESS)
 204         return rc;
 205       ptr->sockaddr_inited= true;
 206     }
 207
 208     use= ptr->address_info;
 209     /* Create the socket */
 210     while (use != NULL)
 211     {
 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)
 214       {
 215         use= use->ai_next;
 216         continue;
 217       }
 218
 219       if ((ptr->fd= socket(use->ai_family,
 220                            use->ai_socktype,
 221                            use->ai_protocol)) < 0)
 222       {
 223         ptr->cached_errno= errno;
 224         WATCHPOINT_ERRNO(errno);
 225         return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE;
 226       }
 227
 228       (void)set_socket_options(ptr);
 229
 230       int flags;
 231       if (ptr->root->connect_timeout)
 232       {
 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);
 236       }
 237
 238       /* connect to server */
 239       while (ptr->fd != -1 &&
 240              connect(ptr->fd, use->ai_addr, use->ai_addrlen) < 0)
 241       {
 242         ptr->cached_errno= errno;
 243         if (errno == EINPROGRESS || /* nonblocking mode - first return, */
 244             errno == EALREADY) /* nonblocking mode - subsequent returns */
 245         {
 246           struct pollfd fds[1] = { [0].fd = ptr->fd, [0].events = POLLOUT };
 247           int error= poll(fds, 1, ptr->root->connect_timeout);
 248
 249           if (error != 1 || fds[0].revents & POLLERR)
 250           {
 251             if (fds[0].revents & POLLERR)
 252             {
 253               int err;
 254               int len = sizeof (err);
 255               (void)getsockopt(ptr->fd, SOL_SOCKET, SO_ERROR, &err, &len);
 256               ptr->cached_errno= errno;
 257             }
 258
 259             (void)close(ptr->fd);
 260             ptr->fd= -1;
 261           }
 262         }
 263         else if (errno == EISCONN) /* we are connected :-) */
 264         {
 265           break;
 266         }
 267         else if (errno != EINTR)
 268         {
 269           (void)close(ptr->fd);
 270           ptr->fd= -1;
 271           break;
 272         }
 273       }
 274
 275       if (ptr->fd != -1)
 276       {
 277         /* restore flags */
 278         if (ptr->root->connect_timeout && (flags & O_NONBLOCK) == 0)
 279           (void)fcntl(ptr->fd, F_SETFL, flags & ~O_NONBLOCK);
 280
 281         WATCHPOINT_ASSERT(ptr->cursor_active == 0);
 282         ptr->server_failure_counter= 0;
 283         return MEMCACHED_SUCCESS;
 284       }
 285       use = use->ai_next;
 286     }
 287   }
 288
 289   if (ptr->fd == -1)
 290   {
 291     /* Failed to connect. schedule next retry */
 292     if (ptr->root->retry_timeout)
 293     {
 294       struct timeval next_time;
 295
 296       if (gettimeofday(&next_time, NULL) == 0)
 297         ptr->next_retry= next_time.tv_sec + ptr->root->retry_timeout;
 298     }
 299     ptr->server_failure_counter+= 1;
 300     return MEMCACHED_ERRNO; /* The last error should be from connect() */
 301   }
 302
 303   ptr->server_failure_counter= 0;
 304   return MEMCACHED_SUCCESS; /* The last error should be from connect() */
 305 }
 306
 307
 308 memcached_return memcached_connect(memcached_server_st *ptr)
 309 {
 310   memcached_return rc= MEMCACHED_NO_SERVERS;
 311   LIBMEMCACHED_MEMCACHED_CONNECT_START();
 312
 313   if (ptr->root->retry_timeout)
 314   {
 315     struct timeval next_time;
 316
 317     gettimeofday(&next_time, NULL);
 318     if (next_time.tv_sec < ptr->next_retry)
 319       return MEMCACHED_TIMEOUT;
 320   }
 321   /* We need to clean up the multi startup piece */
 322   switch (ptr->type)
 323   {
 324   case MEMCACHED_CONNECTION_UNKNOWN:
 325     WATCHPOINT_ASSERT(0);
 326     rc= MEMCACHED_NOT_SUPPORTED;
 327     break;
 328   case MEMCACHED_CONNECTION_UDP:
 329   case MEMCACHED_CONNECTION_TCP:
 330     rc= network_connect(ptr);
 331     break;
 332   case MEMCACHED_CONNECTION_UNIX_SOCKET:
 333     rc= unix_socket_connect(ptr);
 334     break;
 335   default:
 336     WATCHPOINT_ASSERT(0);
 337   }
 338
 339   LIBMEMCACHED_MEMCACHED_CONNECT_END();
 340
 341   return rc;
 342 }