/*
 * Copyright (c) 2013, Cisco Systems, Inc. All rights reserved.
 *
 * This software is available to you under a choice of one of two
 * licenses.  You may choose to be licensed under the terms of the GNU
 * General Public License (GPL) Version 2, available from the file
 * COPYING in the main directory of this source tree, or the
 * BSD license below:
 *
 *     Redistribution and use in source and binary forms, with or
 *     without modification, are permitted provided that the following
 *     conditions are met:
 *
 *      - Redistributions of source code must retain the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer.
 *
 *      - Redistributions in binary form must reproduce the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer in the documentation and/or other materials
 *        provided with the distribution.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 */

#include <linux/init.h>
#include <linux/errno.h>

#include <rdma/ib_user_verbs.h>
#include <rdma/ib_addr.h>

#include "usnic_common_util.h"
#include "usnic_ib.h"
#include "usnic_ib_qp_grp.h"
#include "usnic_vnic.h"
#include "usnic_ib_verbs.h"
#include "usnic_ib_sysfs.h"
#include "usnic_log.h"

static ssize_t board_id_show(struct device *device,
                             struct device_attribute *attr, char *buf)
{
        struct usnic_ib_dev *us_ibdev =
                rdma_device_to_drv_device(device, struct usnic_ib_dev, ib_dev);
        unsigned short subsystem_device_id;

        mutex_lock(&us_ibdev->usdev_lock);
        subsystem_device_id = us_ibdev->pdev->subsystem_device;
        mutex_unlock(&us_ibdev->usdev_lock);

        return sysfs_emit(buf, "%u\n", subsystem_device_id);
}
static DEVICE_ATTR_RO(board_id);

/*
 * Report the configuration for this PF
 */
static ssize_t
config_show(struct device *device, struct device_attribute *attr, char *buf)
{
        struct usnic_ib_dev *us_ibdev =
                rdma_device_to_drv_device(device, struct usnic_ib_dev, ib_dev);
        enum usnic_vnic_res_type res_type;
        int len;

        mutex_lock(&us_ibdev->usdev_lock);
        if (kref_read(&us_ibdev->vf_cnt) > 0) {
                char *busname;
                char *sep = "";
                /*
                 * bus name seems to come with annoying prefix.
                 * Remove it if it is predictable
                 */
                busname = us_ibdev->pdev->bus->name;
                if (strncmp(busname, "PCI Bus ", 8) == 0)
                        busname += 8;

                len = sysfs_emit(buf, "%s: %s:%d.%d, %s, %pM, %u VFs\n",
                                 dev_name(&us_ibdev->ib_dev.dev),
                                 busname,
                                 PCI_SLOT(us_ibdev->pdev->devfn),
                                 PCI_FUNC(us_ibdev->pdev->devfn),
                                 netdev_name(us_ibdev->netdev),
                                 us_ibdev->ufdev->mac,
                                 kref_read(&us_ibdev->vf_cnt));

                len += sysfs_emit_at(buf, len, " Per VF:");
                for (res_type = USNIC_VNIC_RES_TYPE_EOL;
                     res_type < USNIC_VNIC_RES_TYPE_MAX; res_type++) {
                        if (us_ibdev->vf_res_cnt[res_type] == 0)
                                continue;
                        len += sysfs_emit_at(buf, len, "%s %d %s",
                                             sep,
                                             us_ibdev->vf_res_cnt[res_type],
                                             usnic_vnic_res_type_to_str(res_type));
                        sep = ",";
                }
                len += sysfs_emit_at(buf, len, "\n");
        } else {
                len = sysfs_emit(buf, "%s: no VFs\n",
                                 dev_name(&us_ibdev->ib_dev.dev));
        }

        mutex_unlock(&us_ibdev->usdev_lock);

        return len;
}
static DEVICE_ATTR_RO(config);

static ssize_t
iface_show(struct device *device, struct device_attribute *attr, char *buf)
{
        struct usnic_ib_dev *us_ibdev =
                rdma_device_to_drv_device(device, struct usnic_ib_dev, ib_dev);

        return sysfs_emit(buf, "%s\n", netdev_name(us_ibdev->netdev));
}
static DEVICE_ATTR_RO(iface);

static ssize_t
max_vf_show(struct device *device, struct device_attribute *attr, char *buf)
{
        struct usnic_ib_dev *us_ibdev =
                rdma_device_to_drv_device(device, struct usnic_ib_dev, ib_dev);

        return sysfs_emit(buf, "%u\n", kref_read(&us_ibdev->vf_cnt));
}
static DEVICE_ATTR_RO(max_vf);

static ssize_t
qp_per_vf_show(struct device *device, struct device_attribute *attr, char *buf)
{
        struct usnic_ib_dev *us_ibdev =
                rdma_device_to_drv_device(device, struct usnic_ib_dev, ib_dev);
        int qp_per_vf;

        qp_per_vf = max(us_ibdev->vf_res_cnt[USNIC_VNIC_RES_TYPE_WQ],
                        us_ibdev->vf_res_cnt[USNIC_VNIC_RES_TYPE_RQ]);

        return sysfs_emit(buf, "%d\n", qp_per_vf);
}
static DEVICE_ATTR_RO(qp_per_vf);

static ssize_t
cq_per_vf_show(struct device *device, struct device_attribute *attr, char *buf)
{
        struct usnic_ib_dev *us_ibdev =
                rdma_device_to_drv_device(device, struct usnic_ib_dev, ib_dev);

        return sysfs_emit(buf, "%d\n",
                          us_ibdev->vf_res_cnt[USNIC_VNIC_RES_TYPE_CQ]);
}
static DEVICE_ATTR_RO(cq_per_vf);

static struct attribute *usnic_class_attributes[] = {
        &dev_attr_board_id.attr,
        &dev_attr_config.attr,
        &dev_attr_iface.attr,
        &dev_attr_max_vf.attr,
        &dev_attr_qp_per_vf.attr,
        &dev_attr_cq_per_vf.attr,
        NULL
};

const struct attribute_group usnic_attr_group = {
        .attrs = usnic_class_attributes,
};

struct qpn_attribute {
        struct attribute attr;
        ssize_t (*show)(struct usnic_ib_qp_grp *, char *buf);
};

/*
 * Definitions for supporting QPN entries in sysfs
 */
static ssize_t
usnic_ib_qpn_attr_show(struct kobject *kobj, struct attribute *attr, char *buf)
{
        struct usnic_ib_qp_grp *qp_grp;
        struct qpn_attribute *qpn_attr;

        qp_grp = container_of(kobj, struct usnic_ib_qp_grp, kobj);
        qpn_attr = container_of(attr, struct qpn_attribute, attr);

        return qpn_attr->show(qp_grp, buf);
}

static const struct sysfs_ops usnic_ib_qpn_sysfs_ops = {
        .show = usnic_ib_qpn_attr_show
};

#define QPN_ATTR_RO(NAME) \
struct qpn_attribute qpn_attr_##NAME = __ATTR_RO(NAME)

static ssize_t context_show(struct usnic_ib_qp_grp *qp_grp, char *buf)
{
        return sysfs_emit(buf, "0x%p\n", qp_grp->ctx);
}

static ssize_t summary_show(struct usnic_ib_qp_grp *qp_grp, char *buf)
{
        int i, j;
        struct usnic_vnic_res_chunk *res_chunk;
        struct usnic_vnic_res *vnic_res;
        int len;

        len = sysfs_emit(buf, "QPN: %d State: (%s) PID: %u VF Idx: %hu",
                         qp_grp->ibqp.qp_num,
                         usnic_ib_qp_grp_state_to_string(qp_grp->state),
                         qp_grp->owner_pid,
                         usnic_vnic_get_index(qp_grp->vf->vnic));

        for (i = 0; qp_grp->res_chunk_list[i]; i++) {
                res_chunk = qp_grp->res_chunk_list[i];
                for (j = 0; j < res_chunk->cnt; j++) {
                        vnic_res = res_chunk->res[j];
                        len += sysfs_emit_at(buf, len, " %s[%d]",
                                usnic_vnic_res_type_to_str(vnic_res->type),
                                vnic_res->vnic_idx);
                }
        }

        len += sysfs_emit_at(buf, len, "\n");

        return len;
}

static QPN_ATTR_RO(context);
static QPN_ATTR_RO(summary);

static struct attribute *usnic_ib_qpn_default_attrs[] = {
        &qpn_attr_context.attr,
        &qpn_attr_summary.attr,
        NULL
};
ATTRIBUTE_GROUPS(usnic_ib_qpn_default);

static struct kobj_type usnic_ib_qpn_type = {
        .sysfs_ops = &usnic_ib_qpn_sysfs_ops,
        .default_groups = usnic_ib_qpn_default_groups,
};

int usnic_ib_sysfs_register_usdev(struct usnic_ib_dev *us_ibdev)
{
        /* create kernel object for looking at individual QPs */
        kobject_get(&us_ibdev->ib_dev.dev.kobj);
        us_ibdev->qpn_kobj = kobject_create_and_add("qpn",
                        &us_ibdev->ib_dev.dev.kobj);
        if (us_ibdev->qpn_kobj == NULL) {
                kobject_put(&us_ibdev->ib_dev.dev.kobj);
                return -ENOMEM;
        }

        return 0;
}

void usnic_ib_sysfs_unregister_usdev(struct usnic_ib_dev *us_ibdev)
{
        kobject_put(us_ibdev->qpn_kobj);
}

void usnic_ib_sysfs_qpn_add(struct usnic_ib_qp_grp *qp_grp)
{
        struct usnic_ib_dev *us_ibdev;
        int err;

        us_ibdev = qp_grp->vf->pf;

        err = kobject_init_and_add(&qp_grp->kobj, &usnic_ib_qpn_type,
                        kobject_get(us_ibdev->qpn_kobj),
                        "%d", qp_grp->grp_id);
        if (err) {
                kobject_put(us_ibdev->qpn_kobj);
                return;
        }
}

void usnic_ib_sysfs_qpn_remove(struct usnic_ib_qp_grp *qp_grp)
{
        struct usnic_ib_dev *us_ibdev;

        us_ibdev = qp_grp->vf->pf;

        kobject_put(&qp_grp->kobj);
        kobject_put(us_ibdev->qpn_kobj);
}