malloc/test1.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3
   4 #include "mpool.h"
   5
   6 int main(void)
   7 {
   8     mpool_t *mpool;
   9     mpret_t mpret;
  10     char *buffer;
  11
  12     /* Initialize memory pool with 1024 bytes */
  13     mpret = mpool_init(&mpool, 10, 5);
  14     if (mpret == MPOOL_ENOMEM) {
  15         fprintf(stderr, "mpool: not enough memory\n");
  16         exit(EXIT_FAILURE);
  17     }
  18     else if (mpret == MPOOL_EBADVAL) {
  19         fprintf(stderr, "mpool: bad value passed to mpool_init()\n");
  20         exit(EXIT_FAILURE);
  21     }
  22
  23     /* Allocate 32 bytes for buffer */
  24     if ((buffer = mpool_alloc(mpool, 5)) == NULL) {
  25         fprintf(stderr, "mpool: no available block in pool\n");
  26         mpool_destroy(mpool);
  27         exit(EXIT_FAILURE);
  28     }
  29
  30     /* Read input from standard input stream */
  31     fgets(buffer, 1<<5, stdin);
  32
  33     /* Print buffer's contents */
  34     printf("Buffer: %s", buffer);
  35
  36     /*
  37      * Free buffer --
  38      * could be omitted, since we call mpool_destroy() afterwards
  39     */
  40     mpool_free(mpool, buffer);
  41
  42     /* Destroy memory pool and free all resources */
  43     mpool_destroy(mpool);
  44
  45     return EXIT_SUCCESS;
  46 }