static SFLSampler *getSampler(HSP *sp, char *devName, uint32_t ifIndex)
{
SFLSampler *sampler = sfl_agent_getSamplerByIfIndex(sp->sFlow->agent, ifIndex);
if(sampler == NULL) {
SFLDataSource_instance dsi;
SFL_DS_SET(dsi, 0, ifIndex, 0); // ds_class,ds_index,ds_instance
HSPSFlow *sf = sp->sFlow;
// add sampler (with sub-sampling rate), and poller too
uint32_t samplingRate = sf->sFlowSettings->ulogSubSamplingRate;
sampler = sfl_agent_addSampler(sf->agent, &dsi);
sfl_sampler_set_sFlowFsPacketSamplingRate(sampler, samplingRate);
sfl_sampler_set_sFlowFsReceiver(sampler, HSP_SFLOW_RECEIVER_INDEX);
if(devName) {
uint32_t pollingInterval = sf->sFlowSettings ?
sf->sFlowSettings->pollingInterval :
SFL_DEFAULT_POLLING_INTERVAL;
SFLPoller *poller = sfl_agent_addPoller(sf->agent, &dsi, sp, agentCB_getCounters_interface);
sfl_poller_set_sFlowCpInterval(poller, pollingInterval);
sfl_poller_set_sFlowCpReceiver(poller, HSP_SFLOW_RECEIVER_INDEX);
// remember the device name to make the lookups easier later.
// Don't want to point directly to the SFLAdaptor or SFLAdaptorNIO object
// in case it gets freed at some point. The device name is enough.
poller->userData = (void *)my_strdup(devName);
}
}
return sampler;
}
/**
* Searches for existing sampler with ifIndex and returns it. If there is no
* existing sampler with the ifIndex, creates a new one (and a poller) and
* returns it.
*/
static SFLSampler *getSampler(HSP *sp, char *devName, uint32_t ifIndex)
{
SFLSampler *sampler = sfl_agent_getSamplerByIfIndex(sp->sFlow->agent, ifIndex);
if (sampler == NULL) {
SFLDataSource_instance dsi;
SFL_DS_SET(dsi, 0, ifIndex, 0); // ds_class=0 interface
HSPSFlow *sf = sp->sFlow;
uint32_t samplingRate = sp->sFlow->sFlowSettings->samplingRate;
sampler = sfl_agent_addSampler(sf->agent, &dsi);
sampler->userData = (void *)my_strdup(devName);
sfl_sampler_set_sFlowFsPacketSamplingRate(sampler, samplingRate);
sfl_sampler_set_sFlowFsReceiver(sampler, HSP_SFLOW_RECEIVER_INDEX);
}
return sampler;
}
void sfl_agent_resetReceiver(SFLAgent *agent, SFLReceiver *receiver)
{
SFLReceiver *rcv;
SFLSampler *sm;
SFLPoller *pl;
/* tell samplers and pollers to stop sending to this receiver */
/* first get his receiverIndex */
u_int32_t rcvIdx = 0;
for(rcv = agent->receivers; rcv != NULL; rcv = rcv->nxt)
if(rcv == receiver) break;
/* now tell anyone that is using it to stop */
for(sm = agent->samplers; sm != NULL; sm = sm->nxt)
if(sfl_sampler_get_sFlowFsReceiver(sm) == rcvIdx) sfl_sampler_set_sFlowFsReceiver(sm, 0);
for(pl = agent->pollers; pl != NULL; pl = pl->nxt)
if(sfl_poller_get_sFlowCpReceiver(pl) == rcvIdx) sfl_poller_set_sFlowCpReceiver(pl, 0);
}
static SFLSampler *getSampler(HSP *sp, SFLAdaptor *adaptor)
{
HSPAdaptorNIO *adaptorNIO=(HSPAdaptorNIO *)adaptor->userData;
if(adaptorNIO) {
if(adaptorNIO->sampler == NULL) {
SFLDataSource_instance dsi;
SFL_DS_SET(dsi, 0, adaptor->ifIndex, 0); // ds_class,ds_index,ds_instance
HSPSFlow *sf = sp->sFlow;
// add sampler (with sub-sampling rate)
uint32_t samplingRate = sf->sFlowSettings->ulogSubSamplingRate;
adaptorNIO->sampler = sfl_agent_addSampler(sf->agent, &dsi);
sfl_sampler_set_sFlowFsPacketSamplingRate(adaptorNIO->sampler, samplingRate);
sfl_sampler_set_sFlowFsReceiver(adaptorNIO->sampler, HSP_SFLOW_RECEIVER_INDEX);
// and make sure we have a poller too
getPoller(sp, adaptor);
}
return adaptorNIO->sampler;
}
return NULL;
}
void sfl_agent_resetReceiver(SFLAgent *agent, SFLReceiver *receiver)
{
/* tell samplers and pollers to stop sending to this receiver */
/* first get his receiverIndex */
u_int32_t rcvIdx = 0;
SFLReceiver *rcv = agent->receivers;
for(; rcv != NULL; rcv = rcv->nxt) {
rcvIdx++; /* thanks to Diego Valverde for pointing out this bugfix */
if(rcv == receiver) {
/* now tell anyone that is using it to stop */
SFLSampler *sm = agent->samplers;
SFLPoller *pl = agent->pollers;
for(; sm != NULL; sm = sm->nxt)
if(sfl_sampler_get_sFlowFsReceiver(sm) == rcvIdx) sfl_sampler_set_sFlowFsReceiver(sm, 0);
for(; pl != NULL; pl = pl->nxt)
if(sfl_poller_get_sFlowCpReceiver(pl) == rcvIdx) sfl_poller_set_sFlowCpReceiver(pl, 0);
break;
}
}
}
