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

#include <stdio.h>
#include <libintl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <locale.h>
#include <errno.h>
#include <unistd.h>
#include <strings.h>
#include <libscf.h>	/* for SMF error exit codes */

#include <srpt_ioctl.h>


/* Globals */
char	cmdName[] = "svc-srpt";

int
main(int argc, char *argv[])
{
	int		ret;
	boolean_t	do_enable = B_FALSE;
	int		srpt_ioctl;
	int		fd = -1;
	int		saverr;
	char		*txt;
	struct stat	statbuf;

	(void) setlocale(LC_ALL, "");

	if (argc < 2 || argc > 2) {
		goto usage;
	}

	if (strcasecmp(argv[1], "start") == 0) {
		do_enable = B_TRUE;
		srpt_ioctl = SRPT_IOC_ENABLE_SVC;
	} else if (strcasecmp(argv[1], "stop") == 0)  {
		srpt_ioctl = SRPT_IOC_DISABLE_SVC;
	} else {
		goto usage;
	}

	fd = open(SRPT_NODE, O_RDONLY);
	if (fd < 0) {
		saverr = errno;

		(void) fprintf(stderr, "%s: %s (%s): %s\n", cmdName,
		    gettext("Could not open SRP Target pseudodevice"),
		    SRPT_NODE, strerror(saverr));

		if (saverr == ENOENT) {
			/* avoid having the service go into maintenance */
			if ((stat("/devices/ib", &statbuf)) != 0) {
				(void) fprintf(stderr, "%s: %s\n", cmdName,
				    gettext(
				    "No InfiniBand devices on this system."));
			}
			ret = 0;

		} else {
			ret = SMF_EXIT_ERR_FATAL; /* avoid retries */
		}

		(void) fprintf(stderr, "%s: %s\n", cmdName,
		    gettext("SRPT Driver may not be loaded."));

		return (ret);
	}

	ret = ioctl(fd, srpt_ioctl, NULL);
	if (ret != 0) {
		if (do_enable) {
			txt = gettext("Could not enable SRP Target");
		} else {
			txt = gettext("Could not disable SRP Target");
		}

		(void) fprintf(stderr, "%s: %d", txt, ret);
		ret = SMF_EXIT_ERR_FATAL;
	}

	(void) close(fd);

	return (ret);

usage:
	(void) fprintf(stderr, gettext("Usage: %s start|stop"), cmdName);
	(void) fprintf(stderr, "\n");

	return (1);
}