共享内存最大的问题是什么?没错,就是多进程竞争内存的问题,就像类似于我们平时说的线程安全问题。如何解决这个问题?这个时候我们的信号量就上场了。信号量的本质就是一个计数器,用来实现进程之间的互斥与同步。例如信号量的初始值是 1,然后 a 进程来访问内存1的时候,我们就把信号量的值设为 0,然后进程b 也要来访问内存1的时候,看到信号量的值为 0 就知道已经有进程在访问内存1了,这个时候进程 b 就会访问不了内存1。所以说,信号量也是进程之间的一种通信方式。
下面看具体代码实现:
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/sem.h>
#include <time.h>
#ifdef __linux
union semun //macos 系统中在sys/sem.h文件中已定义
{
int val;
struct semid_ds *buf;
unsigned short *array;
};
#endif
static int sem_id = 0;
static int set_semvalue();
static int semaphore_p();
static int semaphore_v();
static void del_semvalue();
void printCurrentTime() //打印当前时间
{
time_t timep;
struct tm *p;
time(&timep);
p = gmtime(&timep);
printf("%d-%d-%d %d:%d:%d\n", (1900 + p->tm_year), (1 + p->tm_mon), p->tm_mday,
(8 + p->tm_hour), p->tm_min, p->tm_sec);
}
int main(int argc, char *argv[])
{
char message = 'x';
int i = 0;
//创建信号量
sem_id = semget((key_t)1234, 1, 0666 | IPC_CREAT);
if(argc > 1)
{
//程序第一次被调用,初始化信号量
if(!set_semvalue())
{
printf("Failed to initialize semaphore\n");
exit(EXIT_FAILURE);
}
//设置要输出到屏幕中的信息,即其参数的第一个字符
message = argv[1][0];
sleep(2);
}
for(int i = 0; i < 10; ++i)
{
//进入临界区
if(!semaphore_p()) //占用资源,别的进程先等待
exit(EXIT_FAILURE);
//向屏幕中输出数据
printf("getpid() = %d i = %d message = %c ", getpid(), i, message);
printCurrentTime();
//清理缓冲区,然后休眠随机时间
fflush(stdout);
sleep(rand() % 3);
//离开临界区前再一次向屏幕输出数据
printf("getpid() = %d i = %d message = %c ", getpid(), i, message);
printCurrentTime();
fflush(stdout);
//离开临界区,休眠随机时间后继续循环
if(!semaphore_v()) //释放资源,别的进程就可以进入临界区
exit(EXIT_FAILURE);
sleep(rand() % 2);
}
printCurrentTime();
sleep(10);
printf("\n%d - finished\n", getpid());
printCurrentTime();
if(argc > 1)
{
//如果程序是第一次被调用,则在退出前删除信号量
sleep(3);
del_semvalue();
}
printCurrentTime();
printf("Hello World!\n");
return 0;
}
static int set_semvalue()
{
//用于初始化信号量,在使用信号量前必须这样做
union semun sem_union;
sem_union.val = 1;
//SETVAL:用来把信号量初始化为一个已知的值
if(semctl(sem_id, 0, SETVAL, sem_union) == -1)
return 0;
return 1;
}
static void del_semvalue()
{
//删除信号量
union semun sem_union;
if(semctl(sem_id, 0, IPC_RMID, sem_union) == -1)
printf("Failed to delete semaphore\n");
}
static int semaphore_p()
{
//对信号量做减1操作,即等待P(sv)
struct sembuf sem_b;
sem_b.sem_num = 0;
sem_b.sem_op = -1; //P()
sem_b.sem_flg = SEM_UNDO;
if(semop(sem_id, &sem_b, 1) == -1)
{
printf("semaphore_p failed\n");
return 0;
}
return 1;
}
static int semaphore_v()
{
//这是一个释放操作,它使信号量变为可用,即发送信号V(sv)
struct sembuf sem_b;
sem_b.sem_num = 0;
sem_b.sem_op = 1;//V()
sem_b.sem_flg = SEM_UNDO;
if(semop(sem_id, &sem_b, 1) == -1)
{
printf("semaphore_v failed\n");
return 0;
}
return 1;
}
Ubuntu16.4运行结果:
Mac10.14.6运行结果:
参考:
https://blog.csdn.net/xiajun07061225/article/details/8475738
https://blog.csdn.net/ljianhui/article/details/10243617