SFLPoller *sfl_agent_getNextPoller(SFLAgent *agent, SFLDataSource_instance *pdsi)
{
/* return the one lexograpically just after it - assume they are sorted
correctly according to the lexographical ordering of the object ids */
SFLPoller *pl = sfl_agent_getPoller(agent, pdsi);
return pl ? pl->nxt : NULL;
}
void removePoller(HSP *sp, SFLAdaptor *adaptor)
{
SFLDataSource_instance dsi;
SFL_DS_SET(dsi, SFL_DSCLASS_IFINDEX, adaptor->ifIndex, 0);
SFLPoller *poller = sfl_agent_getPoller(sp->sFlow->agent, &dsi);
if (poller != NULL) {
if (poller->userData != NULL) {
my_free(poller->userData);
}
removeQueuedPoller(sp, poller);
sfl_agent_removePoller(sp->sFlow->agent, &dsi);
myLog(LOG_INFO, "removePoller: removing poller with ifIndex=%u (portName=%s, %S)",
adaptor->ifIndex,
adaptor->deviceName, ((HVSVPortInfo *)adaptor->userData)->portFriendlyName);
}
}
/**
* Updates the switch port list with the information in pSwitchConfig
* obtained from the filter.
*/
void updateSwitchPorts(HSP *sp, PAllSwitchesConfig config)
{
if (config->revision <= sp->portInfoRevision) {
return;
}
if (sp->vAdaptorList == NULL) {
sp->vAdaptorList = adaptorListNew();
} else {
adaptorListMarkAll(sp->vAdaptorList);
}
PSwitchConfig switchConfig;
for (uint32_t switchNum = 0; switchNum < config->numSwitches; switchNum++) {
if (switchNum == 0) {
switchConfig = GET_FIRST_SWITCH_CONFIG(config);
} else {
switchConfig = GET_NEXT_SWITCH_CONFIG(switchConfig);
}
wchar_t *switchName = ndiswcsdup(&switchConfig->switchName);
uint64_t switchId = switchConfig->switchID;
for (uint32_t portNum = 0; portNum < switchConfig->numPorts; portNum++) {
PPortEntry portEntry = GET_PORT_ENTRY_AT(switchConfig, portNum);
uint32_t portId = portEntry->portID;
wchar_t *portName = ndiswcsdup(&portEntry->portName);
char portGuid[FORMATTED_GUID_LEN+1];
guidToString(portName, (UCHAR *)portGuid, FORMATTED_GUID_LEN);
SFLAdaptor *switchPort = adaptorListGet(sp->vAdaptorList, portGuid);
if (switchPort == NULL) {
//new port so add to the vadaptor list
//convert GUID to uuid format to look up ifIndex/dsIndex
char uuid[16];
hexToBinary((UCHAR *)portGuid, (UCHAR *)uuid, 33);
uint32_t ifIndex = assign_dsIndex(&sp->portStore, uuid, &sp->maxIfIndex, &sp->portStoreInvalid);
switchPort = addVAdaptor(sp->vAdaptorList, portGuid, ifIndex);
HVSVPortInfo *portInfo = (HVSVPortInfo *)switchPort->userData;
portInfo->filterEnabled = TRUE;
portInfo->portId = portId;
portInfo->revision = portEntry->revision;
switchPort->marked = FALSE;
updatePortSwitchName(switchPort, switchName);
portInfo->switchId = switchConfig->switchID;
myLog(LOG_INFO, "updateSwitchPorts: Added new portId=%u ifIndex=%u deviceName=%s switchId=%llu switchName=%S",
portInfo->portId, switchPort->ifIndex, switchPort->deviceName, portInfo->switchId, portInfo->switchName);
addPoller(sp, switchPort);
} else {
//we already know about this port, so make sure we have a poller
//and the current info
SFLDataSource_instance dsi;
SFL_DS_SET(dsi, 0, switchPort->ifIndex, 0);
SFLPoller *poller = sfl_agent_getPoller(sp->sFlow->agent, &dsi);
if (poller == NULL) {
poller = addPoller(sp, switchPort);
}
HVSVPortInfo *portInfo = (HVSVPortInfo *)switchPort->userData;
if (portEntry->revision > portInfo->revision) {
updatePortSwitchName(switchPort, switchName);
portInfo->revision = portEntry->revision;
if (poller != NULL) {
sfl_poller_resetCountersSeqNo(poller);
}
myLog(LOG_INFO, "updateSwitchPorts: revision changed: portId=%u ifIndex=%u deviceName=%s switchId=%llu switchName=%S",
portInfo->portId, switchPort->ifIndex, switchPort->deviceName, portInfo->switchId, portInfo->switchName);
}
portInfo->filterEnabled = TRUE;
switchPort->marked = FALSE;
}
my_free(portName);
}
my_free(switchName);
}
//now sweep
//remove the pollers for non-sampling ports
for (uint32_t i = 0; i < sp->vAdaptorList->num_adaptors; i++) {
SFLAdaptor *vAdaptor = sp->vAdaptorList->adaptors[i];
if (vAdaptor->marked) {
HVSVPortInfo *portInfo = (HVSVPortInfo *)vAdaptor->userData;
if (portInfo->filterEnabled) {
//filter (ie sampling) has been disabled in the switch with this port
((HVSVPortInfo *)vAdaptor->userData)->portId = 0;
removePoller(sp, vAdaptor);
portInfo->filterEnabled = FALSE;
//Clear the mark so this port will not be deleted, the VM and adaptor may still exist.
//If the adaptor does not exist, it will be removed when we next refresh the VMs.
vAdaptor->marked = FALSE;
} else {
//this was a port added for a vm on a switch with the filter disabled, so
//just clear the mark so that it will not be deleted.
vAdaptor->marked = FALSE;
}
}
}
//Now remove the marked adaptors and their port info from the list
adaptorListFreeMarked(sp->vAdaptorList, freePortInfo);
//TODO ageout the persistent ifIndex->GUID mapping and remove from vAdaptor list.
sp->portInfoRevision = config->revision;
readWMISwitchPorts(sp); //update the ifSpeed, MAC, VM name
sp->refreshVms = TRUE;
}
