source: MondoRescue/branches/2.2.10/mondo/src/lib/mr_mem.c @ 2320

Last change on this file since 2320 was 2320, checked in by Bruno Cornec, 11 years ago

r3331@localhost: bruno | 2009-08-06 01:34:32 +0200

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