简单Socket 封装类及应用

最新推荐文章于 2021-01-05 18:55:15 发布

memory_off

最新推荐文章于 2021-01-05 18:55:15 发布

阅读量1.2k

点赞数

CC 4.0 BY-SA版权

分类专栏： MFC 文章标签： socket byte null 服务器 binding stream

本文链接：https://blog.youkuaiyun.com/memory_off/article/details/5330054

MFC 专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一个基于Windows Socket的服务器客户端通信实现案例。该案例详细展示了如何通过C++进行服务器监听及客户端连接，并处理数据收发的过程。服务器能够接受多个客户端连接，并能区分不同客户端的数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/****************************例 **********************************/

LRESULT CVoltaProjDlg::OnServerMessage(WPARAM wParam, LPARAM lParam)

{

SOCKET socket;

int length,i;

ID_CARD_INFO serverdata;

switch(lParam)

{

case FD_ACCEPT:

socket= accept(m_server.m_hSocket,NULL,NULL);

m_server.m_connectionList.AddHead(socket);

return 0;

case FD_READ://接收到客户端信息

length=m_server.m_connectionList.GetCount();

for(i=0;i<length;i++)//循环寻找向服务器发送数据的客户端

{

socket=m_server.m_connectionList.GetAt(m_server.m_connectionList.FindIndex(i));

if(socket==wParam)//找到客户端

{

recv(socket,(char*)&serverdata,DATASIZE,0);//服务器接收客户端发送的数据--s

if((cmd.num!=serverdata.num) || (cmd.on!= serverdata.on))

{

cmd.num=serverdata.num;

cmd.on= serverdata.on;

}

return 0;

}

return 0;

case FD_WRITE:

return 0;

case FD_CLOSE:

AfxMessageBox("主机连接异常关闭！请重新启动主机程序，再启动客户机程序！");

return 0;

default:

return 0;

}

/**************************** Socket.h ***********************************/

#include <math.h>

#include "atlbase.h"

#include <afxtempl.h>

#include "winsock.h"

#include "afxsock.h"

#include <winnetwk.h>

#pragma comment(lib, "mpr.lib")

#pragma comment(lib, "wsock32.lib")

typedef CList <SOCKET,SOCKET&> SOCKET_ARRAY;

typedef struct cardinfo

{

int num;

BOOL on;

BOOL bShutFlag;

}ID_CARD_INFO,*ID_CARD_INFO_PTR;

#define DATASIZE sizeof(ID_CARD_INFO)

#define SER_MESSAGE WM_USER + 100 // 自定义消息

#define CLI_MESSAGE WM_USER + 101

// 在对话框上加上自定义消息响应

// ON_MESSAGE(SER_MESSAGE,OnServerMessage)

class CServer

{

public:

CServer();

virtual ~CServer();

private:

void ServerInit();

public:

BOOL InitAndListen(HWND hwnd,UINT port);

static int GetLocalHostName(CString &sHostName); //获得本地计算机名称

static int GetIpAddress(const CString &sHostName, CString &sIpAddress);//获得本地IP

static int GetIpAddress(const CString &sHostName, BYTE &f0,BYTE &f1,BYTE &f2,BYTE &f3);//获得本地IP

BOOL CreateServer(HWND hwnd,UINT port);

public:

SOCKET m_hSocket;

sockaddr_in m_addr;

UINT m_uPort;

HWND m_hWnd;

SOCKET_ARRAY m_connectionList;//客户机IP地址的第四个值及连接成功标志结构

};

class CClient

{

public:

CClient();

virtual ~CClient();

private:

void ClientInit();

public:

BOOL InitAndConnet(HWND hwnd,UINT port,CString strIP);

static int GetLocalHostName(CString &sHostName); //获得本地计算机名称

static int GetIpAddress(const CString &sHostName, CString &sIpAddress);//获得本地IP

static int GetIpAddress(const CString &sHostName, BYTE &f0,BYTE &f1,BYTE &f2,BYTE &f3);//获得本地IP

BOOL Enumerate(LPNETRESOURCE lpNetRC_p,int depth);

BOOL ConnectServer(HWND hwnd ,UINT port,CString strIP);

public:

void SendData(ID_CARD_INFO &commdata);

void RevData(ID_CARD_INFO commdata);

void GetString(CString & str);

void SendString(CString a);

SOCKET m_hSocket;

sockaddr_in m_addr;

UINT m_uPort;

HWND m_hWnd;

CStringList m_List;

CStringList m_NeighborList;

CString computer[50];

int computer_count;

int count;

};

/**************************** Socket.Cpp **********************/

CServer::CServer() // 初始化版本库

{

m_hSocket=NULL;

WSADATA wsaData;

WORD version = MAKEWORD(2, 0);

int ret = WSAStartup(version, &wsaData);

if(ret != 0)

{

TRACE("Initilize Error!/n"); //初始化失败网络

AfxMessageBox("Failed in initial socket");

}

m_connectionList.RemoveAll();

ASSERT(m_connectionList.GetCount()==0);

}

void CServer::ServerInit() // 设置为非堵塞套接字

// SER_MESSAGE 为自定义消息，指定接收到以下事件时发送// 自定义消息

{

if(WSAAsyncSelect(m_hSocket, m_hWnd, SER_MESSAGE, FD_ACCEPT|FD_READ|FD_WRITE|FD_CLOSE)>0)

AfxMessageBox("error select");

}

BOOL CServer::InitAndListen(HWND hwnd,UINT port)

{

m_uPort=port;

m_hWnd=hwnd;

if(m_hSocket != NULL)

{

//如果已经创建了套接字先关闭原来的

closesocket(m_hSocket);

m_hSocket = NULL;

}

if(m_hSocket == NULL)

{

//创建新的套接字，这里是创建的流类型的套接字

m_hSocket = socket(AF_INET, SOCK_STREAM,0);

ASSERT(m_hSocket != NULL);

ServerInit();

}

m_addr.sin_family = AF_INET;

m_addr.sin_addr.S_un.S_addr = INADDR_ANY;

m_addr.sin_port = htons(m_uPort);

int ret = 0;

int error = 0;

//绑定一个套接字到本机的地址

ret = bind(m_hSocket, (LPSOCKADDR)&m_addr, sizeof(m_addr));

if(ret == SOCKET_ERROR)

{ //绑定错误

AfxMessageBox("Binding Error");

return FALSE;

}

//开始一个侦听过程，等待客户的连接

ret = listen(m_hSocket, 10); //这里的第二个参数表示最多支持的客户连接数

if(ret == SOCKET_ERROR)

{ //listen失败

AfxMessageBox("Listen Error");

return FALSE;

}

return TRUE;

}

BOOL CServer::CreateServer(HWND hwnd,UINT port)

{

if(InitAndListen(hwnd,port)==FALSE)

{

return FALSE;

}

else//把本机设为服务器

{

return TRUE;

}