Linux之select系统调用_1

SYNOPSIS
       /* According to POSIX.1-2001 */
       #include <sys/select.h>

       /* According to earlier standards */
       #include <sys/time.h>
       #include <sys/types.h>
       #include <unistd.h>

       int select(int nfds, fd_set *readfds, fd_set *writefds,
                  fd_set *exceptfds, struct timeval *timeout);

       void FD_CLR(int fd, fd_set *set);
       int  FD_ISSET(int fd, fd_set *set);
       void FD_SET(int fd, fd_set *set);
       void FD_ZERO(fd_set *set);

DESCRIPTION
       select() allow  a  program  to  monitor  multiple  file
       descriptors,  waiting  until one or more of the file descriptors become
       "ready" for some class of I/O operation (e.g., input possible).  A file
       descriptor  is considered ready if it is possible to perform the corre-
       sponding I/O operation (e.g., read(2)) without blocking.

/*************************************************************************
  > File Name: server.c
  > Author: KrisChou
  > Mail:zhoujx0219@163.com 
  > Created Time: Sat 23 Aug 2014 02:37:58 PM CST
 ************************************************************************/

#include <stdio.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <sys/select.h>
#include <sys/time.h>

int main(int argc, char* argv[]) 
{
    /* 打开管道 */
    int fd1, fd2, fd3 ;
    fd1= open(argv[1], O_RDONLY);
    fd2= open(argv[2], O_RDONLY);
    fd3= open(argv[3], O_RDONLY);
    
    printf("server: %d \n", getpid());
    
    /* 设置select需要监听的集合fd_set */
    fd_set readset ,bak_set ;
    FD_ZERO(&bak_set);
    FD_SET(fd1, &bak_set);
    FD_SET(fd2, &bak_set);
    FD_SET(fd3, &bak_set);
    /* 设置select轮巡时间 */
    struct timeval  tm ;
    tm.tv_sec = 10 ;
    tm.tv_usec = 0 ;
    /* 可以在程序开始，使得服务器端同时读到3个消息，此处可有可无 */
    sleep(10);
    
    /* select是通过检查管道是否阻塞，来进行监听的。如果均为阻塞，select在轮巡时间内返回0。
     * 如果有管道不阻塞，则select返回不阻塞的管道个数。
     * 注意：当某个客户端向管道发送数据，或者关闭管道，均属于非阻塞状况。
     * 我们可以根据read的返回值来区分这两种情况                                           */
    int nret ; /* select返回值 */
    char buf[1024];
    while(1){
        /* 每次select后都会改变select所监听集合的值，因此需要在每次select的开始重新设置监听集合。
         * 如果不进行设置，那么上次select后监听集合置1的描述符仍然为1。
         * 注意：在Linux系统下，时间也需要重新设置。                                            */
        readset = bak_set ;
        tm.tv_sec = 10 ;
        tm.tv_usec = 0 ;
        nret = select(6 , &readset, NULL, NULL, &tm );
        if(nret > 0)
        {
            printf("%d active ! \n", nret);
            if(FD_ISSET(fd1, &readset))       /* select返回值大于0时，必然有管道不阻塞了，此时需要查看是哪个管道*/
            {
                memset(buf, 0 , 1024);
                if(0 == read(fd1, buf, 1024))
                {
                    FD_CLR(fd1, &bak_set);    /* 如果一个用户其写端已经关闭，那么将其描述符从监听集合中去除 */
                }else 
                {
                    write(1, buf, strlen(buf));
                }
            }
            if(FD_ISSET(fd2, &readset))
            {
                memset(buf, 0 , 1024);
                if(0 == read(fd2, buf, 1024) )
                {
                    FD_CLR(fd2, &bak_set);
                }else
                {
                write(1, buf, strlen(buf));
                }
            }
            if(FD_ISSET(fd3, &readset))
            {
                memset(buf, 0 , 1024);
                if(read(fd3, buf, 1024) == 0)
                {
                    FD_CLR(fd3, &bak_set);
                }else
                {
                    write(1, buf, strlen(buf));
                }
            }


        }else if(nret == 0) 
        {
            printf("time out ! \n");
        }
    }

}

/*************************************************************************
    > File Name: client.c
    > Author: KrisChou
    > Mail:zhoujx0219@163.com 
    > Created Time: Sat 23 Aug 2014 02:30:54 PM CST
 ************************************************************************/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <time.h>

int main(int argc, char *argv[])
{
    /* 开启管道 */
    printf("%d start ! \n", getpid());
    int fd_sent;
    fd_sent = open(argv[1],O_WRONLY);
    if(fd_sent == -1)
    {
        perror("open");
        exit(1);
    }
    printf("connect success! \n");

    
    /* 客户端向管道发送消息，按ctrl+D退出循环 */
    
    char line[1024]; /* 从标准输入获得要发送的消息 */
    char msg[1024];  /* 给消息头加一个pid，表示身份 */
    
    while(memset(line,0,1024), fgets(line,1024,stdin) != NULL)
    {
        memset(msg,0,1024);
        sprintf(msg, "%d : %s \n", getpid(), line);
        write(fd_sent, msg, strlen(msg));
    }

    close(fd_sent);
    return 0;
}

mkfifo 1.fifo 2.fifo 3.fifo;

./server.exe 1.fifo 2.fifo 3.fifo

./client 1.fifo
./client 2.fifo
./client 3.fifo

if(0 == read(fd1, buf, 1024))
{
    FD_CLR(fd1, &bak_set);     /* 如果一个用户其写端已经关闭，那么将其描述符从监听集合中去除 */
}

1 active!
1 active!
1 active!
...