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

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  LibMemcached
   4  *
   5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
   6  *  All rights reserved.
   7  *
   8  *  Redistribution and use in source and binary forms, with or without
   9  *  modification, are permitted provided that the following conditions are
  10  *  met:
  11  *
  12  *      * Redistributions of source code must retain the above copyright
  13  *  notice, this list of conditions and the following disclaimer.
  14  *
  15  *      * Redistributions in binary form must reproduce the above
  16  *  copyright notice, this list of conditions and the following disclaimer
  17  *  in the documentation and/or other materials provided with the
  18  *  distribution.
  19  *
  20  *      * The names of its contributors may not be used to endorse or
  21  *  promote products derived from this software without specific prior
  22  *  written permission.
  23  *
  24  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  25  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  26  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  27  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  28  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  30  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  31  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  32  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  34  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35  *
  36  */
  37
  38 #include "libmemcached/common.h"
  39
  40 struct memcached_error_st
  41 {
  42   memcached_st *root;
  43   uint64_t query_id;
  44   struct memcached_error_st *next;
  45   memcached_return_t rc;
  46   int local_errno;
  47   size_t size;
  48   char c_str[];
  49 };
  50
  51 static void _set(memcached_st *memc, memcached_string_t *str, const memcached_return_t rc, const int local_errno)
  52 {
  53   WATCHPOINT_ASSERT(memc);
  54   if (! memc)
  55     return;
  56
  57   if (memc->error_messages && memc->error_messages->query_id != memc->query_id)
  58   {
  59     memcached_error_free(memc);
  60   }
  61
  62   memcached_error_st *error;
  63   error= (struct memcached_error_st *)libmemcached_malloc(memc, sizeof(struct memcached_error_st) +(str ? str->size :0) +1);
  64
  65   if (! error)
  66     return;
  67
  68   error->root= memc;
  69   error->query_id= memc->query_id;
  70   error->rc= rc;
  71   error->local_errno= local_errno;
  72
  73   if (str)
  74   {
  75     error->size= str->size;
  76     memcpy(error->c_str, str->c_str, str->size);
  77     error->c_str[str->size]= 0;
  78   }
  79   else
  80   {
  81     error->size= 0;
  82   }
  83
  84   error->next= memc->error_messages;
  85   memc->error_messages= error;
  86 }
  87
  88 memcached_return_t memcached_set_error_string(memcached_st *memc, memcached_return_t rc, const char *str, size_t length)
  89 {
  90   memcached_string_t tmp;
  91   tmp.c_str= str;
  92   tmp.size= length;
  93   return memcached_set_error(memc, rc, &tmp);
  94 }
  95
  96 memcached_return_t memcached_set_error(memcached_st *memc, memcached_return_t rc, memcached_string_t *str)
  97 {
  98   if (rc == MEMCACHED_SUCCESS)
  99     return MEMCACHED_SUCCESS;
 100
 101   _set(memc, str, rc, 0);
 102
 103   return rc;
 104 }
 105
 106 memcached_return_t memcached_set_errno(memcached_st *memc, int local_errno, memcached_string_t *str)
 107 {
 108   _set(memc, str, MEMCACHED_ERRNO, local_errno);
 109
 110   return MEMCACHED_ERRNO;
 111 }
 112
 113 static void _error_print(const memcached_error_st *error)
 114 {
 115   if (! error)
 116     return;
 117
 118   if (! error->size)
 119   {
 120     fprintf(stderr, "%s\n", memcached_strerror(NULL, error->rc) );
 121   }
 122   else
 123   {
 124     fprintf(stderr, "%s %s\n", memcached_strerror(NULL, error->rc), error->c_str);
 125   }
 126
 127   _error_print(error->next);
 128 }
 129
 130 void memcached_error_print(const memcached_st *self)
 131 {
 132   if (! self)
 133     return;
 134
 135   _error_print(self->error_messages);
 136 }
 137
 138 static void _error_free(memcached_error_st *error)
 139 {
 140   if (! error)
 141     return;
 142
 143   _error_free(error->next);
 144
 145   if (error && error->root)
 146   {
 147     libmemcached_free(error->root, error);
 148   }
 149   else if (error)
 150   {
 151     free(error);
 152   }
 153 }
 154
 155 void memcached_error_free(memcached_st *self)
 156 {
 157   if (! self)
 158     return;
 159
 160   _error_free(self->error_messages);
 161 }
 162
 163 const char *memcached_last_error_message(memcached_st *memc)
 164 {
 165   if (! memc)
 166     return memcached_strerror(memc, MEMCACHED_INVALID_ARGUMENTS);
 167
 168   if (! memc->error_messages)
 169     return memcached_strerror(memc, MEMCACHED_SUCCESS);
 170
 171   if (! memc->error_messages->size)
 172   {
 173     return memcached_strerror(memc, memc->error_messages->rc);
 174   }
 175
 176   return memc->error_messages->c_str;
 177 }
 178
 179 memcached_return_t memcached_last_error(memcached_st *memc)
 180 {
 181   if (! memc)
 182     return MEMCACHED_INVALID_ARGUMENTS;
 183
 184   if (! memc->error_messages)
 185     return MEMCACHED_SUCCESS;
 186
 187   return memc->error_messages->rc;
 188 }
 189
 190 int memcached_last_error_errno(memcached_st *memc)
 191 {
 192   if (! memc)
 193     return 0;
 194
 195   if (! memc->error_messages)
 196     return 0;
 197
 198   return memc->error_messages->local_errno;
 199 }