1*7c478bd9Sstevel@tonic-gate /*
2*7c478bd9Sstevel@tonic-gate * CDDL HEADER START
3*7c478bd9Sstevel@tonic-gate *
4*7c478bd9Sstevel@tonic-gate * The contents of this file are subject to the terms of the
5*7c478bd9Sstevel@tonic-gate * Common Development and Distribution License, Version 1.0 only
6*7c478bd9Sstevel@tonic-gate * (the "License"). You may not use this file except in compliance
7*7c478bd9Sstevel@tonic-gate * with the License.
8*7c478bd9Sstevel@tonic-gate *
9*7c478bd9Sstevel@tonic-gate * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*7c478bd9Sstevel@tonic-gate * or http://www.opensolaris.org/os/licensing.
11*7c478bd9Sstevel@tonic-gate * See the License for the specific language governing permissions
12*7c478bd9Sstevel@tonic-gate * and limitations under the License.
13*7c478bd9Sstevel@tonic-gate *
14*7c478bd9Sstevel@tonic-gate * When distributing Covered Code, include this CDDL HEADER in each
15*7c478bd9Sstevel@tonic-gate * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*7c478bd9Sstevel@tonic-gate * If applicable, add the following below this CDDL HEADER, with the
17*7c478bd9Sstevel@tonic-gate * fields enclosed by brackets "[]" replaced with your own identifying
18*7c478bd9Sstevel@tonic-gate * information: Portions Copyright [yyyy] [name of copyright owner]
19*7c478bd9Sstevel@tonic-gate *
20*7c478bd9Sstevel@tonic-gate * CDDL HEADER END
21*7c478bd9Sstevel@tonic-gate */
22*7c478bd9Sstevel@tonic-gate /*
23*7c478bd9Sstevel@tonic-gate * Copyright 2004 Sun Microsystems, Inc. All rights reserved.
24*7c478bd9Sstevel@tonic-gate * Use is subject to license terms.
25*7c478bd9Sstevel@tonic-gate */
26*7c478bd9Sstevel@tonic-gate
27*7c478bd9Sstevel@tonic-gate #pragma ident "%Z%%M% %I% %E% SMI"
28*7c478bd9Sstevel@tonic-gate
29*7c478bd9Sstevel@tonic-gate /*
30*7c478bd9Sstevel@tonic-gate * wracct - write system accounting records
31*7c478bd9Sstevel@tonic-gate */
32*7c478bd9Sstevel@tonic-gate
33*7c478bd9Sstevel@tonic-gate #include <sys/types.h>
34*7c478bd9Sstevel@tonic-gate #include <sys/procset.h>
35*7c478bd9Sstevel@tonic-gate #include <exacct.h>
36*7c478bd9Sstevel@tonic-gate #include <stdio.h>
37*7c478bd9Sstevel@tonic-gate #include <stdlib.h>
38*7c478bd9Sstevel@tonic-gate #include <libintl.h>
39*7c478bd9Sstevel@tonic-gate #include <strings.h>
40*7c478bd9Sstevel@tonic-gate #include <errno.h>
41*7c478bd9Sstevel@tonic-gate
42*7c478bd9Sstevel@tonic-gate #define CMDNAME "wracct"
43*7c478bd9Sstevel@tonic-gate #define USAGE \
44*7c478bd9Sstevel@tonic-gate gettext("Usage: wracct -i id_list [-t partial | interval ] " \
45*7c478bd9Sstevel@tonic-gate "{process | task}\n")
46*7c478bd9Sstevel@tonic-gate #define OPTIONS_STRING "i:t:"
47*7c478bd9Sstevel@tonic-gate
48*7c478bd9Sstevel@tonic-gate static void
usage(void)49*7c478bd9Sstevel@tonic-gate usage(void)
50*7c478bd9Sstevel@tonic-gate {
51*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr, USAGE);
52*7c478bd9Sstevel@tonic-gate exit(2);
53*7c478bd9Sstevel@tonic-gate }
54*7c478bd9Sstevel@tonic-gate
55*7c478bd9Sstevel@tonic-gate static long
Atol(char * p)56*7c478bd9Sstevel@tonic-gate Atol(char *p)
57*7c478bd9Sstevel@tonic-gate {
58*7c478bd9Sstevel@tonic-gate long l;
59*7c478bd9Sstevel@tonic-gate char *q;
60*7c478bd9Sstevel@tonic-gate errno = 0;
61*7c478bd9Sstevel@tonic-gate l = strtol(p, &q, 10);
62*7c478bd9Sstevel@tonic-gate if (errno != 0 || q == p || l < 0 || *q != '\0') {
63*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr, gettext("%s: illegal argument -- %s\n"),
64*7c478bd9Sstevel@tonic-gate CMDNAME, p);
65*7c478bd9Sstevel@tonic-gate exit(2);
66*7c478bd9Sstevel@tonic-gate /*NOTREACHED*/
67*7c478bd9Sstevel@tonic-gate } else {
68*7c478bd9Sstevel@tonic-gate return (l);
69*7c478bd9Sstevel@tonic-gate }
70*7c478bd9Sstevel@tonic-gate }
71*7c478bd9Sstevel@tonic-gate
72*7c478bd9Sstevel@tonic-gate int
main(int argc,char * argv[])73*7c478bd9Sstevel@tonic-gate main(int argc, char *argv[])
74*7c478bd9Sstevel@tonic-gate {
75*7c478bd9Sstevel@tonic-gate idtype_t ent_flag = -1;
76*7c478bd9Sstevel@tonic-gate int rec_flag = EW_PARTIAL;
77*7c478bd9Sstevel@tonic-gate id_t id = -1;
78*7c478bd9Sstevel@tonic-gate char *id_sequence = NULL;
79*7c478bd9Sstevel@tonic-gate char *idlp = NULL;
80*7c478bd9Sstevel@tonic-gate int c, r;
81*7c478bd9Sstevel@tonic-gate
82*7c478bd9Sstevel@tonic-gate while ((c = getopt(argc, argv, OPTIONS_STRING)) != EOF) {
83*7c478bd9Sstevel@tonic-gate switch (c) {
84*7c478bd9Sstevel@tonic-gate case 't':
85*7c478bd9Sstevel@tonic-gate if (strcmp(optarg, "partial") == 0) {
86*7c478bd9Sstevel@tonic-gate rec_flag = EW_PARTIAL;
87*7c478bd9Sstevel@tonic-gate } else if (strcmp(optarg, "interval") == 0) {
88*7c478bd9Sstevel@tonic-gate rec_flag = EW_INTERVAL;
89*7c478bd9Sstevel@tonic-gate } else {
90*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr,
91*7c478bd9Sstevel@tonic-gate gettext("%s: wrong record type\n"),
92*7c478bd9Sstevel@tonic-gate CMDNAME);
93*7c478bd9Sstevel@tonic-gate usage();
94*7c478bd9Sstevel@tonic-gate }
95*7c478bd9Sstevel@tonic-gate break;
96*7c478bd9Sstevel@tonic-gate case 'i':
97*7c478bd9Sstevel@tonic-gate id_sequence = strdup(optarg);
98*7c478bd9Sstevel@tonic-gate break;
99*7c478bd9Sstevel@tonic-gate case '?':
100*7c478bd9Sstevel@tonic-gate default:
101*7c478bd9Sstevel@tonic-gate usage();
102*7c478bd9Sstevel@tonic-gate break;
103*7c478bd9Sstevel@tonic-gate }
104*7c478bd9Sstevel@tonic-gate }
105*7c478bd9Sstevel@tonic-gate
106*7c478bd9Sstevel@tonic-gate if (optind >= argc) {
107*7c478bd9Sstevel@tonic-gate usage();
108*7c478bd9Sstevel@tonic-gate }
109*7c478bd9Sstevel@tonic-gate
110*7c478bd9Sstevel@tonic-gate if (strcmp(argv[optind], "task") == 0) {
111*7c478bd9Sstevel@tonic-gate ent_flag = P_TASKID;
112*7c478bd9Sstevel@tonic-gate } else if (strcmp(argv[optind], "process") == 0 ||
113*7c478bd9Sstevel@tonic-gate strcmp(argv[optind], "proc") == 0) {
114*7c478bd9Sstevel@tonic-gate ent_flag = P_PID;
115*7c478bd9Sstevel@tonic-gate } else {
116*7c478bd9Sstevel@tonic-gate usage();
117*7c478bd9Sstevel@tonic-gate }
118*7c478bd9Sstevel@tonic-gate
119*7c478bd9Sstevel@tonic-gate if (ent_flag == P_PID && rec_flag == EW_INTERVAL) {
120*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr,
121*7c478bd9Sstevel@tonic-gate gettext("%s: interval process records not supported\n"),
122*7c478bd9Sstevel@tonic-gate CMDNAME);
123*7c478bd9Sstevel@tonic-gate exit(2);
124*7c478bd9Sstevel@tonic-gate }
125*7c478bd9Sstevel@tonic-gate
126*7c478bd9Sstevel@tonic-gate if (id_sequence == NULL) {
127*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr,
128*7c478bd9Sstevel@tonic-gate gettext("%s: please use -i option to specify ids\n"),
129*7c478bd9Sstevel@tonic-gate CMDNAME);
130*7c478bd9Sstevel@tonic-gate exit(2);
131*7c478bd9Sstevel@tonic-gate }
132*7c478bd9Sstevel@tonic-gate
133*7c478bd9Sstevel@tonic-gate for (idlp = strtok(id_sequence, ", \t\n"); idlp != NULL;
134*7c478bd9Sstevel@tonic-gate idlp = strtok(NULL, ", \t\n")) {
135*7c478bd9Sstevel@tonic-gate id = Atol(idlp);
136*7c478bd9Sstevel@tonic-gate if (wracct(ent_flag, id, rec_flag) < 0) {
137*7c478bd9Sstevel@tonic-gate r = errno;
138*7c478bd9Sstevel@tonic-gate (void) fprintf(stderr,
139*7c478bd9Sstevel@tonic-gate "%s: operation failed on %s %d: %s\n",
140*7c478bd9Sstevel@tonic-gate CMDNAME, (ent_flag == P_TASKID) ? "taskid" : "pid",
141*7c478bd9Sstevel@tonic-gate (int)id, strerror(r));
142*7c478bd9Sstevel@tonic-gate exit(1);
143*7c478bd9Sstevel@tonic-gate }
144*7c478bd9Sstevel@tonic-gate }
145*7c478bd9Sstevel@tonic-gate
146*7c478bd9Sstevel@tonic-gate return (0);
147*7c478bd9Sstevel@tonic-gate }
148