| 对于某些程序尤其是服务程序,我们一般不希望用户重复运行,禁止程序重复运行的方式有多种,在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; } |