Unix XSI IPC 消息佇列完整範例程式

注意事項

Linux 2.4.22 限制：

• 可發送最長訊息位元組數：8192
• 佇列最大容量位元組數：16384
• 佇列最大佇列容量數：16

key_t ftok(char* path, int id) 使用說明

• ftok 建立一個鍵值，是核心中佇列在外部的 ID 號。由於消息佇列處於核心中，只有建立者和核心知道佇列在核心裡的 ID 號，其他進程無法得知，因此需關聯一個外部鍵值進行關聯
• id 範圍：1-255
• 回傳核心消息佇列的 ID 號

server.c - 伺服器端程式

#include "msg.h" 
#include  
#include  
#include  
 
int main(int argc, char** argv) 
{ 
 int queid = open_msg("/root", 100); 
 
 while(1) 
 { 
 fputs("請輸入要發送的類型：1 or 2\n", stdout); 
 int type; 
 scanf("%d", &type); 
 
 switch(type) 
 { 
 case MYTYPE_ONE: 
 { 
 msg_send(queid, "MYTYPE_ONE", MYTYPE_ONE); 
 break; 
 } 
 case MYTYPE_TWO: 
 { 
 msg_send(queid, "MYTYPE_TWO", MYTYPE_TWO); 
 break; 
 } 
 default: 
 { 
 fputs("輸入類型錯誤，請重新輸入\n", stdout); 
 break; 
 } 
 } 
 
 fputs("輸入：q 為退出，其它表示繼續\n", stdout); 
 if(getchar() == 'q') 
 { 
 fputs("退出成功！\n", stdout); 
 break; 
 } 
 else 
 { 
 fputs("繼續發送消息\n", stdout); 
 } 
 } 
 // 不發送退出需要將佇列移除 
 del_que(queid); 
 
 return 0; 
} 

client.c - 客戶端程式

#include "msg.h" 
#include  
#include  
#include  
 
int main(int argc, char** argv) 
{ 
 int queid = open_msg("/root", 100); 
 
 while(1) 
 { 
 fputs("請接收要發送的類型：1 or 2\n", stdout); 
 int type; 
 scanf("%d", &type); 
 
 switch(type) 
 { 
 case MYTYPE_ONE: 
 { 
 msg_rec(queid, MYTYPE_ONE); 
 break; 
 } 
 case MYTYPE_TWO: 
 { 
 msg_rec(queid, MYTYPE_TWO); 
 break; 
 } 
 default: 
 { 
 fputs("輸入類型錯誤，請重新輸入\n", stdout); 
 break; 
 } 
 } 
 
 fputs("輸入：q 為退出，其它表示繼續\n", stdout); 
 if(getchar() == 'q') 
 { 
 fputs("退出成功！\n", stdout); 
 break; 
 } 
 else 
 { 
 fputs("繼續發送消息\n", stdout); 
 } 
 } 
 // 佇列移除 
 del_que(queid); 
 
 return 0; 
} 

msg.c - 訊息佇列核心操作實作

#include  
#include  
#include  
#include  
#include  
#include  
#include  
#include "msg.h" 
 
 // 如果存在佇列則開啟，沒有則建立 
int open_msg(char* path, int id) 
{ 
 // 獲取 IPC 物件的一個鍵值 
 key_t key = ftok(path, id); 
 if(-1 == key) 
 { 
 perror("ftok\n"); 
 exit(1); 
 } 
 // 建立一個佇列 
 int queid = msgget(key, IPC_CREAT | 0666); 
 if(-1 == queid) 
 { 
 perror("msgget\n"); 
 exit(1); 
 } 
 return queid; 
} 
 
 // 發送消息到佇列 
void msg_send(key_t key, char* text, long msgtype) 
{ 
 // 初始化內容 
 struct MSG tmp; 
 memset(&tmp, sizeof(struct MSG), 0); 
 tmp.mytype = msgtype; 
 strcpy(tmp.mytext, text); 
 // 發送消息 
 if(msgsnd(key, &tmp, TEXTSIZE, 0)) 
 { 
 perror("msgsnd\n"); 
 exit(1); 
 } 
} 
 
 // 從消息佇列獲取消息並顯示 
void msg_rec(key_t key, long msgtype) 
{ 
 struct MSG tmp; 
 if(-1 == msgrcv(key, &tmp, TEXTSIZE, msgtype, MSG_NOERROR)) 
 { 
 perror("msgrcv\n"); 
 exit(1); 
 } 
 printf("receive content: %s\n", tmp.mytext); 
} 
 
 // 刪除佇列，即使佇列裡面還有消息也一起刪除 
void del_que(key_t key) 
{ 
 if(msgctl(key, IPC_RMID, NULL)) 
 { 
 perror("msgsnd\n"); 
 exit(1); 
 } 
} 

msg.h - 標頭檔定義

#ifndef MSG_H 
#define MSG_H 
 
#include  
 
#define TEXTSIZE 100 
#define ARRYSIZE 2 
#define MYTYPE_ONE 1 
#define MYTYPE_TWO 2 
 
struct MSG 
{ 
 long mytype; 
 char mytext[TEXTSIZE]; 
}; 
 
int open_msg(char*, int); 
void msg_send(key_t, char*, long); 
 
#endif // end MSG_H