目录
一、前言
上篇笔记(【学习笔记2】管道通信:输入输出重定向)记录了输入输出重定向的实现方法,本文总结一下匿名管道的内容,希望对自己与各位有所帮助。
二、管道的概念
简单理解,计算机中的管道(pipe)类似现实世界中的水管,从一端放入水流,另一端就会流出来。回到计算机中,这个水流就是数据。
管道(pipe)又分为 匿名管道 和 命名管道。本文主要介绍如何使用匿名管道来实现输入输出重定向功能。
三、匿名管道的创建和使用
3.1 函数原型
3.1.1 CreatePipe
/* 创建管道 */
BOOL WINAPI CreatePipe(
PHANDLE hReadPipe, // 管道输出(读取)端句柄
PHANDLE hWritePipe, // 管道输入(写入)端句柄
LPSECURITY_ATTRIBUTES lpPipeAttributes, // 管道的安全属性
DWORDnSize // 管道缓冲区容量,设置0时使用默认大小
);
管道的安全属性通常设置如下:
SECURITY_ATTRIBUTES sa = {0};
sa.nLength = sizeof(sa);
sa.bInheritHandle = TRUE;
sa.lpSecurityDescriptor = NULL;
注意:不能在管道的输出端写入数据,也不能在管道的输入端读取数据。
3.1.2 ReadFile
/* 从管道中读取数据 */
BOOL ReadFile(
HANDLE hFile, // 管道输出端(读取)句柄
LPVOID lpBuffer, // 数据缓冲区指针,读取的数据将写在该缓冲区中
DWORD nNumberOfBytesToRead, // 指定读取的字节数
LPDWORD lpNumberOfBytesRead, // 返回实际读取到的字节数
LPOVERLAPPED lpOverlapped // 用于异步操作,一般传入NULL即可
);
3.1.3 WriteFile
/* 向管道写入数据 */
BOOL WriteFile(
HANDLE hFile, // 管道输入端(写入)句柄,也可以是CreateFile()接口创建的文件句柄
LPVOID lpBuffer, // 待写入管道的数据缓冲区指针
DWORD nNumberOfBytesToWrite, // 指定写入的字节数
LPDWORD lpNumberOfBytesWritten, // 返回实际写入管道的字节数
LPOVERLAPPED lpOverlapped // 用于异步操作,一般传入NULL即可
);
3.1.4 CloseHandle
/* 关闭管道端口句柄 */
BOOL CloseHandle(
HANDLE hObject // 想要关闭的句柄
);
注意:当管道输入输出端都被关闭后,系统会自动关闭该管道,并回收相关资源。
3.2 管道使用技巧
从以上管道相关的函数原型来看,管道的读写其实和文件的读写非常类似,实际上管道也是一种特殊的文件——内存映射文件。
使用管道时需要注意:读取和写入数据时,一定要注意顺序,如果管道中没有数据,调用ReadFile()会造成阻塞,直到有其它线程将数据写入管道。同样,当有线程正在管道中读取数据时,其它试图将数据写入管道的的线程也会被阻塞,更多内容请查阅MSDN。
3.3 示例代码
此处以【学习笔记2】管道通信:输入输出重定向中的示例程序作为子进程,在父进程中创建两个管道(输入/输出 管道),分别用于存放输入输出数据,将这两个管道重定向到输入输出文件(stdin.txt 和 stdout.txt)。
此时,在父进程中启动子进程,将子进程的标准输入输出改成从上述创建的 输入/输出 管道句柄,即实现了子进程的输入输出重定向,示例代码如下:
#include <windows.h>
#include <stdio.h>
int main()
{
char sz[3][50] = { "示例程序.exe", "infile.txt", "outfile.txt" };
HANDLE hPipeInputRead = NULL;
HANDLE hPipeInputWrite = NULL;
HANDLE hPipeOutputRead = NULL;
HANDLE hPipeOutputWrite = NULL;
SECURITY_ATTRIBUTES sa;
sa.nLength = sizeof(sa);
sa.bInheritHandle = TRUE;
sa.lpSecurityDescriptor = NULL;
CreatePipe(&hPipeInputRead, &hPipeInputWrite, &sa, 0); // 创建输入管道
CreatePipe(&hPipeOutputRead, &hPipeOutputWrite, &sa, 0); // 创建输出管道
HANDLE hInputFile = CreateFile(L"stdin.txt", GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
HANDLE hOutputFile = CreateFile(L"stdout.txt", GENERIC_READ | GENERIC_WRITE, FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
CHAR Buffer[1024 * 4] = { 0 };
DWORD dwNumberOfBytesRead = 0;
DWORD dwNumberOfBytesWritten = 0;
BOOL bRet = FALSE;
while (TRUE) { // 从文件stdin.txt中读取数据,写到输入管道中
bRet = ReadFile(hInputFile, Buffer, sizeof(Buffer), &dwNumberOfBytesRead, NULL);
if (!bRet || dwNumberOfBytesRead == 0)
break;
bRet = WriteFile(hPipeInputWrite, Buffer, dwNumberOfBytesRead, &dwNumberOfBytesWritten, NULL);
if (!bRet)
break;
}
// 关闭输入管道的写句柄
CloseHandle(hInputFile);
CloseHandle(hPipeInputWrite);
// 启动demo.exe作为子进程
STARTUPINFO si;
si.cb = sizeof(si);
GetStartupInfo(&si);
si.hStdInput = hPipeInputRead; // 输入由标准输入 -> 从管道中读取
si.hStdOutput = hPipeOutputWrite; // 输出由标准输出 -> 输出到管道
si.wShowWindow = SW_HIDE;
si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;
PROCESS_INFORMATION pi;
CreateProcess(L"demo.exe", NULL, NULL, NULL, TRUE, 0, NULL, NULL, &si, &pi);
WaitForSingleObject(pi.hProcess, INFINITE);
// 关闭进程相关句柄
CloseHandle(pi.hThread);
CloseHandle(pi.hProcess);
CloseHandle(hPipeInputRead);
CloseHandle(hPipeOutputWrite);
while (TRUE) { // 从输出管道中读取数据,写到文件stdout.txt中
bRet = ReadFile(hPipeOutputRead, Buffer, sizeof(Buffer), &dwNumberOfBytesRead, NULL);
if (!bRet || dwNumberOfBytesRead == 0)
break;
bRet = WriteFile(hOutputFile, Buffer, dwNumberOfBytesRead, &dwNumberOfBytesWritten, NULL);
if (!bRet)
break;
}
//关闭输出管道的读句柄
CloseHandle(hOutputFile);
CloseHandle(hPipeOutputRead);
return 0;
}
四、总结
从上述示例来看,匿名管道可以在不修改程序源码的情况下实现输入输出重定向,也就是说只要有exe文件,即可实现输入输出重定向,并且上述示例中,父进程也可以使用 输入/输出 管道写入或读取数据,即实现了父子进程间的通信。