357fda8c5f8f687a9058ddf3d03f83dd40f76d9b
[awesomized/libmemcached] / 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 }