source: branches/3.2/mondo/src/lib/mr_mem.c @ 3205

Last change on this file since 3205 was 3205, checked in by Bruno Cornec, 7 years ago
  • Fix mondoarchive which now works in text mode. GUI mode still not working and crashing on first update for backup. Needs more

investigation in newt-specific.c

  • Property svn:eol-style set to native
File size: 3.9 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 asprintf */
56void mr_asprintf_int(char **strp, int line, const char *file, const char *fmt, ...) {
57
58    int res = 0;
59    va_list args;
60
61    va_start(args,fmt);
62    res = vasprintf(strp, fmt, args);
63    if (res == -1) {
64        mr_msg_int(1,line,file,"Unable to alloc memory in mr_asprintf\nExiting...");
65        mr_exit(-1,"Unable to alloc memory in mr_asprintf");
66    }
67    va_end(args);
68}
69
70/* encapsulation function for getline */
71void mr_getline_int(char **lineptr, FILE *fd, int line, const char *file) {
72   
73    ssize_t ret;
74    size_t n = 0;
75
76    /*  We reached end of file, allocating empty string */
77    if (feof(fd)) {
78        mr_asprintf_int(lineptr, line, file, "");
79        return;
80    }
81    ret = getline(lineptr,&n,fd);
82    if (ret == -1) {
83        mr_msg_int(1,line,file,"Unable to alloc memory in mr_getline\nExiting...");
84        mr_exit(-1,"Unable to alloc memory in mr_getline");
85    }
86}
87
88/*
89 * Function that properly allocates a string from another one
90 * freeing it before in any case
91 */
92void mr_allocstr_int(char *alloc, const char *orig, int line, const char *file) {
93
94    if (alloc != NULL) {
95        mr_free_int((void **)&alloc, line, file); 
96    }
97    mr_asprintf_int(&alloc, line, file, orig);
98}
99
100/*
101 * Function that properly put a variable in the environment
102 */
103void 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,"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,"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,"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 */
124void 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,"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,"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,"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
Note: See TracBrowser for help on using the repository browser.