Linux 高级编程 - 文件锁定

版权声明：本文为 DLonng 原创文章，可以随意转载，但必须在明确位置注明出处！

文件锁简介

锁是保护共享资源的一种方法。在许多 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 个参数比较重要：

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

cmd 参数

这个参数有 3 种取值方式：

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

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

文件锁分类

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

读锁（L_RDLCK）和写锁（L_WRLCK）

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

如下表所示：

lock_table

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

强制锁和建议锁

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

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.txt

cat 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_test
No process has the lock.

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

./file_lock

Process 11080 has lock.
# while (1) 阻塞

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

./lock_test
Process 11080 has the lock.

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

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

结语

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

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

本文原创首发于微信公号「登龙」，分享机器学习、算法编程、Python、机器人技术等原创文章，扫码即可关注！