xref: /titanic_52/usr/src/cmd/rcap/common/utils.c (revision 7c478bd95313f5f23a4c958a745db2134aa03244)

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

#pragma ident	"%Z%%M%	%I%	%E% SMI"

#include <sys/param.h>
#include <libintl.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <syslog.h>
#include <unistd.h>
#include <errno.h>

#include "utils.h"

static char ERRNO_FMT[] = ": %s";

static char *pname = NULL;
static rcm_level_t message_priority = RCM_WARN;
static rcm_dst_t message_dst = RCD_STD;

static void dmesg(int level, char *msg);

/*PRINTFLIKE2*/
void
dprintfe(int level, char *format, ...)
{
	va_list alist;

	va_start(alist, format);
	vdprintfe(level, format, alist);
	va_end(alist);
}

/*PRINTFLIKE2*/
void
vdprintfe(int level, char *format, va_list alist)
{
	char buf[LINELEN];
	char *c;
	int err = errno;

	*buf = 0;

	if ((strlen(buf) + 1) < LINELEN)
		(void) vsnprintf(buf + strlen(buf), LINELEN - 1 - strlen(buf),
		    format, alist);
	if ((c = strchr(buf, '\n')) == NULL) {
		if ((strlen(buf) + 1) < LINELEN)
			(void) snprintf(buf + strlen(buf), LINELEN - 1 -
			    strlen(buf), gettext(ERRNO_FMT), strerror(err));
	} else
		*c = 0;

	dmesg(level, buf);
}

#ifdef DEBUG_MSG
/*PRINTFLIKE1*/
void
debug(char *format, ...)
{
	va_list alist;

	if (get_message_priority() < RCM_DEBUG)
		return;

	va_start(alist, format);
	vdprintfe(RCM_DEBUG, format, alist);
	va_end(alist);
}

/*PRINTFLIKE1*/
void
debug_high(char *format, ...)
{
	va_list alist;

	if (get_message_priority() < RCM_DEBUG_HIGH)
		return;

	va_start(alist, format);
	vdprintfe(RCM_DEBUG_HIGH, format, alist);
	va_end(alist);
}
#endif /* DEBUG_MSG */

/*PRINTFLIKE1*/
void
warn(char *format, ...)
{
	va_list alist;

	if (get_message_priority() < RCM_WARN)
		return;

	va_start(alist, format);
	vdprintfe(RCM_WARN, format, alist);
	va_end(alist);
}

/*PRINTFLIKE1*/
void
die(char *format, ...)
{
	va_list alist;

	if (get_message_priority() < RCM_ERR)
		return;

	va_start(alist, format);
	vdprintfe(RCM_ERR, format, alist);
	va_end(alist);

	exit(E_ERROR);
}

/*PRINTFLIKE1*/
void
info(char *format, ...)
{
	va_list alist;

	if (get_message_priority() < RCM_INFO)
		return;

	va_start(alist, format);
	vdprintfe(RCM_INFO, format, alist);
	va_end(alist);
}

char *
setprogname(char *arg0)
{
	char *p = strrchr(arg0, '/');

	if (p == NULL)
		p = arg0;
	else
		p++;
	pname = p;
	return (pname);
}

/*
 * Output a message to the controlling tty or log, depending on which is
 * configured.  The message should contain no newlines.
 */
static void
dmesg(int level, char *msg)
{
	if (message_priority >= level) {
		FILE *fp;
		int syslog_severity = -1;

		switch (message_dst) {
		case RCD_STD:
			fp = level >= RCM_DEBUG ? stderr : stdout;

			if (pname != NULL) {
				(void) fputs(pname, fp);
				(void) fputs(": ", fp);
			}
			(void) fputs(msg, fp);
			(void) fputc('\n', fp);
			(void) fflush(fp);
			break;
		case RCD_SYSLOG:
			switch (level) {
			case RCM_ERR:
				syslog_severity = LOG_ERR;
				break;
			case RCM_WARN:
				syslog_severity = LOG_WARNING;
				break;
			case RCM_INFO:
				syslog_severity = LOG_INFO;
				break;
			case RCM_DEBUG:
				syslog_severity = LOG_DEBUG;
				break;
			}
			if (syslog_severity >= 0)
				(void) syslog(syslog_severity, "%s", msg);
			break;
		}
	}
}

rcm_level_t
get_message_priority(void)
{
	return (message_priority);
}

rcm_level_t
set_message_priority(rcm_level_t new_priority)
{
	rcm_level_t old_priority = message_priority;

	message_priority = new_priority;
	return (old_priority);
}

rcm_dst_t
set_message_destination(rcm_dst_t new_dst)
{
	rcm_dst_t old_dst = message_dst;

	if ((message_dst = new_dst) == RCD_SYSLOG)
		openlog(pname, LOG_ODELAY | LOG_PID, LOG_DAEMON);

	return (old_dst);
}

void
hrt2ts(hrtime_t hrt, timestruc_t *tsp)
{
	tsp->tv_sec = hrt / NANOSEC;
	tsp->tv_nsec = hrt % NANOSEC;
}

int
xatoi(char *p)
{
	int i;
	char *q;

	errno = 0;
	i = (int)strtol(p, &q, 10);
	if (errno != 0 || q == p || i < 0 || *q != '\0') {
		warn(gettext("illegal argument -- %s\n"), p);
		return (-1);
	} else {
		return (i);
	}
}