• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (c) 2017 Cyril Hrubis <chrubis@suse.cz>
3  *
4  * This program is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will 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, see <http://www.gnu.org/licenses/>.
16  */
17 
18 #define _GNU_SOURCE
19 #include <unistd.h>
20 #include <errno.h>
21 #include "config.h"
22 #ifdef HAVE_SYS_FANOTIFY_H
23 # include <sys/fanotify.h>
24 #endif
25 #define TST_NO_DEFAULT_MAIN
26 #include "tst_test.h"
27 #include "tst_safe_macros.h"
28 #include "lapi/personality.h"
29 
safe_setpgid(const char * file,const int lineno,pid_t pid,pid_t pgid)30 int safe_setpgid(const char *file, const int lineno, pid_t pid, pid_t pgid)
31 {
32 	int rval;
33 
34 	rval = setpgid(pid, pgid);
35 	if (rval) {
36 		tst_brk(TBROK | TERRNO,
37 		        "%s:%d: setpgid(%i, %i) failed",
38 			file, lineno, pid, pgid);
39 	}
40 
41 	return rval;
42 }
43 
safe_getpgid(const char * file,const int lineno,pid_t pid)44 pid_t safe_getpgid(const char *file, const int lineno, pid_t pid)
45 {
46 	pid_t pgid;
47 
48 	pgid = getpgid(pid);
49 	if (pgid == -1) {
50 		tst_brk(TBROK | TERRNO,
51 			"%s:%d: getpgid(%i) failed", file, lineno, pid);
52 	}
53 
54 	return pgid;
55 }
56 
safe_fanotify_init(const char * file,const int lineno,unsigned int flags,unsigned int event_f_flags)57 int safe_fanotify_init(const char *file, const int lineno,
58 	unsigned int flags, unsigned int event_f_flags)
59 {
60 	int rval;
61 
62 #ifdef HAVE_SYS_FANOTIFY_H
63 	rval = fanotify_init(flags, event_f_flags);
64 
65 	if (rval == -1) {
66 		if (errno == ENOSYS) {
67 			tst_brk(TCONF,
68 				"fanotify is not configured in this kernel.");
69 		}
70 		tst_brk(TBROK | TERRNO,
71 			"%s:%d: fanotify_init() failed", file, lineno);
72 	}
73 #else
74 	tst_brk(TCONF, "Header <sys/fanotify.h> is not present");
75 #endif /* HAVE_SYS_FANOTIFY_H */
76 
77 	return rval;
78 }
79 
safe_personality(const char * filename,unsigned int lineno,unsigned long persona)80 int safe_personality(const char *filename, unsigned int lineno,
81 		    unsigned long persona)
82 {
83 	int prev_persona = personality(persona);
84 
85 	if (prev_persona < 0) {
86 		tst_brk_(filename, lineno, TBROK | TERRNO,
87 			 "persona(%ld) failed", persona);
88 	}
89 
90 	return prev_persona;
91 }
92 
safe_setregid(const char * file,const int lineno,gid_t rgid,gid_t egid)93 int safe_setregid(const char *file, const int lineno,
94 		  gid_t rgid, gid_t egid)
95 {
96 	int rval;
97 
98 	rval = setregid(rgid, egid);
99 	if (rval == -1) {
100 		tst_brk_(file, lineno, TBROK | TERRNO,
101 			 "setregid(%li, %li) failed",
102 			 (long)rgid, (long)egid);
103 	}
104 
105 	return rval;
106 }
107 
108 
safe_setreuid(const char * file,const int lineno,uid_t ruid,uid_t euid)109 int safe_setreuid(const char *file, const int lineno,
110 		  uid_t ruid, uid_t euid)
111 {
112 	int rval;
113 
114 	rval = setreuid(ruid, euid);
115 	if (rval == -1) {
116 		tst_brk_(file, lineno, TBROK | TERRNO,
117 			 "setreuid(%li, %li) failed",
118 			 (long)ruid, (long)euid);
119 	}
120 
121 	return rval;
122 }
123 
124 
safe_sigaction(const char * file,const int lineno,int signum,const struct sigaction * act,struct sigaction * oldact)125 int safe_sigaction(const char *file, const int lineno,
126                    int signum, const struct sigaction *act,
127                    struct sigaction *oldact)
128 {
129 	int rval;
130 
131 	rval = sigaction(signum, act, oldact);
132 
133 	if (rval == -1) {
134 		tst_brk_(file, lineno, TBROK | TERRNO,
135 			"sigaction(%s (%d), %p, %p) failed",
136 			tst_strsig(signum), signum, act, oldact);
137 	}
138 
139 	return rval;
140 }
141 
safe_getgrnam(const char * file,const int lineno,const char * name)142 struct group *safe_getgrnam(const char *file, const int lineno,
143 			    const char *name)
144 {
145 	struct group *rval;
146 
147 	errno = 0;
148 	rval = getgrnam(name);
149 	if (rval == NULL) {
150 		tst_brk_(file, lineno, TBROK | TERRNO,
151 			"getgrnam(%s) failed", name);
152 	}
153 
154 	return rval;
155 }
156 
safe_getgrnam_fallback(const char * file,const int lineno,const char * name,const char * fallback)157 struct group *safe_getgrnam_fallback(const char *file, const int lineno,
158 				     const char *name, const char *fallback)
159 {
160 	struct group *rval;
161 
162 	errno = 0;
163 	rval = getgrnam(name);
164 	if (rval == NULL) {
165 		tst_res_(file, lineno, TINFO,
166 			 "getgrnam(%s) failed - try fallback %s",
167 			 name, fallback);
168 		rval = safe_getgrnam(file, lineno, fallback);
169 	}
170 
171 	return rval;
172 }
173 
safe_getgrgid(const char * file,const int lineno,gid_t gid)174 struct group *safe_getgrgid(const char *file, const int lineno, gid_t gid)
175 {
176 	struct group *rval;
177 
178 	errno = 0;
179 	rval = getgrgid(gid);
180 	if (rval == NULL) {
181 		tst_brk_(file, lineno, TBROK | TERRNO,
182 			"getgrgid(%li) failed", (long)gid);
183 	}
184 
185 	return rval;
186 }
187 
safe_chroot(const char * file,const int lineno,const char * path)188 int safe_chroot(const char *file, const int lineno, const char *path)
189 {
190 	int rval;
191 
192 	rval = chroot(path);
193 	if (rval == -1) {
194 		tst_brk_(file, lineno, TBROK | TERRNO,
195 			 "chroot(%s) failed", path);
196 	}
197 
198 	return rval;
199 }
200