Linux 应用开发

文件锁简介

锁是保护共享资源的一种方法。在许多 UNIX 系统上，如果一个文件同时被多个进程编辑，该文件的最后状态取决于写该文件的最后一个进程。但是对于一些特殊的程序，如数据库有时需要独占一个文件，这时就不能让这个文件被多个进程同时操作了。于是 UNIX 系统提供了文件锁来实现这种独占文件的功能。

文件锁：当一个进程正在读或者或者修改文件的某个部分时，使用锁可以阻止其他进程修改同一个文件区域，文件锁可以锁定整个文件或者文件中的一个区域。

文件锁 API

早期的 BSD 只支持 flock 函数，该函数只能对整个文件加锁，不能对文件中的一部分加锁。但是 POSIX 标准的加锁使用 fcntl，这个函数不仅能够对整个文件加锁，也能对文件区域加锁，还能实现其他的文件控制操作。

这个 fcntl 函数也非常重要，来认真学习下这个函数，你可以通过 man fcntl 来查看帮助手册。

fcntl 函数

函数声明如下：

#include <unistd.h>#include <fcntl.h>/* * fd： 文件描述符 * cmd：F_GETLK，F_SETLK，F_SETLKW * arg：一个取决于 cmd 的可选参数 * return：成功返回 0，失败返回 -1 */int fcntl(int fd, int cmd, ... /* arg */ );

其中 cmd 和 arg 参数比较重要，我们详细介绍这两个参数。

arg 参数

这个参数是一个可选参数，指定文件加锁的详细信息，在加锁文件时我们都会指定这个参数。这个参数是一个指向 struct flock 结构的指针：

struct flock {    ...    short l_type;        short l_whence;     off_t l_start;     off_t l_len;      pid_t l_pid;     ...};

其中 5 个参数比较重要：
1. l_type：希望的锁的类型，F_RDLCK（共享读锁），F_WRLCK（独占性写锁），F_UNLCK（解锁）
2. l_whence：与 lseek 的参数相同，指定 l_start 从文件的何处开始偏移，SEEK_SET，SEEK_CUR, SEEK_END
3. l_start：加锁或者解锁的字节偏移量，与 l_whence 配合使用
4. l_len：要加锁或者解锁的字节长度
5. l_pid：进程 ID 为 l_pid 的进程能够阻塞加锁操作。

cmd 参数

这个参数有 3 种取值方式：
1. F_GETLK：判断当前文件是否被加锁，如果有锁则将现有锁的信息复制到 struct flock 中，否则设置 l_type = F_UNLCK
2. F_SETLK：设置由 arg 所描述的锁
3. F_SETLKW：如果请求的锁不能被授予，那么调用进程会被置为休眠

总体来讲：用 F_GETLK 来测试能够获得一把锁，然后用 F_SETLK 或者 F_SETLKW 尝试加上一把锁。

文件锁分类

按照锁的类型和系统中锁的属性可以分为下面的 2 类。

读锁（L_RDLCK）和写锁（L_WRLCK）

这两个锁的区别是：任意多个进程在一个给定的字节上可以有一把共享的读锁，但是在一个给定的字节上只能有一个进程有一把独占写锁。

如下表所示：

但是这个规则只适用不同进程之间的提出的锁请求。在单个进程下，如果在已有的锁上加新锁，则新锁会替换旧锁，也不管锁是何种类型。例如：读锁可以替换写锁，写锁也可以替换读锁。

强制锁和建议锁

1. 强制锁：内核管理的底层锁，一个进程锁定后其他用户不能打开
2. 建议锁：用户管理的上层锁，一个进程锁定后其他用户可以打开

Linux 默认加的是建议锁，如果需要加强制锁，则需要在文件系统上用 mount 命令的 -o mand 选项来打开强制锁机制，参考 man fcntl。

实例：fcntl 锁定文件

为了更好的理解锁的机制，我们来写一个实际的加锁程序来锁定一个文件，然后测试是否加锁成功。

1. 对文件加建议锁

/* * file_lock.c  */#include <stdio.h>#include <stdlib.h>#include <sys/stat.h>#include <sys/types.h>#include <unistd.h>#include <fcntl.h>int main() {    // 打开文件     int fd = open("./hello.txt", O_RDWR | O_CREAT, 0666)  ;     if(fd < 0) {        printf("file open fail.\n");        exit(1);    }    // 建立一个独占性写锁    struct flock lock;    lock.l_type = F_WRLCK;    lock.l_whence = SEEK_SET;    lock.l_start = 0;    lock.l_len = 0;    lock.l_pid = getpid();     // 对 hello.txt 加锁    if (fcntl(fd, F_SETLK, &lock) < 0)        printf("fcntl error.\n");    printf("Process %d has lock.\n", lock.l_pid);       while(1); }

编译运行：

# 编译gcc file_lock.c -o file_lock# 运行 ./file_lock Process 11080 has lock.# Ctrl - C 结束程序 

然后我们启动另一个终端，尝试向 hello.txt 中写入数据，观察是否会阻塞：

echo hello_world >> hello.txtcat hello.txt# 写入成功了hello_world

我们发现竟然写入成功了，我们不是已经加了锁了吗？其实这是因为默认加的是建议锁，建议锁可以被其他进程直接打开，因此可以写入成功。但是我们在程序中一般都是先用 F_GETLCK 主动判断要打开的文件是否有锁，例如下面这个例子：

2. 测试文件是否加锁成功

/* * lock_test.c */#include <stdio.h>#include <stdlib.h>#include <sys/stat.h>#include <sys/types.h>#include <unistd.h>#include <fcntl.h>int main() {     int fd = open("./hello.txt", O_RDWR | O_CREAT, 0666)  ;     if(fd < 0) {        printf("file open fail.\n");        exit(1);    }    // 如果发现有进程已经对这个文件加锁    // 那么现有锁的信息将重写下面的结构    struct flock lock;    lock.l_type = F_WRLCK;    lock.l_whence = SEEK_SET;    lock.l_start = 0;    lock.l_len = 0;    lock.l_pid = getpid();     // 判断 hello.txt 上是否已经加锁    if (fcntl(fd, F_GETLK, &lock) < 0)        printf("fcntl error.\n");    // 如果没有进程对文件加锁，只将 l_type 设置为 F_UNLCK，其他信息不变    if (lock.l_type == F_UNLCK)        printf("No process has the lock.\n");       else         printf("Process %d has the lock.\n", lock.l_pid);    return 0;}

我们编译运行：

gcc lock_test.c -o lock_test./lock_testNo process has the lock.

发现当前 hello.txt 文件上没有锁，我们运行上一个 ./file_lock 程序：

./file_lockProcess 11080 has lock.# while (1) 阻塞

这时进程 11080 已经对 hello.txt 加上了建议的独占性写锁，我们再次用 ./lock_test 测试这个文件是否被加锁：

./lock_testProcess 11080 has the lock.

可以看到这个文件已经被 11080 进程加锁了，我们之后就不要操作这个文件了。

记住在使用 fcntl 时，一定主要主动判断文件是否被加锁。

结语

文件锁是一个非常重要的话题，这篇博客主要介绍了一个重要的 fcntl 函数和文件锁的相关概念，一定要掌握这个函数的基本用法，其他重要的锁话题，例如死锁等，我们后面再议。

最后，感谢你的阅读，我们下次再见 :)

推荐关注我的微信公众号 CDeveloper，坚持技术原创，只说真话！