Linux进程间的五种通信方式介绍 5

3、For Example

下面这个例子，使用了【共享内存+信号量+消息队列】的组合来实现服务器进程与客户进程间的通信。

• 共享内存用来传递数据；
• 信号量用来同步；
• 消息队列用来 在客户端修改了共享内存后 通知服务器读取。

server.c

复制#include复制#include // shared memory复制#include // semaphore复制#include // message queue复制
复制// 消息队列结构复制struct msg_form {复制    long mtype;复制    char mtext;复制};复制
复制// 联合体，用于semctl初始化复制union semun复制{复制    int              val; /for SETVAL/复制    struct semid_ds buf;复制    unsigned short  *array;复制};复制
复制// 初始化信号量复制int init_sem(int sem_id, int value)复制{复制    union semun tmp;复制    tmp.val = value;复制    if(semctl(sem_id, 0, SETVAL, tmp) == -1)复制    {复制        perror("Init Semaphore Error");复制        return -1;复制    }复制    return 0;复制}复制
复制// P操作:复制//  若信号量值为1，获取资源并将信号量值-1 复制//  若信号量值为0，进程挂起等待复制int sem_p(int sem_id)复制{复制    struct sembuf sbuf;复制    sbuf.sem_num = 0; /序号/复制    sbuf.sem_op = -1; /P操作/复制    sbuf.sem_flg = SEM_UNDO;复制
复制    if(semop(sem_id, &sbuf, 1) == -1)复制    {复制        perror("P operation Error");复制        return -1;复制    }复制    return 0;复制}复制
复制// V操作：复制//  释放资源并将信号量值+1复制//  如果有进程正在挂起等待，则唤醒它们复制int sem_v(int sem_id)复制{复制    struct sembuf sbuf;复制    sbuf.sem_num = 0; /序号/复制    sbuf.sem_op = 1;  /V操作/复制    sbuf.sem_flg = SEM_UNDO;复制
复制    if(semop(sem_id, &sbuf, 1) == -1)复制    {复制        perror("V operation Error");复制        return -1;复制    }复制    return 0;复制}复制
复制// 删除信号量集复制int del_sem(int sem_id)复制{复制    union semun tmp;复制    if(semctl(sem_id, 0, IPC_RMID, tmp) == -1)复制    {复制        perror("Delete Semaphore Error");复制        return -1;复制    }复制    return 0;复制}复制
复制// 创建一个信号量集复制intcreat_sem(key_t key)复制{复制    int sem_id;复制    if((sem_id = semget(key, 1, IPC_CREAT|0666)) == -1)复制    {复制        perror("semget error");复制        exit(-1);复制    }复制    init_sem(sem_id, 1);  /初值设为1资源未占用/复制    return sem_id;复制}复制
复制int main()复制{复制    key_t key;复制    int shmid, semid, msqid;复制    char shm;复制    char data[] = "this is server";复制    struct shmid_ds buf1;  /用于删除共享内存/复制    struct msqid_ds buf2;  /用于删除消息队列/复制    struct msg_form msg;  /消息队列用于通知对方更新了共享内存/复制
复制    // 获取key值复制    if((key = ftok(".", 'z')) < 0)复制    {复制        perror("ftok error");复制        exit(1);复制    }复制
复制    // 创建共享内存复制    if((shmid = shmget(key, 1024, IPC_CREAT|0666)) == -1)复制    {复制        perror("Create Shared Memory Error");复制        exit(1);复制    }复制
复制    // 连接共享内存复制    shm = (char*)shmat(shmid, 0, 0);复制    if((int)shm == -1)复制    {复制        perror("Attach Shared Memory Error");复制        exit(1);复制    }复制
复制    // 创建消息队列复制    if ((msqid = msgget(key, IPC_CREAT|0777)) == -1)复制    {复制        perror("msgget error");复制        exit(1);复制    }复制
复制    // 创建信号量复制    semid = creat_sem(key);复制复制    // 读数据复制    while(1)复制    {复制        msgrcv(msqid, &msg, 1, 888, 0); /读取类型为888的消息/复制        if(msg.mtext == 'q')  /quit - 跳出循环/ 复制            break;复制        if(msg.mtext == 'r')  /read - 读共享内存/复制        {复制            sem_p(semid);复制            printf("%s\\n",shm);复制            sem_v(semid);复制        }复制    }复制
复制    // 断开连接复制    shmdt(shm);复制
复制    /删除共享内存、消息队列、信号量/复制    shmctl(shmid, IPC_RMID, &buf1);复制    msgctl(msqid, IPC_RMID, &buf2);复制    del_sem(semid);复制    return 0;复制}