OpenSolaris Launch

2 files changed, 193 insertions, 0 deletions

diff --git a/usr/src/cmd/wracct/Makefile b/usr/src/cmd/wracct/Makefile
new file mode 100644
index 0000000000..f03ecfdd52
--- /dev/null
+++ b/usr/src/cmd/wracct/Makefile
@@ -0,0 +1,46 @@
+#
+# 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 2004 Sun Microsystems, Inc.  All rights reserved.
+# Use is subject to license terms.
+#
+#ident	"%Z%%M%	%I%	%E% SMI"
+#
+# cmd/wracct/Makefile
+
+PROG=	wracct
+
+include ../Makefile.cmd
+
+.KEEP_STATE:
+
+CFLAGS += $(CCVERBOSE)
+
+all: $(PROG)
+
+install: all $(ROOTPROG)
+
+clean:
+
+lint: lint_PROG
+
+include ../Makefile.targ
diff --git a/usr/src/cmd/wracct/wracct.c b/usr/src/cmd/wracct/wracct.c
new file mode 100644
index 0000000000..0715ccf86a
--- /dev/null
+++ b/usr/src/cmd/wracct/wracct.c
@@ -0,0 +1,147 @@
+/*
+ * 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 2004 Sun Microsystems, Inc.  All rights reserved.
+ * Use is subject to license terms.
+ */
+
+#pragma ident	"%Z%%M%	%I%	%E% SMI"
+
+/*
+ * wracct - write system accounting records
+ */
+
+#include <sys/types.h>
+#include <sys/procset.h>
+#include <exacct.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <libintl.h>
+#include <strings.h>
+#include <errno.h>
+
+#define	CMDNAME "wracct"
+#define	USAGE \
+    gettext("Usage: wracct -i id_list [-t partial | interval ] " \
+	"{process | task}\n")
+#define	OPTIONS_STRING "i:t:"
+
+static void
+usage(void)
+{
+	(void) fprintf(stderr, USAGE);
+	exit(2);
+}
+
+static long
+Atol(char *p)
+{
+	long l;
+	char *q;
+	errno = 0;
+	l = strtol(p, &q, 10);
+	if (errno != 0 || q == p || l < 0 || *q != '\0') {
+		(void) fprintf(stderr, gettext("%s: illegal argument -- %s\n"),
+		    CMDNAME, p);
+		exit(2);
+		/*NOTREACHED*/
+	} else {
+		return (l);
+	}
+}
+
+int
+main(int argc, char *argv[])
+{
+	idtype_t ent_flag = -1;
+	int rec_flag = EW_PARTIAL;
+	id_t id = -1;
+	char *id_sequence = NULL;
+	char *idlp = NULL;
+	int c, r;
+
+	while ((c = getopt(argc, argv, OPTIONS_STRING)) != EOF) {
+		switch (c) {
+			case 't':
+				if (strcmp(optarg, "partial") == 0) {
+					rec_flag = EW_PARTIAL;
+				} else if (strcmp(optarg, "interval") == 0) {
+					rec_flag = EW_INTERVAL;
+				} else {
+					(void) fprintf(stderr,
+					    gettext("%s: wrong record type\n"),
+					    CMDNAME);
+					usage();
+				}
+				break;
+			case 'i':
+				id_sequence = strdup(optarg);
+				break;
+			case '?':
+			default:
+				usage();
+				break;
+		}
+	}
+
+	if (optind >= argc) {
+		usage();
+	}
+
+	if (strcmp(argv[optind], "task") == 0) {
+		ent_flag = P_TASKID;
+	} else if (strcmp(argv[optind], "process") == 0 ||
+	    strcmp(argv[optind], "proc") == 0) {
+		ent_flag = P_PID;
+	} else {
+		usage();
+	}
+
+	if (ent_flag == P_PID && rec_flag == EW_INTERVAL) {
+		(void) fprintf(stderr,
+		    gettext("%s: interval process records not supported\n"),
+		    CMDNAME);
+		exit(2);
+	}
+
+	if (id_sequence == NULL) {
+		(void) fprintf(stderr,
+		    gettext("%s: please use -i option to specify ids\n"),
+		    CMDNAME);
+		exit(2);
+	}
+
+	for (idlp = strtok(id_sequence, ", \t\n"); idlp != NULL;
+	    idlp = strtok(NULL, ", \t\n")) {
+		id = Atol(idlp);
+		if (wracct(ent_flag, id, rec_flag) < 0) {
+			r = errno;
+			(void) fprintf(stderr,
+			    "%s: operation failed on %s %d: %s\n",
+			    CMDNAME, (ent_flag == P_TASKID) ? "taskid" : "pid",
+			    (int)id, strerror(r));
+			exit(1);
+		}
+	}
+
+	return (0);
+}