使用管道编程----利用别人的程序,实现自己的功能

 

 WINDOWS自带的大量控制台工具,是从DOS继承过来的,应用广泛,功能强大.如果写程序能对其加以利用,有时能够收到事倍功半的效果.使用管道机制能很容易的实现这一点.当然也可以自己创建一个DOS工具,再利用管道机制做一个GUI,无疑也是很爽的.

下面以调用netstat程序为例利用管道创建一个应用.

其核心函数是下面这个函数,从一个网友那里学来的:

void CGetProcessDlg::function()
{

//使用下面三个句柄来对第三方程序做输入输出操作


 HANDLE   hSTDINWrite, hSTDINRead;       // 用于重定向子进程输入的句柄  
 HANDLE   hSTDOUTWrite, hSTDOUTRead;     // 用于重定向子进程输出的句柄  
 HANDLE   hSTDERRWrite, hSTDERRRead;     // 用于重定向子进程输出错误的句柄  

 CString tmp;
 
 SECURITY_ATTRIBUTES   sa;  
   
 sa.bInheritHandle = TRUE;  
 sa.lpSecurityDescriptor = NULL;  
 sa.nLength = sizeof(sa);  
 
 // 创建子进程输出匿名管道,下面会通过  hSTDOUTRead来获取控制台输出的内容
 if( !CreatePipe(&hSTDOUTRead, &hSTDOUTWrite, &sa, 0) )  
 {  
  AfxMessageBox("Create   STDOUT   pipe   failed");  
  return;  
 } 
 
 // 创建子进程输入匿名管道

 if( !CreatePipe(&hSTDINRead, &hSTDINWrite, &sa, 0) )  
 {  
  AfxMessageBox("Create   STDIN   pipe   failed");  
  return;  
 }

//下面是创建进程,如果不使用管道,也是可以创建进程的,不过那样我们无法与进程交互
 PROCESS_INFORMATION  pi;
 ZeroMemory(&pi, sizeof(pi));  
 STARTUPINFO  si;
 GetStartupInfo(&si);
 
 si.cb = sizeof(STARTUPINFO);
 si.dwFlags = STARTF_USESTDHANDLES   |   STARTF_USESHOWWINDOW;
 si.wShowWindow = SW_SHOW; //SW_HIDE;     //SW_SHOW;
 si.hStdInput   =   hSTDINRead;      //重定向子进程输入  
 si.hStdOutput   =   hSTDOUTWrite;   //重定向子进程输入   
 si.hStdError = hSTDOUTWrite; //GetStdHandle( STD_ERROR_HANDLE );
 
 char cmd[256] = {0};
 ::strcpy(cmd, "netstat.exe -nao");
 
 if( !::CreateProcess(NULL, cmd, NULL, NULL, TRUE, NORMAL_PRIORITY_CLASS, NULL, NULL, &si, &pi) )  
 {  
  
  int errorNo = ::GetLastError();
  AfxMessageBox("create process failed");  
  return;  
 }
 ::CloseHandle(hSTDOUTWrite);  
 ::CloseHandle(hSTDINRead);
 
 char strData[1024] = {0};
 DWORD dwBytes;
 int i = 0;

//由于是流,不知道其长短,也没有EOF结束符,所以只好分段来获取
 while(::ReadFile(hSTDOUTRead, strData, sizeof(strData), &dwBytes, NULL))
 {    
  tmp.Format(strData);
  m_value += tmp;
 }
 m_value += '/r';
 m_value += '/n';
 DWORD uExitCode;
 ::TerminateProcess(pi.hProcess,uExitCode);
 
 ::WaitForSingleObject(pi.hProcess, INFINITE);  
 ::CloseHandle(hSTDOUTRead);  
 ::CloseHandle(hSTDINWrite);  
 ::CloseHandle(pi.hProcess);  
 ::CloseHandle(pi.hThread); 
}
下面是结合上面函数添加对话框和EDIT控件后的完整程序.本人新学VC,高手就不要看了.

下面主要是创建了一个对话框,对话框的OK键改成执行键,对话框上加了一个EDIT控件,EDIT绑定变量m_value,用来输出获取的数据。EDIT控制属性要加上自动换行。

// getProcessDlg.cpp : implementation file
//

#include "stdafx.h"
#include "getProcess.h"
#include "getProcessDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
 CAboutDlg();

// Dialog Data
 //{{AFX_DATA(CAboutDlg)
 enum { IDD = IDD_ABOUTBOX };
 //}}AFX_DATA

 // ClassWizard generated virtual function overrides
 //{{AFX_VIRTUAL(CAboutDlg)
 protected:
 virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
 //}}AFX_VIRTUAL

// Implementation
protected:
 //{{AFX_MSG(CAboutDlg)
 //}}AFX_MSG
 DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
 //{{AFX_DATA_INIT(CAboutDlg)
 //}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
 CDialog::DoDataExchange(pDX);
 //{{AFX_DATA_MAP(CAboutDlg)
 //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
 //{{AFX_MSG_MAP(CAboutDlg)
  // No message handlers
 //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/
// CGetProcessDlg dialog

CGetProcessDlg::CGetProcessDlg(CWnd* pParent /*=NULL*/)
 : CDialog(CGetProcessDlg::IDD, pParent)
{
 //{{AFX_DATA_INIT(CGetProcessDlg)
 m_value = _T("");
 //}}AFX_DATA_INIT
 // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
 m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CGetProcessDlg::DoDataExchange(CDataExchange* pDX)
{
 CDialog::DoDataExchange(pDX);
 //{{AFX_DATA_MAP(CGetProcessDlg)
 DDX_Text(pDX, IDC_EDIT1, m_value);
 //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CGetProcessDlg, CDialog)
 //{{AFX_MSG_MAP(CGetProcessDlg)
 ON_WM_SYSCOMMAND()
 ON_WM_PAINT()
 ON_WM_QUERYDRAGICON()
 //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/
// CGetProcessDlg message handlers

BOOL CGetProcessDlg::OnInitDialog()
{
 CDialog::OnInitDialog();

 // Add "About..." menu item to system menu.

 // IDM_ABOUTBOX must be in the system command range.
 ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
 ASSERT(IDM_ABOUTBOX < 0xF000);

 CMenu* pSysMenu = GetSystemMenu(FALSE);
 if (pSysMenu != NULL)
 {
  CString strAboutMenu;
  strAboutMenu.LoadString(IDS_ABOUTBOX);
  if (!strAboutMenu.IsEmpty())
  {
   pSysMenu->AppendMenu(MF_SEPARATOR);
   pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  }
 }

 // Set the icon for this dialog.  The framework does this automatically
 //  when the application's main window is not a dialog
 SetIcon(m_hIcon, TRUE);   // Set big icon
 SetIcon(m_hIcon, FALSE);  // Set small icon
 
 // TODO: Add extra initialization here
 
 return TRUE;  // return TRUE  unless you set the focus to a control
}

void CGetProcessDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
 if ((nID & 0xFFF0) == IDM_ABOUTBOX)
 {
  CAboutDlg dlgAbout;
  dlgAbout.DoModal();
 }
 else
 {
  CDialog::OnSysCommand(nID, lParam);
 }
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CGetProcessDlg::OnPaint()
{
 if (IsIconic())
 {
  CPaintDC dc(this); // device context for painting

  SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

  // Center icon in client rectangle
  int cxIcon = GetSystemMetrics(SM_CXICON);
  int cyIcon = GetSystemMetrics(SM_CYICON);
  CRect rect;
  GetClientRect(&rect);
  int x = (rect.Width() - cxIcon + 1) / 2;
  int y = (rect.Height() - cyIcon + 1) / 2;

  // Draw the icon
  dc.DrawIcon(x, y, m_hIcon);
 }
 else
 {
  CDialog::OnPaint();
 }
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CGetProcessDlg::OnQueryDragIcon()
{
 return (HCURSOR) m_hIcon;
}

//这里是添加的一个方法,创建进程
void CGetProcessDlg::function()
{
 HANDLE   hSTDINWrite, hSTDINRead;       // 用于重定向子进程输入的句柄  
 HANDLE   hSTDOUTWrite, hSTDOUTRead;     // 用于重定向子进程输出的句柄  
 HANDLE   hSTDERRWrite, hSTDERRRead;     // 用于重定向子进程输出的句柄  

 CString tmp;
 
 SECURITY_ATTRIBUTES   sa;  
   
 sa.bInheritHandle = TRUE;  
 sa.lpSecurityDescriptor = NULL;  
 sa.nLength = sizeof(sa);  
 
 // 创建子进程输出匿名管道  
 if( !CreatePipe(&hSTDOUTRead, &hSTDOUTWrite, &sa, 0) )  
 {  
  AfxMessageBox("Create   STDOUT   pipe   failed");  
  return;  
 } 
 
 // 创建子进程输入匿名管道  
 if( !CreatePipe(&hSTDINRead, &hSTDINWrite, &sa, 0) )  
 {  
  AfxMessageBox("Create   STDIN   pipe   failed");  
  return;  
 }
 PROCESS_INFORMATION  pi;
 ZeroMemory(&pi, sizeof(pi));  
 STARTUPINFO  si;
 GetStartupInfo(&si);
 
 si.cb = sizeof(STARTUPINFO);
 si.dwFlags = STARTF_USESTDHANDLES   |   STARTF_USESHOWWINDOW;
 si.wShowWindow = SW_SHOW; //SW_HIDE;     //SW_SHOW;
 si.hStdInput   =   hSTDINRead;      //重定向子进程输入  
 si.hStdOutput   =   hSTDOUTWrite;   //重定向子进程输入   
 si.hStdError = hSTDOUTWrite; //GetStdHandle( STD_ERROR_HANDLE );
 
 char cmd[256] = {0};
 ::strcpy(cmd, "netstat.exe -nao");
 
 if( !::CreateProcess(NULL, cmd, NULL, NULL, TRUE, NORMAL_PRIORITY_CLASS, NULL, NULL, &si, &pi) )  
 {  
  
  int errorNo = ::GetLastError();
  AfxMessageBox("create process failed");  
  return;  
 }
 ::CloseHandle(hSTDOUTWrite);  
 ::CloseHandle(hSTDINRead);
 
 char strData[1024] = {0};
 DWORD dwBytes;
 int i = 0;
 while(::ReadFile(hSTDOUTRead, strData, sizeof(strData), &dwBytes, NULL))
 {    
  tmp.Format(strData);
  m_value += tmp;
 }
 m_value += '/r';
 m_value += '/n';
 DWORD uExitCode;
 ::TerminateProcess(pi.hProcess,uExitCode);
 
 ::WaitForSingleObject(pi.hProcess, INFINITE);  
 ::CloseHandle(hSTDOUTRead);  
 ::CloseHandle(hSTDINWrite);  
 ::CloseHandle(pi.hProcess);  
 ::CloseHandle(pi.hThread); 
}

void CGetProcessDlg::OnOK()
{
 // TODO: Add extra validation here
    UpdateData(TRUE);
 function();
 UpdateData(FALSE);
 //CDialog::OnOK();
}

void CGetProcessDlg::OnCancel()
{
 // TODO: Add extra cleanup here
 
 CDialog::OnCancel();
}

可以使用管道将 `ps -ea` 命令的输出作为 `grep httpd` 命令的输入,实现类似于 `ps -ea | grep httpd` 的功能。 具体实现方法如下: ```c #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <fcntl.h> int main() { int fd[2]; pid_t pid; if (pipe(fd) == -1) { perror("pipe"); exit(1); } pid = fork(); if (pid == -1) { perror("fork"); exit(1); } else if (pid == 0) { // 子进程 close(fd[1]); // 关闭写端 dup2(fd[0], STDIN_FILENO); // 将管道的读端复制到标准输入 execlp("grep", "grep", "httpd", NULL); // 执行 grep 命令 perror("execlp"); exit(1); } else { // 父进程 close(fd[0]); // 关闭读端 dup2(fd[1], STDOUT_FILENO); // 将标准输出复制到管道的写端 execlp("ps", "ps", "-ea", NULL); // 执行 ps 命令 perror("execlp"); exit(1); } return 0; } ``` 首先,使用 `pipe()` 函数创建管道,将管道的读端保存到 `fd[0]` 中,将管道的写端保存到 `fd[1]` 中。如果 `pipe()` 函数返回值为 `-1`,则说明创建管道失败,此时使用 `perror()` 函数输出错误信息,并退出程序。 接着,使用 `fork()` 函数创建子进程。如果 `fork()` 函数返回值为 `-1`,则说明创建子进程失败,此时使用 `perror()` 函数输出错误信息,并退出程序。如果 `fork()` 函数返回值为 `0`,则说明当前进程是子进程,此时关闭管道的写端,将管道的读端复制到标准输入,然后执行 `grep httpd` 命令。如果 `execlp()` 函数返回值为 `-1`,则说明执行命令失败,此时使用 `perror()` 函数输出错误信息,并退出子进程。如果 `fork()` 函数返回值大于 `0`,则说明当前进程是父进程,此时关闭管道的读端,将标准输出复制到管道的写端,然后执行 `ps -ea` 命令。如果 `execlp()` 函数返回值为 `-1`,则说明执行命令失败,此时使用 `perror()` 函数输出错误信息,并退出父进程。
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值