锁文件实现

#include <unistd.h>
#include <fcntl.h>


#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>




#define SEQFILE "./seqno"
#define FILE_MODE (S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)
#define LOCKFILE        "./seqno.lock"
#define MAXFILE 64


void my_lock(int fd);
void my_unlock(int fd);


int main(int argc, char **argv)
{
int fd;
long i = 0;
long seqno = 0;
pid_t pid;
ssize_t n;
char line[MAXFILE + 1];


pid = getpid();
printf("pid:%d\n", pid);


fd = open(SEQFILE, O_RDWR | O_CREAT, FILE_MODE);


if(fd < 0)
{
printf("Open %s failed\n", SEQFILE);
return 1;
}


if((pid = fork()) < 0)
{
perror("fork error\n");
return -1;
}


if(pid == 0)
{

my_lock(fd);

/×

lseek(fd, 0L, SEEK_SET);
n = read(fd, line, MAXFILE);
line[n] = '\0';


n = sscanf(line, "%ld\n", &seqno);
printf("%s:pid = %ld, seq# = %ld\n", argv[0], (long)getpid(), seqno);
seqno++;


sprintf(line, "%ld\n", seqno);
lseek(fd, 0L, SEEK_SET);
write(fd, line, strlen(line));
×/
my_unlock(fd);
}else{
my_lock(fd);
/× lseek(fd, 0L, SEEK_SET);
n = read(fd, line, MAXFILE);
line[n] = '\0';


n = sscanf(line, "%ld\n", &seqno);
printf("%s:pid = %ld, seq# = %ld\n", argv[0], (long)getpid(), seqno);
seqno++;


sprintf(line, "%ld\n", seqno);
lseek(fd, 0L, SEEK_SET);
write(fd, line, strlen(line));

×/
my_unlock(fd);

}
close(fd);
return 0;
}


void my_lock(int fd)
{
int tempfd = -1;


while((tempfd = open(LOCKFILE, O_RDWR | O_CREAT | O_EXCL,FILE_MODE)) < 0)//如果文件存在，则打开出错，说明已有进程在操作共享文件
{
if(errno != EEXIST)//判断文件不存在时，创建文件是否成功
{
printf("open error for lock file\n");
}
}
close(tempfd);
}


void my_unlock(int fd)
{
unlink(LOCKFILE);//删除创建的文件
}