---

rmSubagent.c

Go to the documentation of this file.

#include <ucd-snmp/ucd-snmp-config.h>
#ifdef HAVE_NETINET_IN_H
#include <netinet/in.h>
#endif
#include <ucd-snmp/ucd-snmp-includes.h>
#include <ucd-snmp/ucd-snmp-agent-includes.h>
#include <signal.h>

#include <ucd-snmp/agent_trap.h>

#include "rmClient.h"

#include <unistd.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/file.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>

#include "RMSystemInfo.h"
#include "RMGenericConfig.h"
#include "RMThresholdConfig.h"
#include "RMWatermarkConfig.h"
#include "RMLeakyBucketConfig.h"
#include "RMGenericControl.h"
#include "RMThresholdControl.h"
#include "RMLeakyBucketControl.h"
#include "RMMonitors.h"
#include "ResourceMonitor/MonitoringEventSchema.h"

static int keep_running;

RETSIGTYPE
stop_server(int a) {
    keep_running = 0;
}

void daemonize();

void catchNotificationEvent(int signo, siginfo_t *info, void *ignored);

void
catchNotificationEvent(int signo, siginfo_t *info, void *ignored)
{
  posix_log_recid_t recid;
  unsigned long monid;

  posix_log_siginfo_getrecid(info, ignored, &recid);
  monid = (unsigned long) info->si_int;

  addEvent(recid, monid);
}

int main () {

  int agentx_subagent=1; /* change this if you're a master agent */

  struct sigaction act;

  memset(&act, 0, sizeof(act));
  act.sa_flags = SA_SIGINFO | SA_RESTART;
  act.sa_handler = (void (*) (int)) catchNotificationEvent;
  sigaction(SIGRTMIN + 1, &act, NULL);


  daemonize();

  /* print log errors to stderr */
  snmp_enable_stderrlog();

  /* we're an agentx subagent? */
  if (agentx_subagent) {
    /* make us a agentx client. */
    ds_set_boolean(DS_APPLICATION_ID, DS_AGENT_ROLE, 1);
  }

  /* initialize the agent library */
  init_agent("rmSubagent");

  initRMClient();

  init_RMSystemInfo();
  init_RMGenericConfig();
  init_RMThresholdConfig();
  init_RMWatermarkConfig();
  init_RMLeakyBucketConfig();
  init_RMGenericControl();
  init_RMThresholdControl();
  init_RMLeakyBucketControl();
  init_RMMonitors();
  
  /* ustMain will be used to read ustMain.conf files. */
  init_snmp("rmSubagent");

  /* If we're going to be a snmp master agent */
  if (!agentx_subagent)
    init_master_agent(161, NULL, NULL);  /* open port 161 (UDP:snmp) */

  /* In case we recevie a request to stop (kill -TERM or kill -INT) */
  keep_running = 1;
  signal(SIGTERM, stop_server);
  signal(SIGINT, stop_server);

  /* you're main loop here... */
  while(keep_running) {
    /* if you use select(), see snmp_select_info() in snmp_api(3) */
    /*     --- OR ---  */
    agent_check_and_process(1); /* 0 == don't block */

    processEvents();

  }


  /* at shutdown time */
  snmp_shutdown("rmSubagent");
  shutdownRMClient();

  return 0;
}

void
daemonize()
{
  pid_t pid;

  int dpid;
  FILE *pidfile;
  int fd;

  char * pfile;

  pfile = "/var/run/rmsubagentd.pid";


  if((pid = fork()) < 0) {
    fprintf(stderr, "can not fork child process. ");
    exit(1);
  }
  else if (pid > 0) {
    exit(0);
  }

  (void)setpgrp();

  (void)signal(SIGHUP, SIG_IGN);

  if ((pid = fork()) < 0) {
    fprintf(stderr, " can not fork the second child process.");
    exit(1);
  }
  else if(pid > 0) {
    exit(0);
  }

  chdir("/");
  umask(0);

  if ((fd = open(pfile, O_RDWR|O_CREAT, 0644)) == -1) 
    return;

  if(( pidfile = fdopen(fd, "r+")) == NULL) {
    close(fd);
    return;
  }

  dpid = getpid();
  flock(fd, LOCK_EX);
  fprintf(pidfile, "%d\n", dpid);
  flock(fd, LOCK_UN);
  fclose(pidfile);
  close(fd);
}

---