log.h
#ifndef _log_h__
#define _log_h__
#define LEVEL_DEBUG 0
#define LEVEL_TUNNINT 1
#define LEVEL_INFO 2
#define LEVEL_WARN 3
#define LEVEL_ERROR 4
#define TRACE_INFO(...) trace(LEVEL_INFO, __VA_ARGS__);
#define TRACE_DEBUG(...) trace(LEVEL_DEBUG, __VA_ARGS__);
#define TRACE_ERROR(...) trace(LEVEL_ERROR, __VA_ARGS__);
void set_log_path(const char* path);
void trace(int log_level, const char* format, ...);
#endif
log.c
#include<stdio.h>
#include<stdlib.h>
#include<time.h>
#include<string.h>
#include<stdarg.h>
#include<unistd.h>
#include<sys/stat.h>
#include "log.h"
static const char* _log_path = NULL;
static const char* llchar = "DTIWEFM";
void set_log_path(const char* path) {
_log_path = path;
}
static const char* get_log_path() {
return _log_path;
}
static void mkdir_for_log(const char* log_path, int mode) {
if (NULL == log_path)
return;
char* path = strdup(log_path);
char* p = strrchr(path, '/');
if (p) *p = '\0';
if (0 != access(path, 0)) {
p = path;
do {
p = strchr(p + 1, '/');
if (p) *p = '\0';
if (0 != access(path, 0))
mkdir(path, mode);
if (p) *p = '/';
} while(p);
}
free(path);
}
static void write_file(const char* log_path, char* message) {
FILE *fp = fopen(log_path, "a");
if (fp) {
chmod(log_path, 0755);
strcat(message, "\n");
int len = strlen(message);
fwrite(message, len, 1, fp);
fclose(fp);
}
}
static void log_header(int log_level, char* message) {
time_t t = {0};
time(&t);
struct tm* p = localtime(&t);
sprintf(message, "%d-%c %2d-%2d %2d:%2d:%2d ", getpid(), llchar[log_level], p->tm_mon+1, p->tm_mday,
p->tm_hour, p->tm_min, p->tm_sec);
}
void trace(int log_level, const char* format, ...) {
va_list ap;
char message[1024];
if (NULL == format)
return;
log_header(log_level, message);
int len = strlen(message);
va_start(ap, format);
vsnprintf(message + len, sizeof(message) - len - 2, format, ap);
va_end(ap);
mkdir_for_log(get_log_path(), 0700);
write_file(get_log_path(), message);
}