git.m6w6.name Git - awesomized/libmemcached/blob - lib/memcached_behavior.c

   1 #include "common.h"
   2 #include <sys/types.h>
   3 #include <sys/socket.h>
   4 #include <netinet/tcp.h>
   5
   6 /*
   7   This function is used to modify the behabior of running client.
   8
   9   We quit all connections so we can reset the sockets.
  10 */
  11
  12 void set_behavior_flag(memcached_st *ptr, memcached_flags temp_flag, void *data)
  13 {
  14   unsigned int *truefalse= (unsigned int *)data;
  15
  16   memcached_quit(ptr);
  17   if (truefalse)
  18     ptr->flags|= temp_flag;
  19   else
  20     ptr->flags+= temp_flag;
  21 }
  22
  23 memcached_return memcached_behavior_set(memcached_st *ptr,
  24                                         memcached_behavior flag,
  25                                         void *data)
  26 {
  27   switch (flag)
  28   {
  29   case MEMCACHED_BEHAVIOR_NO_BLOCK:
  30     set_behavior_flag(ptr, MEM_NO_BLOCK, data);
  31     break;
  32   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
  33     set_behavior_flag(ptr, MEM_TCP_NODELAY, data);
  34     break;
  35   case MEMCACHED_BEHAVIOR_MD5_HASHING:
  36     set_behavior_flag(ptr, MEM_USE_MD5, data);
  37     break;
  38   case MEMCACHED_BEHAVIOR_CRC_HASHING:
  39     set_behavior_flag(ptr, MEM_USE_CRC, data);
  40     break;
  41   case MEMCACHED_BEHAVIOR_KETAMA:
  42     set_behavior_flag(ptr, MEM_USE_KETAMA, data);
  43     break;
  44   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
  45     {
  46       ptr->send_size= (*((int *)data));
  47       memcached_quit(ptr);
  48       break;
  49     }
  50   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
  51     {
  52       ptr->recv_size= (*((int *)data));
  53       memcached_quit(ptr);
  54       break;
  55     }
  56
  57
  58   }
  59
  60   return MEMCACHED_SUCCESS;
  61 }
  62
  63 unsigned long long memcached_behavior_get(memcached_st *ptr,
  64                                           memcached_behavior flag)
  65 {
  66   memcached_flags temp_flag= 0;
  67
  68   switch (flag)
  69   {
  70   case MEMCACHED_BEHAVIOR_NO_BLOCK:
  71     temp_flag= MEM_NO_BLOCK;
  72     break;
  73   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
  74     temp_flag= MEM_TCP_NODELAY;
  75     break;
  76   case MEMCACHED_BEHAVIOR_MD5_HASHING:
  77     temp_flag= MEM_USE_MD5;
  78     break;
  79   case MEMCACHED_BEHAVIOR_CRC_HASHING:
  80     temp_flag= MEM_USE_CRC;
  81     break;
  82   case MEMCACHED_BEHAVIOR_KETAMA:
  83     temp_flag= MEM_USE_KETAMA;
  84     break;
  85   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
  86     {
  87       int sock_size;
  88       socklen_t sock_length= sizeof(int);
  89
  90       /* We just try the first host, and if it is down we return zero */
  91       if ((memcached_connect(ptr, 0)) != MEMCACHED_SUCCESS)
  92         return 0;
  93
  94       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
  95                      SO_SNDBUF, &sock_size, &sock_length))
  96         return 0; /* Zero means error */
  97
  98       return sock_size;
  99     }
 100   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
 101     {
 102       int sock_size;
 103       socklen_t sock_length= sizeof(int);
 104
 105       /* We just try the first host, and if it is down we return zero */
 106       if ((memcached_connect(ptr, 0)) != MEMCACHED_SUCCESS)
 107         return 0;
 108
 109       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 110                      SO_RCVBUF, &sock_size, &sock_length))
 111         return 0; /* Zero means error */
 112
 113       return sock_size;
 114     }
 115   }
 116
 117   WATCHPOINT_ASSERT(temp_flag); /* Programming mistake if it gets this far */
 118   if (ptr->flags & temp_flag)
 119     return 1;
 120   else
 121     return 0;
 122
 123   return MEMCACHED_SUCCESS;
 124 }