IO信号灯集

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

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

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