xref: /illumos-gate/usr/src/cmd/tsol/plabel/plabel.c (revision f875b4eb)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 #pragma ident	"%Z%%M%	%I%	%E% SMI"
28 
29 /*
30  *	plabel - gets process label.
31  */
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <errno.h>
35 #include <unistd.h>
36 #include <fcntl.h>
37 #include <string.h>
38 #include <locale.h>
39 #include <procfs.h>
40 #include <sys/proc.h>
41 #include <zone.h>
42 
43 #include <sys/tsol/label_macro.h>
44 
45 #include <tsol/label.h>
46 
47 #define	s_flag	0x04
48 #define	S_flag	0x08
49 
50 #define	INIT_ALLOC_LEN	1024
51 #define	MAX_ALLOC_NUM	11
52 
53 static int look(char *);
54 static int perr(char *);
55 static void usage(void);
56 
57 static char procname[64];
58 
59 static unsigned int opt_flag = 0;
60 static char *cmd = NULL;
61 
62 int
main(int argc,char ** argv)63 main(int argc, char **argv)
64 {
65 	int err, rc = 0;
66 	int opt;
67 
68 	(void) setlocale(LC_ALL, "");
69 #if !defined(TEXT_DOMAIN)	/* Should be defined by cc -D */
70 #define	TEXT_DOMAIN	"SYS_TEST"	/* Use this only if it weren't */
71 #endif
72 	(void) textdomain(TEXT_DOMAIN);
73 
74 	if ((cmd = strrchr(argv[0], '/')) == NULL)
75 		cmd = argv[0];
76 	else
77 		cmd++;
78 
79 	/* Error if labeling is not active. */
80 	if (!is_system_labeled()) {
81 		(void) fprintf(stderr,
82 		    gettext("%s: Trusted Extensions must be enabled\n"), cmd);
83 		return (1);
84 	}
85 
86 	while ((opt = getopt(argc, argv, "sS")) != EOF) {
87 		switch (opt) {
88 		case 's':
89 			if (opt_flag & (s_flag | S_flag)) {
90 				usage();
91 				return (1);
92 			}
93 			opt_flag |= s_flag;
94 			break;
95 
96 		case 'S':
97 			if (opt_flag & (s_flag | S_flag)) {
98 				usage();
99 				return (1);
100 			}
101 			opt_flag |= S_flag;
102 			break;
103 		default:
104 			usage();
105 			return (1);
106 		}
107 	}
108 
109 	argc -= optind;
110 	argv += optind;
111 	if (argc == 0) {
112 		char pid[11]; /* 32 bit pids go to 4294967295 plus a NUL */
113 
114 		(void) sprintf(pid, "%d", (int)getpid());
115 		rc = look(pid);
116 	} else {
117 		while (argc-- > 0) {
118 			err = look(*argv++);
119 			if (rc == 0)
120 				rc = err;
121 		}
122 	}
123 	return (rc);
124 }
125 
126 static int
look(char * arg)127 look(char *arg)
128 {
129 	int fd;
130 	m_label_t *plabel;
131 	psinfo_t info;		/* process information from /proc */
132 	char *str;
133 	int wordlen = DEF_NAMES;
134 
135 	if (opt_flag == S_flag)
136 		wordlen = LONG_NAMES;
137 	else if (opt_flag == s_flag)
138 		wordlen = SHORT_NAMES;
139 
140 	if (strchr(arg, '/') != NULL)
141 		(void) strncpy(procname, arg, sizeof (procname));
142 	else {
143 		(void) strcpy(procname, "/proc/");
144 		(void) strncat(procname, arg,
145 		    sizeof (procname) - strlen(procname));
146 	}
147 	(void) strlcat(procname, "/psinfo", sizeof (procname)
148 	    - strlen(procname));
149 
150 	/*
151 	 * Open the process to be examined.
152 	 */
153 retry:
154 	if ((fd = open(procname, O_RDONLY)) < 0) {
155 		/*
156 		 * Make clean message for non-existent process.
157 		 */
158 		if (errno == ENOENT) {
159 			errno = ESRCH;
160 			perror(arg);
161 			return (1);
162 		}
163 		return (perr(NULL));
164 	}
165 
166 
167 	/*
168 	 * Get the info structure for the process and close quickly.
169 	 */
170 	if (read(fd, &info, sizeof (info)) < 0) {
171 		int	saverr = errno;
172 
173 		(void) close(fd);
174 		if (saverr == EAGAIN)
175 			goto retry;
176 		if (saverr != ENOENT)
177 			perror(arg);
178 		return (1);
179 	}
180 	(void) close(fd);
181 
182 	if (info.pr_lwp.pr_state == 0)  /* can't happen? */
183 		return (1);
184 
185 	if ((plabel = getzonelabelbyid(info.pr_zoneid)) == NULL) {
186 		return (1);
187 	}
188 
189 	/*
190 	 * The process label for global zone is admin_high
191 	 */
192 	if (info.pr_zoneid == GLOBAL_ZONEID) {
193 		_BSLHIGH(plabel);
194 	}
195 
196 	if (label_to_str(plabel, &str, M_LABEL, wordlen) != 0) {
197 		perror(arg);
198 		return (2);
199 	}
200 	(void) printf("%s\n", str);
201 	m_label_free(plabel);
202 	free(str);
203 	return (0);
204 }
205 
206 
207 /*
208  * usage()
209  *
210  * This routine is called whenever there is a usage type of error has
211  * occured.  For example, when a invalid option has has been specified.
212  *
213  */
214 static void
usage(void)215 usage(void)
216 {
217 
218 	(void) fprintf(stderr, "Usage: \n");
219 	(void) fprintf(stderr,
220 	    gettext("	%s [pid ...]    \n"), cmd);
221 	(void) fprintf(stderr,
222 	    gettext("	%s -s  [pid ...] \n"), cmd);
223 	(void) fprintf(stderr,
224 	    gettext("	%s -S  [pid ...] \n"), cmd);
225 }
226 
227 static int
perr(char * s)228 perr(char *s) {
229 
230 	if (s)
231 		(void) fprintf(stderr, "%s: ", procname);
232 	else
233 		s = procname;
234 	perror(s);
235 	return (1);
236 }
237