xref: /linux/tools/lib/api/fs/fs.c (revision 507e190946297c34a27d9366b0661d5e506fdd03)
1 #include <ctype.h>
2 #include <errno.h>
3 #include <limits.h>
4 #include <stdbool.h>
5 #include <stdio.h>
6 #include <stdlib.h>
7 #include <string.h>
8 #include <sys/vfs.h>
9 #include <sys/types.h>
10 #include <sys/stat.h>
11 #include <fcntl.h>
12 #include <unistd.h>
13 #include <sys/mount.h>
14 
15 #include "fs.h"
16 #include "debug-internal.h"
17 
18 #define _STR(x) #x
19 #define STR(x) _STR(x)
20 
21 #ifndef SYSFS_MAGIC
22 #define SYSFS_MAGIC            0x62656572
23 #endif
24 
25 #ifndef PROC_SUPER_MAGIC
26 #define PROC_SUPER_MAGIC       0x9fa0
27 #endif
28 
29 #ifndef DEBUGFS_MAGIC
30 #define DEBUGFS_MAGIC          0x64626720
31 #endif
32 
33 #ifndef TRACEFS_MAGIC
34 #define TRACEFS_MAGIC          0x74726163
35 #endif
36 
37 #ifndef HUGETLBFS_MAGIC
38 #define HUGETLBFS_MAGIC        0x958458f6
39 #endif
40 
41 #ifndef BPF_FS_MAGIC
42 #define BPF_FS_MAGIC           0xcafe4a11
43 #endif
44 
45 static const char * const sysfs__fs_known_mountpoints[] = {
46 	"/sys",
47 	0,
48 };
49 
50 static const char * const procfs__known_mountpoints[] = {
51 	"/proc",
52 	0,
53 };
54 
55 #ifndef DEBUGFS_DEFAULT_PATH
56 #define DEBUGFS_DEFAULT_PATH "/sys/kernel/debug"
57 #endif
58 
59 static const char * const debugfs__known_mountpoints[] = {
60 	DEBUGFS_DEFAULT_PATH,
61 	"/debug",
62 	0,
63 };
64 
65 
66 #ifndef TRACEFS_DEFAULT_PATH
67 #define TRACEFS_DEFAULT_PATH "/sys/kernel/tracing"
68 #endif
69 
70 static const char * const tracefs__known_mountpoints[] = {
71 	TRACEFS_DEFAULT_PATH,
72 	"/sys/kernel/debug/tracing",
73 	"/tracing",
74 	"/trace",
75 	0,
76 };
77 
78 static const char * const hugetlbfs__known_mountpoints[] = {
79 	0,
80 };
81 
82 static const char * const bpf_fs__known_mountpoints[] = {
83 	"/sys/fs/bpf",
84 	0,
85 };
86 
87 struct fs {
88 	const char		*name;
89 	const char * const	*mounts;
90 	char			 path[PATH_MAX];
91 	bool			 found;
92 	long			 magic;
93 };
94 
95 enum {
96 	FS__SYSFS   = 0,
97 	FS__PROCFS  = 1,
98 	FS__DEBUGFS = 2,
99 	FS__TRACEFS = 3,
100 	FS__HUGETLBFS = 4,
101 	FS__BPF_FS = 5,
102 };
103 
104 #ifndef TRACEFS_MAGIC
105 #define TRACEFS_MAGIC 0x74726163
106 #endif
107 
108 static struct fs fs__entries[] = {
109 	[FS__SYSFS] = {
110 		.name	= "sysfs",
111 		.mounts	= sysfs__fs_known_mountpoints,
112 		.magic	= SYSFS_MAGIC,
113 	},
114 	[FS__PROCFS] = {
115 		.name	= "proc",
116 		.mounts	= procfs__known_mountpoints,
117 		.magic	= PROC_SUPER_MAGIC,
118 	},
119 	[FS__DEBUGFS] = {
120 		.name	= "debugfs",
121 		.mounts	= debugfs__known_mountpoints,
122 		.magic	= DEBUGFS_MAGIC,
123 	},
124 	[FS__TRACEFS] = {
125 		.name	= "tracefs",
126 		.mounts	= tracefs__known_mountpoints,
127 		.magic	= TRACEFS_MAGIC,
128 	},
129 	[FS__HUGETLBFS] = {
130 		.name	= "hugetlbfs",
131 		.mounts = hugetlbfs__known_mountpoints,
132 		.magic	= HUGETLBFS_MAGIC,
133 	},
134 	[FS__BPF_FS] = {
135 		.name	= "bpf",
136 		.mounts = bpf_fs__known_mountpoints,
137 		.magic	= BPF_FS_MAGIC,
138 	},
139 };
140 
141 static bool fs__read_mounts(struct fs *fs)
142 {
143 	bool found = false;
144 	char type[100];
145 	FILE *fp;
146 
147 	fp = fopen("/proc/mounts", "r");
148 	if (fp == NULL)
149 		return NULL;
150 
151 	while (!found &&
152 	       fscanf(fp, "%*s %" STR(PATH_MAX) "s %99s %*s %*d %*d\n",
153 		      fs->path, type) == 2) {
154 
155 		if (strcmp(type, fs->name) == 0)
156 			found = true;
157 	}
158 
159 	fclose(fp);
160 	return fs->found = found;
161 }
162 
163 static int fs__valid_mount(const char *fs, long magic)
164 {
165 	struct statfs st_fs;
166 
167 	if (statfs(fs, &st_fs) < 0)
168 		return -ENOENT;
169 	else if ((long)st_fs.f_type != magic)
170 		return -ENOENT;
171 
172 	return 0;
173 }
174 
175 static bool fs__check_mounts(struct fs *fs)
176 {
177 	const char * const *ptr;
178 
179 	ptr = fs->mounts;
180 	while (*ptr) {
181 		if (fs__valid_mount(*ptr, fs->magic) == 0) {
182 			fs->found = true;
183 			strcpy(fs->path, *ptr);
184 			return true;
185 		}
186 		ptr++;
187 	}
188 
189 	return false;
190 }
191 
192 static void mem_toupper(char *f, size_t len)
193 {
194 	while (len) {
195 		*f = toupper(*f);
196 		f++;
197 		len--;
198 	}
199 }
200 
201 /*
202  * Check for "NAME_PATH" environment variable to override fs location (for
203  * testing). This matches the recommendation in Documentation/sysfs-rules.txt
204  * for SYSFS_PATH.
205  */
206 static bool fs__env_override(struct fs *fs)
207 {
208 	char *override_path;
209 	size_t name_len = strlen(fs->name);
210 	/* name + "_PATH" + '\0' */
211 	char upper_name[name_len + 5 + 1];
212 	memcpy(upper_name, fs->name, name_len);
213 	mem_toupper(upper_name, name_len);
214 	strcpy(&upper_name[name_len], "_PATH");
215 
216 	override_path = getenv(upper_name);
217 	if (!override_path)
218 		return false;
219 
220 	fs->found = true;
221 	strncpy(fs->path, override_path, sizeof(fs->path));
222 	return true;
223 }
224 
225 static const char *fs__get_mountpoint(struct fs *fs)
226 {
227 	if (fs__env_override(fs))
228 		return fs->path;
229 
230 	if (fs__check_mounts(fs))
231 		return fs->path;
232 
233 	if (fs__read_mounts(fs))
234 		return fs->path;
235 
236 	return NULL;
237 }
238 
239 static const char *fs__mountpoint(int idx)
240 {
241 	struct fs *fs = &fs__entries[idx];
242 
243 	if (fs->found)
244 		return (const char *)fs->path;
245 
246 	return fs__get_mountpoint(fs);
247 }
248 
249 static const char *mount_overload(struct fs *fs)
250 {
251 	size_t name_len = strlen(fs->name);
252 	/* "PERF_" + name + "_ENVIRONMENT" + '\0' */
253 	char upper_name[5 + name_len + 12 + 1];
254 
255 	snprintf(upper_name, name_len, "PERF_%s_ENVIRONMENT", fs->name);
256 	mem_toupper(upper_name, name_len);
257 
258 	return getenv(upper_name) ?: *fs->mounts;
259 }
260 
261 static const char *fs__mount(int idx)
262 {
263 	struct fs *fs = &fs__entries[idx];
264 	const char *mountpoint;
265 
266 	if (fs__mountpoint(idx))
267 		return (const char *)fs->path;
268 
269 	mountpoint = mount_overload(fs);
270 
271 	if (mount(NULL, mountpoint, fs->name, 0, NULL) < 0)
272 		return NULL;
273 
274 	return fs__check_mounts(fs) ? fs->path : NULL;
275 }
276 
277 #define FS(name, idx)				\
278 const char *name##__mountpoint(void)		\
279 {						\
280 	return fs__mountpoint(idx);		\
281 }						\
282 						\
283 const char *name##__mount(void)			\
284 {						\
285 	return fs__mount(idx);			\
286 }						\
287 						\
288 bool name##__configured(void)			\
289 {						\
290 	return name##__mountpoint() != NULL;	\
291 }
292 
293 FS(sysfs,   FS__SYSFS);
294 FS(procfs,  FS__PROCFS);
295 FS(debugfs, FS__DEBUGFS);
296 FS(tracefs, FS__TRACEFS);
297 FS(hugetlbfs, FS__HUGETLBFS);
298 FS(bpf_fs, FS__BPF_FS);
299 
300 int filename__read_int(const char *filename, int *value)
301 {
302 	char line[64];
303 	int fd = open(filename, O_RDONLY), err = -1;
304 
305 	if (fd < 0)
306 		return -1;
307 
308 	if (read(fd, line, sizeof(line)) > 0) {
309 		*value = atoi(line);
310 		err = 0;
311 	}
312 
313 	close(fd);
314 	return err;
315 }
316 
317 /*
318  * Parses @value out of @filename with strtoull.
319  * By using 0 for base, the strtoull detects the
320  * base automatically (see man strtoull).
321  */
322 int filename__read_ull(const char *filename, unsigned long long *value)
323 {
324 	char line[64];
325 	int fd = open(filename, O_RDONLY), err = -1;
326 
327 	if (fd < 0)
328 		return -1;
329 
330 	if (read(fd, line, sizeof(line)) > 0) {
331 		*value = strtoull(line, NULL, 0);
332 		if (*value != ULLONG_MAX)
333 			err = 0;
334 	}
335 
336 	close(fd);
337 	return err;
338 }
339 
340 #define STRERR_BUFSIZE  128     /* For the buffer size of strerror_r */
341 
342 int filename__read_str(const char *filename, char **buf, size_t *sizep)
343 {
344 	size_t size = 0, alloc_size = 0;
345 	void *bf = NULL, *nbf;
346 	int fd, n, err = 0;
347 	char sbuf[STRERR_BUFSIZE];
348 
349 	fd = open(filename, O_RDONLY);
350 	if (fd < 0)
351 		return -errno;
352 
353 	do {
354 		if (size == alloc_size) {
355 			alloc_size += BUFSIZ;
356 			nbf = realloc(bf, alloc_size);
357 			if (!nbf) {
358 				err = -ENOMEM;
359 				break;
360 			}
361 
362 			bf = nbf;
363 		}
364 
365 		n = read(fd, bf + size, alloc_size - size);
366 		if (n < 0) {
367 			if (size) {
368 				pr_warning("read failed %d: %s\n", errno,
369 					 strerror_r(errno, sbuf, sizeof(sbuf)));
370 				err = 0;
371 			} else
372 				err = -errno;
373 
374 			break;
375 		}
376 
377 		size += n;
378 	} while (n > 0);
379 
380 	if (!err) {
381 		*sizep = size;
382 		*buf   = bf;
383 	} else
384 		free(bf);
385 
386 	close(fd);
387 	return err;
388 }
389 
390 int filename__write_int(const char *filename, int value)
391 {
392 	int fd = open(filename, O_WRONLY), err = -1;
393 	char buf[64];
394 
395 	if (fd < 0)
396 		return err;
397 
398 	sprintf(buf, "%d", value);
399 	if (write(fd, buf, sizeof(buf)) == sizeof(buf))
400 		err = 0;
401 
402 	close(fd);
403 	return err;
404 }
405 
406 int procfs__read_str(const char *entry, char **buf, size_t *sizep)
407 {
408 	char path[PATH_MAX];
409 	const char *procfs = procfs__mountpoint();
410 
411 	if (!procfs)
412 		return -1;
413 
414 	snprintf(path, sizeof(path), "%s/%s", procfs, entry);
415 
416 	return filename__read_str(path, buf, sizep);
417 }
418 
419 int sysfs__read_ull(const char *entry, unsigned long long *value)
420 {
421 	char path[PATH_MAX];
422 	const char *sysfs = sysfs__mountpoint();
423 
424 	if (!sysfs)
425 		return -1;
426 
427 	snprintf(path, sizeof(path), "%s/%s", sysfs, entry);
428 
429 	return filename__read_ull(path, value);
430 }
431 
432 int sysfs__read_int(const char *entry, int *value)
433 {
434 	char path[PATH_MAX];
435 	const char *sysfs = sysfs__mountpoint();
436 
437 	if (!sysfs)
438 		return -1;
439 
440 	snprintf(path, sizeof(path), "%s/%s", sysfs, entry);
441 
442 	return filename__read_int(path, value);
443 }
444 
445 int sysfs__read_str(const char *entry, char **buf, size_t *sizep)
446 {
447 	char path[PATH_MAX];
448 	const char *sysfs = sysfs__mountpoint();
449 
450 	if (!sysfs)
451 		return -1;
452 
453 	snprintf(path, sizeof(path), "%s/%s", sysfs, entry);
454 
455 	return filename__read_str(path, buf, sizep);
456 }
457 
458 int sysfs__read_bool(const char *entry, bool *value)
459 {
460 	char *buf;
461 	size_t size;
462 	int ret;
463 
464 	ret = sysfs__read_str(entry, &buf, &size);
465 	if (ret < 0)
466 		return ret;
467 
468 	switch (buf[0]) {
469 	case '1':
470 	case 'y':
471 	case 'Y':
472 		*value = true;
473 		break;
474 	case '0':
475 	case 'n':
476 	case 'N':
477 		*value = false;
478 		break;
479 	default:
480 		ret = -1;
481 	}
482 
483 	free(buf);
484 
485 	return ret;
486 }
487 int sysctl__read_int(const char *sysctl, int *value)
488 {
489 	char path[PATH_MAX];
490 	const char *procfs = procfs__mountpoint();
491 
492 	if (!procfs)
493 		return -1;
494 
495 	snprintf(path, sizeof(path), "%s/sys/%s", procfs, sysctl);
496 
497 	return filename__read_int(path, value);
498 }
499 
500 int sysfs__write_int(const char *entry, int value)
501 {
502 	char path[PATH_MAX];
503 	const char *sysfs = sysfs__mountpoint();
504 
505 	if (!sysfs)
506 		return -1;
507 
508 	if (snprintf(path, sizeof(path), "%s/%s", sysfs, entry) >= PATH_MAX)
509 		return -1;
510 
511 	return filename__write_int(path, value);
512 }
513