C Programming: malloc() inside another function

I need help with malloc() inside another function.

I'm passing a pointer and size to the function from my main() and I would like to allocate memory for that pointer dynamically using malloc() from inside that called function, but what I see is that.... the memory, which is getting allocated, is for the pointer declared within my called function and not for the pointer which is inside the main().

How should I pass a pointer to a function and allocate memory for the passed pointer from inside the called function?

I have written the following code and I get the output as shown below.

SOURCE:

int main()
{
unsigned char *input_image;
unsigned int bmp_image_size = 262144;


if(alloc_pixels(input_image, bmp_image_size)==NULL)
printf("\nPoint2: Memory allocated: %d bytes",_msize(input_image));
else
printf("\nPoint3: Memory not allocated");
return 0;
}


signed char alloc_pixels(unsigned char *ptr, unsigned int size)
{
signed char status = NO_ERROR;
ptr = NULL;


ptr = (unsigned char*)malloc(size);


if(ptr== NULL)
{
status = ERROR;
free(ptr);
printf("\nERROR: Memory allocation did not complete successfully!");
}


printf("\nPoint1: Memory allocated: %d bytes",_msize(ptr));


return status;
}

PROGRAM OUTPUT:

Point1: Memory allocated ptr: 262144 bytes
Point2: Memory allocated input_image: 0 bytes
111811 次浏览
小开
最佳答案

You need to pass a pointer to a pointer as the parameter to your function.

int main()
{
unsigned char *input_image;
unsigned int bmp_image_size = 262144;


if(alloc_pixels(&input_image, bmp_image_size) == NO_ERROR)
printf("\nPoint2: Memory allocated: %d bytes",_msize(input_image));
else
printf("\nPoint3: Memory not allocated");
return 0;
}


signed char alloc_pixels(unsigned char **ptr, unsigned int size)
{
signed char status = NO_ERROR;
*ptr = NULL;


*ptr = (unsigned char*)malloc(size);


if(*ptr== NULL)
{
status = ERROR;
free(*ptr);      /* this line is completely redundant */
printf("\nERROR: Memory allocation did not complete successfully!");
}


printf("\nPoint1: Memory allocated: %d bytes",_msize(*ptr));


return status;
}
小开

If you want your function to modify the pointer itself, you'll need to pass it as a pointer to a pointer. Here's a simplified example:

void allocate_memory(char **ptr, size_t size) {
void *memory = malloc(size);
if (memory == NULL) {
// ...error handling (btw, there's no need to call free() on a null pointer. It doesn't do anything.)
}


*ptr = (char *)memory;
}


int main() {
char *data;
allocate_memory(&data, 16);
}
小开

This does not make sense :

if(alloc_pixels(input_image, bmp_image_size)==NULL)

alloc_pixels returns a signed char (ERROR or NO_ERROR) and you compare it to NULL (which is supposed to be used for pointers).

If you want input_image to be changed, you need to pass a pointer to it to alloc_pixels. alloc_pixels signature would be the following:

signed char alloc_pixels(unsigned char **ptr, unsigned int size)

You would call it like this:

alloc_pixels(&input_image, bmp_image_size);

And the memory allocation

*ptr = malloc(size);
小开

How should I pass a pointer to a function and allocate memory for the passed pointer from inside the called function?

Ask yourself this: if you had to write a function that had to return an int, how would you do it?

You'd either return it directly:

int foo(void)
{
return 42;
}

or return it through an output parameter by adding a level of indirection (i.e., using an int* instead of int):

void foo(int* out)
{
assert(out != NULL);
*out = 42;
}

So when you're returning a pointer type (T*), it's the same thing: you either return the pointer type directly:

T* foo(void)
{
T* p = malloc(...);
return p;
}

or you add one level of indirection:

void foo(T** out)
{
assert(out != NULL);
*out = malloc(...);
}
小开

You need to pass the pointer by reference, not by copy, the parameter in the function alloc_pixels requires the ampersand & to pass back out the address of the pointer - that is call by reference in C speak.

main()
{
unsigned char *input_image;
unsigned int bmp_image_size = 262144;


if(alloc_pixels(&input_image, bmp_image_size)==NULL)
printf("\nPoint2: Memory allocated: %d bytes",_msize(input_image));
else
printf("\nPoint3: Memory not allocated");


}


signed char alloc_pixels(unsigned char **ptr, unsigned int size)
{
signed char status = NO_ERROR;
*ptr = NULL;


*ptr = (unsigned char*)malloc(size);


if((*ptr) == NULL)
{
status = ERROR;
/* free(ptr);
printf("\nERROR: Memory allocation did not complete successfully!"); */
}


printf("\nPoint1: Memory allocated: %d bytes",_msize(*ptr));


return status;
}

I have commented out the two lines free(ptr) and "ERROR: ..." within the alloc_pixels function as that is confusing. You do not need to free a pointer if the memory allocation failed.

Edit: After looking at the msdn link supplied by OP, a suggestion, the code sample is the same as earlier in my answer.... but...change the format specifier to %u for the size_t type, in the printf(...) call in main().

main()
{
unsigned char *input_image;
unsigned int bmp_image_size = 262144;


if(alloc_pixels(&input_image, bmp_image_size)==NULL)
printf("\nPoint2: Memory allocated: %u bytes",_msize(input_image));
else
printf("\nPoint3: Memory not allocated");


}
小开

In your initial code , when you were passing input_image to the function alloc_pixels, compiler was creating a copy of it (i.e. ptr) and storing the value on the stack. You assign the value returned by malloc to ptr. This value is lost once the function returns to main and the stack unwinds. So, the memory is still allocated on heap but the memory location was never stored in (or assigned to )input_image, hence the issue.

You can change the signature of the function alloc_pixels which would be simpler to understand, and you won't require the additional 'status' variable as well.

unsigned char *alloc_pixels(unsigned int size)
{
unsigned char *ptr = NULL;
ptr = (unsigned char *)malloc(size);
if (ptr != NULL)
printf("\nPoint1: Memory allocated: %d bytes",_msize(ptr));
return ptr;
}

You can call the above function in main :

int main()
{
unsigned char *input_image;
unsigned int bmp_image_size = 262144;


if((input_image = alloc_pixels(bmp_image_size))==NULL)
printf("\nPoint3: Memory not allocated");
else
printf("\nPoint2: Memory allocated: %d bytes",_msize(input_image));
return 0;


}
小开

Parameters' assignment will work only if you set the value to its address.

There are 2 points that you should know before you attempt to solve this problem:
1. C Function: All the parameters you passed to the function will be a copy in the function.

That means every assignment that you've made in the function will not affect the variables outside the function, you're working on the copy actually:

int i = 1;
fun(i);
printf("%d\n", i);
//no matter what kind of changes you've made to i in fun, i's value will be 1

So, if you want to change i in the function, you need to know the difference between the thing and its copy:

The copy shared the value with the thing, but not the address.

And that's their only difference.

So the only way to change i in the function is using the address of i.

For example, there's a new function fun_addr:

void fun_addr(int *i) {
*i = some_value;
}

In this way, you could change i's value.

  1. malloc:

The key point in the fun_addr function is, you've passed a address to the function. And you could change the value stored in that address.

What will malloc do?

malloc will allocate a new memory space, and return the pointer pointed to that address back.

Look at this instruction:

int *array = (int*) malloc(sizeof(int) * SIZE);

What you are doing is let array's value equals to the address returned by malloc.

See? This is the same question, permanently assigning value to the parameter passed to the function. At this point, the value is address.

Now, assign the address(returned by malloc) to the address(stores the old address).

So the code should be:

void fun_addr_addr(int **p) {
*p = (int*) malloc(sizeof(int) * SIZE);
}

This one will work.

小开

The only way I could get pointer to a pointer solution to work for a similar problem I was having for this function

    BOOL OpenBitmap2 ( LPCTSTR pszFileName, char** pszBMPFile)

Was by assigning a temporary pointer to store the address

    char* BMPFile;
{ BMPFile = (char*)GlobalAlloc(GPTR, dwFileSize + 1);   // allocate storage using GlobalAlloc + 1 for null term string

then reassigning it

    {* pszBMPFile = BMPFile; return (0);} // Error = False

Any comment on why using "* pszBMPFile" directly with GlobalAlloc didn't work would be appreciated. I answered my own question. I forgot to carry the "*" through with pszBMPFile in the other lines of code. Good lessons from all the contributors. Many thanks.

小开

As mentioned in the other answers, we need a pointer to the pointer. But why?

We need to pass the value by a pointer in order to be able to modify the value. If you want to modify an int, you need to pass it by the int*.

In this question, the value we want to modify is a pointer int* (pointer changed from NULL to the address of the allocated memory), so we need to pass a pointer to the pointer int**.

By doing followed, pInt inside foo(int*) is a copy of the argument. When we allocate memory to the local variable, the one in the main() is intact.

void foo(int* pInt)
{
pInt = malloc(...);
}
int main()
{
int* pInt;
foo(pInt);
return 0;
}

So we need a pointer to pointer,

void foo(int** pInt)
{
*pInt = malloc(...);
}
int main()
{
int* pInt;
foo(&pInt);
return 0;
}

提交答案