2_4_2___test.c

happyzlz

                2_4_2___test.c
#include stdio.h>
#include stdlib.h>
#include string.h>
#include strings.h>
#include getopt.h>
#include sys/select.h>
#include sys/types.h>
#include sys/stat.h>
#include errno.h>
#include fcntl.h>
#include unistd.h>
void read_line( int file , char * buff )
{
        while( 1 ){
                if( read( file, buff, 1 ) == 1 ){
                        if( *buff == '\n' ){
                                break;
                        }
                        buff++;
                }
        }
}
void write_line( int file , char * buff )
{
        while( 1 ){
                if( write( file, buff, 1 ) == 1 ){
                        if( *buff  == '\n' ){
                                break;
                        }
                        buff++;
                }
        }
}
void read_stdin( int pipe )
{
        char    buff[ 512 ];
        fcntl( 0, F_SETFL, O_NONBLOCK );
      
        read_line( 0,buff );
        write_line( pipe, buff );
}
void read_pipe( int pipe )
{
        char    buff[ 512 ];
        
        fcntl( pipe, F_SETFL, O_NONBLOCK );
        
        read_line( pipe,buff );
        write_line( 1, buff );
}
void write_pipe( void )
{
        printf("write pipe\n");
}
void usage( void )
{
        printf( "Usage :\n" );
        printf( "\ttestpipe -f /dev/pipe0 \n" );
        printf( "\tor\n" );
        printf( "\ttestpipe -f /dev/pipe1 \n" );
}
int main( int argc, char * argv[] )
{
        fd_set  rfds;   /* read files */
        fd_set  wfds;   /* write files */
        int     ret;
        char  * path = "";
        char  * opts = "f:";
        int     pipe;
        int     op;
        
        /* get arguments and check it*/
        if( argc != 3 ){
                usage();
                return 0;
        }
        while( ( op = getopt( argc, argv, opts ) ) != EOF ){
                switch( op ){
                        case 'f':
                                path = optarg;
                                break;
                                
                        default:
                                usage();
                                return 0;
                                break;
                }
        }
      
        if( ( strcmp( path, "/dev/pipe0" )  == 0 ) ||
            ( strcmp( path, "/dev/pipe1" )  == 0 ) ){
        }
        else{
                usage();
                return 0;
        }
        
               
        /* open pipe-file and select it */
        pipe = open( path, O_RDWR );
        if( pipe == -1 ){
                printf( "open %s failed : %d \n", path, errno );
                return 0;
        }
      
        /* main loop */
        while( 1 ){
                FD_ZERO( &rfds );
                FD_SET( 0, &rfds );
                FD_SET( pipe, &rfds );
               
                FD_ZERO( &wfds );
                FD_SET( pipe, &wfds );
                ret = select( pipe + 1, &rfds, NULL, NULL, NULL );
                if( ret == -1 ){
                        printf( "select error : %d\n", errno );
                        break;
                }
                else if( ret > 0 ){
                        if( FD_ISSET( 0, &rfds ) ){
                                read_stdin( pipe );
                        }
                        else if( FD_ISSET( pipe, &rfds ) ){
                                read_pipe( pipe );
                        }
                }
        }
        close( pipe );
        
        return 0;
}
               
               
返回目录


本文来自ChinaUnix博客，如果查看原文请点：http://blog.chinaunix.net/u2/68424/showart_681415.html