C++实现的Linux多线程聊天室

/* * Server.cpp * * Created on: 2012-10-16 * Author: blackhole */ #include "Server.h" #include "malloc.h" list<int> Server::clientlist; //服务器 int Server::CreateThread(void *(*fun)(void *), void *__restrict arg) { int res = 0; pthread_t id; res = pthread_create(&id, NULL, fun, arg); return res; } void *Server::SendMessage(void *arg) { printf("Now send...\n"); int len; char *senddata = (char *) arg; list<int>::iterator i; for (i = clientlist.begin(); i!= clientlist.end(); ++i) { len = send(*i, senddata, strlen(senddata), 0); if (-1 == len) { printf("Failed to send"); } printf("发送:%s\n", senddata); } return NULL; } void *Server::RecvMessage(void *connfd) { int len; char recv_buf[1024]; while (1) { memset(recv_buf, 0, sizeof(recv_buf)); printf("Now recv...\n"); len = recv(*(int*) connfd, recv_buf, sizeof(recv_buf), 0); printf("接受%d个字符\n", len); if (-1 == len) { printf("Failed to recv"); } if (0 == len) { printf("break %d\n", *(int*) connfd); list<int>::iterator ii; for (ii = clientlist.begin(); ii != clientlist.end(); ++ii) { if (*ii == *((int *)connfd)) { close(*((int *)connfd)); clientlist.erase(ii); break; } } break; } printf("Receive message:%s\n", recv_buf); SendMessage((void *) recv_buf); } return ((void *) 0); } int Server::StartServerByIpPort(const char* ip, int port) { int res = 0; lisenfd = socket(AF_INET, SOCK_STREAM, 0); if (-1 == lisenfd) { perror("Failed to socket"); return -1; } memset(&ser_addr, 0, sizeof(ser_addr)); ser_addr.sin_family = AF_INET; ser_addr.sin_port = htons(port); //这里输入服务器端口号 ser_addr.sin_addr.s_addr = inet_addr(ip); //htonl(INADDR_ANY); //INADDR_ANY表示本机IP if (-1 == bind(lisenfd, (struct sockaddr *) &ser_addr, sizeof(ser_addr))) { perror("Failed to bind"); return -2; } printf("Now listen...\n"); if (-1 == listen(lisenfd, 10)) { perror("Failed to listen"); return -3; } while (1) { memset(&cli_addr, 0, sizeof(cli_addr)); unsigned int cli_len; cli_len = sizeof(cli_addr); printf("Now accept...\n"); int *connfd = (int *) malloc(sizeof(int)); *connfd = accept(lisenfd, (struct sockaddr *) &cli_addr, &cli_len); if (-1 == *connfd) { perror("Failed to accept"); return -4; } clientlist.push_front(*connfd); pthread_t recvdid; int ctres = pthread_create(&recvdid, NULL, RecvMessage,(void *) connfd); printf("创建接受线程状态%d\n", ctres); if (ctres != 0) { printf("Failed to create thread!"); } } return res; } void Server::StopServer() { list<int>::iterator i; for (i = clientlist.begin(); i != clientlist.end(); ++i) { close(*i); } clientlist.clear(); close(lisenfd); } Server::~Server() { StopServer(); }