C++ 命名管道示例

想做一个 Hook CreateFile 重定向到内存的功能，貌似可以假借命名管道实现这个功能。不熟悉命名管道，做了几个demo，如下：

Server：

// NamedPipeServer.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <iostream>
#include <windows.h>
#include <ctime>
using namespace std;

int _tmain(int argc, _TCHAR* argv[])
{
    HANDLE hPipe = CreateNamedPipe(L"\\\\.\\Pipe\\mypipe",PIPE_ACCESS_DUPLEX,PIPE_TYPE_MESSAGE|PIPE_READMODE_MESSAGE|PIPE_WAIT
        ,PIPE_UNLIMITED_INSTANCES,,,NMPWAIT_WAIT_FOREVER,);

    //waiting to be connected
    if(ConnectNamedPipe(hPipe, NULL) == NULL)
        return ;

    DWORD    dwWrite;
    const char *pStr = "data from server";
    if( !WriteFile(hPipe, pStr, strlen(pStr), &dwWrite, NULL) )
    {
        cout << "write failed..." << endl<< endl;
        return ;
    }
    cout << "sent data: " << endl << pStr<< endl<< endl;

    system("pause");
    return ;
}

Client:

// NamedPipeClient.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <iostream>
#include <windows.h>
#include <ctime>
using namespace std;
#define BUFSIZE 5

int _tmain(int argc, _TCHAR* argv[])
{
    if (WaitNamedPipe(L"\\\\.\\Pipe\\mypipe", NMPWAIT_WAIT_FOREVER) == FALSE)
        return ;

    HANDLE hPipe=CreateFile(L"\\\\.\\Pipe\\mypipe", GENERIC_READ | GENERIC_WRITE, ,
        NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);

    if((long)hPipe == -)
        return ;

    BOOL fSuccess = false;
    DWORD len = ;
    char buffer[BUFSIZE];
    string recvData = "";
    do
    {
        fSuccess = ReadFile(hPipe ,buffer ,BUFSIZE*sizeof(char) ,&len ,NULL);
        char buffer2[BUFSIZE+] = {};
        memcpy(buffer2,buffer,len);
        recvData.append(buffer2);
        if(!fSuccess || len < BUFSIZE)
            break;
    }while(true);

    cout<<"recv data:"<<endl<<recvData.c_str()<<endl<<endl;

    FlushFileBuffers(hPipe);
    DisconnectNamedPipe(hPipe);
    CloseHandle(hPipe);

    system("pause");
    return ;
}

Server & Client:

// MultiThreadDemo.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <iostream>
#include <windows.h>
#include <ctime>
using namespace std;
#define BUFSIZE 5

DWORD WINAPI ThreadFunction(LPVOID lpParam)
{
    wstring* pipeName = static_cast<wstring*>(lpParam);

    HANDLE hPipe = CreateNamedPipe((*pipeName).c_str(),PIPE_ACCESS_DUPLEX,PIPE_TYPE_MESSAGE|PIPE_READMODE_MESSAGE|PIPE_WAIT
        ,PIPE_UNLIMITED_INSTANCES,,,NMPWAIT_WAIT_FOREVER,);

    //waiting to be connected
    if(ConnectNamedPipe(hPipe, NULL) == NULL)
        return ;

    DWORD    dwWrite;
    const char *pStr = "data from server";
    if( !WriteFile(hPipe, pStr, strlen(pStr), &dwWrite, NULL) )
    {
        cout << "write failed..." << endl<< endl;
        return ;
    }
    cout << "sent data: " << endl << pStr<< endl<< endl;
}

int _tmain(int argc, _TCHAR* argv[])
{
    wstring pipeName = L"\\\\.\\Pipe\\mypipe";

    DWORD dwThreadID = ;
    HANDLE hThread = CreateThread(NULL, , ThreadFunction, &pipeName , , &dwThreadID);

    Sleep();

    if (WaitNamedPipe(pipeName.c_str(), NMPWAIT_WAIT_FOREVER) == FALSE)
        return ;

    HANDLE hPipe=CreateFile(pipeName.c_str(), GENERIC_READ | GENERIC_WRITE, ,
        NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);

    if((long)hPipe == -)
        return ;

    BOOL fSuccess = false;
    DWORD len = ;
    char buffer[BUFSIZE];
    string recvData = "";
    do
    {
        fSuccess = ReadFile(hPipe ,buffer ,BUFSIZE*sizeof(char) ,&len ,NULL);
        char buffer2[BUFSIZE+] = {};
        memcpy(buffer2,buffer,len);
        recvData.append(buffer2);
        if(!fSuccess || len < BUFSIZE)
            break;
    }while(true);

    cout<<"recv data:"<<endl<<recvData.c_str()<<endl<<endl;

    FlushFileBuffers(hPipe);
    DisconnectNamedPipe(hPipe);
    CloseHandle(hPipe);

    system("pause");
    return ;
}