windows socket编程入门示例2

// tcpSocketServer.cpp
#include <stdio.h>
#include <winsock2.h>
#include <vector>

#pragma comment(lib,"ws2_32.lib")
#define IP_ADDRESS "127.0.0.1"
#define PORT 5678

static DWORD WINAPI RecvThread(void *Param);
static DWORD WINAPI SendThread(void *Param);

void main()
{
WORD wVersionRequested;
WSADATA wsaData;
int retCode;
SOCKET socketServer;
SOCKADDR_IN addrSrv;
SOCKADDR_IN addrClient;
int len = sizeof(SOCKADDR);
SOCKET socketConnection;
std::vector<HANDLE> threadHandleSet;
std::vector<SOCKET> clientSocketSet;

wVersionRequested = MAKEWORD(2, 2);
retCode = WSAStartup(wVersionRequested, &wsaData);
if (retCode != 0)
{
fprintf(stderr, "Load WSADTATA failed!\n");
return;
}
if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)
{
fprintf(stderr, "WSADTATA version error!\n");
WSACleanup();
return;
}

socketServer = socket(AF_INET, SOCK_STREAM, 0);//创建套接字
if (socketServer == INVALID_SOCKET)
{
fprintf(stderr, "invalid socket !\n");
WSACleanup();
return;
}

//addrSrv.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
addrSrv.sin_addr.S_un.S_addr = inet_addr(IP_ADDRESS);
addrSrv.sin_family = AF_INET;//将sin_family字段设置为AF_INET，代表WinSock使用的是IP地址族
addrSrv.sin_port = htons(PORT);//端口号

retCode = bind(socketServer, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR));//绑定套接字到一个IP地址和一个端口上
if (retCode == SOCKET_ERROR)
{
printf("bind error !");
closesocket(socketServer);
WSACleanup();
return;
}

retCode = listen(socketServer, 2);//将套接字设置为监听模式等待连接请求
if (retCode == SOCKET_ERROR)
{
printf("listen error !");
closesocket(socketServer);
WSACleanup();
return;
}

printf("Tcp server is running....\n");
while (1)
{
//请求到来后，接收连接请求，返回一个新的对应于此次连接的套接字
socketConnection = accept(socketServer, (SOCKADDR*)&addrClient, &len);
if (socketConnection == INVALID_SOCKET)
{
printf("accept error, continue to accept new client !");
continue;
}

printf("接受到一个连接：%s \r\n", inet_ntoa(addrClient.sin_addr));
clientSocketSet.push_back(socketConnection);

//创建一个子线程调用RecvThread函数，专门用来接收消息
HANDLE recvThreadHandle = CreateThread(NULL, 0, RecvThread, &socketConnection, 0, NULL);
if (recvThreadHandle == NULL)
{
fprintf(stderr, "Create recive data thread failed!\n");
}
threadHandleSet.push_back(recvThreadHandle);

HANDLE sendThreadHandle = CreateThread(NULL, 0, SendThread, &socketConnection, 0, NULL);
if (sendThreadHandle == NULL)
{
fprintf(stderr, "Create recive data thread failed!\n");
}
threadHandleSet.push_back(sendThreadHandle);
}

fprintf(stderr, "main thread to wait sub thread to finish.\n");
for (int i = 0; i < threadHandleSet.size(); i++)
{
if (threadHandleSet[i])
{
WaitForSingleObject(threadHandleSet[i], INFINITE);
CloseHandle(threadHandleSet[i]);
fprintf(stderr, "thread %d close()\n", i + 1);
}
}

for (int i = 0; i < clientSocketSet.size(); i++)
{
closesocket(clientSocketSet[i]);
fprintf(stderr, "socket %d close.\n", i + 1);
}

closesocket(socketServer);
WSACleanup();//关闭加载的套接字库
}

static DWORD WINAPI RecvThread(void *Param){
SOCKET socketConnection = *((SOCKET *)Param); //设置socket
char recvBuf[1024];
int ret;
while (1)
{
ret = recv(socketConnection, recvBuf, sizeof(recvBuf), 0);
if (ret > 0)
{
recvBuf[ret] = 0x00;
fprintf(stdout, "recivce data from client success!\n");
printf("客户端说:%s\n", recvBuf);
}
else if (ret == 0 || (ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET))
{
fprintf(stderr, "Client has disconneted, recive data failed!\n");
break;
}
else if (ret == SOCKET_ERROR && WSAGetLastError() == WSAEWOULDBLOCK)
{
fprintf(stdout, "非阻塞类型返回\n");
break;
}
}

fprintf(stderr, "接收数据子线程退出\n");
return 0;
}

static DWORD WINAPI SendThread(void *Param)
{
SOCKET socketConnection = *((SOCKET *)Param);
int retCode;
char sendBuf[128];

while (1)
{
memset(sendBuf, 0, sizeof(sendBuf));
printf("输入要发送给客户端的消息：\n");
fgets(sendBuf, sizeof(sendBuf), stdin);
int length = strlen(sendBuf);
if (length < 2)
{
printf("不能发送空消息.\n");
continue;
}
retCode = send(socketConnection, sendBuf, length, 0);

if (retCode == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET)
{
fprintf(stderr, "Client has disconneted, send data to client failed!\n");
break;
}
else if (retCode == SOCKET_ERROR)
{
printf("Send data to client failed!\n");
break;
}
if (retCode != SOCKET_ERROR)
{
printf("Send data to client finished.\n");
}
}

fprintf(stderr, "发送数据子线程退出\n");
return 0;
}

// tcpSocketClient.cpp
#include <stdio.h>
#include <winsock2.h>

#pragma comment(lib,"ws2_32.lib")
#define IP_ADDRESS "127.0.0.1"
#define PORT 5678

static DWORD WINAPI RecvThread(void *Param);

void main( )
{
WORD wVersionRequested;
WSADATA wsaData;
SOCKET socketClient;
SOCKADDR_IN addrSrv;
char sendBuf[50];
int retCode;

wVersionRequested = MAKEWORD(2, 2);
retCode = WSAStartup(wVersionRequested, &wsaData);
if (retCode != 0)
{
fprintf(stderr, "Load WSADTATA failed!\n");
return;
}
if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2){
fprintf(stderr, "WSADTATA version error!\n");
WSACleanup();
return;
}

socketClient = socket(AF_INET, SOCK_STREAM, 0);
if (socketClient == INVALID_SOCKET)
{
fprintf(stderr, "invalid socket !\n");
WSACleanup();
return;
}

addrSrv.sin_addr.S_un.S_addr = inet_addr(IP_ADDRESS);//IP地址
addrSrv.sin_family = AF_INET; //将sin_family字段设置为AF_INET，代表WinSock使用的是IP地址族
addrSrv.sin_port = htons(PORT); //端口号

//向服务器端发出连接请求
retCode = connect(socketClient, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR));
if (retCode == SOCKET_ERROR)
{
fprintf(stderr, "Connet server error!%d\n", GetLastError());
closesocket(socketClient);
WSACleanup();
return;
}

//创建子线程调用RecvThread函数，专门用来接收服务器端发过来的消息
HANDLE recvDataThreadHandle = CreateThread(NULL, 0, RecvThread, &socketClient, 0, NULL);
if (recvDataThreadHandle == NULL)
fprintf(stderr, "Create recive data thread failed!\n");

//发送消息在主线程当中实现
while (1)
{
memset(sendBuf, 0, sizeof(sendBuf));
printf("请输入要发给服务器端的消息:\n");
fgets(sendBuf, sizeof(sendBuf), stdin);
int length = strlen(sendBuf);
if (length < 2)
{
printf("不能发送空消息.\n");
continue;
}

retCode = send(socketClient, sendBuf, strlen(sendBuf) + 1, 0);//发送消息
if (retCode == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET)
{
fprintf(stderr, "Server has disconneted, send data to server failed!\n");
break;
}
else if (retCode == SOCKET_ERROR)
{
printf("Send data to server failed!\n");
break;
}
if (retCode != SOCKET_ERROR)
{
printf("Send data to server finished.\n");
}
}

if (recvDataThreadHandle)
{
/// 同步子线程，再关闭
WaitForSingleObject(recvDataThreadHandle, INFINITE);
CloseHandle(recvDataThreadHandle);
}

closesocket(socketClient);
WSACleanup();//关闭加载的套接字库
}

static DWORD WINAPI RecvThread(void *Param)
{
SOCKET socketConnection = *((SOCKET *)Param);
char recvBuf[1024];
int ret;

while (1)
{
ret = recv(socketConnection, recvBuf, sizeof(recvBuf), 0);
if (ret > 0)
{
recvBuf[ret] = 0x00;
fprintf(stdout, "recivce data from server success!\n");
printf("服务器说:%s\n", recvBuf);
}
else if (ret == 0 || (ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET))
{
fprintf(stderr, "Server has disconneted, recive data failed!\n");
break;
}
else if (ret == SOCKET_ERROR && WSAGetLastError() == WSAEWOULDBLOCK)
{
fprintf(stdout, "非阻塞类型返回\n");
break;
}
}

return 0;
}