有一些程序在同一服务器同一时间只能运行一个,因此这样的程序应该需要一个排外的处理。当程序再次运行时,检测到该程序已经运行了,那么程序停止结束。
文件和记录锁(fcntl()和F_SETLK(设置记录锁))提供一种方法确保只有程序只被正在运行一次。如果每一个程序对一个固定的文件进行加锁(写,只有这个锁)。连续对该文件追加记录锁会失败,也意味着该程序正在被运行。这个记录锁是互斥的,如果一个程序对文件追加了记录锁,当这个程序结束时,这个记录锁就会被移除(无论是强制退出还是手动退出)。另外,向文件写入内容时,不要忘记了清空前次写入的内容噢。
代码如下所示:
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <errno.h>
#include <sys/stat.h>
#include <syslog.h>
#define LOCKFIFE "/var/run/daemon.pid"
/* 参数mode设定,只有在建立新文件时才会有效
* S_IRUSR: 代表该文件所有者具有可读的权限
* S_IWUSR: 代表该文件所有者具有可写的权限
* S_IRGRP: 代表该文件用户组具有可读的权限
* S_IROTH: 代表该其他用户具有可读的权限 */
#define LOCKMODE (S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH)
int lockfile(int fd)
{
struct flock f1;
f1.l_type = F_WRLCK;
f1.l_start = 0;
f1.l_whence = SEEK_SET;
f1.l_len = 0;
/* F_SETLK: 设置记录锁 */
return (fcntl(fd, F_SETLK, &f1));
}
int already_running(void)
{
int fd;
char buf[16];
/* ORDWR: 以可读写方式打开文件
* OCREAT: 如文件不存在,则创建该文件 */
fd = open(LOCKFIFE, O_RDWR|O_CREAT, LOCKMODE);
if(fd < 0){
exit(1);
}
if(lockfile(fd)){
/* 错误码 errno:
* EAGAIN: Try again
* EACCES: Permission denied */
if(errno == EACCES || errno == EAGAIN){
close(fd);
return 1;
}
exit(1);
}
/* 改变文件大小,清空文件 */
ftruncate(fd, 0);
/* 写入进程号,可供管理员等查看 */
sprintf(buf, "%ld", (long)getpid());
write(fd, buf, strlen(buf)+1);
return 0;
}
int main(int argc, char *argv[])
{
int err;
if(already_running()){
printf("daemon already_running!\n");
return 1;
}
while(1){
sleep(1);
}
return 0;
}