c语言 - CS50 PSET3 恢复分段错误，无法获取代码以输出 JPEG 的

Question

该程序应该从原始文件中恢复JPEG，我已经检查了类似的问题，但似乎无法找出导致分割错误和无法输出图像的代码中的问题。cs50 中的调试器不起作用。

法典：

#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <stdint.h>
typedef uint8_t  BYTE;
bool checkheader(BYTE buffer[512]);
int main(int argc, char *argv[])
{
// ensure proper usage
if (argc != 2)
{
printf("Usage: ./recover imagen");
return 1;
}
// save arguments
char *infile = argv[1];
// open input file
FILE *inptr = fopen(infile, "r");
if(inptr == NULL)
{
printf("File could not be readn");
return 2;
}
//open output file
FILE* img = NULL;

BYTE buffer[512];
int block = 512;
bool newJpeg = false;
bool run = false;
char filename[8];
int filenameCnt = 0;

// run until end of file
while (fread(buffer, 512, 1, inptr) == 1)
{
// read 512 blocks UNTIL FIRST HEADER FOUND, checking to see if first four bytes indicate JPEG
if (newJpeg == false)
{
for (int i = 0; i < block; i++)
{
fread(&buffer[i], 1, 512, inptr);
}
}
// create jpeg file if jpeg header is found
if(checkheader(buffer) == true)
{
sprintf(filename,"%03d.jpg", filenameCnt); // creating a new jpeg
filenameCnt++;
img = fopen(filename, "w"); // making the jpeg writable
run = true;
// write data of block after header
for (int j = 0; j < block - 4; j++)
{
fwrite(&buffer[j+4], 1, 512, img);
}
}
// read and write new blocks into jpeg until new header is found
while (run == true)
{
// read new block
for (int i = 0; i < block; i++)
{
fread(&buffer[i], 1, 512, inptr);
}
// check if new block has header
if(checkheader(buffer) == true)
{
fclose(img); // close current jpeg so new one could be opened
newJpeg = true;
run = false;
}
else
// if not, continue writing new blocks into the current jpeg.
for (int j = 0; j < block; j++)
{
fwrite(&buffer[j], 1, 512, img);
}
}
}
return 0;
}
bool checkheader(BYTE buffer[512])
{
if(buffer[0] == 0xff && buffer[1] == 0xd8 && buffer[2] == 0xff && (buffer[3] & 0xf0) == 0xe0)
return true;
else
return false;
}

它应该输出 50 个 JPEG 文件。任何帮助或意见将不胜感激。谢谢。

Answer 1

这个"cs50 中的调试器不起作用"是否意味着debug50？也许您在段错误之后设置了断点。段错误fread(&buffer[i], 1, 512, inptr);(第 51 行)出现。基本上，对于某些 i，它不能在缓冲区 [i] 的地址"容纳"512 字节的数据。

一般来说，输入文件应该在一个地方读取，如下所示：while (fread(buffer, 512, 1, inptr) == 1)

这个循环是什么：

for (int i = 0; i < block; i++)
{
fread(&buffer[i], 1, 512, inptr);
}

要做(如果它没有段错误！它将从文件中读取 512 个 512 字节的块。哇，它肯定会错过一些非常重要的数据，不是吗？

程序需要在一个地方读取inptr，while是合适的，然后决定如何处理每个块读取。有三种选择：

• 丢弃(未找到第一个 JPEG 签名)
• 写入当前 JPEG
• 开始一个新的 JPEG(它是一个签名块)

请记住，签名块是输出不可或缺的一部分。

// write data of block after header
for (int j = 0; j < block - 4; j++)
{
fwrite(&buffer[j+4], 1, 512, img);
} 

正在丢弃标头，这将导致输出损坏。