3 #include <sys/socket.h>
4 #include <netinet/tcp.h>
7 This function is used to modify the behabior of running client.
9 We quit all connections so we can reset the sockets.
12 void set_behavior_flag(memcached_st
*ptr
, memcached_flags temp_flag
, void *data
)
14 unsigned int *truefalse
= (unsigned int *)data
;
18 ptr
->flags
|= temp_flag
;
20 ptr
->flags
+= temp_flag
;
23 memcached_return
memcached_behavior_set(memcached_st
*ptr
,
24 memcached_behavior flag
,
29 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
30 set_behavior_flag(ptr
, MEM_NO_BLOCK
, data
);
32 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
33 set_behavior_flag(ptr
, MEM_TCP_NODELAY
, data
);
35 case MEMCACHED_BEHAVIOR_MD5_HASHING
:
36 set_behavior_flag(ptr
, MEM_USE_MD5
, data
);
38 case MEMCACHED_BEHAVIOR_CRC_HASHING
:
39 set_behavior_flag(ptr
, MEM_USE_CRC
, data
);
41 case MEMCACHED_BEHAVIOR_KETAMA
:
42 set_behavior_flag(ptr
, MEM_USE_KETAMA
, data
);
44 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
46 ptr
->send_size
= (*((int *)data
));
50 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
52 ptr
->recv_size
= (*((int *)data
));
60 return MEMCACHED_SUCCESS
;
63 unsigned long long memcached_behavior_get(memcached_st
*ptr
,
64 memcached_behavior flag
)
66 memcached_flags temp_flag
= 0;
70 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
71 temp_flag
= MEM_NO_BLOCK
;
73 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
74 temp_flag
= MEM_TCP_NODELAY
;
76 case MEMCACHED_BEHAVIOR_MD5_HASHING
:
77 temp_flag
= MEM_USE_MD5
;
79 case MEMCACHED_BEHAVIOR_CRC_HASHING
:
80 temp_flag
= MEM_USE_CRC
;
82 case MEMCACHED_BEHAVIOR_KETAMA
:
83 temp_flag
= MEM_USE_KETAMA
;
85 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
88 socklen_t sock_length
= sizeof(int);
90 /* We just try the first host, and if it is down we return zero */
91 if ((memcached_connect(ptr
, 0)) != MEMCACHED_SUCCESS
)
94 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
95 SO_SNDBUF
, &sock_size
, &sock_length
))
96 return 0; /* Zero means error */
100 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
103 socklen_t sock_length
= sizeof(int);
105 /* We just try the first host, and if it is down we return zero */
106 if ((memcached_connect(ptr
, 0)) != MEMCACHED_SUCCESS
)
109 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
110 SO_RCVBUF
, &sock_size
, &sock_length
))
111 return 0; /* Zero means error */
117 WATCHPOINT_ASSERT(temp_flag
); /* Programming mistake if it gets this far */
118 if (ptr
->flags
& temp_flag
)
123 return MEMCACHED_SUCCESS
;