首页 > 解决方案 > CS50 pset4 恢复:fwrite 函数中的 Seg-fault

问题描述

每当我运行程序时都会发生分段错误。使用 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");
}
}

标签: csegmentation-faultcs50

解决方案


你正在做一个fread(output,...,...,file)

也就是说,您正在读入一个缓冲区,即您的outputFILE 描述符。可能不是你想要的...

注意第一个参数fread是一个void *指针,所以很高兴接受它(即没有编译错误)。

换句话说,您正在破坏FILE *output指向的内容。


推荐阅读