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

#include <sys/types.h>
#include <sys/sysmacros.h>
#include <sys/ddi.h>
#include <sys/async.h>
#include <sys/sunddi.h>
#include <sys/ddifm.h>
#include <sys/fm/protocol.h>
#include <sys/vmem.h>
#include <sys/intr.h>
#include <sys/ivintr.h>
#include <sys/errno.h>
#include <sys/hypervisor_api.h>
#include <sys/hsvc.h>
#include <sys/pci_cfgacc.h>
#include <sys/pci_cfgacc_4v.h>
#include <px_obj.h>
#include <sys/machsystm.h>
#include "px_lib4v.h"

#define MPS_SET         0
#define MPS_GET         1

static uint64_t hvio_rp_mps(devhandle_t dev_hdl, pci_device_t bdf, int32_t *mps,
        int op);

uint64_t
hvio_get_rp_mps_cap(devhandle_t dev_hdl, pci_device_t bdf, int32_t *mps_cap)
{
        return (hvio_rp_mps(dev_hdl, bdf, mps_cap, MPS_GET));
}

uint64_t
hvio_set_rp_mps(devhandle_t dev_hdl, pci_device_t bdf, int32_t mps)
{
        return (hvio_rp_mps(dev_hdl, bdf, &mps, MPS_SET));
}

uint64_t
hvio_rp_mps(devhandle_t dev_hdl, pci_device_t bdf, int32_t *mps, int op)
{
        uint64_t        data, hdr, hdr_next_ptr, hdr_cap_id;
        uint16_t        offset = PCI_CONF_STAT;
        int             deadcount = 0;
        pci_cfg_data_t  dataw;

        if ((hvio_config_get(dev_hdl, bdf, PCI_CONF_VENID, 4,
            (pci_cfg_data_t *)&data)) != H_EOK)
                return (H_ENOACCESS);

        if ((data & 0xffff) != 0x108e)
                return (H_ENOACCESS);

        if ((hvio_config_get(dev_hdl, bdf, PCI_CONF_COMM, 4,
            (pci_cfg_data_t *)&hdr)) != H_EOK)
                return (H_ENOACCESS);

        if (!(hdr & (PCI_STAT_CAP << 16)))
                return (H_ENOACCESS);

        (void) hvio_config_get(dev_hdl, bdf, PCI_CONF_CAP_PTR, 4,
            (pci_cfg_data_t *)&hdr);

        hdr_next_ptr = hdr & 0xFF;
        hdr_cap_id = 0;

        while ((hdr_next_ptr != PCI_CAP_NEXT_PTR_NULL) &&
            (hdr_cap_id != PCI_CAP_ID_PCI_E)) {

                offset = hdr_next_ptr;

                if (hdr_next_ptr < 0x40)
                        break;

                (void) hvio_config_get(dev_hdl, bdf, hdr_next_ptr, 4,
                    (pci_cfg_data_t *)&hdr);

                hdr_next_ptr = (hdr >> 8) & 0xFF;
                hdr_cap_id = hdr & 0xFF;

                if (deadcount++ > 100)
                        break;
        }

        if (hdr_cap_id != PCI_CAP_ID_PCI_E)
                return (H_ENOACCESS);

        if (op == MPS_SET) {

                /* Write the MPS */

                (void) hvio_config_get(dev_hdl, bdf, offset + PCIE_DEVCTL,
                    4, (pci_cfg_data_t *)&data);

                data = (data & 0xffffff1f) | (*mps << 5);

                dataw.qw = data;

                (void) hvio_config_put(dev_hdl, bdf, offset + PCIE_DEVCTL,
                    4, dataw);
        } else {

                /* Read the MPS Capabilities */

                (void) hvio_config_get(dev_hdl, bdf, offset + PCIE_DEVCAP,
                    4, (pci_cfg_data_t *)&data);

                *mps = data & 0x7;
        }

        return (H_EOK);
}