ptools/pstop/pstop.c

	pstop.c revision b06cdb87d254343cca2e66a21fd421617c3a0b7b
/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (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 2006 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#pragma ident   "%Z%%M% %I% %E% SMI"

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <signal.h>
#include <libproc.h>

static  int stop(char *);
static  int lwpstop(int *, const lwpstatus_t *, const lwpsinfo_t *);

static  char    *command;
static  const char *lwps;
static  struct  ps_prochandle *P;

int
main(int argc, char **argv)
{
    int rc = 0;

    if ((command = strrchr(argv[0], '/')) != NULL)
        command++;
    else
        command = argv[0];

    if (argc <= 1) {
        (void) fprintf(stderr, "usage:\t%s pid[/lwps] ...\n", command);
        (void) fprintf(stderr,
            "  (stop processes or lwps with /proc request)\n");
        return (2);
    }

    while (--argc > 0)
        rc += stop(*++argv);

    return (rc);
}

static int
stop(char *arg)
{
    int gcode;
    int rc = 0;

    if ((P = proc_arg_xgrab(arg, NULL, PR_ARG_PIDS, PGRAB_RETAIN |
        PGRAB_NOSTOP | PGRAB_FORCE, &gcode, &lwps)) == NULL) {
        (void) fprintf(stderr, "%s: cannot control %s: %s\n",
            command, arg, Pgrab_error(gcode));
        return (1);
    } else if (lwps != NULL) {
        /*
         * The user has provided an lwp specification.  Let's consider
         * the lwp specification as a mask.  We iterate over all lwps in
         * the process and stop every lwp, which matches the mask.  If
         * there is no lwp matching the mask or an error occured during
         * the iteration, set the return code to 1 as indication of an
         * error.
         */
        int lwpcount = 0;

        (void) Plwp_iter_all(P, (proc_lwp_all_f *)lwpstop, &lwpcount);
        if (lwpcount == 0) {
            (void) fprintf(stderr, "%s: cannot control %s:"
                " no matching LWPs found\n", command, arg);
            rc = 1;
        } else if (lwpcount == -1)
            rc = 1;
    } else {
        (void) Pdstop(P);   /* Stop the process. */
    }

    /*
     * Prelease could change the tracing flags, use Pfree and unset
     * run-on-last-close flag to prevent the process being set running
     * after detaching from it.
     */
    (void) Punsetflags(P, PR_RLC);
    Pfree(P);
    return (rc);
}

/* ARGSUSED */
static int
lwpstop(int *lwpcount, const lwpstatus_t *status, const lwpsinfo_t *info)
{
    struct ps_lwphandle *L;
    int gcode;

    if (proc_lwp_in_set(lwps, info->pr_lwpid)) {
        /*
         * There is a race between the callback from the iterator and
         * grabbing of the lwp.  If the lwp has already exited, Lgrab
         * will return the error code G_NOPROC.  It's not a real error,
         * only if there is no lwp matching the specification.
         */
        if ((L = Lgrab(P, info->pr_lwpid, &gcode)) != NULL) {
            (void) Ldstop(L);
            Lfree(L);
            if (*lwpcount >= 0)
                (*lwpcount)++;
        } else if (gcode != G_NOPROC) {
            (void) fprintf(stderr, "%s: cannot control %d/%d: %s\n",
                command, (int)Pstatus(P)->pr_pid,
                (int)info->pr_lwpid, Lgrab_error(gcode));
            *lwpcount = -1;
        }
    }
    return (0);
}