getmyipdlg.cpp

獲得本機 IP 地址的程序

// GetMyIPDlg.cpp : implementation file
//

#include "stdafx.h"
#include "GetMyIP.h"
#include "GetMyIPDlg.h"

#include <winsock2.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()

/////////////////////////////////////////////////////////////////////////////
// CGetMyIPDlg dialog

CGetMyIPDlg::CGetMyIPDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CGetMyIPDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CGetMyIPDlg)
	m_IpAddress = _T("");
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CGetMyIPDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CGetMyIPDlg)
	DDX_Text(pDX, IDC_IPADDRESS_EDIT, m_IpAddress);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CGetMyIPDlg, CDialog)
	//{{AFX_MSG_MAP(CGetMyIPDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CGetMyIPDlg message handlers

BOOL CGetMyIPDlg::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
	
	// TODO: Add extra initialization here
	unsigned char MyIp[5] ;
	char HostIP[16];
	GetMyIp(MyIp) ;
	ConvertToIP(MyIp , HostIP) ;

	m_IpAddress = HostIP ;
	UpdateData(FALSE) ;
	return TRUE;  // return TRUE  unless you set the focus to a control
}

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

void CGetMyIPDlg::GetMyIp(unsigned char MyIp[])
{
	unsigned char MyIpAddress[5];
	WORD wVersionRequested = MAKEWORD(1, 1);
	WSADATA wsaData;
	
	if (!WSAStartup(wVersionRequested, &wsaData)) 
	{
		char hostname[256];
		int res = gethostname(hostname, sizeof(hostname));
		
		if (res == 0) 
		{
			hostent* pHostent = gethostbyname(hostname);
			if (pHostent!=NULL) 
			{
				hostent& he = *pHostent;
				sockaddr_in sa;
				
				for (int nAdapter=0; he.h_addr_list[nAdapter]; nAdapter++) 
				{
					memcpy ( &sa.sin_addr.s_addr, he.h_addr_list[nAdapter],he.h_length);
					// Output the machines IP Address.
				}	
				WSACleanup();
				
				MyIpAddress[0] = sa.sin_addr.S_un.S_un_b.s_b1;
				MyIpAddress[1] = sa.sin_addr.S_un.S_un_b.s_b2;
				MyIpAddress[2] = sa.sin_addr.S_un.S_un_b.s_b3;
				MyIpAddress[3] = sa.sin_addr.S_un.S_un_b.s_b4;

				MyIpAddress[4] = '\0' ;
			}
			else
				strcpy((char *)MyIpAddress,TEXT(""));
		}
		else
			strcpy((char *)MyIpAddress,TEXT(""));
	}
	else
		strcpy((char *)MyIpAddress,TEXT(""));

	strcpy((char *)MyIp , (char *)MyIpAddress);
}

void CGetMyIPDlg::ConvertToIP(unsigned char *exten, char HostIp[])
{
	char tempIp[4],ServerIp[16],tempch;
	unsigned char tempCharacter[4];
	int i , j , k , quotient , remainder ;
			
	for(i = 0 ; i < 4 ; i++)
	{
		tempCharacter[i] = exten[i];
	}

	strcpy(ServerIp,TEXT(""));

	for(i = 0 ; i < 4 ; i ++)
	{
		j = 0 ;
			
		quotient  = tempCharacter[i] / 10 ;
		remainder = tempCharacter[i] % 10 ;

		tempIp[j] = remainder + 48 ;
		j++;
		while(quotient && j < 3)
		{
			remainder = quotient % 10  ;
			quotient  = quotient / 10  ;
			tempIp[j] = remainder + 48 ;
			j++;
		}
		tempIp[j] = '\0' ;
		tempIp[3] = '\0' ;

		//revert
				
		for(k = 0 ; k < j / 2 ; k ++)
		{
			tempch		  = tempIp[k] ;
			tempIp[k]	  = tempIp[j - k -1] ;
			tempIp[j - k - 1] = tempch ;
		}

		tempIp[j] = '\0' ;
		tempIp[3] = '\0' ;
		
		strcat(ServerIp,tempIp);
				
		if(i < 3)
			strcat(ServerIp,TEXT("."));
	}
	ServerIp[15] = '\0' ;
	strcpy(HostIp,ServerIp);
}