#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);//删除创建的文件
}