高级IO-非阻塞

一、设置文件描述符标志

//设置文件描述符标志
void set_fl(int fd, int flags)
{
    int val;
    //获取文件描述符标志
    if((val = fcntl(fd, F_GETFL, 0)) < 0)
        printf("fcntl F_GETFL error\n");
    val |= flags;//添加描述符标志flags
    //设置文件描述符标志
    if(fcntl(fd, F_SETFL, val) < 0)
        printf("fcntl F_SETFL error\n");
}

二、清除文件描述符标志

//清除文件描述符标志
void clr_fl(int fd, int flags)
{
    int val;
    //获取文件描述符标志
    if((val = fcntl(fd, F_GETFL, 0)) < 0)
        printf("fcntl F_GETFL error\n");
    val &= ~flags;//清除描述符标志flags
    //设置文件描述符标志
    if(fcntl(fd, F_SETFL, val) < 0)
        printf("fcntl F_SETFL error\n");
}

三、主函数

int main(void)
{
    int ntowrite, nwrite;
    char *ptr;
 
    ntowrite = read(STDIN_FILENO, buf, sizeof(buf));
    fprintf(stderr, "read %d bytes.\n",ntowrite);
 
    set_fl(STDOUT_FILENO, O_NONBLOCK);
 
    ptr = buf;
    while(ntowrite > 0)
    {
        errno = 0;
        nwrite = write(STDOUT_FILENO, ptr, ntowrite);
        fprintf(stderr, "nwrite = %d, errno = %d\n", nwrite, errno);
 
        if(nwrite > 0)
        {
            ptr += nwrite;
            ntowrite -= nwrite;
        }
    }
    clr_fl(STDOUT_FILENO, O_NONBLOCK);
 
    exit(0);
}
 

四、程序

#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>

void set_fl(int fd, int flags);
void clr_fl(int fd, int flags);
 
char buf[500000];
 
int main(void)
{
    int ntowrite, nwrite;
    char *ptr;
 
    ntowrite = read(STDIN_FILENO, buf, sizeof(buf));
    fprintf(stderr, "read %d bytes.\n",ntowrite);
 
    set_fl(STDOUT_FILENO, O_NONBLOCK);
 
    ptr = buf;
    while(ntowrite > 0)
    {
        errno = 0;
        nwrite = write(STDOUT_FILENO, ptr, ntowrite);
        fprintf(stderr, "nwrite = %d, errno = %d\n", nwrite, errno);
 
        if(nwrite > 0)
        {
            ptr += nwrite;
            ntowrite -= nwrite;
        }
    }
    clr_fl(STDOUT_FILENO, O_NONBLOCK);
 
    exit(0);
}
 
 
//设置文件描述符标志
void set_fl(int fd, int flags)
{
    int val;
    //获取文件描述符标志
    if((val = fcntl(fd, F_GETFL, 0)) < 0)
        printf("fcntl F_GETFL error\n");
    val |= flags;//添加描述符标志flags
    //设置文件描述符标志
    if(fcntl(fd, F_SETFL, val) < 0)
        printf("fcntl F_SETFL error\n");
}
//清除文件描述符标志
void clr_fl(int fd, int flags)
{
    int val;
    //获取文件描述符标志
    if((val = fcntl(fd, F_GETFL, 0)) < 0)
        printf("fcntl F_GETFL error\n");
    val &= ~flags;//清除描述符标志flags
    //设置文件描述符标志
    if(fcntl(fd, F_SETFL, val) < 0)
        printf("fcntl F_SETFL error\n");
}

五、编译

编译
gcc -o read read.c   
运行
./read