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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > SNMP编程 > 查看源码
Snmp.cpp
资源名称：TopologyBasedOnVC.rar　[点击查看]
上传用户：yfy060102
上传日期：2021-05-22
资源大小：60k
文件大小：9k
源码类别：

SNMP编程

开发平台：

Visual C++

Snmp.cpp：源码内容
  1. // Snmp1.cpp : implementation file
  3. #include "stdafx.h"
  4. #include "Snmp.h"
  5. #include "Winsock2.h"
  7. #ifdef _DEBUG
  8. #define new DEBUG_NEW
  9. #undef THIS_FILE
  10. static char THIS_FILE[] = __FILE__;
  11. #endif
  13. CSnmp::CSnmp()
  14. {
  15.   if(SnmpStartup(&nMajorVersion,&nMinorVersion,&nLevel,&nTranslateMode,
  16.   &nRetransmitMode)!=SNMPAPI_SUCCESS)
  17.  AfxMessageBox("initilization failure");
  19. if(SnmpSetTranslateMode(SNMPAPI_UNTRANSLATED_V1)!=SNMPAPI_SUCCESS)
  20. AfxMessageBox("SetTranslateMode failure");
  22. if(SnmpSetRetransmitMode(SNMPAPI_ON)!=SNMPAPI_SUCCESS)
  23. AfxMessageBox("SetRetransmitMode failure");
  25. m_hSession=NULL;
  26. m_hvbl=NULL;
  28. state=0;
  29. }
  31. CSnmp::~CSnmp()
  32. {
  33. if(m_hSession!=NULL)
  34. SnmpClose(m_hSession);
  35. SnmpCleanup();
  36. }
  38. BOOL CSnmp::CreateSession()
  39. {
  40. if (m_hSession)
  41. {
  42. AfxMessageBox("m_hSession has been created!");
  43. return TRUE;
  44. }
  46. if((m_hSession=SnmpCreateSession(NULL,0,SnmpCallback, (LPVOID)this))==SNMPAPI_FAILURE)
  47. {
  48. AfxMessageBox("Createm_hSession failure");
  49. return FALSE;
  50. };
  52. return TRUE;
  53. }
  55. //Callback function for snmp notify
  56. SNMPAPI_STATUS CALLBACK CSnmp::SnmpCallback(
  57. HSNMP_SESSION hSession,  // handle to the WinSNMP m_hSession
  58. HWND hWnd,               // handle to the notification window
  59. UINT wMsg,               // window notification message number
  60. WPARAM wParam,           // type of notification
  61. LPARAM lParam,           // request identifier of PDU
  62. LPVOID lpClientData)     // optional application-defined data
  63. {
  64. if (wParam==SNMPAPI_TL_TIMEOUT)
  65. ((CSnmp *)lpClientData)->state=1;
  67. ((CSnmp *)lpClientData)->m_Event.SetEvent();
  68. return SNMPAPI_SUCCESS;
  69. }
  71. BOOL CSnmp::CreateVbl(LPCSTR name,smiLPVALUE pvalue)
  72. {
  73. BOOL bRet=FALSE;
  74. smiLPOID pOid=NULL;
  75. if (name!=NULL)
  76. {
  77. pOid=new smiOID;
  78. if(SnmpStrToOid(name,pOid)==SNMPAPI_FAILURE)
  79. {
  80. AfxMessageBox("CreateVbl failure");
  81. bRet=FALSE;
  82. goto out;
  83. };
  84. };
  85. m_hvbl=SnmpCreateVbl(m_hSession,pOid,pvalue);
  86. if(m_hvbl==SNMPAPI_FAILURE)
  87. {
  88. AfxMessageBox("CreateVbl failure");
  89. bRet=FALSE;
  90. goto out;
  91. }
  93. out:
  94. //dwErr=SnmpGetLastError(m_hSession);
  95. if(pOid!=NULL)
  96. delete pOid;
  97. return bRet;
  98. }
  100. BOOL CSnmp::SetVbl(LPCSTR name,smiLPCVALUE value)
  101. {
  102. BOOL bRet=TRUE;
  103. smiLPOID pOid=new smiOID;
  104. if(SnmpStrToOid(name,pOid)==SNMPAPI_FAILURE)
  105. {
  106. AfxMessageBox("SetVbl failure--oid");
  107. //dwErr=SnmpGetLastError(m_hSession);
  108. bRet=FALSE;
  109. }
  110. if(SnmpSetVb(m_hvbl,0,pOid,NULL)==SNMPAPI_FAILURE)
  111. {
  112. AfxMessageBox("SetVbl failure");
  113. //dwErr=SnmpGetLastError(m_hSession);
  114. bRet=FALSE;
  115. }
  117. if(pOid!=NULL)
  118. delete pOid;
  120. return bRet;
  121. }
  123. BOOL CSnmp::CreatePdu(smiINT PDU_type,smiINT32 request_id,
  124.  smiINT error_status,smiINT error_index)
  125. {
  126. m_hpdu=SnmpCreatePdu(m_hSession,PDU_type,NULL,error_status,error_index,m_hvbl);
  127. if(m_hpdu==SNMPAPI_FAILURE)
  128. {
  129. AfxMessageBox("CreatePdu failure");
  130. return FALSE;
  131. //dwErr=SnmpGetLastError(m_hSession);
  132. }
  133. else if (error_status > 0)
  134.     {
  135. AfxMessageBox("Error: error_status=%d, error_index=%dn",
  136.                        error_status, error_index);
  137. return FALSE;
  138. }
  139. return TRUE;
  140. }
  142. BOOL CSnmp::Send(LPCSTR address)
  143. {
  144. state=0;
  145. BOOL bRet=TRUE;
  146. HSNMP_ENTITY hAgent;
  147. if((hAgent=SnmpStrToEntity(m_hSession,address))==SNMPAPI_FAILURE)
  148. AfxMessageBox("SendMsg failure--entity");
  150. smiOCTETS contextName;
  151. contextName.ptr=(unsigned char *)(m_community.GetBuffer(30));
  152. contextName.len=lstrlen(m_community);
  154. HSNMP_CONTEXT hView;
  155. if((hView=SnmpStrToContext(m_hSession,&contextName))==SNMPAPI_FAILURE)
  156. {
  157. AfxMessageBox("SendMsg failure--context");
  158. bRet=FALSE;
  159. }
  161. if(SnmpSendMsg(m_hSession,NULL,hAgent,hView,m_hpdu)==SNMPAPI_FAILURE)
  162. {
  163. AfxMessageBox("SendMsg failure");
  164. //dwErr=SnmpGetLastError(m_hSession);
  165. CString str;
  166. str.Format("%d",SnmpGetLastError(m_hSession));
  167. AfxMessageBox(str);
  168. bRet=FALSE;
  169. }
  171. if(m_hpdu!=NULL)
  172. SnmpFreePdu(m_hpdu);
  173. if(m_hvbl!=NULL)
  174. SnmpFreeVbl(m_hvbl);
  175. if(hAgent!=NULL)
  176. SnmpFreeEntity(hAgent);
  177. if(hView!=NULL)
  178. SnmpFreeContext(hView);
  180. return bRet;
  181. }
  183. BOOL CSnmp::Register()
  184. {
  185. /* if(SnmpRegister(NULL,NULL,NULL,NULL,NULL,SNMPAPI_ON)==SNMPAPI_FAILURE)
  186. {
  187. AfxMessageBox("Register failure");
  188. if((dwErr=SnmpGetLastError(m_hSession))=SNMPAPI_SUCCESS)
  189. AfxMessageBox("strange");
  190. CString str;
  191. str.Format("%d",dwErr);
  192. AfxMessageBox(str);
  193. }*/
  194. return TRUE;
  195. }
  197. BOOL CSnmp::Receive()
  198. {
  199. if (!Receive(m_strOid,m_value))
  200. return FALSE;
  202. CString strIp,strTemp;
  204. for(int i=0;i<nCount;i++)
  205. {
  206. switch(m_value[i].syntax)
  207. {
  208. case SNMP_SYNTAX_INT: 
  209. // case SNMP_SYNTAX_INT32:
  210. smiINT sNumber;
  211. sNumber=m_value[i].value.sNumber;
  212. m_strValue[i].Format("d",sNumber);
  213. break;
  214. case SNMP_SYNTAX_UINT32:
  215. case SNMP_SYNTAX_CNTR32:
  216. case SNMP_SYNTAX_GAUGE32: 
  217. case SNMP_SYNTAX_TIMETICKS:
  218. smiUINT32 uNumber;
  219. uNumber=m_value[i].value.uNumber;
  220. m_strValue[i].Format("d",uNumber);
  221. break;
  222. case SNMP_SYNTAX_CNTR64:
  223. smiCNTR64 hNumber;
  224. hNumber=m_value[i].value.hNumber;
  225. // m_strValue[i].Format("d",hNumber.hipart*4294967296+hNumber.lopart);
  226. break;
  227. case SNMP_SYNTAX_OCTETS: 
  228. case SNMP_SYNTAX_OPAQUE: 
  229. case SNMP_SYNTAX_NSAPADDR:
  230. m_strValue[i]=(char *)m_value[i].value.string.ptr;
  231. break;
  232. case SNMP_SYNTAX_IPADDR:
  233. in_addr ipAddr;
  234. unsigned long iAddr;
  235. iAddr=*((unsigned long *)(m_value[i].value.string.ptr));
  236. ipAddr.S_un.S_addr=iAddr;
  237. m_strValue[i]=inet_ntoa(ipAddr);
  239. /* ipAddr.S_un.S_addr=ntohs(((unsigned long *)(m_value[i]->value.string.ptr)));
  240. strIp.Format("%d",*m_value[i].value.string.ptr);
  241. strIp+=".";
  242. strTemp.Format("%d",*(m_value[i].value.string.ptr+1));
  243. strIp+=strTemp;
  244. strIp+=".";
  245. strTemp.Format("%d",*(m_value[i].value.string.ptr+2));
  246. strIp+=strTemp;
  247. strIp+=".";
  248. strTemp.Format("%d",*(m_value[i].value.string.ptr+3));
  249. strIp+=strTemp;
  250. m_strValue[i]=strIp;//*/
  251. break;
  252. case SNMP_SYNTAX_OID:
  253. smiOID oid=m_value[i].value.oid;
  254. char str[100];
  255. SnmpOidToStr(&oid,30,str);
  256. m_strValue[i]=str;
  257. //SnmpFreeDescriptor (SNMP_SYNTAX_OID, (smiLPOPAQUE)&oid);
  258. break;
  259. case SNMP_SYNTAX_NULL: 
  260. case SNMP_SYNTAX_NOSUCHOBJECT:
  261. case SNMP_SYNTAX_NOSUCHINSTANCE: 
  262. case SNMP_SYNTAX_ENDOFMIBVIEW:
  263. smiBYTE empty;
  264. empty=m_value[i].value.empty;
  265. m_strValue[i]="No Value Returned";
  266. break;
  267. default:
  268. break;
  269. };
  270. }
  272. return TRUE;
  273. }
  275. BOOL CSnmp::Receive(CString name[],smiVALUE value[])
  276. {
  277. if (state==1)
  278. {
  279. AfxMessageBox("Request TimeOut!");
  280. for (int i=0;i<10;i++)
  281. {
  282. name[i]="Unkown";
  283. value[i].syntax=SNMP_SYNTAX_NULL;
  284. }
  285. return FALSE;
  286. }
  287. BOOL bRet=TRUE;
  288. HSNMP_ENTITY srcEntity; 
  289. HSNMP_ENTITY dstEntity; 
  290. HSNMP_CONTEXT context;
  291. HSNMP_PDU pPdu;
  293. HSNMP_VBL varbindlist;
  294. smiOID Oid;
  296. if(SnmpRecvMsg(m_hSession,&srcEntity,&dstEntity,&context,&pPdu)!=SNMPAPI_SUCCESS)
  297. {
  298. AfxMessageBox("receive failure--recv");
  299. bRet=FALSE;
  300. }
  302. CHAR aa[100]; 
  303. SnmpEntityToStr(srcEntity,100,aa);
  304. strSrcEntity=aa;
  305. SnmpEntityToStr(dstEntity,100,aa);
  306. strDstEntity=aa;
  308. if(SnmpGetPduData(pPdu,&PDU_type,&request_id,&error_status,&error_index,&varbindlist)!=SNMPAPI_SUCCESS)
  309. {
  310. AfxMessageBox("receive failure--getpdu");
  311. CString str;
  312. str.Format("%d",SnmpGetLastError(NULL));
  313. AfxMessageBox(str);
  314. bRet=FALSE;
  315. }
  317. if (PDU_type!=SNMP_PDU_RESPONSE)
  318. {
  319. AfxMessageBox("Unknown package!");
  320. bRet=FALSE;
  321. }
  323. if((nCount=SnmpCountVbl(varbindlist))==SNMPAPI_FAILURE)
  324. {
  325. AfxMessageBox("Count Vbl Error");
  326. bRet=FALSE;
  327. }
  329. if (nCount>10)
  330. {
  331. AfxMessageBox("No buff,The number of parameters returned >10!");
  332. bRet=FALSE;
  333. }
  335. for(int i=0;i<nCount;i++)
  336. {
  337. if(SnmpGetVb(varbindlist,i+1,&Oid,&value[i])!=SNMPAPI_SUCCESS)
  338. {
  339. AfxMessageBox("receive failure--getvb");
  340. CString str;
  341. str.Format("%d",SnmpGetLastError(NULL));
  342. AfxMessageBox(str);
  343. bRet=FALSE;
  344. }
  346. char buff[100];
  347. if(SnmpOidToStr(&Oid,100,buff)==SNMPAPI_FAILURE)
  348. {
  349. AfxMessageBox("Get Vb Error");
  350. CString str;
  351. str.Format("%d",SnmpGetLastError(NULL));
  352. AfxMessageBox(str);
  353. bRet=FALSE;
  354. }
  355. name[i]=buff;
  356. }
  358. SnmpFreeEntity(srcEntity);
  359. SnmpFreeEntity(dstEntity);
  360. SnmpFreeContext(context);
  361. SnmpFreePdu(pPdu);
  362. SnmpFreeVbl(varbindlist);
  363. //SnmpFreeDescriptor(SNMP_SYNTAX_OID,&Oid);
  365. return bRet;
  366. }
  368. void CSnmp::ReportError(DWORD dwError)
  369. {
  370. return;
  371. }
  373. void CSnmp::ReInitial()
  374. {
  375. if(m_hpdu!=NULL)
  376. SnmpFreePdu(m_hpdu);
  377. if(m_hvbl!=NULL)
  378. SnmpFreeVbl(m_hvbl);
  379. }
  381. BOOL CSnmp::WaitForResponse()
  382. {
  383. WaitForSingleObject((HANDLE)m_Event, INFINITE);
  384. m_Event.ResetEvent();
  385. if (state==1)
  386. {
  387. state=0;
  388. return FALSE;
  389. }
  390. return TRUE;
  391. }
  393. BOOL CSnmp::IsCreated()
  394. {
  395. if (m_hSession)
  396. return TRUE;
  398. return FALSE;
  399. }