X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fdo.cc;h=daf1a1150eb2d242303b9826488f9475a6a67283;hb=bb98ce33936edc0dc914652c0227f95727518b16;hp=0d05ea6d4523675a6d89e0987e2ccd44b6b6c837;hpb=facfe08e21d2c3135c508ff098ff603ddaacef3e;p=awesomized%2Flibmemcached diff --git a/libmemcached/do.cc b/libmemcached/do.cc index 0d05ea6d..daf1a115 100644 --- a/libmemcached/do.cc +++ b/libmemcached/do.cc @@ -11,7 +11,60 @@ #include -memcached_return_t memcached_vdo(org::libmemcached::Instance* instance, +static memcached_return_t _vdo_udp(memcached_instance_st* instance, + libmemcached_io_vector_st vector[], + const size_t count) +{ +#ifndef __MINGW32__ + if (vector[0].buffer or vector[0].length) + { + return memcached_set_error(*instance->root, MEMCACHED_NOT_SUPPORTED, MEMCACHED_AT, + memcached_literal_param("UDP messages was attempted, but vector was not setup for it")); + } + + struct msghdr msg; + memset(&msg, 0, sizeof(msg)); + + increment_udp_message_id(instance); + vector[0].buffer= instance->write_buffer; + vector[0].length= UDP_DATAGRAM_HEADER_LENGTH; + + msg.msg_iov= (struct iovec*)vector; +#ifdef __APPLE__ + msg.msg_iovlen= int(count); +#else + msg.msg_iovlen= count; +#endif + + uint32_t retry= 5; + while (--retry) + { + ssize_t sendmsg_length= ::sendmsg(instance->fd, &msg, 0); + if (sendmsg_length > 0) + { + break; + } + else if (sendmsg_length < 0) + { + if (errno == EMSGSIZE) + { + return memcached_set_error(*instance, MEMCACHED_WRITE_FAILURE, MEMCACHED_AT); + } + + return memcached_set_errno(*instance, errno, MEMCACHED_AT); + } + } + + return MEMCACHED_SUCCESS; +#else + (void)instance; + (void)vector; + (void)count; + return MEMCACHED_FAILURE; +#endif +} + +memcached_return_t memcached_vdo(memcached_instance_st* instance, libmemcached_io_vector_st vector[], const size_t count, const bool with_flush) @@ -34,54 +87,21 @@ memcached_return_t memcached_vdo(org::libmemcached::Instance* instance, **/ if (memcached_is_udp(instance->root)) { - if (vector[0].buffer or vector[0].length) - { - return memcached_set_error(*instance->root, MEMCACHED_NOT_SUPPORTED, MEMCACHED_AT, - memcached_literal_param("UDP messages was attempted, but vector was not setup for it")); - } - - struct msghdr msg; - memset(&msg, 0, sizeof(msg)); - - increment_udp_message_id(instance); - vector[0].buffer= instance->write_buffer; - vector[0].length= UDP_DATAGRAM_HEADER_LENGTH; - - msg.msg_iov= (struct iovec*)vector; - msg.msg_iovlen= count; - - uint32_t retry= 5; - while (--retry) - { - ssize_t sendmsg_length= ::sendmsg(instance->fd, &msg, 0); - if (sendmsg_length > 0) - { - break; - } - else if (sendmsg_length < 0) - { - if (errno == EMSGSIZE) - { - return memcached_set_error(*instance, MEMCACHED_WRITE_FAILURE, MEMCACHED_AT); - } - - return memcached_set_errno(*instance, errno, MEMCACHED_AT); - } - } - - return MEMCACHED_SUCCESS; + return _vdo_udp(instance, vector, count); } bool sent_success= memcached_io_writev(instance, vector, count, with_flush); if (sent_success == false) { + //assert(memcached_last_error(instance->root) == MEMCACHED_SUCCESS); if (memcached_last_error(instance->root) == MEMCACHED_SUCCESS) { + //assert(memcached_last_error(instance->root) != MEMCACHED_SUCCESS); return memcached_set_error(*instance, MEMCACHED_WRITE_FAILURE, MEMCACHED_AT); } else { - rc= MEMCACHED_WRITE_FAILURE; + rc= memcached_last_error(instance->root); } } else if (memcached_is_replying(instance->root))