2.5. Додаток ClientSocket

Вихідні тексти додатка ClientSocket, призначеного для спільного використання з тільки що описаним додатком ServerSocket, наведені в листинге 3.2.

Після запуску цього додатка ви повинні створити канал зв'язку з додатком ServerSocket, нажавши на кнопку Connect, після чого можна набирати будь-які повідомлення в текстовому вікні й посилати їх, натискаючи на кнопку Send.

Сервер, одержавши повідомлення, відобразить його на екрані в у текстовому вікні.

Листинг 3.2.

Файл ClientSocket/CClientSocketDlg.h

// ClientSocketDlg.h : header file

//

#if !defined(AFX_CLIENTSOCKETDLG_H__13F49153_B30E_471D_BAEB_20BFC9900E2E__INCLUDED_)

#define AFX_CLIENTSOCKETDLG_H__13F49153_B30E_471D_BAEB_20BFC9900E2E__INCLUDED_

#if _MSC_VER > 1000

#pragma once

#endif // _MSC_VER > 1000

#include <winsock2.h>

/////////////////////////////////////////////////////////////////////////////

// CClientSocketDlg dialog

class CClientSocketDlg : public CDialog

{

// Construction

private:

void SendData( char * );

void SetConnection( char *ipAddress, int port );

public:

CClientSocketDlg(CWnd* pParent = NULL); // standard constructor

// Dialog Data

//{{AFX_DATA(CClientSocketDlg)

enum { IDD = IDD_CLIENTSOCKET_DIALOG };

CIPAddressCtrl m_IpAddress;

int m_portNum;

CString m_edtData;

//}}AFX_DATA

// ClassWizard generated virtual function overrides

//{{AFX_VIRTUAL(CClientSocketDlg)

protected:

virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support

//}}AFX_VIRTUAL

// Implementation

protected:

void CloseSocket();

HICON m_hIcon;

// Методи для роботи із сокетуми

BOOL InitLib();

// Змінні для роботи із сокетом

SOCKET clt_socket;

// Дескриптор клієнтського сокету

// Generated message map functions

//{{AFX_MSG(CClientSocketDlg)

virtual BOOL OnInitDialog();

afx_msg void OnSysCommand(UINT nID, LPARAM lParam);

afx_msg void OnPaint();

afx_msg HCURSOR OnQueryDragIcon();

afx_msg void OnBtnSend();

afx_msg void OnBtnConnect();

afx_msg void OnBtnDisconnect();

afx_msg void OnClose();

//}}AFX_MSG

DECLARE_MESSAGE_MAP()

};

//{{AFX_INSERT_LOCATION}}

// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_CLIENTSOCKETDLG_H__13F49153_B30E_471D_BAEB_20BFC9900E2E__INCLUDED_)

Файл ClientSocket/CClientSocketDlg.cpp

// ClientSocketDlg.cpp : implementation file

//

#include "stdafx.h"

#include "ClientSocket.h"

#include "ClientSocketDlg.h"

#ifdef _DEBUG

#define new DEBUG_NEW

#undef THIS_FILE

static char THIS_FILE[] = __FILE__;

#endif

#define WSA_NETEVENT (WM_USER+2)

/////////////////////////////////////////////////////////////////////////////

// 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()

/////////////////////////////////////////////////////////////////////////////

// CClientSocketDlg dialog

CClientSocketDlg::CClientSocketDlg(CWnd* pParent /*=NULL*/)

: CDialog(CClientSocketDlg::IDD, pParent)

{

//{{AFX_DATA_INIT(CClientSocketDlg)

m_portNum = 0;

m_edtData = _T("");

//}}AFX_DATA_INIT

// Note that LoadIcon does not require a subsequent DestroyIcon in Win32

m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);

}

void CClientSocketDlg::DoDataExchange(CDataExchange* pDX)

{

CDialog::DoDataExchange(pDX);

//{{AFX_DATA_MAP(CClientSocketDlg)

DDX_Control(pDX, IDC_IPADDRESS1, m_IpAddress);

DDX_Text(pDX, IDC_EDT_PORT, m_portNum);

DDV_MinMaxInt(pDX, m_portNum, 0, 65535);

DDX_Text(pDX, IDC_EDIT1, m_edtData);

//}}AFX_DATA_MAP

}

BEGIN_MESSAGE_MAP(CClientSocketDlg, CDialog)

//{{AFX_MSG_MAP(CClientSocketDlg)

ON_WM_SYSCOMMAND()

ON_WM_PAINT()

ON_WM_QUERYDRAGICON()

ON_BN_CLICKED(IDC_BTN_SEND, OnBtnSend)

ON_BN_CLICKED(IDC_BTN_CONNECT, OnBtnConnect)

ON_BN_CLICKED(IDC_BTN_DISCONNECT, OnBtnDisconnect)

ON_WM_CLOSE()

//}}AFX_MSG_MAP

END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////

// CClientSocketDlg message handlers

BOOL CClientSocketDlg::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 < 0x000);

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_IpAddress.SetWindowText("127.0.0.1");

m_portNum = 8888;

UpdateData(FALSE);

if ( !InitLib() )

{

MessageBox("Init error", "Error", MB_OK);

return FALSE;

}

return TRUE; // return TRUE unless you set the focus to a control

}

void CClientSocketDlg::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 CClientSocketDlg::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 CClientSocketDlg::OnQueryDragIcon()

{

return (HCURSOR) m_hIcon;

}

BOOL CClientSocketDlg::InitLib()

{

int rc;

WSADATA WSAData;

char szTemp[128];

// Ініціалізація й перевірка версії Windows Sockets

rc = WSAStartup(MAKEWORD(2, 2), &WSAData);

if(rc != 0)

{

MessageBox("WSAStartup Error", "Error", MB_OK);

return FALSE;

}

// Відображаємо опис і версію системи Windows Sockets

// у заголовку вікна

wsprintf(szTemp, "Client use %s %s",

WSAData.szDescription,WSAData.szSystemStatus);

SetWindowText( szTemp );

return TRUE;

}

void CClientSocketDlg::SetConnection( char *ipAddress, int port )

{

// Створюємо сокет

clt_socket = socket(AF_INET, SOCK_STREAM, 0);

if(clt_socket == INVALID_SOCKET)

{

return;

}

PHOSTENT phe;

SOCKADDR_IN destAddr;

// Установлюємо адресу IP і номер порту

destAddr.sin_family = AF_INET;

// Визначаємо адреса вузла

phe = gethostbyname( ipAddress );

if(phe == NULL)

{

closesocket(clt_socket);

MessageBox("Gethostbyname Error", "Error", MB_OK);

return;

}

// Копіюємо адресу вузла

memcpy((char FAR *)&(destAddr.sin_addr), phe->h_addr, phe->h_length);

// Копіюємо номер порту

destAddr.sin_port = htons(port);

// Установлюємо з'єднання

if(connect(clt_socket, (PSOCKADDR)&destAddr, sizeof(destAddr)) < 0)

{

MessageBox("Connect Error", "Error", MB_OK);

return;

}

}

void CClientSocketDlg::OnBtnSend()

{

UpdateData();

SendData( (char*)(LPCSTR)m_edtData );

}

void CClientSocketDlg::OnBtnConnect()

{

UpdateData();

CString strIpAddress;

m_IpAddress.GetWindowText( strIpAddress );

SetConnection( (char*)(LPCSTR)strIpAddress, m_portNum );

}

void CClientSocketDlg::SendData( char *szBuf )

{

int len = send(clt_socket, szBuf, lstrlen(szBuf), 0);

}

void CClientSocketDlg::CloseSocket()

{

closesocket( clt_socket );

clt_socket = 0;

}

void CClientSocketDlg::OnBtnDisconnect()

{

CloseSocket();

}

void CClientSocketDlg::OnClose()

{

CloseSocket();

CDialog::OnClose();

}

Всі використані в ньому функції були вже описані. Звернути увагу треба тільки на функцію SetConnection, призначену для установки каналу зв'язку із сервером.

Якщо буде виконуватися перевірка роботи додатків ServerSocket і ClientSocket на тому самому комп'ютері, адреса сервера повинен бути зазначений у такий спосіб:

phe = gethostbyname ("localhost") або

phe = gethostbyname ("127.0.0.1")