Posted on 2007-11-26 14:52
lymons 閱讀(2472)
評(píng)論(0) 編輯 收藏 引用 所屬分類:
C 、
Unix/Linux
Linux 文件鎖是建議鎖，也有人把它叫做記錄鎖，是通過(guò)系統(tǒng)調(diào)用fcntl(2)來(lái)實(shí)現(xiàn)的。
這種鎖在鎖定文件時(shí)有兩種模式，分別是阻塞(block)和非阻塞模式。
在編碼時(shí)比較常用的是有一種的非阻塞模式，也就是發(fā)現(xiàn)文件已經(jīng)被其他進(jìn)程
鎖定時(shí)，立即返回不予等待。而阻塞模式則正好與它相反，也就是一直等待直到
其他進(jìn)程釋放文件鎖為止。
注：關(guān)于詳細(xì)內(nèi)容請(qǐng)參看《Unix環(huán)境高級(jí)編程》不過(guò)，有的時(shí)候也會(huì)用到阻塞模式的文件鎖，而且會(huì)要求不能被一直阻塞，等待
了一定時(shí)間后應(yīng)返回。也就是說(shuō)，想給阻塞版本的文件鎖加上一個(gè)超時(shí)時(shí)間(timeout)。
通過(guò)man手冊(cè)，fcntl(2)里面沒(méi)有關(guān)于在阻塞模式時(shí)，設(shè)置超時(shí)時(shí)間的任何描述。
但從man手冊(cè)里我們發(fā)現(xiàn)，文件鎖在阻塞時(shí)會(huì)被信號(hào)(signal)中斷。所以我們就像
可以利用設(shè)置信號(hào)軟中斷來(lái)實(shí)現(xiàn)一個(gè)自己版本的等待超時(shí)呢。
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <unistd.h>
4 #include <fcntl.h>
5 #include <sys/types.h>
6 #include <sys/stat.h>
7 #include <signal.h>
8
9 #define readw_lock(fd) \
10 lock_reg((fd), F_SETLKW, F_RDLCK)
11 #define writew_lock(fd) \
12 lock_reg((fd), F_SETLKW, F_WRLCK)
13 #define unlock(fd) \
14 lock_reg((fd), F_SETLK, F_UNLCK)
15
16 typedef int (*LW_FN)(char *fname);
17
18 int
19 lock_reg(int fd, int cmd, int type)
20 {
21 struct flock lock;
22 lock.l_type = type;
23 lock.l_start = 0;
24 lock.l_whence = SEEK_SET;
25 lock.l_len = 0;
26
27 return fcntl(fd, cmd, &lock);
28 }
29
30 void hander(int signo)
31 {
32 // do nothing
33 return;
34 }
35
36 int lockw(char *fname, LW_FN fn, int timeout)
37 {
38 int ret = 0;
39 int fd;
40 struct sigaction act, oact;
41
42 if ((fd = open(fname, O_CREAT | O_RDWR, 0666)) == -1) {
43 printf("open failed!\n");
44 return -1;
45 }
46
47 // set timer to wakeup fcntl
48 act.sa_handler = hander;
49 sigemptyset(&act.sa_mask);
50 act.sa_flags = 0; // here, must be zero for wakeup fcntl
51 sigaction(SIGALRM, &act, &oact);
52
53 int sec = alarm(timeout);
54
55 if (writew_lock(fd) == 0) {
56 alarm(sec);
57 // recovery signal handler.
58 sigaction(SIGALRM, &oact, NULL);
59
60 printf("locked OK!\n");
61
62 // here, add code about file.
63 #ifdef _TEST
64 getchar();
65 ret = 0;
66 #else
67 ret = fn(fname);
68 #endif
69
70 printf("unlocked!\n");
71 unlock(fd);
72 }
73 else {
74 alarm(sec);
75 // recovery signal handler.
76 sigaction(SIGALRM, &oact, NULL);
77 // lock failed, because of timeout.
78 printf("write lock failed\n");
79 ret = -1;
80 }
81
82 return ret;
83 }
84
85 // test code
86 int func(char *fname)
87 {
88 printf("check file:%s 
\n", fname);
89 getchar();
90 return 0;
91 }
92
93 int main()
94 {
95 return lockw("file.lock", func, 5);
96 }
97
98
該程序的原理是，利用了alarm(2)設(shè)置的定時(shí)器，在一定時(shí)間過(guò)后會(huì)產(chǎn)生SIGALRM信號(hào)，會(huì)使當(dāng)前正在
執(zhí)行的系統(tǒng)調(diào)用中斷，導(dǎo)致該系統(tǒng)調(diào)用(fcntl)返回失敗。
上述代碼有以下的說(shuō)明：
1. 信號(hào)處理函數(shù)hander是一個(gè)空函數(shù)，里面什么也不做。它的存在就是為了接收SIGALRM信號(hào)
2. sigaction的sa_flags成員一定要設(shè)置成0，否則不會(huì)是系統(tǒng)調(diào)用中斷
3. 為了防止把以前設(shè)置的定時(shí)器破壞，不管是加鎖成功還是失敗都立即恢復(fù)以前的定時(shí)器。
4. 因?yàn)闉榱私邮?strong>SIGALRM信號(hào)，我們?cè)O(shè)置了它的信號(hào)處理函數(shù)。那在加鎖失敗和成功后也要恢復(fù)以前的設(shè)定。
注：雖然上面的代碼能實(shí)現(xiàn)文件鎖超時(shí)等待的問(wèn)題，但又引入了另一個(gè)問(wèn)題，就是該代碼會(huì)破壞以前設(shè)定的定時(shí)器，即使是后面也恢復(fù)了以前的定時(shí)器設(shè)置，也會(huì)有一些副作用。比如：當(dāng)為了等待其他進(jìn)程釋放文件鎖，傳遞到lockw函數(shù)里的等待時(shí)間(也就是形參timeout)超過(guò)了以前設(shè)定的定時(shí)器觸發(fā)時(shí)間，那這段期間內(nèi)的以前設(shè)定的定時(shí)器就無(wú)效了。也就是說(shuō), 在調(diào)用lockw之前,該進(jìn)程了已經(jīng)設(shè)定了一個(gè)2秒的定時(shí)器, 而這個(gè)進(jìn)程在調(diào)用lockw時(shí)傳遞的timeout時(shí)間為10秒(鎖定的阻塞時(shí)間為10秒), 那么從調(diào)用lockw的那一刻起,2秒的定時(shí)器就無(wú)效了,知道鎖定成功或者失敗為止.