Merge pull request 'Guarder protected with semaphor' (#1) from semaphor-protection into main
Reviewed-on: #1
This commit was merged in pull request #1.
This commit is contained in:
@@ -4,20 +4,33 @@
|
||||
#include <sys/types.h>
|
||||
#include <sys/shm.h>
|
||||
#include <sys/ipc.h>
|
||||
#include <sys/sem.h>
|
||||
#include <unistd.h>
|
||||
|
||||
class guarder{
|
||||
private:
|
||||
const int shmid;
|
||||
const int semid;
|
||||
|
||||
void semaphoreOp(int op) {
|
||||
struct sembuf sb{};
|
||||
sb.sem_num = 0;
|
||||
sb.sem_op = op;
|
||||
sb.sem_flg = 0;
|
||||
semop(semid, &sb, 1);
|
||||
}
|
||||
|
||||
void setFlag(bool value)
|
||||
{
|
||||
semaphoreOp(-1);
|
||||
bool *flag = static_cast<bool *>(shmat(shmid, nullptr, 0));
|
||||
*flag = value;
|
||||
shmdt(flag);
|
||||
semaphoreOp(1);
|
||||
}
|
||||
|
||||
public:
|
||||
guarder() : shmid(shmget(IPC_PRIVATE, sizeof(bool), 0666 | IPC_CREAT))
|
||||
guarder() : shmid(shmget(IPC_PRIVATE, sizeof(bool), 0666 | IPC_CREAT)), semid(semget(IPC_PRIVATE, 1, 0666 | IPC_CREAT))
|
||||
{
|
||||
setFlagOff();
|
||||
}
|
||||
@@ -34,15 +47,18 @@ public:
|
||||
|
||||
bool isFlagOn()
|
||||
{
|
||||
semaphoreOp(-1);
|
||||
bool *flag = static_cast<bool *>(shmat(shmid, nullptr, 0));
|
||||
bool ret = *flag;
|
||||
shmdt(flag);
|
||||
semaphoreOp(1);
|
||||
return ret;
|
||||
}
|
||||
|
||||
~guarder()
|
||||
{
|
||||
shmctl(shmid, IPC_RMID, nullptr);
|
||||
semctl(semid, 0, IPC_RMID);
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user