select to achieve a personal chat room

#include <func.h>
int main(int argc, char *argv[])
{
    // ./chat1 1.pipe 2.pipe
    ARGS_CHECK(argc,3);
    int fdr = open(argv[1],O_RDONLY);
    int fdw = open(argv[2],O_WRONLY);
    puts("chat1");
    fd_set rdset;
    time_t t;
    while(1){
        FD_ZERO(&rdset);
        FD_SET(STDIN_FILENO,&rdset);
        FD_SET(fdr,&rdset);
        char buf[128];
        char T[256]={0};
        struct timeval timeout;
        //2.5s的超时时限
        timeout.tv_sec = 2;
        timeout.tv_usec = 500000;
        int sret = select(fdr+1,&rdset,NULL,NULL,&timeout);
        if(sret == 0){
            puts("timeout!");
        }
        else{
            if(FD_ISSET(STDIN_FILENO,&rdset)){
                puts("msg from stdin");
                memset(buf,0,sizeof(buf));
                int ret = read(STDIN_FILENO,buf,sizeof(buf)-1);
                if(ret == 0){
                    puts("I quit");
                    write(fdw,"nishigehaoren",13);
                    break;
                }
                strcpy(T,ctime(&t));
                strcat(T,buf);
                write(fdw,T,strlen(T));
            }
            if(FD_ISSET(fdr,&rdset)){
                puts("msg from pipe");
                memset(buf,0,sizeof(buf));
                int ret = read(fdr,buf,sizeof(buf)-1);
                if(ret == 0){
                    puts("over");
                    break;
                }
                puts(buf);
            }
        }
    }
    close(fdw);
    close(fdr);
    return 0;
}

 

#include <func.h>
int main(int argc, char *argv[])
{
    // ./chat2 2.pipe 1.pipe
    ARGS_CHECK(argc,3);
    int fdw = open(argv[1],O_WRONLY);
    int fdr = open(argv[2],O_RDONLY);
    puts("chat2");
    fd_set rdset;
    time_t t;
    while(1){
        FD_ZERO(&rdset);
        FD_SET(STDIN_FILENO,&rdset);
        FD_SET(fdr,&rdset);
        char buf[128];
        char T[256]={0};
        struct timeval timeout;
        //2.5s的超时时限
        timeout.tv_sec = 2;
        timeout.tv_usec = 500000;
        int sret = select(fdr+1,&rdset,NULL,NULL,&timeout);
        if(sret == 0){
            puts("timeout!");
        }
        else{
            if(FD_ISSET(STDIN_FILENO,&rdset)){
                puts("msg from stdin");
                memset(buf,0,sizeof(buf));
                int ret = read(STDIN_FILENO,buf,sizeof(buf)-1);
                if(ret == 0){
                    puts("I quit");
                    write(fdw,"nishigehaoren",13);
                    break;
                }
                strcpy(T,ctime(&t));
                strcat(T,buf);
                write(fdw,T,strlen(T));
            }
            if(FD_ISSET(fdr,&rdset)){
                puts("msg from pipe");
                memset(buf,0,sizeof(buf));
                int ret = read(fdr,buf,sizeof(buf)-1);
                if(ret == 0){
                    puts("over");
                    break;
                }
                puts(buf);
            }
        }
    }
    close(fdw);
    close(fdr);
    return 0;
}

 

Guess you like

Origin blog.csdn.net/qq_43496435/article/details/114240632