信号灯之光借我一用

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

??? return -1;

}

if (!m_OnlyOne.Mark(20040901))

{

??? std::cout

??? return -1;

}