??? 对于某些程序尤其是服务程序，我们一般不希望用户重复运行，禁止程序重复运行的方式有多种，在linux下我选择利用信号量来禁止程序多次运行。
??? 原理：程序执行时在内存中创建一个特定Key的信号量，并对该信号量的值做加1计数操作，并设置为进程退出自动减1。因此程序在启动时判断这个特定Key的信号量的值是否大于0，如大于0则表示已有实例在运行，如等于0或不存在这个信号量则说明尚没有实例运行。
const int SEM_PRMS = 0644;//信号量操作权限，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;
}
|
本类阅读TOP10