时间戳

首页 > 解决方案 > 带有 malloc() 函数的错误检查堆

malloc - 带有 malloc() 函数的错误检查堆

问题描述

我无法理解此源代码的作用。不仅如此，每当我运行此源代码的调试时，我都会不断收到相同的窗口。以下是提供的源代码以及输出错误和弹出窗口的屏幕截图。如果有人可以分解此源代码并指导或指出正确的方向以使此源代码正常工作，我将不胜感激。 

`#include "stdafx.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>

void *errorchecked_malloc(unsigned int);    //Function prototype for errorchecked_malloc()

int main(int argc, char *argv[]) {
    char *char_ptr;     //A char pointr
    int *int_ptr;       //An integer pointer
    int mem_size;

    if (argc < 2)       //If there aren't command line arguments,
        mem_size = 50;  //use 50 as default value.
    else
        mem_size = atoi(argv[1]);

    printf("\t[+] allocating %d bytes of memory on the heap for char_ptr\n", mem_size);
    char_ptr = (char *)errorchecked_malloc(mem_size);   //Allocating heap memory

    /*if (char_ptr == NULL) {   //Error checking, in case of malloc() fails
        fprintf(stderr, "Error: could not allocate heap memmory.\n");
        exit(-1);
    } */

    strcpy_s(char_ptr, char_ptr[36], "This is memory located on the heap.");
    printf("char_ptr (%p) --> '%s'\n", char_ptr, char_ptr);
    printf("\t [+] allocating 12 bytes of memory on the heap for int_ptr\n");
    int_ptr = (int *)errorchecked_malloc(12);   //Allocated heap memory again

    /*if (int_ptr == NULL) {            //Error checking incse malloc() fails
        fprintf(stderr, "Error: could not allocate heap memmory.\n");
        exit(-1);
    }*/

    *int_ptr = 31337;               //Put the Value of 31337 where int_ptr is pointing
    printf("int_ptr (%p) --> '%s'\n", int_ptr, *int_ptr);

    printf("\t [-] freeing char_ptr's heap memory...\n");
    free(char_ptr);                 //Freeing heap memory

    printf("\t [+] allocating another 15 bytes for int_ptr\n");
    char_ptr = (char *)errorchecked_malloc(15); //Allocating more heap memory

    /*if (char_ptr == NULL) {           //Error checking incase malloc() fails
        fprintf(stderr, "Error: could not allocate heap memmory.\n");
        exit(-1);
    }*/

    strcpy_s(char_ptr, char_ptr[11], "new memory");
    printf("char_ptr (%p) --> '%s'\n", char_ptr, char_ptr);

    printf("\t[-] freeing int_ptr's heap memory...\n");
    free(int_ptr);                  //Freeing heap memory
    printf("\t[-] freeing char_ptr's heap memory...\n");
    free(char_ptr);                 //Freeing the other block of heap memory

    getchar();
    return 0;
}

void *errorchecked_malloc(unsigned int size) {      //An errorchecked_malloc function
    void *ptr;
    ptr = malloc(size);
    if (ptr == NULL) {
        fprintf(stderr, "Error: could not allocate heap memory.\n");
        exit(-1);
    }
    return ptr;
}

在此处输入图像描述

在此处输入图像描述

标签： mallocheap-memory

解决方案

在某些情况下，程序可能会从失败的分配存储尝试中正常恢复。因此，虽然某些语言将分配失败视为致命错误，但 C 标准指定malloc()如果无法分配可用存储，则可能返回空指针（标准语言建议必须这样做，除非在一些不实用的环境）。

尽管在malloc()可以进行优雅恢复的情况下，产生一个空指针而不是致命错误可能会有所帮助，但在程序需要一定数量的内存才能做任何有用的事情的情况下，它的用处不大，而且不会没有它就能够有用地行事。在这种情况下，必须对每个malloc()进行空检查将浪费程序员的精力和代码空间。因此，拥有一个行为类似于空间可用时的函数通常很有malloc()用，但它保证永远不会返回除有效指针之外的任何内容；如果它不能返回一个有效的指针，它根本不会返回。上面的代码看起来像是这样一个函数的实现以及它的测试工具。

推荐阅读