122 lines
2.8 KiB
C
122 lines
2.8 KiB
C
/********************************************
|
|
* @Date: 2024 08 12
|
|
* @Description: 日志拦截器
|
|
*********************************************/
|
|
|
|
#include "logging/logging-interceptor.h"
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
static log_Interceptor *G_interceptor = NULL;
|
|
|
|
static char **G_keywords = NULL;
|
|
|
|
static void get_next(char *str, int *next) {
|
|
next[1] = 0;
|
|
int i = 1;
|
|
int j = 0;
|
|
while (i < strlen(str)) {
|
|
if (j == 0 || str[i] == str[j]) {
|
|
next[++i] = ++j;
|
|
} else {
|
|
j = next[j];
|
|
}
|
|
}
|
|
}
|
|
|
|
static bool kmp_search(char *substr, char *master) {
|
|
int i = 0;
|
|
int j = 0;
|
|
int substrlen = strlen(substr);
|
|
int masterlen = strlen(master);
|
|
int *next = (int *)malloc(sizeof(int) * substrlen + 1);
|
|
get_next(substr, next);
|
|
|
|
while (i < masterlen && j < substrlen) {
|
|
if (master[i] == substr[j]) {
|
|
i++;
|
|
j++;
|
|
} else {
|
|
if (j == 0) {
|
|
i++;
|
|
} else {
|
|
j = next[j];
|
|
}
|
|
}
|
|
}
|
|
|
|
free(next);
|
|
if (j == substrlen)
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* @description 处理
|
|
* @param
|
|
* @return
|
|
*/
|
|
static bool _disposeSubstring(char *level, const char *message, ...) {
|
|
int count = 0;
|
|
|
|
if (G_keywords == NULL) {
|
|
return false;
|
|
}
|
|
|
|
while (G_keywords[count] != NULL) {
|
|
if (kmp_search(G_keywords[count], (char *)message)) {
|
|
return true;
|
|
}
|
|
count++;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* @description : 完成拦截器自我释放内存
|
|
*/
|
|
static void _freeSubstring(log_Interceptor *interceptor) {
|
|
int sum = 0;
|
|
while (G_keywords[sum] != NULL) {
|
|
free(G_keywords[sum]);
|
|
sum++;
|
|
}
|
|
free(G_keywords);
|
|
G_keywords = NULL;
|
|
|
|
if (interceptor->handler != NULL) {
|
|
interceptor->handler->_free(interceptor->handler);
|
|
}
|
|
|
|
free(interceptor);
|
|
}
|
|
|
|
/**
|
|
* @description : 子字符串拦截器
|
|
*/
|
|
log_Interceptor *substringInterceptor(char *keywords[],
|
|
int count,
|
|
log_level level,
|
|
log_Handler *handler) {
|
|
log_Interceptor *interceptor =
|
|
(log_Interceptor *)malloc(sizeof(log_Interceptor));
|
|
interceptor->_dispose = _disposeSubstring;
|
|
interceptor->handler = handler;
|
|
interceptor->level = level;
|
|
interceptor->_free = _freeSubstring;
|
|
|
|
G_keywords = (char **)malloc((sizeof(G_keywords) * count) + 1);
|
|
|
|
for (int i = 0; i < count; i++) {
|
|
G_keywords[i] = (char *)malloc(strlen(keywords[i]) + 1);
|
|
strcpy(G_keywords[i], keywords[i]);
|
|
}
|
|
G_keywords[count] = NULL;
|
|
|
|
G_interceptor = interceptor;
|
|
return G_interceptor;
|
|
} |