使用malloc()函数对堆进行错误检查

时间:2018-11-02 14:58:31

标签: malloc heap

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

`#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;
}

enter image description here

enter image description here

1 个答案:

答案 0 :(得分:0)

在某些情况下,程序可能会从分配存储的失败尝试中正常恢复。因此,尽管某些语言将分配失败视为致命错误,但C标准指定malloc()如果无法分配可用存储,则可能返回空指针(标准语言建议执行此操作因此,除了在某些环境中不可行之外。

尽管malloc()失败会产生空指针而不是致命错误,这在可能进行正常恢复的情况下可能会有所帮助,但在程序需要一定数量的内存以便执行以下操作的情况下,它的作用就较小做任何有用的事情,没有它就无法发挥作用。在这种情况下,必须对每个malloc()进行空检查,这将浪费程序员的精力和代码空间。因此,当空间可用时,具有一个类似于malloc()的函数通常很有用,但是可以保证除了有效指针外,它永远不会返回任何东西。如果它不能返回有效的指针,它将根本不会返回。上面的代码看起来像是该功能的实现以及测试功能。