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(8,line,file,"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,"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 | /* encapsulation function for getline */
|
---|
56 | void mr_getline_int(char **lineptr, size_t *n, FILE *fd, int line, const char *file) {
|
---|
57 |
|
---|
58 | ssize_t ret;
|
---|
59 |
|
---|
60 | ret = getline(lineptr,n,fd);
|
---|
61 | if ((ret == -1) && (! feof(fd))) {
|
---|
62 | mr_msg_int(1,line,file,"Unable to alloc memory in mr_getline\nExiting...");
|
---|
63 | mr_exit(-1,"Unable to alloc memory in mr_getline");
|
---|
64 | }
|
---|
65 | }
|
---|
66 |
|
---|
67 | /* encapsulation function for asprintf */
|
---|
68 | void mr_asprintf_int(char **strp, int line, const char *file, const char *fmt, ...) {
|
---|
69 |
|
---|
70 | int res = 0;
|
---|
71 | va_list args;
|
---|
72 |
|
---|
73 | va_start(args,fmt);
|
---|
74 | res = vasprintf(strp, fmt, args);
|
---|
75 | if (res == -1) {
|
---|
76 | mr_msg_int(1,line,file,"Unable to alloc memory in mr_asprintf\nExiting...");
|
---|
77 | mr_exit(-1,"Unable to alloc memory in mr_asprintf");
|
---|
78 | }
|
---|
79 | va_end(args);
|
---|
80 | }
|
---|
81 |
|
---|
82 | /*
|
---|
83 | * Function that properly allocates a string from another one
|
---|
84 | * freeing it before in any case
|
---|
85 | */
|
---|
86 | void mr_allocstr_int(char *alloc, const char *orig, int line, const char *file) {
|
---|
87 |
|
---|
88 | if (alloc != NULL) {
|
---|
89 | mr_free_int((void **)&alloc, line, file);
|
---|
90 | }
|
---|
91 | mr_asprintf_int(&alloc, line, file, orig);
|
---|
92 | }
|
---|
93 |
|
---|
94 | /*
|
---|
95 | * Function that properly put a variable in the environment
|
---|
96 | */
|
---|
97 | void mr_setenv_int(const char *name, const char *value, int line, char *file) {
|
---|
98 |
|
---|
99 | if (name == NULL) {
|
---|
100 | mr_msg_int(1,line,file,"Unable to setenv a NULL variable\nExiting...");
|
---|
101 | mr_exit(-1, "Unable to setenv a NULL variable");
|
---|
102 | }
|
---|
103 | if (value == NULL) {
|
---|
104 | mr_msg_int(1,line,file,"Unable to affect NULL to %s\nExiting...", name);
|
---|
105 | mr_exit(-1, "Unable to affect a NULL variable");
|
---|
106 | }
|
---|
107 | if (setenv(name, value, 1) != 0) {
|
---|
108 | mr_msg_int(1,line,file,"Unable to put %s in environment", name);
|
---|
109 | mr_exit(-1,"Unable to put in environment");
|
---|
110 | }
|
---|
111 | }
|
---|
112 |
|
---|
113 | /*
|
---|
114 | * Equivalent function of strcat but safe
|
---|
115 | * from memory allocation point of view
|
---|
116 | * and richer from an interface point of view
|
---|
117 | */
|
---|
118 | void mr_strcat_int(char **in, int line, const char *file, const char *fmt, ...) {
|
---|
119 |
|
---|
120 | char *fmt2 = NULL;
|
---|
121 | va_list args;
|
---|
122 | int res = 0;
|
---|
123 |
|
---|
124 | if (fmt == NULL) {
|
---|
125 | return;
|
---|
126 | }
|
---|
127 | if (in == NULL) {
|
---|
128 | mr_msg_int(1,line,file,"Unable to add to NULL pointer\nExiting...");
|
---|
129 | mr_exit(-1, "Unable to add to a NULL pointer");
|
---|
130 | }
|
---|
131 | va_start(args,fmt);
|
---|
132 | if (*in == NULL) {
|
---|
133 | res = vasprintf(in, fmt, args);
|
---|
134 | if (res == -1) {
|
---|
135 | mr_msg_int(1,line,file,"Unable to alloc memory in mr_strcat\nExiting...");
|
---|
136 | mr_exit(-1,"Unable to alloc memory in mr_strcat");
|
---|
137 | }
|
---|
138 | } else {
|
---|
139 | mr_asprintf_int(&fmt2, line, file, "%s%s", *in, fmt);
|
---|
140 | mr_free_int((void **)in,line,file);
|
---|
141 | res = vasprintf(in, fmt2, args);
|
---|
142 | if (res == -1) {
|
---|
143 | mr_msg_int(1,line,file,"Unable to alloc memory in mr_strcat\nExiting...");
|
---|
144 | mr_exit(-1,"Unable to alloc memory in mr_strcat");
|
---|
145 | }
|
---|
146 | mr_free_int((void **)&fmt2,line,file);
|
---|
147 | }
|
---|
148 | va_end(args);
|
---|
149 | }
|
---|
150 |
|
---|
151 |
|
---|