Merge lp:~brianaker/libmemcached/1164440 Build: jenkins-Libmemcached-371
[m6w6/libmemcached] / libmemcached / stats.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Libmemcached library
4 *
5 * Copyright (C) 2011-2013 Data Differential, http://datadifferential.com/
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
9 * met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
17 * distribution.
18 *
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
21 * written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 *
35 */
36
37 #include <libmemcached/common.h>
38
39 static const char *memcached_stat_keys[] = {
40 "pid",
41 "uptime",
42 "time",
43 "version",
44 "pointer_size",
45 "rusage_user",
46 "rusage_system",
47 "curr_items",
48 "total_items",
49 "bytes",
50 "curr_connections",
51 "total_connections",
52 "connection_structures",
53 "cmd_get",
54 "cmd_set",
55 "get_hits",
56 "get_misses",
57 "evictions",
58 "bytes_read",
59 "bytes_written",
60 "limit_maxbytes",
61 "threads",
62 NULL
63 };
64
65 struct local_context
66 {
67 memcached_stat_fn func;
68 void *context;
69 const char *args;
70 const size_t args_length;
71
72 local_context(memcached_stat_fn func_arg,
73 void *context_arg,
74 const char *args_arg,
75 const size_t args_length_arg) :
76 func(func_arg),
77 context(context_arg),
78 args(args_arg),
79 args_length(args_length_arg)
80 { }
81 };
82
83
84 static memcached_return_t set_data(memcached_stat_st *memc_stat, const char *key, const char *value)
85 {
86
87 if (strlen(key) < 1)
88 {
89 WATCHPOINT_STRING(key);
90 return MEMCACHED_UNKNOWN_STAT_KEY;
91 }
92 else if (strcmp("pid", key) == 0)
93 {
94 errno= 0;
95 int64_t temp= strtoll(value, (char **)NULL, 10);
96 if (errno != 0)
97 {
98 return MEMCACHED_FAILURE;
99 }
100
101 if (temp <= INT32_MAX and ( sizeof(pid_t) == sizeof(int32_t) ))
102 {
103 memc_stat->pid= pid_t(temp);
104 }
105 else if (temp > -1)
106 {
107 memc_stat->pid= pid_t(temp);
108 }
109 else
110 {
111 // If we got a value less then -1 then something went wrong in the
112 // protocol
113 }
114 }
115 else if (not strcmp("uptime", key))
116 {
117 errno= 0;
118 memc_stat->uptime= strtoul(value, (char **)NULL, 10);
119 if (errno != 0)
120 {
121 return MEMCACHED_FAILURE;
122 }
123 }
124 else if (not strcmp("time", key))
125 {
126 errno= 0;
127 memc_stat->time= strtoul(value, (char **)NULL, 10);
128 if (errno != 0)
129 {
130 return MEMCACHED_FAILURE;
131 }
132 }
133 else if (not strcmp("version", key))
134 {
135 memcpy(memc_stat->version, value, strlen(value));
136 memc_stat->version[strlen(value)]= 0;
137 }
138 else if (not strcmp("pointer_size", key))
139 {
140 errno= 0;
141 memc_stat->pointer_size= strtoul(value, (char **)NULL, 10);
142 if (errno != 0)
143 {
144 return MEMCACHED_FAILURE;
145 }
146 }
147 else if (not strcmp("rusage_user", key))
148 {
149 char *walk_ptr;
150 for (walk_ptr= (char*)value; (!ispunct(*walk_ptr)); walk_ptr++) {};
151 *walk_ptr= 0;
152 walk_ptr++;
153
154 errno= 0;
155 memc_stat->rusage_user_seconds= strtoul(value, (char **)NULL, 10);
156 if (errno != 0)
157 {
158 return MEMCACHED_FAILURE;
159 }
160
161 errno= 0;
162 memc_stat->rusage_user_microseconds= strtoul(walk_ptr, (char **)NULL, 10);
163 if (errno != 0)
164 {
165 return MEMCACHED_FAILURE;
166 }
167 }
168 else if (not strcmp("rusage_system", key))
169 {
170 char *walk_ptr;
171 for (walk_ptr= (char*)value; (!ispunct(*walk_ptr)); walk_ptr++) {};
172 *walk_ptr= 0;
173 walk_ptr++;
174
175 errno= 0;
176 memc_stat->rusage_system_seconds= strtoul(value, (char **)NULL, 10);
177 if (errno != 0)
178 {
179 return MEMCACHED_FAILURE;
180 }
181
182 errno= 0;
183 memc_stat->rusage_system_microseconds= strtoul(walk_ptr, (char **)NULL, 10);
184 if (errno != 0)
185 {
186 return MEMCACHED_FAILURE;
187 }
188 }
189 else if (not strcmp("curr_items", key))
190 {
191 errno= 0;
192 memc_stat->curr_items= strtoul(value, (char **)NULL, 10);
193 if (errno != 0)
194 {
195 return MEMCACHED_FAILURE;
196 }
197 }
198 else if (not strcmp("total_items", key))
199 {
200 errno= 0;
201 memc_stat->total_items= strtoul(value, (char **)NULL, 10);
202 if (errno != 0)
203 {
204 return MEMCACHED_FAILURE;
205 }
206 }
207 else if (not strcmp("bytes_read", key))
208 {
209 errno= 0;
210 memc_stat->bytes_read= strtoull(value, (char **)NULL, 10);
211 if (errno != 0)
212 {
213 return MEMCACHED_FAILURE;
214 }
215 }
216 else if (not strcmp("bytes_written", key))
217 {
218 errno= 0;
219 memc_stat->bytes_written= strtoull(value, (char **)NULL, 10);
220 if (errno != 0)
221 {
222 return MEMCACHED_FAILURE;
223 }
224 }
225 else if (not strcmp("bytes", key))
226 {
227 errno= 0;
228 memc_stat->bytes= strtoull(value, (char **)NULL, 10);
229 if (errno != 0)
230 {
231 return MEMCACHED_FAILURE;
232 }
233 }
234 else if (not strcmp("curr_connections", key))
235 {
236 errno= 0;
237 memc_stat->curr_connections= strtoull(value, (char **)NULL, 10);
238 if (errno != 0)
239 {
240 return MEMCACHED_FAILURE;
241 }
242 }
243 else if (not strcmp("total_connections", key))
244 {
245 errno= 0;
246 memc_stat->total_connections= strtoull(value, (char **)NULL, 10);
247 if (errno != 0)
248 {
249 return MEMCACHED_FAILURE;
250 }
251 }
252 else if (not strcmp("connection_structures", key))
253 {
254 errno= 0;
255 memc_stat->connection_structures= strtoul(value, (char **)NULL, 10);
256 if (errno != 0)
257 {
258 return MEMCACHED_FAILURE;
259 }
260 }
261 else if (not strcmp("cmd_get", key))
262 {
263 errno= 0;
264 memc_stat->cmd_get= strtoull(value, (char **)NULL, 10);
265 if (errno != 0)
266 {
267 return MEMCACHED_FAILURE;
268 }
269 }
270 else if (not strcmp("cmd_set", key))
271 {
272 errno= 0;
273 memc_stat->cmd_set= strtoull(value, (char **)NULL, 10);
274 if (errno != 0)
275 {
276 return MEMCACHED_FAILURE;
277 }
278 }
279 else if (not strcmp("get_hits", key))
280 {
281 errno= 0;
282 memc_stat->get_hits= strtoull(value, (char **)NULL, 10);
283 if (errno != 0)
284 {
285 return MEMCACHED_FAILURE;
286 }
287 }
288 else if (not strcmp("get_misses", key))
289 {
290 errno= 0;
291 memc_stat->get_misses= strtoull(value, (char **)NULL, 10);
292 if (errno != 0)
293 {
294 return MEMCACHED_FAILURE;
295 }
296 }
297 else if (not strcmp("evictions", key))
298 {
299 errno= 0;
300 memc_stat->evictions= strtoull(value, (char **)NULL, 10);
301 if (errno != 0)
302 {
303 return MEMCACHED_FAILURE;
304 }
305 }
306 else if (not strcmp("limit_maxbytes", key))
307 {
308 errno= 0;
309 memc_stat->limit_maxbytes= strtoull(value, (char **)NULL, 10);
310 if (errno != 0)
311 {
312 return MEMCACHED_FAILURE;
313 }
314 }
315 else if (not strcmp("threads", key))
316 {
317 errno= 0;
318 memc_stat->threads= strtoul(value, (char **)NULL, 10);
319 if (errno != 0)
320 {
321 return MEMCACHED_FAILURE;
322 }
323 }
324 else if ((strcmp("delete_misses", key) == 0 or /* New stats in the 1.3 beta */
325 strcmp("delete_hits", key) == 0 or /* Just swallow them for now.. */
326 strcmp("incr_misses", key) == 0 or
327 strcmp("incr_hits", key) == 0 or
328 strcmp("decr_misses", key) == 0 or
329 strcmp("decr_hits", key) == 0 or
330 strcmp("cas_misses", key) == 0 or
331 strcmp("cas_hits", key) == 0 or
332 strcmp("cas_badval", key) == 0 or
333 strcmp("cmd_flush", key) == 0 or
334 strcmp("accepting_conns", key) == 0 or
335 strcmp("listen_disabled_num", key) == 0 or
336 strcmp("conn_yields", key) == 0 or
337 strcmp("auth_cmds", key) == 0 or
338 strcmp("auth_errors", key) == 0 or
339 strcmp("reclaimed", key) == 0) == 0)
340 {
341 WATCHPOINT_STRING(key);
342 /* return MEMCACHED_UNKNOWN_STAT_KEY; */
343 return MEMCACHED_SUCCESS;
344 }
345
346 return MEMCACHED_SUCCESS;
347 }
348
349 char *memcached_stat_get_value(const memcached_st* shell, memcached_stat_st *memc_stat,
350 const char *key, memcached_return_t *error)
351 {
352 memcached_return_t not_used;
353 if (error == NULL)
354 {
355 error= &not_used;
356 }
357
358 if (memc_stat == NULL)
359 {
360 *error= MEMCACHED_INVALID_ARGUMENTS;
361 return NULL;
362 }
363
364 char buffer[SMALL_STRING_LEN];
365 int length;
366
367 *error= MEMCACHED_SUCCESS;
368
369 if (memcmp("pid", key, sizeof("pid") -1) == 0)
370 {
371 length= snprintf(buffer, SMALL_STRING_LEN,"%lld", (signed long long)memc_stat->pid);
372 }
373 else if (not memcmp("uptime", key, sizeof("uptime") -1))
374 {
375 length= snprintf(buffer, SMALL_STRING_LEN,"%lu", memc_stat->uptime);
376 }
377 else if (not memcmp("time", key, sizeof("time") -1))
378 {
379 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->time);
380 }
381 else if (not memcmp("version", key, sizeof("version") -1))
382 {
383 length= snprintf(buffer, SMALL_STRING_LEN,"%s", memc_stat->version);
384 }
385 else if (not memcmp("pointer_size", key, sizeof("pointer_size") -1))
386 {
387 length= snprintf(buffer, SMALL_STRING_LEN,"%lu", memc_stat->pointer_size);
388 }
389 else if (not memcmp("rusage_user", key, sizeof("rusage_user") -1))
390 {
391 length= snprintf(buffer, SMALL_STRING_LEN,"%lu.%lu", memc_stat->rusage_user_seconds, memc_stat->rusage_user_microseconds);
392 }
393 else if (not memcmp("rusage_system", key, sizeof("rusage_system") -1))
394 {
395 length= snprintf(buffer, SMALL_STRING_LEN,"%lu.%lu", memc_stat->rusage_system_seconds, memc_stat->rusage_system_microseconds);
396 }
397 else if (not memcmp("curr_items", key, sizeof("curr_items") -1))
398 {
399 length= snprintf(buffer, SMALL_STRING_LEN,"%lu", memc_stat->curr_items);
400 }
401 else if (not memcmp("total_items", key, sizeof("total_items") -1))
402 {
403 length= snprintf(buffer, SMALL_STRING_LEN,"%lu", memc_stat->total_items);
404 }
405 else if (not memcmp("curr_connections", key, sizeof("curr_connections") -1))
406 {
407 length= snprintf(buffer, SMALL_STRING_LEN,"%lu", memc_stat->curr_connections);
408 }
409 else if (not memcmp("total_connections", key, sizeof("total_connections") -1))
410 {
411 length= snprintf(buffer, SMALL_STRING_LEN,"%lu", memc_stat->total_connections);
412 }
413 else if (not memcmp("connection_structures", key, sizeof("connection_structures") -1))
414 {
415 length= snprintf(buffer, SMALL_STRING_LEN,"%lu", memc_stat->connection_structures);
416 }
417 else if (not memcmp("cmd_get", key, sizeof("cmd_get") -1))
418 {
419 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->cmd_get);
420 }
421 else if (not memcmp("cmd_set", key, sizeof("cmd_set") -1))
422 {
423 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->cmd_set);
424 }
425 else if (not memcmp("get_hits", key, sizeof("get_hits") -1))
426 {
427 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->get_hits);
428 }
429 else if (not memcmp("get_misses", key, sizeof("get_misses") -1))
430 {
431 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->get_misses);
432 }
433 else if (not memcmp("evictions", key, sizeof("evictions") -1))
434 {
435 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->evictions);
436 }
437 else if (not memcmp("bytes_read", key, sizeof("bytes_read") -1))
438 {
439 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->bytes_read);
440 }
441 else if (not memcmp("bytes_written", key, sizeof("bytes_written") -1))
442 {
443 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->bytes_written);
444 }
445 else if (not memcmp("bytes", key, sizeof("bytes") -1))
446 {
447 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->bytes);
448 }
449 else if (not memcmp("limit_maxbytes", key, sizeof("limit_maxbytes") -1))
450 {
451 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->limit_maxbytes);
452 }
453 else if (not memcmp("threads", key, sizeof("threads") -1))
454 {
455 length= snprintf(buffer, SMALL_STRING_LEN,"%lu", memc_stat->threads);
456 }
457 else
458 {
459 Memcached* memc= (Memcached*)memcached2Memcached(shell);
460 *error= memcached_set_error(*memc, MEMCACHED_INVALID_ARGUMENTS, MEMCACHED_AT, memcached_literal_param("Invalid key provided"));
461 return NULL;
462 }
463
464 if (length >= SMALL_STRING_LEN || length < 0)
465 {
466 Memcached* memc= (Memcached*)memcached2Memcached(shell);
467 *error= memcached_set_error(*memc, MEMCACHED_FAILURE, MEMCACHED_AT, memcached_literal_param("Internal failure occured with buffer, please report this bug."));
468 return NULL;
469 }
470
471 // User is responsible for free() memory, so use malloc()
472 char *ret= static_cast<char *>(malloc(size_t(length +1)));
473 memcpy(ret, buffer, (size_t) length);
474 ret[length]= '\0';
475
476 return ret;
477 }
478
479 static memcached_return_t binary_stats_fetch(memcached_stat_st *memc_stat,
480 const char *args,
481 const size_t args_length,
482 memcached_instance_st* instance,
483 struct local_context *check)
484 {
485 char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE];
486 protocol_binary_request_stats request= {}; // = {.bytes= {0}};
487
488 initialize_binary_request(instance, request.message.header);
489
490 request.message.header.request.opcode= PROTOCOL_BINARY_CMD_STAT;
491 request.message.header.request.datatype= PROTOCOL_BINARY_RAW_BYTES;
492
493 if (args_length)
494 {
495 request.message.header.request.keylen= htons(uint16_t(args_length));
496 request.message.header.request.bodylen= htonl(uint32_t( args_length));
497
498 libmemcached_io_vector_st vector[]=
499 {
500 { request.bytes, sizeof(request.bytes) },
501 { args, args_length }
502 };
503
504 if (memcached_vdo(instance, vector, 2, true) != MEMCACHED_SUCCESS)
505 {
506 memcached_io_reset(instance);
507 return MEMCACHED_WRITE_FAILURE;
508 }
509 }
510 else
511 {
512 libmemcached_io_vector_st vector[]=
513 {
514 { request.bytes, sizeof(request.bytes) }
515 };
516
517 if (memcached_vdo(instance, vector, 1, true) != MEMCACHED_SUCCESS)
518 {
519 memcached_io_reset(instance);
520 return MEMCACHED_WRITE_FAILURE;
521 }
522 }
523
524 memcached_server_response_decrement(instance);
525 while (1)
526 {
527 memcached_return_t rc= memcached_response(instance, buffer, sizeof(buffer), NULL);
528
529 if (rc == MEMCACHED_END)
530 {
531 break;
532 }
533
534 if (rc != MEMCACHED_SUCCESS)
535 {
536 memcached_io_reset(instance);
537 return rc;
538 }
539
540 if (check && check->func)
541 {
542 size_t key_length= strlen(buffer);
543
544 check->func(instance,
545 buffer, key_length,
546 buffer+key_length+1, strlen(buffer+key_length+1),
547 check->context);
548 }
549
550 if (memc_stat)
551 {
552 if ((set_data(memc_stat, buffer, buffer + strlen(buffer) + 1)) == MEMCACHED_UNKNOWN_STAT_KEY)
553 {
554 WATCHPOINT_ERROR(MEMCACHED_UNKNOWN_STAT_KEY);
555 WATCHPOINT_ASSERT(0);
556 }
557 }
558 }
559
560 /*
561 * memcached_response will decrement the counter, so I need to reset it..
562 * todo: look at this and try to find a better solution.
563 * */
564 instance->cursor_active_= 0;
565
566 return MEMCACHED_SUCCESS;
567 }
568
569 static memcached_return_t ascii_stats_fetch(memcached_stat_st *memc_stat,
570 const char *args,
571 const size_t args_length,
572 memcached_instance_st* instance,
573 struct local_context *check)
574 {
575 libmemcached_io_vector_st vector[]=
576 {
577 { memcached_literal_param("stats ") },
578 { args, args_length },
579 { memcached_literal_param("\r\n") }
580 };
581
582 memcached_return_t rc= memcached_vdo(instance, vector, 3, true);
583 if (memcached_success(rc))
584 {
585 char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE];
586 while ((rc= memcached_response(instance, buffer, sizeof(buffer), NULL)) == MEMCACHED_STAT)
587 {
588 char *string_ptr= buffer;
589 string_ptr+= 5; /* Move past STAT */
590
591 char *end_ptr;
592 for (end_ptr= string_ptr; isgraph(*end_ptr); end_ptr++) {};
593 char *key= string_ptr;
594 key[size_t(end_ptr-string_ptr)]= 0;
595
596 string_ptr= end_ptr + 1;
597 for (end_ptr= string_ptr; !(isspace(*end_ptr)); end_ptr++) {};
598 char *value= string_ptr;
599 value[(size_t)(end_ptr -string_ptr)]= 0;
600 #if 0
601 bool check_bool= bool(check);
602 bool check_func_bool= bool(check) ? bool(check->func) : false;
603 fprintf(stderr, "%s:%d %s %s %d:%d\n", __FILE__, __LINE__, key, value, check_bool, check_func_bool);
604 #endif
605
606 if (check and check->func)
607 {
608 check->func(instance,
609 key, strlen(key),
610 value, strlen(value),
611 check->context);
612 }
613
614 if (memc_stat)
615 {
616 if((set_data(memc_stat, key, value)) == MEMCACHED_UNKNOWN_STAT_KEY)
617 {
618 WATCHPOINT_ERROR(MEMCACHED_UNKNOWN_STAT_KEY);
619 WATCHPOINT_ASSERT(0);
620 }
621 }
622 }
623 }
624
625 if (rc == MEMCACHED_ERROR)
626 {
627 return MEMCACHED_INVALID_ARGUMENTS;
628 }
629
630 if (rc == MEMCACHED_END)
631 {
632 return MEMCACHED_SUCCESS;
633 }
634
635 return rc;
636 }
637
638 memcached_stat_st *memcached_stat(memcached_st *shell, char *args, memcached_return_t *error)
639 {
640 Memcached* self= memcached2Memcached(shell);
641 memcached_return_t unused;
642 if (error == NULL)
643 {
644 error= &unused;
645 }
646
647 if (memcached_failed(*error= initialize_query(self, true)))
648 {
649 return NULL;
650 }
651
652 if (memcached_is_udp(self))
653 {
654 *error= memcached_set_error(*self, MEMCACHED_NOT_SUPPORTED, MEMCACHED_AT);
655 return NULL;
656 }
657
658 memcached_return_t rc;
659 size_t args_length= 0;
660 if (args)
661 {
662 args_length= strlen(args);
663 if (memcached_failed(rc= memcached_key_test(*self, (const char **)&args, &args_length, 1)))
664 {
665 *error= memcached_set_error(*self, rc, MEMCACHED_AT);
666 return NULL;
667 }
668 }
669
670 WATCHPOINT_ASSERT(error);
671
672 memcached_stat_st *stats= libmemcached_xcalloc(self, memcached_server_count(self), memcached_stat_st);
673 if (stats == NULL)
674 {
675 *error= memcached_set_error(*self, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT);
676 return NULL;
677 }
678
679 WATCHPOINT_ASSERT(rc == MEMCACHED_SUCCESS);
680 rc= MEMCACHED_SUCCESS;
681 for (uint32_t x= 0; x < memcached_server_count(self); x++)
682 {
683 memcached_stat_st* stat_instance= stats +x;
684
685 stat_instance->pid= -1;
686 stat_instance->root= self;
687
688 memcached_instance_st* instance= memcached_instance_fetch(self, x);
689
690 memcached_return_t temp_return;
691 if (memcached_is_binary(self))
692 {
693 temp_return= binary_stats_fetch(stat_instance, args, args_length, instance, NULL);
694 }
695 else
696 {
697 temp_return= ascii_stats_fetch(stat_instance, args, args_length, instance, NULL);
698 }
699
700 // Special case where "args" is invalid
701 if (temp_return == MEMCACHED_INVALID_ARGUMENTS)
702 {
703 rc= MEMCACHED_INVALID_ARGUMENTS;
704 break;
705 }
706
707 if (memcached_failed(temp_return))
708 {
709 rc= MEMCACHED_SOME_ERRORS;
710 }
711 }
712
713 *error= rc;
714
715 return stats;
716 }
717
718 memcached_return_t memcached_stat_servername(memcached_stat_st *memc_stat, char *args,
719 const char *hostname, in_port_t port)
720 {
721 memcached_st memc;
722
723 memcached_stat_st unused_memc_stat;
724 if (memc_stat == NULL)
725 {
726 memc_stat= &unused_memc_stat;
727 }
728
729 memset(memc_stat, 0, sizeof(memcached_stat_st));
730
731 memcached_st *memc_ptr= memcached_create(&memc);
732 if (memc_ptr == NULL)
733 {
734 return MEMCACHED_MEMORY_ALLOCATION_FAILURE;
735 }
736
737 memcached_return_t rc;
738 if (memcached_failed(rc= memcached_server_add(&memc, hostname, port)))
739 {
740 memcached_free(&memc);
741 return rc;
742 }
743
744 if (memcached_success(rc= initialize_query(memc_ptr, true)))
745 {
746 size_t args_length= 0;
747 if (args)
748 {
749 args_length= strlen(args);
750 rc= memcached_key_test(*memc_ptr, (const char **)&args, &args_length, 1);
751 }
752
753 if (memcached_success(rc))
754 {
755 memcached_instance_st* instance= memcached_instance_fetch(memc_ptr, 0);
756 if (memc.flags.binary_protocol)
757 {
758 rc= binary_stats_fetch(memc_stat, args, args_length, instance, NULL);
759 }
760 else
761 {
762 rc= ascii_stats_fetch(memc_stat, args, args_length, instance, NULL);
763 }
764 }
765 }
766
767 memcached_free(&memc);
768
769 return rc;
770 }
771
772 /*
773 We make a copy of the keys since at some point in the not so distant future
774 we will add support for "found" keys.
775 */
776 char ** memcached_stat_get_keys(memcached_st *shell,
777 memcached_stat_st *,
778 memcached_return_t *error)
779 {
780 Memcached* memc= memcached2Memcached(shell);
781 if (memc)
782 {
783 char **list= static_cast<char **>(libmemcached_malloc(memc, sizeof(memcached_stat_keys)));
784 if (list == NULL)
785 {
786 if (error)
787 {
788 *error= memcached_set_error(*memc, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT);
789 }
790
791 return NULL;
792 }
793
794 memcpy(list, memcached_stat_keys, sizeof(memcached_stat_keys));
795
796 if (error)
797 {
798 *error= MEMCACHED_SUCCESS;
799 }
800
801 return list;
802 }
803
804 return NULL;
805 }
806
807 void memcached_stat_free(const memcached_st *, memcached_stat_st *memc_stat)
808 {
809 WATCHPOINT_ASSERT(memc_stat); // Be polite, but when debugging catch this as an error
810 if (memc_stat)
811 {
812 libmemcached_free(memc_stat->root, memc_stat);
813 }
814 }
815
816 static memcached_return_t call_stat_fn(memcached_st *memc,
817 memcached_instance_st* instance,
818 void *context)
819 {
820 if (memc)
821 {
822 local_context *check= (struct local_context *)context;
823
824 if (memcached_is_binary(memc))
825 {
826 return binary_stats_fetch(NULL, check->args, check->args_length, instance, check);
827 }
828 else
829 {
830 return ascii_stats_fetch(NULL, check->args, check->args_length, instance, check);
831 }
832 }
833
834 return MEMCACHED_INVALID_ARGUMENTS;
835 }
836
837 memcached_return_t memcached_stat_execute(memcached_st *shell, const char *args, memcached_stat_fn func, void *context)
838 {
839 Memcached* memc= memcached2Memcached(shell);
840 if (memcached_fatal(memcached_version(memc)))
841 {
842 return memcached_last_error(memc);
843 }
844
845 local_context check(func, context, args, args ? strlen(args) : 0);
846
847 return memcached_server_execute(memc, call_stat_fn, (void *)&check);
848 }