使用select实现进程间的即时聊天

首先使用mkfifo命令新建两个管道文件1.fifo和2.fifo。
首先画图分析:
在这里插入图片描述
如图所示,1.fifo的读端是进程1,写端是进程2;2.fifo的读端是进程2,写端是进程1。且进程1与2均从标准输入流读取数据。所以要注意,进程1的读操作描述符字为0与1.fifo;进程2的读操作描述符字为0与2.fifo.

1号和2号的命令行参数均为1.fifo 2.fifo
1号的代码为:

#include <sys/types.h>
#include <sys/stat.h>
#include <sys/select.h>
#include <sys/time.h>
#include <fcntl.h>
#include <cstdio>
#include <cstring>
#include <unistd.h>

int main(int argc, char *argv[])	 
{
    int fdr=open(argv[1],O_RDONLY);		//1.fifo为1号的read
    int fdw=open(argv[2],O_WRONLY);		//2.fifo为1号的write
    printf("This is No.1\n");
    char buf[128]={0};
    fd_set rdset;						//读操作描述符字集合
    int ret;
    while(1)
    {
        FD_ZERO(&rdset);
        FD_SET(0,&rdset);
        FD_SET(fdr,&rdset);
        ret=select(fdr+1,&rdset,NULL,NULL,NULL);
        if(ret>0)
        {
            if(FD_ISSET(fdr,&rdset))		//如果管道可读
            {
                memset(buf,0,sizeof(buf));
                ret=read(fdr,buf,sizeof(buf));
                if(ret==0)					//若管道无数据,读端阻塞;管道关闭,read返回0
                {
                    printf("Fifo has been closed.\n");
                    close(fdr);
                    close(fdw);
                    break;
                }
                printf("No.2 says:%s\n",buf);
            }
            if(FD_ISSET(0,&rdset))			//从stdin读取数据
            {
                memset(buf,0,sizeof(buf));
                ret=read(0,buf,sizeof(buf));
                if(ret>0)
                {
                    write(fdw,buf,strlen(buf)-1);
                }else{
                    printf("Fifo has been closed.\n");
                    close(fdr);
                    close(fdw);
                    break;
                }
            }
        }
    }
    return 0;
}

2号代码:

#include <sys/types.h>
#include <sys/stat.h>
#include <sys/select.h>
#include <sys/time.h>
#include <fcntl.h>
#include <cstdio>
#include <cstring>
#include <unistd.h>

int main(int argc,char *argv[])
{
    int fdw=open(argv[1],O_WRONLY);		//1.fifo为2号的write
    int fdr=open(argv[2],O_RDONLY);		//2.fifo为2号的read
    printf("This is No.2\n");
    char buf[128]={0};
    int ret;
    fd_set rdset;
    while(1)
    {
        FD_ZERO(&rdset);
        FD_SET(0,&rdset);
        FD_SET(fdr,&rdset);
        ret=select(fdr+1,&rdset,NULL,NULL,NULL);
        if(ret>0)
        {
            if(FD_ISSET(0,&rdset))
            {
                memset(buf,0,sizeof(buf));
                ret=read(0,buf,sizeof(buf));
                if(ret>0)
                {
                    write(fdw,buf,strlen(buf)-1);
                }else{
                    printf("Fifo has been closed.\n");
                    close(fdr);
                    close(fdw);
                    break;
                }
            }
            if(FD_ISSET(fdr,&rdset))
            {
                memset(buf,0,sizeof(buf));
                ret=read(fdr,buf,sizeof(buf));
                if(ret==0)
                {
                    printf("Fifo has been closed.\n");
                    close(fdr);
                    close(fdw);
                    break;
                }
                printf("No.1 says:%s\n",buf);
            }
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值