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;
}



  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值