把信号灯集加到共享内存实现同步：输入输出，quit结束

嵌新程

已于 2024-09-05 00:07:14 修改

阅读量859

点赞数 12

分类专栏：练习文章标签：算法 linux 嵌入式 IO 嵌入式应用层开发

于 2024-09-05 00:06:37 首次发布

本文链接：https://blog.csdn.net/qr70892/article/details/141907312

版权

练习专栏收录该内容

70 篇文章 0 订阅

订阅专栏

题目

把信号灯集加到共享内存实现同步：输入输出，quit结束

思路

一、整体功能概述

这段代码使用共享内存和信号量集实现了两个进程之间的同步通信。一个进程负责从终端输入内容并存入共享内存，另一个进程从共享内存读取内容并输出，当输入为 “quit” 时，两个进程都结束。

二、主要函数和数据结构分析

union semun：

用于设置信号量的值。它包含一个整数成员 val，可以用来初始化或修改信号量的值。

init 函数：

接受信号量集标识符 semid、信号量编号 num 和初始值 val 作为参数。
创建一个 union semun 类型的变量 sem，并将传入的 val 赋值给 sem.val。
使用 semctl 函数将指定编号的信号量初始化为传入的 val。

pv 函数：

接受信号量集标识符 semid、信号量编号 num 和操作值 op 作为参数。
创建一个 struct sembuf 类型的变量 buf，设置信号量编号、操作值和标志位（这里设置为阻塞）。
使用 semop 函数对指定编号的信号量执行操作。

input 进程（主函数）：

生成共享内存和信号量集的键值 key，使用 ftok 函数以指定的文件（"sem.c"）和字符（'a'）生成唯一键值。
创建或打开共享内存，通过 shmget 和 shmat 函数获取共享内存的标识符和映射地址。
创建或打开信号量集，通过 semget 函数获取信号量集标识符。如果信号量集已存在，则初始化其中的两个信号量为 0 和 1。
在一个无限循环中，先对 1 号信号量执行 P 操作（pv(semid, 1, -1)），等待资源可用以进行输入操作。从终端读取输入并存入共享内存，然后对 0 号信号量执行 V 操作（pv(semid, 0, 1)），通知输出进程可以读取共享内存中的内容。如果输入为 “quit”，则跳出循环。
最后取消共享内存的映射，但没有删除共享内存和信号量集（注释掉了相关代码）。

output 进程（主函数）：

与 input 进程类似，生成键值、创建或打开共享内存和信号量集。
在一个无限循环中，先对 0 号信号量执行 P 操作（pv(semid, 0, -1)），等待输入进程将内容存入共享内存。如果读取到的内容为 “quit”，则跳出循环。否则，输出共享内存中的内容，并对 1 号信号量执行 V 操作（pv(semid, 1, 1)），通知输入进程可以进行下一次输入操作。
最后取消共享内存的映射，删除共享内存和信号量集。

三、执行流程分析

两个进程分别执行各自的 main 函数。
首先生成相同的键值，确保能够访问到相同的共享内存和信号量集。
input 进程创建或打开共享内存和信号量集后，等待 1 号信号量资源可用，然后从终端读取输入并存入共享内存，接着释放 0 号信号量通知 output 进程。
output 进程等待 0 号信号量资源可用，然后读取共享内存中的内容并输出，接着释放 1 号信号量通知 input 进程。
如此循环，直到 input 进程输入 “quit”，两个进程都检测到这个字符串并跳出循环，结束执行。

四、应用场景和注意事项

应用场景：

适用于需要两个或多个进程之间进行同步通信的场景，特别是当需要在进程之间共享数据并确保数据的一致性和顺序性时。
可以用于实现生产者 - 消费者模型，其中一个进程作为生产者将数据存入共享内存，另一个进程作为消费者从共享内存读取数据。

注意事项：

确保在使用共享内存和信号量集时进行适当的错误处理，以避免程序出现异常情况。
考虑在程序结束时正确地删除共享内存和信号量集，以释放系统资源。
注意信号量的初始值设置和操作顺序，以确保进程之间的同步正确无误。

代码

input.c

#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <stdio.h>
#include <errno.h>
#include <sys/shm.h>
#include <string.h>

union semun
{
    int val;
};

//初始化
void init(int semid, int num, int val)
{
    union semun sem;
    sem.val = val;
    semctl(semid, num, SETVAL, sem); //把num号信号灯初始化值为val
}

//PV操作
void pv(int semid, int num, int op)
{
    struct sembuf buf; //sembuf结构体人家写好的直接拿来用就可以
    buf.sem_num = num;
    buf.sem_op = op;
    buf.sem_flg = 0;       //阻塞
    semop(semid, &buf, 1); //对num号灯进行op操作
}

int main(int argc, char const *argv[])
{
    key_t key;
    int semid;
    int shmid;
    char *p;

    if ((key = ftok("sem.c", 'a')) < 0)
    {
        perror("ftok err");
        return -1;
    }
    printf("key: %#x\n", key);

    //创建或打开共享内存
    shmid = shmget(key, 128, IPC_CREAT | IPC_EXCL | 0777);
    if (shmid <= 0)
    {
        if (errno == EEXIST)
            shmid = shmget(key, 128, 0777);
        else
        {
            perror("shmget er");
            return -1;
        }
    }
    printf("shmid: %d\n", shmid);

    //映射共享内存
    p = (char *)shmat(shmid, NULL, 0);
    if (p == (char *)-1)
    {
        perror("shmat err");
        return -1;
    }

    //创建或打开信号灯集
    semid = semget(key, 2, IPC_CREAT | IPC_EXCL | 0777);
    if (semid <= 0)
    {
        if (errno == EEXIST)
            semid = semget(key, 2, 0777);
        else
        {
            perror("semget er");
            return -1;
        }
    }
    else //确保对信号灯集中的信号灯始终初始化一次，下次执行程序还继续获得上一次的值。如果修改初值，需要删除信号灯然后重新打开。
    {
        //初始化信号灯
        init(semid, 0, 0); //把0号信号灯初始化值为0
        init(semid, 1, 1); //把1号信号灯初始化值为1
    }
    printf("semid: %d\n", semid);

    //获取信号灯值
    printf("sem0: %d\n", semctl(semid, 0, GETVAL));
    printf("sem1: %d\n", semctl(semid, 1, GETVAL));

    while (1)
    {
        pv(semid, 1, -1);
        scanf("%s", p);
        pv(semid, 0, 1);
        if (strcmp(p, "quit") == 0)
            break;
    }

    shmdt(p);                      //取消映射
    // shmctl(shmid, IPC_RMID, NULL); //删除共享内存
    // semctl(semid, 0, IPC_RMID);    //删除信号灯集
    return 0;
}

output.c

#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <stdio.h>
#include <errno.h>
#include <sys/shm.h>
#include <string.h>

union semun
{
    int val;
};

//初始化
void init(int semid, int num, int val)
{
    union semun sem;
    sem.val = val;
    semctl(semid, num, SETVAL, sem); //把num号信号灯初始化值为val
}

//PV操作
void pv(int semid, int num, int op)
{
    struct sembuf buf; //sembuf结构体人家写好的直接拿来用就可以
    buf.sem_num = num;
    buf.sem_op = op;
    buf.sem_flg = 0;       //阻塞
    semop(semid, &buf, 1); //对num号灯进行op操作
}

int main(int argc, char const *argv[])
{
    key_t key;
    int semid;
    int shmid;
    char *p;

    if ((key = ftok("sem.c", 'a')) < 0)
    {
        perror("ftok err");
        return -1;
    }
    printf("key: %#x\n", key);

    //创建或打开共享内存
    shmid = shmget(key, 128, IPC_CREAT | IPC_EXCL | 0777);
    if (shmid <= 0)
    {
        if (errno == EEXIST)
            shmid = shmget(key, 128, 0777);
        else
        {
            perror("shmget er");
            return -1;
        }
    }
    printf("shmid: %d\n", shmid);

    //映射共享内存
    p = (char *)shmat(shmid, NULL, 0);
    if (p == (char *)-1)
    {
        perror("shmat err");
        return -1;
    }

    //创建或打开信号灯集
    semid = semget(key, 2, IPC_CREAT | IPC_EXCL | 0777);
    if (semid <= 0)
    {
        if (errno == EEXIST)
            semid = semget(key, 2, 0777);
        else
        {
            perror("semget er");
            return -1;
        }
    }
    else //确保对信号灯集中的信号灯始终初始化一次，下次执行程序还继续获得上一次的值。如果修改初值，需要删除信号灯然后重新打开。
    {
        //初始化信号灯
        init(semid, 0, 0); //把0号信号灯初始化值为0
        init(semid, 1, 1); //把1号信号灯初始化值为1
    }
    printf("semid: %d\n", semid);

    //获取信号灯值
    printf("sem0: %d\n", semctl(semid, 0, GETVAL));
    printf("sem1: %d\n", semctl(semid, 1, GETVAL));

    while (1)
    {
        pv(semid, 0, -1);
        if (strcmp(p, "quit") == 0)
            break;
        printf("shm: %s\n", p);
        pv(semid, 1, 1);
    }

    shmdt(p);                      //取消映射
    shmctl(shmid, IPC_RMID, NULL); //删除共享内存
    semctl(semid, 0, IPC_RMID);    //删除信号灯集
    return 0;
}