-2

我试图用来fread()确定文件是否是 jpeg，我的想法是使用fread()读取文件的标题以确保它是 jpeg，然后，如果是 - 复制文件。我知道标准的 jpeg 标头是0xff 0xd8 0xff 0xe(0 to f)，所以我编写了一个程序来读取文件的前四个字节。

#include <stdio.h>
#include <stdint.h>

typedef uint8_t BYTE;

int main (int argc, char *argv[])
{
    FILE *in = fopen(argv[1], "rb");
    
    FILE *out = fopen("output.jpg", "wb");
    
    BYTE buffer [4096];
    
    while (fread(buffer, 1, sizeof(buffer), in))
    {
        if ((buffer[0] == 0xff) & (buffer[1] == 0xd8) & (buffer[2] == 0xff) & ((buffer[3] & 0xf0) == 0xe0))
        {
            fwrite(buffer, 1, sizeof(buffer), out);
        }
    }
    
    fclose(in);
    fclose(out);
}

但是，运行程序后，当我尝试打开时output.jpg，我收到消息：Invalid or Unsupported Image Format.

我决定fwrite()通过将段落更改为：

while (fread(buffer, 1, sizeof(buffer), in))
        {
            fwrite(buffer, 1, sizeof(buffer), out);
        }

在这种情况下，output.jpg 确实打开正确，这让我认为这是我试图用来fread()检查导致问题的文件类型。我应该使用不同的功能，还是只是格式不正确？

4

1 回答 1

5

这里有两个问题。

您的程序在循环中一次从文件中读取最多 4KB，并且您正在检查每个4KB 块的前 4 个字节，以查看它是否是 jpeg 标头，然后再写入文件。您只想在初次读取文件时执行此操作一次。如果它通过检查，则继续循环读取和写入，否则会中断。

另一个问题是您总是将sizeof(buffer)字节写入文件。该fread函数可能读取的字节数少于此，如果是这种情况，您将写入不应该写入的额外数据。您需要捕获 的返回值fread以了解读取了多少，然后将其用作写入的长度fwrite。

于 2021-02-14T04:27:05.893 回答