p9y: bsd
[m6w6/libmemcached] / src / libmemcached / instance.cc
index 751188c720ea421b208a7b3ee3f240deafd2bcce..266eccc70f8d8b6f53b73b1513ab7879ff5447e6 100644 (file)
-/*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
- * 
- *  Libmemcached library
- *
- *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
- *  Copyright (C) 2006-2009 Brian Aker All rights reserved.
- *
- *  Redistribution and use in source and binary forms, with or without
- *  modification, are permitted provided that the following conditions are
- *  met:
- *
- *      * Redistributions of source code must retain the above copyright
- *  notice, this list of conditions and the following disclaimer.
- *
- *      * Redistributions in binary form must reproduce the above
- *  copyright notice, this list of conditions and the following disclaimer
- *  in the documentation and/or other materials provided with the
- *  distribution.
- *
- *      * The names of its contributors may not be used to endorse or
- *  promote products derived from this software without specific prior
- *  written permission.
- *
- *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- */
+/*
+    +--------------------------------------------------------------------+
+    | libmemcached - C/C++ Client Library for memcached                  |
+    +--------------------------------------------------------------------+
+    | Redistribution and use in source and binary forms, with or without |
+    | modification, are permitted under the terms of the BSD license.    |
+    | You should have received a copy of the license in a bundled file   |
+    | named LICENSE; in case you did not receive a copy you can review   |
+    | the terms online at: https://opensource.org/licenses/BSD-3-Clause  |
+    +--------------------------------------------------------------------+
+    | Copyright (c) 2006-2014 Brian Aker   https://datadifferential.com/ |
+    | Copyright (c) 2020 Michael Wallner   <mike@php.net>                |
+    +--------------------------------------------------------------------+
+*/
 
 #include "libmemcached/common.h"
 
-static inline void _server_init(memcached_instance_st* self, Memcached *root,
-                                const memcached_string_t& hostname,
-                                in_port_t port,
-                                uint32_t weight, memcached_connection_t type)
-{
-  self->options.is_shutting_down= false;
-  self->options.is_dead= false;
-  self->options.ready= false;
-  self->_events= 0;
-  self->_revents= 0;
-  self->cursor_active_= 0;
-  self->port_= port;
-  self->fd= INVALID_SOCKET;
-  self->io_bytes_sent= 0;
-  self->request_id= 0;
-  self->server_failure_counter= 0;
-  self->server_failure_counter_query_id= 0;
-  self->server_timeout_counter= 0;
-  self->server_timeout_counter_query_id= 0;
-  self->weight= weight ? weight : 1; // 1 is the default weight value
-  self->io_wait_count.read= 0;
-  self->io_wait_count.write= 0;
-  self->io_wait_count.timeouts= 0;
-  self->io_wait_count._bytes_read= 0;
-  self->major_version= UINT8_MAX;
-  self->micro_version= UINT8_MAX;
-  self->minor_version= UINT8_MAX;
-  self->type= type;
-  self->error_messages= NULL;
-  self->read_ptr= self->read_buffer;
-  self->read_buffer_length= 0;
-  self->write_buffer_offset= 0;
-  self->address_info= NULL;
-  self->address_info_next= NULL;
-
-  self->state= MEMCACHED_SERVER_STATE_NEW;
-  self->next_retry= 0;
-
-  self->root= root;
-  if (root)
-  {
-    self->version= ++root->server_info.version;
-  }
-  else
-  {
-    self->version= UINT_MAX;
+static inline void _server_init(memcached_instance_st *self, Memcached *root,
+                                const memcached_string_t &hostname, in_port_t port, uint32_t weight,
+                                memcached_connection_t type) {
+  self->options.is_shutting_down = false;
+  self->options.is_dead = false;
+  self->options.ready = false;
+  self->_events = 0;
+  self->_revents = 0;
+  self->cursor_active_ = 0;
+  self->port_ = port;
+  self->fd = INVALID_SOCKET;
+  self->io_bytes_sent = 0;
+  self->request_id = 0;
+  self->server_failure_counter = 0;
+  self->server_failure_counter_query_id = 0;
+  self->server_timeout_counter = 0;
+  self->server_timeout_counter_query_id = 0;
+  self->weight = weight ? weight : 1; // 1 is the default weight value
+  self->io_wait_count.read = 0;
+  self->io_wait_count.write = 0;
+  self->io_wait_count.timeouts = 0;
+  self->io_wait_count._bytes_read = 0;
+  self->major_version = UINT8_MAX;
+  self->micro_version = UINT8_MAX;
+  self->minor_version = UINT8_MAX;
+  self->type = type;
+  self->error_messages = NULL;
+  self->read_ptr = self->read_buffer;
+  self->read_buffer_length = 0;
+  self->write_buffer_offset = 0;
+  self->address_info = NULL;
+  self->address_info_next = NULL;
+
+  self->state = MEMCACHED_SERVER_STATE_NEW;
+  self->next_retry = 0;
+
+  self->root = root;
+  if (root) {
+    self->version = ++root->server_info.version;
+  } else {
+    self->version = UINT_MAX;
   }
-  self->limit_maxbytes= 0;
+  self->limit_maxbytes = 0;
   self->hostname(hostname);
 }
 
-static memcached_instance_st* _server_create(memcached_instance_st* self, const memcached_st *memc)
-{
-  if (self == NULL)
-  {
-   self= libmemcached_xmalloc(memc, memcached_instance_st);
+static memcached_instance_st *_server_create(memcached_instance_st *self,
+                                             const memcached_st *memc) {
+  if (self == NULL) {
+    self = libmemcached_xmalloc(memc, memcached_instance_st);
 
-    if (self == NULL)
-    {
+    if (self == NULL) {
       return NULL; /*  MEMCACHED_MEMORY_ALLOCATION_FAILURE */
     }
 
-    self->options.is_allocated= true;
-  }
-  else
-  {
-    self->options.is_allocated= false;
+    self->options.is_allocated = true;
+  } else {
+    self->options.is_allocated = false;
   }
 
-  self->options.is_initialized= true;
+  self->options.is_initialized = true;
 
   return self;
 }
 
-void memcached_instance_st::events(short arg)
-{
-  if ((_events | arg) == _events)
-  {
+void memcached_instance_st::events(short arg) {
+  if ((_events | arg) == _events) {
     return;
   }
 
-  _events|= arg;
+  _events |= arg;
 }
 
-void memcached_instance_st::revents(short arg)
-{
-  if (arg)
-  {
-    options.ready= true;
+void memcached_instance_st::revents(short arg) {
+  if (arg) {
+    options.ready = true;
   }
 
-  _revents= arg;
-  _events&= short(~arg);
+  _revents = arg;
+  _events &= short(~arg);
 }
 
-memcached_instance_st* __instance_create_with(memcached_st *memc,
-                                                    memcached_instance_st* self,
-                                                    const memcached_string_t& hostname,
-                                                    const in_port_t port,
-                                                    uint32_t weight, 
-                                                    const memcached_connection_t type)
-{
-  if (memcached_is_valid_servername(hostname) == false)
-  {
-    memcached_set_error(*memc, MEMCACHED_INVALID_ARGUMENTS, MEMCACHED_AT, memcached_literal_param("Invalid hostname provided"));
+memcached_instance_st *__instance_create_with(memcached_st *memc, memcached_instance_st *self,
+                                              const memcached_string_t &hostname,
+                                              const in_port_t port, uint32_t weight,
+                                              const memcached_connection_t type) {
+  if (memcached_is_valid_servername(hostname) == false) {
+    memcached_set_error(*memc, MEMCACHED_INVALID_ARGUMENTS, MEMCACHED_AT,
+                        memcached_literal_param("Invalid hostname provided"));
     return NULL;
   }
 
-  self= _server_create(self, memc);
+  self = _server_create(self, memc);
 
-  if (self == NULL)
-  {
+  if (self == NULL) {
     return NULL;
   }
 
   _server_init(self, const_cast<memcached_st *>(memc), hostname, port, weight, type);
 
-  if (memc and memcached_is_udp(memc))
-  { 
-    self->write_buffer_offset= UDP_DATAGRAM_HEADER_LENGTH;
+  if (memc and memcached_is_udp(memc)) {
+    self->write_buffer_offset = UDP_DATAGRAM_HEADER_LENGTH;
     memcached_io_init_udp_header(self, 0);
   }
 
   return self;
 }
 
-void __instance_free(memcached_instance_st* self)
-{
+void __instance_free(memcached_instance_st *self) {
   memcached_quit_server(self, false);
 
   self->clear_addrinfo();
@@ -172,47 +131,36 @@ void __instance_free(memcached_instance_st* self)
 
   memcached_error_free(*self);
 
-  if (memcached_is_allocated(self))
-  {
+  if (memcached_is_allocated(self)) {
     libmemcached_free(self->root, self);
-  }
-  else
-  {
-    self->options.is_initialized= false;
+  } else {
+    self->options.is_initialized = false;
   }
 }
 
-void memcached_instance_free(memcached_instance_st* self)
-{
-  if (self)
-  {
+void memcached_instance_free(memcached_instance_st *self) {
+  if (self) {
     __instance_free(self);
   }
 }
 
-memcached_return_t memcached_server_cursor(const memcached_st* shell,
-                                           const memcached_server_fn *callback,
-                                           void *context,
-                                           uint32_t number_of_callbacks)
-{
-  const Memcached* memc= memcached2Memcached(shell);
+memcached_return_t memcached_server_cursor(const memcached_st *shell,
+                                           const memcached_server_fn *callback, void *context,
+                                           uint32_t number_of_callbacks) {
+  const Memcached *memc = memcached2Memcached(shell);
   memcached_return_t rc;
-  if (memcached_failed(rc= initialize_const_query(memc)))
-  {
+  if (memcached_failed(rc = initialize_const_query(memc))) {
     return rc;
   }
 
-  size_t errors= 0;
-  for (uint32_t x= 0; x < memcached_instance_list_count(memc); x++)
-  {
-    memcached_instance_st* instance= memcached_instance_by_position(memc, x);
+  size_t errors = 0;
+  for (uint32_t x = 0; x < memcached_instance_list_count(memc); x++) {
+    memcached_instance_st *instance = memcached_instance_by_position(memc, x);
 
-    for (uint32_t y= 0; y < number_of_callbacks; y++)
-    {
-      memcached_return_t ret= (*callback[y])(memc, instance, context);
+    for (uint32_t y = 0; y < number_of_callbacks; y++) {
+      memcached_return_t ret = (*callback[y])(memc, instance, context);
 
-      if (memcached_failed(ret))
-      {
+      if (memcached_failed(ret)) {
         errors++;
         continue;
       }
@@ -223,139 +171,112 @@ memcached_return_t memcached_server_cursor(const memcached_st* shell,
 }
 
 memcached_return_t memcached_server_execute(memcached_st *memc,
-                                            memcached_server_execute_fn callback,
-                                            void *context)
-{
-  if (callback == NULL)
-  {
+                                            memcached_server_execute_fn callback, void *context) {
+  if (callback == NULL) {
     return MEMCACHED_INVALID_ARGUMENTS;
   }
 
-  bool some_errors= false;;
-  for (uint32_t x= 0; x < memcached_instance_list_count(memc); x++)
-  {
-    memcached_instance_st* instance= memcached_instance_fetch(memc, x);
+  bool some_errors = false;
+  ;
+  for (uint32_t x = 0; x < memcached_instance_list_count(memc); x++) {
+    memcached_instance_st *instance = memcached_instance_fetch(memc, x);
 
-    memcached_return_t rc= (*callback)(memc, instance, context);
-    if (rc == MEMCACHED_INVALID_ARGUMENTS)
-    {
+    memcached_return_t rc = (*callback)(memc, instance, context);
+    if (rc == MEMCACHED_INVALID_ARGUMENTS) {
       return rc;
-    }
-    else if (memcached_fatal(rc))
-    {
-      some_errors= true;
+    } else if (memcached_fatal(rc)) {
+      some_errors = true;
     }
   }
 
-  (void)some_errors;
+  (void) some_errors;
   return MEMCACHED_SUCCESS;
 }
 
-const memcached_instance_st * memcached_server_by_key(memcached_st *shell,
-                                                     const char *key,
-                                                     size_t key_length,
-                                                     memcached_return_t *error)
-{
-  Memcached* memc= memcached2Memcached(shell);
+const memcached_instance_st *memcached_server_by_key(memcached_st *shell, const char *key,
+                                                     size_t key_length, memcached_return_t *error) {
+  Memcached *memc = memcached2Memcached(shell);
   memcached_return_t unused;
-  if (error == NULL)
-  {
-    error= &unused;
+  if (error == NULL) {
+    error = &unused;
   }
 
-
   memcached_return_t rc;
-  if (memcached_failed(rc= initialize_const_query(memc)))
-  {
-    *error= rc;
+  if (memcached_failed(rc = initialize_const_query(memc))) {
+    *error = rc;
     return NULL;
   }
 
-  if (memcached_failed((memcached_key_test(*memc, (const char **)&key, &key_length, 1))))
-  {
-    *error= memcached_last_error(memc);
+  if (memcached_failed((memcached_key_test(*memc, (const char **) &key, &key_length, 1)))) {
+    *error = memcached_last_error(memc);
     return NULL;
   }
 
-  uint32_t server_key= memcached_generate_hash(memc, key, key_length);
+  uint32_t server_key = memcached_generate_hash(memc, key, key_length);
   return memcached_instance_by_position(memc, server_key);
 }
 
 /*
   If we do not have a valid object to clone from, we toss an error.
 */
-static memcached_instance_st* memcached_instance_clone(memcached_instance_st* source)
-{
+static memcached_instance_st *memcached_instance_clone(memcached_instance_st *source) {
   /* We just do a normal create if source is missing */
-  if (source == NULL)
-  {
+  if (source == NULL) {
     return NULL;
   }
 
-  memcached_string_t hostname_= { memcached_string_make_from_cstr(source->hostname()) };
-  return __instance_create_with(source->root,
-                                NULL,
-                                hostname_,
-                                source->port(), source->weight,
+  memcached_string_t hostname_ = {memcached_string_make_from_cstr(source->hostname())};
+  return __instance_create_with(source->root, NULL, hostname_, source->port(), source->weight,
                                 source->type);
 }
 
-void set_last_disconnected_host(memcached_instance_st* self)
-{
+void set_last_disconnected_host(memcached_instance_st *self) {
   assert(self->root);
-  if (self->root)
-  {
-    if (memcached_server_get_last_disconnect(self->root) and
-        memcached_server_get_last_disconnect(self->root)->version == self->version)
+  if (self->root) {
+    if (memcached_server_get_last_disconnect(self->root)
+        and memcached_server_get_last_disconnect(self->root)->version == self->version)
     {
       return;
     }
 
     // const_cast
-    memcached_st *root= (memcached_st *)self->root;
+    memcached_st *root = (memcached_st *) self->root;
 
-    memcached_instance_free((memcached_instance_st*)(root->last_disconnected_server));
+    memcached_instance_free((memcached_instance_st *) (root->last_disconnected_server));
 
     // We set is_parsing so that no lookup happens
-    root->state.is_parsing= true;
-    root->last_disconnected_server= memcached_instance_clone(self);
-    root->state.is_parsing= false;
+    root->state.is_parsing = true;
+    root->last_disconnected_server = memcached_instance_clone(self);
+    root->state.is_parsing = false;
 
-    ((memcached_instance_st*)memcached_server_get_last_disconnect(root))->version= self->version;
+    ((memcached_instance_st *) memcached_server_get_last_disconnect(root))->version = self->version;
   }
 }
 
-const memcached_instance_st * memcached_server_get_last_disconnect(const memcached_st *shell)
-{
-  const Memcached* self= memcached2Memcached(shell);
-  if (self)
-  {
-    return (const memcached_instance_st *)self->last_disconnected_server;
+const memcached_instance_st *memcached_server_get_last_disconnect(const memcached_st *shell) {
+  const Memcached *self = memcached2Memcached(shell);
+  if (self) {
+    return (const memcached_instance_st *) self->last_disconnected_server;
   }
 
   return 0;
 }
 
-void memcached_instance_next_retry(const memcached_instance_st * self, const time_t absolute_time)
-{
+void memcached_instance_next_retry(const memcached_instance_st *self, const time_t absolute_time) {
   WATCHPOINT_ASSERT(self);
-  if (self)
-  {
-    ((memcached_instance_st*)self)->next_retry= absolute_time;
+  if (self) {
+    ((memcached_instance_st *) self)->next_retry = absolute_time;
   }
 }
 
-bool memcached_instance_st::valid() const
-{
-  if (fd == INVALID_SOCKET)
-  {
+bool memcached_instance_st::valid() const {
+  if (fd == INVALID_SOCKET) {
     return false;
   }
 
   return true;
 }
 
-bool memcached_instance_st::is_shutting_down() const
-{
+bool memcached_instance_st::is_shutting_down() const {
   return options.is_shutting_down;
 }