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 2003 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 <stdio.h>
30*0Sstevel@tonic-gate #include <stdlib.h>
31*0Sstevel@tonic-gate #include <unistd.h>
32*0Sstevel@tonic-gate #include <fcntl.h>
33*0Sstevel@tonic-gate #include <string.h>
34*0Sstevel@tonic-gate #include <errno.h>
35*0Sstevel@tonic-gate #include <sys/types.h>
36*0Sstevel@tonic-gate #include <signal.h>
37*0Sstevel@tonic-gate #include <libproc.h>
38*0Sstevel@tonic-gate 
39*0Sstevel@tonic-gate static	int	start(char *);
40*0Sstevel@tonic-gate 
41*0Sstevel@tonic-gate static	char	*command;
42*0Sstevel@tonic-gate 
43*0Sstevel@tonic-gate int
44*0Sstevel@tonic-gate main(int argc, char **argv)
45*0Sstevel@tonic-gate {
46*0Sstevel@tonic-gate 	int rc = 0;
47*0Sstevel@tonic-gate 
48*0Sstevel@tonic-gate 	if ((command = strrchr(argv[0], '/')) != NULL)
49*0Sstevel@tonic-gate 		command++;
50*0Sstevel@tonic-gate 	else
51*0Sstevel@tonic-gate 		command = argv[0];
52*0Sstevel@tonic-gate 
53*0Sstevel@tonic-gate 	if (argc <= 1) {
54*0Sstevel@tonic-gate 		(void) fprintf(stderr, "usage:\t%s pid ...\n", command);
55*0Sstevel@tonic-gate 		(void) fprintf(stderr, "  (set stopped processes running)\n");
56*0Sstevel@tonic-gate 		return (2);
57*0Sstevel@tonic-gate 	}
58*0Sstevel@tonic-gate 
59*0Sstevel@tonic-gate 	while (--argc > 0)
60*0Sstevel@tonic-gate 		rc += start(*++argv);
61*0Sstevel@tonic-gate 
62*0Sstevel@tonic-gate 	return (rc);
63*0Sstevel@tonic-gate }
64*0Sstevel@tonic-gate 
65*0Sstevel@tonic-gate static int
66*0Sstevel@tonic-gate start(char *arg)
67*0Sstevel@tonic-gate {
68*0Sstevel@tonic-gate 	struct ps_prochandle *P;
69*0Sstevel@tonic-gate 	int gcode;
70*0Sstevel@tonic-gate 
71*0Sstevel@tonic-gate 	if ((P = proc_arg_grab(arg, PR_ARG_PIDS,
72*0Sstevel@tonic-gate 	    PGRAB_FORCE | PGRAB_RETAIN | PGRAB_NOSTOP, &gcode)) == NULL) {
73*0Sstevel@tonic-gate 		(void) fprintf(stderr, "%s: cannot control %s: %s\n",
74*0Sstevel@tonic-gate 			command, arg, Pgrab_error(gcode));
75*0Sstevel@tonic-gate 		return (1);
76*0Sstevel@tonic-gate 	}
77*0Sstevel@tonic-gate 
78*0Sstevel@tonic-gate 	/*
79*0Sstevel@tonic-gate 	 * If the victim is stopped because of a job control signal, we
80*0Sstevel@tonic-gate 	 * need to send it SIGCONT to get it moving again, otherwise
81*0Sstevel@tonic-gate 	 * the agent will not be able to run, and so will not be able to
82*0Sstevel@tonic-gate 	 * exit the process.
83*0Sstevel@tonic-gate 	 */
84*0Sstevel@tonic-gate 	(void) kill(Pstatus(P)->pr_pid, SIGCONT);
85*0Sstevel@tonic-gate 
86*0Sstevel@tonic-gate 	/*
87*0Sstevel@tonic-gate 	 * if the agent already exists, Pcreate_agent will adopt the
88*0Sstevel@tonic-gate 	 * extant agent so that we can destroy it
89*0Sstevel@tonic-gate 	 */
90*0Sstevel@tonic-gate 	if (Pstatus(P)->pr_lwp.pr_flags & PR_AGENT) {
91*0Sstevel@tonic-gate 		if (Pcreate_agent(P) != 0) {
92*0Sstevel@tonic-gate 			(void) fprintf(stderr,
93*0Sstevel@tonic-gate 			    "%s: cannot remove agent from %s: %s\n",
94*0Sstevel@tonic-gate 			    command, arg, strerror(errno));
95*0Sstevel@tonic-gate 
96*0Sstevel@tonic-gate 			Prelease(P, 0);
97*0Sstevel@tonic-gate 			return (1);
98*0Sstevel@tonic-gate 		}
99*0Sstevel@tonic-gate 
100*0Sstevel@tonic-gate 		Pdestroy_agent(P);
101*0Sstevel@tonic-gate 	}
102*0Sstevel@tonic-gate 
103*0Sstevel@tonic-gate 	/*
104*0Sstevel@tonic-gate 	 * Prelease could change the tracing flags or leave the victim hung
105*0Sstevel@tonic-gate 	 * so we set the process running and free the handle by hand.
106*0Sstevel@tonic-gate 	 */
107*0Sstevel@tonic-gate 	(void) Psetrun(P, 0, 0);
108*0Sstevel@tonic-gate 	Pfree(P);
109*0Sstevel@tonic-gate 
110*0Sstevel@tonic-gate 	return (0);
111*0Sstevel@tonic-gate }
112