/*
 * 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 2007 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include	<sys/param.h>
#include	<sys/types.h>
#include	<sys/user.h>
#include	<sys/vfs.h>
#include	<sys/vnode.h>
#include	<sys/file.h>
#include	<sys/stream.h>
#include	<sys/stropts.h>
#include	<sys/strsubr.h>
#include	<sys/dlpi.h>
#include	<sys/vnode.h>
#include	<sys/socket.h>
#include	<sys/sockio.h>
#include	<sys/cmn_err.h>
#include	<net/if.h>
#include	<sys/sad.h>
#include	<sys/kstr.h>
#include	<sys/ddi.h>
#include	<sys/sunddi.h>
#include	<sys/sunldi.h>

#include	<sys/cred.h>
#include	<sys/sysmacros.h>

#include	<sys/modctl.h>

/*
 * Routines to allow strplumb() legitimate access
 * to the kernel.
 */
int
kstr_open(major_t maj, minor_t min, vnode_t **vpp, int *fd)
{
	vnode_t		*vp;
	int		error;

	vp = makespecvp(makedevice(maj, min), VCHR);

	/*
	 * Fix for 4170365: only allocate file descriptor entry
	 * if file descriptor is to be returned; otherwise VOP_OPEN.
	 */
	if (fd != NULL)
		error = fassign(&vp, FREAD|FWRITE, fd);
	else
		error = VOP_OPEN(&vp, FREAD|FWRITE, CRED(), NULL);

	/*
	 * Must set vpp after calling fassign()/VOP_OPEN()
	 * since `vp' might change if it's a clone driver.
	 */
	if (vpp != NULL)
		*vpp = vp;

	return (error);
}

int
kstr_plink(vnode_t *vp, int fd, int *mux_id)
{
	int	id;
	int	error;

	if (error = strioctl(vp, I_PLINK, (intptr_t)fd, 0, K_TO_K, CRED(), &id))
		return (error);
	if (mux_id)
		*mux_id = id;
	return (0);
}

int
kstr_unplink(vnode_t *vp, int mux_id)
{
	int	rval;

	return (strioctl(vp, I_PUNLINK, (intptr_t)mux_id, 0,
	    K_TO_K, CRED(), &rval));
}

int
kstr_push(vnode_t *vp, char *mod)
{
	int	rval;

	return (strioctl(vp, I_PUSH, (intptr_t)mod, 0, K_TO_K, CRED(), &rval));
}

int
kstr_pop(vnode_t *vp)
{
	int	rval;

	return (strioctl(vp, I_POP, 0, 0, K_TO_K, CRED(), &rval));
}

int
kstr_close(vnode_t *vp, int fd)
{
	int ret;

	if (vp == (vnode_t *)NULL && fd == -1)
		return (EINVAL);

	if (fd != -1) {
		if (closeandsetf(fd, NULL) == 0) {
			return (0);
		} else {
			return (EINVAL);
		}
	} else {
		ret = VOP_CLOSE(vp, FREAD|FWRITE, 1, (offset_t)0, CRED(), NULL);
		VN_RELE(vp);
		return (ret);
	}
}

int
kstr_ioctl(struct vnode *vp, int cmd, intptr_t arg)
{
	int	rval;

	return (strioctl(vp, cmd, arg, 0, K_TO_K, CRED(), &rval));
}

/*
 * Optionally send data (if smp set) and optionally receive data (if rmp is
 * set). If timeo is NULL the reception will sleep until a message is
 * received; otherwise the sleep is limited to the specified amount of time.
 */
int
kstr_msg(vnode_t *vp, mblk_t *smp, mblk_t **rmp, timestruc_t *timeo)
{
	int			error;
	clock_t			timout;	/* milliseconds */
	uchar_t 		pri;
	int 			pflag;
	rval_t			rval;

	if (rmp == NULL && timeo != NULL &&
	    (timeo->tv_sec != 0 || timeo->tv_nsec != 0))
		return (EINVAL);

	if (smp == NULL && rmp == NULL)
		return (EINVAL);

	if (smp != NULL) {
		/* Send message while honoring flow control */
		(void) kstrputmsg(vp, smp, NULL, 0, 0,
		    MSG_BAND | MSG_HOLDSIG | MSG_IGNERROR, 0);
	}

	if (rmp == NULL) {
		/* No reply wanted by caller */
		return (0);
	}

	/*
	 * Convert from nanoseconds to milliseconds.
	 */
	if (timeo != NULL) {
		timout = timeo->tv_sec * 1000 + timeo->tv_nsec / 1000000;
		if (timout > INT_MAX)
			return (EINVAL);
	} else
		timout = -1;

	/* Wait for timeout millseconds for a message */
	pflag = MSG_ANY;
	pri = 0;
	*rmp = NULL;
	error = kstrgetmsg(vp, rmp, NULL, &pri, &pflag, timout, &rval);
	/* Callers use *rmp == NULL to determine that there was a timeout */
	if (error == ETIME)
		error = 0;
	return (error);
}

#define	SAD_ADM	"/devices/pseudo/sad@0:admin"
#define	SAD_USR	"/devices/pseudo/sad@0:user"

/*
 * It is the callers responsibility to make sure that "mods"
 * conforms to what is required. We do not check it here.
 *
 * "maj", "min", and "lastmin" are value-result parameters.
 * for SET_AUTOPUSH, "anchor" should be set to the place in the stream
 *	to put the anchor, or NULL if no anchor needs to be set.
 * for GET_AUTOPUSH, "anchor" should point to a uint_t to store the
 *	position of the anchor at, or NULL if the caller is not interested.
 */
int
kstr_autopush(int op, major_t *maj, minor_t *min, minor_t *lastmin,
    uint_t *anchor, char *mods[])
{
	ldi_handle_t	lh;
	ldi_ident_t	li;
	struct strapush	push;
	int		i, error, rval;

	li = ldi_ident_from_anon();
	if (op == SET_AUTOPUSH || op == CLR_AUTOPUSH) {
		error = ldi_open_by_name(SAD_ADM, FREAD|FWRITE,
		    kcred, &lh, li);
		if (error) {
			printf("kstr_autopush: open failed error %d\n", error);
			ldi_ident_release(li);
			return (error);
		}
	} else	{
		error = ldi_open_by_name(SAD_USR, FREAD|FWRITE,
		    kcred, &lh, li);
		if (error) {
			printf("kstr_autopush: open failed error %d\n", error);
			ldi_ident_release(li);
			return (error);
		}
	}
	ldi_ident_release(li);

	switch (op) {
	case GET_AUTOPUSH:
		/* Get autopush information */

		push.sap_major = *maj;
		push.sap_minor = *min;

		error = ldi_ioctl(lh, SAD_GAP, (intptr_t)&push,
		    FKIOCTL, kcred, &rval);
		if (error) {
			printf("kstr_autopush: "
			    "ioctl(GET_AUTOPUSH) failed, error %d\n", error);
			(void) ldi_close(lh, FREAD|FWRITE, kcred);
			return (error);
		}
		switch (push.sap_cmd) {
		case SAP_ONE:
			*maj = push.sap_major;
			*min = push.sap_minor;
			*lastmin = 0;
			break;

		case SAP_RANGE:
			*maj = push.sap_major;
			*min = push.sap_minor;
			*lastmin = push.sap_lastminor;
			break;

		case SAP_ALL:
			*maj = push.sap_major;
			*min = (minor_t)-1;
			break;
		}

		if (anchor != NULL)
			*anchor = push.sap_anchor;

		if (push.sap_npush > 1) {
			for (i = 0; i < push.sap_npush &&
			    mods[i] != NULL; i++)
				(void) strcpy(mods[i], push.sap_list[i]);
			mods[i] = NULL;
		}
		(void) ldi_close(lh, FREAD|FWRITE, kcred);
		return (0);

	case CLR_AUTOPUSH:
		/* Remove autopush information */

		push.sap_cmd = SAP_CLEAR;
		push.sap_minor = *min;
		push.sap_major = *maj;

		error = ldi_ioctl(lh, SAD_SAP, (intptr_t)&push,
		    FKIOCTL, kcred, &rval);
		if (error) {
			printf("kstr_autopush: "
			    "ioctl(CLR_AUTOPUSH) failed, error %d\n", error);
		}
		(void) ldi_close(lh, FREAD|FWRITE, kcred);
		return (error);

	case SET_AUTOPUSH:
		/* Set autopush information */

		if (*min == (minor_t)-1) {
			push.sap_cmd = SAP_ALL;
		} else if (*lastmin == 0) {
			push.sap_cmd = SAP_ONE;
		} else	{
			push.sap_cmd = SAP_RANGE;
		}

		if (anchor != NULL)
			push.sap_anchor = *anchor;
		else
			push.sap_anchor = 0;

		push.sap_minor = *min;
		push.sap_major = *maj;
		if (lastmin)
			push.sap_lastminor = *lastmin;
		else
			push.sap_lastminor = 0;

		/* pain */
		for (i = 0; i < MAXAPUSH && mods[i] != (char *)NULL; i++) {
			(void) strcpy(push.sap_list[i], mods[i]);
		}
		push.sap_npush = i;
		push.sap_list[i][0] = '\0';

		error = ldi_ioctl(lh, SAD_SAP, (intptr_t)&push,
		    FKIOCTL, kcred, &rval);
		if (error) {
			printf("kstr_autopush: "
			    "ioctl(SET_AUTOPUSH) failed, error %d\n", error);
		}
		(void) ldi_close(lh, FREAD|FWRITE, kcred);
		return (error);

	default:
		(void) ldi_close(lh, FREAD|FWRITE, kcred);
		return (EINVAL);
	}
}