从C中的文件读取字符

Question

这是我想要做的。 我编写了一个简短的C代码，它将打开一个文件，读取前2个字节（2个字符），并将其与2个字符串进行比较。 这有助于识别文件类型（让我们将前2个字节称为文件签名）。

现在，一旦我从文件中读取了2个字节，我想将其与预定义的签名进行比较，并基于该打印文件类型。

码：

#include <stdio.h>
#include <stdlib.h>

int main(int argc, char *argv[])
{
FILE *fp;
char signature[2];

if(argc!=2)
{
printf("usage: fileio.c <filename>\n");
exit(1);
}

if((fp=fopen(argv[1],"r"))!=NULL)
{
fscanf(fp,"%c %c", &signature[0], &signature[1]);
printf("%x %x\n",signature[0],signature[1]);
}

}

如果我在Windows平台上为可执行文件运行此命令，它将显示为：4a 5d，因为它是MZ签名。

现在，我想做这样的事情：

比较签名数组的2个字节与0x4d5a，如果相等，则打印它是可执行文件。

我认为的另一种方法是，将其与字符串“ MZ”进行比较。 但是然后我需要使用fscanf从文件中读取前2个字节，并将它们存储在字符串中。 然后将其与“ MZ”签名进行比较。

如果我可以使用十六进制字节来做，那就太好了，因为我以后需要对十六进制字节执行一些操作。

谢谢。

Answer 1

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

int main(int argc, char *argv[]){
  FILE *fp;
  const char mz_sig[] =  {0x4d, 0x5a};
  char signature[2];

  fp=fopen(argv[1],"r");
  fscanf(fp,"%c %c", &signature[0], &signature[1]);
  printf("%x %x\n", signature[0], signature[1]);

  if (*(uint16_t*)signature == *(uint16_t*)mz_sig) {
    printf("match\n");
  }

  return 0;
}

Answer 2

首先，您可能应该以二进制模式（ "rb" ）打开文件。

至于读取，您可以使用fread作为单个uint16_t读取前两个字节，然后进行比较。