专栏首页开发与安全linux系统编程之文件与I/O(六):fcntl 函数与文件锁

linux系统编程之文件与I/O(六):fcntl 函数与文件锁

一、fcntl函数

功能:操纵文件描述符,改变已打开的文件的属性

int fcntl(int fd, int cmd, ... /* arg */ );

cmd的取值可以如下:

复制文件描述符 F_DUPFD (long)

设置/获取文件描述符标志 F_GETFD (void) F_SETFD (long)

设置/获取文件状态标志 F_GETFL (void) F_SETFL (long)

获取/设置文件锁 F_GETLK F_SETLK,F_SETLKW

其中复制文件描述符可参见《linux系统编程之文件与I/O(五):打开文件的内核结构file和重定向》,文件描述符的标志只有一个即FD_CLOEXEC,设置/获取文件描述符标志看这里。下面先来看设置/获取文件状态标志。

F_SETFL:

On Linux  this  command can change only the O_APPEND, O_ASYNC, O_DIRECT, O_NOATIME, and O_NONBLOCK flags.

示例程序如下:

/*************************************************************************
    > File Name: file_fcntl.c
    > Author: Simba
    > Mail: dameng34@163.com
    > Created Time: Sat 23 Feb 2013 02:34:02 PM CST
 ************************************************************************/
#include<sys/types.h>
#include<sys/stat.h>
#include<unistd.h>
#include<fcntl.h>
#include<stdio.h>
#include<stdlib.h>
#include<errno.h>
#include<string.h>

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

void set_flag(int, int);
void clr_flag(int, int);

int main(int argc, char *argv[])
{
    char buf[1024] = {0};
    int ret;
    /*
        int flags;
        flags = fcntl(0, F_GETFL, 0);
        if (flags == -1)
            ERR_EXIT("fcntl get flag error");
        ret = fcntl(0, SETFL, flags | O_NONBLOCK); //设置为非阻塞,但不更改其他状态
        if (ret == -1)
            ERR_EXIT("fcntl set flag error");
    */
    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");
}

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");
}

测试输出:

simba@ubuntu:~/Documents/code/linux_programming/APUE/File_IO$ ./file_fcntl  read error: Resource temporarily unavailable

因为将标准输入的状态更改为非阻塞,则read不会阻塞等待输入而立即返回错误,errno将被置为EAGAIN,即可以重新尝试。

二、文件锁结构体

struct flock { ... short l_type;       /* Type of lock: F_RDLCK,         F_WRLCK, F_UNLCK */ short l_whence; /* How to interpret l_start:                                    SEEK_SET, SEEK_CUR, SEEK_END */ off_t l_start;       /* Starting offset for lock */ off_t l_len;         /* Number of bytes to lock */ pid_t l_pid;        /* PID of process blocking our lock                                    (F_GETLK only) */      ... };

文件锁的类型只有两种,一种是写锁也叫排他锁,一种是读锁也就共享锁,可以有多个进程各持有一个读锁,但只能有一个进程持有写锁,只有对文件有对应的读写权限才能施加对应的锁类型。中间三个参数 l_whence,  l_start, l_len 决定了被锁定的文件范围。当fcntl 函数的cmd为F_GETLK时,flock 结构体的 l_pid 参数会返回持有写锁的进程id。进程退出或者文件描述符被关闭时,会释放所有的锁。

示例程序如下:

/*************************************************************************
    > File Name: file_flock.c
    > Author: Simba
    > Mail: dameng34@163.com
    > Created Time: Sat 23 Feb 2013 02:34:02 PM CST
 ************************************************************************/
#include<sys/types.h>
#include<sys/stat.h>
#include<unistd.h>
#include<fcntl.h>
#include<stdio.h>
#include<stdlib.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, 0664);
    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)
    {
        /* 若为F_SETLKW,这时如果锁已经被其他进程占用,则此进程会阻塞直到其他进程释放锁*/
        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; //进程退出会对所有文件解锁
}

测试如下:

我们先在一个 终端执行程序:

simba@ubuntu:~/Documents/code/linux_programming/APUE/File_IO$ ./file_flock  lock success press any key to unlock

现在文件已经被锁住了,而且没有按下任何按键,所以卡在这里,也还没解锁,接着在另一个终端再次执行同个程序:

simba@ubuntu:~/Documents/code/linux_programming/APUE/File_IO$ ./file_flock  lock fail: Resource temporarily unavailable

会立即返回错误,因为我们希望施加的是排他锁,而现在前面一个进程正在占用写锁还没释放,所以尝试施加锁失败,而如果fcntl 函数的cmd 设置为 F_SETLKW,即带w的版本,则此进程会一直阻塞直到前面一个进程释放了锁。

参考:《APUE》

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 80386的各种寄存器一览

    注:本分类下文章大多整理自《深入分析linux内核源代码》一书,另有参考其他一些资料如《linux内核完全剖析》、《linux c 编程一站式学习》等,只是为了...

    s1mba
  • 从零开始学C++之STL(八):函数对象、 函数对象与容器、函数对象与算法

    一、函数对象 1、函数对象(function object)也称为仿函数(functor) 2、一个行为类似函数的对象,它可以没有参数,也可以带有若干参数。 ...

    s1mba
  • 从零开始学C++之从C到C++(二):引用、数组引用与指针引用、内联函数inline、四种类型转换运算符

    一、引用 (1)、引用是给一个变量起别名 定义引用的一般格式:类型  &引用名 = 变量名; 例如:int a=1;  int  &b=a;// b是a的别...

    s1mba
  • Codeforces Round #473 (Div. 2)

    attack
  • leetcode周赛130 | 题解速递

    leetcode周赛题解,后面每周如果有时间就做周赛题目,不单独发leetcode的题目了,做多了感觉都可以在以前的题目里面找到类似的。

    ACM算法日常
  • 【Miscalculation UVALive - 6833 】【模拟】

    题目讲的是给你一个串,里面是加法、乘法混合运算(个人赛中误看成是加减乘除混合运算),有两种算法,一种是乘法优先运算,另一种是依次从左向右运算(不管它是否乘在前还...

    _DIY
  • 洛谷P1567 统计天数

    题目背景 统计天数 题目描述 炎热的夏日,KC非常的不爽。他宁可忍受北极的寒冷,也不愿忍受厦门的夏天。最近,他开始研究天气的变化。他希望用研究的结果预测未来的天...

    attack
  • BZOJ4260: Codechef REBXOR (01Tire树)

    attack
  • Stream篇(1)

    最近在看一个写的很好的博客,为了加深记忆,把自认为重要的东西,一边看,一边记在这里 一、什么是流、字节序列、字节 一条河中有一条鱼游过,这条鱼就是一个字节,这个...

    小端
  • P1801 黑匣子_NOI导刊2010提高(06)

    题目描述 Black Box是一种原始的数据库。它可以储存一个整数数组,还有一个特别的变量i。最开始的时候Black Box是空的.而i等于0。这个Black ...

    attack

扫码关注云+社区

领取腾讯云代金券

玩转腾讯云 有奖征文活动