进程通信—卖票系统的实现

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

typedef struct _shm
{
    int flag;
    int ticket;
}SHM;

void sellTicket(SHM* pshm)
{
    while(1)
    {
        int time = rand() %10 + 1;
        usleep(time *100000);
        if (pshm->flag == 1)
        {
            pshm->flag == 0;
            if (pshm->ticket == 0)
            {
                pshm->flag == 1;
                break;
            }
            printf ("卖掉第 %d张票\n",pshm->ticket);
            pshm->ticket--;
            pshm->flag = 1;
        }
    }
}

int main(int argc,char **argv)
{
    srand((unsigned int)time(NULL));
    //1.创建或者获取一个共享内存
    int shmid = shmget((key_t)1234 , sizeof(SHM), 0666|IPC_CREAT);
    if (shmid == -1)
    {
        perror ("shmget");
        return -1;
    }

    //2.将共享内存映射到当前的进程中
    SHM* pshm = (SHM*)shmat(shmid, NULL, 0);
    if (pshm == (void*)-1)
    {
        perror ("shmat");
        return -1;
    }

    //如果命令行参数为2，负责对共享内存进行初始化
    if (argc == 2)
    {
        pshm->flag = 1;
        pshm->ticket = 100;
    }

    //开始卖票
    sellTicket(pshm);

    //如果命令行参数为2，负责对共享内存进行删除
    if (argc == 2)
    {
        shmctl(shmid, IPC_RMID, NULL);
    }
    return 0;
}