// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
/*
 * Copyright (c) 2016 Mellanox Technologies Ltd. All rights reserved.
 * Copyright (c) 2015 System Fabric Works, Inc. All rights reserved.
 */

#include <rdma/rdma_netlink.h>
#include <net/addrconf.h>
#include "rxe.h"
#include "rxe_loc.h"

MODULE_AUTHOR("Bob Pearson, Frank Zago, John Groves, Kamal Heib");
MODULE_DESCRIPTION("Soft RDMA transport");
MODULE_LICENSE("Dual BSD/GPL");

/* free resources for a rxe device all objects created for this device must
 * have been destroyed
 */
void rxe_dealloc(struct ib_device *ib_dev)
{
        struct rxe_dev *rxe = container_of(ib_dev, struct rxe_dev, ib_dev);

        rxe_pool_cleanup(&rxe->uc_pool);
        rxe_pool_cleanup(&rxe->pd_pool);
        rxe_pool_cleanup(&rxe->ah_pool);
        rxe_pool_cleanup(&rxe->srq_pool);
        rxe_pool_cleanup(&rxe->qp_pool);
        rxe_pool_cleanup(&rxe->cq_pool);
        rxe_pool_cleanup(&rxe->mr_pool);
        rxe_pool_cleanup(&rxe->mw_pool);

        WARN_ON(!RB_EMPTY_ROOT(&rxe->mcg_tree));

        mutex_destroy(&rxe->usdev_lock);
}

static const struct ib_device_ops rxe_ib_dev_odp_ops = {
        .advise_mr = rxe_ib_advise_mr,
};

/* initialize rxe device parameters */
static void rxe_init_device_param(struct rxe_dev *rxe, struct net_device *ndev)
{
        rxe->max_inline_data                    = RXE_MAX_INLINE_DATA;

        rxe->attr.vendor_id                     = RXE_VENDOR_ID;
        rxe->attr.max_mr_size                   = RXE_MAX_MR_SIZE;
        rxe->attr.page_size_cap                 = RXE_PAGE_SIZE_CAP;
        rxe->attr.max_qp                        = RXE_MAX_QP;
        rxe->attr.max_qp_wr                     = RXE_MAX_QP_WR;
        rxe->attr.device_cap_flags              = RXE_DEVICE_CAP_FLAGS;
        rxe->attr.kernel_cap_flags              = IBK_ALLOW_USER_UNREG;
        rxe->attr.max_send_sge                  = RXE_MAX_SGE;
        rxe->attr.max_recv_sge                  = RXE_MAX_SGE;
        rxe->attr.max_sge_rd                    = RXE_MAX_SGE_RD;
        rxe->attr.max_cq                        = RXE_MAX_CQ;
        rxe->attr.max_cqe                       = (1 << RXE_MAX_LOG_CQE) - 1;
        rxe->attr.max_mr                        = RXE_MAX_MR;
        rxe->attr.max_mw                        = RXE_MAX_MW;
        rxe->attr.max_pd                        = RXE_MAX_PD;
        rxe->attr.max_qp_rd_atom                = RXE_MAX_QP_RD_ATOM;
        rxe->attr.max_res_rd_atom               = RXE_MAX_RES_RD_ATOM;
        rxe->attr.max_qp_init_rd_atom           = RXE_MAX_QP_INIT_RD_ATOM;
        rxe->attr.atomic_cap                    = IB_ATOMIC_HCA;
        rxe->attr.max_mcast_grp                 = RXE_MAX_MCAST_GRP;
        rxe->attr.max_mcast_qp_attach           = RXE_MAX_MCAST_QP_ATTACH;
        rxe->attr.max_total_mcast_qp_attach     = RXE_MAX_TOT_MCAST_QP_ATTACH;
        rxe->attr.max_ah                        = RXE_MAX_AH;
        rxe->attr.max_srq                       = RXE_MAX_SRQ;
        rxe->attr.max_srq_wr                    = RXE_MAX_SRQ_WR;
        rxe->attr.max_srq_sge                   = RXE_MAX_SRQ_SGE;
        rxe->attr.max_fast_reg_page_list_len    = RXE_MAX_FMR_PAGE_LIST_LEN;
        rxe->attr.max_pkeys                     = RXE_MAX_PKEYS;
        rxe->attr.local_ca_ack_delay            = RXE_LOCAL_CA_ACK_DELAY;

        if (ndev->addr_len) {
                memcpy(rxe->raw_gid, ndev->dev_addr,
                        min_t(unsigned int, ndev->addr_len, ETH_ALEN));
        } else {
                /*
                 * This device does not have a HW address, but
                 * connection mangagement requires a unique gid.
                 */
                eth_random_addr(rxe->raw_gid);
        }

        addrconf_addr_eui48((unsigned char *)&rxe->attr.sys_image_guid,
                        rxe->raw_gid);

        rxe->max_ucontext                       = RXE_MAX_UCONTEXT;

        if (IS_ENABLED(CONFIG_INFINIBAND_ON_DEMAND_PAGING)) {
                rxe->attr.kernel_cap_flags |= IBK_ON_DEMAND_PAGING;

                /* IB_ODP_SUPPORT_IMPLICIT is not supported right now. */
                rxe->attr.odp_caps.general_caps |= IB_ODP_SUPPORT;

                rxe->attr.odp_caps.per_transport_caps.ud_odp_caps |= IB_ODP_SUPPORT_SEND;
                rxe->attr.odp_caps.per_transport_caps.ud_odp_caps |= IB_ODP_SUPPORT_RECV;
                rxe->attr.odp_caps.per_transport_caps.ud_odp_caps |= IB_ODP_SUPPORT_SRQ_RECV;

                rxe->attr.odp_caps.per_transport_caps.rc_odp_caps |= IB_ODP_SUPPORT_SEND;
                rxe->attr.odp_caps.per_transport_caps.rc_odp_caps |= IB_ODP_SUPPORT_RECV;
                rxe->attr.odp_caps.per_transport_caps.rc_odp_caps |= IB_ODP_SUPPORT_WRITE;
                rxe->attr.odp_caps.per_transport_caps.rc_odp_caps |= IB_ODP_SUPPORT_READ;
                rxe->attr.odp_caps.per_transport_caps.rc_odp_caps |= IB_ODP_SUPPORT_ATOMIC;
                rxe->attr.odp_caps.per_transport_caps.rc_odp_caps |= IB_ODP_SUPPORT_SRQ_RECV;
                rxe->attr.odp_caps.per_transport_caps.rc_odp_caps |= IB_ODP_SUPPORT_FLUSH;
                rxe->attr.odp_caps.per_transport_caps.rc_odp_caps |= IB_ODP_SUPPORT_ATOMIC_WRITE;

                /* set handler for ODP prefetching API - ibv_advise_mr(3) */
                ib_set_device_ops(&rxe->ib_dev, &rxe_ib_dev_odp_ops);
        }
}

/* initialize port attributes */
static void rxe_init_port_param(struct rxe_port *port)
{
        port->attr.state                = IB_PORT_DOWN;
        port->attr.max_mtu              = IB_MTU_4096;
        port->attr.active_mtu           = IB_MTU_256;
        port->attr.gid_tbl_len          = RXE_PORT_GID_TBL_LEN;
        port->attr.port_cap_flags       = RXE_PORT_PORT_CAP_FLAGS;
        port->attr.max_msg_sz           = RXE_PORT_MAX_MSG_SZ;
        port->attr.bad_pkey_cntr        = RXE_PORT_BAD_PKEY_CNTR;
        port->attr.qkey_viol_cntr       = RXE_PORT_QKEY_VIOL_CNTR;
        port->attr.pkey_tbl_len         = RXE_PORT_PKEY_TBL_LEN;
        port->attr.lid                  = RXE_PORT_LID;
        port->attr.sm_lid               = RXE_PORT_SM_LID;
        port->attr.lmc                  = RXE_PORT_LMC;
        port->attr.max_vl_num           = RXE_PORT_MAX_VL_NUM;
        port->attr.sm_sl                = RXE_PORT_SM_SL;
        port->attr.subnet_timeout       = RXE_PORT_SUBNET_TIMEOUT;
        port->attr.init_type_reply      = RXE_PORT_INIT_TYPE_REPLY;
        port->attr.active_width         = RXE_PORT_ACTIVE_WIDTH;
        port->attr.active_speed         = RXE_PORT_ACTIVE_SPEED;
        port->attr.phys_state           = RXE_PORT_PHYS_STATE;
        port->mtu_cap                   = ib_mtu_enum_to_int(IB_MTU_256);
        port->subnet_prefix             = cpu_to_be64(RXE_PORT_SUBNET_PREFIX);
}

/* initialize port state, note IB convention that HCA ports are always
 * numbered from 1
 */
static void rxe_init_ports(struct rxe_dev *rxe, struct net_device *ndev)
{
        struct rxe_port *port = &rxe->port;

        rxe_init_port_param(port);
        addrconf_addr_eui48((unsigned char *)&port->port_guid,
                            rxe->raw_gid);
        spin_lock_init(&port->port_lock);
}

/* init pools of managed objects */
static void rxe_init_pools(struct rxe_dev *rxe)
{
        rxe_pool_init(rxe, &rxe->uc_pool, RXE_TYPE_UC);
        rxe_pool_init(rxe, &rxe->pd_pool, RXE_TYPE_PD);
        rxe_pool_init(rxe, &rxe->ah_pool, RXE_TYPE_AH);
        rxe_pool_init(rxe, &rxe->srq_pool, RXE_TYPE_SRQ);
        rxe_pool_init(rxe, &rxe->qp_pool, RXE_TYPE_QP);
        rxe_pool_init(rxe, &rxe->cq_pool, RXE_TYPE_CQ);
        rxe_pool_init(rxe, &rxe->mr_pool, RXE_TYPE_MR);
        rxe_pool_init(rxe, &rxe->mw_pool, RXE_TYPE_MW);
}

/* initialize rxe device state */
static void rxe_init(struct rxe_dev *rxe, struct net_device *ndev)
{
        /* init default device parameters */
        rxe_init_device_param(rxe, ndev);

        rxe_init_ports(rxe, ndev);
        rxe_init_pools(rxe);

        /* init pending mmap list */
        spin_lock_init(&rxe->mmap_offset_lock);
        spin_lock_init(&rxe->pending_lock);
        INIT_LIST_HEAD(&rxe->pending_mmaps);

        /* init multicast support */
        spin_lock_init(&rxe->mcg_lock);
        rxe->mcg_tree = RB_ROOT;

        mutex_init(&rxe->usdev_lock);
}

void rxe_set_mtu(struct rxe_dev *rxe, unsigned int ndev_mtu)
{
        struct rxe_port *port = &rxe->port;
        enum ib_mtu mtu;

        mtu = eth_mtu_int_to_enum(ndev_mtu);

        /* Make sure that new MTU in range */
        mtu = mtu ? min_t(enum ib_mtu, mtu, IB_MTU_4096) : IB_MTU_256;

        port->attr.active_mtu = mtu;
        port->mtu_cap = ib_mtu_enum_to_int(mtu);
}

/* called by ifc layer to create new rxe device.
 * The caller should allocate memory for rxe by calling ib_alloc_device.
 */
int rxe_add(struct rxe_dev *rxe, unsigned int mtu, const char *ibdev_name,
                        struct net_device *ndev)
{
        rxe_init(rxe, ndev);
        rxe_set_mtu(rxe, mtu);

        return rxe_register_device(rxe, ibdev_name, ndev);
}

static int rxe_newlink(const char *ibdev_name, struct net_device *ndev)
{
        struct rxe_dev *rxe;
        int err = 0;

        if (is_vlan_dev(ndev)) {
                rxe_err("rxe creation allowed on top of a real device only\n");
                err = -EPERM;
                goto err;
        }

        rxe = rxe_get_dev_from_net(ndev);
        if (rxe) {
                ib_device_put(&rxe->ib_dev);
                rxe_err_dev(rxe, "already configured on %s\n", ndev->name);
                err = -EEXIST;
                goto err;
        }

        err = rxe_net_add(ibdev_name, ndev);
        if (err) {
                rxe_err("failed to add %s\n", ndev->name);
                goto err;
        }
err:
        return err;
}

static struct rdma_link_ops rxe_link_ops = {
        .type = "rxe",
        .newlink = rxe_newlink,
};

static int __init rxe_module_init(void)
{
        int err;

        err = rxe_alloc_wq();
        if (err)
                return err;

        err = rxe_net_init();
        if (err) {
                rxe_destroy_wq();
                return err;
        }

        rdma_link_register(&rxe_link_ops);
        pr_info("loaded\n");
        return 0;
}

static void __exit rxe_module_exit(void)
{
        rdma_link_unregister(&rxe_link_ops);
        ib_unregister_driver(RDMA_DRIVER_RXE);
        rxe_net_exit();
        rxe_destroy_wq();

        pr_info("unloaded\n");
}

late_initcall(rxe_module_init);
module_exit(rxe_module_exit);

MODULE_ALIAS_RDMA_LINK("rxe");