linux系统编程之文件与io(五)

上一节中已经学习了文件描述符的复制，复制方法有三种，其中最后一种fcntl还并未使用到，关于这个函数，不光只有复制文件描述符的功能，还有其它一些用法，本节就对其进行一一剖析:

fcntl常用操作:

这里，我们将上节当中用dup或dup2实现复制文件描述符改用fcntl，程序如下：

先将test2.txt的内容清空，以便进行测试,编译运行：

通过man来查看下它的说明：

【说明：关于这一的操作命令，等之后学到进程时再来学习，先这边记录一下】

上一节也有介绍过，先回顾一下都有哪些状态标志：

也就是说，通过这个命令，能更改文件状态标志，说来有些难理解，下面以实例代码来进行一一说明：

编译运行：

当输入内容时，read才读取完，并打印出输入的内容:

这时本来的文件状态，但是，可以fcntl函数，来改变这种阻塞状态为非阻塞状态，如下：

具体代码：

#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#define ERR_EXIT(m) \
    do \
    { \
        perror(m); \
        exit(EXIT_FAILURE); \
    } while(0)
int main(int argc, char *argv[])
{
    char buf[1024] = {0};
    int ret;
    int flags;
    flags = fcntl(0, F_GETFL, 0);//通过F_GETFL来获得标准输入的状态
    if (flags == -1)
        ERR_EXIT("fcntl get flag error");

    ret = fcntl(0, F_SETFL, flags | O_NONBLOCK);//通过F_SETFL来改变文件的状态为非阻塞0_NONBLOCK，但是为了保留其它状态，所以设置之前需获得状态，再进行与操作
    if (ret == -1)
        ERR_EXIT("fcntl set flag error");

    ret = read(0, buf, 1024);
    if (ret == -1)
        ERR_EXIT("read error");

    printf("buf=%s\n", buf);
    return 0;
}

编译运行：

实际上，对于上面这个错误，对应的错误代码是：

【注意：在设置状态时，一定得先用F_GETFL获取状态，然后再去用F_SETFL去设置，因为我们只想设置非阻塞的状态，对于其它状态如：写状态，读状态等想保留】
关于上面这段设置状态的代码，可以进行封装，以便进行复用，如下：

#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>


#define ERR_EXIT(m) \
    do \
    { \
        perror(m); \
        exit(EXIT_FAILURE); \
    } while(0)

void set_flag(int fd, int flags);//设置文件状态标志函数声明

int main(int argc, char *argv[])
{
    char buf[1024] = {0};
    int ret;

    set_flag(0, O_NONBLOCK);//这时，经过代码封装之后，代码就显得比较干净了
    
    ret = read(0, buf, 1024);
    if (ret == -1)
        ERR_EXIT("read error");

    printf("buf=%s\n", buf);
    return 0;
}

//设置文件状态标志
void set_flag(int fd, int flags)
{
    int val;
    val = fcntl(fd, F_GETFL, 0);
    if (val == -1)
                ERR_EXIT("fcntl get flag error");
    val |= flags;
    if (fcntl(fd, F_SETFL, val) < 0)
        ERR_EXIT("fcntl set flag error");
}

另外，我们还可以封装一个清除文件状态标志的函数，跟设置很类似，如下：

#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>


#define ERR_EXIT(m) \
    do \
    { \
        perror(m); \
        exit(EXIT_FAILURE); \
    } while(0)

void set_flag(int fd, int flags);
void clr_flag(int fd, int flags);//清除文件状态标志函数声明

int main(int argc, char *argv[])
{
    char buf[1024] = {0};
    int ret;

    set_flag(0, O_NONBLOCK);
    clr_flag(0, O_NONBLOCK);//清除了非阻塞的状态标记，也就是等于最终还是阻塞状态
    
    ret = read(0, buf, 1024);
    if (ret == -1)
        ERR_EXIT("read error");

    printf("buf=%s\n", buf);
    return 0;
}

void set_flag(int fd, int flags)
{
    int val;
    val = fcntl(fd, F_GETFL, 0);
    if (val == -1)
                ERR_EXIT("fcntl get flag error");
    val |= flags;
    if (fcntl(fd, F_SETFL, val) < 0)
        ERR_EXIT("fcntl set flag error");
}

//清除文件状态标志
void clr_flag(int fd, int flags)
{
    int val;
    val = fcntl(fd, F_GETFL, 0);
    if (val == -1)
                ERR_EXIT("fcntl get flag error");
    val &= ~flags;
    if (fcntl(fd, F_SETFL, val) < 0)
        ERR_EXIT("fcntl set flag error");
}

编译运行：

【提示：设置状态中的：val |= flags;和清除状态中的：val &= ~flags;是怎么一回事，可以看一下C程序中的位操作，这个也可以参考：http://www.cnblogs.com/webor2006/p/3440026.html】

先来解释一下结构体字段：

说了这么多，可能还是有点不是很好理解，下面以实际代码来进行说明：

#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>


#define ERR_EXIT(m) \
    do \
    { \
        perror(m); \
        exit(EXIT_FAILURE); \
    } while(0)

int main(int argc, char *argv[])
{
    int fd;
    fd = open("test2.txt", O_CREAT | O_RDWR | O_TRUNC, 0644);
    if (fd == -1)
        ERR_EXIT("open error");

    struct flock lock;
    memset(&lock, 0, sizeof(lock));
    lock.l_type = F_WRLCK;//加上排它锁
    lock.l_whence = SEEK_SET;
    lock.l_start = 0;
    lock.l_len = 0;

    if (fcntl(fd, F_SETLK, &lock) == 0)
    {
        printf("lock success\n");
        printf("press any key to unlock\n");
        getchar();
        lock.l_type = F_UNLCK;//释放锁
        if (fcntl(fd, F_SETLK, &lock) == 0)
            printf("unlock success\n");
        else
            ERR_EXIT("unlock fail");
    }
    else
        ERR_EXIT("lock fail");
    return 0;
}