1 /*
2  * Copyright (c) 2013 Cyril Hrubis <chrubis@suse.cz>
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License as
6  * published by the Free Software Foundation; either version 2 of
7  * the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it would be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write the Free Software Foundation,
16  * Inc.,  51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17  */
18 
19 #define _GNU_SOURCE
20 #include <stdarg.h>
21 #include <stdio.h>
22 #include <errno.h>
23 #include "test.h"
24 #include "safe_stdio_fn.h"
25 
26 FILE *safe_fopen(const char *file, const int lineno, void (cleanup_fn)(void),
27                  const char *path, const char *mode)
28 {
29 	FILE *f = fopen(path, mode);
30 
31 	if (f == NULL) {
32 		tst_brkm(TBROK | TERRNO, cleanup_fn,
33 			 "%s:%d: fopen(%s,%s) failed",
34 			 file, lineno, path, mode);
35 	}
36 
37 	return f;
38 }
39 
40 int safe_fclose(const char *file, const int lineno, void (cleanup_fn)(void),
41                 FILE *f)
42 {
43 	int ret;
44 
45 	ret = fclose(f);
46 
47 	if (ret) {
48 		tst_brkm(TBROK | TERRNO, cleanup_fn,
49 			 "%s:%d: fclose(%p) failed", file, lineno, f);
50 	}
51 
52 	return ret;
53 }
54 
55 int safe_asprintf(const char *file, const int lineno, void (cleanup_fn)(void),
56                   char **strp, const char *fmt, ...)
57 {
58 	int ret;
59 	va_list va;
60 
61 	va_start(va, fmt);
62 	ret = vasprintf(strp, fmt, va);
63 	va_end(va);
64 
65 	if (ret < 0) {
66 		tst_brkm(TBROK | TERRNO, cleanup_fn,
67 			 "%s:%d: asprintf(%s,...) failed", file, lineno, fmt);
68 	}
69 
70 	return ret;
71 }
72 
73 FILE *safe_popen(const char *file, const int lineno, void (cleanup_fn)(void),
74 		 const char *command, const char *type)
75 {
76 	FILE *stream;
77 	const int saved_errno = errno;
78 
79 	errno = 0;
80 	stream = popen(command, type);
81 
82 	if (stream == NULL) {
83 		if (errno != 0) {
84 			tst_brkm(TBROK | TERRNO, cleanup_fn,
85 				 "%s:%d: popen(%s,%s) failed",
86 				 file, lineno, command, type);
87 		} else {
88 			tst_brkm(TBROK, cleanup_fn,
89 				 "%s:%d: popen(%s,%s) failed: Out of memory",
90 				 file, lineno, command, type);
91 		}
92 	}
93 
94 	errno = saved_errno;
95 
96 	return stream;
97 }
98