中国IT动力,最新最全的IT技术教程
最新100篇 | 推荐100篇 | 专题100篇 | 排行榜 | 搜索 | 在线API文档 | 网通镜像
首 页 | 程序开发 | 操作系统 | 软件应用 | 图形图象 | 网络应用 | 精文荟萃 | 教育认证 | 硬件维护 | 未整理篇 | 站长教程
ASP JS PHP工程 ASP.NET 网站建设 UML J2EESUN .NET VC VB VFP 网络维护 数据库 DB2 SQL2000 Oracle Mysql
服务器 Win2000 Office C DreamWeaver FireWorks Flash PhotoShop 上网宝典 CorelDraw 协议大全 网络安全 微软认证
硬件维护  CPU  主板  硬盘  内存  显卡  显示器  键盘鼠标  声卡音箱  打印机  机箱电源  BIOS  网卡  C#  Java  Delphi  vs.net2005
  当前位置:> 程序开发 > 编程语言 > C/C++
巧用信号量防进程重入
作者:未知 时间:2005-09-13 23:28 出处:Blog.ChinaUnix.net 责编:chinaitpower
              摘要:巧用信号量防进程重入
对于某些程序尤其是服务程序,我们一般不希望用户重复运行,禁止程序重复运行的方式有多种,在linux下我选择利用信号量来禁止程序多次运行。

原理:程序执行时在内存中创建一个特定Key的信号量,并对该信号量的值做加1计数操作,并设置为进程退出自动减1。因此程序在启动时判断这个特定Key的信号量的值是否大于0,如大于0则表示已有实例在运行,如等于0或不存在这个信号量则说明尚没有实例运行。

//信号量操作权限,0644即主用户(属主)可读写、组成员及其它成员可读不可写
const int SEM_PRMS = 0644;

class COnlyOne
{
public:
bool Exists(key_t _nKey);//检查信号量是否已存在
bool Mark(key_t _nKey);//设置信号量
bool Unmark(key_t _nKey);//清除信号量
};

bool gnl_lnx::COnlyOne::Exists(key_t _nKey)
{
int nID = semget(_nKey, 0, 0);
if (nID == -1) return false;

return semctl(nID, 0, GETVAL, NULL) > 0;
}

bool gnl_lnx::COnlyOne::Mark(key_t _nKey)
{
int nID = semget(_nKey, 0, 0);
if (nID == -1)
nID = semget(_nKey, 1, IPC_CREAT | IPC_EXCL | SEM_PRMS);

struct sembuf buf[2];
buf[0].sem_num = 0;
buf[0].sem_op = 0;
buf[0].sem_flg = IPC_NOWAIT;
buf[1].sem_num = 0;
buf[1].sem_op = 1;
buf[1].sem_flg = SEM_UNDO;//进程退出时自动回滚
return semop(nID, &buf[0], 2) == 0;
}

bool gnl_lnx::COnlyOne::Unmark(key_t _nKey)
{
int nID = semget(_nKey, 0, 0);
if (nID == -1) return true;
return semctl(nID, 0, IPC_RMID, 0) == 0;
}

调用示例

if (m_OnlyOne.Exists(20040901))
{
std::cout << "Another program with same ID is running!" << std::endl;
return -1;
}

if (!m_OnlyOne.Mark(20040901))
{
std::cout << "Create semaphore failed." << std::endl;
return -1;
}
关闭本页
 
首页 | 投资与合作 | 服务条款 | 隐私政策 | 收藏本站 | 设为首页 | 新用户注册 | 免责声明 | 使用帮助
Copyright ©2005-2008 chinaitpower.com All rights reserved. www.chinaitpower.com 版权所有