netbiosdlg.cpp

对硬件的netbios编程

// netbiosDlg.cpp : implementation file
//

#include "stdafx.h"
#include "netbios.h"
#include "netbiosDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CNetbiosDlg dialog

CNetbiosDlg::CNetbiosDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CNetbiosDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CNetbiosDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CNetbiosDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CNetbiosDlg)
	DDX_Control(pDX, IDC_LIT_SHOWMAC, c_litctl);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CNetbiosDlg, CDialog)
	//{{AFX_MSG_MAP(CNetbiosDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BT_GETMAC, OnBtGetMAC)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CNetbiosDlg message handlers

BOOL CNetbiosDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 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
	
	//列表控件分两列
	TCHAR rgtsz[4][10] = {_T("网卡序号"),_T("Mac地址"),_T("哈哈")};
	LV_COLUMN lvcolumn;
	CRect rect;
	c_litctl.GetWindowRect(&rect);
	for(int i=0;i<3;i++)
	{
		lvcolumn.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT 
						| LVCF_WIDTH | LVCF_ORDER;
		lvcolumn.fmt = LVCFMT_LEFT;
		lvcolumn.pszText = rgtsz[i];
		lvcolumn.iSubItem = i;
		lvcolumn.iOrder = i;
		switch (i)
		{
		case 0 :
			lvcolumn.cx=rect.Width()/6;
            break;
		case 1 :
			lvcolumn.cx=(rect.Width()*2)/3;
			break;
		case 2 :
			lvcolumn.cx=rect.Width()/6 ;
			break;
		default :
			break;
		}

		c_litctl.InsertColumn(i, &lvcolumn);
	}
	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CNetbiosDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 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.

void CNetbiosDlg::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 CNetbiosDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CNetbiosDlg::OnBtGetMAC() 
{
	NCB f_ncb;
	UCHAR m_code;
	LANA_ENUM m_lenum;
	int m_occurred=0;

	memset(&f_ncb,0,sizeof(f_ncb));
	f_ncb.ncb_command=	NCBENUM;
	f_ncb.ncb_buffer=(UCHAR *)&m_lenum;
	f_ncb.ncb_length=sizeof(m_lenum);

	m_code=Netbios(&f_ncb);

	for(m_occurred=0;m_occurred<m_lenum.length;m_occurred++)
	{
		GetOneMac(m_lenum.lana[m_occurred]);
	}

	int m_item;
	LV_ITEM m_lvitem;
	TCHAR buffer[128];
	for(int i=0;i<m_arrAdapters.GetSize();i++)
	{
		for(int j=0;j<2;j++)
		{
			m_lvitem.mask=LVIF_TEXT|(j==0?LVIF_IMAGE:0);
			m_lvitem.iItem=(j==0)?i:m_item;
			m_lvitem.iSubItem=j;
			m_lvitem.iImage=(i%2)?0:2;
			if(i==0)
			{
				sprintf(buffer,"%d",m_arrAdapters.GetAt(i).nIndex);
				m_lvitem.pszText=buffer;
				m_item=c_litctl.InsertItem(&m_lvitem);
			}
			else
			{
				sprintf(buffer,"%s",m_arrAdapters.GetAt(i).strMac);
				m_lvitem.pszText=buffer;
				c_litctl.SetItem(&m_lvitem);
			}
		}
	}
	// TODO: Add your control notification handler code here	
}



void CNetbiosDlg::GetOneMac(int AdapterIndex)
{
	NCB f_ncb;
	UCHAR m_code;
	ASTAT m_adapter;
	memset(&f_ncb,0,sizeof(f_ncb));
	f_ncb.ncb_command=NCBRESET;
	f_ncb.ncb_lana_num=AdapterIndex;

	m_code=Netbios(&f_ncb);

	memset(&f_ncb,0,sizeof(f_ncb));
	f_ncb.ncb_command=NCBASTAT;
	f_ncb.ncb_lana_num=AdapterIndex;
	
	strcpy((char *)f_ncb.ncb_callname,"*");
	f_ncb.ncb_command=NCBASTAT;
	f_ncb.ncb_length=sizeof(m_adapter);

	m_code=Netbios(&f_ncb);
	if(m_code==0)
	{
		CString m_MacAddr;
		m_MacAddr.Format("%02X%02X-%02X%02x-%02X%02%-%02%02/n",m_adapter.adapt.adapter_address[0],m_adapter.adapt.adapter_address[1],m_adapter.adapt.adapter_address[2],m_adapter.adapt.adapter_address[3],m_adapter.adapt.adapter_address[4],m_adapter.adapt.adapter_address[5]);

		ADPTSTRCT m_adptst;
		m_adptst.nIndex=AdapterIndex;
		m_adptst.strMac=m_MacAddr;
	
		m_arrAdapters.Add(m_adptst);
	}
	// TODO: Add your control notification handler code here
}