?? nbtstatdlg.cpp
字號:
// NBTSTATDlg.cpp : implementation file實現文件
#include "stdafx.h"
#include "NBTSTAT.h"
#include "NBTSTATDlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
#define destPORT 137 //nbtstat name port
#define myPORT 4321
// CNBTSTATDlg dialog
CNBTSTATDlg* pDlg;//定義變量
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};
unsigned char B1[4],B2[4];
HANDLE wait_handle;
CString strOldIP="";
bool bExit=false;
int i;
CNBTSTATDlg::CNBTSTATDlg(CWnd* pParent /*=NULL*/)
: CDialog(CNBTSTATDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CNBTSTATDlg)
//}}AFX_DATA_INIT
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
//---------------------------------------------
pDlg=this;
if(!m_UDPSocket.Create(myPORT,SOCK_DGRAM))
{
AfxMessageBox("Failed Create Socket");//創建接口失敗
}
//---------------------------------------------
}
void CNBTSTATDlg::DoDataExchange(CDataExchange* pDX)//DDX技術實現
{
CDialog::DoDataExchange(pDX);//MFC會自動增加,用戶不必干預
//{{AFX_DATA_MAP(CNBTSTATDlg)
DDX_Control(pDX, IDC_SPIN1, m_spin);
DDX_Control(pDX, IDC_LIST1, m_ListBox);
DDX_Control(pDX, IDC_IPADDRESS2, m_IPEdit2);
DDX_Control(pDX, IDC_LIST3, m_ListView);
DDX_Control(pDX, IDC_IPADDRESS1, m_IPEdit1);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CNBTSTATDlg, CDialog)//消息映射函數
//{{AFX_MSG_MAP(CNBTSTATDlg)
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
ON_BN_CLICKED(IDC_BTN_SEND, OnBtnSend)
ON_NOTIFY(IPN_FIELDCHANGED, IDC_IPADDRESS1, OnFieldchangedIpaddress1)
ON_BN_CLICKED(IDC_BTN_EXIT, OnBtnExit)
ON_BN_CLICKED(IDC_BTN_SAVE, OnBtnSave)
ON_NOTIFY(NM_DBLCLK, IDC_LIST3, OnDblclkListView)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
// CNBTSTATDlg message handlers 消息句柄
BOOL CNBTSTATDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// Set the icon for this dialog. The framework does this automatically
// when the application's main window is not a dialog
// 當應用程序的主窗口不是一個對話框時,框架類會自動為對話框設置圖標
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
m_IPEdit1.SetAddress(172,22,56,1); //設置ip地址默認范圍
m_IPEdit2.SetAddress(172,22,56,255);
wait_handle=CreateEvent(NULL,true,false,"receive data");//創建處于非觸發狀態的事件。類型為手動
GetDlgItem(IDC_BTN_EXIT)->EnableWindow(false);
m_spin.SetRange(100,10000);
m_spin.SetPos(100);
// -----------------ListView 初始化---------------------
DWORD dwStyle=GetWindowLong(m_ListView.GetSafeHwnd(),GWL_STYLE);//通過函數調用取得計算機的相關信息
dwStyle&=~LVS_TYPEMASK;
dwStyle|=LVS_REPORT;
SetWindowLong(m_ListView.GetSafeHwnd(),GWL_STYLE,dwStyle);//創建列表框項風格
m_ListView.InsertColumn(0,"MAC地址",LVCFMT_LEFT,120);
m_ListView.InsertColumn(0,"用戶\\其它",LVCFMT_LEFT,100);
m_ListView.InsertColumn(0,"主機",LVCFMT_LEFT,80);
m_ListView.InsertColumn(0,"工作組",LVCFMT_LEFT,80);
m_ListView.InsertColumn(0,"IP地址",LVCFMT_LEFT,100);
m_ListView.SetExtendedStyle(LVS_EX_GRIDLINES);
::SendMessage(m_ListView.m_hWnd, LVM_SETEXTENDEDLISTVIEWSTYLE,
LVS_EX_FULLROWSELECT, LVS_EX_FULLROWSELECT);//設置列表框風格
return TRUE; // return TRUE unless you set the focus to a control
}//如果沒有獲得新的鍵盤焦點,就返加一個真值.
// If you add a minimize button to your dialog, you will need the code below
// to draw the icon. For MFC applications using the document/view model,
// this is automatically done for you by the framework.如果你增加一個最小化按鈕,
// 你會需要下面的代碼去繪畫出這個圖標. 通過MFC應用程序的文檔或視圖模式,這個主窗
// 口會自動選擇.
void CNBTSTATDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
// Center icon in client rectangle 默認圖標范圍
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);// 居中
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// Draw the icon
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
// The system calls this to obtain the cursor to display while the user drags
// the minimized window.系統通過這個去獲得游標去顯示用戶拖動的最小化窗口.
HCURSOR CNBTSTATDlg::OnQueryDragIcon()
{
return (HCURSOR) m_hIcon;//返回游標指針m_hIcon
}
//--------------------------nbtstat線程----------------------
UINT NbtstatThread(LPVOID param)
{
//循環對要查詢的ip發數據
do
{
if(bExit) //是否退出線程
{
AfxMessageBox("退出線程!");
pDlg->GetDlgItem(IDC_BTN_SEND)->EnableWindow(true);
pDlg->GetDlgItem(IDC_EDIT1)->EnableWindow(true);
pDlg->GetDlgItem(IDC_SPIN1)->EnableWindow(true);
pDlg->GetDlgItem(IDC_IPADDRESS1)->EnableWindow(true);
pDlg->GetDlgItem(IDC_IPADDRESS2)->EnableWindow(true);
pDlg->GetDlgItem(IDC_BTN_EXIT)->EnableWindow(false);
bExit=false;
return 1;
}
pDlg->m_strIP.Format("%d.%d.%d.%d",B1[0],B1[1],B1[2],B1[3]);//得到ip
pDlg->m_ListBox.InsertString(0,pDlg->m_strIP);//將該ip插入ListView的ip字段
if(B1[3]!=0&&B1[2]!=0)
pDlg->m_UDPSocket.SendTo((void*)bs,50,destPORT,pDlg->m_strIP,0);//向指定的ip發數據報
int nWait=pDlg->m_spin.GetPos();//設置超時
WaitForSingleObject(
wait_handle, // 等待事件的句柄
nWait // 超時
);
ResetEvent(wait_handle);//將事件重新置回非觸發狀態
//=============================================
//生成下一個要查詢的ip
if(B1[2]<=B2[2])
{
if(B1[3]<B2[3])B1[3]++;
else if(B1[2]<B2[2]&&B1[3]<255)B1[3]++;
else if(B1[2]<B2[2]&&B1[3]==255)
{
B1[3]=0;
B1[2]++;
}
}
else break;
if(B1[3]>=B2[3]&&B1[2]>=B2[2])break;
}while(B1[2]<=255&&B1[3]<=255);
pDlg->m_ListBox.InsertString(0,"-----完成!-----");
pDlg->GetDlgItem(IDC_BTN_SEND)->EnableWindow(true);//設置按鈕狀態
pDlg->GetDlgItem(IDC_EDIT1)->EnableWindow(true);
pDlg->GetDlgItem(IDC_SPIN1)->EnableWindow(true);
pDlg->GetDlgItem(IDC_IPADDRESS1)->EnableWindow(true);
pDlg->GetDlgItem(IDC_IPADDRESS2)->EnableWindow(true);
pDlg->GetDlgItem(IDC_BTN_EXIT)->EnableWindow(false);
return 0;
}
//-----------------------------------------------------------
void CNBTSTATDlg::OnBtnSend()
{
//從ip控件得到要查詢的ip范圍
m_IPEdit1.GetAddress(B1[0],B1[1],B1[2],B1[3]);
m_IPEdit2.GetAddress(B2[0],B2[1],B2[2],B2[3]);
//判斷ip范圍是否合法
if(B2[2]<B1[2]){AfxMessageBox("終止地址應大于起始地址！"); return ;}
else if(B2[2]==B1[2]&&B2[3]<B1[3]){AfxMessageBox("終止地址應大于起始地址！"); return ;}
if(B2[0]!=B1[0]||B2[1]!=B1[1]){AfxMessageBox("不支持A類或B類網！"); return ;}
//設置相關按鈕狀態
GetDlgItem(IDC_BTN_SEND)->EnableWindow(false);//設置按鈕狀態
GetDlgItem(IDC_EDIT1)->EnableWindow(false);
GetDlgItem(IDC_SPIN1)->EnableWindow(false);
GetDlgItem(IDC_IPADDRESS1)->EnableWindow(false);
GetDlgItem(IDC_IPADDRESS2)->EnableWindow(false);
GetDlgItem(IDC_BTN_EXIT)->EnableWindow(true);
//啟動線程
AfxBeginThread(NbtstatThread,this->GetSafeHwnd(),THREAD_PRIORITY_NORMAL);
}
void CNBTSTATDlg::OnReceive()
{
BYTE Buf[500];
CString str,strIP,strHost,strHex,strMac,Host,Group,User;
UINT dport;
m_UDPSocket.ReceiveFrom(Buf,500,strIP,dport,0);//接收數據
//如果接收到的ip為空或者與原來接收到的ip相同，則返回
if(strIP==(char)NULL||strIP==strOldIP)return;
strOldIP=strIP;
int index=m_ListView.InsertItem(0,strIP);//將ip插入ListView
strHost=""; //機器名字
strHex="";//MAC地址
User="?";//
Host="\\";
int tem=0,num=0;
bool bAdd=true;
//根據數據報規則取出相應的信息
for(i=57;i<500;i++) //57-72
{
if(Buf[i]==0xcc)break;
if(Buf[i]==0x20)bAdd=false;
if(bAdd)
{
str.Format("%c",Buf[i]);
if(Buf[i]>=' ')strHost+=str;
str.Format("%02x.",Buf[i]);
strHex+=str;
}
if((++tem)%18==0)
{
bAdd=true;
strHost.TrimRight((char)NULL);//清除右邊的空格
if(strHost=="")
{
strMac.Delete(17,strMac.GetLength()-17);
m_ListView.SetItem(index,4,LVIF_TEXT,strMac, 0, 0, 0,0);
break;
}
if(num==0&&strHost!="")
{
m_ListView.SetItem(index,2,LVIF_TEXT,strHost, 0, 0, 0,0);
Host=strHost;
num++;
}
else
{
if(Host!=strHost&&num==1&&strHost!="")
{
m_ListView.SetItem(index,1,LVIF_TEXT,strHost, 0, 0, 0,0);
Group=strHost;
num++;
}
else
{
if(strHost!=Host&&strHost!=Group&&num==2&&strHost!="")
{
User=strHost;
if(User!="__MSBROWSE__")
{
m_ListView.SetItem(index,3,LVIF_TEXT,User, 0, 0, 0,0);
num++;
}
}
}
}
strMac=strHex;
strHost="";
strHex="";
}
}
//觸發事件，導致線程函數的繼續執行
SetEvent(wait_handle);
}
void CNBTSTATDlg::OnFieldchangedIpaddress1(NMHDR* pNMHDR, LRESULT* pResult) //IP地址編輯函數
{
unsigned char AddrB[4];
m_IPEdit1.GetAddress(AddrB[0],AddrB[1],AddrB[2],AddrB[3]);
m_IPEdit2.SetAddress(AddrB[0],AddrB[1],AddrB[2],255);
*pResult = 0;
}
void CNBTSTATDlg::OnBtnExit() //退出線程
{
bExit=true;
}
CString MakeLen(CString str,int len)//獲得字串長度
{
if(str.GetLength()<len)
{
for(int i=str.GetLength();i<len;i++)str+=' ';
}
else str+=' ';
return str;
}
void CNBTSTATDlg::OnBtnSave() //保存獲得的計算機信息
{
int Count=m_ListView.GetItemCount();
CFile f;
CFileException e;
CString strFileName;
CTime t=CTime::GetCurrentTime();
CString strTime = t.Format( "%d-%H-%M-%S" );
strTime="NBT_LOG-"+strTime;
CFileDialog fileDlg(FALSE,"txt",strTime,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,"txt",NULL); //TRUE 為打開對話框
if(fileDlg.DoModal()==IDOK)
{
UpdateData(true);
strFileName=fileDlg.GetFileName();
UpdateData(false);
if( f.Open(strFileName, CFile::modeCreate | CFile::modeWrite, &e ) )
{
CString str;
str="IP地址 工作組 主機 用戶 MAC地址\r\n\r\n";
f.Write( str, str.GetLength() );
for(int i=0;i<Count;i++)
{
str=MakeLen(m_ListView.GetItemText(i,0),17)
+MakeLen(m_ListView.GetItemText(i,1),14)
+MakeLen(m_ListView.GetItemText(i,2),14)
+MakeLen(m_ListView.GetItemText(i,3),14)
+MakeLen(m_ListView.GetItemText(i,4),17);
str+="\r\n\r\n";
f.Write( str, str.GetLength() );
}
f.Close();
}
}
}
void CNBTSTATDlg::OnDblclkListView(NMHDR* pNMHDR, LRESULT* pResult) //雙擊列表框編輯數
{
CString str;
POSITION pos=m_ListView.GetFirstSelectedItemPosition();//獲得第一次選擇的列表位置
if(pos==NULL)
{
AfxMessageBox("請選擇想要打開的主機。");
return;
}
int SelIndex = m_ListView.GetNextSelectedItem(pos);
str=m_ListView.GetItemText(SelIndex,2);
str="\\\\"+str;
char* host=&char(*str);
::ShellExecute(this->GetSafeHwnd(),"open",host,NULL,"c:\\",SW_SHOW);//日志記錄存放位置
*pResult = 0;
}
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -