套接口复用实现的聊天室的程序_其他_编程开发_程序员俱乐部

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

/*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;
}