Make use of new vdo().
[awesomized/libmemcached] / libmemcached / stats.c
1 /*
2 */
3
4 #include "common.h"
5
6 static const char *memcached_stat_keys[] = {
7 "pid",
8 "uptime",
9 "time",
10 "version",
11 "pointer_size",
12 "rusage_user",
13 "rusage_system",
14 "curr_items",
15 "total_items",
16 "bytes",
17 "curr_connections",
18 "total_connections",
19 "connection_structures",
20 "cmd_get",
21 "cmd_set",
22 "get_hits",
23 "get_misses",
24 "evictions",
25 "bytes_read",
26 "bytes_written",
27 "limit_maxbytes",
28 "threads",
29 NULL
30 };
31
32
33 static memcached_return_t set_data(memcached_stat_st *memc_stat, char *key, char *value)
34 {
35
36 if (strlen(key) < 1)
37 {
38 WATCHPOINT_STRING(key);
39 return MEMCACHED_UNKNOWN_STAT_KEY;
40 }
41 else if (!strcmp("pid", key))
42 {
43 memc_stat->pid= (uint32_t) strtol(value, (char **)NULL, 10);
44 }
45 else if (!strcmp("uptime", key))
46 {
47 memc_stat->uptime= (uint32_t) strtol(value, (char **)NULL, 10);
48 }
49 else if (!strcmp("time", key))
50 {
51 memc_stat->time= (uint32_t) strtol(value, (char **)NULL, 10);
52 }
53 else if (!strcmp("version", key))
54 {
55 memcpy(memc_stat->version, value, strlen(value));
56 memc_stat->version[strlen(value)]= 0;
57 }
58 else if (!strcmp("pointer_size", key))
59 {
60 memc_stat->pointer_size= (uint32_t) strtol(value, (char **)NULL, 10);
61 }
62 else if (!strcmp("rusage_user", key))
63 {
64 char *walk_ptr;
65 for (walk_ptr= value; (!ispunct(*walk_ptr)); walk_ptr++);
66 *walk_ptr= 0;
67 walk_ptr++;
68 memc_stat->rusage_user_seconds= (uint32_t) strtol(value, (char **)NULL, 10);
69 memc_stat->rusage_user_microseconds= (uint32_t) strtol(walk_ptr, (char **)NULL, 10);
70 }
71 else if (!strcmp("rusage_system", key))
72 {
73 char *walk_ptr;
74 for (walk_ptr= value; (!ispunct(*walk_ptr)); walk_ptr++);
75 *walk_ptr= 0;
76 walk_ptr++;
77 memc_stat->rusage_system_seconds= (uint32_t) strtol(value, (char **)NULL, 10);
78 memc_stat->rusage_system_microseconds= (uint32_t) strtol(walk_ptr, (char **)NULL, 10);
79 }
80 else if (!strcmp("curr_items", key))
81 {
82 memc_stat->curr_items= (uint32_t) strtol(value, (char **)NULL, 10);
83 }
84 else if (!strcmp("total_items", key))
85 {
86 memc_stat->total_items= (uint32_t) strtol(value, (char **)NULL, 10);
87 }
88 else if (!strcmp("bytes_read", key))
89 {
90 memc_stat->bytes_read= (uint32_t) strtoll(value, (char **)NULL, 10);
91 }
92 else if (!strcmp("bytes_written", key))
93 {
94 memc_stat->bytes_written= (uint32_t) strtoll(value, (char **)NULL, 10);
95 }
96 else if (!strcmp("bytes", key))
97 {
98 memc_stat->bytes= (uint32_t) strtoll(value, (char **)NULL, 10);
99 }
100 else if (!strcmp("curr_connections", key))
101 {
102 memc_stat->curr_connections= (uint32_t) strtoll(value, (char **)NULL, 10);
103 }
104 else if (!strcmp("total_connections", key))
105 {
106 memc_stat->total_connections= (uint32_t) strtoll(value, (char **)NULL, 10);
107 }
108 else if (!strcmp("connection_structures", key))
109 {
110 memc_stat->connection_structures= (uint32_t) strtol(value, (char **)NULL, 10);
111 }
112 else if (!strcmp("cmd_get", key))
113 {
114 memc_stat->cmd_get= (uint64_t) strtoll(value, (char **)NULL, 10);
115 }
116 else if (!strcmp("cmd_set", key))
117 {
118 memc_stat->cmd_set= (uint64_t) strtoll(value, (char **)NULL, 10);
119 }
120 else if (!strcmp("get_hits", key))
121 {
122 memc_stat->get_hits= (uint64_t) strtoll(value, (char **)NULL, 10);
123 }
124 else if (!strcmp("get_misses", key))
125 {
126 memc_stat->get_misses= (uint64_t)strtoll(value, (char **)NULL, 10);
127 }
128 else if (!strcmp("evictions", key))
129 {
130 memc_stat->evictions= (uint64_t)strtoll(value, (char **)NULL, 10);
131 }
132 else if (!strcmp("limit_maxbytes", key))
133 {
134 memc_stat->limit_maxbytes= (uint64_t) strtoll(value, (char **)NULL, 10);
135 }
136 else if (!strcmp("threads", key))
137 {
138 memc_stat->threads= (uint32_t) strtol(value, (char **)NULL, 10);
139 }
140 else if (!(strcmp("delete_misses", key) == 0 ||/* New stats in the 1.3 beta */
141 strcmp("delete_hits", key) == 0 ||/* Just swallow them for now.. */
142 strcmp("incr_misses", key) == 0 ||
143 strcmp("incr_hits", key) == 0 ||
144 strcmp("decr_misses", key) == 0 ||
145 strcmp("decr_hits", key) == 0 ||
146 strcmp("cas_misses", key) == 0 ||
147 strcmp("cas_hits", key) == 0 ||
148 strcmp("cas_badval", key) == 0 ||
149 strcmp("cmd_flush", key) == 0 ||
150 strcmp("accepting_conns", key) == 0 ||
151 strcmp("listen_disabled_num", key) == 0 ||
152 strcmp("conn_yields", key) == 0 ||
153 strcmp("auth_cmds", key) == 0 ||
154 strcmp("auth_errors", key) == 0 ||
155 strcmp("reclaimed", key) == 0))
156 {
157 WATCHPOINT_STRING(key);
158 return MEMCACHED_UNKNOWN_STAT_KEY;
159 }
160
161 return MEMCACHED_SUCCESS;
162 }
163
164 char *memcached_stat_get_value(const memcached_st *ptr, memcached_stat_st *memc_stat,
165 const char *key, memcached_return_t *error)
166 {
167 char buffer[SMALL_STRING_LEN];
168 int length;
169 char *ret;
170
171 *error= MEMCACHED_SUCCESS;
172
173 if (!memcmp("pid", key, strlen("pid")))
174 length= snprintf(buffer, SMALL_STRING_LEN,"%u", memc_stat->pid);
175 else if (!memcmp("uptime", key, strlen("uptime")))
176 length= snprintf(buffer, SMALL_STRING_LEN,"%u", memc_stat->uptime);
177 else if (!memcmp("time", key, strlen("time")))
178 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->time);
179 else if (!memcmp("version", key, strlen("version")))
180 length= snprintf(buffer, SMALL_STRING_LEN,"%s", memc_stat->version);
181 else if (!memcmp("pointer_size", key, strlen("pointer_size")))
182 length= snprintf(buffer, SMALL_STRING_LEN,"%u", memc_stat->pointer_size);
183 else if (!memcmp("rusage_user", key, strlen("rusage_user")))
184 length= snprintf(buffer, SMALL_STRING_LEN,"%u.%u", memc_stat->rusage_user_seconds, memc_stat->rusage_user_microseconds);
185 else if (!memcmp("rusage_system", key, strlen("rusage_system")))
186 length= snprintf(buffer, SMALL_STRING_LEN,"%u.%u", memc_stat->rusage_system_seconds, memc_stat->rusage_system_microseconds);
187 else if (!memcmp("curr_items", key, strlen("curr_items")))
188 length= snprintf(buffer, SMALL_STRING_LEN,"%u", memc_stat->curr_items);
189 else if (!memcmp("total_items", key, strlen("total_items")))
190 length= snprintf(buffer, SMALL_STRING_LEN,"%u", memc_stat->total_items);
191 else if (!memcmp("curr_connections", key, strlen("curr_connections")))
192 length= snprintf(buffer, SMALL_STRING_LEN,"%u", memc_stat->curr_connections);
193 else if (!memcmp("total_connections", key, strlen("total_connections")))
194 length= snprintf(buffer, SMALL_STRING_LEN,"%u", memc_stat->total_connections);
195 else if (!memcmp("connection_structures", key, strlen("connection_structures")))
196 length= snprintf(buffer, SMALL_STRING_LEN,"%u", memc_stat->connection_structures);
197 else if (!memcmp("cmd_get", key, strlen("cmd_get")))
198 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->cmd_get);
199 else if (!memcmp("cmd_set", key, strlen("cmd_set")))
200 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->cmd_set);
201 else if (!memcmp("get_hits", key, strlen("get_hits")))
202 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->get_hits);
203 else if (!memcmp("get_misses", key, strlen("get_misses")))
204 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->get_misses);
205 else if (!memcmp("evictions", key, strlen("evictions")))
206 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->evictions);
207 else if (!memcmp("bytes_read", key, strlen("bytes_read")))
208 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->bytes_read);
209 else if (!memcmp("bytes_written", key, strlen("bytes_written")))
210 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->bytes_written);
211 else if (!memcmp("bytes", key, strlen("bytes")))
212 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->bytes);
213 else if (!memcmp("limit_maxbytes", key, strlen("limit_maxbytes")))
214 length= snprintf(buffer, SMALL_STRING_LEN,"%llu", (unsigned long long)memc_stat->limit_maxbytes);
215 else if (!memcmp("threads", key, strlen("threads")))
216 length= snprintf(buffer, SMALL_STRING_LEN,"%u", memc_stat->threads);
217 else
218 {
219 *error= MEMCACHED_NOTFOUND;
220 return NULL;
221 }
222
223 ret= libmemcached_malloc(ptr, (size_t) (length + 1));
224 memcpy(ret, buffer, (size_t) length);
225 ret[length]= '\0';
226
227 return ret;
228 }
229
230 static memcached_return_t binary_stats_fetch(memcached_stat_st *memc_stat,
231 char *args,
232 memcached_server_write_instance_st instance)
233 {
234 memcached_return_t rc;
235
236 char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE];
237 protocol_binary_request_stats request= {.bytes= {0}};
238 request.message.header.request.magic= PROTOCOL_BINARY_REQ;
239 request.message.header.request.opcode= PROTOCOL_BINARY_CMD_STAT;
240 request.message.header.request.datatype= PROTOCOL_BINARY_RAW_BYTES;
241
242 if (args != NULL)
243 {
244 size_t len= strlen(args);
245
246 rc= memcached_validate_key_length(len, true);
247 unlikely (rc != MEMCACHED_SUCCESS)
248 return rc;
249
250 request.message.header.request.keylen= htons((uint16_t)len);
251 request.message.header.request.bodylen= htonl((uint32_t) len);
252
253 struct __write_vector_st vector[]=
254 {
255 { .length= sizeof(request.bytes), .buffer= request.bytes },
256 { .length= len, .buffer= args }
257 };
258
259 if (memcached_vdo(instance, vector, 2, true) != MEMCACHED_SUCCESS)
260 {
261 memcached_io_reset(instance);
262 return MEMCACHED_WRITE_FAILURE;
263 }
264 }
265 else
266 {
267 if (memcached_do(instance, request.bytes,
268 sizeof(request.bytes), true) != MEMCACHED_SUCCESS)
269 {
270 memcached_io_reset(instance);
271 return MEMCACHED_WRITE_FAILURE;
272 }
273 }
274
275 memcached_server_response_decrement(instance);
276 do
277 {
278 rc= memcached_response(instance, buffer, sizeof(buffer), NULL);
279
280 if (rc == MEMCACHED_END)
281 break;
282
283 unlikely (rc != MEMCACHED_SUCCESS)
284 {
285 memcached_io_reset(instance);
286 return rc;
287 }
288
289 unlikely((set_data(memc_stat, buffer, buffer + strlen(buffer) + 1)) == MEMCACHED_UNKNOWN_STAT_KEY)
290 {
291 WATCHPOINT_ERROR(MEMCACHED_UNKNOWN_STAT_KEY);
292 WATCHPOINT_ASSERT(0);
293 }
294 } while (1);
295
296 /* shit... memcached_response will decrement the counter, so I need to
297 ** reset it.. todo: look at this and try to find a better solution.
298 */
299 instance->cursor_active= 0;
300
301 return MEMCACHED_SUCCESS;
302 }
303
304 static memcached_return_t ascii_stats_fetch(memcached_stat_st *memc_stat,
305 char *args,
306 memcached_server_write_instance_st instance)
307 {
308 memcached_return_t rc;
309 char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE];
310 size_t send_length;
311
312 if (args)
313 send_length= (size_t) snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE,
314 "stats %s\r\n", args);
315 else
316 send_length= (size_t) snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE,
317 "stats\r\n");
318
319 if (send_length >= MEMCACHED_DEFAULT_COMMAND_SIZE)
320 return MEMCACHED_WRITE_FAILURE;
321
322 rc= memcached_do(instance, buffer, send_length, true);
323 if (rc != MEMCACHED_SUCCESS)
324 goto error;
325
326 while (1)
327 {
328 rc= memcached_response(instance, buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, NULL);
329
330 if (rc == MEMCACHED_STAT)
331 {
332 char *string_ptr, *end_ptr;
333 char *key, *value;
334
335 string_ptr= buffer;
336 string_ptr+= 5; /* Move past STAT */
337 for (end_ptr= string_ptr; isgraph(*end_ptr); end_ptr++);
338 key= string_ptr;
339 key[(size_t)(end_ptr-string_ptr)]= 0;
340
341 string_ptr= end_ptr + 1;
342 for (end_ptr= string_ptr; !(isspace(*end_ptr)); end_ptr++);
343 value= string_ptr;
344 value[(size_t)(end_ptr-string_ptr)]= 0;
345 string_ptr= end_ptr + 2;
346 unlikely((set_data(memc_stat, key, value)) == MEMCACHED_UNKNOWN_STAT_KEY)
347 {
348 WATCHPOINT_ERROR(MEMCACHED_UNKNOWN_STAT_KEY);
349 WATCHPOINT_ASSERT(0);
350 }
351 }
352 else
353 break;
354 }
355
356 error:
357 if (rc == MEMCACHED_END)
358 return MEMCACHED_SUCCESS;
359 else
360 return rc;
361 }
362
363 memcached_stat_st *memcached_stat(memcached_st *ptr, char *args, memcached_return_t *error)
364 {
365 memcached_return_t rc;
366 memcached_stat_st *stats;
367
368 unlikely (ptr->flags.use_udp)
369 {
370 *error= MEMCACHED_NOT_SUPPORTED;
371 return NULL;
372 }
373
374 stats= libmemcached_calloc(ptr, memcached_server_count(ptr), sizeof(memcached_stat_st));
375
376 if (! stats)
377 {
378 *error= MEMCACHED_MEMORY_ALLOCATION_FAILURE;
379 return NULL;
380 }
381
382 rc= MEMCACHED_SUCCESS;
383 for (uint32_t x= 0; x < memcached_server_count(ptr); x++)
384 {
385 memcached_return_t temp_return;
386 memcached_server_write_instance_st instance;
387 memcached_stat_st *stat_instance;
388
389 stat_instance= stats + x;
390
391 stat_instance->root= ptr;
392
393 instance= memcached_server_instance_fetch(ptr, x);
394
395 if (ptr->flags.binary_protocol)
396 {
397 temp_return= binary_stats_fetch(stat_instance, args, instance);
398 }
399 else
400 {
401 temp_return= ascii_stats_fetch(stat_instance, args, instance);
402 }
403
404 if (temp_return != MEMCACHED_SUCCESS)
405 rc= MEMCACHED_SOME_ERRORS;
406 }
407
408 *error= rc;
409 return stats;
410 }
411
412 memcached_return_t memcached_stat_servername(memcached_stat_st *memc_stat, char *args,
413 const char *hostname, in_port_t port)
414 {
415 memcached_return_t rc;
416 memcached_st memc;
417 memcached_st *memc_ptr;
418 memcached_server_write_instance_st instance;
419
420 memset(memc_stat, 0, sizeof(memcached_stat_st));
421
422 memc_ptr= memcached_create(&memc);
423 WATCHPOINT_ASSERT(memc_ptr);
424
425 memcached_server_add(&memc, hostname, port);
426
427 instance= memcached_server_instance_fetch(memc_ptr, 0);
428
429 if (memc.flags.binary_protocol)
430 {
431 rc= binary_stats_fetch(memc_stat, args, instance);
432 }
433 else
434 {
435 rc= ascii_stats_fetch(memc_stat, args, instance);
436 }
437
438 memcached_free(&memc);
439
440 return rc;
441 }
442
443 /*
444 We make a copy of the keys since at some point in the not so distant future
445 we will add support for "found" keys.
446 */
447 char ** memcached_stat_get_keys(const memcached_st *ptr,
448 memcached_stat_st *memc_stat,
449 memcached_return_t *error)
450 {
451 char **list;
452 size_t length= sizeof(memcached_stat_keys);
453
454 (void)memc_stat;
455
456 list= libmemcached_malloc(ptr, length);
457
458 if (! list)
459 {
460 *error= MEMCACHED_MEMORY_ALLOCATION_FAILURE;
461 return NULL;
462 }
463
464 memcpy(list, memcached_stat_keys, sizeof(memcached_stat_keys));
465
466 *error= MEMCACHED_SUCCESS;
467
468 return list;
469 }
470
471 void memcached_stat_free(const memcached_st *ptr, memcached_stat_st *memc_stat)
472 {
473 if (memc_stat == NULL)
474 {
475 WATCHPOINT_ASSERT(0); /* Be polite, but when debugging catch this as an error */
476 return;
477 }
478
479 if (memc_stat->root)
480 {
481 libmemcached_free(memc_stat->root, memc_stat);
482 }
483 else if (ptr)
484 {
485 libmemcached_free(ptr, memc_stat);
486 }
487 else
488 {
489 free(memc_stat);
490 }
491 }