Linux進程間的五種通信方式介紹 5

3、For Example

下面這個例子，使用了【共享內(nèi)存+信號量+消息隊列】的組合來實現(xiàn)服務(wù)器進程與客戶進程間的通信。

• 共享內(nèi)存用來傳遞數(shù)據(jù)；
• 信號量用來同步；
• 消息隊列用來 在客戶端修改了共享內(nèi)存后 通知服務(wù)器讀取。

server.c

#include#include // shared memory#include // semaphore#include // message queue
// 消息隊列結(jié)構(gòu)struct msg_form {    long mtype;    char mtext;};
// 聯(lián)合體，用于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;}
// 創(chuàng)建一個信號量集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);  /初值設(shè)為1資源未占用/    return sem_id;}
int main(){    key_t key;    int shmid, semid, msqid;    char shm;    char data[] = "this is server";    struct shmid_ds buf1;  /用于刪除共享內(nèi)存/    struct msqid_ds buf2;  /用于刪除消息隊列/    struct msg_form msg;  /消息隊列用于通知對方更新了共享內(nèi)存/
    // 獲取key值    if((key = ftok(".", 'z')) < 0)    {        perror("ftok error");        exit(1);    }
    // 創(chuàng)建共享內(nèi)存    if((shmid = shmget(key, 1024, IPC_CREAT|0666)) == -1)    {        perror("Create Shared Memory Error");        exit(1);    }
    // 連接共享內(nèi)存    shm = (char*)shmat(shmid, 0, 0);    if((int)shm == -1)    {        perror("Attach Shared Memory Error");        exit(1);    }
    // 創(chuàng)建消息隊列    if ((msqid = msgget(key, IPC_CREAT|0777)) == -1)    {        perror("msgget error");        exit(1);    }
    // 創(chuàng)建信號量    semid = creat_sem(key);    // 讀數(shù)據(jù)    while(1)    {        msgrcv(msqid, &msg, 1, 888, 0); /讀取類型為888的消息/        if(msg.mtext == 'q')  /quit - 跳出循環(huán)/             break;        if(msg.mtext == 'r')  /read - 讀共享內(nèi)存/        {            sem_p(semid);            printf("%s\\n",shm);            sem_v(semid);        }    }
    // 斷開連接    shmdt(shm);
    /刪除共享內(nèi)存、消息隊列、信號量/    shmctl(shmid, IPC_RMID, &buf1);    msgctl(msqid, IPC_RMID, &buf2);    del_sem(semid);    return 0;}