1 /* -*- Mode: C; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2 #include <stdarg.h>
3 #include <stdio.h>
4 #include <string.h>
5 #include <memcached/extension.h>
6 #include <memcached/extension_loggers.h>
7 #include <memcached/engine.h>
8
9 static EXTENSION_LOG_LEVEL current_log_level = EXTENSION_LOG_WARNING;
10 SERVER_HANDLE_V1 *sapi;
11
stderror_get_name(void)12 static const char *stderror_get_name(void) {
13 return "standard error";
14 }
15
stderror_logger_log(EXTENSION_LOG_LEVEL severity,const void * client_cookie,const char * fmt,...)16 static void stderror_logger_log(EXTENSION_LOG_LEVEL severity,
17 const void* client_cookie,
18 const char *fmt, ...)
19 {
20 if (severity >= current_log_level) {
21 (void)client_cookie;
22 int len = strlen(fmt);
23 bool needlf = (len > 0 && fmt[len - 1] != '\n');
24 va_list ap;
25 va_start(ap, fmt);
26 vfprintf(stderr, fmt, ap);
27 va_end(ap);
28 if (needlf) {
29 fprintf(stderr, "\n");
30 }
31 fflush(stderr);
32 }
33 }
34
35 static EXTENSION_LOGGER_DESCRIPTOR stderror_logger_descriptor = {
36 .get_name = stderror_get_name,
37 .log = stderror_logger_log
38 };
39
on_log_level(const void * cookie,ENGINE_EVENT_TYPE type,const void * event_data,const void * cb_data)40 static void on_log_level(const void *cookie,
41 ENGINE_EVENT_TYPE type,
42 const void *event_data,
43 const void *cb_data) {
44 if (sapi != NULL) {
45 current_log_level = sapi->log->get_level();
46 }
47 }
48
memcached_initialize_stderr_logger(GET_SERVER_API get_server_api)49 EXTENSION_ERROR_CODE memcached_initialize_stderr_logger(GET_SERVER_API get_server_api) {
50 sapi = get_server_api();
51 if (sapi == NULL) {
52 return EXTENSION_FATAL;
53 }
54
55 current_log_level = sapi->log->get_level();
56 sapi->callback->register_callback(NULL, ON_LOG_LEVEL,
57 on_log_level, NULL);
58
59 return EXTENSION_SUCCESS;
60 }
61
null_get_name(void)62 static const char *null_get_name(void) {
63 return "/dev/null";
64 }
65
null_logger_log(EXTENSION_LOG_LEVEL severity,const void * client_cookie,const char * fmt,...)66 static void null_logger_log(EXTENSION_LOG_LEVEL severity,
67 const void* client_cookie,
68 const char *fmt, ...)
69 {
70 (void)severity;
71 (void)client_cookie;
72 (void)fmt;
73 /* EMPTY */
74 }
75
76 static EXTENSION_LOGGER_DESCRIPTOR null_logger_descriptor = {
77 .get_name = null_get_name,
78 .log = null_logger_log
79 };
80
get_null_logger(void)81 EXTENSION_LOGGER_DESCRIPTOR* get_null_logger(void) {
82 return &null_logger_descriptor;
83 }
84
get_stderr_logger(void)85 EXTENSION_LOGGER_DESCRIPTOR* get_stderr_logger(void) {
86 return &stderror_logger_descriptor;
87 }
88