/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (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 2005 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*LINTLIBRARY*/

#include <sys/types.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/param.h>
#include <sys/stat.h>
#include <sys/sysmacros.h>
#include <sys/vfstab.h>
#include <sys/lofi.h>
#include <sys/ramdisk.h>
#include <sys/fssnap_if.h>
#include "libadm.h"

/*
 * Globals:
 *	getfullrawname - returns a fully-qualified raw device name
 *	getfullblkname - returns a fully-qualified block device name
 *
 * These two routines take a device pathname and return corresponding
 * the raw or block device name.
 *
 * First the device name is fully qualified:
 * 	If the device name does not start with a '/' or starts with
 *	'./' then the current working directory is added to the beginning
 *	of the pathname.
 *
 *	If the device name starts with a '../' then all but the last
 *	sub-directory of the current working directory is added to the
 *	the beginning of the pathname.
 *
 * Second if the fully-qualified device name given is the raw/block
 * device that is being asked for then the fully-qualified device name is
 * returned.
 *
 * Third if an entry is found in /etc/vfstab which matches the given name
 * then the corresponding raw/block device is returned.  This allows
 * non-standard names to be converted (.i.e., block device "/dev/joe" can
 * be converted to raw device "/dev/fred", via this mechanism).
 *
 * Last standard names are converted.  Standard names are those
 * with a '/dsk/' for block or '/rdsk/' for raw sub-directory components
 * in the device name. Or, the filename component has an 'r' for raw or
 * no 'r' for block (e.g., rsd0a <=> sd0a).
 *
 * Caveat:
 * It is assumed that the block and raw devices have the
 * same device number, and this is used to verify the conversion
 * happened corretly.  If this happens not to be true, due to mapping
 * of minor numbers or sometheing, then entries can be put in the
 * the '/etc/vfstab' file to over-ride this checking.
 *
 *
 * Return Values:
 * 	raw/block device name	- (depending on which routine is used)
 *	null string		- When the conversion failed
 *	null pointer		- malloc problems
 *
 * It is up to the user of these routines to free the memory, of
 * the device name or null string returned by these library routines,
 * when appropriate by the application.
 */
#define	GET_BLK	0
#define	GET_RAW	1

static int test_if_blk(char *, dev_t);
static int test_if_raw(char *, dev_t);
static char *getblkcomplete(char *, struct stat64 *);
static char *getrawcomplete(char *, struct stat64 *);

/*
 * getfullname() - Builds a fully qualified pathname.
 *		   This handles . and .. as well.
 *		   NOTE: This is different from realpath(3C) because
 *			 it does not follow links.
 */
static char *
getfullname(char *path)
{
	char	cwd[MAXPATHLEN];
	char	*c;
	char	*wa;
	size_t	len;

	if (*path == '/')
		return (strdup(path));

	if (getcwd(cwd, sizeof (cwd)) == NULL)
		return (strdup(""));

	/* handle . and .. */
	if (strncmp(path, "./", 2) == 0) {
		/* strip the ./ from the given path */
		path += 2;
	} else if (strncmp(path, "../", 3) == 0) {
		/* strip the last directory component from cwd */
		c = strrchr(cwd, '/');
		*c = '\0';

		/* strip the ../ from the given path */
		path += 3;
	}

	/*
	 * Adding 2 takes care of slash and null terminator.
	 */
	len = strlen(cwd) + strlen(path) + 2;
	if ((wa = malloc(len)) == NULL)
		return (NULL);

	(void) strcpy(wa, cwd);
	(void) strcat(wa, "/");
	(void) strcat(wa, path);

	return (wa);
}

/*
 * test the path/fname to see if is blk special
 */
static int
test_if_blk(char *new_path, dev_t raw_dev)
{
	struct stat64	buf;

	/* check if we got a char special file */
	if (stat64(new_path, &buf) != 0)
		return (0);

	if (!S_ISBLK(buf.st_mode))
		return (0);

	if (raw_dev != buf.st_rdev)
		return (0);

	return (1);
}

/*
 * test the path/fname to see if is char special
 */
static int
test_if_raw(char *new_path, dev_t blk_dev)
{
	struct stat64	buf;

	/* check if we got a char special file */
	if (stat64(new_path, &buf) != 0)
		return (0);

	if (!S_ISCHR(buf.st_mode))
		return (0);

	if (blk_dev != buf.st_rdev)
		return (0);

	return (1);
}

/*
 * complete getblkrawname() for blk->raw to handle volmgt devices
 */

static char *
getblkcomplete(char *cp, struct stat64 *dat)
{
	char 		*dp;
	char		*new_path;
	char		c;

	/* ok, so we either have a bad device or a floppy */

	/* try the rfd# form */
	if ((dp = strstr(cp, "/rfd")) != NULL) {
		if ((new_path = malloc(strlen(cp))) == NULL)
			return (NULL);

		c = *++dp;			/* save the 'r' */
		*dp = '\0';			/* replace it with a null */
		(void) strcpy(new_path, cp);	/* save first part of it */
		*dp++ = c;			/* give the 'r' back */
		(void) strcat(new_path, dp);	/* copy, skipping the 'r' */

		if (test_if_blk(new_path, dat->st_rdev))
			return (new_path);

		free(new_path);
		return (strdup(""));
	}

	/* try the rdiskette form */
	if ((dp = strstr(cp, "/rdiskette")) != NULL) {
		if ((new_path = malloc(strlen(cp))) == NULL)
			return (NULL);

		c = *++dp;			/* save the 'r' */
		*dp = '\0';			/* replace it with a null */
		(void) strcpy(new_path, cp);	/* save first part of it */
		*dp++ = c;			/* give the 'r' back */
		(void) strcat(new_path, dp);	/* copy, skipping the 'r' */

		if (test_if_blk(new_path, dat->st_rdev))
			return (new_path);

		free(new_path);
		return (strdup(""));
	}

	/* no match found */
	return (strdup(""));
}

/*
 * complete getfullrawname() for raw->blk to handle volmgt devices
 */

static char *
getrawcomplete(char *cp, struct stat64 *dat)
{
	char 		*dp;
	char		*new_path;
	char		c;

	/* ok, so we either have a bad device or a floppy */

	/* try the fd# form */
	if ((dp = strstr(cp, "/fd")) != NULL) {
		/* malloc path for new_path to hold raw */
		if ((new_path = malloc(strlen(cp)+2)) == NULL)
			return (NULL);

		c = *++dp;			/* save the 'f' */
		*dp = '\0';			/* replace it with a null */
		(void) strcpy(new_path, cp);	/* save first part of it */
		*dp = c;			/* put the 'f' back */
		(void) strcat(new_path, "r");	/* insert an 'r' */
		(void) strcat(new_path, dp);	/* copy the rest */

		if (test_if_raw(new_path, dat->st_rdev))
			return (new_path);

		free(new_path);
	}

	/* try the diskette form */
	if ((dp = strstr(cp, "/diskette")) != NULL) {
		/* malloc path for new_path to hold raw */
		if ((new_path = malloc(strlen(cp)+2)) == NULL)
			return (NULL);

		c = *++dp;			/* save at 'd' */
		*dp = '\0';			/* replace it with a null */
		(void) strcpy(new_path, cp);	/* save first part */
		*dp = c;			/* put the 'd' back */
		(void) strcat(new_path, "r");	/* insert an 'r' */
		(void) strcat(new_path, dp);	/* copy the rest */

		if (test_if_raw(new_path, dat->st_rdev))
			return (new_path);

		free(new_path);
		return (strdup(""));
	}

	/* failed to build raw name, return null string */
	return (strdup(""));



}

static char *
getvfsspecial(char *path, int raw_special)
{
	FILE		*fp;
	struct vfstab	vp;
	struct vfstab	ref_vp;

	if ((fp = fopen("/etc/vfstab", "r")) == NULL)
		return (NULL);

	(void) memset(&ref_vp, 0, sizeof (struct vfstab));

	if (raw_special)
		ref_vp.vfs_special = path;
	else
		ref_vp.vfs_fsckdev = path;

	if (getvfsany(fp, &vp, &ref_vp)) {
		(void) fclose(fp);
		return (NULL);
	}

	(void) fclose(fp);

	if (raw_special)
		return (vp.vfs_fsckdev);

	return (vp.vfs_special);
}

/*
 * change the device name to a block device name
 */
char *
getfullblkname(char *cp)
{
	struct stat64	buf;
	char		*dp;
	char		*new_path;
	dev_t		raw_dev;

	if (cp == NULL)
		return (strdup(""));

	/*
	 * Create a fully qualified name.
	 */
	if ((cp = getfullname(cp)) == NULL)
		return (NULL);

	if (*cp == '\0')
		return (cp);

	if (stat64(cp, &buf) != 0) {
		free(cp);
		return (strdup(""));
	}

	if (S_ISBLK(buf.st_mode))
		return (cp);

	if (!S_ISCHR(buf.st_mode)) {
		free(cp);
		return (strdup(""));
	}

	if ((dp = getvfsspecial(cp, GET_BLK)) != NULL) {
		free(cp);
		return (strdup(dp));
	}

	raw_dev = buf.st_rdev;

	/*
	 * We have a raw device name, go find the block name.
	 */
	if ((dp = strstr(cp, "/rdsk/")) == NULL &&
	    (dp = strstr(cp, "/" LOFI_CHAR_NAME "/")) == NULL &&
	    (dp = strstr(cp, "/" RD_CHAR_NAME "/")) == NULL &&
	    (dp = strstr(cp, "/" SNAP_CHAR_NAME "/")) == NULL &&
	    (dp = strrchr(cp, '/')) == NULL) {
		/* this is not really possible */
		free(cp);
		return (strdup(""));
	}
	dp++;
	if (*dp != 'r') {
		dp = getblkcomplete(cp, &buf);
		free(cp);
		return (dp);
	}
	if ((new_path = malloc(strlen(cp))) == NULL) {
		free(cp);
		return (NULL);
	}
	(void) strncpy(new_path, cp, dp - cp);

	/* fill in the rest of the unraw name */
	(void) strcpy(new_path + (dp - cp), dp + 1);

	if (test_if_blk(new_path, raw_dev)) {
		free(cp);
		/* block name was found, return it here */
		return (new_path);
	}
	free(new_path);

	dp = getblkcomplete(cp, &buf);
	free(cp);
	return (dp);
}

/*
 * change the device name to a raw devname
 */
char *
getfullrawname(char *cp)
{
	struct stat64	buf;
	char		*dp;
	char		*new_path;
	dev_t		blk_dev;

	if (cp == NULL)
		return (strdup(""));

	/*
	 * Create a fully qualified name.
	 */
	if ((cp = getfullname(cp)) == NULL)
		return (NULL);

	if (*cp == '\0')
		return (cp);

	if (stat64(cp, &buf) != 0) {
		free(cp);
		return (strdup(""));
	}

	if (S_ISCHR(buf.st_mode))
		return (cp);

	if (!S_ISBLK(buf.st_mode)) {
		free(cp);
		return (strdup(""));
	}

	blk_dev = buf.st_rdev;

	if ((dp = getvfsspecial(cp, GET_RAW)) != NULL) {
		free(cp);
		return (strdup(dp));
	}

	/*
	 * We have a block device name, go find the raw name.
	 */
	if ((dp = strstr(cp, "/dsk/")) == NULL &&
	    (dp = strstr(cp, "/" LOFI_BLOCK_NAME "/")) == NULL &&
	    (dp = strstr(cp, "/" RD_BLOCK_NAME "/")) == NULL &&
	    (dp = strstr(cp, "/" SNAP_BLOCK_NAME "/")) == NULL &&
	    (dp = strrchr(cp, '/')) == NULL) {
		/* this is not really possible */
		free(cp);
		return (strdup(""));
	}
	dp++;

	if ((new_path = malloc(strlen(cp)+2)) == NULL) {
		free(cp);
		return (NULL);
	}
	(void) strncpy(new_path, cp, dp - cp);
	/* fill in the rest of the raw name */
	new_path[dp - cp] = 'r';
	(void) strcpy(new_path + (dp - cp) + 1, dp);

	if (test_if_raw(new_path, blk_dev)) {
		free(cp);
		return (new_path);
	}
	free(new_path);

	dp = getrawcomplete(cp, &buf);
	free(cp);
	return (dp);
}