/* Signal Handler. USR1 increases verbosity, USR2 decreases verbosity.
HUP re-reads target list */
void *sig_handler(void *arg)
{
sigset_t *signal_set = (sigset_t *) arg;
int sig_number;
while (1) {
sigwait(signal_set, &sig_number);
switch (sig_number) {
case SIGHUP:
if(lock) {
waiting = TRUE;
}
else {
entries = hash_target_file(target_file);
waiting = FALSE;
}
break;
case SIGUSR1:
set->verbose++;
break;
case SIGUSR2:
set->verbose--;
break;
case SIGTERM:
case SIGINT:
case SIGQUIT:
debug(LOW, "Quiting: received signal %d.\n", sig_number);
#if HAVE_MYSQL
#ifndef FEATURES
mysql_dbdisconnect(&mysql);
#endif
#endif
unlink(pid_file);
exit(1);
break;
}
}
}
/* Main rated */
int main(int argc, char *argv[]) {
crew_t crew;
pthread_t sig_thread;
sigset_t signal_set;
struct timeval begin_time, end_time;
unsigned long sleep_time;
unsigned long poll_time;
unsigned long long polls;
unsigned long long last_poll;
double rate;
char *conf_file = NULL;
char *table;
char errstr[BUFSIZE];
int ch, i, freed;
struct timespec ts;
dfp = stderr;
/* Check argument count */
if (argc < 3)
usage(argv[0]);
/* Set default environment */
config_defaults(set);
/* Parse the command-line. */
while ((ch = getopt(argc, argv, "c:p:dhmDt:vz")) != EOF)
switch ((char) ch) {
case 'c':
conf_file = optarg;
break;
case 'd':
set->dbon = FALSE;
break;
case 'D':
set->daemon = FALSE;
break;
case 'h':
usage(argv[0]);
break;
case 'm':
set->multiple++;
break;
case 'p':
pid_file = optarg;
break;
case 't':
target_file = optarg;
break;
case 'v':
set->verbose++;
break;
case 'z':
set->withzeros = TRUE;
break;
}
debug(LOW, "rated version %s starting.\n", VERSION);
if (set->daemon) {
if (daemon_init() < 0)
fatal("Could not fork daemon!\n");
debug(LOW, "Daemon detached\n");
}
pthread_mutex_init(&stats.mutex, NULL);
/* Initialize signal handler */
sigemptyset(&signal_set);
sigaddset(&signal_set, SIGHUP);
sigaddset(&signal_set, SIGUSR1);
sigaddset(&signal_set, SIGUSR2);
sigaddset(&signal_set, SIGTERM);
sigaddset(&signal_set, SIGINT);
sigaddset(&signal_set, SIGQUIT);
if (!set->multiple)
checkPID(pid_file, set);
if (pthread_sigmask(SIG_BLOCK, &signal_set, NULL) != 0)
fatal("pthread_sigmask error\n");
/* start a thread to catch signals */
if (pthread_create(&sig_thread, NULL, sig_handler, (void *) &(signal_set)) != 0)
fatal("pthread_create error\n");
/* Read configuration file to establish local environment */
if (conf_file) {
if ((read_rated_config(conf_file, set)) < 0)
fatal("Could not read config file: %s\n", conf_file);
} else {
conf_file = malloc(BUFSIZE);
if (!conf_file)
fatal("Fatal malloc error!\n");
for(i=0;i<CONFIG_PATHS;i++) {
snprintf(conf_file, BUFSIZE, "%s%s", config_paths[i], DEFAULT_CONF_FILE);
if (read_rated_config(conf_file, set) >= 0)
break;
if (i == CONFIG_PATHS-1) {
snprintf(conf_file, BUFSIZE, "%s%s", config_paths[0], DEFAULT_CONF_FILE);
if ((write_rated_config(conf_file, set)) < 0)
fatal("Couldn't write config file.\n");
}
}
}
/* these probably aren't thread safe*/
init_snmp("rated");
/* TODO only do this if we're debugging or not daemonised? */
snmp_enable_stderrlog();
/* output oids numerically - this is equivalent to -On in the snmp tools */
netsnmp_ds_set_int(NETSNMP_DS_LIBRARY_ID, NETSNMP_DS_LIB_OID_OUTPUT_FORMAT, NETSNMP_OID_OUTPUT_NUMERIC);
if (set->dbon) {
/* load the database driver */
/* we need a db connection before we parse the targets file so we can check and create tables */
if (!(db_init(set) && db_connect(set)))
fatal("** Database error - check configuration.\n");
/* create our own internal tables */
table = db_check_and_create_data_table(RATED);
if (table == NULL)
fatal("** Database error - couldn't create rated table.\n");
else
free(table);
if (!db_check_and_create_oids_table(OIDS))
fatal("** Database error - couldn't create oids table.\n");
}
/* build list of hosts to be polled */
head = hash_target_file(target_file);
if (head == NULL)
fatal("Error updating target list.");
if (hosts < set->threads) {
debug(LOW, "Number of hosts is less than configured number of threads, defaulting to %i.\n", hosts);
set->threads = hosts;
}
debug(LOW, "Initializing threads (%d).\n", set->threads);
pthread_mutex_init(&(crew.mutex), NULL);
pthread_cond_init(&(crew.done), NULL);
pthread_cond_init(&(crew.go), NULL);
crew.current = NULL;
debug(HIGH, "Starting threads...");
crew.running = set->threads;
for (i = 0; i < set->threads; i++) {
crew.member[i].index = i;
crew.member[i].crew = &crew;
if (pthread_create(&(crew.member[i].thread), NULL, poller, (void *) &(crew.member[i])) != 0)
fatal("pthread_create error\n");
debug(HIGH, " %i", i);
}
debug(HIGH, " done\n");
/* spin waiting for all threads to start up */
debug(HIGH, "Waiting for thread startup.\n");
ts.tv_sec = 0;
ts.tv_nsec = 10000000; /* 10 ms */
gettimeofday(&begin_time, NULL);
while (crew.running > 0 ) {
nanosleep(&ts, NULL);
}
gettimeofday(&end_time, NULL);
debug(HIGH, "Waited %lu milliseconds for thread startup.\n", timediff(end_time, begin_time));
debug(LOW, "rated ready.\n");
/* Loop Forever Polling Target List */
while (1) {
/* check if we've been signalled */
if (quitting) {
debug(LOW, "Quitting: received signal %i.\n", quit_signal);
if (set->dbon)
db_disconnect();
/* one final stat output */
print_stats(stats, set);
unlink(pid_file);
exit(1);
} else if (waiting) {
debug(HIGH, "Processing pending SIGHUP.\n");
/* this just rebuilds the target list
* so all of the targets will reset to a first poll */
/* none of the threads should be running at this point so we shouldn't need a lock */
freed = free_host_list(head);
debug(HIGH, "Freed %i hosts\n", freed);
head = hash_target_file(target_file);
waiting = FALSE;
}
last_poll = stats.polls;
gettimeofday(&begin_time, NULL);
PT_MUTEX_LOCK(&(crew.mutex));
crew.current = head;
debug(LOW, "Queue ready, broadcasting thread go condition.\n");
PT_COND_BROAD(&(crew.go));
PT_MUTEX_UNLOCK(&(crew.mutex));
/*
* wait for signals from threads finishing
* we have to use a do loop because when this starts up the running count will be zero
* so wait at least once until we get a signal that some thread is done before checking for zero
*/
PT_MUTEX_LOCK(&(crew.mutex));
do {
PT_COND_WAIT(&(crew.done), &(crew.mutex));
} while (crew.running > 0);
PT_MUTEX_UNLOCK(&(crew.mutex));
if (quitting_now)
continue;
gettimeofday(&end_time, NULL);
poll_time = timediff(end_time, begin_time);
polls = stats.polls - last_poll;
rate = ((double) polls / poll_time) * 1000;
/* don't underflow */
if (poll_time < set->interval) {
sleep_time = set->interval - poll_time;
} else {
sleep_time = 0;
stats.slow++;
}
/* have to call this before we increment the round counter */
calc_stats(&stats, poll_time);
stats.round++;
debug(LOW, "Poll round %d completed %llu getnexts in %lu ms (%.0f/s).\n", stats.round, polls, poll_time, rate);
/* insert the internal poll data for this round into the rated table */
if (set->dbon)
db_insert(RATED, 0, end_time, stats.polls - last_poll, rate);
if (set->verbose >= LOW) {
print_stats(stats, set);
}
if (sleep_time > 0) {
sleepy(sleep_time, set);
} else {
debug(LOW, "Slow poll, not sleeping\n");
}
} /* while(1) */
exit(0);
}
/* Main rtgpoll */
int main(int argc, char *argv[]) {
crew_t crew;
pthread_t sig_thread;
sigset_t signal_set;
struct timeval now;
double begin_time, end_time, sleep_time;
char *conf_file = NULL;
char errstr[BUFSIZE];
int ch, i;
dfp = stderr;
/* Check argument count */
if (argc < 3)
usage(argv[0]);
/* Set default environment */
config_defaults(set);
/* Parse the command-line. */
while ((ch = getopt(argc, argv, "c:p:dhmDt:vz")) != EOF)
switch ((char) ch) {
case 'c':
conf_file = optarg;
break;
case 'd':
set->dboff = TRUE;
break;
case 'D':
set->daemon = FALSE;
break;
case 'h':
usage(argv[0]);
break;
case 'm':
set->multiple++;
break;
case 'p':
pid_file = optarg;
break;
case 't':
target_file = optarg;
break;
case 'v':
set->verbose++;
break;
case 'z':
set->withzeros = TRUE;
break;
}
debug(LOW, "RTG version %s starting.\n", VERSION);
if (set->daemon) {
if (daemon_init() < 0)
fatal("Could not fork daemon!\n");
debug(LOW, "Daemon detached\n");
}
/* Initialize signal handler */
sigemptyset(&signal_set);
sigaddset(&signal_set, SIGHUP);
sigaddset(&signal_set, SIGUSR1);
sigaddset(&signal_set, SIGUSR2);
sigaddset(&signal_set, SIGTERM);
sigaddset(&signal_set, SIGINT);
sigaddset(&signal_set, SIGQUIT);
if (!set->multiple)
checkPID(pid_file, set);
if (pthread_sigmask(SIG_BLOCK, &signal_set, NULL) != 0)
printf("pthread_sigmask error\n");
/* Read configuration file to establish local environment */
if (conf_file) {
if ((read_rtg_config(conf_file, set)) < 0)
fatal("Could not read config file: %s\n", conf_file);
} else {
conf_file = malloc(BUFSIZE);
if (!conf_file)
fatal("Fatal malloc error!\n");
for(i=0;i<CONFIG_PATHS;i++) {
snprintf(conf_file, BUFSIZE, "%s%s", config_paths[i], DEFAULT_CONF_FILE);
if (read_rtg_config(conf_file, set) >= 0)
break;
if (i == CONFIG_PATHS-1) {
snprintf(conf_file, BUFSIZE, "%s%s", config_paths[0], DEFAULT_CONF_FILE);
if ((write_rtg_config(conf_file, set)) < 0)
fatal("Couldn't write config file.\n");
}
}
}
/* hash list of targets to be polled */
init_hash();
entries = hash_target_file(target_file);
if (entries <= 0)
fatal("Error updating target list.");
debug(LOW, "Initializing threads (%d).\n", set->threads);
pthread_mutex_init(&(crew.mutex), NULL);
pthread_cond_init(&(crew.done), NULL);
pthread_cond_init(&(crew.go), NULL);
crew.work_count = 0;
/* Initialize the SNMP session */
debug(LOW, "Initializing SNMP (port %d).\n", set->snmp_port);
init_snmp("RTG");
/* Attempt to connect to the MySQL Database */
#ifndef FEATURES
#if HAVE_MYSQL
if (!(set->dboff)) {
if (mysql_dbconnect(set->dbdb, &mysql) < 0)
fatal("** Database error - check configuration.\n");
if (!mysql_ping(&mysql))
debug(LOW, "connected.\n");
else
fatal("server not responding.\n");
}
#endif
#else
#if HAVE_MYSQL
my_init();
#endif
#endif
debug(HIGH, "\nStarting threads.\n");
for (i = 0; i < set->threads; i++) {
crew.member[i].index = i;
crew.member[i].crew = &crew;
if (pthread_create(&(crew.member[i].thread), NULL, poller, (void *) &(crew.member[i])) != 0)
printf("pthread_create error\n");
}
if (pthread_create(&sig_thread, NULL, sig_handler, (void *) &(signal_set)) != 0)
printf("pthread_create error\n");
/* give threads time to start up */
sleep(1);
debug(LOW, "RTG Ready.\n");
/* Loop Forever Polling Target List */
while (1) {
lock = TRUE;
gettimeofday(&now, NULL);
begin_time = (double) now.tv_usec / 1000000 + now.tv_sec;
PT_MUTEX_LOCK(&(crew.mutex));
init_hash_walk();
current = getNext();
crew.work_count = entries;
PT_MUTEX_UNLOCK(&(crew.mutex));
if (set->verbose >= LOW) {
if (set->daemon)
sysloginfo("Queue ready, broadcasting thread go condition.");
else
timestamp("Queue ready, broadcasting thread go condition.");
}
PT_COND_BROAD(&(crew.go));
PT_MUTEX_LOCK(&(crew.mutex));
while (crew.work_count > 0) {
PT_COND_WAIT(&(crew.done), &(crew.mutex));
}
PT_MUTEX_UNLOCK(&(crew.mutex));
gettimeofday(&now, NULL);
lock = FALSE;
end_time = (double) now.tv_usec / 1000000 + now.tv_sec;
stats.poll_time = end_time - begin_time;
stats.round++;
sleep_time = set->interval - stats.poll_time;
if (waiting) {
debug(HIGH, "Processing pending SIGHUP.\n");
entries = hash_target_file(target_file);
waiting = FALSE;
}
if (set->verbose >= LOW) {
snprintf(errstr, sizeof(errstr), "Poll round %d complete.", stats.round);
if (set->daemon)
sysloginfo(errstr);
else
timestamp(errstr);
print_stats(stats, set);
}
if (sleep_time <= 0)
stats.slow++;
else
sleepy(sleep_time, set);
} /* while */
#ifndef FEATURES
#if HAVE_MYSQL
/* Disconnect from the MySQL Database, exit. */
if (!(set->dboff))
mysql_dbdisconnect(&mysql);
#endif
#endif
exit(0);
}
