windows socket入门例程

windows socket入门例程

这里只粘贴了TEST.CPP的内容, 关键是说明IP地址的取得.

// TestDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Test.h"
#include "TestDlg.h"
#include <afxsock.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()

/////////////////////////////////////////////////////////////////////////////
// CTestDlg dialog

CTestDlg::CTestDlg(CWnd* pParent /*=NULL*/)
 : CDialog(CTestDlg::IDD, pParent)
{
 //{{AFX_DATA_INIT(CTestDlg)
 m_Info = _T("");
 m_Output = _T("");
 m_Input = _T("");
 m_Connect = _T("");
 m_IPAddress = _T("");
 m_Port = 0;
 m_Status = -1;
 //}}AFX_DATA_INIT
 // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
 m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CTestDlg::DoDataExchange(CDataExchange* pDX)
{
 CDialog::DoDataExchange(pDX);
 //{{AFX_DATA_MAP(CTestDlg)
 DDX_Text(pDX, IDC_OUTPUTEDIT, m_Output);
 DDX_Text(pDX, IDC_INPUTEDIT, m_Input);
 DDX_Text(pDX, IDC_CONNECTEDIT, m_Connect);
 DDX_Text(pDX, IDC_IPADDRESS, m_IPAddress);
 DDV_MaxChars(pDX, m_IPAddress, 15);
 DDX_Text(pDX, IDC_PORT, m_Port);
 DDX_Radio(pDX, IDC_SERVERRADIO, m_Status);
 //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CTestDlg, CDialog)
 //{{AFX_MSG_MAP(CTestDlg)
 ON_WM_SYSCOMMAND()
 ON_WM_PAINT()
 ON_WM_QUERYDRAGICON()
 ON_BN_CLICKED(IDC_CONNECTBUTTON, OnConnect)
 ON_BN_CLICKED(IDC_DISCONNECTBUTTON, OnDisconnect)
 ON_BN_CLICKED(IDC_SENDBUTTON, OnSend)
 ON_BN_CLICKED(IDC_SERVERRADIO, OnServerradio)
 ON_BN_CLICKED(IDC_CLIENTRADIO, OnClientradio)
 ON_BN_CLICKED(IDC_SENDCLEARBUTTON, OnSendclear)
 ON_BN_CLICKED(IDC_RECEIVECLEARBUTTON, OnReceiveclear)
 //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTestDlg message handlers

BOOL CTestDlg::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
 
    m_Status=-1;
    m_ServerSocket=NULL;
 m_ClientSocket=NULL;
 m_arIn=NULL;
 m_arOut=NULL;
    m_file=NULL;
 m_Connect="";
 m_IPAddress="127.0.0.1";
 m_Port=5000;
    GetDlgItem(IDC_IPADDRESS)->EnableWindow(FALSE);
 GetDlgItem(IDC_PORT)->EnableWindow(FALSE);
 UpdateData(FALSE);
 return TRUE;  // return TRUE  unless you set the focus to a control
}

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

void CTestDlg::OnConnect()
{
 CString msg;
    UpdateData(TRUE);
 if (m_Status==0 )  //server
 {
  if  ( m_ServerSocket!=NULL)
        {
   m_Connect="Please disconnect!";
   UpdateData(FALSE);
  }
  else
  {
           m_Connect="Waiting for Client...";
     UpdateData(FALSE);
     if(!AfxSocketInit())
     {
              MessageBox("WindowsSocket initial failed!","Send",MB_ICONSTOP);
              return;
     }
           m_ServerSocket=new CNewSocket;
           m_ServerSocket->m_Status=m_Status;
     m_ServerSocket->GetDlg(this); 
     if(!m_ServerSocket->Create(m_Port))
              MessageBox("SendSocket create failed!", "Send",MB_ICONSTOP);
           else
     {
              m_ServerSocket->Listen();
     }
  }
 }
 else
 {
  if (m_Status==1)
  {
   if (m_ClientSocket!=NULL)
   {         
    m_Connect="Please disconnect!";
    UpdateData(FALSE);
   }
   else
   {
    m_Connect="Connect to the Server...";
    UpdateData(FALSE);
    if(!AfxSocketInit())
    {
     MessageBox("WindowsSocket initial failed!", "Receive",MB_ICONSTOP);
                    return;
    }
                m_ClientSocket=new CNewSocket;
                m_ClientSocket->GetDlg(this);
    m_ClientSocket->m_Status=m_Status;
          if(!m_ClientSocket->Create())
    {
     MessageBox("ReceiveSocket create failed!","Receive",MB_ICONSTOP);
     return;
    }
    else
    {
     if (!m_ClientSocket->Connect(m_IPAddress,m_Port))
     {
      CString str=m_Connect;
      SocketReset();
      m_Connect=str;
      m_Connect+="Error!";
      UpdateData(FALSE);
     }      
     else  
     {         
      m_Connect+="OK!";
      m_file=new CSocketFile(m_ClientSocket);
      m_arIn=new CArchive(m_file, CArchive::load);
      m_arOut=new CArchive(m_file, CArchive::store);
     }
     UpdateData(FALSE);
    }
   }
  }
 }
 if (m_Status==-1)
 {
  msg="Please choose the status!";
  AfxMessageBox(msg);   
 }
}

void CTestDlg::OnSend()
{
  if (m_arOut)
 {
    if (m_Status==0)
  {
          UpdateData(TRUE);
    *m_arOut<<m_Output;
          m_arOut->Flush(); 
  }
     else
  {
           UpdateData(TRUE);
        *m_arOut<<m_Output;
           m_arOut->Flush(); 
  }
 }
 else AfxMessageBox("Not connected!");
}

void CTestDlg::OnAccept()
{
 struct sockaddr_in Addr;
 CString ss;
 int addrlen;
 m_Connect+="OK!";
 UpdateData(FALSE);
 m_ClientSocket=new CNewSocket;
 m_ClientSocket->GetDlg(this);

 m_ServerSocket->Accept(*m_ClientSocket,(struct sockaddr*)&Addr,&addrlen);

//必须加入"&addrlen"一项才能正确取得IP地址

 ss.Format("%s",::inet_ntoa(Addr.sin_addr));//inet_ntoa(Addr.sin_addr);
 GetDlgItem(IDC_EDIT3)->SetWindowText(ss);

 m_ClientSocket->m_Status=m_ServerSocket->m_Status;
    m_file=new CSocketFile(m_ClientSocket);
 m_arIn=new CArchive(m_file, CArchive::load);
    m_arOut=new CArchive(m_file, CArchive::store);
}

void CTestDlg::OnReceive()
{
 *m_arIn>>m_Input;  
 UpdateData(FALSE);  
}

void CTestDlg::OnDisconnect()
{
    if (m_arOut!=NULL)
 {
  SocketReset();
  m_Connect="Disconnected!";
  UpdateData(FALSE);
 }
}

void CTestDlg::OnClose()
{
    if (m_ClientSocket->m_Status==0) m_Connect="Client ";
 else m_Connect="Server ";
 m_Connect+="has disconnected!"; 
 UpdateData(FALSE);
}

void CTestDlg::SocketReset()
{
 if (m_arIn!=NULL)
 {
  delete m_arIn;
  m_arIn=NULL;
 }
 if (m_arOut!=NULL)
 {
  delete m_arOut;
  m_arOut=NULL;
 } 
 if (m_file!=NULL) 
 {
  delete m_file;
  m_file=NULL;
 }
 if (m_ClientSocket!=NULL) 
 {
  delete m_ClientSocket;
  m_ClientSocket=NULL;
 }
 if (m_ServerSocket!=NULL)
 {   
  delete m_ServerSocket;
  m_ServerSocket=NULL;
 }
 m_Connect="";
 UpdateData(FALSE);
}

void CTestDlg::OnServerradio()
{
    UpdateData(TRUE);
    GetDlgItem(IDC_IPADDRESS)->EnableWindow(FALSE);
 GetDlgItem(IDC_PORT)->EnableWindow(TRUE);
    UpdateData(FALSE);
}

void CTestDlg::OnClientradio()
{ 
    UpdateData(TRUE);
    GetDlgItem(IDC_IPADDRESS)->EnableWindow(TRUE);
 GetDlgItem(IDC_PORT)->EnableWindow(TRUE);
    UpdateData(FALSE);
}

void CTestDlg::OnSendclear()
{
    m_Output="";
    UpdateData(FALSE);
}

void CTestDlg::OnReceiveclear()
{
 m_Input="";
 UpdateData(FALSE);
}