2008年(884)
分类: C/C++
2008-08-06 09:57:03
这显然是一种具备潜在价值的功能。利用这一技术,我们至少可以
1. 编写出自己的IDE,如果我们有足够的耐心的话;
2. 在我们自己的应用程序里嵌入全文检索功能(调用Borland C 里的grep.exe工具),或者压缩和解压缩功能(调用控制台方式的压缩解压程序,比如arj.exe、pkzip.exe等);
3. 连接其他人编写的,或者我们自己很久以前编写的控制台程序——我经常因为难以调用一个功能强大但又没有源码的控制台程序而苦恼万分。
这样好的功能是如何实现的呢?
首先,如果我们想做的是用一个控制台程序调用另一个控制台程序,那就再简单不过了。我们只消把父进程的stdout重定向到某个匿名管道的WRITE端,然后启动子进程,这时,子进程的stdout因为继承的关系也连在了管道的WRITE端,子进程的所有标准输出都写入了管道,父进程则在管道的另一端随时“侦听”——这一技术叫做输入输出的重定向。
可现在的问题是,GUI方式的Windows程序根本没有控制台,没有stdin、stdout之类的东西,子进程又是别人写好的东西无法更改,这重定向该从何谈起呢?
还有另外一招:我们可以直接在调用子进程时用命令行中的管道指令“>”将子进程的标准输出重定向到一个文件,子进程运行完毕后再去读取文件内容。这种方法当然可行,但它的问题是,我们很难实时监控子进程的输出,如果子进程不是随时刷新stdout的话,那我们只能等一整块数据实际写入文件之后才能看到运行结果;况且,访问磁盘文件的开销也远比内存中的管道操作来得大。
我这里给出的方案其实很简单:既然控制台程序可以调用另一个控制台程序并完成输入输出的重定向,那我们完全可以编写一个中介程序,这个中介程序调用我们需要调用的工具程序并随时获取该程序的输出信息,然后直接将信息用约定的进程间通讯方式(比如匿名管道)传回GUI程序,就象下图中这样:
图中,工具程序和中介程序都是以隐藏的方式运行的。工具程序原本输出到stdout的信息被重定向到中介程序开辟的管道中,中介程序再利用GUI程序创建的管道将信息即时传递到GUI程序的一个后台线程里,后台线程负责刷新GUI程序的用户界面(使用后台线程的原因是,只有这样才可以保证信息在GUI界面中随时输出时不影响用户正在进行的其他操作,就象我们在Visual
Studio中执行耗时较长的编译功能那样)。
我写的中介程序名字叫wSpawn,这个名字来自Visual Studio里完成类似功能的中介程序VcSpawn(你可以在Visual Studio的安装目录中找到它)。我的wSpawn非常简单,它利用系统调用_popen()同时完成创建子进程和输入输出重定向两件工作。GUI程序则使用一种特殊的命令行方式调用wSpawn:
wspawn –h其中,-h后跟的是GUI程序提供的管道句柄,由GUI程序自动将其转换为十进制数字,wSpawn运行时将信息写入该句柄中,随后的内容是GUI程序真正要执行的命令行,例如调用C 编译器cl.exe的方式大致如下:[arg1] [arg2] ...
wspawn –h 1903 cl /Id:\myInclude Test.cppwspawn.cpp的程序清单如下:
#include下面,我们就利用wSpawn程序,写一个简单的“IDE”工具。我们选择Visual Studio 6.0作为开发环境(本文给出的代码也在Visual Studio.NET 7.0中做过测试)。首先,创建Visual C 工程myIDE,工程类型为MFC AppWizard(EXE)中的Dialog based类型,即创建了一个主窗口为对话框的GUI程序。工程myIDE的主对话框类是CMyIDEDlg。现在我们要在资源编辑器中为主对话框添加一个足够大的多行编辑框(Edit Box),它的控制ID是IDC_EDIT1,必须为IDC_EDIT1设置以下属性:#include #include #include #include using namespace std; void exit_friendly(void) { puts("请不要单独运行wSpawn."); exit(0); } int main( int argc, char *argv[] ) { HANDLE hWrite = NULL; DWORD dwWrited; int i = 0, ret = 0, len = 0; char psBuffer[256]; FILE* child_output; string command_line = ""; // 检查命令行,如存在管道句柄,则将其转换为HANDLE类型 if (argc < 2) exit_friendly(); if (!stricmp(argv[1], "-h")) { if (argc < 4) exit_friendly(); hWrite = (HANDLE)atoi(argv[2]); i = 3; } else i = 1; // 提取要执行的命令 for (; i < argc; i ) { command_line = argv[i]; command_line = " "; } // 使用_popen创建子进程并重定向其标准输出到文件指针中 if( (child_output = _popen( command_line.c_str(), "rt" )) == NULL ) exit( 1 ); while( !feof( child_output ) ) { if( fgets( psBuffer, 255, child_output ) != NULL ) { if (hWrite) { // 将子进程的标准输出写入管道,提供给自己的父进程 // 格式是先写数据块长度(0表示结束),再写数据块内容 len = strlen(psBuffer); WriteFile(hWrite, &len, sizeof(int), &dwWrited, NULL); WriteFile(hWrite, psBuffer, len, &dwWrited, NULL); } else // 如命令行未提供管道句柄,则直接打印输出 printf(psBuffer); } } // 写“0”表示所有数据都已写完 len = 0; if (hWrite) WriteFile(hWrite, &len, sizeof(int), &dwWrited, NULL); return _pclose( child_output ); }
CEdit m_edit1;使用ClassWizard为“确定”按钮添加消息响应方法OnOK(),编辑该方法:
void CMyIDEDlg::OnOK() { AfxBeginThread(myThread, this); InvalidateRect(NULL); UpdateWindow(); }也就是说,我们在“确定”按钮按下时,启动了后台线程myThread(),那么,myThread()到底做了些什么呢?我们先在CMyIDEDlg类的头文件myIDEDlg.h中加上一个成员函数声明:
protected: static UINT myThread(LPVOID pParam);然后,在CMyIDEDlg类的实现文件myIDEDlg.cpp里添加myThread()的实现代码:
UINT CMyIDEDlg::myThread(LPVOID pParam) { PROCESS_INFORMATION pi; STARTUPINFO siStartInfo; SECURITY_ATTRIBUTES saAttr; CString Output, tmp; char command_line[200]; DWORD dwRead; char* buf; int len; HANDLE hRead, hWrite; CMyIDEDlg* pDlg = (CMyIDEDlg*)pParam; // 创建与wSpawn.exe通讯的可继承的匿名管道 saAttr.nLength = sizeof(SECURITY_ATTRIBUTES); saAttr.bInheritHandle = TRUE; saAttr.lpSecurityDescriptor = NULL; if (!CreatePipe(&hRead, &hWrite, &saAttr, 0)) { AfxMessageBox("创建管道失败"); return 0; } // 准备wSpawn的命令行,在命令行给出写管道句柄和要wSpawn执行的命令 memset(&pi, 0, sizeof(pi)); sprintf(command_line, "wspawn -h %d cl /?", (unsigned int)hWrite); // 子进程以隐藏方式运行 ZeroMemory( &siStartInfo, sizeof(STARTUPINFO) ); siStartInfo.cb = sizeof(STARTUPINFO); siStartInfo.wShowWindow = SW_HIDE; siStartInfo.dwFlags = STARTF_USESHOWWINDOW; // 创建wSpawn子进程 if (!CreateProcess( NULL, command_line, NULL, NULL, TRUE, 0, NULL, NULL, &siStartInfo, &pi)) { AfxMessageBox("调用wSpawn时失败"); return 0; } // 读管道,并显示wSpawn从管道中返回的输出信息 if(!ReadFile( hRead, &len, sizeof(int), &dwRead, NULL) || dwRead == 0) return 0; while(len) { buf = new char[len 1]; memset(buf, 0, len 1); if(!ReadFile( hRead, buf, len, &dwRead, NULL) || dwRead == 0) return 0; // 将返回信息中的"\n"替换为Edit Box可识别的"\r\n" tmp = buf; tmp.Replace("\n", "\r\n"); Output = tmp; // 将结果显示在Edit Box中,并刷新对话框 pDlg->m_edit1.SetWindowText(Output); pDlg->InvalidateRect(NULL); pDlg->UpdateWindow(); delete[] buf; if(!ReadFile( hRead, &len, sizeof(int), &dwRead, NULL) || dwRead == 0) return 0; } // 等待wSpawn结束 WaitForSingleObject(pi.hProcess, 30000); // 关闭管道句柄 CloseHandle(hRead); CloseHandle(hWrite); return 0; }