每当我运行程序时,都会发生分段错误。通过使用printf函数,我缩小了在第二个while循环的第一次迭代中,在代码fwrite(buffer,sizeof(BYTE),block,output)的最后一行中,在我的fwrite函数中出现段错误。我已经尝试了很多东西,但是我无法使段错误消失。知道是什么原因造成的,以及如何解决?
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <cs50.h>
#include <string.h>
typedef uint8_t BYTE;
const int block = 512;
int main(int argc, char *argv[])
{
if (argc != 2)
{
printf("Usage: ./recover card.raw");
return 1;
}
//open file for reading
char *infile = argv[1];
FILE *file = fopen(infile, "r");
if (file == NULL)
{
printf("Error while opening file");
return 1;
}
//initializing
BYTE buffer[block];
memset(buffer, 0x00, block);
BYTE x = 0;
memset(&x, 0x00, 1);
BYTE JPEG[3] = {0xff, 0xd8, 0xff};
char filename[8];
int imagecount = 0;
//open file output for writing
FILE* output = NULL;
output = fopen("000.jpg", "w");
while (1)
{
fread(&x, sizeof(BYTE), 1, file);
if (x == JPEG[0])
{
fread(&x, sizeof(BYTE), 1, file);
if (x == JPEG[1])
{
fread(&x, sizeof(BYTE), 1, file);
if (x == JPEG[2])
{
fread(&x, sizeof(BYTE), 1, file);
int y = (int) x;
if ((y >= 224) && (y <= 239))
{
break;
}
}
}
}
}
fwrite(JPEG, sizeof(BYTE), 3, output);
fwrite(&x, sizeof(BYTE), 1, output);
fread(output, sizeof(BYTE), block - 4, file);
while (1)
{
int n = fread(&buffer, sizeof(BYTE), block, file);
if (n == 0)
{
break;
}
if (buffer[0] == JPEG[0] && buffer[1] == JPEG[1] && buffer[2] == JPEG[2] && output != NULL)
{
fclose(output);
imagecount++;
sprintf(filename, "%03i.jpg", imagecount);
output = fopen(filename, "w");
}
printf("z");
fwrite(buffer, sizeof(BYTE), block, output);
printf("z");
}
}