正文
Socket网络编程--聊天程序(5)
小程序:扫一扫查出行
【扫一扫了解最新限行尾号】
复制小程序
【扫一扫了解最新限行尾号】
复制小程序
上一小节我们讲了使用select来避免使用多进程的资源浪费问题。上次只是实现了从多个客户端发送数据给服务器端,接下来就要实现从服务器端发送数据给各个客户端。
使用select多路转换处理聊天程序2
client.c 使用上一节用的那个,在那个基础上修改下面几句
//send-recv 一些返回指没有判断,具体可以看server.c
if((pid=fork())<)
{
perror("fork error\n");
}
else if(pid==)/*child*/
{
while()
{
fgets(sendBuf,MAX_BUF,stdin);
if(send(sockfd,sendBuf,strlen(sendBuf),)==-)
{
perror("fail to receive datas.");
}
memset(sendBuf,,sizeof(sendBuf));
}
}
else
{
while()
{
if((recvSize=recv(sockfd,recvBuf,MAX_BUF,)==-))
{
printf("Server maybe shutdown!");
break;
}
printf("Server:%s\n",recvBuf);
memset(recvBuf,,sizeof(recvBuf));
}
kill(pid,SIGKILL);
} close(sockfd); return ;
}
server.c 我们在上一小节的基础上加上一个处理服务器stdin读取数据,然后遍历所有fd_A里面还连接着的客户端,然后一个一个的发送数据。
while()
{
FD_ZERO(&servfd);//清空所有server的fd
FD_ZERO(&recvfd);//清空所有client的fd
FD_SET(sockfd,&servfd);
//timeout.tv_sec=30;//可以减少判断的次数
switch(select(max_servfd+,&servfd,NULL,NULL,&timeout))
{
...
}
//FD_COPY(recvfd,servfd);
for(i=;i<MAX_CON_NO;i++)//最大队列进行判断,优化的话,可以使用链表
{
if(fd_A[i]!=)
{
FD_SET(fd_A[i],&recvfd);
}
} switch(select(max_recvfd+,&recvfd,NULL,NULL,&timeout))
{
... } /*用于检测stdin是否有数据*/
FD_ZERO(&servfd);
FD_SET(STDIN_FILENO,&servfd); switch(select(STDIN_FILENO+,&servfd,NULL,NULL,&timeout))
{
case -:
break;
case :
break;
default:
/*send datas to client*/
if(FD_ISSET(STDIN_FILENO,&servfd))
{
fgets(sendBuf,MAX_DATA_SIZE,stdin);
for(i=;i<MAX_CON_NO;i++)
{
if(fd_A[i]!=)
{
printf("数据发往%d,",fd_A[i]);
if((sendSize=send(fd_A[i],sendBuf,strlen(sendBuf),))!=strlen(sendBuf))
{
perror("fail");
exit();
}
else
{
printf("Success\n");
}
}
}
memset(sendBuf,,MAX_DATA_SIZE);
fflush(stdin);
}
break;
}
}
return ;
}
废话不多说,直接上运行时的截图
程序运行的顺序是,运行server,然后运行client1,server发送data1,client2连接server,server发送data2,client3连接server,server发送data3,此时client1退出连接,server获取fd 4 close,发送data4的时候没有发送给fd4。然后client4连接上去。分配为fd7。(虽然fd4已经关闭了,按说可以给client4分配fd4,不过太麻烦我这里就没有实现。)
实现聊天室功能
接下来就是实现聊天室功能。上面server.c中有三个select,第一个是处理随时可能来的客户端连接,第二个select是处理随时可能从客户器端来的数据,第三个select是处理随时可能从控制台输入数据,并send到各个客户端。要实现聊天室功能,就对第二个和第三个select进行合并。具体不多说,直接上代码
client.c 基本不变
server.c
#include <stdio.h>
...
int main(int argc,char *argv[])
{
...
while()
{
FD_ZERO(&servfd);//清空所有server的fd
FD_ZERO(&recvfd);//清空所有client的fd
FD_SET(sockfd,&servfd);
//timeout.tv_sec=30;//可以减少判断的次数
switch(select(max_servfd+,&servfd,NULL,NULL,&timeout))
{
...
}
...
switch(select(max_recvfd+,&recvfd,NULL,NULL,&timeout))
{
case -:
//select error
break;
case :
//timeout
break;
default:
for(i=;i<conn_amount;i++)
{
if(FD_ISSET(fd_A[i],&recvfd))
{
/*receive datas from client*/
if((recvSize=recv(fd_A[i],recvBuf,MAX_DATA_SIZE,))==- || recvSize==)
{
//perror("fail to receive datas");
//表示该client是关闭的
printf("fd %d close\n",fd_A[i]);
FD_CLR(fd_A[i],&recvfd);
fd_A[i]=;
}
else//客户端发送数据过来,然后这里进行转发
{
/*send datas to client*/
for(j=;j<MAX_CON_NO;j++)
{
if(fd_A[j]!=&&i!=j)
{
printf("数据发往%d,",fd_A[j]);
if((sendSize=send(fd_A[j],recvBuf,strlen(recvBuf),))!=strlen(recvBuf))
{
perror("fail");
exit();
}
else
{
printf("Success\n");
}
}
}
//可以判断recvBuf是否为bye来判断是否可以close
memset(recvBuf,,MAX_DATA_SIZE);
}
}
}
break;
}
}
return ;
}
直接上运行时图片
各个程序运行的循序是先运行server,然后运行client1,client2,然后cli1发送数据data1,此时cli2接收到server发来的转发数据,然后... ...
好了,到现在为止已经完成了聊天室功能了,还可以随时进入随时出来。好像还不错的样子。下一节就介绍引入用户名登陆的字符界面(由于我不会图形化界面,所以就不花时间去学那个了,至于字符界面下有个curses.h据说可以弄得很好看,我这里就不弄了。有兴趣的可以去弄一下),还有一些人性化的中文提示,然后整理一下代码。由于基本没有什么技术含量,所以如果到时候篇幅不够就再加个数据库设计,考虑使用数据库来保存用户(为了方便就使用mysql吧)。
小小剧透一下,如果接下来有时间将会实现以下功能 用户功能,指令功能,私聊功能,vip功能,文件发送功能,多服务器问题
本文地址: http://www.cnblogs.com/wunaozai/p/3871563.html