LOGC.c

#include "LOGC.h"

/*
 * iLOG3Lite - log function library written in c
 * author	: calvin
 * email	: calvinwilliams@163.com
 * LastVersion	: v1.0.15
 *
 * Licensed under the LGPL v2.1, see the file LICENSE in base directory.
 */

/* 代码宏 */
#define OFFSET_BUFPTR(_buffer_,_bufptr_,_len_,_buflen_,_remain_len_) \
	if( _len_ > 0 && _len_ < _remain_len_ ) \
	{ \
		_bufptr_ += _len_ ; \
		_buflen_ += _len_ ; \
		_remain_len_ -= _len_ ; \
	} \

#define OFFSET_BUFPTR_IN_LOOP(_buffer_,_bufptr_,_len_,_buflen_,_remain_len_) \
	if( _len_ > 0 && _len_ < _remain_len_ ) \
	{ \
		_bufptr_ += _len_ ; \
		_buflen_ += _len_ ; \
		_remain_len_ -= _len_ ; \
	} \
	else \
	{ \
		break; \
	} \

/* 日志文件名 */
TLS char	_g_logc_pathfilename[ MAXLEN_FILENAME + 1 ] = "" ;
TLS int		_g_logc_level = LOGCLEVEL_NOLOG ;
TLS pid_t	_g_logc_pid = 0 ;
TLS char	_g_logc_cust_label1[ LOGC_MAXLEN_CUST_LABEL + 1 ] = "" ;
TLS char	_g_logc_cust_label2[ LOGC_MAXLEN_CUST_LABEL + 1 ] = "" ;
TLS char	_g_logc_cust_label3[ LOGC_MAXLEN_CUST_LABEL + 1 ] = "" ;
TLS char	_g_logc_cust_label4[ LOGC_MAXLEN_CUST_LABEL + 1 ] = "" ;
TLS char	_g_logc_cust_label5[ LOGC_MAXLEN_CUST_LABEL + 1 ] = "" ;

static const char _logc_level_itoa[][7] = { "DEBUG" , "INFO" , "NOTICE" , "WARN" , "ERROR" , "FATAL" , "NOLOG" } ;

/* 设置日志文件名 */
void SetLogcFile( char *format , ... )
{
	va_list		valist ;
	
	va_start( valist , format );
	SetLogcFileV( format , valist );
	va_end( valist );
	
	_g_logc_pid = getpid() ;
	
	return;
}

void SetLogcFileV( char *format , va_list valist )
{
	VSNPRINTF( _g_logc_pathfilename , sizeof(_g_logc_pathfilename)-1 , format , valist );
	_g_logc_pid = getpid() ;
	return;
}

char *GetLogcFilePtr()
{
	return _g_logc_pathfilename;
}

/* 设置日志等级 */
void SetLogcLevel( int log_level )
{
	_g_logc_level = log_level ;
	return;
}

int GetLogcLevel()
{
	return _g_logc_level;
}

/* 自定义标签 */
void SetCustLabel1( char *cust_label )
{
	memset( _g_logc_cust_label1 , 0x00 , sizeof(_g_logc_cust_label1) );
	strncpy( _g_logc_cust_label1 , cust_label , sizeof(_g_logc_cust_label1)-1 );
	return;
}

void SetCustLabel2( char *cust_label )
{
	memset( _g_logc_cust_label2 , 0x00 , sizeof(_g_logc_cust_label2) );
	strncpy( _g_logc_cust_label2 , cust_label , sizeof(_g_logc_cust_label2)-1 );
	return;
}

void SetCustLabel3( char *cust_label )
{
	memset( _g_logc_cust_label3 , 0x00 , sizeof(_g_logc_cust_label3) );
	strncpy( _g_logc_cust_label3 , cust_label , sizeof(_g_logc_cust_label3)-1 );
	return;
}

void SetCustLabel4( char *cust_label )
{
	memset( _g_logc_cust_label4 , 0x00 , sizeof(_g_logc_cust_label4) );
	strncpy( _g_logc_cust_label4 , cust_label , sizeof(_g_logc_cust_label4)-1 );
	return;
}

void SetCustLabel5( char *cust_label )
{
	memset( _g_logc_cust_label5 , 0x00 , sizeof(_g_logc_cust_label5) );
	strncpy( _g_logc_cust_label5 , cust_label , sizeof(_g_logc_cust_label5)-1 );
	return;
}

/* 输出日志 */
int WriteLogcBaseV( int log_level , char *c_filename , long c_fileline , char *format , va_list valist )
{
	char			c_filename_copy[ MAXLEN_FILENAME + 1 ] ;
	char			*p_c_filename = NULL ;
	
	struct timeval		tv ;
	struct tm		stime ;
	static struct timeval	last_tv = { 0 , 0 } ;
	static char		last_time_buf[ 10+1+8 + 1 ] = "" ;
	
	char			log_buffer[ 4096 + 1 ] ;
	char			*log_bufptr = NULL ;
	size_t			log_buflen ;
	size_t			log_buf_remain_len ;
	size_t			len ;
	
	/* 处理源代码文件名 */
	memset( c_filename_copy , 0x00 , sizeof(c_filename_copy) );
	strncpy( c_filename_copy , c_filename , sizeof(c_filename_copy)-1 );
	p_c_filename = strrchr( c_filename_copy , '\\' ) ;
	if( p_c_filename )
		p_c_filename++;
	else
		p_c_filename = c_filename_copy ;

	/* 填充行日志 */
#if ( defined __linux__ ) || ( defined __unix ) || ( defined _AIX )
	gettimeofday( & tv , NULL );
	if( tv.tv_sec != last_tv.tv_sec )
	{
		localtime_r( &(tv.tv_sec) , & stime );
		strftime( last_time_buf , sizeof(last_time_buf) , "%Y-%m-%d %H:%M:%S" , & stime ) ;
		last_tv.tv_sec = tv.tv_sec ;
	}
#elif ( defined _WIN32 )
	{
	SYSTEMTIME	stNow ;
	GetLocalTime( & stNow );
	tv.tv_usec = stNow.wMilliseconds * 1000 ;
	stime.tm_year = stNow.wYear - 1900 ;
	stime.tm_mon = stNow.wMonth - 1 ;
	stime.tm_mday = stNow.wDay ;
	stime.tm_hour = stNow.wHour ;
	stime.tm_min = stNow.wMinute ;
	stime.tm_sec = stNow.wSecond ;
	}
#endif

	memset( log_buffer , 0x00 , sizeof(log_buffer) );
	log_bufptr = log_buffer ;
	log_buflen = 0 ;
	log_buf_remain_len = sizeof(log_buffer) - 1 - sizeof(NEWLINE) ;
	
	len = SNPRINTF( log_bufptr , log_buf_remain_len , "%s.%06ld | %-6s | " , last_time_buf , (long)(tv.tv_usec) , _logc_level_itoa[log_level] ) ;
	OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	if( _g_logc_cust_label1[0] )
	{
		len = SNPRINTF( log_bufptr , log_buf_remain_len , "%s " ,  _g_logc_cust_label1 ) ;
		OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	}
	if( _g_logc_cust_label2[0] )
	{
		len = SNPRINTF( log_bufptr , log_buf_remain_len , "%s " ,  _g_logc_cust_label2 ) ;
		OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	}
	if( _g_logc_cust_label3[0] )
	{
		len = SNPRINTF( log_bufptr , log_buf_remain_len , "%s " ,  _g_logc_cust_label3 ) ;
		OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	}
	if( _g_logc_cust_label4[0] )
	{
		len = SNPRINTF( log_bufptr , log_buf_remain_len , "%s " ,  _g_logc_cust_label4 ) ;
		OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	}
	if( _g_logc_cust_label5[0] )
	{
		len = SNPRINTF( log_bufptr , log_buf_remain_len , "%s " ,  _g_logc_cust_label5 ) ;
		OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	}
	if( _g_logc_cust_label1[0] || _g_logc_cust_label2[0] || _g_logc_cust_label3[0] || _g_logc_cust_label4[0] || _g_logc_cust_label5[0] )
	{
		len = SNPRINTF( log_bufptr , log_buf_remain_len , "| " ) ;
		OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	}
	len = SNPRINTF( log_bufptr , log_buf_remain_len , "%d:%s:%ld | " ,  _g_logc_pid , p_c_filename , c_fileline ) ;
	OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	len = VSNPRINTF( log_bufptr , log_buf_remain_len , format , valist );
	OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	log_buf_remain_len += sizeof(NEWLINE) ;
	len = SNPRINTF( log_bufptr , log_buf_remain_len , NEWLINE ) ;
	OFFSET_BUFPTR( log_buffer , log_bufptr , len , log_buflen , log_buf_remain_len );
	
	/* 输出行日志 */
	if( STRCMP( _g_logc_pathfilename , == , "#stdout" ) )
	{
		WRITE( 1 , log_buffer , log_buflen );
	}
	else if( _g_logc_pathfilename[0] )
	{
		int		fd ;
		
#if ( defined __linux__ ) || ( defined __unix ) || ( defined _AIX )
		fd = OPEN( _g_logc_pathfilename , O_CREAT | O_WRONLY | O_APPEND , S_IRWXU | S_IRWXG | S_IRWXO ) ;
#elif ( defined _WIN32 )
		fd = OPEN( _g_logc_pathfilename , _O_CREAT | _O_WRONLY | _O_APPEND | _O_BINARY , _S_IREAD | _S_IWRITE ) ;
#endif
		if( fd == -1 )
			return -1;
		
		WRITE( fd , log_buffer , log_buflen );
		
		CLOSE( fd );
	}
	
	return 0;
}

int WriteLogcBase( int log_level , char *c_filename , long c_fileline , char *format , ... )
{
	va_list		valist ;
	
	va_start( valist , format );
	WriteLogcBaseV( log_level , c_filename , c_fileline , format , valist );
	va_end( valist );
	
	return 0;
}

#if ( defined __STDC_VERSION__ ) && ( __STDC_VERSION__ >= 199901 )

#else

int WriteLevelLogc( int log_level , char *c_filename , long c_fileline , char *format , ... )
{
	va_list		valist ;
	
	if( log_level < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteLogcBaseV( log_level , c_filename , c_fileline , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteFatalLogc( char *c_filename , long c_fileline , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_FATAL < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteLogcBaseV( LOGCLEVEL_FATAL , c_filename , c_fileline , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteErrorLogc( char *c_filename , long c_fileline , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_ERROR < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteLogcBaseV( LOGCLEVEL_ERROR , c_filename , c_fileline , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteWarnLogc( char *c_filename , long c_fileline , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_WARN < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteLogcBaseV( LOGCLEVEL_WARN , c_filename , c_fileline , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteNoticeLogc( char *c_filename , long c_fileline , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_NOTICE < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteLogcBaseV( LOGCLEVEL_NOTICE , c_filename , c_fileline , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteInfoLogc( char *c_filename , long c_fileline , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_INFO < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteLogcBaseV( LOGCLEVEL_INFO , c_filename , c_fileline , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteDebugLogc( char *c_filename , long c_fileline , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_DEBUG < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteLogcBaseV( LOGCLEVEL_DEBUG , c_filename , c_fileline , format , valist );
	va_end( valist );
	
	return 0;
}

#endif

int WriteHexLogcBaseV( int log_level , char *c_filename , long c_fileline , char *buf , long buflen , char *format , va_list valist )
{
	char		hexlog_buffer[ 10*4096 + 1 ] ;
	char		*hexlog_bufptr = NULL ;
	size_t		hexlog_buflen ;
	size_t		hexlog_buf_remain_len ;
	size_t		len ;
	
	int		row_offset , col_offset ;
	
	if( buf == NULL && buflen <= 0 )
		return 0;
	if( buflen > sizeof(hexlog_buffer) - 1 )
		return -1;
	
	/* 输出行日志 */
	WriteLogcBaseV( log_level , c_filename , c_fileline , format , valist );
	
	/* 填充十六进制块日志 */
	memset( hexlog_buffer , 0x00 , sizeof(hexlog_buffer) );
	hexlog_bufptr = hexlog_buffer ;
	hexlog_buflen = 0 ;
	hexlog_buf_remain_len = sizeof(hexlog_buffer) - 1 ;
	
	len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , "             0  1  2  3  4  5  6  7  8  9  A  B  C  D  E  F    0123456789ABCDEF" ) ;
	OFFSET_BUFPTR( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
	len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , NEWLINE ) ;
	OFFSET_BUFPTR( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
	
	row_offset = 0 ;
	col_offset = 0 ;
	while( hexlog_buf_remain_len > 0 )
	{
		len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , "0x%08X   " , row_offset * 16 ) ;
		OFFSET_BUFPTR_IN_LOOP( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
		for( col_offset = 0 ; col_offset < 16 ; col_offset++ )
		{
			if( row_offset * 16 + col_offset < buflen )
			{
				len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , "%02X " , *((unsigned char *)buf+row_offset*16+col_offset)) ;
				OFFSET_BUFPTR_IN_LOOP( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
			}
			else
			{
				len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , "   " ) ;
				OFFSET_BUFPTR_IN_LOOP( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
			}
		}
		len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , "  " ) ;
		OFFSET_BUFPTR_IN_LOOP( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
		for( col_offset = 0 ; col_offset < 16 ; col_offset++ )
		{
			if( row_offset * 16 + col_offset < buflen )
			{
				if( isprint( (int)*(buf+row_offset*16+col_offset) ) )
				{
					len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , "%c" , *((unsigned char *)buf+row_offset*16+col_offset) ) ;
					OFFSET_BUFPTR_IN_LOOP( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
				}
				else
				{
					len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , "." ) ;
					OFFSET_BUFPTR_IN_LOOP( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
				}
			}
			else
			{
				len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , " " ) ;
				OFFSET_BUFPTR_IN_LOOP( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
			}
		}
		len = SNPRINTF( hexlog_bufptr , hexlog_buf_remain_len , NEWLINE ) ;
		OFFSET_BUFPTR_IN_LOOP( hexlog_buffer , hexlog_bufptr , len , hexlog_buflen , hexlog_buf_remain_len );
		if( row_offset * 16 + col_offset >= buflen )
			break;
		row_offset++;
	}
	
	if( STRNCMP( hexlog_bufptr-(sizeof(NEWLINE)-1) , != , NEWLINE , sizeof(NEWLINE)-1 ) )
	{
		memcpy( hexlog_bufptr-(sizeof(NEWLINE)-1) , NEWLINE , sizeof(NEWLINE)-1 );
	}
	
	/* 输出十六进制块日志 */
	if( STRCMP( _g_logc_pathfilename , == , "#stdout" ) )
	{
		WRITE( 1 , hexlog_buffer , hexlog_buflen );
	}
	else if( _g_logc_pathfilename[0] )
	{
		int		fd ;
		
#if ( defined __linux__ ) || ( defined __unix ) || ( defined _AIX )
		fd = OPEN( _g_logc_pathfilename , O_CREAT | O_WRONLY | O_APPEND , S_IRWXU | S_IRWXG | S_IRWXO ) ;
#elif ( defined _WIN32 )
		fd = OPEN( _g_logc_pathfilename , _O_CREAT | _O_WRONLY | _O_APPEND | _O_BINARY , _S_IREAD | _S_IWRITE ) ;
#endif
		if( fd == -1 )
			return -1;
		
		WRITE( fd , hexlog_buffer , hexlog_buflen );
		
		CLOSE( fd );
	}
	
	return 0;
}

int WriteHexLogcBase( int log_level , char *c_filename , long c_fileline , char *buf , long buflen , char *format , ... )
{
	va_list		valist ;
	
	va_start( valist , format );
	WriteHexLogcBaseV( log_level , c_filename , c_fileline , buf , buflen , format , valist );
	va_end( valist );
	
	return 0;
}

#if ( defined __STDC_VERSION__ ) && ( __STDC_VERSION__ >= 199901 )

#else

int WriteLevelHexLogc( int log_level , char *c_filename , long c_fileline , char *buf , long buflen , char *format , ... )
{
	va_list		valist ;
	
	if( log_level < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteHexLogcBaseV( log_level , c_filename , c_fileline , buf , buflen , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteFatalHexLogc( char *c_filename , long c_fileline , char *buf , long buflen , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_FATAL < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteHexLogcBaseV( LOGCLEVEL_FATAL , c_filename , c_fileline , buf , buflen , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteErrorHexLogc( char *c_filename , long c_fileline , char *buf , long buflen , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_ERROR < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteHexLogcBaseV( LOGCLEVEL_ERROR , c_filename , c_fileline , buf , buflen , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteWarnHexLogc( char *c_filename , long c_fileline , char *buf , long buflen , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_WARN < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteHexLogcBaseV( LOGCLEVEL_WARN , c_filename , c_fileline , buf , buflen , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteNoticeHexLogc( char *c_filename , long c_fileline , char *buf , long buflen , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_NOTICE < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteHexLogcBaseV( LOGCLEVEL_NOTICE , c_filename , c_fileline , buf , buflen , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteInfoHexLogc( char *c_filename , long c_fileline , char *buf , long buflen , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_INFO < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteHexLogcBaseV( LOGCLEVEL_INFO , c_filename , c_fileline , buf , buflen , format , valist );
	va_end( valist );
	
	return 0;
}

int WriteDebugHexLogc( char *c_filename , long c_fileline , char *buf , long buflen , char *format , ... )
{
	va_list		valist ;
	
	if( LOGCLEVEL_DEBUG < _g_logc_level )
		return 0;
	
	va_start( valist , format );
	WriteHexLogcBaseV( LOGCLEVEL_DEBUG , c_filename , c_fileline , buf , buflen , format , valist );
	va_end( valist );
	
	return 0;
}

#endif