456 lines
14 KiB
C
Executable File
456 lines
14 KiB
C
Executable File
#include <stdio.h>
|
||
#include <string.h>
|
||
#include <errno.h>
|
||
#include <unistd.h>
|
||
#include <sys/stat.h>
|
||
|
||
#include "ulog.h"
|
||
#include "log_common.h"
|
||
#include "ulog_in.h"
|
||
#include "sev_sched.h"
|
||
|
||
#define FILTER_CONTENT ":msg,contains,\""MODULE_FMT"\"\n"
|
||
#define DEFAULT_CONFIG_FILE_TMP "/etc/log-sched.conf.tmp"
|
||
|
||
|
||
static int copy_file(const char *src, const char *dst)
|
||
{
|
||
//return -1;
|
||
int ret = -1;
|
||
FILE *src_fp = NULL,*dst_fp = NULL;
|
||
src_fp = fopen(src, "r");
|
||
if (src_fp == NULL) {
|
||
ULOG_ERR(g_log, "Opening src file:%s is failure:%s", src, strerror(errno));
|
||
goto END;
|
||
}
|
||
dst_fp = fopen(dst, "w");
|
||
if (dst_fp == NULL) {
|
||
ULOG_ERR(g_log, "Opening dst file:%s is failure:%s", src, strerror(errno));
|
||
goto END;
|
||
}
|
||
|
||
char buff[1024];
|
||
int len;
|
||
int n;
|
||
while((len = fread(buff, 1, sizeof(buff), src_fp)) > 0)
|
||
{
|
||
n = fwrite(buff, 1, len ,dst_fp);
|
||
if (n != len) {
|
||
ULOG_ERR(g_log, "Configure file which is written[length:%d, actual len:%n] is failure:%s", n, len, strerror(errno));
|
||
goto END;
|
||
}
|
||
|
||
if (feof(src_fp) == 0) {
|
||
break;
|
||
}
|
||
}
|
||
|
||
ret = 0;
|
||
END:
|
||
if (src_fp != NULL) {
|
||
fclose(src_fp);
|
||
}
|
||
if (dst_fp != NULL) {
|
||
fclose(dst_fp);
|
||
}
|
||
|
||
return ret;
|
||
}
|
||
|
||
static int __log_conf(const char *mode, const u8 level, const char *conf_path, const char *conf_file, const char *filter_mod,
|
||
int (*cb_content)(FILE *fp, const u8 level, const char *filter_mod, void *arg), void *arg)
|
||
{
|
||
FILE *fp = NULL;
|
||
int ret = -1;
|
||
u8 exist_backup = 1;
|
||
|
||
/********** rsyslog configure **********/
|
||
char conf_path_file[MAX_PATH_SZ], bak_file[MAX_PATH_SZ];
|
||
|
||
snprintf(conf_path_file, sizeof(conf_path_file), "%s%s", conf_path, conf_file);
|
||
snprintf(bak_file, sizeof(bak_file), BAK_FILE, conf_file);
|
||
|
||
//if (rename(conf_path_file, bak_file) < 0) {
|
||
if (copy_file(conf_path_file, bak_file) != 0) {
|
||
if (errno == ENOENT) {
|
||
exist_backup = 0;
|
||
ULOG_INFO(g_log, "Been not exist, Configure file:%s don't need to backup", conf_path_file);
|
||
} else {
|
||
ULOG_ERR(g_log, "Baking configure file:%s is failure:%s", conf_path_file, strerror(errno));
|
||
return ret;
|
||
}
|
||
}
|
||
|
||
fp = fopen(conf_path_file, mode);
|
||
if (fp == NULL) {
|
||
ULOG_ERR(g_log, "Opening log configure file:%s is failure:%s", conf_path_file, strerror(errno));
|
||
goto END;
|
||
}
|
||
|
||
ULOG_DEBUG(g_log, "cb_content:%x", cb_content);
|
||
if (cb_content(fp, level, filter_mod, arg) != 0) {
|
||
ULOG_ERR(g_log, "Callback log content is failure");
|
||
goto END;
|
||
}
|
||
|
||
ret = 0;
|
||
END:
|
||
if (fp != NULL) {
|
||
fclose(fp);
|
||
}
|
||
|
||
if (exist_backup == 1) {
|
||
if (ret == -1) {
|
||
// 恢复备份配置
|
||
if (rename(bak_file, conf_path_file) < 0) {
|
||
ULOG_ERR(g_log, "Restoring configure file:%s is failure:%s", conf_path_file, strerror(errno));
|
||
}
|
||
} else {
|
||
if (unlink(bak_file) < 0) {
|
||
ULOG_WARNING(g_log, "Delete back file:%s is failure", bak_file);
|
||
}
|
||
}
|
||
}
|
||
|
||
return ret;
|
||
}
|
||
|
||
int log_conf(const u8 level, const char *conf_path, const char *conf_file, const char *filter_mod,
|
||
int (*cb_content)(FILE *fp, const u8 level, const char *filter_mod, void *arg), void *arg)
|
||
{
|
||
return __log_conf("w", level, conf_path, conf_file, filter_mod, cb_content, arg);
|
||
}
|
||
|
||
int log_conf_append(const u8 level, const char *conf_path, const char *conf_file, const char *filter_mod,
|
||
int (*cb_content)(FILE *fp, const u8 level, const char *filter_mod, void *arg), void *arg)
|
||
{
|
||
return __log_conf("a", level, conf_path, conf_file, filter_mod, cb_content, arg);
|
||
}
|
||
|
||
int log_level_to_str(const u8 level, char *str, u32 len)
|
||
{
|
||
u8 i;
|
||
char tmp[20];
|
||
for (i = 0; i <= level;) {
|
||
if (snprintf(tmp, sizeof(tmp), "*.=%s", g_level_array[i].str) < 0) {
|
||
ULOG_ERR(g_log, "Setting content of log file configure is failure");
|
||
return -1;
|
||
}
|
||
strcat(str, tmp);
|
||
i++;
|
||
if (level >= i) {
|
||
strcat(str, ";");
|
||
}
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
|
||
int write_conf_content(FILE *fp, const u8 level, const char *filter_mod, void *arg)
|
||
{
|
||
int ret = -1;
|
||
char line[MAX_LINE_SZ + 100] = {0};
|
||
ULOG_DEBUG(g_log, "filter module:%s\n", filter_mod);
|
||
if ((filter_mod != NULL) && (strlen(filter_mod) > 0)) {
|
||
snprintf(line, sizeof(line), FILTER_CONTENT, filter_mod);
|
||
if (fputs(line, fp) == EOF) {
|
||
ULOG_ERR(g_log, "Message filter:%s of configure file which is written is failure:%s",
|
||
filter_mod, strerror(errno));
|
||
goto END;
|
||
}
|
||
}
|
||
|
||
line[0] = '\0'; // 清零
|
||
if (log_level_to_str(level, line, sizeof(line)) != 0) {
|
||
ULOG_ERR(g_log, "Setting content of log file configure is failure");
|
||
goto END;
|
||
}
|
||
strcat(line, REDIRECT_SEPERATE);
|
||
|
||
strcat(line, (const char *)arg);
|
||
strcat(line, "\n");
|
||
|
||
//if (fputs(line, fp) == EOF) {
|
||
int n = fwrite(line, 1, strlen(line), fp);
|
||
if (n != strlen(line)) {
|
||
ULOG_ERR(g_log, "Configure file which is written[length:%d] is failure:%s", n, strerror(errno));
|
||
goto END;
|
||
}
|
||
|
||
ret = 0;
|
||
END:
|
||
return ret;
|
||
}
|
||
|
||
int modify_authorizing(const char *redirect_path)
|
||
{
|
||
if (chmod(redirect_path, S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH) < 0) {
|
||
ULOG_ERR(g_log, "Authorizing of %s which is modified is failure:%s", redirect_path, strerror(errno));
|
||
return -1;
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
int write_conf_content_authorizing(FILE *fp, const u8 level, const char *filter_mod, void *arg)
|
||
{
|
||
if (write_conf_content(fp, level, filter_mod, arg) != 0) {
|
||
ULOG_ERR(g_log, "configure of log conosle which is written is failure");
|
||
return -1;
|
||
}
|
||
|
||
if (modify_authorizing((const char *)arg) != 0) {
|
||
ULOG_ERR(g_log, "Modifying authorizing of %s is failure", (const char *)arg);
|
||
return -1;
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
int log_off_with_file(const char *path, const char *file_name)
|
||
{
|
||
char file[MAX_PATH_SZ];
|
||
int ret = -1;
|
||
|
||
if (snprintf(file, sizeof(file), "%s%s", path, file_name) < 0) {
|
||
ULOG_ERR(g_log, "Setting file path(dir:%s, file:%s) is failure",
|
||
path, file_name);
|
||
return ret;
|
||
}
|
||
if (unlink(file) < 0) {
|
||
if (errno != ENOENT) {
|
||
ULOG_ERR(g_log, "Deleting configure file(dir:%s, file:%s) is failure:%s",
|
||
path, file_name, strerror(errno));
|
||
return ret;
|
||
} else {
|
||
ULOG_DEBUG(g_log, "File(dir:%s, file:%s) is not exist", path, file_name);
|
||
}
|
||
}
|
||
ret = 0;
|
||
return ret;
|
||
}
|
||
|
||
void rpc_conf_proc(rpc_conn *conn, pointer input, int input_len, int need_len, rpc_cb cb, void *arg)
|
||
{
|
||
char str_err[1024];
|
||
|
||
if (input == NULL) {
|
||
strncpy(str_err, "Input is not null", sizeof(str_err));
|
||
ULOG_WARNING(g_log, str_err);
|
||
goto FAIL;
|
||
}
|
||
|
||
if (input_len < need_len) {
|
||
if (snprintf(str_err, sizeof(str_err),
|
||
"The input paramter of rpc log is needed length of %d, but the actual length is %d",
|
||
need_len, input_len) < 0) {
|
||
strncpy(str_err, "Setting error message is failure", sizeof(str_err));
|
||
ULOG_ERR(g_log, str_err);
|
||
} else {
|
||
ULOG_WARNING(g_log, str_err);
|
||
}
|
||
goto FAIL;
|
||
}
|
||
|
||
// ret == 1,不需要重启服务;ret == 0，需要重启服务
|
||
int ret = cb(input, arg, str_err, sizeof(str_err));
|
||
if (ret < 0) {
|
||
ULOG_ERR(g_log, str_err);
|
||
goto FAIL;
|
||
}
|
||
|
||
if ((ret == 0) && (log_sev_restart() != 0)) {
|
||
if (snprintf(str_err, sizeof(str_err),
|
||
"Restarting log service is failure") < 0) {
|
||
strncpy(str_err, "Setting error message is failure", sizeof(str_err));
|
||
ULOG_ERR(g_log, str_err);
|
||
}
|
||
goto FAIL;
|
||
}
|
||
rpc_return_null(conn);
|
||
return;
|
||
FAIL:
|
||
rpc_return_error(conn, RET_ERR, str_err);
|
||
}
|
||
|
||
/* 写配置到配置文件。追加在配置文件最后 */
|
||
int write_log_file_conf(const char *p_key_str, const char *p_value_str)
|
||
{
|
||
char conf_value_str[MAX_LINE_SZ] = "";
|
||
|
||
char buffer[MAX_LINE_SZ] = "";
|
||
char *p_pos = NULL;
|
||
int is_exist = 0;
|
||
FILE *tmp_conf_fp = NULL;
|
||
|
||
if (NULL == p_key_str || '\0' == p_key_str[0] || NULL == p_value_str) {
|
||
ULOG_ERR(g_log, "Bad input");
|
||
return -1;
|
||
}
|
||
|
||
/* 定位到起始位置 */
|
||
if (fseek(g_conf_fp, 0, SEEK_SET) == -1) {
|
||
ULOG_ERR(g_log, "Seeknig config to begin is faiure:%s", strerror(errno));
|
||
return -1;
|
||
}
|
||
|
||
/* 查看配置是否已存在 */
|
||
while (fgets(buffer,MAX_LINE_SZ,g_conf_fp)!=NULL) {
|
||
p_pos = strstr(buffer, p_key_str);
|
||
if (p_pos != NULL) {
|
||
memset(conf_value_str, 0, sizeof(conf_value_str));
|
||
strcpy(conf_value_str, p_pos+strlen(p_key_str));
|
||
|
||
is_exist = 1;
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (is_exist == 1) {
|
||
/* 判断已有配置是否与当前传入配置相同 */
|
||
if (strlen(conf_value_str) == strlen(p_value_str) && 0 == strcmp(conf_value_str, p_value_str)) {
|
||
/* 无需做任何变更，本函数返回 */
|
||
return 0;
|
||
}
|
||
|
||
/* 定位到起始位置 */
|
||
if (fseek(g_conf_fp, 0, SEEK_SET) == -1) {
|
||
ULOG_ERR(g_log, "Seeknig config to begin is faiure:%s", strerror(errno));
|
||
return -1;
|
||
}
|
||
|
||
/* 打开临时文件 */
|
||
tmp_conf_fp = fopen(DEFAULT_CONFIG_FILE_TMP, "w+");
|
||
if (NULL == tmp_conf_fp){
|
||
ULOG_ERR(g_log, "Open temp config file failed: %s", strerror(errno));
|
||
return -1;
|
||
}
|
||
|
||
/* 将原文件内容写入临时文件，但跳过当前配置相关项 */
|
||
while (fgets(buffer,MAX_LINE_SZ,g_conf_fp)!=NULL) {
|
||
if (strstr(buffer, p_key_str) != NULL) {
|
||
continue;
|
||
}
|
||
fputs(buffer,tmp_conf_fp);
|
||
}
|
||
fflush(tmp_conf_fp);
|
||
|
||
/* 将临时文件写回原配置文件 */
|
||
truncate(g_conf_file, 0);
|
||
// fclose(g_conf_fp);
|
||
// g_conf_fp = fopen(g_conf_file, "w+");
|
||
|
||
if (fseek(g_conf_fp, 0, SEEK_SET) == -1) {
|
||
ULOG_ERR(g_log, "Seeknig config to begin is faiure:%s", strerror(errno));
|
||
return -1;
|
||
}
|
||
|
||
if (fseek(tmp_conf_fp, 0, SEEK_SET) == -1) {
|
||
ULOG_ERR(g_log, "Seeknig config to begin is faiure:%s", strerror(errno));
|
||
return -1;
|
||
}
|
||
|
||
while (fgets(buffer,MAX_LINE_SZ,tmp_conf_fp)!=NULL) {
|
||
fputs(buffer,g_conf_fp);
|
||
}
|
||
fflush(g_conf_fp);
|
||
|
||
/* 关闭临时文件 */
|
||
fclose(tmp_conf_fp);
|
||
tmp_conf_fp = NULL;
|
||
}
|
||
else {
|
||
/* 定位到结束位置 */
|
||
if (fseek(g_conf_fp, 0, SEEK_END) == -1) {
|
||
ULOG_ERR(g_log, "Seeknig config to end is faiure:%s", strerror(errno));
|
||
return -1;
|
||
}
|
||
}
|
||
|
||
/* 向配置文件中追加配置项 */
|
||
char fmt[MAX_LINE_SZ]="";
|
||
snprintf(fmt, sizeof(fmt), "%s=%s\n", p_key_str, p_value_str);
|
||
ULOG_DEBUG(g_log, "Configure content is %s", fmt);
|
||
int len = strlen(fmt);
|
||
int n = fwrite(fmt, 1, len, g_conf_fp);
|
||
if (len != n) {
|
||
ULOG_ERR(g_log, "Writing conf is failure:%s", strerror(errno));
|
||
return -1;
|
||
}
|
||
fflush(g_conf_fp);
|
||
|
||
return 0;
|
||
}
|
||
|
||
ret_code get_log_file_conf(const char *key_str, char *value_str, int value_len)
|
||
{
|
||
ret_code ret = RET_OK;
|
||
char *pos = NULL;
|
||
char *pos2 = NULL;
|
||
size_t str_len = 0;
|
||
|
||
if (fseek(g_conf_fp, 0, SEEK_SET) == -1) {
|
||
ULOG_ERR(g_log, "Seeknig config to begin is faiure:%s", strerror(errno));
|
||
return ret;
|
||
}
|
||
|
||
ssize_t n;
|
||
char *line = NULL;
|
||
while ((getline(&line, &n, g_conf_fp)) != -1)
|
||
{
|
||
if (strstr(line, key_str) == NULL )
|
||
{
|
||
continue;
|
||
}
|
||
|
||
pos = strstr(line, "=");
|
||
if (pos != NULL)
|
||
{
|
||
if (strlen(pos) <= 1)
|
||
{
|
||
continue;
|
||
}
|
||
|
||
pos ++;
|
||
if ((pos2 = strstr(pos, " ")) != NULL)
|
||
{
|
||
str_len = strlen(pos) - strlen(pos2);
|
||
}
|
||
else
|
||
{
|
||
str_len = strlen(pos);
|
||
}
|
||
|
||
if (str_len >= value_len)
|
||
{
|
||
ULOG_ERR(g_log, "Get log-sched config %s, but invaild value", key_str);
|
||
ret = RET_ERR;
|
||
break;
|
||
}
|
||
|
||
memset(value_str, 0, value_len);
|
||
strncpy(value_str, pos, str_len);
|
||
|
||
/* 滤除掉段尾的回车换行 */
|
||
if ('\n' == value_str[strlen(value_str)-1])
|
||
{
|
||
value_str[strlen(value_str)-1] = '\0';
|
||
}
|
||
|
||
ULOG_DEBUG(g_log, "Get log-sched config %s:%s", key_str, value_str);
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (NULL!= line)
|
||
{
|
||
free(line);
|
||
}
|
||
return ret;
|
||
}
|
||
|
||
|
||
|
||
|
||
|