src/interfaces/ecpg/ecpglib/memory.c

   1 /* $PostgreSQL$ */
   2
   3 #define POSTGRES_ECPG_INTERNAL
   4 #include "postgres_fe.h"
   5
   6 #include "ecpg-pthread-win32.h"
   7 #include "ecpgtype.h"
   8 #include "ecpglib.h"
   9 #include "ecpgerrno.h"
  10 #include "extern.h"
  11
  12 void
  13 ecpg_free(void *ptr)
  14 {
  15         free(ptr);
  16 }
  17
  18 char *
  19 ecpg_alloc(long size, int lineno)
  20 {
  21         char       *new = (char *) calloc(1L, size);
  22
  23         if (!new)
  24         {
  25                 ecpg_raise(lineno, ECPG_OUT_OF_MEMORY, ECPG_SQLSTATE_ECPG_OUT_OF_MEMORY, NULL);
  26                 return NULL;
  27         }
  28
  29         return (new);
  30 }
  31
  32 char *
  33 ecpg_realloc(void *ptr, long size, int lineno)
  34 {
  35         char       *new = (char *) realloc(ptr, size);
  36
  37         if (!new)
  38         {
  39                 ecpg_raise(lineno, ECPG_OUT_OF_MEMORY, ECPG_SQLSTATE_ECPG_OUT_OF_MEMORY, NULL);
  40                 return NULL;
  41         }
  42
  43         return (new);
  44 }
  45
  46 char *
  47 ecpg_strdup(const char *string, int lineno)
  48 {
  49         char       *new;
  50
  51         if (string == NULL)
  52                 return NULL;
  53
  54         new = strdup(string);
  55         if (!new)
  56         {
  57                 ecpg_raise(lineno, ECPG_OUT_OF_MEMORY, ECPG_SQLSTATE_ECPG_OUT_OF_MEMORY, NULL);
  58                 return NULL;
  59         }
  60
  61         return (new);
  62 }
  63
  64 /* keep a list of memory we allocated for the user */
  65 struct auto_mem
  66 {
  67         void       *pointer;
  68         struct auto_mem *next;
  69 };
  70
  71 #ifdef ENABLE_THREAD_SAFETY
  72 static pthread_key_t auto_mem_key;
  73 static pthread_once_t auto_mem_once = PTHREAD_ONCE_INIT;
  74
  75 static void
  76 auto_mem_destructor(void *arg)
  77 {
  78         ECPGfree_auto_mem();
  79 }
  80
  81 static void
  82 auto_mem_key_init(void)
  83 {
  84         pthread_key_create(&auto_mem_key, auto_mem_destructor);
  85 }
  86
  87 static struct auto_mem *
  88 get_auto_allocs(void)
  89 {
  90         pthread_once(&auto_mem_once, auto_mem_key_init);
  91         return (struct auto_mem *) pthread_getspecific(auto_mem_key);
  92 }
  93
  94 static void
  95 set_auto_allocs(struct auto_mem * am)
  96 {
  97         pthread_setspecific(auto_mem_key, am);
  98 }
  99 #else
 100 static struct auto_mem *auto_allocs = NULL;
 101
 102 #define get_auto_allocs()               (auto_allocs)
 103 #define set_auto_allocs(am)             do { auto_allocs = (am); } while(0)
 104 #endif
 105
 106 void
 107 ecpg_add_mem(void *ptr, int lineno)
 108 {
 109         struct auto_mem *am = (struct auto_mem *) ecpg_alloc(sizeof(struct auto_mem), lineno);
 110
 111         am->pointer = ptr;
 112         am->next = get_auto_allocs();
 113         set_auto_allocs(am);
 114 }
 115
 116 void
 117 ECPGfree_auto_mem(void)
 118 {
 119         struct auto_mem *am = get_auto_allocs();
 120
 121         /* free all memory we have allocated for the user */
 122         if (am)
 123         {
 124                 do
 125                 {
 126                         struct auto_mem *act = am;
 127
 128                         am = am->next;
 129                         ecpg_free(act->pointer);
 130                         ecpg_free(act);
 131                 } while (am);
 132                 set_auto_allocs(NULL);
 133         }
 134 }
 135
 136 void
 137 ecpg_clear_auto_mem(void)
 138 {
 139         struct auto_mem *am = get_auto_allocs();
 140
 141         /* only free our own structure */
 142         if (am)
 143         {
 144                 do
 145                 {
 146                         struct auto_mem *act = am;
 147
 148                         am = am->next;
 149                         ecpg_free(act);
 150                 } while (am);
 151                 set_auto_allocs(NULL);
 152         }
 153 }