gfwleak/mesa-framework-mesa-handle-logger
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

使用mkstemp生成临时conf文件

Browse Source
This commit is contained in:
“yangwei”
2019-09-04 20:04:43 +08:00
parent 117720197e
commit 635bb180bf
1 changed files with 114 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

149
src/MESA_handle_logger.c
View File

@@ -1,11 +1,15 @@
#include "MESA_handle_logger.h"
#include "zlog.h"
#include <fcntl.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
static int g_zlog_conf_fp = -1;
char tmp_conf_filepath[1024] = "";
#define MAX_HANDLE_LOG_PATH 4096
#define GLOB_ZLOG_CONF "GLOB_ZLOG_CONF"
@@ -113,61 +117,131 @@ static zlog_category_t *init_zlog(const char *conf_path, const char *category)
return zc;
}
static int get_filepath(int fp, char *buf, int buflen)
{
if(buf == NULL || buflen <=0)return -1;
char proclnk[0xFFF] = "";
//int fno = fileno(fp);
int fno = fp;
sprintf(proclnk, "/proc/self/fd/%d", fno);
int r = readlink(proclnk, buf, buflen);
return r;
}
static char tmp_template[] = "/tmp/handle_loggger.XXXXXX";
static void escape_for_zlog(char *in_buf, int buflen)
{
if(in_buf == NULL || buflen <= 0)return;
int i = 0;
for(i = 0; i < buflen; i++)
{
if ((!isalnum(in_buf[i])) && (in_buf[i] != '_') && (in_buf[i] != '-') && (in_buf[i] != '*') && (in_buf[i] != '!'))
{
in_buf[i] = '_';
}
}
return;
}
void *MESA_create_runtime_log_handle(const char *file_path, int level)
{
if(file_path == NULL)
return NULL;
int rc;
zlog_category_t *zc;
int rc = -1;
zlog_category_t *zc = NULL;
FILE *fp = NULL;
log_handle_t *p_handle = NULL;
const char *p_conf_path = NULL;
const char *p_name = NULL;
char zlog_rule_conf_content[MAX_HANDLE_LOG_PATH + 1];
char handle_name[MAX_HANDLE_LOG_PATH];
char *p_path_end = rindex(file_path, '/');
int path_len = 0;
if (p_path_end != NULL)
{
path_len = p_path_end - file_path;
if(path_len == 0)return NULL;
p_name = p_path_end+1;
}
else
{
p_name = file_path;
}
char *pathvar = getenv(GLOB_ZLOG_CONF);
char *p_name = p_path_end+1;
strcpy(handle_name, file_path);
escape_for_zlog(handle_name, strlen(handle_name));
p_name = handle_name;
if (pathvar == NULL)
{
//creating file_path failed, return NULL
if (path_len > 0 && create_path(file_path, path_len) < 0)return NULL;
char default_zlog_conf_path[MAX_HANDLE_LOG_PATH + 5];
snprintf(default_zlog_conf_path, sizeof(default_zlog_conf_path), "%s.conf", file_path);
if (access(default_zlog_conf_path, R_OK) != 0)
if (create_path(file_path, p_path_end - file_path) < 0)
return NULL;
if (g_zlog_conf_fp == -1)
{
fp = fopen(default_zlog_conf_path, "w");
if(fp == NULL)return NULL;
char zlog_rule_conf_content[MAX_HANDLE_LOG_PATH + 1];
snprintf(zlog_rule_conf_content, sizeof(zlog_rule_conf_content),
"[global]\ndefault format = \"%%d(%%c), %%V, %%m%%n\" \n[levels]\nDEBUG=10\nINFO=20\nFATAL=30\n[rules]\n %s.* \"%s.%%d(%%F)\"",
p_name, file_path);
fwrite(zlog_rule_conf_content, strlen(zlog_rule_conf_content), 1, fp);
fclose(fp);
//g_zlog_conf_fp = tmpfile();
char temp_filename[1024]="";
sprintf(temp_filename, "/tmp/MESA_handle_logger_%d.XXXXXX", getpid());
g_zlog_conf_fp = mkstemp(temp_filename);
if (g_zlog_conf_fp == -1)
{
return NULL;
}
if (get_filepath(g_zlog_conf_fp, tmp_conf_filepath, sizeof(tmp_conf_filepath)) < 0)
{
goto error;
}
snprintf(zlog_rule_conf_content, sizeof(zlog_rule_conf_content),
"[global]\ndefault format = \"%%d(%%c), %%V, %%m%%n\" \n[levels]\nDEBUG=10\nINFO=20\nFATAL=30\n[rules]\n%s.* \"%s.%%d(%%F)\"",
p_name, file_path);
write(g_zlog_conf_fp, zlog_rule_conf_content, strlen(zlog_rule_conf_content));
fsync(g_zlog_conf_fp);
int rc = zlog_init(tmp_conf_filepath);
if(rc)
{
goto error;
}
zc = zlog_get_category(p_name);
if (!zc)
{
printf("get zlog category %s in %s fail\n", file_path, tmp_conf_filepath);
error:
unlink(tmp_conf_filepath);
close(g_zlog_conf_fp);
g_zlog_conf_fp = -1;
memset(tmp_conf_filepath, 0, sizeof(tmp_conf_filepath));
zlog_fini();
return NULL;
}
}
else
{
zc = zlog_get_category(p_name);
if (!zc)
{
printf("get zlog category %s in %s fail\n", file_path, tmp_conf_filepath);
return NULL;
}
else
{
//if (get_filepath(g_zlog_conf_fp, tmp_conf_filepath, sizeof(tmp_conf_filepath)) < 0)
// return NULL;
snprintf(zlog_rule_conf_content, sizeof(zlog_rule_conf_content),
"\n%s.* \"%s.%%d(%%F)\"",
p_name, file_path);
write(g_zlog_conf_fp, zlog_rule_conf_content, strlen(zlog_rule_conf_content));
fsync(g_zlog_conf_fp);
int rc = zlog_reload(tmp_conf_filepath);
if (rc)return NULL;
zc = zlog_get_category(p_name);
if (!zc)
{
printf("get zlog category %s in %s fail\n", file_path, tmp_conf_filepath);
return NULL;
}
}
}
p_conf_path = default_zlog_conf_path;
}
else
{
if (access(pathvar, R_OK) != 0)
{
return NULL;
}
p_conf_path = pathvar;
zc = init_zlog(pathvar, p_name);
if (zc == NULL)
goto error;
}
zc = init_zlog(p_conf_path, p_name);
if (zc == NULL)return NULL;
p_handle = (log_handle_t *)calloc(sizeof(log_handle_t), 1);
strncpy(p_handle->runtime_log_file, file_path, sizeof(p_handle->runtime_log_file) - 1);
p_handle->runtime_log_level = level;
@@ -182,6 +256,11 @@ void MESA_destroy_runtime_log_handle(void *handle)
{
log_handle_t *p_handle = (log_handle_t *)handle;
zlog_fini();
if(g_zlog_conf_fp != -1)
{
unlink(tmp_conf_filepath);
close(g_zlog_conf_fp);
}
free(handle);
handle = NULL;
}