[people/ms/rstp.git] / main.c

/*****************************************************************************
  Copyright (c) 2006 EMC Corporation.

  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.
  
  The full GNU General Public License is included in this distribution in the
  file called LICENSE.

  Authors: Srinivas Aji <Aji_Srinivas@emc.com>

******************************************************************************/

#include "epoll_loop.h"
#include "bridge_ctl.h"
#include "ctl_socket_server.h"
#include "netif_utils.h"
#include "packet.h"
#include "log.h"

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <getopt.h>
#include <syslog.h>
#include <errno.h>

static int become_daemon = 1;
static int is_daemon = 0;
int log_level = LOG_LEVEL_DEFAULT;

int main(int argc, char *argv[])
{
	int c,ret;
	while ((c = getopt(argc, argv, "dv:")) != -1) {
		switch (c) {
		case 'd':
			become_daemon = 0;
			break;
		case 'v':
			{
				char *end;
				long l;
				l = strtoul(optarg, &end, 0);
				if (*optarg == 0 || *end != 0
				    || l > LOG_LEVEL_MAX) {
					ERROR("Invalid loglevel %s", optarg);
					exit(1);
				}
				log_level = l;
			}
			break;
		default:
			return -1;
		}
	}

	TST(init_epoll() == 0, -1);
	TST(ctl_socket_init() == 0, -1);
	TST(packet_sock_init() == 0, -1);
	TST(netsock_init() == 0, -1);
	TST(init_bridge_ops() == 0, -1);
	if (become_daemon) {
		FILE *f = fopen("/var/run/rstpd.pid", "w");
		if (!f) {
			ERROR("can't open /var/run/rstp.pid");
			return -1;
		}
		openlog("rstpd", 0, LOG_DAEMON);
		ret = daemon(0, 0);
		if (ret) {
			ERROR("daemon() failed");
			return -1;		    
		}
		is_daemon = 1;
		fprintf(f, "%d", getpid());
		fclose(f);
	}
	return epoll_main_loop();
}

/*********************** Logging *********************/

#include <stdarg.h>
#include <time.h>

void vDprintf(int level, const char *fmt, va_list ap)
{
	if (level > log_level)
		return;

	if (!is_daemon) {
		char logbuf[256];
		logbuf[255] = 0;
		time_t clock;
		struct tm *local_tm;
		time(&clock);
		local_tm = localtime(&clock);
		int l =
		    strftime(logbuf, sizeof(logbuf) - 1, "%F %T ", local_tm);
		vsnprintf(logbuf + l, sizeof(logbuf) - l - 1, fmt, ap);
		printf("%s\n", logbuf);
	} else {
		vsyslog((level <= LOG_LEVEL_INFO) ? LOG_INFO : LOG_DEBUG, fmt,
			ap);
	}
}

void Dprintf(int level, const char *fmt, ...)
{
	va_list ap;
	va_start(ap, fmt);
	vDprintf(level, fmt, ap);
	va_end(ap);
}
Commit	Line	Data
ad02a0eb SH	1	/*****************************************************************************
	2	Copyright (c) 2006 EMC Corporation.
	3
	4	This program is free software; you can redistribute it and/or modify it
	5	under the terms of the GNU General Public License as published by the Free
	6	Software Foundation; either version 2 of the License, or (at your option)
	7	any later version.
	8
	9	This program is distributed in the hope that it will be useful, but WITHOUT
	10	ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	11	FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
	12	more details.
	13
	14	You should have received a copy of the GNU General Public License along with
	15	this program; if not, write to the Free Software Foundation, Inc., 59
	16	Temple Place - Suite 330, Boston, MA 02111-1307, USA.
	17
	18	The full GNU General Public License is included in this distribution in the
	19	file called LICENSE.
	20
	21	Authors: Srinivas Aji <Aji_Srinivas@emc.com>
	22
	23	******************************************************************************/
	24
	25	#include "epoll_loop.h"
	26	#include "bridge_ctl.h"
	27	#include "ctl_socket_server.h"
	28	#include "netif_utils.h"
fcea43ad	29	#include "packet.h"
ad02a0eb SH	30	#include "log.h"
	31
	32	#include <stdio.h>
	33	#include <stdlib.h>
	34	#include <unistd.h>
	35	#include <getopt.h>
	36	#include <syslog.h>
358260ff	37	#include <errno.h>
ad02a0eb SH	38
	39	static int become_daemon = 1;
	40	static int is_daemon = 0;
	41	int log_level = LOG_LEVEL_DEFAULT;
	42
	43	int main(int argc, char *argv[])
	44	{
c768cf44	45	int c,ret;
11904a35 SH	46	while ((c = getopt(argc, argv, "dv:")) != -1) {
	47	switch (c) {
	48	case 'd':
	49	become_daemon = 0;
	50	break;
	51	case 'v':
	52	{
	53	char *end;
	54	long l;
	55	l = strtoul(optarg, &end, 0);
	56	if (optarg == 0 \|\| end != 0
	57	\|\| l > LOG_LEVEL_MAX) {
	58	ERROR("Invalid loglevel %s", optarg);
	59	exit(1);
	60	}
	61	log_level = l;
	62	}
	63	break;
	64	default:
	65	return -1;
	66	}
	67	}
	68
	69	TST(init_epoll() == 0, -1);
	70	TST(ctl_socket_init() == 0, -1);
fcea43ad	71	TST(packet_sock_init() == 0, -1);
11904a35 SH	72	TST(netsock_init() == 0, -1);
	73	TST(init_bridge_ops() == 0, -1);
	74	if (become_daemon) {
358260ff SH	75	FILE *f = fopen("/var/run/rstpd.pid", "w");
	76	if (!f) {
	77	ERROR("can't open /var/run/rstp.pid");
	78	return -1;
	79	}
11904a35	80	openlog("rstpd", 0, LOG_DAEMON);
c768cf44 DF	81	ret = daemon(0, 0);
	82	if (ret) {
	83	ERROR("daemon() failed");
	84	return -1;
	85	}
11904a35	86	is_daemon = 1;
358260ff SH	87	fprintf(f, "%d", getpid());
358260ff SH	88	fclose(f);
11904a35 SH	89	}
11904a35 SH	90	return epoll_main_loop();
ad02a0eb SH	91	}
	92
	93	/********************* Logging *******************/
	94
	95	#include <stdarg.h>
	96	#include <time.h>
	97
11904a35	98	void vDprintf(int level, const char *fmt, va_list ap)
ad02a0eb	99	{
11904a35 SH	100	if (level > log_level)
	101	return;
	102
	103	if (!is_daemon) {
	104	char logbuf[256];
	105	logbuf[255] = 0;
	106	time_t clock;
	107	struct tm *local_tm;
	108	time(&clock);
	109	local_tm = localtime(&clock);
	110	int l =
	111	strftime(logbuf, sizeof(logbuf) - 1, "%F %T ", local_tm);
	112	vsnprintf(logbuf + l, sizeof(logbuf) - l - 1, fmt, ap);
	113	printf("%s\n", logbuf);
	114	} else {
	115	vsyslog((level <= LOG_LEVEL_INFO) ? LOG_INFO : LOG_DEBUG, fmt,
	116	ap);
	117	}
ad02a0eb SH	118	}
ad02a0eb SH	119
11904a35	120	void Dprintf(int level, const char *fmt, ...)
ad02a0eb	121	{
11904a35 SH	122	va_list ap;
	123	va_start(ap, fmt);
	124	vDprintf(level, fmt, ap);
	125	va_end(ap);
ad02a0eb	126	}