本节通过socket实现一个简单的聊天室功能
聊天室中如果有人说话，则服务器负责将内容传送给聊天室的其他人
那么就需要客户端和服务端两个程序，客户端负责发送消息，服务端负责接收和转发消息
客户端代码：
#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/shm.h>
#define MYPORT 8606
#define BUFFER_SIZE 1024
int main()
{
int sock_cli;
fd_set rfds;
struct timeval tv;
int retval, maxfd;
///定义sockfd
sock_cli = socket(AF_INET,SOCK_STREAM, 0);
///定义sockaddr_in
struct sockaddr_in servaddr;
memset(&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(MYPORT); ///服务器端口
servaddr.sin_addr.s_addr = inet_addr("127.0.0.1"); ///服务器ip
//连接服务器，成功返回0，错误返回-1
if (connect(sock_cli, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0)
{
perror("connect");
exit(1);
}
while(1){
/*把可读文件描述符的集合清空*/
FD_ZERO(&rfds);
/*把标准输入的文件描述符加入到集合中*/
FD_SET(0, &rfds);
maxfd = 0;
/*把当前连接的文件描述符加入到集合中*/
FD_SET(sock_cli, &rfds);
/*找出文件描述符集合中最大的文件描述符*/
if(maxfd < sock_cli)
maxfd = sock_cli;
/*设置超时时间*/
tv.tv_sec = 10;
tv.tv_usec = 0;
/*等待聊天*/
retval = select(maxfd+1, &rfds, NULL, NULL, &tv);
if(retval == -1){
printf("select出错，客户端程序退出\n");
break;
}else if(retval == 0){
//printf("客户端没有任何输入信息，并且服务器也没有信息到来，waiting...\n");
continue;
}else{
/*服务器发来了消息*/
if(FD_ISSET(sock_cli,&rfds)){
char recvbuf[BUFFER_SIZE];
int len;
len = recv(sock_cli, recvbuf, sizeof(recvbuf),0);
if(len > 0)
{
printf("%s",recvbuf);
}
memset(recvbuf, 0, sizeof(recvbuf));
}
/*用户输入信息了,开始处理信息并发送*/
if(FD_ISSET(0, &rfds)){
char sendbuf[BUFFER_SIZE];
fgets(sendbuf, sizeof(sendbuf), stdin);
send(sock_cli, sendbuf, strlen(sendbuf),0); //发送
memset(sendbuf, 0, sizeof(sendbuf));
}
}
}
close(sock_cli);
return 0;
}服务端代码：
#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/shm.h>
#include <iostream>
#include <thread>
#include <list>
#include <iostream>
#define PORT 8606
#define IP "127.0.0.1"
int mSocket;
struct sockaddr_in servaddr;
socklen_t len;
std::list<int> connList;//用list来存放套接字，没有限制套接字的容量就可以实现一个server跟若干个client通信
void sendMsg(int sender,char* msg,int length)
{
char buf[1024];
memset(buf, 0 ,sizeof(buf));
sprintf(buf,"玩家%d:%s",sender ,msg);
std::list<int>::iterator it;
for(it=connList.begin(); it!=connList.end(); ++it)
{
//给其他人发送消息
if (sender != *it)
{
send(*it, buf, strlen(buf), 0);
}
}
}
void getConn()
{
while(1)
{
int conn = accept(mSocket, (struct sockaddr*)&servaddr, &len);
connList.push_back(conn);
//printf("%d\n", conn);
printf("玩家 %d 进入房间 \n",conn);
}
}
void getData()
{
struct timeval tv;
tv.tv_sec = 10;//设置倒计时时间
tv.tv_usec = 0;
while(1)
{
std::list<int>::iterator it;
for(it=connList.begin(); it!=connList.end(); ++it)
{
fd_set rfds;
FD_ZERO(&rfds);
int maxfd = 0;
int retval = 0;
FD_SET(*it, &rfds);
if(maxfd < *it)
{
maxfd = *it;
}
retval = select(maxfd+1, &rfds, NULL, NULL, &tv);
if(retval == -1)
{
printf("select error\n");
}
else if(retval == 0)
{
//printf("not message\n");
}
else
{
char buf[1024];
memset(buf, 0 ,sizeof(buf));
int len = recv(*it, buf, sizeof(buf), 0);
if (len > 0) {
printf("收到 %d的消息 %s", *it ,buf);
sendMsg(*it, buf, len);
}
}
}
//sleep(1);
}
}
int main()
{
//new socket
std::cout<<"1.创建socket"<<std::endl;
mSocket = socket(AF_INET, SOCK_STREAM, 0);
memset(&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
//设置服务器监听的端口
servaddr.sin_port = htons(PORT);
//如果是局域网，这里可以使用IP127.0.0.1
//servaddr.sin_addr.s_addr = inet_addr(IP);
//如果部署到linux服务器 这里使用htonl(INADDR_ANY)
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
std::cout<<"2.绑定socket和端口号"<<std::endl;
if(bind(mSocket, (struct sockaddr* ) &servaddr, sizeof(servaddr))==-1)
{
perror("bind");
exit(1);
}
std::cout<<"3.监听端口号"<<std::endl;
if(listen(mSocket, 20) == -1)
{
perror("listen");
exit(1);
}
len = sizeof(servaddr);
//thread : while ==>> accpet
std::thread t(getConn);
t.detach();//detach的话后面的线程不同等前面的进程完成后才能进行，如果这里改为join则前面的线程无法判断结束，就会
//一直等待，导致后面的线程无法进行就无法实现操作
//printf("done\n");
//thread : recv ==>> show
std::thread t2(getData);
t2.detach();
while(1)//做一个死循环使得主线程不会提前退出
{
}
return 0;
}首先启动服务器程序；
这里由于有多个客户端，需要找到Xcode编译出来的可运行文件
鼠标右键->选择show in finder，即可找到客户单的可执行文件，双击即可打开一个客户端的可运行程序
可以看到两个客户端程序可以进行简单的文字交谈了 ，小明可能接下来会去小红家修电脑去了，但是小红的电脑具体有没有坏我们就不得而知了，懂得都懂 ～～ 嘿嘿