封装信号灯操作并结合共享内存应用

---

前言

本程序皆为进程间通信，若需测试可运行两个相同进程
实现方法参照https://www.freecplus.net/91049192da9e435a92209b287a220af8.html

---

一、信号灯操作类

// 此类封装了信号灯操作
class CSEM
{
private:
	union semun
	{
		int val;
		struct semid_ds *buf;
		unsigned short *arry;
	};
	
	int sem_id;		// 信号灯描述符

public:
	bool init(key_t key); 	// 如果信号灯已存在，获取信号灯；如果信号灯不存在，则创建信号灯并初始化。
	bool wait();          // 等待信号灯挂出。
	bool post();          // 挂出信号灯。
    bool destroy();       // 销毁信号灯。

// CSEM类成员函数

bool CSEM::init(key_t key)
{
	// 获取信号灯
	if ( (sem_id = semget(key,1,0640)) == -1)
	{
		// 如果信号灯不存在，则创建它
		if (errno == 2)
		{
			if ( (sem_id = semget(key,1,0640|IPC_CREAT)) ==-1) { perror("init 1 semget()"); return false; }

			// 信号灯创建成功后，还需要把它初始化成可用的状态。
			union semun sem_union;
			sem_union.val = 1;
			if (semctl(sem_id,0,SETVAL,sem_union) < 0) { perror("init semctl()"); return false; }
		}
		else
		{ perror("init 2 semget()"); return false; }
	}

	return true;
}

bool CSEM::destroy()
{
	if (semctl(sem_id,0,IPC_RMID) == -1) { perror("destroy semctl()"); return false; }
	
	return true;
}

bool CSEM::wait()
{
	struct sembuf sem_b;
	sem_b.sem_num = 0;
	sem_b.sem_op = -1;
	sem_b.sem_flg = SEM_UNDO;
	if (semop(sem_id, &sem_b, 1) == -1) { perror("wait semop()"); return false; }
	
	return true;
}

bool CSEM::post()
{
	struct sembuf sem_b;
	sem_b.sem_num = 0;
	sem_b.sem_op = 1;
	sem_b.sem_flg = SEM_UNDO;
	if (semop(sem_id, &sem_b, 1) == -1) { perror("wait semop()"); return false; }

	return true;
}
};

二、操作主函数

/*
 * 此程序用于练习共享内存与信号量
 * 作者：zhy 
 * 日期：2020/4/10
 */
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/ipc.h>
#include <sys/sem.h>

int main()
{
	int shmid;	// 共享内存标识符
	CSEM sem;	// 声明信号量类
	

	// 创建共享内存，键值为0x5005，共1024字节。
	if ( (shmid = shmget( (key_t)0x5006,1024,0644|IPC_CREAT)) == -1 )
	{
		printf("shmat(0x5005) failed\n"); return -1; 
	}

	if (sem.init(0x5000) == false)  { printf("sem.init failed. \n"); return -1; }
	printf ("sem.init ok\n");

	char *ptext = (char *)shmat(shmid,0,0);	// 连接至共享内存

	// 等待信信号挂出，等待成功后，将持有锁。
	if (sem.wait() == false)  { printf("sem.wait failed. \n"); return -1; }
	printf ("sem.wait ok\n");

	printf("写入前：%s\n",ptext);
	sprintf(ptext,"本程序的进程号是：%d",getpid());
	printf("写入后：%s\n",ptext);

	sleep(15);
	// 释放锁
	if (sem.post() == false) { printf("sem.post failed. \n"); return -1; }
	printf ("sem.post ok\n");
}