要在程序结束时自动释放通过 `malloc` 分配的内存,你可以使用 `atexit` 函数注册一个退出处理函数。以下是一个简单的示例代码:
#include <stdio.h>
#include <stdlib.h>
// 定义一个指针来保存分配的内存地址
int *array;
// 退出处理函数,用来释放内存
void release_memory() {
if (array != NULL) {
free(array);
printf("Memory has been released.\n");
}
}
int main() {
// 指定数组大小
size_t size = 10;
// 通过 malloc 分配内存
array = (int *)malloc(size * sizeof(int));
if (array == NULL) {
fprintf(stderr, "Memory allocation failed.\n");
return EXIT_FAILURE;
}
// 注册退出处理函数
if (atexit(release_memory) != 0) {
fprintf(stderr, "Cannot set exit function.\n");
free(array); // 如果注册失败手动释放并退出
return EXIT_FAILURE;
}
// 在程序中可以使用数组
for (size_t i = 0; i < size; i++) {
array[i] = i;
}
// 正常退出程序时,release_memory 会被调用
return EXIT_SUCCESS;
}
### 解释
1. **内存分配**:使用 `malloc` 分配一个整数数组,并将地址存储在指针 `array` 中。
2. **注册退出函数**:使用 `atexit(release_memory)` 注册 `release_memory` 作为程序退出时的处理函数。
3. **释放内存**:在 `release_memory` 函数中,调用 `free` 释放分配的内存。
4. **退出处理**:当程序正常退出时,`atexit` 注册的函数会被调用,确保内存被释放。
以上回复来自 FishC AI 小助理,如未能正确解答您的问题,请继续追问。 |