C++用win socket api实现的简单的聊天室程序

#include <cstdio> #include <cstring> #include <vector> #include "winsock2.h" using namespace std; #define HOST_ADDRESS "127.0.0.1" #define HOST_PORT 8888 #define DEFAULT_BUFFER 4096 //客户连接池，保存各个客户的连接 vector<SOCKET> clients; //结构object 保存每个客户端socket 及对应 地址结构sockaddr_in ，保存地址信息是因为显示信息时 //要标明是谁发表的信息 typedef struct { SOCKET sock; sockaddr_in addr; } object; //对于每个客户端的信息处理线程，先获取信息，在通过线程池向各个线程发送信息 DWORD WINAPI clientThread(LPVOID lpParam) { object* obj = (object*)lpParam; SOCKET sock = obj->sock; char buf[DEFAULT_BUFFER] = ""; char sendback[DEFAULT_BUFFER] = ""; int ret, left, idx = 0; //循环处理客户端发送的信息 while(1) { ret = recv(sock, buf, DEFAULT_BUFFER,0); if(ret == 0) return 1; if(ret == SOCKET_ERROR) { printf("receive message failed:%d\n",WSAGetLastError()); return 1; } buf[ret] = '\0'; sprintf(sendback, "[%s said]:%s\n",inet_ntoa(obj->addr.sin_addr),buf); printf("[The message from %s]:%s\n", inet_ntoa(obj->addr.sin_addr), buf); printf("[The sendback message is]: %s\n", sendback); //向各个客户端发送通告信息 for(int i=0; i < clients.size(); i++) { left = strlen(sendback); while(left > 0) { ret = send(clients[i], &sendback[idx], left, 0); if(ret == 0) return 1; else if( ret == SOCKET_ERROR) { printf("send back message failed:%d\n",WSAGetLastError()); return 1; } left -= ret; idx += ret; } idx = 0; } memset(buf, 0, DEFAULT_BUFFER); memset(sendback,0, DEFAULT_BUFFER); } return 0; } int main() { WSADATA data; SOCKET listenSocket; sockaddr_in serverAddr; sockaddr_in clientAddr; SOCKET client; HANDLE hThread; DWORD threadId; int addrLen; //加载winsocket动态链接库 int result = WSAStartup(MAKEWORD(2,2), &data); if(result != NO_ERROR) { printf("Load winsocket dll wrong\n"); return 1; } //建立服务器端监听socket listenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); if(listenSocket == INVALID_SOCKET) { printf("Create listen socket failed:%d\n", WSAGetLastError()); WSACleanup(); return 1; } //服务器端socket 绑定监听端口和地址 serverAddr.sin_family = AF_INET; serverAddr.sin_addr.s_addr = inet_addr(HOST_ADDRESS); serverAddr.sin_port = htons(HOST_PORT); if(bind(listenSocket,(sockaddr*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) { printf("bind error:%s\n",WSAGetLastError()); closesocket(listenSocket); WSACleanup(); return 1; } //监听端口 listen(listenSocket, 10); //阻塞获取客户socket while(1) { addrLen = sizeof(clientAddr); client = accept(listenSocket, (struct sockaddr*)&clientAddr, &addrLen); clients.push_back(client); printf("num of client is %d\n", clients.size()); if(client == INVALID_SOCKET) { printf("create client socket failed:%d\n",WSAGetLastError()); break; } object obj; obj.addr = clientAddr; obj.sock = client; //为每个客户建立处理线程 hThread = CreateThread(NULL,0,clientThread,(LPVOID)&obj, 0, &threadId); if(hThread == NULL) { printf("create thread failed:%d\n",WSAGetLastError()); break; } } closesocket(listenSocket); WSACleanup(); getchar(); return 0; }