/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2022 Marcel Telka <marcel@telka.sk>
 */

/*
 * Test for fchmodat(AT_SYMLINK_NOFOLLOW)
 */

#include <sys/param.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

int
main(void)
{
	int ret = 0;

	char template[MAXPATHLEN];
	char *path;
	char file[MAXPATHLEN];
	char link[MAXPATHLEN];

	/* prepare template for temporary directory */
	if (strlcpy(template, "/tmp/XXXXXX", sizeof (template))
	    >= sizeof (template)) {
		(void) printf("FAIL: Template copy failed\n");
		exit(EXIT_FAILURE);
	}

	/* create temporary directory */
	if ((path = mkdtemp(template)) == NULL) {
		(void) printf("FAIL: Temporary directory creation failed\n");
		exit(EXIT_FAILURE);
	}

	/* format file and link paths */
	(void) snprintf(file, sizeof (file), "%s/file", path);
	(void) snprintf(link, sizeof (link), "%s/link", path);

	/* create the file */
	int fd = open(file, O_WRONLY | O_CREAT, 0644);
	if (fd < 0) {
		(void) printf("FAIL: File %s creation failed\n", file);
		(void) rmdir(path);
		exit(EXIT_FAILURE);
	}
	(void) close(fd);

	/* create symlink */
	if (symlink("file", link) != 0) {
		(void) printf("FAIL: Symlink %s creation failed\n", link);
		(void) unlink(file);
		(void) rmdir(path);
		exit(EXIT_FAILURE);
	}

	/* test fchmodat(AT_SYMLINK_NOFOLLOW) for symlink */
	if (fchmodat(AT_FDCWD, link, 0666, AT_SYMLINK_NOFOLLOW) == 0) {
		(void) printf("FAIL: fchmodat(AT_SYMLINK_NOFOLLOW) "
		    "unexpectedly succeeded for symlink\n");
		ret = EXIT_FAILURE;
	}
	/* test fchmodat(AT_SYMLINK_NOFOLLOW) for regular file */
	if (fchmodat(AT_FDCWD, file, 0666, AT_SYMLINK_NOFOLLOW) != 0) {
		(void) printf("FAIL: fchmodat(AT_SYMLINK_NOFOLLOW) failed for "
		    "regular file\n");
		ret = EXIT_FAILURE;
	}

	/* cleanup */
	(void) unlink(link);
	(void) unlink(file);
	(void) rmdir(path);

	return (ret);
}