sendudpDlg.cpp
上传用户:lx6688
上传日期:2016-01-30
资源大小:46k
文件大小:7k
源码类别:

网络截获/分析

开发平台:

Visual C++

  1. // sendudpDlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "sendudp.h"
  5. #include "sendudpDlg.h"
  6. #include "TypeChange.h"
  7. #ifdef _DEBUG
  8. #define new DEBUG_NEW
  9. #undef THIS_FILE
  10. static char THIS_FILE[] = __FILE__;
  11. #endif
  12. /////////////////////////////////////////////////////////////////////////////
  13. // CAboutDlg dialog used for App About
  14. class CAboutDlg : public CDialog
  15. {
  16. public:
  17. CAboutDlg();
  18. // Dialog Data
  19. //{{AFX_DATA(CAboutDlg)
  20. enum { IDD = IDD_ABOUTBOX };
  21. //}}AFX_DATA
  22. // ClassWizard generated virtual function overrides
  23. //{{AFX_VIRTUAL(CAboutDlg)
  24. protected:
  25. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  26. //}}AFX_VIRTUAL
  27. // Implementation
  28. protected:
  29. //{{AFX_MSG(CAboutDlg)
  30. //}}AFX_MSG
  31. DECLARE_MESSAGE_MAP()
  32. };
  33. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  34. {
  35. //{{AFX_DATA_INIT(CAboutDlg)
  36. //}}AFX_DATA_INIT
  37. }
  38. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  39. {
  40. CDialog::DoDataExchange(pDX);
  41. //{{AFX_DATA_MAP(CAboutDlg)
  42. //}}AFX_DATA_MAP
  43. }
  44. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  45. //{{AFX_MSG_MAP(CAboutDlg)
  46. // No message handlers
  47. //}}AFX_MSG_MAP
  48. END_MESSAGE_MAP()
  49. /////////////////////////////////////////////////////////////////////////////
  50. // CSendudpDlg dialog
  51. CSendudpDlg::CSendudpDlg(CWnd* pParent /*=NULL*/)
  52. : CDialog(CSendudpDlg::IDD, pParent)
  53. {
  54. //{{AFX_DATA_INIT(CSendudpDlg)
  55. m_portdest = 0;
  56. m_portsrc = 0;
  57. m_hexSend = FALSE;
  58. //}}AFX_DATA_INIT
  59. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  60. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  61. isListen = false;
  62. }
  63. void CSendudpDlg::DoDataExchange(CDataExchange* pDX)
  64. {
  65. CDialog::DoDataExchange(pDX);
  66. //{{AFX_DATA_MAP(CSendudpDlg)
  67. DDX_Control(pDX, IDC_EDIT4, m_rcvTxt);
  68. DDX_Control(pDX, IDC_EDIT3, m_data);
  69. DDX_Control(pDX, IDC_IPADDRESS1, m_IP);
  70. DDX_Text(pDX, IDC_EDIT2, m_portdest);
  71. DDX_Text(pDX, IDC_EDIT1, m_portsrc);
  72. DDX_Check(pDX, IDC_CHECK1, m_hexSend);
  73. //}}AFX_DATA_MAP
  74. }
  75. BEGIN_MESSAGE_MAP(CSendudpDlg, CDialog)
  76. //{{AFX_MSG_MAP(CSendudpDlg)
  77. ON_WM_SYSCOMMAND()
  78. ON_WM_PAINT()
  79. ON_WM_QUERYDRAGICON()
  80. ON_BN_CLICKED(IDC_BUTTON1, OnButton1)
  81. ON_BN_CLICKED(IDC_BUTTON2, OnButton2)
  82. ON_BN_CLICKED(IDC_BUTTON3, OnBtnClear)
  83. //}}AFX_MSG_MAP
  84. END_MESSAGE_MAP()
  85. /////////////////////////////////////////////////////////////////////////////
  86. // CSendudpDlg message handlers
  87. BOOL CSendudpDlg::OnInitDialog()
  88. {
  89. CDialog::OnInitDialog();
  90. // Add "About..." menu item to system menu.
  91. // IDM_ABOUTBOX must be in the system command range.
  92. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  93. ASSERT(IDM_ABOUTBOX < 0xF000);
  94. CMenu* pSysMenu = GetSystemMenu(FALSE);
  95. if (pSysMenu != NULL)
  96. {
  97. CString strAboutMenu;
  98. strAboutMenu.LoadString(IDS_ABOUTBOX);
  99. if (!strAboutMenu.IsEmpty())
  100. {
  101. pSysMenu->AppendMenu(MF_SEPARATOR);
  102. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  103. }
  104. }
  105. // Set the icon for this dialog.  The framework does this automatically
  106. //  when the application's main window is not a dialog
  107. SetIcon(m_hIcon, TRUE); // Set big icon
  108. SetIcon(m_hIcon, FALSE); // Set small icon
  109. m_serverIP ="";
  110. return TRUE;  // return TRUE  unless you set the focus to a control
  111. }
  112. void CSendudpDlg::OnSysCommand(UINT nID, LPARAM lParam)
  113. {
  114. if(nID == SC_CLOSE)
  115. CDialog::OnCancel();
  116. else
  117. {
  118. CDialog::OnSysCommand(nID, lParam);
  119. }
  120. }
  121. // If you add a minimize button to your dialog, you will need the code below
  122. //  to draw the icon.  For MFC applications using the document/view model,
  123. //  this is automatically done for you by the framework.
  124. void CSendudpDlg::OnPaint() 
  125. {
  126. if (IsIconic())
  127. {
  128. CPaintDC dc(this); // device context for painting
  129. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  130. // Center icon in client rectangle
  131. int cxIcon = GetSystemMetrics(SM_CXICON);
  132. int cyIcon = GetSystemMetrics(SM_CYICON);
  133. CRect rect;
  134. GetClientRect(&rect);
  135. int x = (rect.Width() - cxIcon + 1) / 2;
  136. int y = (rect.Height() - cyIcon + 1) / 2;
  137. // Draw the icon
  138. dc.DrawIcon(x, y, m_hIcon);
  139. }
  140. else
  141. {
  142. CDialog::OnPaint();
  143. }
  144. }
  145. // The system calls this to obtain the cursor to display while the user drags
  146. //  the minimized window.
  147. HCURSOR CSendudpDlg::OnQueryDragIcon()
  148. {
  149. return (HCURSOR) m_hIcon;
  150. }
  151. void CSendudpDlg::OnButton1() 
  152. {
  153. SOCKET sockClient=socket(AF_INET,SOCK_DGRAM,0);
  154. SOCKADDR_IN addSrv;
  155. CString str;
  156. CTypeChange myChang;
  157. char tmp[200];
  158. int len;
  159. m_IP.GetWindowText(m_serverIP);
  160. addSrv.sin_addr.S_un.S_addr = inet_addr(m_serverIP);
  161. addSrv.sin_family = AF_INET;
  162. addSrv.sin_port = htons(m_portdest);
  163. UpdateData(TRUE);
  164. m_data.GetWindowText(str);
  165. if(m_hexSend)
  166. {
  167. len = myChang.cstr2hex(str, tmp, 200);
  168. if(len>0)
  169. sendto(sockClient, tmp, len, 0, (SOCKADDR*)&addSrv, sizeof(SOCKADDR));
  170. }
  171. else
  172. {
  173. sendto(sockClient,str,str.GetLength(),0,(SOCKADDR*)&addSrv,sizeof(SOCKADDR));
  174. }
  175. closesocket(sockClient);
  176. }
  177. /*
  178. void AppendTxt(unsigned char *buf, int len, CString dTxt)
  179. {
  180. int i, j;
  181. char tmp[4];
  182. dTxt += "n";
  183. for(i=0; i<len; i++)
  184. {
  185. for(j=0; j<16; j++)
  186. {
  187. if(i*16+j>=len) break;
  188. sprintf(tmp, "%02x ", buf[i*16+j]);
  189. dTxt += tmp;
  190. }
  191. dTxt += "n";
  192. if(j<16) break;
  193. }
  194. }*/
  195. void AppendTxt(unsigned char *buf, int len, char *dTxt)
  196. {
  197. int i, j;
  198. char tmp[4];
  199. strcat(dTxt, "rn");
  200. for(i=0; i<len; i++)
  201. {
  202. strcat(dTxt, "    ");
  203. for(j=0; j<16; j++)
  204. {
  205. if(i*16+j>=len) break;
  206. sprintf(tmp, "%02X ", buf[i*16+j]);
  207. strcat(dTxt, tmp);
  208. }
  209. strcat(dTxt, "rn");
  210. if(j<16) break;
  211. }
  212. }
  213. char txtBox[5000];
  214. UINT recvThread(LPVOID lParam)
  215. {
  216. CSendudpDlg *pdlg = (CSendudpDlg *)lParam;
  217. unsigned char buf[500];
  218. int iDataLen;
  219. SOCKADDR_IN enet_receive_addr;
  220. SOCKADDR_IN enet_cur_addr;
  221. SOCKET enet_receive_socket;
  222. int    len = sizeof(enet_cur_addr);
  223. SYSTEMTIME st;
  224. char       tmp[20];
  225. enet_receive_addr.sin_family       = AF_INET;
  226. enet_receive_addr.sin_port         = htons(pdlg->m_portsrc);
  227. enet_receive_addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
  228. if ((enet_receive_socket=socket (AF_INET, SOCK_DGRAM, 0)) == INVALID_SOCKET ) 
  229.     { 
  230.         pdlg->MessageBox("cannot create enet_receive_socketn");  
  231. return -1;
  232. }
  233. if (bind (enet_receive_socket, (struct sockaddr *)&enet_receive_addr, sizeof(enet_receive_addr)) == SOCKET_ERROR)
  234. {
  235. pdlg->MessageBox ("Bind enet_receive_socket failedn");
  236. return -1;
  237. }
  238. txtBox[0] = 0;
  239. while(pdlg->isListen)
  240. {
  241. if((iDataLen = recvfrom(enet_receive_socket, (char *)buf, 500, 0,
  242. (struct sockaddr *)&enet_cur_addr,&len)) != ERROR)
  243. {
  244. GetLocalTime(&st);
  245. sprintf(tmp, "nrnr%2d:%2d:%2dnr", st.wHour,st.wMinute,st.wSecond);
  246. strcat(txtBox, tmp);
  247. AppendTxt(buf, iDataLen, txtBox);
  248. pdlg->m_rcvTxt.SetWindowText(txtBox);
  249. }
  250. }
  251. closesocket(enet_receive_socket);
  252. shutdown (enet_receive_socket, 0x00);
  253. return 0;
  254. }
  255. void CSendudpDlg::OnButton2() 
  256. {
  257. if (isListen) /* 正在侦听 */
  258. {
  259. isListen = false;
  260. GetDlgItem(IDC_BUTTON2)->SetWindowText("侦听");
  261. }
  262. else
  263. {
  264. UpdateData(TRUE);
  265. if (m_portsrc<100 || m_portsrc>9999)
  266. {
  267. MessageBox("端口越界", "error");
  268. return;
  269. }
  270. ::AfxBeginThread(recvThread, this);
  271. isListen = true;
  272. GetDlgItem(IDC_BUTTON2)->SetWindowText("停止");
  273. }
  274. }
  275. void CSendudpDlg::OnBtnClear() 
  276. {
  277. // TODO: Add your control notification handler code here
  278. txtBox[0] = 0;
  279. m_rcvTxt.SetWindowText(txtBox);
  280. }