c - 使用管道连接到控制台程序时如何处理缓冲

Question

我有一个 win32 程序(是的，我一直在使用 win32)启动一个子进程并通过匿名管道连接到它的标准输入/标准输出。大多数时候我不拥有子进程(因此，我无权访问代码)。我发现 child 的标准输出缓冲妨碍了。在我终止进程之前，我看不到 child 的任何输出；但是，我知道输入在收到时会被处理。

所以，我读过 http://support.microsoft.com/kb/190351/en-us ，它警告子进程使用 printf(等)关于缓冲的问题，微软的 brilliant 修复是用 fflush() 跟踪子进程中的所有 printfs(更简单的方法是只需使用 setbuf(stdout,0); 但是，这假定您可以访问子源代码。

显然，子进程在连接到父管道时不会刷新标准输出。有没有办法解决这个问题而不需要修改 child ？

我已经用一个简单的 child 验证了这一点:

    #include <stdio.h>
    #include <fcntl.h>

    char line[256];

    int
    main(int argc, char *argv[])
    {
            char *lp;

    //      setbuf(stdout,0);
            while(1) {
                    printf("Prompt:");
                    lp = gets(line);
                    printf("Got: <%s>\n",lp);
                    if (strcmp(lp,"yada") == 0) {
                            int fd = open("C:/tmp/yada123",O_BINARY|O_RDWR|O_CREAT);
                            if (fd > 0) {
                                    write (fd,"hi!\n",4);
                                    close(fd);
                            }
                    }
                    if (strcmp(lp,"exit") == 0) {
                            printf("bye!\n");
                            break;
                    }
            }
}

如果我取消注释顶部的 setbuf() 调用，那么交互是干净的；但是请注意，我通常没有修改子代码的选项。另外，我验证了另一个方向没有被缓冲(即使我没有看到输出)；因为文件是在收到字符串“yada”时写入的。

Answer 1

前言:在问题的评论中概述了代码注入(inject)可能会解决问题。这篇文章展示了 dll 注入(inject)的一种可能实现，因此补充了评论。我将代码作为普通帖子发布，以便更好地展示多行代码。

以下代码可能是 dll 注入(inject)的良好起点。它使用一个全局钩子(Hook)。这是一个非常简单的方法，但它会将您的 dll 注入(inject)到所有正在运行的进程中(仅注入(inject)您有权访问的进程)。因此，您必须询问当前程序名称并调用 setbuf(stdout,0); 只有在您处于所需应用程序中的情况下:-) 也许你需要添加一些头文件(我只是把一些代码片段放在一起，没有测试它)。

void InstallHook(void) 和 void RemoveHook(void) 函数必须声明为 dll 导出函数，因此您要么使用 *.def 文件，要么声明它们作为 __declspec(dllexport)。这可能取决于您的编译器和设置。

#include <windows.h>
#include <cstdlib>
#include <cstdio>

BOOL WINAPI DllMain( HINSTANCE hModule, DWORD  ul_reason_for_call, LPVOID lpReserved)
{
  UNREFERENCED_PARAMETER(lpReserved);

  switch (ul_reason_for_call)
  {
  case DLL_PROCESS_ATTACH:
  {
    wchar_t buffer [1024];
    // get the path and executable name of the current process
    GetModuleFileNameW( GetModuleHandle( NULL ), buffer, 1024 );
    //test if we are in the target app
    if ( _wcsicmp( buffer, "C:/some_dir/some_app.exe" ) == 0 )
    {
      setbuf(stdout,0);
    }
    break;
  }
  case DLL_PROCESS_DETACH:
    break;
  default:
    break;
  }

  return (true);
}

HHOOK gl_hHook = NULL;
LRESULT CALLBACK HookProc(int nCode, WPARAM wParam, LPARAM lParam)
{
  return (CallNextHookEx( gl_hHook, nCode, wParam, lParam));
}

// call this function before or after you have started the child process
// the hook gets installed and your dll is loaded into each running process
// and the DllMain functions gets called by each process
void InstallHook(void)
{
  gl_hHook = SetWindowsHookEx( WH_CBT, HookProc, gl_hThisInstance, 0 );
}

void RemoveHook(void)
{
  UnhookWindowsHookEx( gl_hHook );
}