【嵌入式学习】IO进程线程day02.27

#ifndef _SEM_H_
#define _SEM_H_

int open_sem(int semcount);

int P(int semid, int semno);

int V(int semid, int semno);

int del_sem(int semid);

#endif

#include<myhead.h>

union semun {
    int              val;    /* Value for SETVAL */
    struct semid_ds *buf;    /* Buffer for IPC_STAT, IPC_SET */
    unsigned short  *array;  /* Array for GETALL, SETALL */
    struct seminfo  *__buf;  /* Buffer for IPC_INFO
                                (Linux-specific) */
};


int init_semno(int semid, int semno)
{
    union semun buf;
    printf("请输入要给编号为%d的灯设置的值：", semno);
    scanf("%d", &buf.val);

    //调用控制函数
    if(semctl(semid, semno, SETVAL, buf) == -1)
    {
        perror("semctl error");
        return -1;
    }

    return 0;
}

int open_sem(int semcount)
{
    //创建key值
    key_t key = -1;
    if((key = ftok("/", 's')) == -1)
    {
        perror("ftok error");
        return -1;
    }

    //通过key值创建一个信号灯集
    int semid = -1;
    if((semid = semget(key, semcount, IPC_CREAT|IPC_EXCL|0664)) == -1)
    {
        //对错误码进行判断，如果错误码为EEXIST，说明信号灯集已经存在
        if(errno == EEXIST)
        {
            //直接打开信号灯集即可
            semid = semget(key, semcount, IPC_CREAT|0664);
            return semid;
        }

        perror("semget error");
        return -1;

    }

    //给信号灯集中的每个灯进行初始化操作
    for(int i=0; i<semcount; i++)
    {
        init_semno(semid, i);           
    }

    return semid;

}

int P(int semid, int semno)
{
    //定义要进行操作的结构体变量
    struct sembuf buf;
    buf.sem_num = semno;
    buf.sem_op = -1;           //表示申请资源
    buf.sem_flg = 0;          //表示阻塞方式申请

    //执行函数
    if(semop(semid, &buf, 1) ==-1)
    {
        perror("P error");
        return -1;
    }

    return 0;
}


int V(int semid, int semno)
{
    //定义要进行操作的结构体变量
    struct sembuf buf;
    buf.sem_num = semno;
    buf.sem_op = 1;         
    buf.sem_flg = 0;        

    //执行函数
    if(semop(semid, &buf, 1) ==-1)
    {
        perror("V error");
        return -1;
    }

    return 0;
}


int del_sem(int semid)
{
    
    if(semctl(semid, 0, IPC_RMID) == -1)
    {
        perror("delete error");
        return -1;
    }

    return 0;
}

#include<myhead.h>
#include"sem.h"

#define PAGE_SIZE 4096     //一页的大小

int main(int argc, const char *argv[])
{
    //11、创建信号灯集
    int semid = open_sem(2);

    //1、创建key值
    key_t key = -1;
    if((key = ftok("/", 'k')) == -1)
    {
        perror("ftok error");
        return -1;
    }
    printf("key = %d\n", key);

    //2、通过key值创建一个共享内存
    int shmid = -1;
    if((shmid = shmget(key, PAGE_SIZE, IPC_CREAT|0664)) == -1)
    {
        perror("shmget error");
        return -1;
    }
    printf("shmid = %d\n", shmid);

    //3、将共享内存段的数据映射到用户空间
    char *addr = shmat(shmid, NULL, 0);
    //参数2：让系统自动选择对齐页
    //参数3：对共享内存的操作为读写操作
    if(addr == (void*)-1)
    {
        perror("shmat error");
        return -1;
    }
    printf("addr = %p\n", addr);          //?

    //4、使用共享内存
    while(1)
    {
        //22、申请1号灯的资源
        P(semid, 1);

        printf("收到的数据为：%s\n", addr);


        //判断输入内容
        if(strcmp(addr, "quit") == 0)
        {
            break;
        }

        //33、释放0号灯的资源
        V(semid, 0);


    }

    //5、取消映射关系
    if(shmdt(addr) == -1)
    {
        perror("shmdt error");
        return -1;
    }

    //44、删除信号灯集
    del_sem(semid);





    return 0;
}