Unix学习笔记——进程间通信之消息队列

学习消息队列时候写的小练习，纯娱乐性质...

 

程序功能：

       服务端进程接受消息类型为1001的消息，然后取消息数据域内的原客户进程号，将消息的消息类型更改为原客户进程进程号，再把消息返回给客户端。

       客户端往服务端发送消息，并接受从服务端返回消息类型等于客户端进程号的消息。

 

---

程序用到的头文件

 

 #ifndef __IPCDEF_H__#define __IPCDEF_H__#include <stdio.h>#include <unistd.h>#include <string.h>/************************************** *File name : ipcdef.h * Author : Sai * Modify date : 2010/06/10 * Version : Initial * Description : * **************************************//*common head file*/#include <stdlib.h>#include <sys/types.h>#include <sys/ipc.h>#include <sys/msg.h>#include <errno.h>/*macro for my own program*/#define MSG_MAX_SIZE 100#define MY_SND_MSG_KEY (key_t)19861212#define MY_RCV_MSG_KEY (key_t)19861213#define MSGTYPE 1001L/*typedef message struct*/typedef struct MyMsg{long msgtype;char msgdata[MSG_MAX_SIZE];}MyMsg_Def;#endif

---

初始化消息队列程序

 

 /************************************** *File name : initmsgque.c * Author : Sai * Modify date : 2010/06/10 * Version : Initial * Description : Create message queue * **************************************/#include "ipcdef.h"int main(void){int snd_msgid = 0;int rcv_msgid = 0;MyMsg_Def msgbuf;]memset(&msgbuf, 0, sizeof(MyMsg_Def));snd_msgid = msgget(MY_SND_MSG_KEY, 0666|IPC_CREAT);if(snd_msgid<0){fprintf(stderr, "Init send_queue failed!/n");exit(1);}rcv_msgid = msgget(MY_RCV_MSG_KEY, 0666|IPC_CREAT);if(rcv_msgid<0){fprintf(stderr, "Init receve_queue failed!/n");exit(1);}fprintf(stderr, "Init Message Queue Successful!/n");return 0;} 

---

消息队列删除程序

 

/************************************** *File name : delmsgque.c * Author : Sai * Modify date : 2010/06/10 * Version : Initial * Description : Delete message queue * **************************************/#include "ipcdef.h"int main(void){int snd_msgid = 0;int rcv_msgid = 0;MyMsg_Def msgdef;memset(&msgdef, 0, sizeof(MyMsg_Def));/*delete send_queue*/snd_msgid = msgget(MY_SND_MSG_KEY, 0);if(snd_msgid>=0)msgctl(snd_msgid, IPC_RMID, NULL);/*delete receve_queue*/rcv_msgid = msgget(MY_RCV_MSG_KEY, 0);if(rcv_msgid>=0)msgctl(rcv_msgid, IPC_RMID, NULL);fprintf(stderr, "Delete Message Queue Successful!/n");return 0;}

---

服务端程序

 

/************************************** *File name : server.c * Author : Sai * Modify date : 2010/06/10 * Version : Initial * Description : * **************************************/#include "ipcdef.h"int main(int argc, char *argv[]){MyMsg_Def msgbuf;;int snd_msgid = 0;int rcv_msgid = 0;int ret = 0;memset(&msgbuf, 0, sizeof(MyMsg_Def));snd_msgid = msgget(MY_SND_MSG_KEY, 0666|IPC_CREAT);if(snd_msgid<0){fprintf(stderr, "FILE-[%s]:Create SND_MSG %d error!/n", __FILE__, MY_SND_MSG_KEY);exit(1);}rcv_msgid = msgget(MY_RCV_MSG_KEY, 0666|IPC_CREAT);if(rcv_msgid<0){fprintf(stderr, "FILE-[%s]:Create RCV_MSG %d error!/n", __FILE__, MY_RCV_MSG_KEY);exit(1);}fprintf(stderr, "FILE-[%s]:Begin to Listen/n", __FILE__);while(1){ret = msgrcv(rcv_msgid, &msgbuf, MSG_MAX_SIZE, MSGTYPE, MSG_NOERROR);if(ret<0){fprintf(stderr, "FILE-[%s]:MSG receve error!/n", __FILE__);exit(1);}msgbuf.msgtype = atol(msgbuf.msgdata);ret = msgsnd(snd_msgid, &msgbuf, MSG_MAX_SIZE, MSG_NOERROR);if(ret){fprintf(stderr, "FILE-[%s] send MSG error!/n", __FILE__);exit(1);}}return 0;}

---

客户端程序

 

/************************************** *File name : client.c * Author : Sai * Modify date : 2010/06/10 * Version : Initial * Description : * **************************************/#include "ipcdef.h"int main(int argc, char *argv[]){MyMsg_Def msgbuf;int ret = 0;int snd_msgid = 0;int rcv_msgid = 0;char data[MSG_MAX_SIZE];int pid = 0;memset(&msgbuf, 0, sizeof(msgbuf));memset(&data, 0, sizeof(data));if(argc!=2){fprintf(stderr, "FILE-[%s]:The number of argc is not enough!/n", __FILE__);exit(1);}snd_msgid = msgget(MY_RCV_MSG_KEY, 0666);if(snd_msgid<0){fprintf(stderr, "FILE-[%s], LINE-[%d]:Create SND_MSG error!/n",__FILE__,__LINE__);exit(1);}rcv_msgid = msgget(MY_SND_MSG_KEY, 0666);if(rcv_msgid<0){fprintf(stderr, "FILE-[%s], LINE-[%d]:Create RCV_MSG error!/n",__FILE__,__LINE__);exit(1);}msgbuf.msgtype = MSGTYPE;pid = getpid();sprintf(data, "%d", pid);strcpy(msgbuf.msgdata, data);ret = msgsnd(snd_msgid, &msgbuf, MSG_MAX_SIZE, IPC_NOWAIT);if(ret){fprintf(stderr, "FILE-[%s], LINE-[%d]:Send MSG error!/n", __FILE__, __LINE__);exit(1);}fprintf(stdout, "FILE-[%s], LINE-[%d]:Send MSG success!/n", __FILE__, __LINE__);ret = msgrcv(rcv_msgid, &msgbuf, MSG_MAX_SIZE, getpid(), IPC_NOWAIT);if(ret<0){fprintf(stderr, "FILE-[%s], LINE-[%d]:Receve MSG error!/n", __FILE__, __LINE__);exit(1);}fprintf(stdout, "FILE-[%s], LINE-[%d]:Receve MSG successful!/n", __FILE__, __LINE__);fprintf(stdout, "MSG_TYPE=[%ld], MSG_DATA=[%s]/n", msgbuf.msgtype, msgbuf.msgdata);return 0;}

---

Makefile文件

 

##############################################File name : server.c# Author : Sai# Modify date : 2010/06/10# Version : Initial# Description : makefile for all############################################# cflags = -Wall -g -I./ -L./all:clean client server delmsgque initmsgque-rm -rf *.o@echo "########## build successful ###########"client:client.ogcc $(cflags) -o $@ $^server:server.ogcc $(cflags) -o $@ $^delmsgque:delmsgque.ogcc $(cflags) -o $@ $^initmsgque:initmsgque.ogcc $(cflags) -o $@ $^%.o:%.cgcc $(cflags) -c $^.PHONY:cleanclean:-rm -rf client server delmsgque *.o */~@echo "######### clean up finished! ###########"

 

提示：先在后台启动服务器程序，然后运行客户端.