开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > SNMP编程 > 查看源码
SNMP1.cpp
资源名称：SNMP_source.rar　[点击查看]
上传用户：lvjun8202
上传日期：2013-04-30
资源大小：797k
文件大小：8k
源码类别：

SNMP编程

开发平台：

C/C++

SNMP1.cpp：源码内容
  1. // SNMP1.cpp: implementation of the CSNMP1 class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "oam.h"
  7. #include "SNMP1.h"
  11. #ifdef _DEBUG
  12. #undef THIS_FILE
  13. static char THIS_FILE[]=__FILE__;
  14. #define new DEBUG_NEW
  15. #endif
  17. //////////////////////////////////////////////////////////////////////
  18. // Construction/Destruction
  19. //////////////////////////////////////////////////////////////////////
  22. CSNMP1::CSNMP1()
  23. {
  24. nMajorVersion=new unsigned long;
  25. nMinorVersion=new unsigned long;
  26. nLevel=new unsigned long;
  27. nTranslateMode=new unsigned long;
  28. nRetransmitMode=new unsigned long;
  30. /* Init MIB var_table , the following tasks should be done:
  32. 1. Design the table structure
  33. 2. Design the table managment service routines
  34. 3. design the data access support routines for each mib var.
  36. In this table, the operations for each specific var 
  37. should be defined, and their function pointers should be stored here 
  38. */
  40.   if(SnmpStartup(nMajorVersion,nMinorVersion,nLevel,nTranslateMode,
  41. nRetransmitMode)!=SNMPAPI_SUCCESS) /*初始化*/
  42. {
  43. //dwErr=SnmpGetLastError(NULL);
  45. switch (SnmpGetLastError(NULL))
  46. {
  47. case SNMPAPI_NOT_INITIALIZED :
  48. AfxMessageBox("The SnmpStartup function did not complete successfully.");
  49. break;
  50. case SNMPAPI_ALLOC_ERROR:
  51. AfxMessageBox("An error occurred during memory allocation.");
  52. break;
  53. case SNMPAPI_OTHER_ERROR :
  54. AfxMessageBox("An unknown or undefined error occurred. ");
  55. break;
  56. default:
  57. break;
  58. }
  59. AfxMessageBox("initilization failure");
  60. }
  62. if(SnmpSetTranslateMode(SNMPAPI_UNTRANSLATED_V1)!=SNMPAPI_SUCCESS)
  63. //if(SnmpSetTranslateMode(SNMPAPI_TRANSLATED)!=SNMPAPI_SUCCESS) 
  64. AfxMessageBox("SetTranslateMode failure");
  65. if(SnmpSetRetransmitMode(SNMPAPI_ON)!=SNMPAPI_SUCCESS)/*设置重传机制*/
  66. AfxMessageBox("SetRetransmitMode failure");
  68. sessionID=FALSE; 
  71. }
  73. CSNMP1::~CSNMP1()
  74. {
  75. if(nRetransmitMode!=NULL)
  76. delete nRetransmitMode;
  77. if(nTranslateMode!=NULL)
  78. delete nTranslateMode;
  79. if(nLevel!=NULL)
  80. delete nLevel;
  81. if(nMinorVersion!=NULL)
  82. delete nMinorVersion;
  83. if(nMajorVersion!=NULL)
  84. delete nMajorVersion;
  85. if(session!=NULL)
  86. SnmpClose(session);
  87. SnmpCleanup();
  90. }
  92. CSNMP1::CreateSession(HWND hWnd,UINT wMsg)
  93. {
  94. if((session=SnmpCreateSession(hWnd,wMsg,NULL,NULL))==SNMPAPI_FAILURE)
  95. AfxMessageBox("CreateSession failure");
  96. }
  97. //ADD BY qIBING
  98. CSNMP1::CreateSession_Trap(HWND hWnd,UINT wMsg)
  99. {
  100. if((session_trap=SnmpCreateSession(hWnd,wMsg,NULL,NULL))==SNMPAPI_FAILURE)
  101. AfxMessageBox("CreateSession trap failure");
  102. }
  104. CSNMP1::CreateVbl(LPCSTR name,smiLPVALUE pvalue)
  105. {
  106. smiLPOID pOid=new smiOID;
  107. if(SnmpStrToOid(name,pOid)==SNMPAPI_FAILURE)
  108. {AfxMessageBox("CreateVbl failure");
  109. //dwErr=SnmpGetLastError(session);
  110. }
  111. m_hvbl=SnmpCreateVbl(session,pOid,pvalue);
  112. if(m_hvbl==SNMPAPI_FAILURE)
  113. {AfxMessageBox("CreateVbl failure");
  114. //dwErr=SnmpGetLastError(session);
  115. }
  116. if(pOid!=NULL)
  117. delete pOid;
  118.     
  119. }
  121. CSNMP1::SetVbl(LPCSTR name,smiLPVALUE pvalue)
  122. {
  123. smiLPOID pOid=new smiOID;
  125. //if(SnmpStrToOid(name,pOid)==SNMPAPI_FAILURE)
  126.     /******************qibing*/
  127. // SnmpMgrStrToOid(name,pOid);
  128. /*********************qibing*/
  129. if(SnmpStrToOid(name,pOid)==SNMPAPI_FAILURE)
  130. {AfxMessageBox("SetVbl failure--oid");
  131. /* { //dwErr=SnmpGetLastError(session);
  132. switch( dwErr)
  133. {
  134. case SNMPAPI_NOT_INITIALIZED:
  135. AfxMessageBox(" The SnmpStartup function did not complete successfully!");
  136. break;
  137. case SNMPAPI_ALLOC_ERROR:
  138. AfxMessgeBox("error in memory allocation");
  139. break;
  140. case SNMPAPI_OID_INVALID:
  141. AfxMessgeBox("invalid Oid ");
  142. break;
  143. case SNMPAPI_OTHER_ERROR :
  144. AfxMessgeBox( " other error");
  145. break;
  146. default:
  147. break
  148. }*/
  149. } 
  150. if(SnmpSetVb(m_hvbl,0,pOid,NULL)==SNMPAPI_FAILURE) /*0--1*/
  151. {AfxMessageBox("SetVbl failure");
  152. //dwErr=SnmpGetLastError(session);
  153. }
  154. if(pOid!=NULL)
  155. delete pOid;
  156. }
  159. CSNMP1::CreatePdu(smiINT PDU_type,smiINT32 request_id,
  160.  smiINT error_status,smiINT error_index)
  161. {
  162. m_hpdu=SnmpCreatePdu(session,PDU_type,NULL,error_status,error_index,m_hvbl);
  163. if(m_hpdu==SNMPAPI_FAILURE)
  164. {
  165. AfxMessageBox("CreatePdu failure");
  166. //dwErr=SnmpGetLastError(session);
  167. }
  168. else if (error_status > 0)/* 进对getBulk*/
  169.                 {
  170.                // AfxMessageBox("Error: error_status=%d, error_index=%dn",
  171.                //      error_status, error_index);
  172.             }
  173. }
  175. CSNMP1::Send(LPCSTR address,const char * community)
  176. {
  177. // oldMemState.Checkpoint();
  178. HSNMP_ENTITY hAgent;
  179. if((hAgent=SnmpStrToEntity(session,address))==SNMPAPI_FAILURE)
  180. AfxMessageBox("SendMsg failure--entity");
  181. smiOCTETS contextName;
  182. contextName.ptr=(unsigned char *)community;
  183. contextName.len=lstrlen(community);
  184. HSNMP_CONTEXT hView;
  185. if((hView=SnmpStrToContext(session,&contextName))==SNMPAPI_FAILURE)
  186. AfxMessageBox("SendMsg failure--context");
  189.    if(SnmpSendMsg(session,NULL,hAgent,hView,m_hpdu)==SNMPAPI_FAILURE)
  190. {
  191. AfxMessageBox("SendMsg failure");
  192. //dwErr=SnmpGetLastError(session);
  193. CString str;
  194. str.Format("%d",SnmpGetLastError(session));
  195. AfxMessageBox(str);
  196. }
  197. if(m_hpdu!=NULL)
  198. SnmpFreePdu(m_hpdu);
  199. if(m_hvbl!=NULL)
  200. SnmpFreeVbl(m_hvbl);
  201. if(hAgent!=NULL)
  202. SnmpFreeEntity(hAgent);
  203. if(hView!=NULL)
  204. SnmpFreeContext(hView);
  206. }
  208. CSNMP1::Register()
  209. {
  210. SNMPAPI_STATUS  dwErr;
  211. if(SnmpRegister(session,NULL,NULL,NULL,NULL,SNMPAPI_ON)==SNMPAPI_FAILURE)
  212. {
  213. AfxMessageBox("Register failure");
  214. if((dwErr=SnmpGetLastError(session_trap))==SNMPAPI_SUCCESS)
  215. AfxMessageBox("strange");
  216. CString str;
  217. str.Format("%d",dwErr);
  218. AfxMessageBox(str);
  219. }
  220. }
  223. int CSNMP1::Receive(LPTSTR *name,smiLPVALUE *value,smiLPINT PDU_type,char* pSrc)
  224. {
  225. HSNMP_ENTITY srcEntity; 
  226. HSNMP_ENTITY dstEntity; 
  227. HSNMP_CONTEXT context; // store for the COMMUNITY field
  228. HSNMP_PDU pPdu;
  229.    smiLPINT32 request_id=new smiINT32;
  230. smiLPINT error_status=new smiINT;
  231. smiLPINT error_index=new smiINT;
  232. HSNMP_VBL varbindlist;
  233. smiLPOID pOid=new smiOID;
  234. if(SnmpRecvMsg(session,&srcEntity,&dstEntity,&context,&pPdu)!=SNMPAPI_SUCCESS)
  235. {
  236. // CString str;
  237. // str.Format("%d",SnmpGetLastError(NULL));
  238. // AfxMessageBox(str);
  239. //AfxMessageBox("receive failure--recv");
  240. return 0;
  241. }
  243. if(SnmpGetPduData(pPdu,PDU_type,request_id,error_status,error_index,&varbindlist)!=SNMPAPI_SUCCESS)
  244. {
  245. /* AfxMessageBox("receive failure--getpdu");
  246. CString str;
  247. str.Format("%d",SnmpGetLastError(NULL));
  248. AfxMessageBox(str);*/
  249. return 0;
  251. }
  253. if((nCount=SnmpCountVbl(varbindlist))==SNMPAPI_FAILURE)
  254. //AfxMessageBox("Count Vbl Error");
  255. return 0;
  257. for(int i=1;i<=nCount;i++)
  258. {
  259. if(SnmpGetVb(varbindlist,i,pOid,value[i])!=SNMPAPI_SUCCESS)
  260. {
  261. /* AfxMessageBox("receive failure--getvb");
  262. CString str;
  263. str.Format("%d",SnmpGetLastError(NULL));
  264. AfxMessageBox(str);*/
  265. return 0;
  266. }
  267. if(SnmpOidToStr(pOid,100,name[i])==SNMPAPI_FAILURE)
  268. {  
  269. /*AfxMessageBox("Get Vb Error");
  270. CString str;
  271. str.Format("%d",SnmpGetLastError(NULL));
  272. AfxMessageBox(str);*/
  273. return 0;
  274. }
  276. }
  278. //SnmpOidToStr(pOid,100,name[i]); 
  279.   /* insert here the processing routines, such as :
  280. 1.Check whether the PDU_type is GetResponse PDU or Trap PDU  
  281.     2.if PDU_type is GetResponse PDU, verify its requestid.  else goto step 5
  282. 3.Record error_status, error_index, into the MIB-II snmp group 
  283.     4.Store requested data into database 
  284. 5. intriger Trap handle process.*/
  287. //if (*PDU_type==SNMP_PDU_TRAP )
  288. //{
  290.       SnmpEntityToStr(srcEntity,100,pSrc);
  292. /* smiOCTETS* a;
  293. SnmpContextToStr(context,a);
  294. CString test
  295. =a->ptr;
  296. AfxMessageBox(test);*/
  299.         
  301. //}
  305. SnmpFreeEntity(srcEntity);
  306. SnmpFreeEntity(dstEntity);
  307. SnmpFreeContext(context);
  308. SnmpFreePdu(pPdu);
  309. SnmpFreeVbl(varbindlist);
  310. //SnmpFreeDescriptor(SNMP_SYNTAX_OID,*pOid);
  311. // delete PDU_type; 
  312. delete request_id;
  313. delete error_status;
  314. delete error_index;
  315. delete pOid;
  316. return 1;
  318.  }