xref: /onnv-gate/usr/src/cmd/pgrep/psexp.c (revision 0:68f95e015346)
1*0Sstevel@tonic-gate /*
2*0Sstevel@tonic-gate  * CDDL HEADER START
3*0Sstevel@tonic-gate  *
4*0Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*0Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
6*0Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
7*0Sstevel@tonic-gate  * with the License.
8*0Sstevel@tonic-gate  *
9*0Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*0Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
11*0Sstevel@tonic-gate  * See the License for the specific language governing permissions
12*0Sstevel@tonic-gate  * and limitations under the License.
13*0Sstevel@tonic-gate  *
14*0Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
15*0Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*0Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
17*0Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
18*0Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
19*0Sstevel@tonic-gate  *
20*0Sstevel@tonic-gate  * CDDL HEADER END
21*0Sstevel@tonic-gate  */
22*0Sstevel@tonic-gate /*
23*0Sstevel@tonic-gate  * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
24*0Sstevel@tonic-gate  * Use is subject to license terms.
25*0Sstevel@tonic-gate  */
26*0Sstevel@tonic-gate 
27*0Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
28*0Sstevel@tonic-gate 
29*0Sstevel@tonic-gate #include <string.h>
30*0Sstevel@tonic-gate #include <stdlib.h>
31*0Sstevel@tonic-gate #include <alloca.h>
32*0Sstevel@tonic-gate #include <libuutil.h>
33*0Sstevel@tonic-gate 
34*0Sstevel@tonic-gate #include "idtab.h"
35*0Sstevel@tonic-gate #include "psexp.h"
36*0Sstevel@tonic-gate 
37*0Sstevel@tonic-gate void
psexp_create(psexp_t * psexp)38*0Sstevel@tonic-gate psexp_create(psexp_t *psexp)
39*0Sstevel@tonic-gate {
40*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_euids);
41*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_ruids);
42*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_rgids);
43*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_ppids);
44*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_pgids);
45*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_sids);
46*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_ttys);
47*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_projids);
48*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_taskids);
49*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_zoneids);
50*0Sstevel@tonic-gate 	idtab_create(&psexp->ps_ctids);
51*0Sstevel@tonic-gate 
52*0Sstevel@tonic-gate 	psexp->ps_pat = NULL;
53*0Sstevel@tonic-gate }
54*0Sstevel@tonic-gate 
55*0Sstevel@tonic-gate void
psexp_destroy(psexp_t * psexp)56*0Sstevel@tonic-gate psexp_destroy(psexp_t *psexp)
57*0Sstevel@tonic-gate {
58*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_euids);
59*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_ruids);
60*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_rgids);
61*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_ppids);
62*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_pgids);
63*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_sids);
64*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_ttys);
65*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_projids);
66*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_taskids);
67*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_zoneids);
68*0Sstevel@tonic-gate 	idtab_destroy(&psexp->ps_ctids);
69*0Sstevel@tonic-gate 
70*0Sstevel@tonic-gate 	if (psexp->ps_pat)
71*0Sstevel@tonic-gate 		regfree(&psexp->ps_reg);
72*0Sstevel@tonic-gate }
73*0Sstevel@tonic-gate 
74*0Sstevel@tonic-gate int
psexp_compile(psexp_t * psexp)75*0Sstevel@tonic-gate psexp_compile(psexp_t *psexp)
76*0Sstevel@tonic-gate {
77*0Sstevel@tonic-gate 	size_t nbytes;
78*0Sstevel@tonic-gate 	char *buf;
79*0Sstevel@tonic-gate 	int err;
80*0Sstevel@tonic-gate 
81*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_euids);
82*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_ruids);
83*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_rgids);
84*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_ppids);
85*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_pgids);
86*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_sids);
87*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_ttys);
88*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_projids);
89*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_taskids);
90*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_zoneids);
91*0Sstevel@tonic-gate 	idtab_sort(&psexp->ps_ctids);
92*0Sstevel@tonic-gate 
93*0Sstevel@tonic-gate 	if (psexp->ps_pat != NULL) {
94*0Sstevel@tonic-gate 		if ((err = regcomp(&psexp->ps_reg, psexp->ps_pat,
95*0Sstevel@tonic-gate 		    REG_EXTENDED)) != 0) {
96*0Sstevel@tonic-gate 
97*0Sstevel@tonic-gate 			nbytes = regerror(err, &psexp->ps_reg, NULL, 0);
98*0Sstevel@tonic-gate 			buf = alloca(nbytes + 1);
99*0Sstevel@tonic-gate 			(void) regerror(err, &psexp->ps_reg, buf, nbytes);
100*0Sstevel@tonic-gate 			(void) strcat(buf, "\n");
101*0Sstevel@tonic-gate 			uu_warn(buf);
102*0Sstevel@tonic-gate 			return (-1);
103*0Sstevel@tonic-gate 		}
104*0Sstevel@tonic-gate 	}
105*0Sstevel@tonic-gate 
106*0Sstevel@tonic-gate 	return (0);
107*0Sstevel@tonic-gate }
108*0Sstevel@tonic-gate 
109*0Sstevel@tonic-gate #define	NOMATCH(__f1, __f2) \
110*0Sstevel@tonic-gate 	psexp->__f1.id_data && !idtab_search(&psexp->__f1, psinfo->__f2)
111*0Sstevel@tonic-gate 
112*0Sstevel@tonic-gate int
psexp_match(psexp_t * psexp,psinfo_t * psinfo,int flags)113*0Sstevel@tonic-gate psexp_match(psexp_t *psexp, psinfo_t *psinfo, int flags)
114*0Sstevel@tonic-gate {
115*0Sstevel@tonic-gate 	regmatch_t pmatch;
116*0Sstevel@tonic-gate 	const char *s;
117*0Sstevel@tonic-gate 
118*0Sstevel@tonic-gate 	if (NOMATCH(ps_euids, pr_euid))
119*0Sstevel@tonic-gate 		return (0);
120*0Sstevel@tonic-gate 	if (NOMATCH(ps_ruids, pr_uid))
121*0Sstevel@tonic-gate 		return (0);
122*0Sstevel@tonic-gate 	if (NOMATCH(ps_rgids, pr_gid))
123*0Sstevel@tonic-gate 		return (0);
124*0Sstevel@tonic-gate 	if (NOMATCH(ps_ppids, pr_ppid))
125*0Sstevel@tonic-gate 		return (0);
126*0Sstevel@tonic-gate 	if (NOMATCH(ps_pgids, pr_pgid))
127*0Sstevel@tonic-gate 		return (0);
128*0Sstevel@tonic-gate 	if (NOMATCH(ps_sids, pr_sid))
129*0Sstevel@tonic-gate 		return (0);
130*0Sstevel@tonic-gate 	if (NOMATCH(ps_ttys, pr_ttydev))
131*0Sstevel@tonic-gate 		return (0);
132*0Sstevel@tonic-gate 	if (NOMATCH(ps_projids, pr_projid))
133*0Sstevel@tonic-gate 		return (0);
134*0Sstevel@tonic-gate 	if (NOMATCH(ps_taskids, pr_taskid))
135*0Sstevel@tonic-gate 		return (0);
136*0Sstevel@tonic-gate 	if (NOMATCH(ps_zoneids, pr_zoneid))
137*0Sstevel@tonic-gate 		return (0);
138*0Sstevel@tonic-gate 	if (NOMATCH(ps_ctids, pr_contract))
139*0Sstevel@tonic-gate 		return (0);
140*0Sstevel@tonic-gate 
141*0Sstevel@tonic-gate 	if (psexp->ps_pat != NULL) {
142*0Sstevel@tonic-gate 		s = (flags & PSEXP_PSARGS) ?
143*0Sstevel@tonic-gate 		    psinfo->pr_psargs : psinfo->pr_fname;
144*0Sstevel@tonic-gate 
145*0Sstevel@tonic-gate 		if (regexec(&psexp->ps_reg, s, 1, &pmatch, 0) != 0)
146*0Sstevel@tonic-gate 			return (0);
147*0Sstevel@tonic-gate 
148*0Sstevel@tonic-gate 		if ((flags & PSEXP_EXACT) &&
149*0Sstevel@tonic-gate 		    (pmatch.rm_so != 0 || s[pmatch.rm_eo] != '\0'))
150*0Sstevel@tonic-gate 			return (0);
151*0Sstevel@tonic-gate 	}
152*0Sstevel@tonic-gate 
153*0Sstevel@tonic-gate 	return (1);
154*0Sstevel@tonic-gate }
155