1 | /*
|
---|
2 | * $Id$
|
---|
3 | *
|
---|
4 | * Code (c)2006 Bruno Cornec <bruno@mondorescue.org>
|
---|
5 | *
|
---|
6 | * Main file of mr_mem : a very small and simple
|
---|
7 | * library for memory management
|
---|
8 | *
|
---|
9 | * Provided under the GPLv2
|
---|
10 | */
|
---|
11 |
|
---|
12 | #ifndef _GNU_SOURCE
|
---|
13 | #define _GNU_SOURCE
|
---|
14 | #endif
|
---|
15 |
|
---|
16 | #include <stdio.h>
|
---|
17 | #include <stdlib.h>
|
---|
18 |
|
---|
19 | #include "mr_err.h"
|
---|
20 | #include "mr_msg.h"
|
---|
21 |
|
---|
22 | /*
|
---|
23 | * Function that frees memory if necessary
|
---|
24 | * A pointer to the memory pointed is passed to it.
|
---|
25 | * *allocated variable points then to the original content
|
---|
26 | * pointed to by the caller
|
---|
27 | * In case of NULL pointer it just logs it
|
---|
28 | */
|
---|
29 | void mr_free_int(void **allocated, int line, const char *file) {
|
---|
30 |
|
---|
31 | /* free man pages says that if allocated is NULL
|
---|
32 | * nothing happens
|
---|
33 | */
|
---|
34 | if (*allocated != NULL) {
|
---|
35 | free(*allocated);
|
---|
36 | *allocated = NULL;
|
---|
37 | } else {
|
---|
38 | mr_msg_int(98,line,file,"mr_free_int","Attempt to free NULL pointer");
|
---|
39 | }
|
---|
40 | }
|
---|
41 |
|
---|
42 | /* encapsulation function for malloc */
|
---|
43 | void *mr_malloc_int(size_t size, int line, const char *file) {
|
---|
44 |
|
---|
45 | void *ret;
|
---|
46 |
|
---|
47 | ret = malloc(size);
|
---|
48 | if (ret == NULL) {
|
---|
49 | mr_msg_int(1,line,file,"mr_malloc_int","Unable to alloc memory in mr_malloc\nExiting...");
|
---|
50 | mr_exit(-1,"Unable to alloc memory in mr_malloc");
|
---|
51 | }
|
---|
52 | return(ret);
|
---|
53 | }
|
---|
54 |
|
---|
55 | /*
|
---|
56 | * Function that properly allocates a string from a format and params by encapsulating asprintf
|
---|
57 | * freeing it before in any case
|
---|
58 | */
|
---|
59 | void mr_asprintf_int(char **strp, int line, const char *file, const char *fmt, ...) {
|
---|
60 |
|
---|
61 | int res = 0;
|
---|
62 | va_list args;
|
---|
63 |
|
---|
64 | if ((strp != NULL) && (*strp != NULL)) {
|
---|
65 | mr_free_int((void **)strp, line, file);
|
---|
66 | }
|
---|
67 |
|
---|
68 | va_start(args,fmt);
|
---|
69 | res = vasprintf(strp, fmt, args);
|
---|
70 | if (res == -1) {
|
---|
71 | mr_msg_int(1,line,file,"mr_asprintf_int","Unable to alloc memory in mr_asprintf\nExiting...");
|
---|
72 | mr_exit(-1,"Unable to alloc memory in mr_asprintf");
|
---|
73 | }
|
---|
74 | va_end(args);
|
---|
75 | }
|
---|
76 |
|
---|
77 | /* encapsulation function for getline */
|
---|
78 | void mr_getline_int(char **lineptr, FILE *fd, int line, const char *file) {
|
---|
79 |
|
---|
80 | ssize_t ret;
|
---|
81 | size_t n = 0;
|
---|
82 |
|
---|
83 | if (*lineptr != NULL) {
|
---|
84 | /* free a variable which should be already free */
|
---|
85 | mr_msg_int(1,line,file,"mr_getline_int","mr_getline found a non freed variable here");
|
---|
86 | mr_free_int((void **)lineptr,line,file);
|
---|
87 | }
|
---|
88 | ret = getline(lineptr,&n,fd);
|
---|
89 | if ((ret == -1) && (! feof(fd))) {
|
---|
90 | mr_msg_int(1,line,file,"mr_getline_int","Unable to alloc memory in mr_getline\nExiting...");
|
---|
91 | mr_exit(-1,"Unable to alloc memory in mr_getline");
|
---|
92 | }
|
---|
93 | /* We reached end of file, allocating empty string */
|
---|
94 | if (ret == -1) {
|
---|
95 | mr_free_int((void **)lineptr,line,file);
|
---|
96 | mr_asprintf_int(lineptr,line,file,"");
|
---|
97 | }
|
---|
98 | }
|
---|
99 |
|
---|
100 | /*
|
---|
101 | * Function that properly put a variable in the environment
|
---|
102 | */
|
---|
103 | void mr_setenv_int(const char *name, const char *value, int line, char *file) {
|
---|
104 |
|
---|
105 | if (name == NULL) {
|
---|
106 | mr_msg_int(1,line,file,"mr_setenv_int","Unable to setenv a NULL variable\nExiting...");
|
---|
107 | mr_exit(-1, "Unable to setenv a NULL variable");
|
---|
108 | }
|
---|
109 | if (value == NULL) {
|
---|
110 | mr_msg_int(1,line,file,"mr_setenv_int","Unable to affect NULL to %s\nExiting...", name);
|
---|
111 | mr_exit(-1, "Unable to affect a NULL variable");
|
---|
112 | }
|
---|
113 | if (setenv(name, value, 1) != 0) {
|
---|
114 | mr_msg_int(1,line,file,"mr_setenv_int","Unable to put %s in environment", name);
|
---|
115 | mr_exit(-1,"Unable to put in environment");
|
---|
116 | }
|
---|
117 | }
|
---|
118 |
|
---|
119 | /*
|
---|
120 | * Equivalent function of strcat but safe
|
---|
121 | * from memory allocation point of view
|
---|
122 | * and richer from an interface point of view
|
---|
123 | */
|
---|
124 | void mr_strcat_int(char **in, int line, const char *file, const char *fmt, ...) {
|
---|
125 | char *fmt2 = NULL;
|
---|
126 | va_list args;
|
---|
127 | int res = 0;
|
---|
128 |
|
---|
129 | if (fmt == NULL) {
|
---|
130 | return;
|
---|
131 | }
|
---|
132 | if (in == NULL) {
|
---|
133 | mr_msg_int(1,line,file,"mr_strcat_int","Unable to add to NULL pointer\nExiting...");
|
---|
134 | mr_exit(-1, "Unable to add to a NULL pointer");
|
---|
135 | }
|
---|
136 | va_start(args,fmt);
|
---|
137 | if (*in == NULL) {
|
---|
138 | res = vasprintf(in, fmt, args);
|
---|
139 | if (res == -1) {
|
---|
140 | mr_msg_int(1,line,file,"mr_strcat_int","Unable to alloc memory in mr_strcat\nExiting...");
|
---|
141 | mr_exit(-1,"Unable to alloc memory in mr_strcat");
|
---|
142 | }
|
---|
143 | } else {
|
---|
144 | mr_asprintf_int(&fmt2, line, file, "%s%s", *in, fmt);
|
---|
145 | mr_free_int((void **)in,line,file);
|
---|
146 | res = vasprintf(in, fmt2, args);
|
---|
147 | if (res == -1) {
|
---|
148 | mr_msg_int(1,line,file,"mr_strcat_int","Unable to alloc memory in mr_strcat\nExiting...");
|
---|
149 | mr_exit(-1,"Unable to alloc memory in mr_strcat");
|
---|
150 | }
|
---|
151 | mr_free_int((void **)&fmt2,line,file);
|
---|
152 | }
|
---|
153 | va_end(args);
|
---|
154 | }
|
---|
155 |
|
---|
156 |
|
---|