Linux下防止程序重新启动

有一些程序在同一服务器同一时间只能运行一个,因此这样的程序应该需要一个排外的处理。当程序再次运行时,检测到该程序已经运行了,那么程序停止结束。

   文件和记录锁(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;
}

猜你喜欢

转载自blog.csdn.net/Ivan804638781/article/details/81660623
今日推荐