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

目录

题目

思路

代码

input.c

output.c

---

题目

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

思路

一、整体功能概述

 

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

 

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

 

1. union semun：

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

2. init 函数：

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

3. pv 函数：

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

4. input 进程（主函数）：

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

5. output 进程（主函数）：

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

 

三、执行流程分析

 

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

 

四、应用场景和注意事项

 

1. 应用场景：

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

2. 注意事项：

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

代码

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;
}