server:
// TcpServer.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <WINSOCK2.H>
#include <iostream>
#pragma comment(lib,"WS2_32.lib")
#define PORT 9990
#define DATA_BUFSIZE 8192
// 定义套接字信息
typedef struct _SOCKET_INFORMATION {
CHAR Buffer[DATA_BUFSIZE]; // 发送和接收数据的缓冲区
WSABUF DataBuf; // 定义发送和接收数据缓冲区的结构体,包括缓冲区的长度和内容
SOCKET Socket; // 与客户端进行通信的套接字
DWORD BytesSEND; // 保存套接字发送的字节数
DWORD BytesRECV; // 保存套接字接收的字节数
} SOCKET_INFORMATION, * LPSOCKET_INFORMATION;
DWORD TotalSockets = 0; // 记录正在使用的套接字总数量
LPSOCKET_INFORMATION SocketArray[FD_SETSIZE]; // 保存Socket信息对象的数组,FD_SETSIZE表示SELECT模型中允许的最大套接字数量
// 创建SOCKET信息
BOOL CreateSocketInformation(SOCKET s)
{
LPSOCKET_INFORMATION SI; // 用于保存套接字的信息
// printf("Accepted socket number %d\n", s); // 打开已接受的套接字编号
// 为SI分配内存空间
if ((SI = (LPSOCKET_INFORMATION) GlobalAlloc(GPTR,
sizeof(SOCKET_INFORMATION))) == NULL)
{
printf("GlobalAlloc() failed with error %d\n", GetLastError());
return FALSE;
}
// 初始化SI的值
SI->Socket = s;
SI->BytesSEND = 0;
SI->BytesRECV = 0;
// 在SocketArray数组中增加一个新元素,用于保存SI对象
SocketArray[TotalSockets] = SI;
TotalSockets++; // 增加套接字数量
return(TRUE);
}
// 从数组SocketArray中删除指定的LPSOCKET_INFORMATION对象
void FreeSocketInformation(DWORD Index)
{
LPSOCKET_INFORMATION SI = SocketArray[Index]; // 获取指定索引对应的LPSOCKET_INFORMATION对象
DWORD i;
// 关闭套接字
closesocket(SI->Socket);
//printf("Closing socket number %d\n", SI->Socket);
// 释放指定LPSOCKET_INFORMATION对象资源
GlobalFree(SI);
// 将数组中index索引后面的元素前移
for (i = Index; i < TotalSockets; i++)
{
SocketArray[i] = SocketArray[i+1];
}
TotalSockets--; // 套接字总数减1
}
// 主函数,启动服务器
int _tmain(int argc, _TCHAR* argv[])
{
SOCKET ListenSocket; // 监听套接字
SOCKET AcceptSocket; // 与客户端进行通信的套接字
SOCKADDR_IN InternetAddr; // 服务器的地址
WSADATA wsaData; // 用于初始化套接字环境
INT Ret; // WinSock API的返回值
FD_SET WriteSet; // 获取可写性的套接字集合
FD_SET ReadSet; // 获取可读性的套接字集合
DWORD Total = 0; // 处于就绪状态的套接字数量
DWORD SendBytes; // 发送的字节数
DWORD RecvBytes; // 接收的字节数
// 初始化WinSock环境
if ((Ret = WSAStartup(0x0202,&wsaData)) != 0)
{
printf("WSAStartup() failed with error %d\n", Ret);
WSACleanup();
return -1;
}
// 创建用于监听的套接字
if ((ListenSocket = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0,
WSA_FLAG_OVERLAPPED)) == INVALID_SOCKET)
{
printf("WSASocket() failed with error %d\n", WSAGetLastError());
return -1;
}
// 设置监听地址和端口号
InternetAddr.sin_family = AF_INET;
InternetAddr.sin_addr.s_addr = htonl(INADDR_ANY);
InternetAddr.sin_port = htons(PORT);
// 绑定监听套接字到本地地址和端口
if(bind(ListenSocket, (PSOCKADDR)&InternetAddr, sizeof(InternetAddr)) == SOCKET_ERROR)
{
printf("bind() failed with error %d\n", WSAGetLastError());
return -1;
}
// 开始监听
if (listen(ListenSocket, 5))
{
printf("listen() failed with error %d\n", WSAGetLastError());
return -1;
}
/* // 设置为非阻塞模式
ULONG NonBlock = 1;
if(ioctlsocket(ListenSocket, FIONBIO, &NonBlock) == SOCKET_ERROR)
{
printf("ioctlsocket() failed with error %d\n", WSAGetLastError());
return -1;
}
// 为ListenSocket套接字创建对应的SOCKET_INFORMATION
// 这样就可以把ListenSocket添加到SocketArray数组中
*/
CreateSocketInformation(ListenSocket);
while(TRUE)
{
// 准备用于网络I/O通知的读/写套接字集合
FD_ZERO(&ReadSet);
FD_ZERO(&WriteSet);
// 向ReadSet集合中添加监听套接字ListenSocket
FD_SET(ListenSocket, &ReadSet);
// 将SocketArray数组中的所有套接字添加到WriteSet和ReadSet集合中
// SocketArray数组中保存着监听套接字和所有与客户端进行通信的套接字
// 这样就可以使用select()判断哪个套接字有接入数据或者读取/写入数据
for (DWORD i=0; i<TotalSockets; i++)
{
LPSOCKET_INFORMATION SocketInfo = SocketArray[i];
FD_SET(SocketInfo->Socket, &WriteSet);
FD_SET(SocketInfo->Socket, &ReadSet);
}
// 判断读/写套接字集合中就绪的套接字
if((Total = select(0, &ReadSet, &WriteSet, NULL, NULL)) == SOCKET_ERROR)
{
printf("select() returned with error %d\n", WSAGetLastError());
return -1;
}
// 依次处理所有套接字。本服务器是一个回应服务器,即将从客户端收到的字符串再发回到客户端。
for (DWORD i=0; i<TotalSockets; i++)
{
LPSOCKET_INFORMATION SocketInfo = SocketArray[i]; // SocketInfo为当前要处理的套接字信息
// 判断当前套接字的可读性,即是否有接入的连接请求或者可以接收数据
if (FD_ISSET(SocketInfo->Socket, &ReadSet))
{
if(SocketInfo->Socket == ListenSocket) // 对于监听套接字来说,可读表示有新的连接请求
{
Total--; // 就绪的套接字减1
// 接受连接请求,得到与客户端进行通信的套接字AcceptSocket
if((AcceptSocket = accept(ListenSocket, NULL, NULL)) != INVALID_SOCKET)
{
printf("连接成功!\n");
/*
// 设置套接字AcceptSocket为非阻塞模式
// 这样服务器在调用WSASend()函数发送数据时就不会被阻塞
NonBlock = 1;
if(ioctlsocket(AcceptSocket, FIONBIO, &NonBlock) == SOCKET_ERROR)
{
printf("ioctlsocket() failed with error %d\n", WSAGetLastError());
return -1;
}
*/
// 创建套接字信息,初始化LPSOCKET_INFORMATION结构体数据,将AcceptSocket添加到SocketArray数组中
if(CreateSocketInformation(AcceptSocket) == FALSE)
return -1;
}
else
{
if(WSAGetLastError() != WSAEWOULDBLOCK)
{
printf("accept() failed with error %d\n", WSAGetLastError());
return -1;
}
}
}
else // 接收数据
{
// 如果当前套接字在ReadSet集合中,则表明该套接字上有可以读取的数据
if (FD_ISSET(SocketInfo->Socket, &ReadSet))
{
Total--; // 减少一个处于就绪状态的套接字
memset(SocketInfo->Buffer, ' ', DATA_BUFSIZE);
ZeroMemory(SocketInfo->DataBuf.buf,DATA_BUFSIZE);// 初始化缓冲区
SocketInfo->DataBuf.buf = SocketInfo->Buffer; // 初始化缓冲区位置
SocketInfo->DataBuf.len = DATA_BUFSIZE; // 初始化缓冲区长度
// 接收数据
DWORD Flags = 0;
if(WSARecv(SocketInfo->Socket, &(SocketInfo->DataBuf), 1, &RecvBytes, &Flags,
NULL, NULL) == SOCKET_ERROR)
{
// 错误编码等于WSAEWOULDBLOCK表示暂没有数据,否则表示出现异常
if(WSAGetLastError() != WSAEWOULDBLOCK)
{
printf("WSARecv() failed with error %d\n", WSAGetLastError());
FreeSocketInformation(i); // 释放套接字信息
}
continue;
}
else // 接收数据
{
SocketInfo->BytesRECV = RecvBytes; // 记录接收数据的字节数
if(RecvBytes == 0) // 如果接收到0个字节,则表示对方关闭连接
{
FreeSocketInformation(i);
continue;
}
else // 如果成功接收数据,则打印收到的数据
{
printf(SocketInfo->DataBuf.buf);
printf("\n");
//
}
}
}
}
}
else
{
// 如果当前套接字在WriteSet集合中,则表明该套接字的内部数据缓冲区中有数据可以发送
if(FD_ISSET(SocketInfo->Socket, &WriteSet))
{
Total--; // 减少一个处于就绪状态的套接字
SocketInfo->DataBuf.buf = SocketInfo->Buffer + SocketInfo->BytesSEND; // 初始化缓冲区位置
SocketInfo->DataBuf.len = SocketInfo->BytesRECV - SocketInfo->BytesSEND; // 初始化缓冲区长度
if(SocketInfo->DataBuf.len > 0) // 如果有需要发送的数据,则发送数据
{
if(WSASend(SocketInfo->Socket, &(SocketInfo->DataBuf), 1, &SendBytes, 0,
NULL, NULL) == SOCKET_ERROR)
{
// 错误编码等于WSAEWOULDBLOCK表示暂没有数据,否则表示出现异常
if(WSAGetLastError() != WSAEWOULDBLOCK)
{
printf("WSASend() failed with error %d\n", WSAGetLastError());
FreeSocketInformation(i); // 释放套接字信息
}
continue;
}
else
{
SocketInfo->BytesSEND += SendBytes; // 记录发送数据的字节数
// 如果从客户端接收到的数据都已经发回到客户端,则将发送和接收的字节数量设置为0
if (SocketInfo->BytesSEND == SocketInfo->BytesRECV)
{
SocketInfo->BytesSEND = 0;
SocketInfo->BytesRECV = 0;
}
}
}
}
} // 如果ListenSocket未就绪,并且返回的错误不是WSAEWOULDBLOCK(该错误表示没有接收的连接请求),则出现异常
}
}
// 暂停,按任意键退出
system("pause");
return 0;
}
client:
// TcpClient.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include <Winsock2.H>
#include <string>
#include <iostream>
#pragma comment(lib,"WS2_32.lib")
#define BUF_SIZE 64 // 缓冲区大小
int _tmain(int argc, _TCHAR* argv[])
{
WSADATA wsd; // 用于初始化Windows Socket
SOCKET sHost; // 与服务器进行通信的套接字
SOCKADDR_IN servAddr; // 服务器地址
char buf[BUF_SIZE]; // 用于接受数据缓冲区
int retVal; // 调用各种Socket函数的返回值
// 初始化Windows Socket
if(WSAStartup(MAKEWORD(2,2),&wsd) != 0)
{
printf("WSAStartup failed !\n");
return 1;
}
// 创建套接字
sHost = socket(AF_INET,SOCK_STREAM,IPPROTO_IP);
if(INVALID_SOCKET == sHost)
{
printf("socket failed !\n");
WSACleanup();
return -1;
}
// 设置服务器地址
servAddr.sin_family = AF_INET;
servAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1"); // 用户需要根据实际情况修改
servAddr.sin_port = htons(9990); // 在实际应用中,建议将服务器的IP地址和端口号保存在配置文件中
int sServerAddlen = sizeof(servAddr); // 计算地址的长度
// 循环等待
while(true)
{
// 连接服务器
Sleep( 200 );
retVal = connect(sHost,(LPSOCKADDR)&servAddr,sizeof(servAddr));
Sleep( 200 );
if(SOCKET_ERROR == retVal)
{
int err = WSAGetLastError();
if(err == WSAEWOULDBLOCK || err == WSAEINVAL) // 无法立即完成非阻塞套接字上的操作
{
//Sleep(500);
continue;
}
else if(err == WSAEISCONN) // 已建立连接
{
break;
}
else
{
continue;
//printf("connect failed !\n");
//closesocket(sHost);
//WSACleanup();
//return -1;
}
}
}
// 循环向服务器发送字符串,并显示反馈信息。
// 发送quit将使服务器程序退出,同时客户端程序自身也将退出
while(true)
{
// 向服务器发送数据
printf("Please input a string to send: ");
// 接收输入的数据
std::string str;
std::getline(std::cin, str);
// 将用户输入的数据复制到buf中
ZeroMemory(buf,BUF_SIZE);
ZeroMemory(buf,BUF_SIZE);
strcpy(buf,str.c_str());
// 循环等待
while(true)
{
// 向服务器发送数据
retVal = send(sHost,buf,strlen(buf),0);
if(SOCKET_ERROR == retVal)
{
int err = WSAGetLastError();
if(err == WSAEWOULDBLOCK) // 无法立即完成非阻塞套接字上的操作
{
continue;
}
else
{
printf("send failed !\n");
closesocket(sHost);
WSACleanup();
return -1;
}
}
break;
}
while(true)
{
ZeroMemory(buf,BUF_SIZE); // 清空接收数据的缓冲区
retVal = recv(sHost,buf,sizeof(buf)+1,0); // 接收服务器回传的数据
if(SOCKET_ERROR == retVal)
{
int err = WSAGetLastError(); // 获取错误编码
if(err == WSAEWOULDBLOCK) // 接收数据缓冲区暂无数据
{
Sleep(100);
printf("waiting back msg !\n");
continue;
}
else if(err == WSAETIMEDOUT || err == WSAENETDOWN)
{
printf("recv failed !\n");
closesocket(sHost);
WSACleanup();
return -1;
}
break;
}
break;
}
//ZeroMemory(buf,BUF_SIZE); // 清空接收数据的缓冲区
//retVal = recv(sHost,buf,sizeof(buf)+1,0); // 接收服务器回传的数据
printf("Recv From Server: %s\n",buf);
// 如果收到quit,则退出
if(strcmp(buf, "quit") == 0)
{
printf("quit!\n");
break;
}
}
// 释放资源
closesocket(sHost);
WSACleanup();
// 暂停,按任意键继续
system("pause");
return 0;
}