18c2ecf20Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-or-later
28c2ecf20Sopenharmony_ci/* fd-based mount test.
38c2ecf20Sopenharmony_ci *
48c2ecf20Sopenharmony_ci * Copyright (C) 2017 Red Hat, Inc. All Rights Reserved.
58c2ecf20Sopenharmony_ci * Written by David Howells (dhowells@redhat.com)
68c2ecf20Sopenharmony_ci */
78c2ecf20Sopenharmony_ci
88c2ecf20Sopenharmony_ci#include <stdio.h>
98c2ecf20Sopenharmony_ci#include <stdlib.h>
108c2ecf20Sopenharmony_ci#include <unistd.h>
118c2ecf20Sopenharmony_ci#include <errno.h>
128c2ecf20Sopenharmony_ci#include <fcntl.h>
138c2ecf20Sopenharmony_ci#include <sys/prctl.h>
148c2ecf20Sopenharmony_ci#include <sys/wait.h>
158c2ecf20Sopenharmony_ci#include <linux/mount.h>
168c2ecf20Sopenharmony_ci#include <linux/unistd.h>
178c2ecf20Sopenharmony_ci
188c2ecf20Sopenharmony_ci#define E(x) do { if ((x) == -1) { perror(#x); exit(1); } } while(0)
198c2ecf20Sopenharmony_ci
208c2ecf20Sopenharmony_cistatic void check_messages(int fd)
218c2ecf20Sopenharmony_ci{
228c2ecf20Sopenharmony_ci	char buf[4096];
238c2ecf20Sopenharmony_ci	int err, n;
248c2ecf20Sopenharmony_ci
258c2ecf20Sopenharmony_ci	err = errno;
268c2ecf20Sopenharmony_ci
278c2ecf20Sopenharmony_ci	for (;;) {
288c2ecf20Sopenharmony_ci		n = read(fd, buf, sizeof(buf));
298c2ecf20Sopenharmony_ci		if (n < 0)
308c2ecf20Sopenharmony_ci			break;
318c2ecf20Sopenharmony_ci		n -= 2;
328c2ecf20Sopenharmony_ci
338c2ecf20Sopenharmony_ci		switch (buf[0]) {
348c2ecf20Sopenharmony_ci		case 'e':
358c2ecf20Sopenharmony_ci			fprintf(stderr, "Error: %*.*s\n", n, n, buf + 2);
368c2ecf20Sopenharmony_ci			break;
378c2ecf20Sopenharmony_ci		case 'w':
388c2ecf20Sopenharmony_ci			fprintf(stderr, "Warning: %*.*s\n", n, n, buf + 2);
398c2ecf20Sopenharmony_ci			break;
408c2ecf20Sopenharmony_ci		case 'i':
418c2ecf20Sopenharmony_ci			fprintf(stderr, "Info: %*.*s\n", n, n, buf + 2);
428c2ecf20Sopenharmony_ci			break;
438c2ecf20Sopenharmony_ci		}
448c2ecf20Sopenharmony_ci	}
458c2ecf20Sopenharmony_ci
468c2ecf20Sopenharmony_ci	errno = err;
478c2ecf20Sopenharmony_ci}
488c2ecf20Sopenharmony_ci
498c2ecf20Sopenharmony_cistatic __attribute__((noreturn))
508c2ecf20Sopenharmony_civoid mount_error(int fd, const char *s)
518c2ecf20Sopenharmony_ci{
528c2ecf20Sopenharmony_ci	check_messages(fd);
538c2ecf20Sopenharmony_ci	fprintf(stderr, "%s: %m\n", s);
548c2ecf20Sopenharmony_ci	exit(1);
558c2ecf20Sopenharmony_ci}
568c2ecf20Sopenharmony_ci
578c2ecf20Sopenharmony_ci/* Hope -1 isn't a syscall */
588c2ecf20Sopenharmony_ci#ifndef __NR_fsopen
598c2ecf20Sopenharmony_ci#define __NR_fsopen -1
608c2ecf20Sopenharmony_ci#endif
618c2ecf20Sopenharmony_ci#ifndef __NR_fsmount
628c2ecf20Sopenharmony_ci#define __NR_fsmount -1
638c2ecf20Sopenharmony_ci#endif
648c2ecf20Sopenharmony_ci#ifndef __NR_fsconfig
658c2ecf20Sopenharmony_ci#define __NR_fsconfig -1
668c2ecf20Sopenharmony_ci#endif
678c2ecf20Sopenharmony_ci#ifndef __NR_move_mount
688c2ecf20Sopenharmony_ci#define __NR_move_mount -1
698c2ecf20Sopenharmony_ci#endif
708c2ecf20Sopenharmony_ci
718c2ecf20Sopenharmony_ci
728c2ecf20Sopenharmony_cistatic inline int fsopen(const char *fs_name, unsigned int flags)
738c2ecf20Sopenharmony_ci{
748c2ecf20Sopenharmony_ci	return syscall(__NR_fsopen, fs_name, flags);
758c2ecf20Sopenharmony_ci}
768c2ecf20Sopenharmony_ci
778c2ecf20Sopenharmony_cistatic inline int fsmount(int fsfd, unsigned int flags, unsigned int ms_flags)
788c2ecf20Sopenharmony_ci{
798c2ecf20Sopenharmony_ci	return syscall(__NR_fsmount, fsfd, flags, ms_flags);
808c2ecf20Sopenharmony_ci}
818c2ecf20Sopenharmony_ci
828c2ecf20Sopenharmony_cistatic inline int fsconfig(int fsfd, unsigned int cmd,
838c2ecf20Sopenharmony_ci			   const char *key, const void *val, int aux)
848c2ecf20Sopenharmony_ci{
858c2ecf20Sopenharmony_ci	return syscall(__NR_fsconfig, fsfd, cmd, key, val, aux);
868c2ecf20Sopenharmony_ci}
878c2ecf20Sopenharmony_ci
888c2ecf20Sopenharmony_cistatic inline int move_mount(int from_dfd, const char *from_pathname,
898c2ecf20Sopenharmony_ci			     int to_dfd, const char *to_pathname,
908c2ecf20Sopenharmony_ci			     unsigned int flags)
918c2ecf20Sopenharmony_ci{
928c2ecf20Sopenharmony_ci	return syscall(__NR_move_mount,
938c2ecf20Sopenharmony_ci		       from_dfd, from_pathname,
948c2ecf20Sopenharmony_ci		       to_dfd, to_pathname, flags);
958c2ecf20Sopenharmony_ci}
968c2ecf20Sopenharmony_ci
978c2ecf20Sopenharmony_ci#define E_fsconfig(fd, cmd, key, val, aux)				\
988c2ecf20Sopenharmony_ci	do {								\
998c2ecf20Sopenharmony_ci		if (fsconfig(fd, cmd, key, val, aux) == -1)		\
1008c2ecf20Sopenharmony_ci			mount_error(fd, key ?: "create");		\
1018c2ecf20Sopenharmony_ci	} while (0)
1028c2ecf20Sopenharmony_ci
1038c2ecf20Sopenharmony_ciint main(int argc, char *argv[])
1048c2ecf20Sopenharmony_ci{
1058c2ecf20Sopenharmony_ci	int fsfd, mfd;
1068c2ecf20Sopenharmony_ci
1078c2ecf20Sopenharmony_ci	/* Mount a publically available AFS filesystem */
1088c2ecf20Sopenharmony_ci	fsfd = fsopen("afs", 0);
1098c2ecf20Sopenharmony_ci	if (fsfd == -1) {
1108c2ecf20Sopenharmony_ci		perror("fsopen");
1118c2ecf20Sopenharmony_ci		exit(1);
1128c2ecf20Sopenharmony_ci	}
1138c2ecf20Sopenharmony_ci
1148c2ecf20Sopenharmony_ci	E_fsconfig(fsfd, FSCONFIG_SET_STRING, "source", "#grand.central.org:root.cell.", 0);
1158c2ecf20Sopenharmony_ci	E_fsconfig(fsfd, FSCONFIG_CMD_CREATE, NULL, NULL, 0);
1168c2ecf20Sopenharmony_ci
1178c2ecf20Sopenharmony_ci	mfd = fsmount(fsfd, 0, MOUNT_ATTR_RDONLY);
1188c2ecf20Sopenharmony_ci	if (mfd < 0)
1198c2ecf20Sopenharmony_ci		mount_error(fsfd, "fsmount");
1208c2ecf20Sopenharmony_ci	E(close(fsfd));
1218c2ecf20Sopenharmony_ci
1228c2ecf20Sopenharmony_ci	if (move_mount(mfd, "", AT_FDCWD, "/mnt", MOVE_MOUNT_F_EMPTY_PATH) < 0) {
1238c2ecf20Sopenharmony_ci		perror("move_mount");
1248c2ecf20Sopenharmony_ci		exit(1);
1258c2ecf20Sopenharmony_ci	}
1268c2ecf20Sopenharmony_ci
1278c2ecf20Sopenharmony_ci	E(close(mfd));
1288c2ecf20Sopenharmony_ci	exit(0);
1298c2ecf20Sopenharmony_ci}
130