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