util.c

   1 /*
   2  * Copyright (C) 2012
   3  *      Dale Weiler
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a copy of
   6  * this software and associated documentation files (the "Software"), to deal in
   7  * the Software without restriction, including without limitation the rights to
   8  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
   9  * of the Software, and to permit persons to whom the Software is furnished to do
  10  * so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in all
  13  * copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21  * SOFTWARE.
  22  */
  23 #include <stdint.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include "gmqcc.h"
  27
  28 struct memblock_t {
  29         const char  *file;
  30         unsigned int line;
  31         unsigned int byte;
  32 };
  33
  34 void *util_memory_a(unsigned int byte, unsigned int line, const char *file) {
  35         struct memblock_t *data = malloc(sizeof(struct memblock_t) + byte);
  36         if (!data) return NULL;
  37         data->line = line;
  38         data->byte = byte;
  39         data->file = file;
  40
  41         printf("[MEM] allocation: %08u (bytes) at %s:%u\n", byte, file, line);
  42         return (void*)((uintptr_t)data+sizeof(struct memblock_t));
  43 }
  44
  45 void util_memory_d(void *ptrn, unsigned int line, const char *file) {
  46         if (!ptrn) return;
  47         void              *data = (void*)((uintptr_t)ptrn-sizeof(struct memblock_t));
  48         struct memblock_t *info = (struct memblock_t*)data;
  49
  50         printf("[MEM] released:   %08u (bytes) at %s:%u\n", info->byte, file, line);
  51         free(data);
  52 }
  53
  54 #ifndef mem_d
  55 #define mem_d(x) util_memory_d((x), __LINE__, __FILE__)
  56 #endif
  57 #ifndef mem_a
  58 #define mem_a(x) util_memory_a((x), __LINE__, __FILE__)
  59 #endif
  60
  61 /*
  62  * Some string utility functions, because strdup uses malloc, and we want
  63  * to track all memory (without replacing malloc).
  64  */
  65 char *util_strdup(const char *s) {
  66         size_t  len;
  67         char   *ptr;
  68
  69         if (!s)
  70                 return NULL;
  71
  72         len = strlen(s);
  73         ptr = mem_a (len+1);
  74
  75         if (ptr && len) {
  76                 memcpy(ptr, s, len);
  77                 ptr[len] = '\0';
  78         }
  79
  80         return ptr;
  81 }