// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (C) 2014 Cyril Hrubis chrubis@suse.cz
 */

#define _GNU_SOURCE
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <sys/mount.h>
#include <mntent.h>
#include <errno.h>
#include <unistd.h>
#include <stdlib.h>
#include <linux/loop.h>
#include <stdint.h>
#include <inttypes.h>
#include <sys/sysmacros.h>
#include <linux/btrfs.h>
#include <linux/limits.h>
#include <sys/statfs.h>
#include "lapi/syscalls.h"
#include "test.h"
#include "safe_macros.h"
#include "tst_device.h"

#ifndef LOOP_CTL_GET_FREE
# define LOOP_CTL_GET_FREE 0x4C82
#endif

#define LOOP_CONTROL_FILE "/dev/loop-control"

#define DEV_FILE "test_dev.img"
#define DEV_SIZE_MB 300u
#define UUID_STR_SZ 37
#define UUID_FMT "%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x"

static char dev_path[PATH_MAX];
static int device_acquired;
static unsigned long prev_dev_sec_write;

static const char * const dev_loop_variants[] = {
	"/dev/loop%i",
	"/dev/loop/%i",
	"/dev/block/loop%i"
};

static const char * const dev_variants[] = {
	"/dev/%s",
	"/dev/block/%s"
};

static int set_dev_loop_path(int dev, char *path, size_t path_len)
{
	unsigned int i;
	struct stat st;

	for (i = 0; i < ARRAY_SIZE(dev_loop_variants); i++) {
		snprintf(path, path_len, dev_loop_variants[i], dev);

		if (stat(path, &st) == 0 && S_ISBLK(st.st_mode))
			return 0;
	}

	return 1;
}

static int set_dev_path(char *dev, char *path, size_t path_len)
{
	unsigned int i;
	struct stat st;

	for (i = 0; i < ARRAY_SIZE(dev_variants); i++) {
		snprintf(path, path_len, dev_variants[i], dev);

		if (stat(path, &st) == 0 && S_ISBLK(st.st_mode))
			return 0;
	}

	return 1;
}

int tst_find_free_loopdev(char *path, size_t path_len)
{
	int ctl_fd, dev_fd, rc, i;
	struct loop_info loopinfo;
	char buf[PATH_MAX];

	/* since Linux 3.1 */
	ctl_fd = open(LOOP_CONTROL_FILE, O_RDWR);

	if (ctl_fd > 0) {
		rc = ioctl(ctl_fd, LOOP_CTL_GET_FREE);
		close(ctl_fd);
		if (rc >= 0) {
			if (path && set_dev_loop_path(rc, path, path_len))
				tst_brkm(TBROK, NULL, "Could not stat loop device %i", rc);
			tst_resm(TINFO, "Found free device %d '%s'",
				rc, path ?: "");
			return rc;
		}
		tst_resm(TINFO, "Couldn't find free loop device");
		return -1;
	}

	switch (errno) {
	case ENOENT:
	break;
	case EACCES:
		tst_resm(TINFO | TERRNO,
			 "Not allowed to open " LOOP_CONTROL_FILE ". "
			 "Are you root?");
	break;
	default:
		tst_resm(TBROK | TERRNO, "Failed to open " LOOP_CONTROL_FILE);
	}

	/*
	 * Older way is to iterate over /dev/loop%i and /dev/loop/%i and try
	 * LOOP_GET_STATUS ioctl() which fails for free loop devices.
	 */
	for (i = 0; i < 256; i++) {

		if (set_dev_loop_path(i, buf, sizeof(buf)))
			continue;

		dev_fd = open(buf, O_RDONLY);

		if (dev_fd < 0)
			continue;

		if (ioctl(dev_fd, LOOP_GET_STATUS, &loopinfo) == 0) {
			tst_resm(TINFO, "Device '%s' in use", buf);
		} else {
			if (errno != ENXIO)
				continue;
			tst_resm(TINFO, "Found free device '%s'", buf);
			close(dev_fd);
			if (path != NULL) {
				strncpy(path, buf, path_len);
				path[path_len-1] = '\0';
			}
			return i;
		}

		close(dev_fd);
	}

	tst_resm(TINFO, "No free devices found");

	return -1;
}

int tst_attach_device(const char *dev, const char *file)
{
	int dev_fd, file_fd;
	struct loop_info loopinfo;

	dev_fd = open(dev, O_RDWR);
	if (dev_fd < 0) {
		tst_resm(TWARN | TERRNO, "open('%s', O_RDWR) failed", dev);
		return 1;
	}

	file_fd = open(file, O_RDWR);
	if (file_fd < 0) {
		tst_resm(TWARN | TERRNO, "open('%s', O_RDWR) failed", file);
		close(dev_fd);
		return 1;
	}

	if (ioctl(dev_fd, LOOP_SET_FD, file_fd) < 0) {
		close(dev_fd);
		close(file_fd);
		tst_resm(TWARN | TERRNO, "ioctl(%s, LOOP_SET_FD, %s) failed",
			 dev, file);
		return 1;
	}

	/* Old mkfs.btrfs use LOOP_GET_STATUS instead of backing_file to get
	 * associated filename, so we need to set up the device by calling
	 * LOOP_SET_FD and LOOP_SET_STATUS.
	 */
	memset(&loopinfo, 0, sizeof(loopinfo));
	strcpy(loopinfo.lo_name, file);

	if (ioctl(dev_fd, LOOP_SET_STATUS, &loopinfo)) {
		close(dev_fd);
		close(file_fd);
		tst_resm(TWARN | TERRNO,
			 "ioctl(%s, LOOP_SET_STATUS, %s) failed", dev, file);
		return 1;
	}

	close(dev_fd);
	close(file_fd);
	return 0;
}

uint64_t tst_get_device_size(const char *dev_path)
{
	int fd;
	uint64_t size;
	struct stat st;

	if (!dev_path)
		tst_brkm(TBROK, NULL, "No block device path");

	if (stat(dev_path, &st)) {
		tst_resm(TWARN | TERRNO, "stat() failed");
		return -1;
	}

	if (!S_ISBLK(st.st_mode)) {
		tst_resm(TWARN, "%s is not a block device", dev_path);
		return -1;
	}

	fd = open(dev_path, O_RDONLY);
	if (fd < 0) {
		tst_resm(TWARN | TERRNO,
				"open(%s, O_RDONLY) failed", dev_path);
		return -1;
	}

	if (ioctl(fd, BLKGETSIZE64, &size)) {
		tst_resm(TWARN | TERRNO,
				"ioctl(fd, BLKGETSIZE64, ...) failed");
		close(fd);
		return -1;
	}

	if (close(fd)) {
		tst_resm(TWARN | TERRNO,
				"close(fd) failed");
		return -1;
	}

	return size/1024/1024;
}

int tst_detach_device_by_fd(const char *dev, int dev_fd)
{
	int ret, i;

	/* keep trying to clear LOOPDEV until we get ENXIO, a quick succession
	 * of attach/detach might not give udev enough time to complete
	 *
	 * Since 18048c1af783 ("loop: Fix a race between loop detach and loop open")
	 * device is detached only after last close.
	 */
	for (i = 0; i < 40; i++) {
		ret = ioctl(dev_fd, LOOP_CLR_FD, 0);

		if (ret && (errno == ENXIO)) {
			SAFE_CLOSE(NULL, dev_fd);
			return 0;
		}

		if (ret && (errno != EBUSY)) {
			tst_resm(TWARN,
				 "ioctl(%s, LOOP_CLR_FD, 0) unexpectedly failed with: %s",
				 dev, tst_strerrno(errno));
			SAFE_CLOSE(NULL, dev_fd);
			return 1;
		}

		usleep(50000);
	}

	tst_resm(TWARN,
		"ioctl(%s, LOOP_CLR_FD, 0) no ENXIO for too long", dev);
	SAFE_CLOSE(NULL, dev_fd);
	return 1;
}

int tst_detach_device(const char *dev)
{
	int dev_fd, ret;

	dev_fd = open(dev, O_RDONLY);
	if (dev_fd < 0) {
		tst_resm(TWARN | TERRNO, "open(%s) failed", dev);
		return 1;
	}

	ret = tst_detach_device_by_fd(dev, dev_fd);
	return ret;
}

int tst_dev_sync(int fd)
{
	return syscall(__NR_syncfs, fd);
}

const char *tst_acquire_loop_device(unsigned int size, const char *filename)
{
	unsigned int acq_dev_size = size ? size : DEV_SIZE_MB;

	if (tst_prealloc_file(filename, 1024 * 1024, acq_dev_size)) {
		tst_resm(TWARN | TERRNO, "Failed to create %s", filename);
		return NULL;
	}

	if (tst_find_free_loopdev(dev_path, sizeof(dev_path)) == -1)
		return NULL;

	if (tst_attach_device(dev_path, filename))
		return NULL;

	return dev_path;
}

const char *tst_acquire_device__(unsigned int size)
{
	const char *dev;
	unsigned int acq_dev_size;
	uint64_t ltp_dev_size;

	acq_dev_size = size ? size : DEV_SIZE_MB;

	dev = getenv("LTP_DEV");

	if (dev) {
		tst_resm(TINFO, "Using test device LTP_DEV='%s'", dev);

		ltp_dev_size = tst_get_device_size(dev);

		if (acq_dev_size <= ltp_dev_size)
			return dev;

		tst_resm(TINFO, "Skipping $LTP_DEV size %"PRIu64"MB, requested size %uMB",
				ltp_dev_size, acq_dev_size);
	}

	dev = tst_acquire_loop_device(acq_dev_size, DEV_FILE);

	if (dev)
		device_acquired = 1;

	return dev;
}

const char *tst_acquire_device_(void (cleanup_fn)(void), unsigned int size)
{
	const char *device;

	if (device_acquired) {
		tst_brkm(TBROK, cleanup_fn, "Device already acquired");
		return NULL;
	}

	if (!tst_tmpdir_created()) {
		tst_brkm(TBROK, cleanup_fn,
			 "Cannot acquire device without tmpdir() created");
		return NULL;
	}

	device = tst_acquire_device__(size);

	if (!device) {
		tst_brkm(TBROK, cleanup_fn, "Failed to acquire device");
		return NULL;
	}

	return device;
}

int tst_release_device(const char *dev)
{
	int ret;

	if (!device_acquired)
		return 0;

	/*
	 * Loop device was created -> we need to detach it.
	 *
	 * The file image is deleted in tst_rmdir();
	 */
	ret = tst_detach_device(dev);

	device_acquired = 0;

	return ret;
}

int tst_clear_device(const char *dev)
{
	if (tst_fill_file(dev, 0, 1024, 512)) {
		tst_resm(TWARN, "Failed to clear 512k block on %s", dev);
		return 1;
	}

	return 0;
}

int tst_umount(const char *path)
{
	int err, ret, i;

	for (i = 0; i < 50; i++) {
		ret = umount(path);
		err = errno;

		if (!ret)
			return 0;

		if (err != EBUSY) {
			tst_resm(TWARN, "umount('%s') failed with %s",
				 path, tst_strerrno(err));
			errno = err;
			return ret;
		}

		tst_resm(TINFO, "umount('%s') failed with %s, try %2i...",
			 path, tst_strerrno(err), i+1);

		if (i == 0) {
			tst_resm(TINFO, "Likely gvfsd-trash is probing newly "
				 "mounted fs, kill it to speed up tests.");
		}

		usleep(100000);
	}

	tst_resm(TWARN, "Failed to umount('%s') after 50 retries", path);
	errno = err;
	return -1;
}

int tst_mount_has_opt(const char *path, const char *opt)
{
	char line[PATH_MAX];
	char abspath[PATH_MAX];
	FILE *file;
	int ret = 0;

	if (!realpath(path, abspath)) {
		tst_resm(TWARN | TERRNO, "realpath(%s) failed", path);
		return 0;
	}

	file = SAFE_FOPEN(NULL, "/proc/mounts", "r");

	while (fgets(line, sizeof(line), file)) {
		char mount_point[PATH_MAX], options[PATH_MAX];

		if (sscanf(line, "%*s %s %*s %s", mount_point, options) < 2)
			continue;

		if (strcmp(mount_point, abspath) != 0)
			continue;

		if (!opt) {
			ret = 1;
			break;
		}

		char *tok = strtok(options, ",");
		while (tok) {
			if (strcmp(tok, opt) == 0) {
				ret = 1;
				break;
			}
			tok = strtok(NULL, ",");
		}
		if (ret)
			break;
	}

	SAFE_FCLOSE(NULL, file);
	return ret;
}

int tst_is_mounted(const char *path)
{
	int ret = tst_mount_has_opt(path, NULL);
	if (!ret)
		tst_resm(TINFO, "No device is mounted at %s", path);

	return ret;
}

int tst_is_mounted_ro(const char *path)
{
	return tst_mount_has_opt(path, "ro");
}

int tst_is_mounted_rw(const char *path)
{
	return tst_mount_has_opt(path, "rw");
}

int tst_is_mounted_at_tmpdir(const char *path)
{
	char cdir[PATH_MAX], mpath[PATH_MAX];
	int ret;

	if (!getcwd(cdir, PATH_MAX)) {
		tst_resm(TWARN | TERRNO, "Failed to find current directory");
		return 0;
	}

	ret = snprintf(mpath, PATH_MAX, "%s/%s", cdir, path);
	if (ret < 0 || ret >= PATH_MAX) {
		tst_resm(TWARN | TERRNO,
			 "snprintf() should have returned %d instead of %d",
			 PATH_MAX, ret);
		return 0;
	}

	return tst_is_mounted(mpath);
}

static int find_stat_file(const char *dev, char *path, size_t path_len)
{
	const char *devname = strrchr(dev, '/') + 1;

	snprintf(path, path_len, "/sys/block/%s/stat", devname);

	if (!access(path, F_OK))
		return 1;

	DIR *dir = SAFE_OPENDIR(NULL, "/sys/block/");
	struct dirent *ent;

	while ((ent = readdir(dir))) {
		snprintf(path, path_len, "/sys/block/%s/%s/stat", ent->d_name, devname);

		if (!access(path, F_OK)) {
			SAFE_CLOSEDIR(NULL, dir);
			return 1;
		}
	}

	SAFE_CLOSEDIR(NULL, dir);
	return 0;
}

unsigned long tst_dev_bytes_written(const char *dev)
{
	unsigned long dev_sec_write = 0, dev_bytes_written, io_ticks = 0;
	char dev_stat_path[PATH_MAX];

	if (!find_stat_file(dev, dev_stat_path, sizeof(dev_stat_path)))
		tst_brkm(TCONF, NULL, "Test device stat file: %s not found",
			 dev_stat_path);

	SAFE_FILE_SCANF(NULL, dev_stat_path,
			"%*s %*s %*s %*s %*s %*s %lu %*s %*s %lu",
			&dev_sec_write, &io_ticks);

	if (!io_ticks)
		tst_brkm(TCONF, NULL, "Test device stat file: %s broken",
			 dev_stat_path);

	dev_bytes_written = (dev_sec_write - prev_dev_sec_write) * 512;

	prev_dev_sec_write = dev_sec_write;

	return dev_bytes_written;
}

static void btrfs_get_uevent_path(char *tmp_path, char *uevent_path)
{
	int fd;
	struct btrfs_ioctl_fs_info_args args = {0};
	char btrfs_uuid_str[UUID_STR_SZ];
	struct dirent *d;
	char bdev_path[PATH_MAX];
	DIR *dir;

	tst_resm(TINFO, "Use BTRFS specific strategy");

	fd = SAFE_OPEN(NULL, tmp_path, O_DIRECTORY);
	if (!ioctl(fd, BTRFS_IOC_FS_INFO, &args)) {
		sprintf(btrfs_uuid_str,
			UUID_FMT,
			args.fsid[0], args.fsid[1],
			args.fsid[2], args.fsid[3],
			args.fsid[4], args.fsid[5],
			args.fsid[6], args.fsid[7],
			args.fsid[8], args.fsid[9],
			args.fsid[10], args.fsid[11],
			args.fsid[12], args.fsid[13],
			args.fsid[14], args.fsid[15]);
		sprintf(bdev_path,
			"/sys/fs/btrfs/%s/devices", btrfs_uuid_str);
	} else {
		tst_brkm(TBROK | TERRNO, NULL, "BTRFS ioctl on %s failed.", tmp_path);
	}
	SAFE_CLOSE(NULL, fd);

	dir = SAFE_OPENDIR(NULL, bdev_path);
	while ((d = SAFE_READDIR(NULL, dir))) {
		if (d->d_name[0] != '.')
			break;
	}

	uevent_path[0] = '\0';

	if (d)
		sprintf(uevent_path, "%s/%s/uevent", bdev_path, d->d_name);
	else
		tst_brkm(TBROK | TERRNO, NULL, "No backing device found while looking in %s", bdev_path);

	if (SAFE_READDIR(NULL, dir))
		tst_resm(TINFO, "Warning: used first of multiple backing device.");

	SAFE_CLOSEDIR(NULL, dir);
}

static char *overlay_mount_from_dev(dev_t dev)
{
	unsigned int dev_major, dev_minor, mnt_major, mnt_minor;
	FILE *fp;
	char line[4096];
	char *mountpoint;
	int ret;

	dev_major = major(dev);
	dev_minor = minor(dev);

	fp = SAFE_FOPEN(NULL, "/proc/self/mountinfo", "r");
	while (fgets(line, sizeof(line), fp) != NULL) {
		ret = sscanf(line, "%*d %*d %u:%u %*s %ms",
			     &mnt_major, &mnt_minor, &mountpoint);
		if (ret != 3)
			tst_brkm(TBROK, NULL,
				 "failed to parse mountinfo line: '%s'",
				 line);
		if (mnt_major == dev_major && mnt_minor == dev_minor)
			break;
		free(mountpoint);
		mountpoint = NULL;
	}
	SAFE_FCLOSE(NULL, fp);
	if (!mountpoint)
		tst_brkm(TBROK, NULL,
			 "Unable to find mount entry for device %u:%u",
			 dev_major, dev_minor);

	return mountpoint;
}

static char *overlay_get_upperdir(char *mountpoint)
{
	FILE *mntf;
	struct mntent *mnt;
	char *optstr, *optstart, *optend;
	char *upperdir = NULL;

	mntf = setmntent("/proc/self/mounts", "r");
	if (!mntf)
		tst_brkm(TBROK | TERRNO, NULL, "Can't open /proc/self/mounts");

	while ((mnt = getmntent(mntf)) != NULL) {
		if (strcmp(mnt->mnt_dir, mountpoint))
			continue;

		if (strcmp(mnt->mnt_type, "overlay"))
			tst_brkm(TBROK, NULL,
				 "expected overlayfs on mount point \"%s\", but it is of type %s.",
				 mountpoint, mnt->mnt_type);

		optstr = hasmntopt(mnt, "upperdir");

		if (optstr) {
			optstart = strchr(optstr, '=');
			optstart++;
			optend = strchrnul(optstr, ',');
			upperdir = strndup(optstart, optend - optstart);
			break;
		}

		tst_brkm(TBROK, NULL,
			 "mount point %s does not contain an upperdir",
			 mountpoint);
	}
	endmntent(mntf);

	if (!upperdir)
		tst_brkm(TBROK, NULL,
			 "Unable to find mount point \"%s\" in mount table",
			 mountpoint);

	return upperdir;
}

/*
 * To get from a file or directory on an overlayfs to a device
 * for an underlying mountpoint requires the following steps:
 *
 * 1. stat() the pathname and pick out st_dev.
 * 2. use the st_dev to look up the mount point of the file
 *    system in /proc/self/mountinfo
 *
 * Because 'mountinfo' is a much more complicated file format than
 * 'mounts', we switch to looking up the mount point in /proc/mounts,
 * and use the mntent.h helpers to parse the entries.
 *
 * 3. Using getmntent(), find the entry for the mount point identified
 *    in step 2.
 * 4. Call hasmntopt() to find the upperdir option, and parse that
 *    option to get to the path name for the upper directory.
 * 5. Call stat on the upper directory.  This should now contain
 *    the major and minor number for the underlying device (assuming
 *    that there aren't nested overlay file systems).
 * 6. Populate the uevent path.
 *
 * Example /proc/self/mountinfo line for overlayfs:
 *    471 69 0:48 / /tmp rw,relatime shared:242 - overlay overlay rw,seclabel,lowerdir=/tmp,upperdir=/mnt/upper/upper,workdir=/mnt/upper/work,uuid=null
 *
 * See section 3.5 of the kernel's Documentation/filesystems/proc.rst file
 * for a detailed explanation of the mountinfo format.
 */
static void overlay_get_uevent_path(char *tmp_path, char *uevent_path)
{
	struct stat st;
	char *mountpoint, *upperdir;

	tst_resm(TINFO, "Use OVERLAYFS specific strategy");

	SAFE_STAT(NULL, tmp_path, &st);

	mountpoint = overlay_mount_from_dev(st.st_dev);
	upperdir = overlay_get_upperdir(mountpoint);
	free(mountpoint);

	SAFE_STAT(NULL, upperdir, &st);
	free(upperdir);

	tst_resm(TINFO, "WARNING: used first of multiple backing devices");
	sprintf(uevent_path, "/sys/dev/block/%d:%d/uevent",
		major(st.st_dev), minor(st.st_dev));
}

__attribute__((nonnull))
void tst_find_backing_dev(const char *path, char *dev, size_t dev_size)
{
	struct stat buf;
	struct statfs fsbuf;
	char uevent_path[PATH_MAX+PATH_MAX+10]; //10 is for the static uevent path
	char dev_name[NAME_MAX];
	char tmp_path[PATH_MAX];
	unsigned int dev_major, dev_minor;

	if (stat(path, &buf) < 0)
		tst_brkm(TWARN | TERRNO, NULL, "stat() failed");

	strncpy(tmp_path, path, PATH_MAX-1);
	tmp_path[PATH_MAX-1] = '\0';
	if (S_ISREG(buf.st_mode))
		dirname(tmp_path);

	dev_major = major(buf.st_dev);
	dev_minor = minor(buf.st_dev);
	*dev = '\0';

	if (statfs(path, &fsbuf) < 0)
		tst_brkm(TBROK | TERRNO, NULL, "statfs() failed");

	if (fsbuf.f_type == TST_BTRFS_MAGIC) {
		btrfs_get_uevent_path(tmp_path, uevent_path);
	} else if (fsbuf.f_type == TST_OVERLAYFS_MAGIC) {
		overlay_get_uevent_path(tmp_path, uevent_path);
	} else if (dev_major == 0) {
		tst_brkm(TBROK, NULL, "%s resides on an unsupported pseudo-file system", path);
	} else {
		tst_resm(TINFO, "Use uevent strategy");
		sprintf(uevent_path,
			"/sys/dev/block/%d:%d/uevent", dev_major, dev_minor);
	}

	if (!access(uevent_path, R_OK)) {
		FILE_LINES_SCANF(NULL, uevent_path, "DEVNAME=%s", dev_name);

		if (!dev_name[0] || set_dev_path(dev_name, dev, dev_size))
			tst_brkm(TBROK, NULL, "Could not stat backing device %s", dev);

	} else {
		tst_brkm(TBROK, NULL, "uevent file (%s) access failed", uevent_path);
	}
}

void tst_stat_mount_dev(const char *const mnt_path, struct stat *const st)
{
	struct mntent *mnt;
	FILE *mntf = setmntent("/proc/self/mounts", "r");

	if (!mntf) {
		tst_brkm(TBROK | TERRNO, NULL, "Can't open /proc/self/mounts");
		return;
	}

	mnt = getmntent(mntf);
	if (!mnt) {
		tst_brkm(TBROK | TERRNO, NULL, "Can't read mounts or no mounts?");
		return;
	}

	do {
		if (strcmp(mnt->mnt_dir, mnt_path)) {
			mnt = getmntent(mntf);
			continue;
		}

		if (stat(mnt->mnt_fsname, st)) {
			tst_brkm(TBROK | TERRNO, NULL,
				 "Can't stat '%s', mounted at '%s'",
				 mnt->mnt_fsname, mnt_path);
		}

		return;
	} while (mnt);

	tst_brkm(TBROK, NULL, "Could not find mount device");
}

int tst_dev_block_size(const char *path)
{
	int fd;
	int size;
	char dev_name[PATH_MAX];

	tst_find_backing_dev(path, dev_name, sizeof(dev_name));

	fd = SAFE_OPEN(NULL, dev_name, O_RDONLY);
	SAFE_IOCTL(NULL, fd, BLKSSZGET, &size);
	SAFE_CLOSE(NULL, fd);

	return size;
}