重定向的标准输入 Lua 对于 Windows 过早结束

Question

我在 Windows 下的 cmd window 中使用 Lua。我使用“cat”（来自 UnxUtils）将文件提供给 Lua 脚本。 该脚本使用“io.read(1)”一次读取一个字节。

local b, n ;
n = -1 ;
b = true ;
while b do
  n = n + 1 ;
  b = io.read(1) ;
end ;
print( n, "bytes read" ) ;

当我向脚本提供一个 333K.EXE 文件时，它声称“读取了 24025 个字节”。 将 same.EXE 提供给“wc”（另一个 UnxUtils），wc 正确显示 333008。

> cat "Firefox Installer.exe" | lua count.lua
24025   bytes read
cat: write error: Invalid argument
> cat "Firefox Installer.exe" | wc
   1408    8674  333008

因为我在“cat | wc”时得到了预期的答案，所以我认为“cat”程序或 Windows 的重定向实现没有任何问题。

我不是在寻找有关如何提高 Lua 脚本效率的建议。 我不需要关于如何使脚本直接从文件中读取（按预期工作）的建议。 我正在寻找关于在哪里寻找我不能使用 Lua 编写过滤器（并且能够信任结果）的原因的线索。

我查看了输入文件，看看 Ctrl-Z 或 Ctrl-D 是否是提前关闭的原因——它们出现在文件的早期。

我尝试在“io.read()”返回“false”后读取：脚本承认看到了更多字节，但仍然不超过 333K 输入文件的 45K。

Answer 1

从我的评论中复制：

可能是 Windows 问题（参见例如这个答案）。 Windows 以不同方式对待二进制和文本“流”/文件。 我假设你的程序的标准输入默认是文本 stream； 稍后使用普通 Lua 无法将 stdin 的模式更改为二进制，您需要一个库。 像lfs = require("lfs"); lfs.setmode(io.stdin, "binary") lfs = require("lfs"); lfs.setmode(io.stdin, "binary")可能有效（使用LuaFileSystem库）。

您还可以尝试修复脚本调用以使用脚本设置正确的模式，该脚本在调用 Lua 脚本之前将stdin更改为二进制模式：

./stdbin.c ：

#include <stdio.h>
#include <unistd.h>
#include <assert.h>

int main(int argc, char** argv) {
    if (argc < 1) {
        printf("Arguments: <program> {args}\n");
        return 1;
    }

    // See  https://learn.microsoft.com/en-us/cpp/c-runtime-library/reference/setmode?redirectedfrom=MSDN&view=msvc-170
    if (_setmode(_fileno(stdin), _O_BINARY) == -1)
        perror("_setmode failed");

    execvp("lua", ++argv);
    // execvp only returns if there is an error
    perror("execvp failed");
    return 1;
}

注意：未经测试。 用法： ./stdbin lua count.lua 。

Answer 2

（这是对 LMD 答案的补充）
在 LuaJIT 中不需要外部库和可执行文件：

local ffi = require"ffi"
ffi.cdef"int _setmode(int,int)"
ffi.C._setmode(0, 0x8000)
-- Now io.read() will return binary data