|
root / base / usr / src / cmd / srptsvc / srptsvc.c
srptsvc.c C 118 lines 2.7 KB
  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
/*
 * 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);
}