/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
 */

/*
 * This is a test program that uses ioctls to the ZFS Unit Test driver
 * to perform readdirs or lookups using flags not normally available
 * to user-land programs.  This allows testing of the flags'
 * behavior outside of a complicated consumer, such as the SMB driver.
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stropts.h>
#include <errno.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/dirent.h>
#include <sys/attr.h>
#include <stddef.h>
#include <fcntl.h>
#include <string.h>
#include <time.h>

#define	_KERNEL

#include <sys/fs/zut.h>
#include <sys/extdirent.h>

#undef	_KERNEL

#define	MAXBUF (64 * 1024)
#define	BIGBUF 4096
#define	LILBUF (sizeof (dirent_t))

#define	DIRENT_NAMELEN(reclen)	\
	((reclen) - (offsetof(dirent_t, d_name[0])))

static void
usage(char *pnam)
{
	(void) fprintf(stderr, "Usage:\n    %s -l [-is] dir-to-look-in "
	    "file-in-dir [xfile-on-file]\n", pnam);
	(void) fprintf(stderr, "    %s -i [-ls] dir-to-look-in "
	    "file-in-dir [xfile-on-file]\n", pnam);
	(void) fprintf(stderr, "    %s -s [-il] dir-to-look-in "
	    "file-in-dir [xfile-on-file]\n", pnam);
	(void) fprintf(stderr, "\t    Perform a lookup\n");
	(void) fprintf(stderr, "\t    -l == lookup\n");
	(void) fprintf(stderr, "\t    -i == request FIGNORECASE\n");
	(void) fprintf(stderr, "\t    -s == request stat(2) and xvattr info\n");
	(void) fprintf(stderr, "    %s -r [-ea] [-b buffer-size-in-bytes] "
	    "dir-to-look-in [file-in-dir]\n", pnam);
	(void) fprintf(stderr, "    %s -e [-ra] [-b buffer-size-in-bytes] "
	    "dir-to-look-in [file-in-dir]\n", pnam);
	(void) fprintf(stderr, "    %s -a [-re] [-b buffer-size-in-bytes] "
	    "dir-to-look-in [file-in-dir]\n", pnam);
	(void) fprintf(stderr, "\t    Perform a readdir\n");
	(void) fprintf(stderr, "\t    -r == readdir\n");
	(void) fprintf(stderr, "\t    -e == request extended entries\n");
	(void) fprintf(stderr, "\t    -a == request access filtering\n");
	(void) fprintf(stderr, "\t    -b == buffer size (default 4K)\n");
	(void) fprintf(stderr, "    %s -A path\n", pnam);
	(void) fprintf(stderr, "\t    Look up _PC_ACCESS_FILTERING "
	    "for path with pathconf(2)\n");
	(void) fprintf(stderr, "    %s -E path\n", pnam);
	(void) fprintf(stderr, "\t    Look up _PC_SATTR_EXISTS "
	    "for path with pathconf(2)\n");
	(void) fprintf(stderr, "    %s -S path\n", pnam);
	(void) fprintf(stderr, "\t    Look up _PC_SATTR_EXISTS "
	    "for path with pathconf(2)\n");
	exit(EINVAL);
}

static void
print_extd_entries(zut_readdir_t *r)
{
	struct edirent *eodp;
	char *bufstart;

	eodp = (edirent_t *)(uintptr_t)r->zr_buf;
	bufstart = (char *)eodp;
	while ((char *)eodp < bufstart + r->zr_bytes) {
		char *blanks = "                ";
		int i = 0;
		while (i < EDIRENT_NAMELEN(eodp->ed_reclen)) {
			if (!eodp->ed_name[i])
				break;
			(void) printf("%c", eodp->ed_name[i++]);
		}
		if (i < 16)
			(void) printf("%.*s", 16 - i, blanks);
		(void) printf("\t%x\n", eodp->ed_eflags);
		eodp = (edirent_t *)((intptr_t)eodp + eodp->ed_reclen);
	}
}

static void
print_entries(zut_readdir_t *r)
{
	dirent64_t *dp;
	char *bufstart;

	dp = (dirent64_t *)(intptr_t)r->zr_buf;
	bufstart = (char *)dp;
	while ((char *)dp < bufstart + r->zr_bytes) {
		int i = 0;
		while (i < DIRENT_NAMELEN(dp->d_reclen)) {
			if (!dp->d_name[i])
				break;
			(void) printf("%c", dp->d_name[i++]);
		}
		(void) printf("\n");
		dp = (dirent64_t *)((intptr_t)dp + dp->d_reclen);
	}
}

static void
print_stats(struct stat64 *sb)
{
	char timebuf[512];

	(void) printf("st_mode\t\t\t%04lo\n", (unsigned long)sb->st_mode);
	(void) printf("st_ino\t\t\t%llu\n", (unsigned long long)sb->st_ino);
	(void) printf("st_nlink\t\t%lu\n", (unsigned long)sb->st_nlink);
	(void) printf("st_uid\t\t\t%d\n", sb->st_uid);
	(void) printf("st_gid\t\t\t%d\n", sb->st_gid);
	(void) printf("st_size\t\t\t%lld\n", (long long)sb->st_size);
	(void) printf("st_blksize\t\t%ld\n", (long)sb->st_blksize);
	(void) printf("st_blocks\t\t%lld\n", (long long)sb->st_blocks);

	timebuf[0] = 0;
	if (ctime_r(&sb->st_atime, timebuf, 512)) {
		(void) printf("st_atime\t\t");
		(void) printf("%s", timebuf);
	}
	timebuf[0] = 0;
	if (ctime_r(&sb->st_mtime, timebuf, 512)) {
		(void) printf("st_mtime\t\t");
		(void) printf("%s", timebuf);
	}
	timebuf[0] = 0;
	if (ctime_r(&sb->st_ctime, timebuf, 512)) {
		(void) printf("st_ctime\t\t");
		(void) printf("%s", timebuf);
	}
}

static void
print_xvs(uint64_t xvs)
{
	uint_t bits;
	int idx = 0;

	if (xvs == 0)
		return;

	(void) printf("-------------------\n");
	(void) printf("Attribute bit(s) set:\n");
	(void) printf("-------------------\n");

	bits = xvs & ((1 << F_ATTR_ALL) - 1);
	while (bits) {
		uint_t rest = bits >> 1;
		if (bits & 1) {
			(void) printf("%s", attr_to_name((f_attr_t)idx));
			if (rest)
				(void) printf(", ");
		}
		idx++;
		bits = rest;
	}
	(void) printf("\n");
}

int
main(int argc, char **argv)
{
	zut_lookup_t lk = {0};
	zut_readdir_t rd = {0};
	boolean_t checking = B_FALSE;
	boolean_t looking = B_FALSE;
	boolean_t reading = B_FALSE;
	boolean_t bflag = B_FALSE;
	long rddir_bufsize = BIGBUF;
	int error = 0;
	int check;
	int fd;
	int c;

	while ((c = getopt(argc, argv, "lisaerb:ASE")) != -1) {
		switch (c) {
		case 'l':
			looking = B_TRUE;
			break;
		case 'i':
			lk.zl_reqflags |= ZUT_IGNORECASE;
			looking = B_TRUE;
			break;
		case 's':
			lk.zl_reqflags |= ZUT_GETSTAT;
			looking = B_TRUE;
			break;
		case 'a':
			rd.zr_reqflags |= ZUT_ACCFILTER;
			reading = B_TRUE;
			break;
		case 'e':
			rd.zr_reqflags |= ZUT_EXTRDDIR;
			reading = B_TRUE;
			break;
		case 'r':
			reading = B_TRUE;
			break;
		case 'b':
			reading = B_TRUE;
			bflag = B_TRUE;
			rddir_bufsize = strtol(optarg, NULL, 0);
			break;
		case 'A':
			checking = B_TRUE;
			check = _PC_ACCESS_FILTERING;
			break;
		case 'S':
			checking = B_TRUE;
			check = _PC_SATTR_ENABLED;
			break;
		case 'E':
			checking = B_TRUE;
			check = _PC_SATTR_EXISTS;
			break;
		case '?':
		default:
			usage(argv[0]);		/* no return */
		}
	}

	if ((checking && looking) || (checking && reading) ||
	    (looking && reading) || (!reading && bflag) ||
	    (!checking && !reading && !looking))
		usage(argv[0]);		/* no return */

	if (rddir_bufsize < LILBUF || rddir_bufsize > MAXBUF) {
		(void) fprintf(stderr, "Sorry, buffer size "
		    "must be >= %d and less than or equal to %d bytes.\n",
		    (int)LILBUF, MAXBUF);
		exit(EINVAL);
	}

	if (checking) {
		char pathbuf[MAXPATHLEN];
		long result;

		if (argc - optind < 1)
			usage(argv[0]);		/* no return */
		(void) strlcpy(pathbuf, argv[optind], MAXPATHLEN);
		result = pathconf(pathbuf, check);
		(void) printf("pathconf(2) check for %s\n", pathbuf);
		switch (check) {
		case _PC_SATTR_ENABLED:
			(void) printf("System attributes ");
			if (result != 0)
				(void) printf("Enabled\n");
			else
				(void) printf("Not enabled\n");
			break;
		case _PC_SATTR_EXISTS:
			(void) printf("System attributes ");
			if (result != 0)
				(void) printf("Exist\n");
			else
				(void) printf("Do not exist\n");
			break;
		case _PC_ACCESS_FILTERING:
			(void) printf("Access filtering ");
			if (result != 0)
				(void) printf("Available\n");
			else
				(void) printf("Not available\n");
			break;
		}
		return (result);
	}

	if ((fd = open(ZUT_DEV, O_RDONLY)) < 0) {
		perror(ZUT_DEV);
		return (ENXIO);
	}

	if (reading) {
		char *buf;

		if (argc - optind < 1)
			usage(argv[0]);		/* no return */

		(void) strlcpy(rd.zr_dir, argv[optind], MAXPATHLEN);
		if (argc - optind > 1) {
			(void) strlcpy(rd.zr_file, argv[optind + 1],
			    MAXNAMELEN);
			rd.zr_reqflags |= ZUT_XATTR;
		}

		if ((buf = malloc(rddir_bufsize)) == NULL) {
			error = errno;
			perror("malloc");
			(void) close(fd);
			return (error);
		}

		rd.zr_buf = (uint64_t)(uintptr_t)buf;
		rd.zr_buflen = rddir_bufsize;

		while (!rd.zr_eof) {
			int ierr;

			if ((ierr = ioctl(fd, ZUT_IOC_READDIR, &rd)) != 0) {
				(void) fprintf(stderr,
				    "IOCTL error: %s (%d)\n",
				    strerror(ierr), ierr);
				free(buf);
				(void) close(fd);
				return (ierr);
			}
			if (rd.zr_retcode) {
				(void) fprintf(stderr,
				    "readdir result: %s (%d)\n",
				    strerror(rd.zr_retcode), rd.zr_retcode);
				free(buf);
				(void) close(fd);
				return (rd.zr_retcode);
			}
			if (rd.zr_reqflags & ZUT_EXTRDDIR)
				print_extd_entries(&rd);
			else
				print_entries(&rd);
		}
		free(buf);
	} else {
		int ierr;

		if (argc - optind < 2)
			usage(argv[0]);		/* no return */

		(void) strlcpy(lk.zl_dir, argv[optind], MAXPATHLEN);
		(void) strlcpy(lk.zl_file, argv[optind + 1], MAXNAMELEN);
		if (argc - optind > 2) {
			(void) strlcpy(lk.zl_xfile,
			    argv[optind + 2], MAXNAMELEN);
			lk.zl_reqflags |= ZUT_XATTR;
		}

		if ((ierr = ioctl(fd, ZUT_IOC_LOOKUP, &lk)) != 0) {
			(void) fprintf(stderr,
			    "IOCTL error: %s (%d)\n",
			    strerror(ierr), ierr);
			(void) close(fd);
			return (ierr);
		}

		(void) printf("\nLookup of ");
		if (lk.zl_reqflags & ZUT_XATTR) {
			(void) printf("extended attribute \"%s\" of ",
			    lk.zl_xfile);
		}
		(void) printf("file \"%s\" ", lk.zl_file);
		(void) printf("in directory \"%s\" ", lk.zl_dir);
		if (lk.zl_retcode) {
			(void) printf("failed: %s (%d)\n",
			    strerror(lk.zl_retcode), lk.zl_retcode);
			(void) close(fd);
			return (lk.zl_retcode);
		}

		(void) printf("succeeded.\n");
		if (lk.zl_reqflags & ZUT_IGNORECASE) {
			(void) printf("----------------------------\n");
			(void) printf("dirent flags: 0x%0x\n", lk.zl_deflags);
			(void) printf("real name: %s\n", lk.zl_real);
		}
		if (lk.zl_reqflags & ZUT_GETSTAT) {
			(void) printf("----------------------------\n");
			print_stats(&lk.zl_statbuf);
			print_xvs(lk.zl_xvattrs);
		}
	}

	(void) close(fd);
	return (0);
}