access_log.c

spook是一个linux下开源的流媒体服务器

/*
 * Copyright (C) 2004 Nathan Lutchansky <[e-mail]>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#include <sys/types.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <time.h>
#include <errno.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <event.h>
#include <log.h>
#include <frame.h>
#include <stream.h>
#include <rtp.h>
#include <conf_parse.h>

static int log_fd = -1;

void write_access_log( char *path, struct sockaddr *addr, int code, char *req,
		int length, char *referer, char *user_agent )
{
	char s[128];
	time_t now;
	int req_len;

	if( log_fd < 0 ) return;

	for( req_len = 0; req[req_len] && req[req_len] != '\r' &&
			req[req_len] != '\n'; ++req_len );
	if( ! referer || ! referer[0] ) referer = "-";
	if( ! user_agent || ! user_agent[0] ) user_agent = "-";

	time( &now );
	strcpy( s, inet_ntoa( ((struct sockaddr_in *)addr)->sin_addr ) );
	write( log_fd, s, strlen( s ) );
	write( log_fd, s, 
		strftime( s, sizeof( s ), " - - [%d/%b/%Y:%T %z] \"",
				localtime( &now ) ) );
	write( log_fd, req, req_len );
	write( log_fd, s, sprintf( s, "\" %d %d \"", code, length ) );
	write( log_fd, referer, strlen( referer ) );
	write( log_fd, "\" \"", 3 );
	write( log_fd, user_agent, strlen( user_agent ) );
	write( log_fd, "\"\n", 2 );
}

/************************ CONFIGURATION DIRECTIVES ************************/

static int set_file( int num_tokens, struct token *tokens, void *d )
{
	if( log_fd >= 0 )
	{
		spook_log( SL_ERR, "log file: only one log file is allowed" );
		return -1;
	}
	if( ( log_fd = open( tokens[1].v.str,
				O_CREAT | O_APPEND | O_WRONLY, 0666 ) ) < 0 )
	{
		spook_log( SL_ERR,
			"can't open log file: %s", strerror( errno ) );
		return -1;
	}
	spook_log( SL_VERBOSE, "writing access logs to %s", tokens[1].v.str );
	return 0;
}

static struct statement config_statements[] = {
	/* directive name, process function, min args, max args, arg types */
	{ "file", set_file, 1, 1, { TOKEN_STR } },

	/* empty terminator -- do not remove */
	{ NULL, NULL, 0, 0, {} }
};

int access_log_init(void)
{
	register_config_context( "log", "file", NULL, NULL,
					config_statements );
	return 0;
}