blob: 32e6844c5ddadc43f3545dbfe3c461ed31aeb87e [file] [log] [blame]
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <netinet/in.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <time.h>
#include <unistd.h>
#include <errno.h>
#include <stdlib.h>
#include <pthread.h>
#include <string.h>
#include "log_config.h"
#define _MOPEN_RILD_CONNET_NUM 5
#define _MOPEN_RILD_SOCKET "/tmp/logd_socket"
static int client_sockfd[_MOPEN_RILD_CONNET_NUM];
static char* log_file;
static int log_size = 1 * 1024 * 1024;
static log_config_entry* config = NULL;
static char tmp_log[48] = {0};
int tcp_connect(char* ip, int port);
char *strstr_tail(const char *dst, const char *src);
#if 0
static int fileter_log(int pri, char* tag, struct filter_list_t* filter)
{
struct filter_list_t* _filter = filter;
while (_filter) {
// _filter->priority
// 获取 筛选的等级 p
int p = 0;
if (_filter->tag) {
int len = strlen(_filter->tag);
// tag and priority
if (0 == memcmp(_filter->tag, tag, len) && ((pri > p) || (pri == p))) {
return 0;
}
} else { // have no tag
if (pri < p) {
return -1;
} else {
return 0;
}
}
_filter = _filter->next;
}
return -1;
}
#endif
int socket_log_print(
int fd,
struct file_list_t* _file_list,
char* entry)
{
// char priChar;
// char timeBuf[32];
// time_t timetemp; // 定义一个时间结构体变量
char defaultBuffer[512];
size_t totalLen;
int fd_new = fd;
struct stat s;
int ret = 0;
if (log_size && (!stat(tmp_log, &s)) && (s.st_size > log_size)) {
fd_new = get_rotate_file(fd_new, log_file, _file_list);
if (fd_new < 0) {
fprintf(stderr, "failed to open %s: %s\n", log_file, strerror(errno));
exit(-1);
}
}
// if(fileter_log(entry->priority, entry->tag, config->filter_list))
// {
// printf("%s %d: fileter pri:%d tag:%s!\n", __FUNCTION__, __LINE__, entry->priority, entry->tag);
// return -1;
// }
// time(&timetemp); // 获得时间参数
// struct tm* ptm = localtime(&timetemp);
// strftime(timeBuf, sizeof(timeBuf), "%Y-%m-%d %H:%M:%S", ptm);
totalLen = snprintf(defaultBuffer, sizeof(defaultBuffer), "%s", entry);
ret = write(fd_new, defaultBuffer, totalLen);
return ret;
}
static int open_local_socket(char *path)
{
int listen_fd;
int ret;
struct sockaddr_un srv_addr;
listen_fd = socket(AF_UNIX, SOCK_STREAM, 0);
if (listen_fd < 0) {
printf("connect creat communication socket");
return -1;
}
unlink(path);
srv_addr.sun_family = AF_UNIX;
strcpy(srv_addr.sun_path, path);
ret = bind(listen_fd, (struct sockaddr*)&srv_addr, sizeof(srv_addr));
if (ret < 0) {
printf("cannot bind server socket");
close(listen_fd);
unlink(path);
return -1;
}
return listen_fd;
}
void* socket_log_thread(void* argv)
{
int listen_fd;
int com_fd, fd_max, fd_num;
int ret, log_fd;
int i, n, str_len;
fd_set reads, cpy_reads;
struct timeval timeout;
char buf[512] = {0};
static struct file_list_t file_list;
config = (log_config_entry*)argv;
pthread_detach(pthread_self());
if (NULL == argv || NULL == config->name) {
return NULL;
}
memset(&file_list, 0, sizeof(struct file_list_t));
file_list.total = config->rotate_file_count;
log_file = config->out_path;
if (config->ip && config->port) {
int port = atoi(config->port);
printf("%s %d : %s:%s\n", __FUNCTION__, __LINE__, config->ip, config->port);
log_fd = tcp_connect(config->ip, port);
} else if (log_file) {
sprintf(tmp_log, "/tmp/log%s", strstr_tail(log_file, "/"));
// 先将文件保存到 /tmp/log/ 目录下,后面到达 rotate_file_size 后,转移到out_path
log_fd = open(tmp_log, O_CREAT | O_WRONLY | O_APPEND, 0600);
if (log_fd < 0) {
fprintf(stderr, "failed to open %s: %s\n", tmp_log, strerror(errno));
exit(-1);
}
} else {
log_fd = STDOUT_FILENO;
}
if (config->rotate_file_size) {
log_size = config->rotate_file_size;
}
listen_fd = open_local_socket(_MOPEN_RILD_SOCKET);
if (listen_fd < 0 || listen_fd == 0) {
return NULL;
}
ret = listen(listen_fd, 1);
if (ret < 0) {
printf("cannot listen sockfd");
close(listen_fd);
unlink(_MOPEN_RILD_SOCKET);
return NULL;
}
FD_ZERO(&reads);
FD_SET(listen_fd, &reads);
fd_max = listen_fd;
printf("socket log start...\n");
memset(client_sockfd, 0, sizeof(client_sockfd));
for (;;) {
cpy_reads = reads;
timeout.tv_sec = 5;
timeout.tv_usec = 5000;
if ((fd_num = select(fd_max + 1, &cpy_reads, 0, 0, &timeout)) == -1) {
perror("select error");
break;
}
if (fd_num == 0) {
continue;
}
for (n = 0; n < fd_max + 1; n++) {
if (FD_ISSET(n, &cpy_reads)) {
if (n == listen_fd) {
com_fd = accept(listen_fd, NULL, NULL);
FD_SET(com_fd, &reads);
if (fd_max < com_fd) {
fd_max = com_fd;
}
printf("accept accept fd [%d]", com_fd);
for (i = 0; i < _MOPEN_RILD_CONNET_NUM; i++) {
if (client_sockfd[i] <= 0) {
client_sockfd[i] = com_fd;
break;
}
}
} else {
str_len = read(n, buf, sizeof(buf));
if (str_len == 0) {
for (i = 0; i < _MOPEN_RILD_CONNET_NUM; i++) {
if (client_sockfd[i] == n) {
client_sockfd[i] = 0;
break;
}
}
FD_CLR(n, &reads);
close(n);
printf("closed client: %d \n", n);
} else {
socket_log_print(log_fd, &file_list, buf);
memset(buf, 0, sizeof(buf));
}
}
}
}
}
close(listen_fd);
close(log_fd);
unlink(_MOPEN_RILD_SOCKET);
printf("%s exit \n", __FUNCTION__);
pthread_exit(NULL);
return NULL;
}