print/bsd-sysv-commands/disable.c

/*
 * 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 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 *
 */

/* $Id: disable.c 146 2006-03-24 00:26:54Z njacobs $ */


#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <locale.h>
#include <libintl.h>
#include <papi.h>
#include "common.h"

static void
usage(char *program)
{
    char *name;

    if ((name = strrchr(program, '/')) == NULL)
        name = program;
    else
        name++;

    fprintf(stdout,
        gettext("Usage: %s [-c] [-W] [-r reason] destination ...\n"),
        name);
    exit(1);
}

static void
cancel_active_job(papi_service_t svc, char *dest)
{
    papi_status_t status;
    papi_job_t *j = NULL;
    char *req_attrs[] = { "job-state", "job-id", NULL };

    status = papiPrinterListJobs(svc, dest, req_attrs, 0, 0, &j);
    if ((status == PAPI_OK) && (j != NULL)) {
        int i;

        for (i = 0; j[i] != NULL; j++) {
            papi_attribute_t **a = papiJobGetAttributeList(j[i]);
            int state = 0;

            if (a == NULL)
                continue;

            (void) papiAttributeListGetInteger(a, NULL,
                "job-state", &state);
            if (state & 0x082A) { /* If state is RS_ACTIVE */
                int32_t id = papiJobGetId(j[i]);

                (void) papiJobCancel(svc, dest, id);
            }
        }
        papiJobListFree(j);
    }
}

int
main(int ac, char *av[])
{
    papi_status_t status;
    papi_service_t svc = NULL;
    papi_encryption_t encryption = PAPI_ENCRYPT_NEVER;
    int exit_status = 0;
    int cancel = 0;
    int pending = 0;    /* not implemented */
    char *reason = NULL;
    int c;

    (void) setlocale(LC_ALL, "");
    (void) textdomain("SUNW_OST_OSCMD");

    while ((c = getopt(ac, av, "EcWr:")) != EOF)
        switch (c) {
        case 'c':   /* cancel active job first */
            cancel = 1;
            break;
        case 'W':   /* wait for active request, not implemented */
            pending = 1;
            break;
        case 'r':   /* reason */
            reason = optarg;
            break;
        case 'E':
            encryption = PAPI_ENCRYPT_NEVER;
            break;
        default:
            usage(av[0]);
        }

    if (ac <= optind)
        usage(av[0]);

    while (optind < ac) {
        char *printer = av[optind++];

        status = papiServiceCreate(&svc, printer, NULL, NULL,
            cli_auth_callback, encryption, NULL);
        if (status != PAPI_OK) {
            fprintf(stderr, gettext(
                "Failed to contact service for %s: %s\n"),
                printer, verbose_papi_message(svc, status));
            exit_status = 1;
        }

        status = papiPrinterDisable(svc, printer, reason);
        if (status == PAPI_OK) {
            printf(gettext("printer \"%s\" now disabled\n"),
                printer);
        } else if (status == PAPI_NOT_ACCEPTING) {
            fprintf(stderr, gettext(
                "Destination \"%s\" was already disabled.\n"),
                printer);
            exit_status = 1;
        } else {
            /* The operation is not supported in lpd protocol */
            if (status == PAPI_OPERATION_NOT_SUPPORTED) {
                fprintf(stderr,
                    verbose_papi_message(svc, status));
            } else {
                fprintf(stderr, gettext("disable: %s: %s\n"),
                    printer, verbose_papi_message(svc, status));
            }
            exit_status = 1;
        }

        if (cancel != 0)
            cancel_active_job(svc, printer);

        papiServiceDestroy(svc);
    }

    return (exit_status);
}