Question

几个小时以来，我一直在尝试找出代码的问题所在。我指的是别人的代码，但我仍然不知道出什么问题了。代码可以正常编译，但是调试器说分段错误发生在第44行（在注释下方）。也欢迎提出改进我的代码的建议。请帮助，谢谢。

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

typedef uint8_t BYTE;

int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        printf("Usage: ./recover image\n");
        return 1;
    }
    FILE *f = fopen(argv[1],"r");
    if (!f)
    {
        return 2;
    }
    BYTE buffer[512];
    FILE* img = NULL;
    int i = 0;
    while (fread(buffer, 512, 1, f) == 1)
    {
        if (buffer[0] == 0xff && buffer[1] == 0xd8 && buffer[2] == 0xff
        && (buffer[3] & 0xf0) == 0xe0)
        {
            //create jpeg
            char filename[8];
            if (img != NULL)
            {
                fclose(img);
                i++;

                sprintf(filename, "%03i.jpg",i);
                img = fopen(filename, "w");
                fwrite(buffer, 512 , 1, img);
            }
            else
            {

                sprintf(filename, "%03i.jpg",i);
                img = fopen(filename, "w");
                //error in the line below :(
                fwrite(buffer, 512 , 1, img);
            }
        }
        else if (i > 0)
        {
            fwrite(buffer, 512 , 1, img);
        }

    }
    fclose(img);
    fclose(f);
}

Answer 1

您似乎认为该行最可能的失败原因是文件句柄无效。因此，我要做的 first 首先是检查每个文件是否正确打开，例如：

img = fopen(filename, "w");
if (img == NULL) {
    printf("Error opening file '%s'\n", filename);
    exit(1);
}

检查函数中的失败条件通常是一个好主意，这通常是个好主意，其中还包括fwrite调用：

if (fwrite(buffer, 512 , 1, img) != 1) {
    printf("Error writing to file\n");
    exit(1);
}

CS50中的分段错误恢复

1 个答案: