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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 进程与线程 > 查看源码
ScanLanComputerDlg.cpp
资源名称：ScanLanComputer.rar　[点击查看]
上传用户：tskangjie
上传日期：2022-07-13
资源大小：39k
文件大小：12k
源码类别：

进程与线程

开发平台：

Visual C++

ScanLanComputerDlg.cpp：源码内容
  1. // ScanLanComputerDlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "ScanLanComputer.h"
  5. #include "ScanLanComputerDlg.h"
  7. #ifdef _DEBUG
  8. #define new DEBUG_NEW
  9. #undef THIS_FILE
  10. static char THIS_FILE[] = __FILE__;
  11. #endif
  13. #define DESTPORT 137  //ScanLanComputer name port
  14. #define MYPORT 4321
  16. /////////////////////////////////////////////////////////////////////////////
  17. // CScanLanComputerDlg dialog
  18. CScanLanComputerDlg* pDlg;
  19. BYTE bs[50] = {0x0,0x00,0x0,0x10,0x0,0x1,0x0,0x0,0x0,0x0,0x0,0x0,0x20,0x43,0x4b,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x0,0x0,0x21,0x0,0x1};
  20. unsigned char B1[4],B2[4];
  21. HANDLE wait_handle;
  23. CString strOldIP = "";
  24. bool bExit = false;
  25. int i;
  27. //------------------------------------------------------------------
  28. CScanLanComputerDlg::CScanLanComputerDlg(CWnd* pParent /*=NULL*/)
  29. : CDialog(CScanLanComputerDlg::IDD, pParent)
  30. {
  31. //{{AFX_DATA_INIT(CScanLanComputerDlg)
  32. // NOTE: the ClassWizard will add member initialization here
  33. //}}AFX_DATA_INIT
  34. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  35. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  37. pDlg = this;
  38. if ( !m_UDPSocket.Create(MYPORT,SOCK_DGRAM) )
  39. {
  40. AfxMessageBox(_T("Failed to create socket!"));
  41. }
  42. }
  44. void CScanLanComputerDlg::DoDataExchange(CDataExchange* pDX)
  45. {
  46. CDialog::DoDataExchange(pDX);
  47. //{{AFX_DATA_MAP(CScanLanComputerDlg)
  48. DDX_Control(pDX, IDC_LIST_IP_ADDRESS, m_ListBoxIP);
  49. DDX_Control(pDX, IDC_IPADDRESS2, m_IPEdit2);
  50. DDX_Control(pDX, IDC_LIST_COMPUTER_INFO, m_ListCtrlComputerInfo);
  51. DDX_Control(pDX, IDC_IPADDRESS1, m_IPEdit1);
  52. //}}AFX_DATA_MAP
  53. }
  55. BEGIN_MESSAGE_MAP(CScanLanComputerDlg, CDialog)
  56. //{{AFX_MSG_MAP(CScanLanComputerDlg)
  57. ON_WM_PAINT()
  58. ON_WM_QUERYDRAGICON()
  59. ON_BN_CLICKED(IDC_BTN_SEND, OnBtnStart)
  60. ON_NOTIFY(IPN_FIELDCHANGED, IDC_IPADDRESS1, OnFieldchangedIpaddress1)
  61. ON_BN_CLICKED(IDC_BTN_EXIT, OnBtnExit)
  62. ON_BN_CLICKED(IDC_BTN_SAVE, OnBtnSave)
  63. ON_NOTIFY(NM_DBLCLK, IDC_LIST3, OnDblclkListView)
  64. ON_BN_CLICKED(IDC_BUTTON_CANCEL, OnButtonCancel)
  65. ON_BN_CLICKED(IDC_BUTTON_CLEAR, OnButtonClear)
  66. //}}AFX_MSG_MAP
  67. END_MESSAGE_MAP()
  69. /////////////////////////////////////////////////////////////////////////////
  70. // CScanLanComputerDlg message handlers
  72. BOOL CScanLanComputerDlg::OnInitDialog()
  73. {
  74. CDialog::OnInitDialog();
  76. // Set the icon for this dialog.  The framework does this automatically
  77. //  when the application's main window is not a dialog
  78. SetIcon(m_hIcon, TRUE); // Set big icon
  79. SetIcon(m_hIcon, FALSE); // Set small icon
  81. // Set IP Address segment.
  82. m_IPEdit1.SetAddress( 172, 25, 68, 1 );
  83. m_IPEdit2.SetAddress( 172, 25, 68,  255 );
  85. // Create event.
  86. wait_handle = CreateEvent( NULL,true,false,_T("receive data") );
  88. // Disable exit thread button.
  89. GetDlgItem( IDC_BTN_EXIT )->EnableWindow( FALSE );
  91. // Initialize list control. 
  92.     DWORD dwStyle = GetWindowLong( m_ListCtrlComputerInfo.GetSafeHwnd(), GWL_STYLE );
  93. dwStyle &= ~LVS_TYPEMASK;
  94. dwStyle |= LVS_REPORT;
  95. SetWindowLong( m_ListCtrlComputerInfo.GetSafeHwnd(), GWL_STYLE, dwStyle );
  97.     m_ListCtrlComputerInfo.InsertColumn( 0, _T("MAC地址"), LVCFMT_LEFT, 150 );
  98. m_ListCtrlComputerInfo.InsertColumn( 0, _T("用户"), LVCFMT_LEFT, 160 );
  99. m_ListCtrlComputerInfo.InsertColumn( 0, _T("主机"), LVCFMT_LEFT, 120 );
  100. m_ListCtrlComputerInfo.InsertColumn( 0, _T("工作组"), LVCFMT_LEFT, 120 );
  101. m_ListCtrlComputerInfo.InsertColumn( 0, _T("IP地址"), LVCFMT_LEFT, 100 );
  102.     
  103. m_ListCtrlComputerInfo.SetExtendedStyle( LVS_EX_GRIDLINES );
  104.     ::SendMessage( m_ListCtrlComputerInfo.m_hWnd, LVM_SETEXTENDEDLISTVIEWSTYLE,
  105.       LVS_EX_FULLROWSELECT, LVS_EX_FULLROWSELECT );
  107. return TRUE;  // return TRUE  unless you set the focus to a control
  108. }
  110. // If you add a minimize button to your dialog, you will need the code below
  111. //  to draw the icon.  For MFC applications using the document/view model,
  112. //  this is automatically done for you by the framework.
  114. void CScanLanComputerDlg::OnPaint() 
  115. {
  116. if (IsIconic())
  117. {
  118. CPaintDC dc(this); // device context for painting
  120. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  122. // Center icon in client rectangle
  123. int cxIcon = GetSystemMetrics(SM_CXICON);
  124. int cyIcon = GetSystemMetrics(SM_CYICON);
  125. CRect rect;
  126. GetClientRect(&rect);
  127. int x = (rect.Width() - cxIcon + 1) / 2;
  128. int y = (rect.Height() - cyIcon + 1) / 2;
  130. // Draw the icon
  131. dc.DrawIcon(x, y, m_hIcon);
  132. }
  133. else
  134. {
  135. CDialog::OnPaint();
  136. }
  137. }
  139. // The system calls this to obtain the cursor to display while the user drags
  140. //  the minimized window.
  141. HCURSOR CScanLanComputerDlg::OnQueryDragIcon()
  142. {
  143. return (HCURSOR) m_hIcon;
  144. }
  146. // ScanLanComputer线程
  147. UINT ScanLanComputerThread(LPVOID param)
  148. {
  149. do
  150. {
  151. if ( bExit )
  152. {
  153. AfxMessageBox(_T("暂时停止查找网络邻居!"));
  155. pDlg->GetDlgItem( IDC_BTN_SEND )->EnableWindow( TRUE );
  156. pDlg->GetDlgItem( IDC_IPADDRESS1 )->EnableWindow( TRUE );
  157. pDlg->GetDlgItem( IDC_IPADDRESS2 )->EnableWindow( TRUE );
  158. pDlg->GetDlgItem( IDC_BTN_EXIT )->EnableWindow( FALSE );
  159. bExit = false;
  161. return 1;
  162. }
  164. pDlg->m_strIP.Format( "%d.%d.%d.%d", B1[0], B1[1], B1[2], B1[3] );
  165. pDlg->m_ListBoxIP.InsertString( 0, pDlg->m_strIP );
  166. if ( B1[3] != 0 && B1[2] != 0 )
  167. pDlg->m_UDPSocket.SendTo( (void*)bs, 50, DESTPORT, pDlg->m_strIP, 0 );
  170. if ( B1[2] <= B2[2] )
  171. {
  172.    if( B1[3] < B2[3] ) B1[3]++;
  173.    else if( B1[2] < B2[2] && B1[3] < 255 )B1[3]++;
  174.    else if( B1[2] < B2[2] && B1[3] == 255 )
  175.    {
  176.    B1[3] = 0;
  177.    B1[2]++;
  178.    }
  179. }
  180. else break;
  181. if( B1[3] >= B2[3] && B1[2] >= B2[2] ) break;
  183. } while( B1[2] <= 255 && B1[3] <= 255 );
  185. pDlg->m_ListBoxIP.InsertString( 0,_T("-----查找网络邻居结束!-----"));
  187. pDlg->GetDlgItem( IDC_BTN_SEND )->EnableWindow( TRUE );
  188. pDlg->GetDlgItem( IDC_IPADDRESS1 )->EnableWindow( TRUE );
  189. pDlg->GetDlgItem( IDC_IPADDRESS2 )->EnableWindow( TRUE );
  190. pDlg->GetDlgItem( IDC_BTN_EXIT )->EnableWindow( FALSE );
  191. pDlg->GetDlgItem( IDC_BUTTON_CLEAR )->EnableWindow( TRUE );
  193. return 0;
  194. }
  196. void CScanLanComputerDlg::OnBtnStart() 
  197. {
  198. GetDlgItem( IDC_BUTTON_CLEAR )->EnableWindow( FALSE );
  200. m_IPEdit1.GetAddress( B1[0], B1[1], B1[2], B1[3] );
  201. m_IPEdit2.GetAddress( B2[0], B2[1], B2[2], B2[3] );
  203. // Check data.
  204. if( B2[2] < B1[2] )
  205. {
  206. AfxMessageBox( _T("终止地址应大于起始地址") ); 
  207. return;
  208. }
  209. else if( B2[2] == B1[2] && B2[3] < B1[3] )
  210. {
  211. AfxMessageBox( _T("终止地址应大于起始地址") ); 
  212. return;
  213. }
  214. else
  215. {
  216. // Do nothing.
  217. }
  219. if( B2[0] != B1[0] || B2[1] != B1[1] )
  220. {
  221. AfxMessageBox( _T("不支持A类或B类网") ); 
  222. return;
  223. }
  224.  
  225. GetDlgItem( IDC_BTN_SEND )->EnableWindow( false );
  226. GetDlgItem( IDC_IPADDRESS1 )->EnableWindow( false );
  227. GetDlgItem( IDC_IPADDRESS2 )->EnableWindow( false );
  228. GetDlgItem( IDC_BTN_EXIT )->EnableWindow( true );
  230. // 启动线程
  231. AfxBeginThread( ScanLanComputerThread, this->GetSafeHwnd(), THREAD_PRIORITY_NORMAL );
  233. }
  235. void CScanLanComputerDlg::OnReceive()
  236. {
  237. BYTE Buf[500];
  239. CString str,strIP,strHost,strHex,strMac,Host,Group,User;
  240. UINT dport;
  241. m_UDPSocket.ReceiveFrom( Buf,500,strIP,dport,0 );
  243. if( strIP == (char)NULL || strIP == strOldIP ) return;
  244. strOldIP = strIP;
  246. int index = m_ListCtrlComputerInfo.InsertItem( 0,strIP );
  248. strHost = _T("");
  249. strHex = _T("");
  250. User = _T("?");
  251. Host = _T("\");
  253. int tem = 0,num = 0;
  254. bool bAdd = true;
  256. for ( i = 57; i < 500; i++ ) 
  257. {
  258. if ( Buf[i] == 0xcc ) break; 
  260. if ( bAdd )
  261. {
  262. str.Format( _T("%c"),Buf[i] );
  263. if( Buf[i] >= ' ' ) strHost += str;
  265. str.Format( _T("%02X-"),Buf[i] );
  266. strHex += str;
  267. }
  269. if ( (++tem)%18 == 0 )
  270. {
  271.             bAdd = true; 
  272. strHost.TrimRight( (char)NULL );
  274. if ( strHost == _T("") )
  275. {
  276.     strMac.Delete( 17,strMac.GetLength() - 17 );
  277. m_ListCtrlComputerInfo.SetItem( index,4,LVIF_TEXT,strMac, 0, 0, 0,0 );
  278. break;
  279. }
  281. if ( num == 0 && strHost != _T("") )
  282. {
  283. m_ListCtrlComputerInfo.SetItem( index,2,LVIF_TEXT,strHost, 0, 0, 0,0 );
  284. Host = strHost;
  285. num++;
  286. }
  287. else
  288. {
  289. if( Host != strHost && num == 1 && strHost != _T("") )
  290. {
  291. m_ListCtrlComputerInfo.SetItem( index,1,LVIF_TEXT,strHost, 0, 0, 0,0 );
  292. Group = strHost;
  293.     num++;
  294. }
  295. else 
  296. {
  297. if( strHost != Host && strHost != Group && num == 2 && strHost != _T(""))
  298. {
  299. User = strHost;
  300. if( User != _T("__MSBROWSE__") )
  301. {
  302. m_ListCtrlComputerInfo.SetItem( index,3,LVIF_TEXT,User, 0, 0, 0,0 );
  303. num++;
  304. }
  305. }
  306. }
  307. }
  309. strMac = strHex;
  310. strHost = _T("");
  311. strHex = _T("");
  312. }
  313. }
  315. SetEvent( wait_handle );
  316. }
  318. void CScanLanComputerDlg::OnFieldchangedIpaddress1( NMHDR* pNMHDR, LRESULT* pResult ) 
  319. {
  320. unsigned char AddrB[4];
  321. m_IPEdit1.GetAddress( AddrB[0], AddrB[1], AddrB[2], AddrB[3] );
  322. m_IPEdit2.SetAddress( AddrB[0], AddrB[1], AddrB[2], 255 );
  324. *pResult = 0;
  325. }
  327. void CScanLanComputerDlg::OnBtnExit() 
  328. {
  329. GetDlgItem( IDC_BUTTON_CLEAR )->EnableWindow( TRUE );
  331. bExit = true;
  332. }
  334. CString MakeLen( CString csDst,int nLength )
  335. {
  336. if( csDst.GetLength() < nLength )
  337. {
  338. for( int i = csDst.GetLength(); i < nLength; i++ )
  339. {
  340. csDst += ' ';
  341. }
  342. }
  343. else 
  344. {
  345. csDst += ' ';
  346. }
  348. return csDst;
  349. }
  351. void CScanLanComputerDlg::OnBtnSave() 
  352. {
  353. int nCount = m_ListCtrlComputerInfo.GetItemCount();
  354. CFile f;
  355. CFileException e;
  356. CString csFileName = _T("");
  358. if ( 0 == m_ListCtrlComputerInfo.GetItemCount())
  359. {
  360. AfxMessageBox( _T("当前列表中没有内容,保存没有意义!"));
  361. return;
  362. }
  364. CTime t = CTime::GetCurrentTime();
  365. CString csTime = t.Format( _T("%Y年%m月%d日-%H时%M分%S秒"));
  366. csTime = _T("网络邻居_") + csTime;
  368. CFileDialog FileDlg( FALSE,_T("txt"),csTime,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,_T("txt"),NULL );
  369. if ( FileDlg.DoModal() == IDOK )
  370. {
  371. UpdateData( true );
  372. csFileName = FileDlg.GetFileName();
  373. UpdateData( false );
  375. if ( f.Open( csFileName, CFile::modeCreate | CFile::modeWrite, &e) )
  376. {
  378. CString csTitle = _T("");
  379. csTitle = _T("IP地址           工作组          主机            用户             MAC地址rn");
  380. csTitle += _T("-----------------------------------------------------------------------------------rn");
  381. f.Write( csTitle, csTitle.GetLength() );
  383. for ( int i = 0; i < nCount; i++ )
  384. {
  385. csTitle = MakeLen( m_ListCtrlComputerInfo.GetItemText(i,0),17 )
  386. +MakeLen( m_ListCtrlComputerInfo.GetItemText(i,1),14 )
  387.     +MakeLen( m_ListCtrlComputerInfo.GetItemText(i,2),14 )
  388. +MakeLen( m_ListCtrlComputerInfo.GetItemText(i,3),14 )
  389. +MakeLen( m_ListCtrlComputerInfo.GetItemText(i,4),17 );
  390. csTitle += "rnrn";
  391. f.Write( csTitle, csTitle.GetLength() );
  392. }
  393. f.Close();
  394. }
  395. }
  396. }
  398. void CScanLanComputerDlg::OnDblclkListView( NMHDR* pNMHDR, LRESULT* pResult ) 
  399. { 
  400. if ( 0 == m_ListCtrlComputerInfo.GetItemCount() )
  401. {
  402. AfxMessageBox(_T("没有主机显示!"));
  403. return;
  404. }
  406. POSITION pos = m_ListCtrlComputerInfo.GetFirstSelectedItemPosition();
  407. if ( NULL == pos )
  408. {
  409. AfxMessageBox(_T("请选择要打开的主机!"));
  410. return;
  411. }
  413.     int nSelIndex = m_ListCtrlComputerInfo.GetNextSelectedItem( pos );
  415. CString csHostName = _T("");
  416. csHostName = m_ListCtrlComputerInfo.GetItemText( nSelIndex, 2 );
  417. csHostName = "\\" + csHostName;
  418. char* host = &char( *csHostName );
  419. ::ShellExecute( this->GetSafeHwnd(), _T("open"), host, NULL, _T("c:\"), SW_SHOW );
  421. *pResult = 0;
  422. }
  424. void CScanLanComputerDlg::OnButtonClear() 
  425. {
  426. if ( (0 == m_ListBoxIP.GetCount()) && (0 == m_ListBoxIP.GetCount()) )
  427. {
  428. AfxMessageBox( _T("所有列表都已经为空，清除没有意义！"));
  429. return;
  430. }
  432. // 清空 listbox.
  433. m_ListBoxIP.ResetContent();
  436. // 清空 listctrl.
  437. m_ListCtrlComputerInfo.DeleteAllItems();
  439. }
  441. void CScanLanComputerDlg::OnButtonCancel() 
  442. {
  443. CDialog::OnCancel();
  444. }