Linux--命名函数实现

Makefile:

.PHONY:all
all:client mutiServer

client:client.cc
	g++ -o $@ $^ -std=c++11
mutiServer:server.cc
	g++ -o $@ $^ -std=c++11

.PHONY:clean
clean:
	rm -f client mutiServer

server.cc

#include "comm.hpp"
#include <sys/wait.h>

static void getMessage(int fd)
{
    char buffer[SIZE];
    while(true)
    {
        memset(buffer,'\0',sizeof (buffer));
        ssize_t s=read(fd,buffer,sizeof(buffer)-1);
        if(s>0)
        {
            cout<<"["<<getpid()<<"] "<<"client say> "<<buffer<<endl;
        }
        else if(s==0)
        {
            //end of file
            cerr<<"["<<getpid()<<"] "<<"read end of file, client quit, server quit too!"<<endl;
            break;
        }
        else
        {
            //read error
            perror("read");
            break;
        }
    }
}

int main()
{
    //1.创建管道文件
    if(mkfifo(ipcPath.c_str(),MODE)<0)
    {
        perror("mkfifo");
        exit(1);
    }

    Log("创建管道文件成功",Debug)<<" step 1 "<<endl;

    //2.打开文件
    int fd=open(ipcPath.c_str(),O_RDONLY);
    if(fd<0)
    {
        perror("open");
        exit(2);
    }

    Log("打开管道文件成功",Debug)<<" step 2 "<<endl;

    int nums=3;
    for(int i=0;i<nums;i++)
    {
        pid_t id=fork();

        if(id==0)
        {
            //3.编写正常的通信代码
            getMessage(fd);
            exit(1);
        }
    }

    for(int i=0;i<nums;i++)
    {
        waitpid(-1,nullptr,0);
    }
   
    //4.关闭文件
    close(fd);

    Log("关闭管道文件成功",Debug)<<" step 3 "<<endl;

    unlink(ipcPath.c_str());//通信完毕

    Log("删除管道文件成功",Debug)<<" step 4 "<<endl;

    return 0;
}

client.cc

#include "comm.hpp"

int main()
{
    //1.获取管道
    int fd=open(ipcPath.c_str(),O_WRONLY);
    if(fd<0)
    {
        perror("open");
        exit(1);
    }
    
    //2.ipc过程
    string buffer;
    while(true)
    {
        cout<<"please enter your message line:>";
        getline(cin,buffer);
        write(fd,buffer.c_str(),buffer.size());
    }

    //3.关闭管道文件
    close(fd);
    return 0;
}

comm.hpp

#ifndef _COMM_H_
#define _COMM_H_

#include <iostream>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <cstring>
#include "Log.hpp"

using namespace std;

#define MODE 0666
#define SIZE 128
string ipcPath = "./fifo.ipc";

#endif

Log.hpp

#ifndef _LOG_H
#define _LOG_H

#include <iostream>
#include <ctime>
#include <string>
using namespace std;

#define Debug 0
#define Notice 1
#define Warning 2
#define Error 3

const string msg[]={
    "Debug",
    "Notice",
    "Warning",
    "Error"
};



ostream &Log(string message,int level)
{
    cout<<" | "<<(unsigned)time(NULL)<<" | "<<msg[level]<<" | "<<message;
    return cout;
}

#endif