www.pudn.com > SnmpExpriment1.rar > Snmp.cpp

// Snmp1.cpp: implementation of the CSnmp class. 
// 
////////////////////////////////////////////////////////////////////// 
 
#include "stdafx.h" 
#include "snmp.h" 
#include "SnmpExpriment.h" 
 
#ifdef _DEBUG 
//CMemoryState oldMemState,newMemState,diffMemState; 
#undef THIS_FILE 
static char THIS_FILE[]=__FILE__; 
#define new DEBUG_NEW 
#endif 
 
 
 
////////////////////////////////////////////////////////////////////// 
// Construction/Destruction 
////////////////////////////////////////////////////////////////////// 
 
CSnmp::CSnmp() 
{ 
	nMajorVersion=new unsigned long; 
	nMinorVersion=new unsigned long; 
	nLevel=new unsigned long; 
	nTranslateMode=new unsigned long; 
	nRetransmitMode=new unsigned long; 
 
	if(SnmpStartup(nMajorVersion,nMinorVersion,nLevel,nTranslateMode, 
		nRetransmitMode)!=SNMPAPI_SUCCESS) 
	{//dwErr=SnmpGetLastError(NULL); 
		AfxMessageBox("initilization failure"); 
	} 
 
	if(SnmpSetTranslateMode(SNMPAPI_UNTRANSLATED_V1)!=SNMPAPI_SUCCESS) 
		AfxMessageBox("SetTranslateMode failure"); 
 
	if(SnmpSetRetransmitMode(SNMPAPI_ON)!=SNMPAPI_SUCCESS) 
		AfxMessageBox("SetRetransmitMode failure"); 
 
	sessionID=FALSE; 
} 
 
CSnmp::~CSnmp() 
{ 
	if(nRetransmitMode!=NULL) 
		delete nRetransmitMode; 
 
	if(nTranslateMode!=NULL) 
		delete nTranslateMode; 
 
	if(nLevel!=NULL) 
		delete nLevel; 
 
	if(nMinorVersion!=NULL) 
		delete nMinorVersion; 
 
	if(nMajorVersion!=NULL) 
		delete nMajorVersion; 
 
	if(session!=NULL) 
		SnmpClose(session); 
 
	SnmpCleanup();	 
} 
 
CSnmp::CloseSnmp() 
{ 
	if(nRetransmitMode!=NULL) 
		delete nRetransmitMode; 
	 
	if(nTranslateMode!=NULL) 
		delete nTranslateMode; 
	 
	if(nLevel!=NULL) 
		delete nLevel; 
	 
	if(nMinorVersion!=NULL) 
		delete nMinorVersion; 
	 
	if(nMajorVersion!=NULL) 
		delete nMajorVersion; 
	 
	if(session!=NULL) 
		SnmpClose(session); 
	 
	SnmpCleanup(); 
} 
 
CSnmp::CreateSession(HWND hWnd,UINT wMsg) 
{ 
	if((session=SnmpCreateSession(hWnd,wMsg,NULL,NULL))==SNMPAPI_FAILURE) 
		AfxMessageBox("CreateSession failure"); 
} 
 
CSnmp::CreateVbl(LPCSTR name,smiLPVALUE pvalue) 
{ 
	smiLPOID pOid=new smiOID; 
 
	if(SnmpStrToOid(name,pOid)==SNMPAPI_FAILURE) 
	{ 
		AfxMessageBox("CreateVbl:converts the dotted numeric string format failure!"); 
	//dwErr=SnmpGetLastError(session); 
	} 
	m_hvbl=SnmpCreateVbl(session,pOid,pvalue); 
 
	if(m_hvbl==SNMPAPI_FAILURE) 
	{ 
		AfxMessageBox("CreateVbl failure"); 
	//dwErr=SnmpGetLastError(session); 
	} 
 
	if(pOid!=NULL) 
		delete pOid; 
} 
 
CSnmp::SetVbl(LPCSTR name) 
{ 
	smiLPOID pOid=new smiOID; 
 
	if(SnmpStrToOid(name,pOid)==SNMPAPI_FAILURE) 
	{ 
		AfxMessageBox("SetVbl failure--oid"); 
	//dwErr=SnmpGetLastError(session); 
	} 
	 
	if(SnmpSetVb(m_hvbl,0,pOid,NULL)==SNMPAPI_FAILURE) 
	{ 
		AfxMessageBox("SetVbl failure"); 
	//dwErr=SnmpGetLastError(session); 
	} 
 
	if(pOid!=NULL) 
		delete pOid; 
} 
 
 
CSnmp::CreatePdu(smiINT PDU_type,smiINT32 request_id, 
				 smiINT error_status,smiINT error_index) 
{ 
	m_hpdu=SnmpCreatePdu(session,PDU_type,NULL,error_status,error_index,m_hvbl); 
	 
	if(m_hpdu==SNMPAPI_FAILURE) 
	{ 
		AfxMessageBox("CreatePdu failure"); 
		//dwErr=SnmpGetLastError(session); 
	} 
	else if (error_status > 0) 
	{ 
		AfxMessageBox("Error: error_status=%d, error_index=%d\n", 
			error_status, error_index); 
	} 
} 
 
CSnmp::Send(LPCSTR address,const char * community) 
{ 
	//	oldMemState.Checkpoint(); 
	HSNMP_ENTITY hAgent; 
 
	if((hAgent=SnmpStrToEntity(session,address))==SNMPAPI_FAILURE) 
		AfxMessageBox("SendMsg failure--entity"); 
 
	smiOCTETS contextName; 
	contextName.ptr=(unsigned char *)community; 
	contextName.len=lstrlen(community); 
	HSNMP_CONTEXT hView; 
 
	if((hView=SnmpStrToContext(session,&contextName))==SNMPAPI_FAILURE) 
		AfxMessageBox("SendMsg failure--context"); 
 
	if(SnmpSendMsg(session,NULL,hAgent,hView,m_hpdu)==SNMPAPI_FAILURE) 
	{ 
		AfxMessageBox("SendMsg failure"); 
		//dwErr=SnmpGetLastError(session); 
		CString str; 
		str.Format("%d",SnmpGetLastError(session)); 
		AfxMessageBox(str); 
	} 
	 
	if(m_hpdu!=NULL) 
		SnmpFreePdu(m_hpdu); 
	if(m_hvbl!=NULL) 
		SnmpFreeVbl(m_hvbl); 
	if(hAgent!=NULL) 
		SnmpFreeEntity(hAgent); 
	if(hView!=NULL) 
		SnmpFreeContext(hView); 
		/*#ifdef _DEBUG 
		newMemState.Checkpoint(); 
		if(diffMemState.Difference(oldMemState,newMemState)) 
		{TRACE("diffrence between fire and noe\n\n"); 
		diffMemState.DumpStatistics(); 
		} 
#endif*/ 
} 
 
CSnmp::Register() 
{ 
/*	if(SnmpRegister(NULL,NULL,NULL,NULL,NULL,SNMPAPI_ON)==SNMPAPI_FAILURE) 
{	AfxMessageBox("Register failure"); 
if((dwErr=SnmpGetLastError(session))=SNMPAPI_SUCCESS) 
AfxMessageBox("strange"); 
CString str; 
str.Format("%d",dwErr); 
AfxMessageBox(str); 
}*/ 
} 
 
CSnmp::Receive(LPTSTR *name,smiLPVALUE *value) 
{	 
	HSNMP_ENTITY srcEntity;  
	HSNMP_ENTITY dstEntity;  
	HSNMP_CONTEXT context; 
	HSNMP_PDU pPdu; 
	smiLPINT PDU_type=new smiINT;  
	smiLPINT32 request_id=new smiINT32; 
	smiLPINT error_status=new smiINT; 
	smiLPINT error_index=new smiINT; 
	HSNMP_VBL varbindlist; 
	smiLPOID pOid=new smiOID; 
	 
	if(SnmpRecvMsg(session,&srcEntity,&dstEntity,&context,&pPdu)!=SNMPAPI_SUCCESS) 
		AfxMessageBox("receive failure--recv"); 
	 
	if(SnmpGetPduData(pPdu,PDU_type,request_id,error_status,error_index,&varbindlist)!=SNMPAPI_SUCCESS) 
	{ 
		AfxMessageBox("receive failure--getpdu"); 
		CString str; 
		str.Format("%d",SnmpGetLastError(NULL)); 
		AfxMessageBox(str); 
	} 
	 
	if((nCount=SnmpCountVbl(varbindlist))==SNMPAPI_FAILURE) 
		AfxMessageBox("Count Vbl Error"); 
	 
	for(int i=1;i<=nCount;i++) 
	{ 
		if(SnmpGetVb(varbindlist,i,pOid,value[i])!=SNMPAPI_SUCCESS) 
		{ 
			AfxMessageBox("receive failure--getvb"); 
			CString str; 
			str.Format("%d",SnmpGetLastError(NULL)); 
			AfxMessageBox(str); 
		} 
		if(SnmpOidToStr(pOid,100,name[i])==SNMPAPI_FAILURE) 
		{ 
			AfxMessageBox("Get Vb Error"); 
			CString str; 
			str.Format("%d",SnmpGetLastError(NULL)); 
			AfxMessageBox(str); 
		} 
	} 
	 
	SnmpFreeEntity(srcEntity); 
	SnmpFreeEntity(dstEntity); 
	SnmpFreeContext(context); 
	SnmpFreePdu(pPdu); 
	SnmpFreeVbl(varbindlist); 
	//SnmpFreeDescriptor(SNMP_SYNTAX_OID,*pOid); 
	delete PDU_type;  
	delete request_id; 
	delete error_status; 
	delete error_index; 
	delete pOid; 
}