套接口复用实现的聊天室的程序

//只要按要求将程序中的注释部份改为注释所述就可以运行了

/*VC下编写多线程程序除了包含头文件
#include <process.h>
一定要记住如下的设定，不然编译会出错的：//Project->Settings->C/C++->Category：Code Generation，Use run-time library：Multithreaded */

//TCP:client.cpp

#pragma comment(lib, "ws2_32.lib")
#include <winsock2.h>
#include <stdio.h>
#include <process.h>
#define MSG_MAX_SIZE 1024

void __cdecl Get_keybeard_input(void*param)
{
SOCKET Sock=(SOCKET)param;
while(1)
{
char info[MSG_MAX_SIZE];
gets(info);
int ret=send(Sock,info,MSG_MAX_SIZE,0);
if(ret==SOCKET_ERROR)
{
printf("send()failed:%d/n",WSAGetLastError());
closesocket(Sock);
WSACleanup();
continue;
}
}
}

int main()
{
WSADATA wsaData;
int iResult = WSAStartup(MAKEWORD(2,2), &wsaData);
if (iResult != 0)
{
printf("Error at WSAStartup()/n");
return -1;
}

SOCKET Sock;
Sock = socket(AF_INET, SOCK_STREAM, 0);
if (Sock == INVALID_SOCKET)
{
printf("Error at socket(): %ld/n", WSAGetLastError());
WSACleanup();
return -1;
}

sockaddr_in service;
service.sin_family = AF_INET;
service.sin_port = htons(27015);
service.sin_addr.s_addr = inet_addr("ip"); //将此处的IP改为服务器程序运行的主机IP(用点分十进制)

int err=connect(Sock,(SOCKADDR*) &service, sizeof(service));
if(err==SOCKET_ERROR)
{
printf("connect() failed: %ld/n", WSAGetLastError());
closesocket(Sock);
WSACleanup();
return -1;
}

_beginthread(Get_keybeard_input,0,(void*)Sock);
char recv_buf[MSG_MAX_SIZE];
while(1)
{
memset(recv_buf,0,MSG_MAX_SIZE);
int rlen=recv(Sock,recv_buf,MSG_MAX_SIZE,0);
if(rlen==0) break;
if(rlen==SOCKET_ERROR)
{
printf("recv() failed: %ld/n", WSAGetLastError());
closesocket(Sock);
WSACleanup();
return -1;
}
puts(recv_buf);
}
closesocket(Sock);
WSACleanup();
return 0;
}

//TCP:server.cpp

#pragma comment(lib, "ws2_32.lib")
#include <winsock2.h>
#include <stdio.h>
int main()
{
WSADATA wsaData;
int iResult = WSAStartup(MAKEWORD(2,2), &wsaData);
if (iResult != 0)
{
printf("Error at WSAStartup()/n");
return -1;
}

SOCKET ListenSocket;
ListenSocket = socket(AF_INET, SOCK_STREAM, 0);
if (ListenSocket == INVALID_SOCKET)
{
printf("Error at socket(): %ld/n", WSAGetLastError());
WSACleanup();
return -1;
}

sockaddr_in service;
service.sin_family = AF_INET;
service.sin_port = htons(27015);
service.sin_addr.s_addr = inet_addr("IP"); //将此处的IP改为的主机IP(用点分十进制表示)

if (bind( ListenSocket, (SOCKADDR*) &service, sizeof(service)) == SOCKET_ERROR)
{
printf("bind() failed: %ld/n", WSAGetLastError());
closesocket(ListenSocket);
WSACleanup();
return -1;
}
if (listen( ListenSocket, 1 ) == SOCKET_ERROR)
{
printf("listen() failed: %ld/n", WSAGetLastError());
closesocket(ListenSocket);
WSACleanup();
return -1;
}

SOCKET client[FD_SETSIZE - 1];
for(int i = 0;i < FD_SETSIZE;i++)
{
client[i] =INVALID_SOCKET;
}

fd_set rset, allset; //allset--所有套接口， rset--可读套接口
FD_ZERO(&allset);
FD_SET(ListenSocket, &allset);

#define MSG_MAX_SIZE 1024
char buf[FD_SETSIZE - 1][MSG_MAX_SIZE];

for(;;)
{
rset = allset; //structure assignment
int ret = select(0,&rset,NULL,NULL,NULL);
if (ret == SOCKET_ERROR)
{
printf("select() failed: %ld/n", WSAGetLastError());
break;
}
if (ret == 0) {
continue;
}
if(FD_ISSET(ListenSocket, &rset))
{
struct sockaddr_in cliaddr;
int clilen = sizeof(cliaddr);
SOCKET sockConn = accept(ListenSocket,(sockaddr *)&cliaddr, &clilen);
if (sockConn == SOCKET_ERROR)
{
printf("accept() failed: %ld/n", WSAGetLastError());
break; //continue
}
for(i = 0; i < FD_SETSIZE - 1; i++)
{
if(client[i] == INVALID_SOCKET)
{
client[i] = sockConn;
break;
}
} //for end
if(i < FD_SETSIZE - 1)
{
FD_SET(sockConn, &allset);
}
else{
printf("too many clients");
closesocket(sockConn);
}
//为了理解容易，可以删除下面4行
ret --;
if(ret <= 0)
{
continue;
}
}// if (FD_ISSET(ListenSocket, &rset)) end

for(i = 0; i < FD_SETSIZE - 1; i++)
{
if ((client[i] != INVALID_SOCKET) &&
FD_ISSET(client[i], &rset))
{
memset(buf[i], 0 , MSG_MAX_SIZE);
int nByte = recv(client[i], buf[i], MSG_MAX_SIZE, 0);
if (nByte == SOCKET_ERROR)
{
printf("recv() failed: %ld/n", WSAGetLastError());
FD_CLR(client[i],&allset);
closesocket(client[i]);
client[i] = INVALID_SOCKET;
continue;
}
if (nByte == 0)
{
FD_CLR(client[i],&allset);
closesocket(client[i]);
client[i] = INVALID_SOCKET;
continue;
}
//do the request
for(int j=0;j<FD_SETSIZE-1;j++)
{
if(i!=j)
int ret=send(client[j],buf[i],MSG_MAX_SIZE,0);
if(ret==SOCKET_ERROR)
{
printf("send()failed:%d/n",WSAGetLastError());
continue;
}

}
}
}//loop client end
}// for(;;) end
closesocket(ListenSocket);
WSACleanup();
return 0;