利用消息队列实现ECHO_SRV

server端代码:

server 从队列的mtype=1接收数据 在发到mtype=pid(client的进程id)



#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <errno.h>
#define MSGMAX 8192
#define ERR_EXIT(m) \
    do { \
        perror(m);\
        exit(EXIT_FAILURE);\
    }while(0)

struct msgbuf{
    long mtype;
    char mtext[MSGMAX];   //消息缓冲区
};


void echo_srv(int msgid)
{
    int n;
    struct msgbuf msg;
    memset(&msg, 0, sizeof(msg));
    while(1)
    {
        if((n = msgrcv(msgid, &msg, MSGMAX, 1, 0)) <0)
            ERR_EXIT("msgsnd");
        int pid;
        pid = *((int*)msg.mtext);

        fputs(msg.mtext+4, stdout);
        msg.mtype = pid;
        msgsnd(msgid, &msg, n, 0);
    }
}

int main(int argc, const char *argv[])
{
    int msgid;
    msgid = msgget(1111, 0);
    if(msgid == -1)
        ERR_EXIT("msgget");
    echo_srv(msgid);

    printf("read \n");
    return 0;
}

 

client端:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <errno.h>
#define MSGMAX 8192
#define ERR_EXIT(m) \
    do { \
        perror(m);\
        exit(EXIT_FAILURE);\
    }while(0)

struct msgbuf{
    long mtype;
    char mtext[MSGMAX];   //消息缓冲区
};




void echo_cli(int msgid)
{
    int pid;
    pid = getpid();
    struct msgbuf msg;
    memset(&msg, 0, sizeof(msg));
    msg.mtype = 1;
    *(int*)msg.mtext = pid;
    while(fgets(msg.mtext+4, MSGMAX, stdin) != NULL)
    {
//        msg.mtype = 1;
        if((msgsnd(msgid, &msg, 4+strlen(msg.mtext+4), IPC_NOWAIT)) <0)
            ERR_EXIT("msgsnd");
//
        memset(msg.mtext+4, 0, MSGMAX);
        if((msgrcv(msgid, &msg, MSGMAX, pid, 0)) <0)
            ERR_EXIT("msgrcv");

        fputs(msg.mtext+4, stdout);
        memset(msg.mtext+4, 0, MSGMAX-4);
    }

}


int main(int argc, const char *argv[])
{
    int msgid;
    msgid = msgget(1111, 0);
    if(msgid == -1)
        ERR_EXIT("msgget");
    
    echo_cli(msgid);

    return 0;
}

就是这个套路,熟练即可

posted @ 2015-03-01 18:25  tilly_chang  阅读(150)  评论(0编辑  收藏  举报