---

rmClient.cpp

Go to the documentation of this file.

#include <stdio.h>
#include <wchar.h>
#include <string.h>

#include <exception>

// #include <cassert>

#include <ResourceMonitor/ResourceMonitor.h>

#include "rmClient.h"

#include "rmClientError.h"
#include "rmSysInfo.h"
#include "rmConfig.h"
#include "rmControl.h"
#include "rmMonitor.h"
#include "rmEvent.h"

#include "eventLog.h"

#include "uuid/uuid.h"

#include <tcpd.h>

int allow_severity       = LOG_INFO;
int deny_severity        = LOG_WARNING;

class subagentBase
{
  public:
    subagentBase(){}
    virtual ~subagentBase(){}

    virtual int init() = 0;

    virtual void shutdown() = 0;

    virtual int getInteger32(RMEntity aEntity, RMAttribute aName, unsigned long aID, long & aValue);
    virtual int setInteger32(RMEntity aEntity, RMAttribute aName, unsigned long aID, long aValue);

    virtual int getUnsigned(RMEntity aEntity, RMAttribute aName, unsigned long aID, unsigned long & aValue);
    virtual int setUnsigned(RMEntity aEntity, RMAttribute aName, unsigned long aID, unsigned long aValue);

    virtual int getUnsigned64(RMEntity aEntity, RMAttribute aName, unsigned long aID, u_int64_t & aValue);
    virtual int setUnsigned64(RMEntity aEntity, RMAttribute aName, unsigned long aID, u_int64_t aValue);

    virtual int getUUID(RMEntity aEntity, RMAttribute aName, unsigned long aID, uuid_t aUUID);

    virtual int getUUIDString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * aBuf);
    virtual int setUUIDString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * aUUID);

    virtual int getString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * & aBuf);
    virtual int setString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * aBuf);

    virtual int getBinary(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * & aBuf, unsigned long & aLength);

    virtual int newEntry(RMEntity aEntity, unsigned long aID);

    virtual int deleteEntry(RMEntity aEntity, unsigned long aID);

    virtual unsigned long getNewId(RMEntity aEntity);

    virtual int findId(RMEntity aEntity, unsigned long aID);

    virtual int getNextId(RMEntity aEntity, unsigned long aID, unsigned long & aNextID);

    virtual size_t getCount(RMEntity aEntity);

    virtual void addEvent(posix_log_recid_t aRecid, unsigned long aId) = 0;

    virtual void processEvents() = 0;

  protected:
    virtual IBaseAccessor * getEntity(RMEntity aEntity) = 0;
};

class rmClient : public subagentBase
{
  public:
    rmClient(){}
    virtual ~rmClient();

    virtual int init();
    virtual void shutdown();

    virtual void addEvent(posix_log_recid_t aRecid, unsigned long aMonid)
    {
      return mEvent->addEvent(aRecid, aMonid);
    }

    virtual void processEvents()
    {
      return mEvent->processEvents();
    }

  protected:
    virtual IBaseAccessor * getEntity(RMEntity aEntity);

  private:
    subsystemInfo * mSubsystemInfo;
    resourceInfo * mResourceInfo;
    statisticInfo * mStatisticInfo;
    dataCaptureInfo * mDataCaptureInfo;
    rmThresholdMonitorConfig * mThresholdConfig;
    rmWatermarkMonitorConfig * mWatermarkConfig;
    rmLeakyBucketMonitorConfig * mLeakyBucketConfig;
    rmGenericMonitorConfig * mGenericConfig;
    rmThresholdMonitorControl * mThresholdControl;
    rmLeakyBucketMonitorControl * mLeakyBucketControl;
    rmGenericMonitorControl * mGenericControl;
    rmMonitor * mMonitor;
    rmEvent * mEvent;
    eventLog * mEventLog;
};

int
subagentBase::getInteger32(RMEntity aEntity, RMAttribute aName, unsigned long aID, long & aValue)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getInteger32(aName, aID, aValue);
}

int
subagentBase::setInteger32(RMEntity aEntity, RMAttribute aName, unsigned long aID, long aValue)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->setInteger32(aName, aID, aValue);
}

int
subagentBase::getUnsigned(RMEntity aEntity, RMAttribute aName, unsigned long aID, unsigned long & aValue)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getUnsigned(aName, aID, aValue);
}

int 
subagentBase::setUnsigned(RMEntity aEntity, RMAttribute aName, unsigned long aID, unsigned long aValue)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->setUnsigned(aName, aID, aValue);
}

int 
subagentBase::getUnsigned64(RMEntity aEntity, RMAttribute aName, unsigned long aID, u_int64_t & aValue)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getUnsigned64(aName, aID, aValue);
}

int 
subagentBase::setUnsigned64(RMEntity aEntity, RMAttribute aName, unsigned long aID, u_int64_t aValue)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->setUnsigned64(aName, aID, aValue);
}

int 
subagentBase::getUUID(RMEntity aEntity, RMAttribute aName, unsigned long aID, uuid_t aUUID)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getUUID(aName, aID, aUUID);
}

int 
subagentBase::getUUIDString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * aBuf)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getUUIDString(aName, aID, aBuf);
}

int 
subagentBase::setUUIDString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * aUUID)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->setUUIDString(aName, aID, aUUID);
}

int 
subagentBase::getString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * & aBuf)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getString(aName, aID, aBuf);
}

int 
subagentBase::setString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * aBuf)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->setString(aName, aID, aBuf);
}

int 
subagentBase::getBinary(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * & aBuf, unsigned long & aLength)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getBinary(aName, aID, aBuf, aLength);
}

int 
subagentBase::newEntry(RMEntity aEntity, unsigned long aID)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->addDefaultElement(aID);
}

int 
subagentBase::deleteEntry(RMEntity aEntity, unsigned long aID)
{
  IBaseAccessor * pBase = getEntity(aEntity);

  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->removeElement(aID);
}

unsigned long 
subagentBase::getNewId(RMEntity aEntity)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getNewId();
}

int 
subagentBase::findId(RMEntity aEntity, unsigned long aID)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->findExactId(aID);
}

int 
subagentBase::getNextId(RMEntity aEntity, unsigned long aID, unsigned long & aNextID)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getNextId(aID, aNextID);
}

size_t 
subagentBase::getCount(RMEntity aEntity)
{
  IBaseAccessor * pBase = getEntity(aEntity);
  if(!pBase)
    return RMCLIENT_ERROR_NOSUCHENTITY;

  return pBase->getCount();
}

rmClient::~rmClient()
{
  shutdown();
}

int
rmClient::init()
{

  int ret;

  long startupTime;
  struct timezone tz;
  struct timeval tv;

  gettimeofday(&tv, &tz);
  startupTime = tv.tv_sec; // assign the initial ID number.

  int i = 0;

  try{
    // init the global rmSysinfo class.
    mSubsystemInfo = new subsystemInfo(startupTime);
    mResourceInfo = new resourceInfo();
    mStatisticInfo = new statisticInfo();

    // because mSubsystemInfo also contains a pointer to mResourceInfo and
    // mStatisticInfo and so on, we have to do init() this way.
    ret = mSubsystemInfo->init(mResourceInfo, mStatisticInfo);
    if(ret)
      return ret;

    ret = mResourceInfo->init(mSubsystemInfo, mStatisticInfo);
    if(ret)
      return ret;

    ret = mStatisticInfo->init(mSubsystemInfo, mResourceInfo);
    if(ret)
      return ret;

    mDataCaptureInfo = new dataCaptureInfo(mSubsystemInfo);
    ret = mDataCaptureInfo->init();
    if(ret)
      return ret;

    mThresholdConfig = new rmThresholdMonitorConfig;
    ret = mThresholdConfig->init();
    if(ret) {
      return ret;
    }

    mWatermarkConfig = new rmWatermarkMonitorConfig;
    ret = mWatermarkConfig->init();
    if(ret)
      return ret;

    mLeakyBucketConfig = new rmLeakyBucketMonitorConfig;
    ret = mLeakyBucketConfig->init();
    if(ret)
      return ret;

    // this has to be done after both gThresholdConfig and gWatermarkConfig
    // are constructed.

    mGenericConfig = new rmGenericMonitorConfig(mSubsystemInfo, mThresholdConfig, mWatermarkConfig, mLeakyBucketConfig);
    ret = mGenericConfig->init();
    if(ret) {
      return ret;
    }

    mThresholdControl = new rmThresholdMonitorControl;
    ret = mThresholdControl->init();
    if(ret)
      return ret;
 
    mLeakyBucketControl = new rmLeakyBucketMonitorControl;
    ret = mLeakyBucketControl->init();
    if(ret)
      return ret;

    mGenericControl = new rmGenericMonitorControl(mThresholdControl, mLeakyBucketControl);
    ret = mGenericControl->init();
    if(ret)
      return ret;
  
    mMonitor = new rmMonitor(startupTime, mSubsystemInfo, mGenericConfig, mThresholdConfig, mWatermarkConfig, mLeakyBucketConfig, mGenericControl, mThresholdControl, mLeakyBucketControl);
    ret = mMonitor->init();
    if(ret)
      return ret;
  
    mEvent = new rmEvent(mSubsystemInfo, mResourceInfo, mDataCaptureInfo);
    ret = mEvent->init();
    if(ret) 
      return ret;

    // rmEvent requires eventLog to be initialized.
    mEventLog = new eventLog(NULL);
    ret = mEventLog->init();
    if(ret)
      return ret;

    return RMCLIENT_SUCCESS;

  }
  catch(...) {
    shutdown();

    return RMCLIENT_ERROR_MEMORY;
  }
}

void
rmClient::shutdown()
{
  delete mGenericConfig;
  delete mThresholdConfig;
  delete mWatermarkConfig;
  delete mLeakyBucketConfig;
  delete mGenericControl;
  delete mThresholdControl;
  delete mLeakyBucketControl;
  delete mMonitor;
  delete mSubsystemInfo;
  delete mResourceInfo;
  delete mStatisticInfo;
  delete mDataCaptureInfo;
  delete mEvent;
  delete mEventLog;
}

IBaseAccessor *
rmClient::getEntity(RMEntity aEntity)
{
  IBaseAccessor * pBase = NULL;

  switch(aEntity) {

    case SUBSYSTEMINFO: 
      pBase = mSubsystemInfo;
      break;

    case RESORUCEINFO: 
      pBase = mResourceInfo;
      break;

    case STATISTICINFO: 
      pBase = mStatisticInfo;
      break;

    case DATACAPTUREPLUGININFO: 
      pBase = mDataCaptureInfo;
      break;

    case GENERICCONFIG: 
      pBase = mGenericConfig;
      break;

    case THRESHOLDCONFIG: 
      pBase = mThresholdConfig;
      break;

    case WATERMARKCONFIG: 
      pBase = mWatermarkConfig;
      break;

    case LEAKYBUCKETCONFIG: 
      pBase = mLeakyBucketConfig;
      break;

    case GENERICCONTROL: 
      pBase = mGenericControl;
      break;

    case THRESHOLDCONTROL: 
      pBase = mThresholdControl;
      break;

    case LEAKYBUCKETCONTROL: 
      pBase = mLeakyBucketControl;
      break;

    case MONITOR: 
      pBase = mMonitor;
      break;

    case EVENTLOG: 
      pBase = mEventLog;
      break;

    default:
      return NULL;
  }

  return pBase;
}



//
// begin of external c functions.
//

subagentBase * gSubagent(NULL);

int
initRMClient()
{
  int ret;
  gSubagent = new rmClient();
  return gSubagent->init();
}

void
shutdownRMClient()
{
  gSubagent->shutdown();
}

int
getRMVersion()
{
  return ResourceMonitor::rmGetResourceMonitorVersion();
}


/*************************************
 *
 * The new interface implementation
 *
 *************************************/

int
getInteger32(RMEntity aEntity, RMAttribute aName, unsigned long aID, long * aValue)
{
  return gSubagent->getInteger32(aEntity, aName, aID, *aValue);
}

int
setInteger32(RMEntity aEntity, RMAttribute aName, unsigned long aID, long aValue)
{
  return gSubagent->setInteger32(aEntity, aName, aID, aValue);
}

int
getUnsigned(RMEntity aEntity, RMAttribute aName, unsigned long aID, unsigned long * aValue)
{
  return gSubagent->getUnsigned(aEntity, aName, aID, *aValue);
}

int
setUnsigned(RMEntity aEntity, RMAttribute aName, unsigned long aID, unsigned long aValue)
{
  return gSubagent->setUnsigned(aEntity, aName, aID, aValue);
}

int
getUnsigned64(RMEntity aEntity, RMAttribute aName, unsigned long aID, u_int64_t * aValue)
{
  return gSubagent->getUnsigned64(aEntity, aName, aID, *aValue);
}

int
setUnsigned64(RMEntity aEntity, RMAttribute aName, unsigned long aID, u_int64_t aValue)
{
  return gSubagent->setUnsigned64(aEntity, aName, aID, aValue);
}

int
getUUIDString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * aBuf)
{
  return gSubagent->getUUIDString(aEntity, aName, aID, aBuf);
}

int
getUUID(RMEntity aEntity, RMAttribute aName, unsigned long aID, uuid_t aUUID)
{
  return gSubagent->getUUID(aEntity, aName, aID, aUUID);
}

int
setUUIDString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * aUUID)
{
  return gSubagent->setUUIDString(aEntity, aName, aID, aUUID);
}

int
getString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * * aBuf)
{
  return gSubagent->getString(aEntity, aName, aID, *aBuf);
}


int
setString(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * aBuf)
{
  return gSubagent->setString(aEntity, aName, aID, aBuf);
}


int
newEntry(RMEntity aEntity, unsigned long aID)
{
  return gSubagent->newEntry(aEntity, aID);
}

int
deleteEntry(RMEntity aEntity, unsigned long aID)
{
  return gSubagent->deleteEntry(aEntity, aID);
}

unsigned long 
getNewId(RMEntity aEntity)
{
  return gSubagent->getNewId(aEntity);
}


int
findId(RMEntity aEntity, unsigned long aID)
{
  return gSubagent->findId(aEntity, aID);
}

int
getNextId(RMEntity aEntity, unsigned long aID, unsigned long * aNextID)
{
  return gSubagent->getNextId(aEntity, aID, *aNextID);
}


size_t
getCount(RMEntity aEntity)
{
  return gSubagent->getCount(aEntity);
}

int 
getBinary(RMEntity aEntity, RMAttribute aName, unsigned long aID, char * * aBuf, unsigned long * aLength)
{
  return gSubagent->getBinary(aEntity, aName, aID, *aBuf, *aLength);
}


void
addEvent(posix_log_recid_t aRecid, unsigned long aId)
{
  return gSubagent->addEvent(aRecid, aId);
}


void
processEvents()
{
  return gSubagent->processEvents();
}

---