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

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

SNMP编程

开发平台：

Visual C++

MIB.cpp：源码内容
  1. // MIB.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "SNMPApp.h"
  6. #include "snmp.h"
  7. #include "MIB.h"
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  15. void strPrintf(const char * pstr,int style=0,int color=0);
  16. enum {normal=0,bold=1,italic=2};
  17. enum {black=0,blue=0x00ff0000,green=0x0000ff00,red=0x000000ff};
  19. #define WM_ROUTERFINISH WM_USER+78
  21. CRouter::CRouter()
  22. {
  23. m_psnmp=NULL;
  24. }
  26. CRouter::~CRouter()
  27. {
  28. int i;
  29. for (i=0;i<m_ifTable.m_ifTableEntries.GetSize();i++)
  30. delete m_ifTable.m_ifTableEntries.GetAt(i);
  31. for (i=0;i<m_ipAddrTable.m_ipAddrTableEntries.GetSize();i++)
  32. delete m_ipAddrTable.m_ipAddrTableEntries.GetAt(i);
  33. for (i=0;i<m_ipRouteTable.m_ipRouteTableEntries.GetSize();i++)
  34. delete m_ipRouteTable.m_ipRouteTableEntries.GetAt(i);
  35. }
  37. BOOL CRouter::GetSystem(CSnmp* pSnmp)
  38. {
  39. m_psnmp=pSnmp;
  41. //internet(1.3.6.1).mgmt(2).mib2(1)
  42. m_CurrOid[0]="1.3.6.1.2.1.1.1.0";//system(1).sysDescr(1).0
  43. m_CurrOid[1]="1.3.6.1.2.1.1.2.0";//system(1).sysObjectID(2).0
  44. m_CurrOid[2]="1.3.6.1.2.1.1.5.0";//system(1).sysName(3).0
  46. m_psnmp->CreateVbl(NULL,NULL);
  47. for(int i=0;i<3;i++)
  48. m_psnmp->SetVbl(m_CurrOid[i],NULL);
  50. m_psnmp->CreatePdu(SNMP_PDU_GET,NULL,NULL,NULL);
  51. m_psnmp->Send((LPCTSTR)m_ipAddr);
  52. if (!m_psnmp->WaitForResponse())
  53. {
  54. m_sys.sysName="none";
  55. m_sys.sysDescr="Unkown";
  56. m_sys.sysObjectID=0;
  57. return FALSE;
  58. }
  59. m_psnmp->Receive();
  60. for (i=0;i<3;i++)
  61. {
  62. if(strcmp(m_psnmp->m_strOid[i],m_CurrOid[0])==0)
  63. m_sys.sysDescr=m_psnmp->m_strValue[i];
  64. else if (strcmp(m_psnmp->m_strOid[i],m_CurrOid[1])==0)
  65. m_sys.sysObjectID=atoi(m_psnmp->m_strValue[i]);
  66. else if (strcmp(m_psnmp->m_strOid[i],m_CurrOid[2])==0)
  67. m_sys.sysName=m_psnmp->m_strValue[i];
  68. }
  69. return TRUE;
  70. }
  72. void CRouter::GetifTable(CSnmp* pSnmp)
  73. {
  74. m_psnmp=pSnmp;
  75. ASSERT(m_psnmp->IsCreated());
  77. m_CurrOid[0]="1.3.6.1.2.1.2.1";
  78. m_psnmp->CreateVbl(NULL,NULL);
  79. m_psnmp->SetVbl(m_CurrOid[0],NULL);
  80. m_psnmp->CreatePdu(SNMP_PDU_GET,NULL,NULL,NULL);
  81. m_psnmp->Send((LPCTSTR)m_ipAddr);
  82. if (!m_psnmp->WaitForResponse())
  83. {
  84. AfxMessageBox("Get ifTable failure!");
  85. return;
  86. }
  87. m_psnmp->Receive();
  88. m_ifTable.ifNumber=atoi(m_psnmp->m_strValue[0]);
  90. //internet(1.3.6.1).mgmt(2).mib2(1).interfaces(2)
  91. CString firstStr="1.3.6.1.2.1.2.2.1.1";//ifIndex:index
  92. CString nextStr="1.3.6.1.2.1.2.2.1.1";//ifIndex
  93. do
  94. {
  95. if (!GetTableCol(nextStr,firstStr,pSnmp))
  96. break;
  97. CifTableEntry* pNewEntry=new CifTableEntry();
  98. pNewEntry->ifIndex=atoi(m_psnmp->m_strValue[0]);
  99. m_ifTable.m_ifTableEntries.Add(pNewEntry);;
  100. }while(1);
  102. m_CurrOid[0]="1.3.6.1.2.1.2.2.1.2";//ifDescr
  103. m_CurrOid[1]="1.3.6.1.2.1.2.2.1.3";//ifType
  104. m_CurrOid[2]="1.3.6.1.2.1.2.2.1.6";//ifPhysAddress
  105. char buff[10];
  106. for (int i=0;i<m_ifTable.m_ifTableEntries.GetSize();i++)
  107. {
  108. m_psnmp->CreateVbl(NULL,NULL);
  109. CifTableEntry* pEntry=
  110. (CifTableEntry*)(m_ifTable.m_ifTableEntries.GetAt(i));
  111. itoa(pEntry->ifIndex,buff,10);
  112. m_psnmp->SetVbl(m_CurrOid[0]+'.'+buff,NULL);
  113. m_psnmp->SetVbl(m_CurrOid[1]+'.'+buff,NULL);
  114. m_psnmp->CreatePdu(SNMP_PDU_GET,NULL,NULL,NULL);
  115. m_psnmp->Send((LPCTSTR)m_ipAddr);
  116. m_psnmp->WaitForResponse();
  117. m_psnmp->Receive();
  118. pEntry->ifDescr=m_psnmp->m_strValue[0];
  119. pEntry->ifType=atoi(m_psnmp->m_strValue[1]);
  120. pEntry->ifPhysAddress=atoi(m_psnmp->m_strValue[2]);
  121. }
  122. }
  124. void CRouter::GetipAddrTable(CSnmp* pSnmp)
  125. {
  126. m_psnmp=pSnmp;
  128. //internet(1.3.6.1).mgmt(2).mib2(1).
  129. //ip(4).ipAddrTable(20).ipAddrEntry(1)
  130. CString firstStr="1.3.6.1.2.1.4.20.1.1";//ipAdEntAddr:index
  131. CString nextStr="1.3.6.1.2.1.4.20.1.1";
  132. do
  133. {
  134. if (!GetTableCol(nextStr,firstStr,pSnmp))
  135. break;
  136. CipAddrTableEntry* pNewEntry=new CipAddrTableEntry();
  137. pNewEntry->ipAdEntAddr=m_psnmp->m_strValue[0];
  138. m_ipAddrTable.m_ipAddrTableEntries.Add(pNewEntry);
  139. }while(1);
  141. m_CurrOid[0]="1.3.6.1.2.1.4.20.1.2";//ipAdEntIfIndex
  142. m_CurrOid[1]="1.3.6.1.2.1.4.20.1.3";//ipAdEntNetMask
  143. for (int i=0;i<m_ipAddrTable.m_ipAddrTableEntries.GetSize();i++)
  144. {
  145. m_psnmp->CreateVbl(NULL,NULL);
  146. CipAddrTableEntry* pEntry=
  147. (CipAddrTableEntry*)(m_ipAddrTable.m_ipAddrTableEntries.GetAt(i));
  148. m_psnmp->SetVbl(m_CurrOid[0]+'.'+pEntry->ipAdEntAddr,NULL);
  149. m_psnmp->SetVbl(m_CurrOid[1]+'.'+pEntry->ipAdEntAddr,NULL);
  150. m_psnmp->CreatePdu(SNMP_PDU_GET,NULL,NULL,NULL);
  151. m_psnmp->Send((LPCTSTR)m_ipAddr);
  152. m_psnmp->WaitForResponse();
  153. m_psnmp->Receive();
  154. pEntry->ipAdEntIfIndex=atoi(m_psnmp->m_strValue[0]);
  155. pEntry->ipAdEntNetMask=m_psnmp->m_strValue[1];
  156. }
  157. }
  159. void CRouter::GetipRouteTable(CSnmp* pSnmp)
  160. {
  161. m_psnmp=pSnmp;
  162. ASSERT(m_psnmp->IsCreated());
  164. //internet(1.3.6.1).mgmt(2).mib2(1).
  165. //ip(4).ipRouteTable(21).ipRouteEntry(1)
  166. CString firstStr="1.3.6.1.2.1.4.21.1.1";//ipRouteDest:index
  167. CString nextStr="1.3.6.1.2.1.4.21.1.1";
  168. do
  169. {
  170. if (!GetTableCol(nextStr,firstStr,pSnmp))
  171. break;
  172. CipRouteTableEntry* pNewEntry=new CipRouteTableEntry();
  173. pNewEntry->ipRouteDst=m_psnmp->m_strValue[0];
  174. m_ipRouteTable.m_ipRouteTableEntries.Add(pNewEntry);
  175. }while(1);
  177. m_CurrOid[0]="1.3.6.1.2.1.4.21.1.2";//ipRouteIfIndex
  178. m_CurrOid[1]="1.3.6.1.2.1.4.21.1.7";//ipRouteNextHop
  179. for (int i=0;i<m_ipRouteTable.m_ipRouteTableEntries.GetSize();i++)
  180. {
  181. m_psnmp->CreateVbl(NULL,NULL);
  182. CipRouteTableEntry* pEntry=
  183. (CipRouteTableEntry*)(m_ipRouteTable.m_ipRouteTableEntries.GetAt(i));
  184. m_psnmp->SetVbl(m_CurrOid[0]+'.'+pEntry->ipRouteDst,NULL);
  185. m_psnmp->SetVbl(m_CurrOid[1]+'.'+pEntry->ipRouteDst,NULL);
  186. m_psnmp->CreatePdu(SNMP_PDU_GET,NULL,NULL,NULL);
  187. m_psnmp->Send((LPCTSTR)m_ipAddr);
  188. m_psnmp->WaitForResponse();
  189. m_psnmp->Receive();
  190. pEntry->ipRouteIfIndex=atoi(m_psnmp->m_strValue[0]);
  191. pEntry->ipRouteNextHop=m_psnmp->m_strValue[1];
  192. }
  193. }
  195. BOOL CRouter::GetRouterInfo(CSnmp* pSnmp)
  196. {
  197. ASSERT(pSnmp);
  198. m_psnmp=pSnmp;
  199. if (!GetSystem(pSnmp))
  200. return FALSE;
  201. GetifTable(pSnmp);
  202. GetipAddrTable(pSnmp);
  203. GetipRouteTable(pSnmp);
  204. m_psnmp=NULL;
  205. return TRUE;
  206. }
  208. BOOL CRouter::GetTableCol(CString &nStr, CString &fStr,CSnmp* psnmp)
  209. {
  210. m_psnmp=psnmp;
  211. ASSERT(m_psnmp->IsCreated());
  212. m_psnmp->CreateVbl(NULL,NULL);
  213. m_psnmp->SetVbl(nStr,NULL);
  215. m_psnmp->CreatePdu(SNMP_PDU_GETNEXT,NULL,NULL,NULL);
  216. m_psnmp->Send((LPCTSTR)m_ipAddr);
  217. if (!m_psnmp->WaitForResponse())
  218. return FALSE;
  219. m_psnmp->Receive();
  221. if(m_psnmp->PDU_type!=SNMP_PDU_RESPONSE)
  222. return FALSE;//接收到错误信息
  224. //我认为已经到达表的最后一行，任务完成
  225. if (m_psnmp->error_status==SNMP_ERROR_NOSUCHNAME)
  226. return FALSE;
  228. int len=fStr.GetLength();
  229. CString str=m_psnmp->m_strOid[0].Left(len);
  230. if (fStr.CompareNoCase((LPCTSTR)str)!=0)
  231. return FALSE;
  233. nStr=m_psnmp->m_strOid[0];
  234. return TRUE;
  235. }
  237. void CRouter::PrintRouterInfo()
  238. {
  239. CString temp;
  240. strPrintf("///The information of router:",italic,0x00dd00ff);
  241. strPrintf(m_ipAddr,italic,blue);
  242. strPrintf("////n",italic,0x00dd00ff);
  243. strPrintf("Router name:",bold,blue);
  244. strPrintf(m_sys.sysName);
  245. strPrintf("nRouter ObjectID:",bold,blue);
  246. temp.Format("%dn",m_sys.sysObjectID);
  247. strPrintf(temp);
  248. strPrintf("Router descriptor:",bold,blue);
  249. temp.Format(":n%sn",m_sys.sysDescr);
  250. strPrintf(temp);
  251. strPrintf("Router interface Ips:n   ",bold,blue);
  252. for (int i=0;i<m_ipAddrTable.m_ipAddrTableEntries.GetSize();i++)
  253. {
  254. CipAddrTableEntry* pEntry=
  255. (CipAddrTableEntry*)m_ipAddrTable.m_ipAddrTableEntries.GetAt(i);
  256. strPrintf(pEntry->ipAdEntAddr);
  257. strPrintf("t");
  258. }
  260. strPrintf("n///Router Information end.///nn",italic,0x00dd00ff);
  261. }