linux文件锁例程

#include <stdio.h>
#include <unistd.h>
#include <signal.h>
#include <fcntl.h>
#include <sys/wait.h>
#include <string.h>


#define LOCKMODE (S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)
/*
 * success return -
 * failed return -1(if cmd = F_SETLK err = EACCES or EAGAIN)
 * cmd (F_GETLK F_SETLK F_SETLKW)
 * l_type (F_RDLCK F_WRLCK F_UNLCK)
 */
static int lock_reg(int fd, int cmd, int type, off_t offset, int whence, off_t len)
{
struct flock lock;


lock.l_type = type;
lock.l_start = offset;
lock.l_whence = whence;
lock.l_len = len;

return(fcntl(fd, cmd, &lock));
}


#define READ_LOCK(fd, offset, whence, len) \
lock_reg((fd), F_SETLK, F_RDLCK, (offset), (whence), (len))
#define READ_LOCKW(fd, offset, whence, len) \
lock_reg((fd), F_SETLKW, F_RDLCK, (offset), (whence), (len))
#define WRITE_LOCK(fd, offset, whence, len) \
lock_reg((fd), F_SETLK, F_WRLCK, (offset), (whence), (len))
#define WRITE_LOCKW(fd, offset, whence, len) \
lock_reg((fd), F_SETLKW, F_WRLCK, (offset), (whence), (len))
#define UN_LOCK(fd, offset, whence, len) \
lock_reg((fd), F_SETLK, F_UNLCK, (offset), (whence), (len))


static int lock_test(int fd, int type, off_t offset, int whence, off_t len)
{
struct flock lock;


lock.l_type = type;
lock.l_start = offset,
lock.l_whence = whence,
lock.l_len = len;


if(fcntl(fd, F_GETLK, &lock) < 0){
printf("fcntl error\n");
}
if(lock.l_type == F_UNLCK){
return 0;
}
return (lock.l_pid);
}
#define IS_READ_LOCK(fd, offset, whence, len) \
(lock_test((fd), F_RDLCK, (offset), (whence), (len)) == 0)
#define IS_WRITE_LOCK(fd, offset, whence, len) \
(lock_test((fd), F_WRLCK, (offset), (whence), (len)) == 0)
int main(int argc, char *argv[])
{
int fd;
char *buf = "hello world\n";
pid_t pid;


if(argc != 2){
printf("usage: %s lock_file\n", argv[0]);
_exit(-1);
}
if((pid =fork()) < 0){
printf("%s(%u): fork error\n", argv[0], (unsigned int)getpid());
_exit(-3);
}else if(pid == 0){
if((fd = open(argv[1], O_WRONLY | O_CREAT, LOCKMODE)) < 0){
printf("%s(%u): open %s error\n", argv[0], (unsigned int)getpid(), argv[1]);
_exit(-2);
}
while(1){
sleep(2);
if(IS_WRITE_LOCK(fd, 0, SEEK_SET, 0)){
printf("%s(%u): be locked\n", argv[1], (unsigned int)getpid());
close(fd);
_exit(1);
}else{
printf("%s(%u): unlocked\n", argv[1], (unsigned int)getpid());
}
}
}
if((fd = open(argv[1], O_WRONLY | O_CREAT, LOCKMODE)) < 0){
printf("%s(%u): open %s error\n", argv[0], (unsigned int)getpid(), argv[1]);
_exit(-2);
}
if(write(fd, buf, strlen(buf)) < 0){
printf("%s(%u): write error\n", argv[0], (unsigned int)getpid());
}
WRITE_LOCK(fd, 0, SEEK_SET, 0);
sleep(4);
UN_LOCK(fd, 0, SEEK_SET, 0);
close(fd);
waitpid(pid, NULL, 0);

return 0;
}