检查 Windows 文件是否被重定向到自身

Question

我试图弄清楚如何测试文件是否被重定向到自身，例如.\command.exe file1 > file1

在 *nix 世界中，我只会使用这样的东西：

// Language agnostic...
if (file_dev == out_dev && file_ino == out_ino) {
    printf("%s\n", "same file!");
}

但在 Windows 中，如果我尝试这样做：

// This (language) is Go...
// create fileStat...

// now stat stdout
outStat, err := os.Stdout.Stat()
// error check

if os.SameFile(fileStat, outStat) {
    fmt.Println("same file!")
}

...我收到 IncorrectFunction 错误。

我阅读了这个 (https://stackoverflow.com/a/21070042/2967113) 问题，据我所知，您无法统计 stdout？

这是一个主要与语言无关的问题——我可以将任何内容翻译成 Go（我正在使用的语言）。我最关心的是如何，使用 Windows 的 ABI（API？），我会找到 stdout 被重定向到的位置。

Answer 1

这个答案是特定于 Windows 的，但是当你标记为“windows”时，我认为没关系。

我对 Go 无能为力，但在 C/C++ 中你可以这样做：

#include <tchar.h>
#include <Windows.h>
#include <iostream>

int _tmain(int argc, _TCHAR* argv[])
{
    wchar_t chPath[MAX_PATH];
    if (GetFinalPathNameByHandle(GetStdHandle(STD_OUTPUT_HANDLE), chPath, MAX_PATH, 0))
        std::wcout << L"stdout = " << chPath << std::endl;
    else
        std::cout << "stdout not redirected" << std::endl;

    return 0;
}

如果 stdout 是控制台句柄，GetFinalPathNameByHandle 将失败，但如果它被重定向到文件，它将返回文件路径。

Answer 2

您可以调用HANDLE GetStdHandle( DWORD stdHandle ) with STD_INPUT_HANDLE and STD_OUTPUT_HANDLE 打开文件句柄。

然后拨打DWORD GetFileType( HANDLE hFile ) to check if the returned type is FILE_TYPE_DISK

最后，调用

DWORD WINAPI GetFinalPathNameByHandle( _In_ HANDLE hFile, _Out_ LPTSTR lpszFilePath, _In_ DWORD cchFilePath, _In_ DWORD dwFlags );

获取文件路径名并比较名称是否等效。

Answer 3

嗯，首先，我不认为你在 UNIX 领域的方法实际上会保护你。

当您的代码开始检查设备和 inode 时，shell 已经已经截断了文件。 它负责处理输出重定向，它会在您的程序启动之前执行此操作，以便为您的输出提供一个新文件。

我怀疑你在 Windows 中也会遇到同样的问题，cmd.exe 会在你的脚本开始运行之前截断你的文件。

话虽如此，我相信在某些时候你将不得不相信用户知道他们在做什么:-)

当然，另一种选择是不做输出重定向，而是要求输入和输出文件作为参数：

cmd.exe myscipt myscript

这样，您可以检测用户是否要写入输入文件（使用规范化的文件名或 inode）并阻止它。

虽然仍然不会阻止用户做一些愚蠢的事情：

cmd.exe myscipt >myscript

在您有机会通知他们应该提供两个参数而不是一个参数之前在吹掉你的脚本。

我认为底线是，如果用户进行了输出重定向，那么您的 程序 无法在为时已晚之前捕捉到它。