diff options
Diffstat (limited to 'storage/mroonga/vendor/groonga/lib/windows_event_logger.c')
| -rw-r--r-- | storage/mroonga/vendor/groonga/lib/windows_event_logger.c | 203 |
1 files changed, 203 insertions, 0 deletions
diff --git a/storage/mroonga/vendor/groonga/lib/windows_event_logger.c b/storage/mroonga/vendor/groonga/lib/windows_event_logger.c new file mode 100644 index 00000000..fb8229ad --- /dev/null +++ b/storage/mroonga/vendor/groonga/lib/windows_event_logger.c @@ -0,0 +1,203 @@ +/* -*- c-basic-offset: 2 -*- */ +/* + Copyright(C) 2015-2017 Brazil + + This library is free software; you can redistribute it and/or + modify it under the terms of the GNU Lesser General Public + License version 2.1 as published by the Free Software Foundation. + + This library is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + Lesser General Public License for more details. + + You should have received a copy of the GNU Lesser General Public + License along with this library; if not, write to the Free Software + Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335 USA +*/ + +#include "grn_logger.h" +#include "grn_ctx.h" +#include "grn_windows.h" + +#include <string.h> + +#ifdef WIN32 + +typedef struct _grn_windows_event_logger_data { + char *event_source_name; + HANDLE event_source; +} grn_windows_event_logger_data; + +static void +windows_event_logger_log(grn_ctx *ctx, grn_log_level level, + const char *timestamp, const char *title, + const char *message, const char *location, + void *user_data) +{ + grn_windows_event_logger_data *data = user_data; + WORD type; + WORD category = 0; + DWORD event_id = 0; + PSID user_sid = NULL; + WORD n_strings = 1; + DWORD event_data_size = 0; + const WCHAR *strings[1]; + LPVOID event_data = NULL; + const char level_marks[] = " EACewnid-"; + grn_obj formatted_buffer; + UINT code_page; + DWORD convert_flags = 0; + int n_converted_chars; + + switch (level) { + case GRN_LOG_NONE : + return; + break; + case GRN_LOG_EMERG : + case GRN_LOG_ALERT : + case GRN_LOG_CRIT : + case GRN_LOG_ERROR : + type = EVENTLOG_ERROR_TYPE; + break; + case GRN_LOG_WARNING : + type = EVENTLOG_WARNING_TYPE; + break; + case GRN_LOG_NOTICE : + case GRN_LOG_INFO : + case GRN_LOG_DEBUG : + case GRN_LOG_DUMP : + type = EVENTLOG_INFORMATION_TYPE; + break; + default : + type = EVENTLOG_ERROR_TYPE; + break; + } + + if (data->event_source == INVALID_HANDLE_VALUE) { + data->event_source = RegisterEventSourceA(NULL, data->event_source_name); + if (data->event_source == INVALID_HANDLE_VALUE) { + return; + } + } + + GRN_TEXT_INIT(&formatted_buffer, 0); + if (location && location[0]) { + grn_text_printf(ctx, &formatted_buffer, "%s|%c|%s %s %s", + timestamp, level_marks[level], title, message, location); + } else { + grn_text_printf(ctx, &formatted_buffer, "%s|%c|%s %s", + timestamp, level_marks[level], title, message); + } + + code_page = grn_windows_encoding_to_code_page(ctx->encoding); + + n_converted_chars = MultiByteToWideChar(code_page, + convert_flags, + GRN_TEXT_VALUE(&formatted_buffer), + GRN_TEXT_LEN(&formatted_buffer), + NULL, + 0); +#define CONVERTED_BUFFER_SIZE 512 + if (n_converted_chars < CONVERTED_BUFFER_SIZE) { + WCHAR converted_buffer[CONVERTED_BUFFER_SIZE]; + n_converted_chars = MultiByteToWideChar(code_page, + convert_flags, + GRN_TEXT_VALUE(&formatted_buffer), + GRN_TEXT_LEN(&formatted_buffer), + converted_buffer, + CONVERTED_BUFFER_SIZE); + converted_buffer[n_converted_chars] = L'\0'; + strings[0] = converted_buffer; + ReportEventW(data->event_source, type, category, event_id, user_sid, + n_strings, event_data_size, + strings, event_data); +#undef CONVERTED_BUFFER_SIZE + } else { + WCHAR *converted; + converted = GRN_MALLOCN(WCHAR, n_converted_chars); + n_converted_chars = MultiByteToWideChar(code_page, + convert_flags, + GRN_TEXT_VALUE(&formatted_buffer), + GRN_TEXT_LEN(&formatted_buffer), + converted, + n_converted_chars); + converted[n_converted_chars] = L'\0'; + strings[0] = converted; + ReportEventW(data->event_source, type, category, event_id, user_sid, + n_strings, event_data_size, + strings, event_data); + GRN_FREE(converted); + } + GRN_OBJ_FIN(ctx, &formatted_buffer); +} + +static void +windows_event_logger_reopen(grn_ctx *ctx, void *user_data) +{ +} + +static void +windows_event_logger_fin(grn_ctx *ctx, void *user_data) +{ + grn_windows_event_logger_data *data = user_data; + + free(data->event_source_name); + if (data->event_source != INVALID_HANDLE_VALUE) { + DeregisterEventSource(data->event_source); + } + free(data); +} +#endif /* WIN32 */ + +grn_rc +grn_windows_event_logger_set(grn_ctx *ctx, const char *event_source_name) +{ +#ifdef WIN32 + grn_rc rc; + grn_logger windows_event_logger; + grn_windows_event_logger_data *data; + + if (ctx) { + GRN_API_ENTER; + } + + data = malloc(sizeof(grn_windows_event_logger_data)); + if (!data) { + if (ctx) { + ERR(GRN_NO_MEMORY_AVAILABLE, + "failed to allocate user data for Windows event logger"); + GRN_API_RETURN(ctx->rc); + } else { + return GRN_NO_MEMORY_AVAILABLE; + } + } + + if (event_source_name) { + data->event_source_name = grn_strdup_raw(event_source_name); + } else { + data->event_source_name = grn_strdup_raw("libgroonga"); + } + data->event_source = INVALID_HANDLE_VALUE; + + windows_event_logger.max_level = GRN_LOG_DEFAULT_LEVEL; + windows_event_logger.flags = GRN_LOG_TIME | GRN_LOG_MESSAGE; + windows_event_logger.user_data = data; + windows_event_logger.log = windows_event_logger_log; + windows_event_logger.reopen = windows_event_logger_reopen; + windows_event_logger.fin = windows_event_logger_fin; + + rc = grn_logger_set(ctx, &windows_event_logger); + if (rc != GRN_SUCCESS) { + windows_event_logger.fin(ctx, windows_event_logger.user_data); + } + + if (ctx) { + GRN_API_RETURN(rc); + } else { + return rc; + } +#else /* WIN32 */ + return GRN_FUNCTION_NOT_IMPLEMENTED; +#endif /* WIN32 */ +} |