/*
 * 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.
 */


/*LINTLIBRARY*/

#include <stdio.h>
#include <stdarg.h>
#include <libintl.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>


/* lpsched include files */
#include "lp.h"
#include "msgs.h"
#include "printers.h"
#include "class.h"

#include <papi_impl.h>


/*
 * Format and send message to lpsched (die if any errors occur)
 */
/*VARARGS1*/
int
snd_msg(service_t *svc, int type, ...)
{
        int rc = -1;
        va_list ap;

        if (svc == NULL)
                return (-1);

        /* fill the message buffer */
        va_start(ap, type);
        rc = _putmessage(svc->msgbuf, type, ap);
        va_end(ap);
        if (rc < 0) {
                detailed_error(svc,
                    gettext("unable to build message for scheduler: %s"),
                    strerror(errno));
                return (rc);
        }

        /* write the message */
        while (((rc = mwrite(svc->md, svc->msgbuf)) < 0) && (errno == EINTR)) {
        }

        if (rc < 0)
                detailed_error(svc,
                    gettext("unable to send message to scheduler: %s"),
                    strerror(errno));
        return (rc);
}

/*
 * Receive message from lpsched (die if any errors occur)
 */
int
rcv_msg(service_t *svc, int type, ...)
{
        int rc = -1;

        if (svc == NULL)
                return (-1);

        /* read the message */
        while (((rc = mread(svc->md, svc->msgbuf, svc->msgbuf_size)) < 0) &&
            (errno == EINTR)) {
        }

        if (rc < 0)
                detailed_error(svc,
                    gettext("unable to read message from scheduler: %s"),
                    strerror(errno));
        else {
                va_list ap;

                va_start(ap, type);
                rc = _getmessage(svc->msgbuf, type, ap);
                va_end(ap);

                if (rc < 0)
                        detailed_error(svc,
                        gettext("unable to parse message from scheduler: %s"),
                            strerror(errno));
        }

        return (rc);
}

papi_status_t
lpsched_status_to_papi_status(int status)
{
        switch (status) {
        case MNOMEM:
                return (PAPI_TEMPORARY_ERROR);
        case MNOFILTER:
                return (PAPI_DOCUMENT_FORMAT_ERROR);
        case MNOOPEN:
                return (PAPI_DOCUMENT_ACCESS_ERROR);
        case MERRDEST:
        case MDENYDEST:
                return (PAPI_NOT_ACCEPTING);
        case MNOMEDIA:
                return (PAPI_PRINT_SUPPORT_FILE_NOT_FOUND);
        case MDENYMEDIA:
        case MNOPERM:
                return (PAPI_NOT_AUTHORIZED);
        case MUNKNOWN:
        case MNODEST:
        case MNOINFO:
                return (PAPI_NOT_FOUND);
        case MTRANSMITERR:
                return (PAPI_SERVICE_UNAVAILABLE);
        case M2LATE:
                return (PAPI_GONE);
        case MBUSY:
                return (PAPI_PRINTER_BUSY);
        case MOK:
        case MOKMORE:
                return (PAPI_OK);
        }

        return (PAPI_INTERNAL_ERROR);
}

char *
lpsched_status_string(short status)
{
                switch (status) {
        case MNOMEM:
                return (gettext("lpsched: out of memory"));
        case MNOFILTER:
                return (gettext("No filter available to convert job"));
        case MNOOPEN:
                return (gettext("lpsched: could not open request"));
        case MERRDEST:
                return (gettext("queue disabled"));
        case MDENYDEST:
                return (gettext("destination denied request"));
        case MNOMEDIA:
                return (gettext("unknown form specified in job"));
        case MDENYMEDIA:
                return (gettext("access denied to form specified in job"));
        case MUNKNOWN:
                return (gettext("no such resource"));
        case MNODEST:
                return (gettext("unknown destination"));
        case MNOPERM:
                return (gettext("permission denied"));
        case MNOINFO:
                return (gettext("no information available"));
        case MTRANSMITERR:
                return (gettext("failure to communicate with lpsched"));
        default: {
                static char result[16];

                snprintf(result, sizeof (result), gettext("status: %d"),
                    status);
                return (result);
                }
        }
}

papi_status_t
lpsched_alloc_files(papi_service_t svc, int number, char **prefix)
{
        papi_status_t result = PAPI_OK;
        short status = MOK;

        if ((svc == NULL) || (prefix == NULL))
                return (PAPI_BAD_ARGUMENT);

        if ((snd_msg(svc, S_ALLOC_FILES, number) < 0) ||
            (rcv_msg(svc, R_ALLOC_FILES, &status, prefix) < 0))
                status = MTRANSMITERR;

        if (status != MOK) {
                detailed_error(svc,
                gettext("failed to allocate %d file(s) for request: %s"),
                    number, lpsched_status_string(status));
                result = lpsched_status_to_papi_status(status);
        }

        return (result);
}

papi_status_t
lpsched_commit_job(papi_service_t svc, char *job, char **tmp)
/* job is host/req-id */
{
        papi_status_t result = PAPI_OK;
        short status = MOK;
        long bits;

        if ((svc == NULL) || (job == NULL) || (tmp == NULL))
                return (PAPI_BAD_ARGUMENT);

        if ((snd_msg(svc, S_PRINT_REQUEST, job) < 0) ||
            (rcv_msg(svc, R_PRINT_REQUEST, &status, tmp, &bits) < 0))
                status = MTRANSMITERR;

        if (status != MOK) {
                detailed_error(svc, gettext("failed to commit job (%s): %s"),
                        job, lpsched_status_string(status));
                result = lpsched_status_to_papi_status(status);
        }

        return (result);
}

papi_status_t
lpsched_start_change(papi_service_t svc, char *printer, int32_t job_id,
                char **tmp)
{
        papi_status_t result = PAPI_OK;
        short status = MOK;
        char req[BUFSIZ];
        char *dest;

        if ((svc == NULL) || (printer == NULL) || (job_id < 0))
                return (PAPI_BAD_ARGUMENT);

        dest = printer_name_from_uri_id(printer, job_id);
        snprintf(req, sizeof (req), "%s-%d", dest, job_id);
        free(dest);

        if ((snd_msg(svc, S_START_CHANGE_REQUEST, req) < 0) ||
            (rcv_msg(svc, R_START_CHANGE_REQUEST, &status, tmp) < 0))
                status = MTRANSMITERR;

        if (status != MOK) {
                detailed_error(svc,
                gettext("failed to initiate change for job (%s-%d): %s"),
                    printer,
                    job_id, lpsched_status_string(status));
                result = lpsched_status_to_papi_status(status);
        }

        return (result);
}

papi_status_t
lpsched_end_change(papi_service_t svc, char *printer, int32_t job_id)
{
        papi_status_t result = PAPI_OK;
        short status = MOK;
        long bits;
        char req[BUFSIZ];
        char *dest;

        if ((svc == NULL) || (printer == NULL) || (job_id < 0))
                return (PAPI_BAD_ARGUMENT);

        dest = printer_name_from_uri_id(printer, job_id);
        snprintf(req, sizeof (req), "%s-%d", dest, job_id);
        free(dest);

        if ((snd_msg(svc, S_END_CHANGE_REQUEST, req) < 0) ||
            (rcv_msg(svc, R_END_CHANGE_REQUEST, &status, &bits) < 0))
                status = MTRANSMITERR;

        if (status != MOK) {
                detailed_error(svc,
                gettext("failed to commit change for job (%s-%d): %s"), printer,
                    job_id, lpsched_status_string(status));
                result = lpsched_status_to_papi_status(status);
        }

        return (result);
}

papi_status_t
lpsched_accept_printer(papi_service_t svc, char *printer)
{
        papi_status_t result = PAPI_OK;
        short   status = MOK;
        char    *req_id;
        char *dest;

        if ((svc == NULL) || (printer == NULL))
                return (PAPI_BAD_ARGUMENT);

        dest = printer_name_from_uri_id(printer, -1);
        if ((snd_msg(svc, S_ACCEPT_DEST, dest) < 0) ||
            (rcv_msg(svc, R_ACCEPT_DEST, &status, &req_id) < 0))
                status = MTRANSMITERR;
        free(dest);

        if ((status != MOK) && (status != MERRDEST)) {
                detailed_error(svc, "%s: %s", printer,
                    lpsched_status_string(status));
        }
        result = lpsched_status_to_papi_status(status);

        return (result);
}

papi_status_t
lpsched_reject_printer(papi_service_t svc, char *printer, char *message)
{
        papi_status_t result = PAPI_OK;
        short    status = MOK;
        char    *req_id;
        char *dest;

        if ((svc == NULL) || (printer == NULL))
                return (PAPI_BAD_ARGUMENT);

        if (message == NULL)
                message = "stopped by user";

        dest = printer_name_from_uri_id(printer, -1);
        if ((snd_msg(svc, S_REJECT_DEST, dest, message, 0) < 0) ||
            (rcv_msg(svc, R_REJECT_DEST, &status, &req_id) < 0))
                status = MTRANSMITERR;
        free(dest);

        if ((status != MOK) && (status != MERRDEST)) {
                detailed_error(svc, "%s: %s", printer,
                    lpsched_status_string(status));
        }
        result = lpsched_status_to_papi_status(status);

        return (result);
}

papi_status_t
lpsched_enable_printer(papi_service_t svc, char *printer)
{
        papi_status_t result = PAPI_OK;
        short    status = MOK;
        char    *req_id;
        char *dest;

        if ((svc == NULL) || (printer == NULL))
                return (PAPI_BAD_ARGUMENT);

        dest = printer_name_from_uri_id(printer, -1);
        if ((snd_msg(svc, S_ENABLE_DEST, dest) < 0) ||
            (rcv_msg(svc, R_ENABLE_DEST, &status, &req_id) < 0))
                status = MTRANSMITERR;
        free(dest);

        if ((status != MOK) && (status != MERRDEST)) {
                detailed_error(svc, "%s: %s", printer,
                    lpsched_status_string(status));
        }
        result = lpsched_status_to_papi_status(status);

        return (result);
}

papi_status_t
lpsched_disable_printer(papi_service_t svc, char *printer, char *message)
{
        papi_status_t result = PAPI_OK;
        short    status = MOK;
        char    *req_id;
        char *dest;

        if ((svc == NULL) || (printer == NULL))
                return (PAPI_BAD_ARGUMENT);

        if (message == NULL)
                message = "stopped by user";

        dest = printer_name_from_uri_id(printer, -1);
        if ((snd_msg(svc, S_DISABLE_DEST, dest, message, 0) < 0) ||
            (rcv_msg(svc, R_DISABLE_DEST, &status, &req_id) < 0))
                status = MTRANSMITERR;
        free(dest);

        if ((status != MOK) && (status != MERRDEST)) {
                detailed_error(svc, "%s: %s", printer,
                    lpsched_status_string(status));
        }
        result = lpsched_status_to_papi_status(status);

        return (result);
}

papi_status_t
lpsched_load_unload_dest(papi_service_t handle, char *dest, int type)
{
        service_t *svc = handle;
        papi_status_t result;
        short status = MOK;

        /* tell the scheduler it's going */
        if (snd_msg(svc, type, dest, "", "") < 0)
                return (PAPI_SERVICE_UNAVAILABLE);

        switch (type) {
        case S_LOAD_PRINTER:
                type = R_LOAD_PRINTER;
                break;
        case S_UNLOAD_PRINTER:
                type = R_UNLOAD_PRINTER;
                break;
        case S_LOAD_CLASS:
                type = R_LOAD_CLASS;
                break;
        case S_UNLOAD_CLASS:
                type = R_UNLOAD_CLASS;
        }

        if (rcv_msg(svc, type, &status) < 0)
                return (PAPI_SERVICE_UNAVAILABLE);

        result = lpsched_status_to_papi_status(status);

        return (result);
}

papi_status_t
lpsched_remove_class(papi_service_t handle, char *dest)
{
        papi_status_t result;

        /* tell the scheduler it's going */
        result = lpsched_load_unload_dest(handle, dest, S_UNLOAD_CLASS);

        if (result == PAPI_OK) {
                /* remove the scheduler config files */
                if (delclass(dest) == -1)
                        result = PAPI_SERVICE_UNAVAILABLE;
        }

        return (result);
}

static void
remove_from_class(papi_service_t handle, char *dest, CLASS *cls)
{
        if (dellist(&cls->members, dest) == 0) {
                if (cls->members != NULL) {
                        if (putclass(cls->name, cls) == 0)
                                (void) lpsched_load_unload_dest(handle,
                                    cls->name, S_LOAD_CLASS);
                } else
                        (void) lpsched_remove_class(handle, cls->name);
        }
}

papi_status_t
lpsched_remove_printer(papi_service_t handle, char *dest)
{

        papi_status_t result;

        /* tell the scheduler it's going */
        result = lpsched_load_unload_dest(handle, dest, S_UNLOAD_PRINTER);

        if (result == PAPI_OK) {
                CLASS *cls;
                char *dflt;

                /* remove the scheduler config files */
                if (delprinter(dest) == -1)
                        return (PAPI_SERVICE_UNAVAILABLE);

                /* remove from any classes */
                while ((cls = getclass(NAME_ALL)) != NULL) {
                        if (searchlist(dest, cls->members) != 0)
                                remove_from_class(handle, dest, cls);
                        freeclass(cls);
                }

                /* reset the default if it needs to be done */
                if (((dflt = getdefault()) != NULL) &&
                    (strcmp(dflt, dest) == 0))
                        putdefault(NAME_NONE);
        }

        return (result);
}

papi_status_t
lpsched_add_modify_class(papi_service_t handle, char *dest,
                papi_attribute_t **attributes)
{
        papi_status_t result;
        void *iter = NULL;
        char **members = NULL;
        char *member;

        /*
         * The only attribute that we can modify for a class is the set of
         * members.  Anything else will be ignored.
         */
        for (result = papiAttributeListGetString(attributes, &iter,
            "member-names", &member);
            result == PAPI_OK;
            result = papiAttributeListGetString(attributes, &iter,
            NULL, &member))
                addlist(&members, member);

        if (members != NULL) {
                /* modify the configuration file */
                CLASS class;

                memset(&class, 0, sizeof (class));
                class.name = dest;
                class.members = members;

                if (putclass(dest, &class) == -1) {
                        if ((errno == EPERM) || (errno == EACCES))
                                result = PAPI_NOT_AUTHORIZED;
                        else
                                result = PAPI_NOT_POSSIBLE;
                } else
                        result = PAPI_OK;

                freelist(members);
        } else
                result = PAPI_ATTRIBUTES;

        /* tell the scheduler about the changes */
        if (result == PAPI_OK)
                result = lpsched_load_unload_dest(handle, dest, S_LOAD_CLASS);

        return (result);
}

papi_status_t
lpsched_add_printer(papi_service_t handle, char *dest,
                papi_attribute_t **attributes)
{
        PRINTER *p;
        papi_status_t result = PAPI_TEMPORARY_ERROR;

        if ((p = calloc(1, sizeof (*p))) != NULL) {
                p->name = strdup(dest);
                p->banner = BAN_ALWAYS;
                p->interface = strdup("/usr/lib/lp/model/uri");
                p->fault_alert.shcmd = strdup("mail");

                attributes_to_printer(attributes, p);

                if (putprinter(dest, p) == -1) {
                        if ((errno == EPERM) || (errno == EACCES))
                                result = PAPI_NOT_AUTHORIZED;
                        else
                                result = PAPI_NOT_POSSIBLE;
                } else
                        result = PAPI_OK;

                freeprinter(p);
        }

        /* tell the scheduler about the changes */
        if (result == PAPI_OK)
                result = lpsched_load_unload_dest(handle, dest, S_LOAD_PRINTER);

        return (result);
}

papi_status_t
lpsched_add_modify_printer(papi_service_t handle, char *dest,
                papi_attribute_t **attributes, int type)
{
        PRINTER *p;
        papi_status_t result;

        if (type == 0) {
                if ((p = calloc(1, sizeof (*p))) != NULL) {
                        p->name = strdup(dest);
                        p->banner = BAN_ALWAYS;
                        p->interface = strdup("/usr/lib/lp/model/uri");
                        p->fault_alert.shcmd = strdup("mail");
                }
        } else
                p = getprinter(dest);

        if (p != NULL) {
                attributes_to_printer(attributes, p);

                if (putprinter(dest, p) == -1) {
                        if ((errno == EPERM) || (errno == EACCES))
                                result = PAPI_NOT_AUTHORIZED;
                        else
                                result = PAPI_NOT_POSSIBLE;
                } else
                        result = PAPI_OK;

                freeprinter(p);
        } else
                result = PAPI_NOT_POSSIBLE;

        /* tell the scheduler about the changes */
        if (result == PAPI_OK)
                result = lpsched_load_unload_dest(handle, dest, S_LOAD_PRINTER);

        return (result);
}