c - fclose()导致崩溃

时间:2015-02-27 00:19:15

标签: c fclose

我在这段代码中遇到问题,它成功执行,直到达到fclose(fp)语句,它崩溃了。

void read_file(const char *filename) {
    FILE *fp;
    int num, i=0;

    fp = fopen("numbers.txt","r");

    if (fp == NULL) {
        printf("Couldn't open numbers.txt for reading.\n");
        exit(0);
    }

    int *random = malloc(sizeof(int));
    if (random == NULL) {
        printf("Error allocating memory!\n");
        return;
    }
    while (fscanf(fp, "%d", &num) > 0) {
        random[i] = num;
        i++;
    }
    printf("\nTEST Before close");

    fclose(fp);

    printf("\nTEST After fclose");
}

句子TEST关闭前成功打印然后控制台停止打印所以TEST fclose没有打印并且光标开始闪烁!

有任何帮助吗? 提前谢谢。

1 个答案:

答案 0 :(得分:3)

问题在于

 int *random = malloc(sizeof(int));

仅分配 ONE 单个整数。

如果文件中有多个整数,则while将增加i索引,并且您将变量写入无效位置,从而导致内存损坏。这可以触发段错误,但它也可能在任何时候导致奇怪的行为,如您的情况。

   random[i] = num;  /* oops !!! if i>0 memory corruption */ 

<强>解决方案:

如果您知道number整数,则可以立即分配正确的内存量。我建议 calloc() 用于此目的,因为它用于数组分配,并将分配的内存初始化为0:

int *random = calloc(number, sizeof(int));  

如果您不知道数字,可以使用 realloc() 逐渐扩展数组的大小:

int number = 100;  /* arbitrary initial size*/ 
int *random = malloc(number*sizeof(int));  
...
while (fscanf(fp, "%d", &num) > 0) {
    if (i==number-1) {
        number  += 100;  /* consider allocating 100 more items */
        random = realloc (random, number*sizeof(int)); 
        if (random==NULL) {
            printf("Not enough momory for reading all the numbers.\n");
            exit(1);
         }
    }
    random[i] = num;
    i++;
}

最后一种方法是根据文件大小推断出最大整数:

fseek (fp, 0, SEEK_END);  // go to the end 
long size=ftell (fp);     // get the length of file
fseek (fp, 0, SEEK_SET);  // go to start
long number = size/2 + 1;  // each number is at least 1 digit folowed by a space, except the las one
if (n > INT_MAX) {
     printf("File too big !\n");
     exit(1);
}
int *random = calloc((size_t)number, sizeof(int));  
...

这当然是切实可行的,但遗憾的是SEEK_END并非所有库实现都支持。

相关问题