mbtk/mbtk_logd/common.c

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <netdb.h>
#include <fcntl.h>
#include <time.h>
#include <unistd.h>
#include <time.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
#include "log_config.h"

/* 文件大小和修改时间 */
static int get_file_size_time(const char *filename)
{
	struct stat statbuf;
	/* 判断未打开文件 */
	if(stat(filename,&statbuf)==-1)
	{
		printf("Get stat on %s Error: %s\n", filename, strerror(errno));
		return(-1);
	}
	if(S_ISDIR(statbuf.st_mode)) // 目录
		return(1);
	if(S_ISREG(statbuf.st_mode)) // 文件
		printf("%s size: %ld bytes\tmodified at %s", filename, statbuf.st_size, ctime(&statbuf.st_mtime));
	return(0);
}

int __main(int argc,char **argv)
{
	DIR *dirp;
	struct dirent *direntp;
	int stats;

	if(argc!=2)
	{
		printf("Usage: %s filename\n\a", argv[0]);
		exit(1);
	}

	if(((stats=get_file_size_time(argv[1]))==0)||(stats==-1)) // 文件或出现错误
		exit(1);

	/* 打开目录 */
	if((dirp=opendir(argv[1]))==NULL)
	{
		printf("Open Directory %s Error: %s\n", argv[1], strerror(errno));
		exit(1);
	}

	/* 返回目录中文件大小和修改时间 */
	while((direntp=readdir(dirp))!=NULL)
	{
		/* 给文件或目录名添加路径:argv[1]+"/"+direntp->d_name */
		char dirbuf[512];
		memset(dirbuf,0,sizeof(dirbuf));
		strcpy(dirbuf,argv[1]);
		strcat(dirbuf,"/");
		strcat(dirbuf,direntp->d_name);
		if(get_file_size_time(dirbuf)==-1) break;
	}

	closedir(dirp);
	exit(1);
}
// 从后往前找
char *strstr_tail(const char *dst, const char *src)
{
    if(NULL == dst || NULL == src)
        return NULL;
    const char *pdst = dst;
    const char *psrc = src;
    char *tmp = NULL ;
    while (*pdst)                          //因为要从后向前找，则首先让pdst指向'\0'
    {
       pdst++;
    }

    while (pdst >= dst )                     //当pdst大于dst则表明dst这个字符串还没有找完
    {
        if (tmp=(char *)strstr(pdst, psrc = src))  //使用strstr帮助寻找，找到保存到tmp
        return tmp;

        pdst--;
     }

     return NULL ;
}

char* strcpy_malloc(char* src)
{
    if (NULL == src || 0 == strlen(src)) {
        return NULL;
    }
    int len = strlen(src);
    char* temp = malloc(len + 1);
    if (NULL == temp) {
        return NULL;
    }
    memset(temp, 0, len + 1);
    strcpy(temp, src);

    return temp;
}

int get_time (char *out_time)
{
    time_t rawtime;
    struct tm *info;
    char buffer[16] = {0};

    time( &rawtime );
    info = localtime(&rawtime);
    strftime(buffer, sizeof(buffer), "%Y%m%d%H%M%S \n", info);
    if(out_time)
        memcpy(out_time, &buffer[2], 12);

    return(0);
}

int tcp_connect(char* ip, int port)
{
    int sockfd;
    struct hostent* he;
    struct sockaddr_in server;

    he = gethostbyname(ip);

    if (he == NULL) {
        printf("gethostbyname error\n");
        exit(1);
    }

    sockfd = socket(AF_INET, SOCK_STREAM, 0);

    if (sockfd == -1) {
        printf("socket error\n");
        exit(1);
    }

    bzero(&server, sizeof(server));
    server.sin_family = AF_INET;
    server.sin_port = htons(port);
    server.sin_addr = *((struct in_addr*)he->h_addr);

    if (connect(sockfd, (struct sockaddr*)&server, sizeof(server)) == -1) {
        printf("connect error\n");
        exit(1);
    }

    return sockfd;
}
int movefile(char *src, char *dst)
{
    if (strcmp(src, dst) == 0)
    {
        printf("Destination file name and source file name can't be the same!");
        return -1;
    }
    FILE *fsrc = fopen(src, "rb"); //一定要“rb”否则复制的文件有时会不能使用
    FILE *fdst = fopen(dst, "wb"); //一定要“wb”否则复制的文件有时会不能使用
    int buflen = 1024;
    char buf[1024];
    int nread = 0;
    while ((nread = fread(buf, 1, buflen, fsrc)) > 0) //最好把读文件的代码放这里
    {
        fwrite(buf, 1, nread, fdst);
    }

    fclose(fsrc);
    fsrc = NULL;
    fclose(fdst);
    fdst = NULL;
    remove(src);//删除源文件

    return 0;
}
/**
 * @brief      get_rotate_file
 *
 * @details    转移log文件
 *
 * @param      param
 *
 * @return     return type
 */
int get_rotate_file(int fd, char *base_file, struct file_list_t *_file_list)
{
    char _time[16] = {0};
    int num = 0;
    char tmp_buf[48] = {0};
    char *current_file = _file_list->file[_file_list->current];
    int len = strlen(base_file);
    char *old = malloc(len + 20);
    int rotate_file_count_max;

    if(_file_list->total == -1 || _file_list->total == 0 || _file_list->total > ROTATE_FILE_COUNT_MAX)
        rotate_file_count_max = ROTATE_FILE_COUNT_MAX;
    else
        rotate_file_count_max = _file_list->total;


    if(0 == _file_list->current && NULL == current_file)
        _file_list->current = rotate_file_count_max - 1;
    if(NULL == old)
    {
        fprintf(stderr, "failed to malloc %s\n", strerror(errno));
        exit(-1);
    }
    close(fd);
    sprintf(tmp_buf, "/tmp/log%s", strstr_tail(base_file, "/"));
    get_time(_time);
    sprintf(old, "%s.%s", base_file, _time);
    if (current_file && strstr(current_file, old)) {
        int tmp_len = strlen(current_file);
        //  以时间命名的文件是否已经存在
        if(current_file[tmp_len - 3] == '.'){
            num = atoi(&current_file[tmp_len - 2]);
            num++;
            sprintf(old, "%s.%02d", old, num);
        }else{
            sprintf(old, "%s.%02d", old, num);
        }
    }
    printf("MBTK_LOGD: tmp_buf is %s, old is %s\n", tmp_buf, old);
    if (movefile(tmp_buf, old) != 0){
        printf("%s %d: Renamed error ,%s to %s.\n", __FUNCTION__, __LINE__, tmp_buf, old);
    }
    _file_list->current++;
    // ringbuf
    if(_file_list->current > (rotate_file_count_max - 1))
    {
        _file_list->current = 0;
    }
    // 如果之前已经存在，先释放之前存放的资源
    if(_file_list->file[_file_list->current])
    {
        // printf("%s %d: remove %s\n", __FUNCTION__, __LINE__, _file_list->file[_file_list->current]);
        // 如果 不限制文件数目，则不删除文件
        if(_file_list->total != -1)
            unlink(_file_list->file[_file_list->current]);
        free(_file_list->file[_file_list->current]);
    }
    // 将文件名保存下来
    _file_list->file[_file_list->current] = strdup(old);
    free(old);
    int fd_new = open(tmp_buf, O_CREAT | O_WRONLY | O_APPEND, 0600);
    if (fd_new < 0) {
        fprintf(stderr, "failed to open %s: %s\n", tmp_buf, strerror(errno));
        exit(-1);
    }

    return fd_new;
}

#if 0
void get_rotate_file_test()
{
    struct file_list_t file_list;
    int i;
    char *test_path = "/tmp/log/m_log.txt";

    memset(&file_list, 0, sizeof(struct file_list_t));
    file_list.total = 3;

    int fd = open(test_path, O_CREAT | O_WRONLY | O_APPEND, 0600);
    if (fd < 0) {
        fprintf(stderr, "failed to open %s: %s\n", test_path, strerror(errno));
        exit(-1);
    }
    write(fd, "test1111", 8);
    for (i = 0; i < 6; ++i) {
        // sleep(1);
        fd = get_rotate_file(fd, test_path, &file_list);
        write(fd, "test2222", 8);
    }
    close(fd);
}
#endif