#include <stdio.h>
#include <wait.h>
#include <stdlib.h>
#include <string.h> 
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <unistd.h>

#define SIZE 1024

const long id = 1000;//這裡規定型別為1000的訊息

typedef struct msgqueue{
        long mytype;//訊息型別
        char mytext[SIZE];//訊息內容
}msgqueue;

int main(){
    key_t k = ftok(".",0x1234);//申請key，確保兩個需要通訊的程序使用同一個key值獲取到的佇列id
    if(k < 0){
        perror("ftok");
        return -1;
    }
    int msgid = msgget(k,IPC_CREAT|IPC_EXCL|0666);//使用key值獲取id
    if(msgid < 0){
        perror("msgget");
        return -1;
    }
    pid_t pid = fork();
    if(pid < 0){
        perror("fork");
        return -1;
    }
    if(pid > 0){//父程序發訊息
        msgqueue queue;
        queue.mytype = id;//傳送的訊息型別為1000
        char msg[SIZE];
        while(1){//迴圈輸入，此時輸入的每一條訊息，訊息型別都是1000
            printf("father says# ");
//            fflush(stdout);
            fgets(msg,sizeof(msg)+1,stdin);//從標準輸入獲取字串，以回車換行結束
            strcpy(queue.mytext,msg);
            if(msgsnd(msgid,&queue,sizeof(msg),0) == -1){//將資訊掛載到訊息佇列裡
                perror("msgsnd");
                return -1;
            }
            if(strncmp(msg,"end",3) == 0){//只要輸入end，認為結束通訊
                break;
            }
            usleep(500000);
        }
        waitpid(pid,NULL,0);//等待子程序退出
    }else if(pid == 0){//子程序取訊息，注意訊息佇列裡只要有人拿了訊息，那麼拿走的訊息就不在訊息佇列裡了
        msgqueue queue;
        while(1){//迴圈拿訊息
            if(msgrcv(msgid,&queue,sizeof(queue.mytext),id,0) == -1){//從訊息佇列裡取訊息，取一個少一個
                perror("msgrcv");
                return -1;
            }
            if(strncmp(queue.mytext,"end",3)==0){//end就退出
                printf("father will quit,i quit too\n");
                break;
            }
            printf("child recv:%s\n",queue.mytext);
        }
        exit(EXIT_SUCCESS);
    }
    if(msgctl(msgid,IPC_RMID,NULL)<0){//刪除訊息佇列，因為訊息佇列建立後不會自己刪除，必須人為刪除或者重啟
        perror("msgctl");
        return -1;
    }
    printf("\n done \n");//沒什麼含義，只是為了表示這次的傳輸是正常退出的
    return 0;
}