/* SPDX-License-Identifier: MIT */
/*
 * Description: test io_uring linkat handling
 */
#include <fcntl.h>
#include <stdio.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>

#include "liburing.h"
#include "helpers.h"

static int do_linkat(struct io_uring *ring, int olddirfd, const char *oldname,
		     const char *newname, int flags)
{
	struct io_uring_sqe *sqe;
	struct io_uring_cqe *cqe;
	int ret;

	sqe = io_uring_get_sqe(ring);
	if (!sqe) {
		fprintf(stderr, "sqe get failed\n");
		return 1;
	}
	io_uring_prep_linkat(sqe, olddirfd, oldname, AT_FDCWD, newname, flags);

	ret = io_uring_submit(ring);
	if (ret != 1) {
		fprintf(stderr, "submit failed: %d\n", ret);
		return 1;
	}

	ret = io_uring_wait_cqes(ring, &cqe, 1, 0, 0);
	if (ret) {
		fprintf(stderr, "wait_cqe failed: %d\n", ret);
		return 1;
	}
	ret = cqe->res;
	io_uring_cqe_seen(ring, cqe);
	return ret;
}

static int files_linked_ok(const char* fn1, const char *fn2)
{
	struct stat s1, s2;

	if (stat(fn1, &s1)) {
		fprintf(stderr, "stat(%s): %s\n", fn1, strerror(errno));
		return 0;
	}
	if (stat(fn2, &s2)) {
		fprintf(stderr, "stat(%s): %s\n", fn2, strerror(errno));
		return 0;
	}
	if (s1.st_dev != s2.st_dev || s1.st_ino != s2.st_ino) {
		fprintf(stderr, "linked files have different device / inode numbers\n");
		return 0;
	}
	if (s1.st_nlink != 2 || s2.st_nlink != 2) {
		fprintf(stderr, "linked files have unexpected links count\n");
		return 0;
	}
	return 1;
}

int main(int argc, char *argv[])
{
	static const char target[] = "io_uring-linkat-test-target";
	static const char emptyname[] = "io_uring-linkat-test-empty";
	static const char linkname[] = "io_uring-linkat-test-link";
	static const char symlinkname[] = "io_uring-linkat-test-symlink";
	struct io_uring ring;
	int ret, fd, exit_status = T_EXIT_FAIL;

	if (argc > 1)
		return T_EXIT_SKIP;

	ret = io_uring_queue_init(8, &ring, 0);
	if (ret) {
		fprintf(stderr, "queue init failed: %d\n", ret);
		return ret;
	}

	ret = fd = open(target, O_CREAT | O_RDWR | O_EXCL, 0600);
	if (ret < 0) {
		perror("open");
		goto out;
	}
	if (write(fd, "linktest", 8) != 8) {
		close(fd);
		goto out;
	}
	if(geteuid()) {
		fprintf(stdout, "not root, skipping AT_EMPTY_PATH test\n");
	} else {
		ret = do_linkat(&ring, fd, "", emptyname, AT_EMPTY_PATH);
		if (ret < 0) {
			if (ret == -EBADF || ret == -EINVAL) {
				fprintf(stdout, "linkat not supported, skipping\n");
				exit_status = T_EXIT_SKIP;
				goto out;
			}
			fprintf(stderr, "linkat: %s\n", strerror(-ret));
			goto out;
		} else if (ret) {
			goto out;
		}
		if (!files_linked_ok(emptyname, target))
			goto out;
		unlinkat(AT_FDCWD, emptyname, 0);
	}
	close(fd);

	ret = symlink(target, symlinkname);
	if (ret < 0) {
		perror("open");
		goto out;
	}

	ret = do_linkat(&ring, AT_FDCWD, target, linkname, 0);
	if (ret < 0) {
		if (ret == -EBADF || ret == -EINVAL) {
			fprintf(stdout, "linkat not supported, skipping\n");
			exit_status = T_EXIT_SKIP;
			goto out;
		}
		fprintf(stderr, "linkat: %s\n", strerror(-ret));
		goto out;
	} else if (ret) {
		goto out;
	}

	if (!files_linked_ok(linkname, target))
		goto out;

	unlinkat(AT_FDCWD, linkname, 0);

	ret = do_linkat(&ring, AT_FDCWD, symlinkname, linkname, AT_SYMLINK_FOLLOW);
	if (ret < 0) {
		fprintf(stderr, "linkat: %s\n", strerror(-ret));
		goto out;
	} else if (ret) {
		goto out;
	}

	if (!files_linked_ok(symlinkname, target))
		goto out;

	ret = do_linkat(&ring, AT_FDCWD, target, linkname, 0);
	if (ret != -EEXIST) {
		fprintf(stderr, "test_linkat linkname already exists failed: %d\n", ret);
		goto out;
	}

	ret = do_linkat(&ring, AT_FDCWD, target, "surely/this/does/not/exist", 0);
	if (ret != -ENOENT) {
		fprintf(stderr, "test_linkat no parent failed: %d\n", ret);
		goto out;
	}
	exit_status = T_EXIT_PASS;
out:
	unlinkat(AT_FDCWD, symlinkname, 0);
	unlinkat(AT_FDCWD, linkname, 0);
	unlinkat(AT_FDCWD, emptyname, 0);
	unlinkat(AT_FDCWD, target, 0);
	io_uring_queue_exit(&ring);
	return exit_status;
}