added zabbix_trapperd

git-svn-id: svn://svn.zabbix.com/trunk@192 97f52cf1-0a1b-0410-bd0e-c28be96e8082

3 files changed, 368 insertions, 41 deletions

diff --git a/src/zabbix_trapper/Makefile.in b/src/zabbix_trapper/Makefile.in
index 4aab0122..e90cf04b 100644
--- a/src/zabbix_trapper/Makefile.in
+++ b/src/zabbix_trapper/Makefile.in
@@ -6,8 +6,10 @@ LIBS    = @LIBS@ @MYSQL_LFLAGS@ @PGSQL_LFLAGS@
 all:
 
 	@CC@ -o ../../bin/zabbix_trapper @CFLAGS@ $(INCLUDE) -I../../include  zabbix_trapper.c ../../include/db.c ../../include/functions.c ../../include/expression.c -Wall $(LIBS)
+	@CC@ -o ../../bin/zabbix_trapperd @CFLAGS@ $(INCLUDE) -I../../include  zabbix_trapperd.c ../../include/db.c ../../include/functions.c ../../include/expression.c -Wall $(LIBS)
 
 clean:
 	rm -fv *.o 
 	rm -fv ../../bin/zabbix_trapper
+	rm -fv ../../bin/zabbix_trapperd
 
diff --git a/src/zabbix_trapper/zabbix_trapper.c b/src/zabbix_trapper/zabbix_trapper.c
index f31ee3d7..e39262cb 100644
--- a/src/zabbix_trapper/zabbix_trapper.c
+++ b/src/zabbix_trapper/zabbix_trapper.c
@@ -43,41 +43,6 @@ void	signal_handler( int sig )
 	exit( FAIL );
 }
 
-int	check_security(void)
-{
-	char	*sname;
-	char	*config;
-	struct	sockaddr_in name;
-	int	i;
-	int	file;
-
-	config=(char *)malloc(16);
-
-	file=open("/etc/zabbix/zabbix_agent.conf",O_RDONLY);
-	if(file == -1)
-	{
-//		printf("Open failed");
-		return FAIL;
-	}
-	i=read(file, config, 16);
-	config[i-1]=0;
-	close(file);
-
-	i=sizeof(struct sockaddr_in);
-
-	if(getpeername(0,  (struct sockaddr *)&name, &i) == 0)
-	{
-//		printf("%d\n",name.sin_port);
-		sname=inet_ntoa(name.sin_addr);
-//		printf("From:=%s=\n",sname);
-		if(strcmp(sname,config)!=0)
-		{
-			return	FAIL;
-		}
-	}
-	return	SUCCEED;
-}
-
 int	process_data(char *server,char *key, double value)
 {
 	char	sql[1024];
@@ -147,12 +112,6 @@ int	main()
 
 	int	ret=SUCCEED;
 
-
-//	if(check_security() == FAIL)
-//	{
-//		exit(FAIL);
-//	}
-
 	signal( SIGINT,  signal_handler );
 	signal( SIGQUIT, signal_handler );
 	signal( SIGTERM, signal_handler );
diff --git a/src/zabbix_trapper/zabbix_trapperd.c b/src/zabbix_trapper/zabbix_trapperd.c
new file mode 100644
index 00000000..16e7ae86
--- /dev/null
+++ b/src/zabbix_trapper/zabbix_trapperd.c
@@ -0,0 +1,366 @@
+#include "config.h"
+
+#include <string.h>
+
+#include <netdb.h>
+
+#include <syslog.h>
+
+#include <stdlib.h>
+#include <stdio.h>
+
+#include <time.h>
+
+#include <unistd.h>
+#include <signal.h>
+
+#include <errno.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+
+/* For config file operations */
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+
+#include "common.h"
+#include "db.h"
+#include "functions.h"
+
+#define	LISTENQ 1024
+
+static pid_t *pids;
+
+void	signal_handler( int sig )
+{
+	if( SIGALRM == sig )
+	{
+		signal( SIGALRM, signal_handler );
+ 
+//		fprintf(stderr,"Timeout while executing operation.");
+	}
+ 
+	if( SIGQUIT == sig || SIGINT == sig || SIGTERM == sig )
+	{
+//		fprintf(stderr,"\nGot QUIT or INT or TERM signal. Exiting..." );
+	}
+	exit( FAIL );
+}
+
+int	process_data(char *server,char *key, double value)
+{
+	char	sql[1024];
+	int	itemid;
+	double	lastvalue;
+	int	now;
+
+	DB_RESULT       *result;
+
+	sprintf(sql,"select i.itemid,i.lastvalue from items i,hosts h where h.status=0 and h.hostid=i.hostid and h.host='%s' and i.key_='%s' and i.status=2;",server,key);
+	result = DBselect(sql);
+	 
+	if(result==NULL)
+	{
+		DBfree_result(result);
+		return  FAIL;
+	}
+	if(DBnum_rows(result)==0)
+	{
+		DBfree_result(result);
+		return  FAIL;
+	}
+
+	if( DBget_field(result,0,0) == NULL )
+	{
+		DBfree_result(result);
+		return  FAIL;
+	}
+
+	itemid=atoi(DBget_field(result,0,0));
+
+	now = time(NULL);
+	sprintf(sql,"insert into history (itemid,clock,value) values (%d,%d,%g);",itemid,now,value);
+	DBexecute(sql);
+
+	if(NULL == DBget_field(result,0,1))
+	{
+		now = time(NULL);
+		sprintf(sql,"update items set lastvalue=%g,lastclock=%d where itemid=%d;",value,now,itemid);
+	}
+	else
+	{
+		lastvalue=atof(DBget_field(result,0,1));
+		now = time(NULL);
+		sprintf(sql,"update items set prevvalue=%g,lastvalue=%g,lastclock=%d where itemid=%d;",lastvalue,value,now,itemid);
+	}
+
+	DBexecute(sql);
+
+	if( update_functions( itemid ) == FAIL)
+	{
+		return FAIL;
+	}
+
+	update_triggers( itemid );
+ 
+	DBfree_result(result);
+
+	return SUCCEED;
+}
+
+int	process(char *s)
+{
+	char	*p;
+	char	*server,*key,*value_string;
+	double	value;
+
+	int	ret=SUCCEED;
+
+	for( p=s+strlen(s)-1; p>s && ( *p=='\r' || *p =='\n' || *p == ' ' ); --p );
+	p[1]=0;
+
+	server=(char *)strtok(s,":");
+	if(NULL == server)
+	{
+		return FAIL;
+	}
+
+	key=(char *)strtok(NULL,":");
+	if(NULL == key)
+	{
+		return FAIL;
+	}
+
+	value_string=(char *)strtok(NULL,":");
+	if(NULL == value_string)
+	{
+		return FAIL;
+	}
+	value=atof(value_string);
+
+	ret=process_data(server,key,value);
+
+	return ret;
+}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+void    daemon_init(void)
+{
+	int     i;
+	pid_t   pid;
+
+	if( (pid = fork()) != 0 )
+	{
+		exit( 0 );
+	}
+
+	setsid();
+	
+	signal( SIGHUP, SIG_IGN );
+
+	if( (pid = fork()) !=0 )
+	{
+		exit( 0 );
+	}
+
+	chdir("/");
+	umask(0);
+
+	for(i=0;i<MAXFD;i++)
+	{
+		close(i);
+	}
+}
+
+void	process_child(int sockfd)
+{
+	ssize_t	nread;
+	char	line[1024];
+	char	result[1024];
+//	void	*sigfunc;
+
+	for(;;)
+	{
+//		sigfunc = signal( SIGALRM, signal_handler );
+//		alarm(AGENT_TIMEOUT);
+
+		if( (nread = read(sockfd, line, 1024)) == 0)
+			return;
+
+//		alarm(0);
+//		signal(SIGALRM, sigfunc);
+
+		line[nread-1]=0;
+
+//		printf("Got line:{%s}\n",line);
+
+		syslog( LOG_DEBUG, "Got line:%s", line);
+		if( SUCCEED == process(line) )
+		{
+			sprintf(result,"OK\n");
+		}
+		else
+		{
+			result[0]=0;
+		}
+		syslog( LOG_DEBUG, "Sending back:%s", result);
+		write(sockfd,result,strlen(result));
+	}
+}
+
+int	tcp_listen(const char *host, const char *serv, socklen_t *addrlenp)
+{
+	int		listenfd, n;
+	const int	on=1;
+	struct addrinfo	hints, *res, *ressave;
+
+	bzero(&hints,sizeof(struct addrinfo));
+	hints.ai_flags = AI_PASSIVE;
+	hints.ai_family = AF_UNSPEC;
+	hints.ai_socktype = SOCK_STREAM;
+
+	if( (n = getaddrinfo(host,serv, &hints, &res)) != 0)
+	{
+		perror("getaddrinfo()");
+		exit(1);
+	}
+
+	ressave = res;
+
+	do {
+		listenfd = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
+		if( listenfd <0)
+				continue;
+		if(setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) !=0 )
+		{
+			perror("setsockopt()");
+			exit(1);
+		}
+		if(bind(listenfd,res->ai_addr,res->ai_addrlen) == 0)
+			break;
+		close(listenfd);
+	} while ((res = res ->ai_next) != NULL);
+
+	if (res == NULL)
+	{
+		perror("tcp_listen");
+		exit(1);
+	}
+
+	if(listen(listenfd, LISTENQ) !=0 )
+	{
+		perror("listen()");
+		exit(1);
+	}
+
+	if(addrlenp)
+		*addrlenp = res->ai_addrlen;
+
+	freeaddrinfo(ressave);
+
+	return	(listenfd);
+}
+
+void	child_main(int i,int listenfd, int addrlen)
+{
+	int	connfd;
+	socklen_t	clilen;
+	struct sockaddr *cliaddr;
+
+	cliaddr=malloc(addrlen);
+
+//	printf("child %ld started\n",(long)getpid());
+	DBconnect();
+
+	for(;;)
+	{
+		clilen = addrlen;
+		connfd=accept(listenfd,cliaddr, &clilen);
+
+		process_child(connfd);
+
+		close(connfd);
+	}
+}
+
+pid_t	child_make(int i,int listenfd, int addrlen)
+{
+	pid_t	pid;
+
+	if((pid = fork()) >0)
+	{
+			return (pid);
+	}
+
+	/* never returns */
+	child_main(i, listenfd, addrlen);
+
+	/* avoid compilator warning */
+	return 0;
+}
+
+int	main()
+{
+	int		listenfd;
+	socklen_t	addrlen;
+	int		i, ret;
+
+	char		host[128];
+	char		*port="10001";
+	
+	daemon_init();
+
+	signal( SIGINT,  signal_handler );
+	signal( SIGQUIT, signal_handler );
+	signal( SIGTERM, signal_handler );
+//	signal( SIGALRM, signal_handler );
+
+        openlog("zabbix_trapperd",LOG_PID,LOG_USER);
+//	ret=setlogmask(LOG_UPTO(LOG_DEBUG));
+	ret=setlogmask(LOG_UPTO(LOG_WARNING));
+
+	syslog( LOG_WARNING, "zabbix_trapperd started");
+
+	if(gethostname(host,127) != 0)
+	{
+		syslog( LOG_CRIT, "gethostname() failed");
+		exit(FAIL);
+	}
+
+	listenfd = tcp_listen(host,port,&addrlen);
+
+	pids = calloc(10, sizeof(pid_t));
+
+	for(i = 0; i< 10; i++)
+	{
+		pids[i] = child_make(i, listenfd, addrlen);
+		syslog( LOG_WARNING, "zabbix_trapperd #%d started", pids[i]);
+	}
+
+	for(;;)
+	{
+			pause();
+	}
+
+	return SUCCEED;
+}