1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
#define MOS_FORCE_PREFIXES
#include "moslib/safe.h"
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#ifndef _WIN32
#include <unistd.h>
#else
#include <io.h>
#endif
#define ERR_MESS(...) err_mess(info, __VA_ARGS__)
static void err_mess(MosSafeErrInfo info, const char *msg, ...) {
va_list args;
va_start(args, msg);
fprintf(stderr, "%s:%d - fatal error in function %s:\n", info.file, info.line, info.func);
vfprintf(stderr, msg, args);
fprintf(stderr, "\n");
va_end(args);
}
void *mos_safe_malloc_fn(size_t size, MosSafeErrInfo info) {
void *p = malloc(size);
if (p)
return p;
ERR_MESS("failed to malloc %zu bytes", size);
exit(EXIT_FAILURE);
}
void *mos_safe_calloc_fn(size_t count, size_t size, MosSafeErrInfo info) {
void *p = calloc(count, size);
if (p)
return p;
ERR_MESS("failed to calloc %zu items of %zu size", count, size);
exit(EXIT_FAILURE);
}
void *mos_safe_realloc_fn(void *ptr, size_t size, MosSafeErrInfo info) {
void *p = realloc(ptr, size);
if (p)
return p;
ERR_MESS("failed to realloc %p to %zu bytes", ptr, size);
exit(EXIT_FAILURE);
}
FILE *mos_safe_fopen_fn(const char *path, const char *mode, MosSafeErrInfo info) {
FILE *p = fopen(path, mode);
if (p)
return p;
ERR_MESS("failed to fopen %s with mode %s", path, mode);
exit(EXIT_FAILURE);
}
FILE *mos_safe_tmpfile_fn(MosSafeErrInfo info) {
FILE *p = tmpfile();
if (p)
return p;
ERR_MESS("failed to create a tmpfile");
exit(EXIT_FAILURE);
}
int mos_safe_dup_fn(int fd, MosSafeErrInfo info) {
#ifndef _WIN32
int res = dup(fd);
#else
int res = _dup(fd);
#endif
if (res != -1)
return res;
ERR_MESS("failed to dup %d", fd);
exit(EXIT_FAILURE);
}
int mos_safe_dup2_fn(int fd, int fd2, MosSafeErrInfo info) {
#ifndef _WIN32
int res = dup2(fd, fd2);
#else
int res = _dup2(fd, fd2);
#endif
if (res != -1)
return res;
ERR_MESS("failed to dup %d 2 %d", fd, fd2);
exit(EXIT_FAILURE);
}
#ifndef _WIN32
pid_t mos_safe_fork_fn(MosSafeErrInfo info) {
pid_t pid = fork();
if (pid != -1)
return pid;
ERR_MESS("failed to fork");
exit(EXIT_FAILURE);
}
int mos_safe_pipe_fn(int fd[2], MosSafeErrInfo info) {
int res = pipe(fd);
if (res != -1)
return res;
ERR_MESS("failed to pipe");
exit(EXIT_FAILURE);
}
#endif
|
