Socket 學習之 MFC:簡單通訊
阿新 • • 發佈:2019-01-23
最終介面如圖所示:
客戶端程式碼如下:
// ClientDlg.h : 標頭檔案 // #pragma once UINT Recv_Th(LPVOID p); // CClientDlg 對話方塊 class CClientDlg : public CDialog { // 構造 public: CClientDlg(CWnd* pParent = NULL); // 標準建構函式 // 對話方塊資料 enum { IDD = IDD_CLIENT_DIALOG }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支援 // 實現 protected: HICON m_hIcon; // 生成的訊息對映函式 virtual BOOL OnInitDialog(); afx_msg void OnSysCommand(UINT nID, LPARAM lParam); afx_msg void OnPaint(); afx_msg HCURSOR OnQueryDragIcon(); DECLARE_MESSAGE_MAP() public: void Update(CString str); private: CEdit * edit_ip; CEdit * edit_send; CEdit * edit_show; CButton * btn_conn; public: afx_msg void OnBnClickedBsend(); afx_msg void OnBnClickedBconn(); };
// ClientDlg.cpp : 實現檔案 // #include <winsock2.h> #include "stdafx.h" #include "Client.h" #include "ClientDlg.h" #pragma comment(lib, "WS2_32.lib") #ifdef _DEBUG #define new DEBUG_NEW #endif SOCKET sock; // 用於應用程式“關於”選單項的 CAboutDlg 對話方塊 class CAboutDlg : public CDialog { public: CAboutDlg(); // 對話方塊資料 enum { IDD = IDD_ABOUTBOX }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支援 // 實現 protected: DECLARE_MESSAGE_MAP() }; CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD) { EnableActiveAccessibility(); } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CAboutDlg, CDialog) END_MESSAGE_MAP() // CClientDlg 對話方塊 CClientDlg::CClientDlg(CWnd* pParent /*=NULL*/) : CDialog(CClientDlg::IDD, pParent) { EnableActiveAccessibility(); m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CClientDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CClientDlg, CDialog) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() //}}AFX_MSG_MAP ON_BN_CLICKED(IDC_BSEND, &CClientDlg::OnBnClickedBsend) ON_BN_CLICKED(IDC_BCONN, &CClientDlg::OnBnClickedBconn) END_MESSAGE_MAP() // CClientDlg 訊息處理程式 BOOL CClientDlg::OnInitDialog() { CDialog::OnInitDialog(); // 將“關於...”選單項新增到系統選單中。 // IDM_ABOUTBOX 必須在系統命令範圍內。 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); } } // 設定此對話方塊的圖示。當應用程式主視窗不是對話方塊時,框架將自動 // 執行此操作 SetIcon(m_hIcon, TRUE); // 設定大圖示 SetIcon(m_hIcon, FALSE); // 設定小圖示 ShowWindow(SW_MINIMIZE); // TODO: 在此新增額外的初始化程式碼 edit_show = (CEdit *)GetDlgItem(IDC_ESHOW); edit_send = (CEdit *)GetDlgItem(IDC_ESEND); btn_conn = (CButton *)GetDlgItem(IDC_BCONN); edit_ip = (CEdit *)GetDlgItem(IDC_EIP); edit_ip->SetWindowText("172.16.70.80"); if (!AfxSocketInit()) { AfxMessageBox(IDP_SOCKETS_INIT_FAILED); return FALSE; } return TRUE; // 除非將焦點設定到控制元件,否則返回 TRUE } void CClientDlg::OnSysCommand(UINT nID, LPARAM lParam) { if ((nID & 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout; dlgAbout.DoModal(); } else { CDialog::OnSysCommand(nID, lParam); } } // 如果向對話方塊新增最小化按鈕,則需要下面的程式碼 // 來繪製該圖示。對於使用文件/檢視模型的 MFC 應用程式, // 這將由框架自動完成。 void CClientDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // 用於繪製的裝置上下文 SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0); // 使圖示在工作區矩形中居中 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; // 繪製圖標 dc.DrawIcon(x, y, m_hIcon); } else { CDialog::OnPaint(); } } //當用戶拖動最小化視窗時系統呼叫此函式取得游標 //顯示。 HCURSOR CClientDlg::OnQueryDragIcon() { return static_cast<HCURSOR>(m_hIcon); } void CClientDlg::Update(CString str) { edit_show->ReplaceSel(str + "\r\n"); } UINT Recv_Th(LPVOID p) { int res; char msg[1024]; CString str; CClientDlg * dlg = (CClientDlg *) AfxGetApp()->GetMainWnd(); dlg->Update("Initialization"); while(1) { if( (res = recv(sock, msg, 1024, 0)) == -1) { dlg->Update("失去連線"); break; } else { msg[res] = '\0'; dlg->Update("server:" + CString(msg)); } } return 0; } void CClientDlg::OnBnClickedBsend() { // TODO: 在此新增控制元件通知處理程式程式碼 CString str; char * msg; edit_send->GetWindowText(str); msg = str.GetBuffer(str.GetLength()); if(send(sock, msg, strlen(msg), 0) == SOCKET_ERROR) { Update("傳送失敗"); } else if(str == "") { MessageBox("請輸入資訊"); } else { Update("client:" + str);//訊息上屏,清空輸入,並重獲焦點 edit_send->SetWindowText(""); edit_send->SetFocus(); } } void CClientDlg::OnBnClickedBconn() { // TODO: 在此新增控制元件通知處理程式程式碼 WSADATA wsaData; SOCKADDR_IN server_addr; WORD wVersion; wVersion = MAKEWORD(2,2); WSAStartup(wVersion,&wsaData); CString ip; edit_ip->GetWindowText(ip);//取得伺服器的IP地址 server_addr.sin_addr.s_addr = inet_addr(ip); server_addr.sin_family = AF_INET; server_addr.sin_port = htons(5150); if( (sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET) { edit_show->ReplaceSel("create socket error\r\n"); } if( connect(sock, (struct sockaddr *) &server_addr, sizeof(SOCKADDR_IN)) == SOCKET_ERROR) { edit_show->ReplaceSel("連線失敗\r\n"); } else { edit_show->ReplaceSel("連線成功\r\n"); AfxBeginThread(&Recv_Th, 0); btn_conn->EnableWindow(FALSE);//按鈕變灰 } }
伺服器端程式碼如下:
// ServerDlg.h : 標頭檔案 // #pragma once UINT Server_Th(LPVOID p); // CServerDlg 對話方塊 class CServerDlg : public CDialog { // 構造 public: CServerDlg(CWnd* pParent = NULL); // 標準建構函式 // 對話方塊資料 enum { IDD = IDD_SERVER_DIALOG }; protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支援 // 實現 protected: HICON m_hIcon; // 生成的訊息對映函式 virtual BOOL OnInitDialog(); afx_msg void OnSysCommand(UINT nID, LPARAM lParam); afx_msg void OnPaint(); afx_msg HCURSOR OnQueryDragIcon(); DECLARE_MESSAGE_MAP() public: void Update(CString str); private: CEdit * show_edit; CEdit * send_edit; public: afx_msg void OnBnClickedBsend(); };
// ServerDlg.cpp : 實現檔案
//
#include "winsock2.h"
#pragma comment(lib, "WS2_32.lib")
#include "stdafx.h"
#include "Server.h"
#include "ServerDlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
SOCKET listen_sock;
SOCKET sock;
// 用於應用程式“關於”選單項的 CAboutDlg 對話方塊
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// 對話方塊資料
enum { IDD = IDD_ABOUTBOX };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支援
// 實現
protected:
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
EnableActiveAccessibility();
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()
// CServerDlg 對話方塊
CServerDlg::CServerDlg(CWnd* pParent /*=NULL*/)
: CDialog(CServerDlg::IDD, pParent)
{
EnableActiveAccessibility();
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CServerDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CServerDlg, CDialog)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
ON_BN_CLICKED(IDC_BSEND, &CServerDlg::OnBnClickedBsend)
END_MESSAGE_MAP()
// CServerDlg 訊息處理程式
BOOL CServerDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// 將“關於...”選單項新增到系統選單中。
// IDM_ABOUTBOX 必須在系統命令範圍內。
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);
}
}
// 設定此對話方塊的圖示。當應用程式主視窗不是對話方塊時,框架將自動
// 執行此操作
SetIcon(m_hIcon, TRUE); // 設定大圖示
SetIcon(m_hIcon, FALSE); // 設定小圖示
ShowWindow(SW_MINIMIZE);
// TODO: 在此新增額外的初始化程式碼
AfxBeginThread(&Server_Th, 0);
show_edit = (CEdit *)GetDlgItem(IDC_ESHOW);
send_edit = (CEdit *)GetDlgItem(IDC_ESEND);
send_edit->SetFocus();
char name[80];
CString IP;
hostent * pHost;
gethostname(name, 128);//獲得主機名
pHost = gethostbyname(name);//獲得主機結構
IP = inet_ntoa(*(in_addr *)pHost->h_addr);
Update("本機IP地址:" + IP);
return TRUE; // 除非將焦點設定到控制元件,否則返回 TRUE
}
void CServerDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialog::OnSysCommand(nID, lParam);
}
}
// 如果向對話方塊新增最小化按鈕,則需要下面的程式碼
// 來繪製該圖示。對於使用文件/檢視模型的 MFC 應用程式,
// 這將由框架自動完成。
void CServerDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // 用於繪製的裝置上下文
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
// 使圖示在工作區矩形中居中
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;
// 繪製圖標
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
//當用戶拖動最小化視窗時系統呼叫此函式取得游標
//顯示。
HCURSOR CServerDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
void CServerDlg::Update(CString str)
{
show_edit->ReplaceSel(str + "\r\n");
}
UINT Server_Th(LPVOID p)
{
WSADATA wsaData;
WORD wVersion;
wVersion = MAKEWORD(2,2);
WSAStartup(wVersion,&wsaData);
SOCKADDR_IN local_addr;
SOCKADDR_IN client_addr;
int iaddrSize = sizeof(SOCKADDR_IN);
int res;
char msg[1024];
CServerDlg * dlg = (CServerDlg *)AfxGetApp()->GetMainWnd();
local_addr.sin_family = AF_INET;
local_addr.sin_port = htons(5150);
local_addr.sin_addr.s_addr = htonl(INADDR_ANY);
if( (listen_sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET )
{
dlg->Update("建立監聽失敗");
}
if( bind(listen_sock, (struct sockaddr*) &local_addr, sizeof(SOCKADDR_IN)) )
{
dlg->Update("繫結錯誤");
}
listen(listen_sock, 1);
if( (sock = accept(listen_sock, (struct sockaddr *)&client_addr, &iaddrSize)) == INVALID_SOCKET)
{
dlg->Update("accept 失敗");
}
else
{
CString port;
port.Format("%d", int(ntohs(client_addr.sin_port)));
dlg->Update( "已連線來自:" + CString(inet_ntoa(client_addr.sin_addr)) + " 埠:" + port);
}
//接收資料
while(1)
{
if( (res = recv(sock, msg, 1024, 0)) == -1 )
{
dlg->Update("失去連線");
break;
}
else
{
msg[res] = '\0';
dlg->Update("client:" + CString(msg));
}
}
return 0;
}
void CServerDlg::OnBnClickedBsend()
{
// TODO: 在此新增控制元件通知處理程式程式碼
CString str;
char * msg;
send_edit->GetWindowText(str);
msg = str.GetBuffer(str.GetLength());
if(send(sock, msg, strlen(msg), 0) == SOCKET_ERROR)
{
show_edit->ReplaceSel("傳送失敗\r\n");
}
else if(str == "")
{
MessageBox("請輸入資訊");
}
else
{
show_edit->ReplaceSel("server:" + str + "\r\n");//訊息上屏,清空輸入,並重獲焦點
send_edit->SetWindowText("");
send_edit->SetFocus();
}
}