buddy/test1.c

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