|
|
/*++
Copyright (c) 1999, Microsoft Corporation
Module Name:
sample\mibmanager.c
Abstract:
The file contains IP Sample's MIB implementation.
Revision History:
MohitT June-15-1999 Created
--*/
#include "pchsample.h"
#pragma hdrstop
DWORDWINAPIMM_MibSet ( IN PIPSAMPLE_MIB_SET_INPUT_DATA pimsid)/*++
Routine Description Set IPSAMPLE's global or interface configuration.
Locks Acquires shared (g_ce.pneNetworkEntry)->rwlLock Releases (g_ce.pneNetworkEntry)->rwlLock
Arguments pimsid input data, contains global/interface configuration Return Value NO_ERROR success Error Code o/w --*/{ DWORD dwErr = NO_ERROR; ROUTING_PROTOCOL_EVENTS rpeEvent; MESSAGE mMessage = {0, 0, 0}; PINTERFACE_ENTRY pie;
if (!ENTER_SAMPLE_API()) { return ERROR_CAN_NOT_COMPLETE; }
do // breakout loop
{ // set global configuration
if (pimsid->IMSID_TypeID is IPSAMPLE_GLOBAL_CONFIG_ID) { if (pimsid->IMSID_BufferSize < sizeof(IPSAMPLE_GLOBAL_CONFIG)) { dwErr = ERROR_INVALID_PARAMETER; break; }
dwErr = CM_SetGlobalInfo((PVOID) pimsid->IMSID_Buffer); if (dwErr != NO_ERROR) break;
rpeEvent = SAVE_GLOBAL_CONFIG_INFO; } // set interface configuration
else if (pimsid->IMSID_TypeID is IPSAMPLE_IF_CONFIG_ID) { if (pimsid->IMSID_BufferSize < sizeof(IPSAMPLE_IF_CONFIG)) { dwErr = ERROR_INVALID_PARAMETER; break; }
dwErr = NM_SetInterfaceInfo(pimsid->IMSID_IfIndex, (PVOID) pimsid->IMSID_Buffer); if (dwErr != NO_ERROR) break;
rpeEvent = SAVE_INTERFACE_CONFIG_INFO; mMessage.InterfaceIndex = pimsid->IMSID_IfIndex; } // error, unexpected type
else { dwErr = ERROR_INVALID_PARAMETER; break; }
// notify router manager
if (EnqueueEvent(rpeEvent, mMessage) is NO_ERROR) SetEvent(g_ce.hMgrNotificationEvent);
} while(FALSE);
LEAVE_SAMPLE_API();
return dwErr;}
DWORDWINAPIMM_MibGet ( IN PIPSAMPLE_MIB_GET_INPUT_DATA pimgid, OUT PIPSAMPLE_MIB_GET_OUTPUT_DATA pimgod, IN OUT PULONG pulOutputSize, IN MODE mMode)/*++
Routine Description Handles the structure accesses required to read MIB data. Supports three modes of querying: EXACT, FIRST, and NEXT, which correspond to MibGet(), MibGetFirst(), and MibGetNext() respectively.
Locks Acquires shared (g_ce.pneNetworkEntry)->rwlLock Releases (g_ce.pneNetworkEntry)->rwlLock
Arguments pimgid input data pimgod output buffer pulOutputSize IN size of output buffer given OUT size of output buffer needed mMode query type Return Value NO_ERROR success Error Code o/w --*/{ DWORD dwErr = NO_ERROR; ULONG ulSizeGiven = 0; ULONG ulSizeNeeded = 0; PINTERFACE_ENTRY pie;
if (!ENTER_SAMPLE_API()) { return ERROR_CAN_NOT_COMPLETE; }
// compute the size of the buffer available for storing
// returned structures (the size of IMGOD_Buffer)
if (*pulOutputSize < sizeof(IPSAMPLE_MIB_GET_OUTPUT_DATA)) ulSizeGiven = 0; else ulSizeGiven = *pulOutputSize - sizeof(IPSAMPLE_MIB_GET_OUTPUT_DATA);
switch (pimgid->IMGID_TypeID) { // the global statistics struct is fixed-length.
// there is a single instance.
case IPSAMPLE_GLOBAL_STATS_ID: { PIPSAMPLE_GLOBAL_STATS pigsdst, pigssrc; // only GET_EXACT and GET_FIRST are valid for the
// global stats object since there is only one entry
if (mMode is GET_NEXT) { dwErr = ERROR_NO_MORE_ITEMS; break; }
// set the output size required for this entry
ulSizeNeeded = sizeof(IPSAMPLE_GLOBAL_STATS);
// check that the output buffer is big enough
if (ulSizeGiven < ulSizeNeeded) { dwErr = ERROR_INSUFFICIENT_BUFFER; break; }
pimgod->IMGOD_TypeID = IPSAMPLE_GLOBAL_STATS_ID;
// since access to this structure is synchronized through
// locked increments/decrements, we must copy it field by field
pigssrc = &(g_ce.igsStats); pigsdst = (PIPSAMPLE_GLOBAL_STATS) pimgod->IMGOD_Buffer;
pigsdst->ulNumInterfaces = pigssrc->ulNumInterfaces;
break; }
// the global configuration structure may be of variable size.
// there is a single instance.
case IPSAMPLE_GLOBAL_CONFIG_ID: { // only GET_EXACT and GET_FIRST are valid for the
// global configuration object since there is only one entry
if (mMode is GET_NEXT) { dwErr = ERROR_NO_MORE_ITEMS; break; }
// CM_GetGlobalInfo() decides whether the buffer size is
// sufficient. if so, it retrieves the global configuration.
// either case it sets the required output buffer size.
dwErr = CM_GetGlobalInfo ((PVOID) pimgod->IMGOD_Buffer, &ulSizeGiven, NULL, NULL, NULL); ulSizeNeeded = ulSizeGiven;
// check that the output buffer was big enough
if (dwErr != NO_ERROR) break;
pimgod->IMGOD_TypeID = IPSAMPLE_GLOBAL_CONFIG_ID; break; }
// the interface statistics struct is fixed-length.
// there may be multiple instances.
case IPSAMPLE_IF_STATS_ID: { PIPSAMPLE_IF_STATS piisdst, piissrc;
ACQUIRE_READ_LOCK(&(g_ce.pneNetworkEntry)->rwlLock);
do // breakout loop
{ // retrieve the interface whose stats are to be read
dwErr = IE_GetIndex(pimgid->IMGID_IfIndex, mMode, &pie); if (dwErr != NO_ERROR) break; // set the output size required for this entry
ulSizeNeeded = sizeof(IPSAMPLE_IF_STATS);
// check that the output buffer is big enough
if (ulSizeGiven < ulSizeNeeded) { dwErr = ERROR_INSUFFICIENT_BUFFER; break; }
pimgod->IMGOD_TypeID = IPSAMPLE_IF_STATS_ID; pimgod->IMGOD_IfIndex = pie->dwIfIndex;
// access to this structure is synchronized through locked
// increments/decrements, so we copy it field by field
piissrc = &(pie->iisStats); piisdst = (PIPSAMPLE_IF_STATS) pimgod->IMGOD_Buffer;
piisdst->ulNumPackets = piissrc->ulNumPackets; } while (FALSE);
RELEASE_READ_LOCK(&(g_ce.pneNetworkEntry)->rwlLock);
break; }
// the interface configuration structure may be of variable size.
// there may be multiple instances.
case IPSAMPLE_IF_CONFIG_ID: { // get the queried interface's index
ACQUIRE_READ_LOCK(&(g_ce.pneNetworkEntry)->rwlLock);
do // breakout loop
{ dwErr = IE_GetIndex(pimgid->IMGID_IfIndex, mMode, &pie); if (dwErr != NO_ERROR) break;
// read lock acuired again, which is fine :)
dwErr = NM_GetInterfaceInfo(pie->dwIfIndex, (PVOID) pimgod->IMGOD_Buffer, &ulSizeGiven, NULL, NULL, NULL); ulSizeNeeded = ulSizeGiven;
// check that the output buffer was big enough
if (dwErr != NO_ERROR) break; pimgod->IMGOD_TypeID = IPSAMPLE_IF_CONFIG_ID; pimgod->IMGOD_IfIndex = pie->dwIfIndex; } while (FALSE);
RELEASE_READ_LOCK(&(g_ce.pneNetworkEntry)->rwlLock);
break; }
// the interface binding structure is of variable size.
// there may be multiple instances.
case IPSAMPLE_IF_BINDING_ID: { PIPSAMPLE_IF_BINDING piib; PIPSAMPLE_IP_ADDRESS piia;
ACQUIRE_READ_LOCK(&(g_ce.pneNetworkEntry)->rwlLock);
do // breakout loop
{ // retrieve the interface whose stats are to be read
dwErr = IE_GetIndex(pimgid->IMGID_IfIndex, mMode, &pie); if (dwErr != NO_ERROR) break; // set the output size required for this entry
ulSizeNeeded = sizeof(IPSAMPLE_IF_BINDING) + pie->ulNumBindings * sizeof(IPSAMPLE_IP_ADDRESS);
// check that the output buffer is big enough
if (ulSizeGiven < ulSizeNeeded) { dwErr = ERROR_INSUFFICIENT_BUFFER; break; }
pimgod->IMGOD_TypeID = IPSAMPLE_IF_BINDING_ID; pimgod->IMGOD_IfIndex = pie->dwIfIndex;
piib = (PIPSAMPLE_IF_BINDING) pimgod->IMGOD_Buffer; piia = IPSAMPLE_IF_ADDRESS_TABLE(piib);
if (INTERFACE_IS_ACTIVE(pie)) piib->dwState |= IPSAMPLE_STATE_ACTIVE;
if (INTERFACE_IS_BOUND(pie)) piib->dwState |= IPSAMPLE_STATE_BOUND;
piib->ulCount = pie->ulNumBindings; CopyMemory((PVOID) piia, // address,mask pairs
(PVOID) pie->pbeBindingTable, // ditto
pie->ulNumBindings * sizeof(IPSAMPLE_IP_ADDRESS)); } while (FALSE);
RELEASE_READ_LOCK(&(g_ce.pneNetworkEntry)->rwlLock);
break; }
default: { dwErr = ERROR_INVALID_PARAMETER; break; } }
*pulOutputSize = sizeof(IPSAMPLE_MIB_GET_OUTPUT_DATA) + ulSizeNeeded;
LEAVE_SAMPLE_API();
return dwErr;}
|
