c - C コードを拡張して 3D 配列を動的に割り当てる

Question

2D 配列を作成する次のコードに基づいて 3D 配列を動的に割り当てたいと考えています。

void** malloc_2d (size_t nrows, size_t ncols, int elementsize) {
size_t i;
void ** ptr;

if ( (ptr = (void**)malloc(nrows * sizeof(void *))) == NULL ) {

  fprintf(stderr, "malloc_2d: out of memory\n");

  exit(1);

}


 if ( (ptr[0] = malloc(nrows * ncols * elementsize)) == NULL ) {

  fprintf(stderr, "malloc_2d: out of memory\n");

  exit(1);

 }

for (i=1; i<nrows; i++) 

  ptr[i] = (char*)ptr[0] + i * ncols * elementsize;

 return ptr;
 }

ありがとう