// SPDX-License-Identifier: GPL-2.0

/* Texas Instruments ICSSG Ethernet Driver
 *
 * Copyright (C) 2018-2022 Texas Instruments Incorporated - https://www.ti.com/
 *
 */

#include <linux/bitops.h>
#include <linux/clk.h>
#include <linux/delay.h>
#include <linux/dma-mapping.h>
#include <linux/dma/ti-cppi5.h>
#include <linux/etherdevice.h>
#include <linux/genalloc.h>
#include <linux/if_hsr.h>
#include <linux/if_vlan.h>
#include <linux/interrupt.h>
#include <linux/io-64-nonatomic-hi-lo.h>
#include <linux/kernel.h>
#include <linux/mfd/syscon.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/of_mdio.h>
#include <linux/of_net.h>
#include <linux/platform_device.h>
#include <linux/phy.h>
#include <linux/property.h>
#include <linux/remoteproc/pruss.h>
#include <linux/regmap.h>
#include <linux/remoteproc.h>
#include <net/switchdev.h>

#include "icssg_prueth.h"
#include "icssg_mii_rt.h"
#include "icssg_switchdev.h"
#include "../k3-cppi-desc-pool.h"

#define PRUETH_MODULE_DESCRIPTION "PRUSS ICSSG Ethernet driver"

#define DEFAULT_VID             1
#define DEFAULT_PORT_MASK       1
#define DEFAULT_UNTAG_MASK      1

#define NETIF_PRUETH_HSR_OFFLOAD_FEATURES       (NETIF_F_HW_HSR_FWD | \
                                                 NETIF_F_HW_HSR_DUP | \
                                                 NETIF_F_HW_HSR_TAG_INS | \
                                                 NETIF_F_HW_HSR_TAG_RM)

#define PRUETH_RX_DMA_ATTR                      (DMA_ATTR_SKIP_CPU_SYNC |\
                                                 DMA_ATTR_WEAK_ORDERING)

/* CTRLMMR_ICSSG_RGMII_CTRL register bits */
#define ICSSG_CTRL_RGMII_ID_MODE                BIT(24)

static void emac_adjust_link(struct net_device *ndev);

static int emac_get_tx_ts(struct prueth_emac *emac,
                          struct emac_tx_ts_response *rsp)
{
        struct prueth *prueth = emac->prueth;
        int slice = prueth_emac_slice(emac);
        int addr;

        addr = icssg_queue_pop(prueth, slice == 0 ?
                               ICSSG_TS_POP_SLICE0 : ICSSG_TS_POP_SLICE1);
        if (addr < 0)
                return addr;

        memcpy_fromio(rsp, prueth->shram.va + addr, sizeof(*rsp));
        /* return buffer back for to pool */
        icssg_queue_push(prueth, slice == 0 ?
                         ICSSG_TS_PUSH_SLICE0 : ICSSG_TS_PUSH_SLICE1, addr);

        return 0;
}

static void tx_ts_work(struct prueth_emac *emac)
{
        struct skb_shared_hwtstamps ssh;
        struct emac_tx_ts_response tsr;
        struct sk_buff *skb;
        int ret = 0;
        u32 hi_sw;
        u64 ns;

        /* There may be more than one pending requests */
        while (1) {
                ret = emac_get_tx_ts(emac, &tsr);
                if (ret) /* nothing more */
                        break;

                if (tsr.cookie >= PRUETH_MAX_TX_TS_REQUESTS ||
                    !emac->tx_ts_skb[tsr.cookie]) {
                        netdev_err(emac->ndev, "Invalid TX TS cookie 0x%x\n",
                                   tsr.cookie);
                        break;
                }

                skb = emac->tx_ts_skb[tsr.cookie];
                emac->tx_ts_skb[tsr.cookie] = NULL;     /* free slot */
                if (!skb) {
                        netdev_err(emac->ndev, "Driver Bug! got NULL skb\n");
                        break;
                }

                hi_sw = readl(emac->prueth->shram.va +
                              TIMESYNC_FW_WC_COUNT_HI_SW_OFFSET_OFFSET);
                ns = icssg_ts_to_ns(hi_sw, tsr.hi_ts, tsr.lo_ts,
                                    IEP_DEFAULT_CYCLE_TIME_NS);

                memset(&ssh, 0, sizeof(ssh));
                ssh.hwtstamp = ns_to_ktime(ns);

                skb_tstamp_tx(skb, &ssh);
                dev_consume_skb_any(skb);

                if (atomic_dec_and_test(&emac->tx_ts_pending))  /* no more? */
                        break;
        }
}

static irqreturn_t prueth_tx_ts_irq(int irq, void *dev_id)
{
        struct prueth_emac *emac = dev_id;

        /* currently only TX timestamp is being returned */
        tx_ts_work(emac);

        return IRQ_HANDLED;
}

static int prueth_start(struct rproc *rproc, const char *fw_name)
{
        int ret;

        ret = rproc_set_firmware(rproc, fw_name);
        if (ret)
                return ret;
        return rproc_boot(rproc);
}

static void prueth_shutdown(struct rproc *rproc)
{
        rproc_shutdown(rproc);
}

static int prueth_emac_start(struct prueth *prueth)
{
        struct icssg_firmwares *firmwares;
        struct device *dev = prueth->dev;
        int ret, slice;

        if (prueth->is_switch_mode)
                firmwares = prueth->icssg_switch_firmwares;
        else if (prueth->is_hsr_offload_mode && HSR_V1 == prueth->hsr_prp_version)
                firmwares = prueth->icssg_hsr_firmwares;
        else if (prueth->is_hsr_offload_mode && PRP_V1 == prueth->hsr_prp_version)
                firmwares = prueth->icssg_prp_firmwares;
        else
                firmwares = prueth->icssg_emac_firmwares;

        for (slice = 0; slice < PRUETH_NUM_MACS; slice++) {
                ret = prueth_start(prueth->pru[slice], firmwares[slice].pru);
                if (ret) {
                        dev_err(dev, "failed to boot PRU%d: %d\n", slice, ret);
                        goto unwind_slices;
                }

                ret = prueth_start(prueth->rtu[slice], firmwares[slice].rtu);
                if (ret) {
                        dev_err(dev, "failed to boot RTU%d: %d\n", slice, ret);
                        rproc_shutdown(prueth->pru[slice]);
                        goto unwind_slices;
                }

                ret = prueth_start(prueth->txpru[slice], firmwares[slice].txpru);
                if (ret) {
                        dev_err(dev, "failed to boot TX_PRU%d: %d\n", slice, ret);
                        rproc_shutdown(prueth->rtu[slice]);
                        rproc_shutdown(prueth->pru[slice]);
                        goto unwind_slices;
                }
        }

        return 0;

unwind_slices:
        while (--slice >= 0) {
                prueth_shutdown(prueth->txpru[slice]);
                prueth_shutdown(prueth->rtu[slice]);
                prueth_shutdown(prueth->pru[slice]);
        }

        return ret;
}

static void prueth_emac_stop(struct prueth *prueth)
{
        int slice;

        for (slice = 0; slice < PRUETH_NUM_MACS; slice++) {
                prueth_shutdown(prueth->txpru[slice]);
                prueth_shutdown(prueth->rtu[slice]);
                prueth_shutdown(prueth->pru[slice]);
        }
}

static void icssg_enable_fw_offload(struct prueth *prueth)
{
        struct prueth_emac *emac;
        int mac;

        for (mac = PRUETH_MAC0; mac < PRUETH_NUM_MACS; mac++) {
                emac = prueth->emac[mac];
                if (prueth->is_hsr_offload_mode) {
                        if (emac->ndev->features & NETIF_F_HW_HSR_TAG_RM)
                                icssg_set_port_state(emac, ICSSG_EMAC_HSR_RX_OFFLOAD_ENABLE);
                        else
                                icssg_set_port_state(emac, ICSSG_EMAC_HSR_RX_OFFLOAD_DISABLE);
                }

                if (prueth->is_switch_mode || prueth->is_hsr_offload_mode) {
                        if (netif_running(emac->ndev)) {
                                icssg_fdb_add_del(emac, eth_stp_addr, prueth->default_vlan,
                                                  ICSSG_FDB_ENTRY_P0_MEMBERSHIP |
                                                  ICSSG_FDB_ENTRY_P1_MEMBERSHIP |
                                                  ICSSG_FDB_ENTRY_P2_MEMBERSHIP |
                                                  ICSSG_FDB_ENTRY_BLOCK,
                                                  true);
                                icssg_vtbl_modify(emac, emac->port_vlan | DEFAULT_VID,
                                                  BIT(emac->port_id) | DEFAULT_PORT_MASK,
                                                  BIT(emac->port_id) | DEFAULT_UNTAG_MASK,
                                                  true);
                                if (prueth->is_hsr_offload_mode)
                                        icssg_vtbl_modify(emac, DEFAULT_VID,
                                                          DEFAULT_PORT_MASK,
                                                          DEFAULT_UNTAG_MASK, true);
                                icssg_set_pvid(prueth, emac->port_vlan, emac->port_id);
                                if (prueth->is_switch_mode)
                                        icssg_set_port_state(emac, ICSSG_EMAC_PORT_VLAN_AWARE_ENABLE);
                        }
                }
        }
}

static int prueth_emac_common_start(struct prueth *prueth)
{
        struct prueth_emac *emac;
        int ret = 0;
        int slice;

        if (!prueth->emac[ICSS_SLICE0] && !prueth->emac[ICSS_SLICE1])
                return -EINVAL;

        /* clear SMEM and MSMC settings for all slices */
        memset_io(prueth->msmcram.va, 0, prueth->msmcram.size);
        memset_io(prueth->shram.va, 0, ICSSG_CONFIG_OFFSET_SLICE1 * PRUETH_NUM_MACS);

        icssg_class_default(prueth->miig_rt, ICSS_SLICE0, 0, false);
        icssg_class_default(prueth->miig_rt, ICSS_SLICE1, 0, false);

        if (prueth->is_switch_mode || prueth->is_hsr_offload_mode)
                icssg_init_fw_offload_mode(prueth);
        else
                icssg_init_emac_mode(prueth);

        for (slice = 0; slice < PRUETH_NUM_MACS; slice++) {
                emac = prueth->emac[slice];
                if (!emac)
                        continue;
                ret = icssg_config(prueth, emac, slice);
                if (ret)
                        goto disable_class;

                /* Reset link state to force reconfiguration in
                 * emac_adjust_link(). Without this, if the link was already up
                 * before restart, emac_adjust_link() won't detect any state
                 * change and will skip critical configuration like writing
                 * speed to firmware.
                 */
                emac->link = 0;

                mutex_lock(&emac->ndev->phydev->lock);
                emac_adjust_link(emac->ndev);
                mutex_unlock(&emac->ndev->phydev->lock);
        }

        ret = prueth_emac_start(prueth);
        if (ret)
                goto disable_class;

        emac = prueth->emac[ICSS_SLICE0] ? prueth->emac[ICSS_SLICE0] :
               prueth->emac[ICSS_SLICE1];
        ret = icss_iep_init(emac->iep, &prueth_iep_clockops,
                            emac, IEP_DEFAULT_CYCLE_TIME_NS);
        if (ret) {
                dev_err(prueth->dev, "Failed to initialize IEP module\n");
                goto stop_pruss;
        }

        return 0;

stop_pruss:
        prueth_emac_stop(prueth);

disable_class:
        icssg_class_disable(prueth->miig_rt, ICSS_SLICE0);
        icssg_class_disable(prueth->miig_rt, ICSS_SLICE1);

        return ret;
}

static int prueth_emac_common_stop(struct prueth *prueth)
{
        struct prueth_emac *emac;

        if (!prueth->emac[ICSS_SLICE0] && !prueth->emac[ICSS_SLICE1])
                return -EINVAL;

        icssg_class_disable(prueth->miig_rt, ICSS_SLICE0);
        icssg_class_disable(prueth->miig_rt, ICSS_SLICE1);

        prueth_emac_stop(prueth);

        emac = prueth->emac[ICSS_SLICE0] ? prueth->emac[ICSS_SLICE0] :
               prueth->emac[ICSS_SLICE1];
        icss_iep_exit(emac->iep);

        return 0;
}

/* called back by PHY layer if there is change in link state of hw port*/
static void emac_adjust_link(struct net_device *ndev)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        struct phy_device *phydev = ndev->phydev;
        struct prueth *prueth = emac->prueth;
        bool new_state = false;
        unsigned long flags;

        if (phydev->link) {
                /* check the mode of operation - full/half duplex */
                if (phydev->duplex != emac->duplex) {
                        new_state = true;
                        emac->duplex = phydev->duplex;
                }
                if (phydev->speed != emac->speed) {
                        new_state = true;
                        emac->speed = phydev->speed;
                }
                if (!emac->link) {
                        new_state = true;
                        emac->link = 1;
                }
        } else if (emac->link) {
                new_state = true;
                emac->link = 0;

                /* f/w should support 100 & 1000 */
                emac->speed = SPEED_1000;

                /* half duplex may not be supported by f/w */
                emac->duplex = DUPLEX_FULL;
        }

        if (new_state) {
                phy_print_status(phydev);

                /* update RGMII and MII configuration based on PHY negotiated
                 * values
                 */
                if (emac->link) {
                        if (emac->duplex == DUPLEX_HALF)
                                icssg_config_half_duplex(emac);
                        /* Set the RGMII cfg for gig en and full duplex */
                        icssg_update_rgmii_cfg(prueth->miig_rt, emac);

                        /* update the Tx IPG based on 100M/1G speed */
                        spin_lock_irqsave(&emac->lock, flags);
                        icssg_config_ipg(emac);
                        spin_unlock_irqrestore(&emac->lock, flags);
                        icssg_config_set_speed(emac);
                        icssg_set_port_state(emac, ICSSG_EMAC_PORT_FORWARD);

                } else {
                        icssg_set_port_state(emac, ICSSG_EMAC_PORT_DISABLE);
                }
        }

        if (emac->link) {
                /* reactivate the transmit queue */
                netif_tx_wake_all_queues(ndev);
        } else {
                netif_tx_stop_all_queues(ndev);
                prueth_cleanup_tx_ts(emac);
        }
}

static enum hrtimer_restart emac_rx_timer_callback(struct hrtimer *timer)
{
        struct prueth_emac *emac =
                        container_of(timer, struct prueth_emac, rx_hrtimer);
        int rx_flow = PRUETH_RX_FLOW_DATA;

        if (emac->rx_chns.irq_disabled) {
                /* re-enable the RX IRQ */
                emac->rx_chns.irq_disabled = false;
                enable_irq(emac->rx_chns.irq[rx_flow]);
        }
        return HRTIMER_NORESTART;
}

static int emac_phy_connect(struct prueth_emac *emac)
{
        struct prueth *prueth = emac->prueth;
        struct net_device *ndev = emac->ndev;
        /* connect PHY */
        ndev->phydev = of_phy_connect(emac->ndev, emac->phy_node,
                                      &emac_adjust_link, 0,
                                      emac->phy_if);
        if (!ndev->phydev) {
                dev_err(prueth->dev, "couldn't connect to phy %s\n",
                        emac->phy_node->full_name);
                return -ENODEV;
        }

        if (!emac->half_duplex) {
                dev_dbg(prueth->dev, "half duplex mode is not supported\n");
                phy_remove_link_mode(ndev->phydev, ETHTOOL_LINK_MODE_10baseT_Half_BIT);
                phy_remove_link_mode(ndev->phydev, ETHTOOL_LINK_MODE_100baseT_Half_BIT);
        }

        /* remove unsupported modes */
        phy_remove_link_mode(ndev->phydev, ETHTOOL_LINK_MODE_1000baseT_Half_BIT);
        phy_remove_link_mode(ndev->phydev, ETHTOOL_LINK_MODE_Pause_BIT);
        phy_remove_link_mode(ndev->phydev, ETHTOOL_LINK_MODE_Asym_Pause_BIT);

        if (emac->phy_if == PHY_INTERFACE_MODE_MII)
                phy_set_max_speed(ndev->phydev, SPEED_100);

        return 0;
}

static u64 prueth_iep_gettime(void *clockops_data, struct ptp_system_timestamp *sts)
{
        u32 hi_rollover_count, hi_rollover_count_r;
        struct prueth_emac *emac = clockops_data;
        struct prueth *prueth = emac->prueth;
        void __iomem *fw_hi_r_count_addr;
        void __iomem *fw_count_hi_addr;
        u32 iepcount_hi, iepcount_hi_r;
        unsigned long flags;
        u32 iepcount_lo;
        u64 ts = 0;

        fw_count_hi_addr = prueth->shram.va + TIMESYNC_FW_WC_COUNT_HI_SW_OFFSET_OFFSET;
        fw_hi_r_count_addr = prueth->shram.va + TIMESYNC_FW_WC_HI_ROLLOVER_COUNT_OFFSET;

        local_irq_save(flags);
        do {
                iepcount_hi = icss_iep_get_count_hi(emac->iep);
                iepcount_hi += readl(fw_count_hi_addr);
                hi_rollover_count = readl(fw_hi_r_count_addr);
                ptp_read_system_prets(sts);
                iepcount_lo = icss_iep_get_count_low(emac->iep);
                ptp_read_system_postts(sts);

                iepcount_hi_r = icss_iep_get_count_hi(emac->iep);
                iepcount_hi_r += readl(fw_count_hi_addr);
                hi_rollover_count_r = readl(fw_hi_r_count_addr);
        } while ((iepcount_hi_r != iepcount_hi) ||
                 (hi_rollover_count != hi_rollover_count_r));
        local_irq_restore(flags);

        ts = ((u64)hi_rollover_count) << 23 | iepcount_hi;
        ts = ts * (u64)IEP_DEFAULT_CYCLE_TIME_NS + iepcount_lo;

        return ts;
}

static void prueth_iep_settime(void *clockops_data, u64 ns)
{
        struct icssg_setclock_desc __iomem *sc_descp;
        struct prueth_emac *emac = clockops_data;
        struct icssg_setclock_desc sc_desc;
        u64 cyclecount;
        u32 cycletime;
        int timeout;

        sc_descp = emac->prueth->shram.va + TIMESYNC_FW_WC_SETCLOCK_DESC_OFFSET;

        cycletime = IEP_DEFAULT_CYCLE_TIME_NS;
        cyclecount = ns / cycletime;

        memset(&sc_desc, 0, sizeof(sc_desc));
        sc_desc.margin = cycletime - 1000;
        sc_desc.cyclecounter0_set = cyclecount & GENMASK(31, 0);
        sc_desc.cyclecounter1_set = (cyclecount & GENMASK(63, 32)) >> 32;
        sc_desc.iepcount_set = ns % cycletime;
        /* Count from 0 to (cycle time) - emac->iep->def_inc */
        sc_desc.CMP0_current = cycletime - emac->iep->def_inc;

        memcpy_toio(sc_descp, &sc_desc, sizeof(sc_desc));

        writeb(1, &sc_descp->request);

        timeout = 5;    /* fw should take 2-3 ms */
        while (timeout--) {
                if (readb(&sc_descp->acknowledgment))
                        return;

                usleep_range(500, 1000);
        }

        dev_err(emac->prueth->dev, "settime timeout\n");
}

static int prueth_perout_enable(void *clockops_data,
                                struct ptp_perout_request *req, int on,
                                u64 *cmp)
{
        struct prueth_emac *emac = clockops_data;
        u32 reduction_factor = 0, offset = 0;
        struct timespec64 ts;
        u64 current_cycle;
        u64 start_offset;
        u64 ns_period;

        if (!on)
                return 0;

        /* Any firmware specific stuff for PPS/PEROUT handling */
        ts.tv_sec = req->period.sec;
        ts.tv_nsec = req->period.nsec;
        ns_period = timespec64_to_ns(&ts);

        /* f/w doesn't support period less than cycle time */
        if (ns_period < IEP_DEFAULT_CYCLE_TIME_NS)
                return -ENXIO;

        reduction_factor = ns_period / IEP_DEFAULT_CYCLE_TIME_NS;
        offset = ns_period % IEP_DEFAULT_CYCLE_TIME_NS;

        /* f/w requires at least 1uS within a cycle so CMP
         * can trigger after SYNC is enabled
         */
        if (offset < 5 * NSEC_PER_USEC)
                offset = 5 * NSEC_PER_USEC;

        /* if offset is close to cycle time then we will miss
         * the CMP event for last tick when IEP rolls over.
         * In normal mode, IEP tick is 4ns.
         * In slow compensation it could be 0ns or 8ns at
         * every slow compensation cycle.
         */
        if (offset > IEP_DEFAULT_CYCLE_TIME_NS - 8)
                offset = IEP_DEFAULT_CYCLE_TIME_NS - 8;

        /* we're in shadow mode so need to set upper 32-bits */
        *cmp = (u64)offset << 32;

        writel(reduction_factor, emac->prueth->shram.va +
                TIMESYNC_FW_WC_SYNCOUT_REDUCTION_FACTOR_OFFSET);

        current_cycle = icssg_read_time(emac->prueth->shram.va +
                                        TIMESYNC_FW_WC_CYCLECOUNT_OFFSET);

        /* Rounding of current_cycle count to next second */
        start_offset = roundup(current_cycle, MSEC_PER_SEC);

        hi_lo_writeq(start_offset, emac->prueth->shram.va +
                     TIMESYNC_FW_WC_SYNCOUT_START_TIME_CYCLECOUNT_OFFSET);

        return 0;
}

const struct icss_iep_clockops prueth_iep_clockops = {
        .settime = prueth_iep_settime,
        .gettime = prueth_iep_gettime,
        .perout_enable = prueth_perout_enable,
};

static void prueth_destroy_xdp_rxqs(struct prueth_emac *emac)
{
        struct xdp_rxq_info *rxq = &emac->rx_chns.xdp_rxq;

        if (xdp_rxq_info_is_reg(rxq))
                xdp_rxq_info_unreg(rxq);
}

static int prueth_create_xdp_rxqs(struct prueth_emac *emac)
{
        struct xdp_rxq_info *rxq = &emac->rx_chns.xdp_rxq;
        struct page_pool *pool = emac->rx_chns.pg_pool;
        struct prueth_rx_chn *rx_chn = &emac->rx_chns;
        int ret;

        ret = xdp_rxq_info_reg(rxq, emac->ndev, 0, emac->napi_rx.napi_id);
        if (ret)
                return ret;

        if (rx_chn->xsk_pool) {
                ret = xdp_rxq_info_reg_mem_model(rxq, MEM_TYPE_XSK_BUFF_POOL, NULL);
                if (ret)
                        goto xdp_unreg;
                xsk_pool_set_rxq_info(rx_chn->xsk_pool, rxq);
        } else {
                ret = xdp_rxq_info_reg_mem_model(rxq, MEM_TYPE_PAGE_POOL, pool);
                if (ret)
                        goto xdp_unreg;
        }

        return 0;

xdp_unreg:
        prueth_destroy_xdp_rxqs(emac);
        return ret;
}

static int icssg_prueth_add_mcast(struct net_device *ndev, const u8 *addr)
{
        struct net_device *real_dev;
        struct prueth_emac *emac;
        int port_mask;
        u8 vlan_id;

        vlan_id = is_vlan_dev(ndev) ? vlan_dev_vlan_id(ndev) : PRUETH_DFLT_VLAN_MAC;
        real_dev = is_vlan_dev(ndev) ? vlan_dev_real_dev(ndev) : ndev;
        emac = netdev_priv(real_dev);

        port_mask = BIT(emac->port_id) | icssg_fdb_lookup(emac, addr, vlan_id);
        icssg_fdb_add_del(emac, addr, vlan_id, port_mask, true);
        icssg_vtbl_modify(emac, vlan_id, port_mask, port_mask, true);

        return 0;
}

static int icssg_prueth_del_mcast(struct net_device *ndev, const u8 *addr)
{
        struct net_device *real_dev;
        struct prueth_emac *emac;
        int other_port_mask;
        int port_mask;
        u8 vlan_id;

        vlan_id = is_vlan_dev(ndev) ? vlan_dev_vlan_id(ndev) : PRUETH_DFLT_VLAN_MAC;
        real_dev = is_vlan_dev(ndev) ? vlan_dev_real_dev(ndev) : ndev;
        emac = netdev_priv(real_dev);

        port_mask = BIT(emac->port_id);
        other_port_mask = port_mask ^ icssg_fdb_lookup(emac, addr, vlan_id);

        icssg_fdb_add_del(emac, addr, vlan_id, port_mask, false);
        icssg_vtbl_modify(emac, vlan_id, port_mask, port_mask, false);

        if (other_port_mask) {
                icssg_fdb_add_del(emac, addr, vlan_id, other_port_mask, true);
                icssg_vtbl_modify(emac, vlan_id, other_port_mask,
                                  other_port_mask, true);
        }

        return 0;
}

static void icssg_prueth_hsr_fdb_add_del(struct prueth_emac *emac,
                                         const u8 *addr, u8 vid, bool add)
{
        icssg_fdb_add_del(emac, addr, vid,
                          ICSSG_FDB_ENTRY_P0_MEMBERSHIP |
                          ICSSG_FDB_ENTRY_P1_MEMBERSHIP |
                          ICSSG_FDB_ENTRY_P2_MEMBERSHIP |
                          ICSSG_FDB_ENTRY_BLOCK, add);

        if (add)
                icssg_vtbl_modify(emac, vid, BIT(emac->port_id),
                                  BIT(emac->port_id), add);
}

static int icssg_prueth_hsr_add_mcast(struct net_device *ndev, const u8 *addr)
{
        struct net_device *real_dev, *port_dev;
        struct prueth_emac *emac;
        u8 vlan_id, i;

        vlan_id = is_vlan_dev(ndev) ? vlan_dev_vlan_id(ndev) : PRUETH_DFLT_VLAN_HSR;
        real_dev = is_vlan_dev(ndev) ? vlan_dev_real_dev(ndev) : ndev;

        if (is_hsr_master(real_dev)) {
                for (i = HSR_PT_SLAVE_A; i < HSR_PT_INTERLINK; i++) {
                        port_dev = hsr_get_port_ndev(real_dev, i);
                        emac = netdev_priv(port_dev);
                        if (!emac) {
                                dev_put(port_dev);
                                return -EINVAL;
                        }
                        icssg_prueth_hsr_fdb_add_del(emac, addr, vlan_id,
                                                     true);
                        dev_put(port_dev);
                }
        } else {
                emac = netdev_priv(real_dev);
                icssg_prueth_hsr_fdb_add_del(emac, addr, vlan_id, true);
        }

        return 0;
}

static int icssg_prueth_hsr_del_mcast(struct net_device *ndev, const u8 *addr)
{
        struct net_device *real_dev, *port_dev;
        struct prueth_emac *emac;
        u8 vlan_id, i;

        vlan_id = is_vlan_dev(ndev) ? vlan_dev_vlan_id(ndev) : PRUETH_DFLT_VLAN_HSR;
        real_dev = is_vlan_dev(ndev) ? vlan_dev_real_dev(ndev) : ndev;

        if (is_hsr_master(real_dev)) {
                for (i = HSR_PT_SLAVE_A; i < HSR_PT_INTERLINK; i++) {
                        port_dev = hsr_get_port_ndev(real_dev, i);
                        emac = netdev_priv(port_dev);
                        if (!emac) {
                                dev_put(port_dev);
                                return -EINVAL;
                        }
                        icssg_prueth_hsr_fdb_add_del(emac, addr, vlan_id,
                                                     false);
                        dev_put(port_dev);
                }
        } else {
                emac = netdev_priv(real_dev);
                icssg_prueth_hsr_fdb_add_del(emac, addr, vlan_id, false);
        }

        return 0;
}

static int icssg_update_vlan_mcast(struct net_device *vdev, int vid,
                                   void *args)
{
        struct prueth_emac *emac = args;

        if (!vdev || !vid)
                return 0;

        netif_addr_lock_bh(vdev);
        __hw_addr_sync_multiple(&emac->vlan_mcast_list[vid], &vdev->mc,
                                vdev->addr_len);
        netif_addr_unlock_bh(vdev);

        if (emac->prueth->is_hsr_offload_mode)
                __hw_addr_sync_dev(&emac->vlan_mcast_list[vid], vdev,
                                   icssg_prueth_hsr_add_mcast,
                                   icssg_prueth_hsr_del_mcast);
        else
                __hw_addr_sync_dev(&emac->vlan_mcast_list[vid], vdev,
                                   icssg_prueth_add_mcast,
                                   icssg_prueth_del_mcast);

        return 0;
}

static void prueth_set_xsk_pool(struct prueth_emac *emac, u16 queue_id)
{
        struct prueth_tx_chn *tx_chn = &emac->tx_chns[queue_id];
        struct prueth_rx_chn *rx_chn = &emac->rx_chns;

        if (emac->xsk_qid != queue_id) {
                rx_chn->xsk_pool = NULL;
                tx_chn->xsk_pool = NULL;
        } else {
                rx_chn->xsk_pool = xsk_get_pool_from_qid(emac->ndev, queue_id);
                tx_chn->xsk_pool = xsk_get_pool_from_qid(emac->ndev, queue_id);
        }
}

static void prueth_destroy_txq(struct prueth_emac *emac)
{
        int ret, i;

        atomic_set(&emac->tdown_cnt, emac->tx_ch_num);
        /* ensure new tdown_cnt value is visible */
        smp_mb__after_atomic();
        /* tear down and disable UDMA channels */
        reinit_completion(&emac->tdown_complete);
        for (i = 0; i < emac->tx_ch_num; i++)
                k3_udma_glue_tdown_tx_chn(emac->tx_chns[i].tx_chn, false);

        ret = wait_for_completion_timeout(&emac->tdown_complete,
                                          msecs_to_jiffies(1000));
        if (!ret)
                netdev_err(emac->ndev, "tx teardown timeout\n");

        for (i = 0; i < emac->tx_ch_num; i++) {
                napi_disable(&emac->tx_chns[i].napi_tx);
                hrtimer_cancel(&emac->tx_chns[i].tx_hrtimer);
                k3_udma_glue_reset_tx_chn(emac->tx_chns[i].tx_chn,
                                          &emac->tx_chns[i],
                                          prueth_tx_cleanup);
                k3_udma_glue_disable_tx_chn(emac->tx_chns[i].tx_chn);
        }
}

static void prueth_destroy_rxq(struct prueth_emac *emac)
{
        int i, ret;

        /* tear down and disable UDMA channels */
        reinit_completion(&emac->tdown_complete);
        k3_udma_glue_tdown_rx_chn(emac->rx_chns.rx_chn, true);

        /* When RX DMA Channel Teardown is initiated, it will result in an
         * interrupt and a Teardown Completion Marker (TDCM) is queued into
         * the RX Completion queue. Acknowledging the interrupt involves
         * popping the TDCM descriptor from the RX Completion queue via the
         * RX NAPI Handler. To avoid timing out when waiting for the TDCM to
         * be popped, schedule the RX NAPI handler to run immediately.
         */
        if (!napi_if_scheduled_mark_missed(&emac->napi_rx)) {
                if (napi_schedule_prep(&emac->napi_rx))
                        __napi_schedule(&emac->napi_rx);
        }

        ret = wait_for_completion_timeout(&emac->tdown_complete,
                                          msecs_to_jiffies(1000));
        if (!ret)
                netdev_err(emac->ndev, "rx teardown timeout\n");

        for (i = 0; i < PRUETH_MAX_RX_FLOWS; i++) {
                napi_disable(&emac->napi_rx);
                hrtimer_cancel(&emac->rx_hrtimer);
                k3_udma_glue_reset_rx_chn(emac->rx_chns.rx_chn, i,
                                          &emac->rx_chns,
                                          prueth_rx_cleanup);
        }

        prueth_destroy_xdp_rxqs(emac);
        k3_udma_glue_disable_rx_chn(emac->rx_chns.rx_chn);
}

static int prueth_create_txq(struct prueth_emac *emac)
{
        int ret, i;

        for (i = 0; i < emac->tx_ch_num; i++) {
                ret = k3_udma_glue_enable_tx_chn(emac->tx_chns[i].tx_chn);
                if (ret)
                        goto reset_tx_chan;
                napi_enable(&emac->tx_chns[i].napi_tx);
        }
        return 0;

reset_tx_chan:
        /* Since interface is not yet up, there is wouldn't be
         * any SKB for completion. So set false to free_skb
         */
        prueth_reset_tx_chan(emac, i, false);
        return ret;
}

static int prueth_create_rxq(struct prueth_emac *emac)
{
        int ret;

        ret = prueth_prepare_rx_chan(emac, &emac->rx_chns, PRUETH_MAX_PKT_SIZE);
        if (ret)
                return ret;

        ret = k3_udma_glue_enable_rx_chn(emac->rx_chns.rx_chn);
        if (ret)
                goto reset_rx_chn;

        ret = prueth_create_xdp_rxqs(emac);
        if (ret)
                goto reset_rx_chn;

        napi_enable(&emac->napi_rx);
        return 0;

reset_rx_chn:
        prueth_reset_rx_chan(&emac->rx_chns, PRUETH_MAX_RX_FLOWS, false);
        return ret;
}

/**
 * emac_ndo_open - EMAC device open
 * @ndev: network adapter device
 *
 * Called when system wants to start the interface.
 *
 * Return: 0 for a successful open, or appropriate error code
 */
static int emac_ndo_open(struct net_device *ndev)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        int ret, num_data_chn = emac->tx_ch_num;
        struct icssg_flow_cfg __iomem *flow_cfg;
        struct prueth *prueth = emac->prueth;
        int slice = prueth_emac_slice(emac);
        struct device *dev = prueth->dev;
        int max_rx_flows;
        int rx_flow;

        /* set h/w MAC as user might have re-configured */
        ether_addr_copy(emac->mac_addr, ndev->dev_addr);

        icssg_class_set_mac_addr(prueth->miig_rt, slice, emac->mac_addr);
        icssg_ft1_set_mac_addr(prueth->miig_rt, slice, emac->mac_addr);

        /* Notify the stack of the actual queue counts. */
        ret = netif_set_real_num_tx_queues(ndev, num_data_chn);
        if (ret) {
                dev_err(dev, "cannot set real number of tx queues\n");
                return ret;
        }

        emac->xsk_qid = -EINVAL;
        init_completion(&emac->cmd_complete);
        ret = prueth_init_tx_chns(emac);
        if (ret) {
                dev_err(dev, "failed to init tx channel: %d\n", ret);
                return ret;
        }

        max_rx_flows = PRUETH_MAX_RX_FLOWS;
        ret = prueth_init_rx_chns(emac, &emac->rx_chns, "rx",
                                  max_rx_flows, PRUETH_MAX_RX_DESC);
        if (ret) {
                dev_err(dev, "failed to init rx channel: %d\n", ret);
                goto cleanup_tx;
        }

        ret = prueth_ndev_add_tx_napi(emac);
        if (ret)
                goto cleanup_rx;

        /* we use only the highest priority flow for now i.e. @irq[3] */
        rx_flow = PRUETH_RX_FLOW_DATA;
        ret = request_irq(emac->rx_chns.irq[rx_flow], prueth_rx_irq,
                          IRQF_TRIGGER_HIGH, dev_name(dev), emac);
        if (ret) {
                dev_err(dev, "unable to request RX IRQ\n");
                goto cleanup_napi;
        }

        if (!prueth->emacs_initialized) {
                ret = prueth_emac_common_start(prueth);
                if (ret)
                        goto free_rx_irq;
                icssg_enable_fw_offload(prueth);
        }

        flow_cfg = emac->dram.va + ICSSG_CONFIG_OFFSET + PSI_L_REGULAR_FLOW_ID_BASE_OFFSET;
        writew(emac->rx_flow_id_base, &flow_cfg->rx_base_flow);
        ret = emac_fdb_flow_id_updated(emac);

        if (ret) {
                netdev_err(ndev, "Failed to update Rx Flow ID %d", ret);
                goto stop;
        }

        icssg_mii_update_mtu(prueth->mii_rt, slice, ndev->max_mtu);

        ret = request_threaded_irq(emac->tx_ts_irq, NULL, prueth_tx_ts_irq,
                                   IRQF_ONESHOT, dev_name(dev), emac);
        if (ret)
                goto stop;

        /* Prepare RX */
        ret = prueth_create_rxq(emac);
        if (ret)
                goto free_tx_ts_irq;

        ret = prueth_create_txq(emac);
        if (ret)
                goto destroy_rxq;

        /* start PHY */
        phy_start(ndev->phydev);

        prueth->emacs_initialized++;

        queue_work(system_long_wq, &emac->stats_work.work);

        return 0;

destroy_rxq:
        prueth_destroy_rxq(emac);
free_tx_ts_irq:
        free_irq(emac->tx_ts_irq, emac);
stop:
        if (!prueth->emacs_initialized)
                prueth_emac_common_stop(prueth);
free_rx_irq:
        free_irq(emac->rx_chns.irq[rx_flow], emac);
cleanup_napi:
        prueth_ndev_del_tx_napi(emac, emac->tx_ch_num);
cleanup_rx:
        prueth_cleanup_rx_chns(emac, &emac->rx_chns, max_rx_flows);
cleanup_tx:
        prueth_cleanup_tx_chns(emac);

        return ret;
}

/**
 * emac_ndo_stop - EMAC device stop
 * @ndev: network adapter device
 *
 * Called when system wants to stop or down the interface.
 *
 * Return: Always 0 (Success)
 */
static int emac_ndo_stop(struct net_device *ndev)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        struct prueth *prueth = emac->prueth;

        /* inform the upper layers. */
        netif_tx_stop_all_queues(ndev);

        /* block packets from wire */
        if (ndev->phydev)
                phy_stop(ndev->phydev);

        if (emac->prueth->is_hsr_offload_mode)
                __dev_mc_unsync(ndev, icssg_prueth_hsr_del_mcast);
        else
                __dev_mc_unsync(ndev, icssg_prueth_del_mcast);

        prueth_destroy_txq(emac);
        prueth_destroy_rxq(emac);

        cancel_work_sync(&emac->rx_mode_work);

        /* Destroying the queued work in ndo_stop() */
        cancel_delayed_work_sync(&emac->stats_work);

        /* stop PRUs */
        if (prueth->emacs_initialized == 1)
                prueth_emac_common_stop(prueth);

        free_irq(emac->tx_ts_irq, emac);

        free_irq(emac->rx_chns.irq[PRUETH_RX_FLOW_DATA], emac);
        prueth_ndev_del_tx_napi(emac, emac->tx_ch_num);

        prueth_cleanup_rx_chns(emac, &emac->rx_chns, PRUETH_MAX_RX_FLOWS);
        prueth_cleanup_tx_chns(emac);

        prueth->emacs_initialized--;

        return 0;
}

static void emac_ndo_set_rx_mode_work(struct work_struct *work)
{
        struct prueth_emac *emac = container_of(work, struct prueth_emac, rx_mode_work);
        struct net_device *ndev = emac->ndev;
        bool promisc, allmulti;

        if (!netif_running(ndev))
                return;

        promisc = ndev->flags & IFF_PROMISC;
        allmulti = ndev->flags & IFF_ALLMULTI;
        icssg_set_port_state(emac, ICSSG_EMAC_PORT_UC_FLOODING_DISABLE);
        icssg_set_port_state(emac, ICSSG_EMAC_PORT_MC_FLOODING_DISABLE);

        if (promisc) {
                icssg_set_port_state(emac, ICSSG_EMAC_PORT_UC_FLOODING_ENABLE);
                icssg_set_port_state(emac, ICSSG_EMAC_PORT_MC_FLOODING_ENABLE);
                return;
        }

        if (allmulti) {
                icssg_set_port_state(emac, ICSSG_EMAC_PORT_MC_FLOODING_ENABLE);
                return;
        }

        if (emac->prueth->is_hsr_offload_mode) {
                __dev_mc_sync(ndev, icssg_prueth_hsr_add_mcast,
                              icssg_prueth_hsr_del_mcast);
                if (rtnl_trylock()) {
                        vlan_for_each(emac->prueth->hsr_dev,
                                      icssg_update_vlan_mcast, emac);
                        rtnl_unlock();
                }
        } else {
                __dev_mc_sync(ndev, icssg_prueth_add_mcast,
                              icssg_prueth_del_mcast);
                if (rtnl_trylock()) {
                        vlan_for_each(ndev, icssg_update_vlan_mcast, emac);
                        rtnl_unlock();
                }
        }
}

/**
 * emac_ndo_set_rx_mode - EMAC set receive mode function
 * @ndev: The EMAC network adapter
 *
 * Called when system wants to set the receive mode of the device.
 *
 */
static void emac_ndo_set_rx_mode(struct net_device *ndev)
{
        struct prueth_emac *emac = netdev_priv(ndev);

        schedule_work(&emac->rx_mode_work);
}

static netdev_features_t emac_ndo_fix_features(struct net_device *ndev,
                                               netdev_features_t features)
{
        /* hsr tag insertion offload and hsr dup offload are tightly coupled in
         * firmware implementation. Both these features need to be enabled /
         * disabled together.
         */
        if (!(ndev->features & (NETIF_F_HW_HSR_DUP | NETIF_F_HW_HSR_TAG_INS)))
                if ((features & NETIF_F_HW_HSR_DUP) ||
                    (features & NETIF_F_HW_HSR_TAG_INS))
                        features |= NETIF_F_HW_HSR_DUP |
                                    NETIF_F_HW_HSR_TAG_INS;

        if ((ndev->features & NETIF_F_HW_HSR_DUP) ||
            (ndev->features & NETIF_F_HW_HSR_TAG_INS))
                if (!(features & NETIF_F_HW_HSR_DUP) ||
                    !(features & NETIF_F_HW_HSR_TAG_INS))
                        features &= ~(NETIF_F_HW_HSR_DUP |
                                      NETIF_F_HW_HSR_TAG_INS);

        return features;
}

static int emac_ndo_vlan_rx_add_vid(struct net_device *ndev,
                                    __be16 proto, u16 vid)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        struct prueth *prueth = emac->prueth;
        int port_mask = BIT(emac->port_id);
        int untag_mask = 0;

        if (prueth->is_hsr_offload_mode)
                port_mask |= BIT(PRUETH_PORT_HOST);

        __hw_addr_init(&emac->vlan_mcast_list[vid]);
        netdev_dbg(emac->ndev, "VID add vid:%u port_mask:%X untag_mask %X\n",
                   vid, port_mask, untag_mask);

        icssg_vtbl_modify(emac, vid, port_mask, untag_mask, true);
        icssg_set_pvid(emac->prueth, vid, emac->port_id);

        return 0;
}

static int emac_ndo_vlan_rx_del_vid(struct net_device *ndev,
                                    __be16 proto, u16 vid)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        struct prueth *prueth = emac->prueth;
        int port_mask = BIT(emac->port_id);
        int untag_mask = 0;

        if (prueth->is_hsr_offload_mode)
                port_mask = BIT(PRUETH_PORT_HOST);

        netdev_dbg(emac->ndev, "VID del vid:%u port_mask:%X untag_mask  %X\n",
                   vid, port_mask, untag_mask);
        icssg_vtbl_modify(emac, vid, port_mask, untag_mask, false);

        return 0;
}

/**
 * emac_xdp_xmit - Implements ndo_xdp_xmit
 * @dev: netdev
 * @n: number of frames
 * @frames: array of XDP buffer pointers
 * @flags: XDP extra info
 *
 * Return: number of frames successfully sent. Failed frames
 * will be free'ed by XDP core.
 *
 * For error cases, a negative errno code is returned and no-frames
 * are transmitted (caller must handle freeing frames).
 **/
static int emac_xdp_xmit(struct net_device *dev, int n, struct xdp_frame **frames,
                         u32 flags)
{
        struct prueth_emac *emac = netdev_priv(dev);
        struct net_device *ndev = emac->ndev;
        struct netdev_queue *netif_txq;
        int cpu = smp_processor_id();
        struct xdp_frame *xdpf;
        unsigned int q_idx;
        int nxmit = 0;
        u32 err;
        int i;

        q_idx = cpu % emac->tx_ch_num;
        netif_txq = netdev_get_tx_queue(ndev, q_idx);

        if (unlikely(flags & ~XDP_XMIT_FLAGS_MASK))
                return -EINVAL;

        __netif_tx_lock(netif_txq, cpu);
        for (i = 0; i < n; i++) {
                xdpf = frames[i];
                err = emac_xmit_xdp_frame(emac, xdpf, q_idx,
                                          PRUETH_TX_BUFF_TYPE_XDP_NDO);
                if (err != ICSSG_XDP_TX) {
                        ndev->stats.tx_dropped++;
                        break;
                }
                nxmit++;
        }
        __netif_tx_unlock(netif_txq);

        return nxmit;
}

/**
 * emac_xdp_setup - add/remove an XDP program
 * @emac: emac device
 * @bpf: XDP program
 *
 * Return: Always 0 (Success)
 **/
static int emac_xdp_setup(struct prueth_emac *emac, struct netdev_bpf *bpf)
{
        struct bpf_prog *prog = bpf->prog;

        if (!emac->xdpi.prog && !prog)
                return 0;

        WRITE_ONCE(emac->xdp_prog, prog);

        xdp_attachment_setup(&emac->xdpi, bpf);

        return 0;
}

static int prueth_xsk_pool_enable(struct prueth_emac *emac,
                                  struct xsk_buff_pool *pool, u16 queue_id)
{
        struct prueth_rx_chn *rx_chn = &emac->rx_chns;
        u32 frame_size;
        int ret;

        if (queue_id >= PRUETH_MAX_RX_FLOWS ||
            queue_id >= emac->tx_ch_num) {
                netdev_err(emac->ndev, "Invalid XSK queue ID %d\n", queue_id);
                return -EINVAL;
        }

        frame_size = xsk_pool_get_rx_frame_size(pool);
        if (frame_size < PRUETH_MAX_PKT_SIZE)
                return -EOPNOTSUPP;

        ret = xsk_pool_dma_map(pool, rx_chn->dma_dev, PRUETH_RX_DMA_ATTR);
        if (ret) {
                netdev_err(emac->ndev, "Failed to map XSK pool: %d\n", ret);
                return ret;
        }

        if (netif_running(emac->ndev)) {
                /* stop packets from wire for graceful teardown */
                ret = icssg_set_port_state(emac, ICSSG_EMAC_PORT_DISABLE);
                if (ret)
                        return ret;
                prueth_destroy_rxq(emac);
        }

        emac->xsk_qid = queue_id;
        prueth_set_xsk_pool(emac, queue_id);

        if (netif_running(emac->ndev)) {
                ret = prueth_create_rxq(emac);
                if (ret) {
                        netdev_err(emac->ndev, "Failed to create RX queue: %d\n", ret);
                        return ret;
                }
                ret = icssg_set_port_state(emac, ICSSG_EMAC_PORT_FORWARD);
                if (ret) {
                        prueth_destroy_rxq(emac);
                        return ret;
                }
                ret = prueth_xsk_wakeup(emac->ndev, queue_id, XDP_WAKEUP_RX);
                if (ret)
                        return ret;
        }

        return 0;
}

static int prueth_xsk_pool_disable(struct prueth_emac *emac, u16 queue_id)
{
        struct xsk_buff_pool *pool;
        int ret;

        if (queue_id >= PRUETH_MAX_RX_FLOWS ||
            queue_id >= emac->tx_ch_num) {
                netdev_err(emac->ndev, "Invalid XSK queue ID %d\n", queue_id);
                return -EINVAL;
        }

        if (emac->xsk_qid != queue_id) {
                netdev_err(emac->ndev, "XSK queue ID %d not registered\n", queue_id);
                return -EINVAL;
        }

        pool = xsk_get_pool_from_qid(emac->ndev, queue_id);
        if (!pool) {
                netdev_err(emac->ndev, "No XSK pool registered for queue %d\n", queue_id);
                return -EINVAL;
        }

        if (netif_running(emac->ndev)) {
                /* stop packets from wire for graceful teardown */
                ret = icssg_set_port_state(emac, ICSSG_EMAC_PORT_DISABLE);
                if (ret)
                        return ret;
                prueth_destroy_rxq(emac);
        }

        xsk_pool_dma_unmap(pool, PRUETH_RX_DMA_ATTR);
        emac->xsk_qid = -EINVAL;
        prueth_set_xsk_pool(emac, queue_id);

        if (netif_running(emac->ndev)) {
                ret = prueth_create_rxq(emac);
                if (ret) {
                        netdev_err(emac->ndev, "Failed to create RX queue: %d\n", ret);
                        return ret;
                }
                ret = icssg_set_port_state(emac, ICSSG_EMAC_PORT_FORWARD);
                if (ret) {
                        prueth_destroy_rxq(emac);
                        return ret;
                }
        }

        return 0;
}

/**
 * emac_ndo_bpf - implements ndo_bpf for icssg_prueth
 * @ndev: network adapter device
 * @bpf: XDP program
 *
 * Return: 0 on success, error code on failure.
 **/
static int emac_ndo_bpf(struct net_device *ndev, struct netdev_bpf *bpf)
{
        struct prueth_emac *emac = netdev_priv(ndev);

        switch (bpf->command) {
        case XDP_SETUP_PROG:
                return emac_xdp_setup(emac, bpf);
        case XDP_SETUP_XSK_POOL:
                return bpf->xsk.pool ?
                        prueth_xsk_pool_enable(emac, bpf->xsk.pool, bpf->xsk.queue_id) :
                        prueth_xsk_pool_disable(emac, bpf->xsk.queue_id);
        default:
                return -EINVAL;
        }
}

int prueth_xsk_wakeup(struct net_device *ndev, u32 qid, u32 flags)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        struct prueth_tx_chn *tx_chn = &emac->tx_chns[qid];
        struct prueth_rx_chn *rx_chn = &emac->rx_chns;

        if (emac->xsk_qid != qid) {
                netdev_err(ndev, "XSK queue %d not registered\n", qid);
                return -EINVAL;
        }

        if (qid >= PRUETH_MAX_RX_FLOWS || qid >= emac->tx_ch_num) {
                netdev_err(ndev, "Invalid XSK queue ID %d\n", qid);
                return -EINVAL;
        }

        if (!tx_chn->xsk_pool) {
                netdev_err(ndev, "XSK pool not registered for queue %d\n", qid);
                return -EINVAL;
        }

        if (!rx_chn->xsk_pool) {
                netdev_err(ndev, "XSK pool not registered for RX queue %d\n", qid);
                return -EINVAL;
        }

        if (flags & XDP_WAKEUP_TX) {
                if (!napi_if_scheduled_mark_missed(&tx_chn->napi_tx)) {
                        if (likely(napi_schedule_prep(&tx_chn->napi_tx)))
                                __napi_schedule(&tx_chn->napi_tx);
                }
        }

        if (flags & XDP_WAKEUP_RX) {
                if (!napi_if_scheduled_mark_missed(&emac->napi_rx)) {
                        if (likely(napi_schedule_prep(&emac->napi_rx)))
                                __napi_schedule(&emac->napi_rx);
                }
        }

        return 0;
}

static const struct net_device_ops emac_netdev_ops = {
        .ndo_open = emac_ndo_open,
        .ndo_stop = emac_ndo_stop,
        .ndo_start_xmit = icssg_ndo_start_xmit,
        .ndo_set_mac_address = eth_mac_addr,
        .ndo_validate_addr = eth_validate_addr,
        .ndo_tx_timeout = icssg_ndo_tx_timeout,
        .ndo_set_rx_mode = emac_ndo_set_rx_mode,
        .ndo_eth_ioctl = phy_do_ioctl,
        .ndo_get_stats64 = icssg_ndo_get_stats64,
        .ndo_get_phys_port_name = icssg_ndo_get_phys_port_name,
        .ndo_fix_features = emac_ndo_fix_features,
        .ndo_vlan_rx_add_vid = emac_ndo_vlan_rx_add_vid,
        .ndo_vlan_rx_kill_vid = emac_ndo_vlan_rx_del_vid,
        .ndo_bpf = emac_ndo_bpf,
        .ndo_xdp_xmit = emac_xdp_xmit,
        .ndo_hwtstamp_get = icssg_ndo_get_ts_config,
        .ndo_hwtstamp_set = icssg_ndo_set_ts_config,
        .ndo_xsk_wakeup = prueth_xsk_wakeup,
};

static int prueth_netdev_init(struct prueth *prueth,
                              struct device_node *eth_node)
{
        int ret, num_tx_chn = PRUETH_MAX_TX_QUEUES;
        struct prueth_emac *emac;
        struct net_device *ndev;
        enum prueth_port port;
        const char *irq_name;
        enum prueth_mac mac;

        port = prueth_node_port(eth_node);
        if (port == PRUETH_PORT_INVALID)
                return -EINVAL;

        mac = prueth_node_mac(eth_node);
        if (mac == PRUETH_MAC_INVALID)
                return -EINVAL;

        ndev = alloc_etherdev_mq(sizeof(*emac), num_tx_chn);
        if (!ndev)
                return -ENOMEM;

        emac = netdev_priv(ndev);
        emac->prueth = prueth;
        emac->ndev = ndev;
        emac->port_id = port;
        emac->xdp_prog = NULL;
        emac->ndev->pcpu_stat_type = NETDEV_PCPU_STAT_TSTATS;
        INIT_WORK(&emac->rx_mode_work, emac_ndo_set_rx_mode_work);

        INIT_DELAYED_WORK(&emac->stats_work, icssg_stats_work_handler);

        ret = pruss_request_mem_region(prueth->pruss,
                                       port == PRUETH_PORT_MII0 ?
                                       PRUSS_MEM_DRAM0 : PRUSS_MEM_DRAM1,
                                       &emac->dram);
        if (ret) {
                dev_err(prueth->dev, "unable to get DRAM: %d\n", ret);
                ret = -ENOMEM;
                goto free_ndev;
        }

        emac->tx_ch_num = 1;

        irq_name = "tx_ts0";
        if (emac->port_id == PRUETH_PORT_MII1)
                irq_name = "tx_ts1";
        emac->tx_ts_irq = platform_get_irq_byname_optional(prueth->pdev, irq_name);
        if (emac->tx_ts_irq < 0) {
                ret = dev_err_probe(prueth->dev, emac->tx_ts_irq, "could not get tx_ts_irq\n");
                goto free;
        }

        SET_NETDEV_DEV(ndev, prueth->dev);
        spin_lock_init(&emac->lock);
        mutex_init(&emac->cmd_lock);

        emac->phy_node = of_parse_phandle(eth_node, "phy-handle", 0);
        if (!emac->phy_node && !of_phy_is_fixed_link(eth_node)) {
                dev_err(prueth->dev, "couldn't find phy-handle\n");
                ret = -ENODEV;
                goto free;
        } else if (of_phy_is_fixed_link(eth_node)) {
                ret = of_phy_register_fixed_link(eth_node);
                if (ret) {
                        dev_err_probe(prueth->dev, ret, "failed to register fixed-link phy\n");
                        goto free;
                }

                emac->phy_node = eth_node;
        }

        ret = of_get_phy_mode(eth_node, &emac->phy_if);
        if (ret) {
                dev_err(prueth->dev, "could not get phy-mode property\n");
                goto free;
        }

        if (emac->phy_if != PHY_INTERFACE_MODE_MII &&
            !phy_interface_mode_is_rgmii(emac->phy_if)) {
                dev_err(prueth->dev, "PHY mode unsupported %s\n", phy_modes(emac->phy_if));
                ret = -EINVAL;
                goto free;
        }

        /* AM65 SR2.0 has TX Internal delay always enabled by hardware
         * and it is not possible to disable TX Internal delay. The below
         * switch case block describes how we handle different phy modes
         * based on hardware restriction.
         */
        switch (emac->phy_if) {
        case PHY_INTERFACE_MODE_RGMII_ID:
                emac->phy_if = PHY_INTERFACE_MODE_RGMII_RXID;
                break;
        case PHY_INTERFACE_MODE_RGMII_TXID:
                emac->phy_if = PHY_INTERFACE_MODE_RGMII;
                break;
        case PHY_INTERFACE_MODE_RGMII:
        case PHY_INTERFACE_MODE_RGMII_RXID:
                dev_err(prueth->dev, "RGMII mode without TX delay is not supported");
                ret = -EINVAL;
                goto free;
        default:
                break;
        }

        /* get mac address from DT and set private and netdev addr */
        ret = of_get_ethdev_address(eth_node, ndev);
        if (!is_valid_ether_addr(ndev->dev_addr)) {
                eth_hw_addr_random(ndev);
                dev_warn(prueth->dev, "port %d: using random MAC addr: %pM\n",
                         port, ndev->dev_addr);
        }
        ether_addr_copy(emac->mac_addr, ndev->dev_addr);

        ndev->dev.of_node = eth_node;
        ndev->min_mtu = PRUETH_MIN_PKT_SIZE;
        ndev->max_mtu = PRUETH_MAX_MTU;
        ndev->netdev_ops = &emac_netdev_ops;
        ndev->ethtool_ops = &icssg_ethtool_ops;
        ndev->hw_features = NETIF_F_SG;
        ndev->features = ndev->hw_features | NETIF_F_HW_VLAN_CTAG_FILTER;
        ndev->hw_features |= NETIF_PRUETH_HSR_OFFLOAD_FEATURES;
        xdp_set_features_flag(ndev,
                              NETDEV_XDP_ACT_BASIC |
                              NETDEV_XDP_ACT_REDIRECT |
                              NETDEV_XDP_ACT_NDO_XMIT |
                              NETDEV_XDP_ACT_XSK_ZEROCOPY);

        netif_napi_add(ndev, &emac->napi_rx, icssg_napi_rx_poll);
        hrtimer_setup(&emac->rx_hrtimer, &emac_rx_timer_callback, CLOCK_MONOTONIC,
                      HRTIMER_MODE_REL_PINNED);
        prueth->emac[mac] = emac;

        return 0;

free:
        pruss_release_mem_region(prueth->pruss, &emac->dram);
free_ndev:
        emac->ndev = NULL;
        prueth->emac[mac] = NULL;
        free_netdev(ndev);

        return ret;
}

bool prueth_dev_check(const struct net_device *ndev)
{
        if (ndev->netdev_ops == &emac_netdev_ops && netif_running(ndev)) {
                struct prueth_emac *emac = netdev_priv(ndev);

                return emac->prueth->is_switch_mode;
        }

        return false;
}

static void prueth_offload_fwd_mark_update(struct prueth *prueth)
{
        int set_val = 0;
        int i;

        if (prueth->br_members == (BIT(PRUETH_PORT_MII0) | BIT(PRUETH_PORT_MII1)))
                set_val = 1;

        dev_dbg(prueth->dev, "set offload_fwd_mark %d\n", set_val);

        for (i = PRUETH_MAC0; i < PRUETH_NUM_MACS; i++) {
                struct prueth_emac *emac = prueth->emac[i];

                if (!emac || !emac->ndev)
                        continue;

                emac->offload_fwd_mark = set_val;
        }
}

static int prueth_emac_restart(struct prueth *prueth)
{
        struct prueth_emac *emac0 = prueth->emac[PRUETH_MAC0];
        struct prueth_emac *emac1 = prueth->emac[PRUETH_MAC1];
        int ret;

        /* Detach the net_device for both PRUeth ports*/
        if (netif_running(emac0->ndev))
                netif_device_detach(emac0->ndev);
        if (netif_running(emac1->ndev))
                netif_device_detach(emac1->ndev);

        /* Disable both PRUeth ports */
        ret = icssg_set_port_state(emac0, ICSSG_EMAC_PORT_DISABLE);
        ret |= icssg_set_port_state(emac1, ICSSG_EMAC_PORT_DISABLE);
        if (ret)
                return ret;

        /* Stop both pru cores for both PRUeth ports*/
        ret = prueth_emac_common_stop(prueth);
        if (ret) {
                dev_err(prueth->dev, "Failed to stop the firmwares");
                return ret;
        }

        /* Start both pru cores for both PRUeth ports */
        ret = prueth_emac_common_start(prueth);
        if (ret) {
                dev_err(prueth->dev, "Failed to start the firmwares");
                return ret;
        }

        /* Enable forwarding for both PRUeth ports */
        ret = icssg_set_port_state(emac0, ICSSG_EMAC_PORT_FORWARD);
        ret |= icssg_set_port_state(emac1, ICSSG_EMAC_PORT_FORWARD);

        /* Attache net_device for both PRUeth ports */
        netif_device_attach(emac0->ndev);
        netif_device_attach(emac1->ndev);

        return ret;
}

static void icssg_change_mode(struct prueth *prueth)
{
        int ret;

        ret = prueth_emac_restart(prueth);
        if (ret) {
                dev_err(prueth->dev, "Failed to restart the firmwares, aborting the process");
                return;
        }

        icssg_enable_fw_offload(prueth);
}

static int prueth_netdevice_port_link(struct net_device *ndev,
                                      struct net_device *br_ndev,
                                      struct netlink_ext_ack *extack)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        struct prueth *prueth = emac->prueth;
        int err;

        if (!prueth->br_members) {
                prueth->hw_bridge_dev = br_ndev;
        } else {
                /* This is adding the port to a second bridge, this is
                 * unsupported
                 */
                if (prueth->hw_bridge_dev != br_ndev)
                        return -EOPNOTSUPP;
        }

        err = switchdev_bridge_port_offload(br_ndev, ndev, emac,
                                            &prueth->prueth_switchdev_nb,
                                            &prueth->prueth_switchdev_bl_nb,
                                            false, extack);
        if (err)
                return err;

        prueth->br_members |= BIT(emac->port_id);

        if (!prueth->is_switch_mode) {
                if (prueth->br_members & BIT(PRUETH_PORT_MII0) &&
                    prueth->br_members & BIT(PRUETH_PORT_MII1)) {
                        prueth->is_switch_mode = true;
                        prueth->default_vlan = PRUETH_DFLT_VLAN_SW;
                        emac->port_vlan = prueth->default_vlan;
                        icssg_change_mode(prueth);
                }
        }

        prueth_offload_fwd_mark_update(prueth);

        return NOTIFY_DONE;
}

static void prueth_netdevice_port_unlink(struct net_device *ndev)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        struct prueth *prueth = emac->prueth;
        int ret;

        prueth->br_members &= ~BIT(emac->port_id);

        if (prueth->is_switch_mode) {
                prueth->is_switch_mode = false;
                emac->port_vlan = 0;
                ret = prueth_emac_restart(prueth);
                if (ret) {
                        dev_err(prueth->dev, "Failed to restart the firmwares, aborting the process");
                        return;
                }
        }

        prueth_offload_fwd_mark_update(prueth);

        if (!prueth->br_members)
                prueth->hw_bridge_dev = NULL;
}

static int prueth_hsr_port_link(struct net_device *ndev)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        struct prueth *prueth = emac->prueth;
        struct prueth_emac *emac0;
        struct prueth_emac *emac1;

        emac0 = prueth->emac[PRUETH_MAC0];
        emac1 = prueth->emac[PRUETH_MAC1];

        if (prueth->is_switch_mode)
                return -EOPNOTSUPP;

        prueth->hsr_members |= BIT(emac->port_id);
        if (!prueth->is_hsr_offload_mode) {
                if (prueth->hsr_members & BIT(PRUETH_PORT_MII0) &&
                    prueth->hsr_members & BIT(PRUETH_PORT_MII1)) {
                        if (!(emac0->ndev->features &
                              NETIF_PRUETH_HSR_OFFLOAD_FEATURES) &&
                            !(emac1->ndev->features &
                              NETIF_PRUETH_HSR_OFFLOAD_FEATURES))
                                return -EOPNOTSUPP;
                        prueth->is_hsr_offload_mode = true;
                        prueth->default_vlan = PRUETH_DFLT_VLAN_HSR;
                        emac0->port_vlan = prueth->default_vlan;
                        emac1->port_vlan = prueth->default_vlan;
                        icssg_change_mode(prueth);
                        netdev_dbg(ndev, "Enabling HSR offload mode\n");
                }
        }

        return 0;
}

static void prueth_hsr_port_unlink(struct net_device *ndev)
{
        struct prueth_emac *emac = netdev_priv(ndev);
        struct prueth *prueth = emac->prueth;
        struct prueth_emac *emac0;
        struct prueth_emac *emac1;
        int ret;

        emac0 = prueth->emac[PRUETH_MAC0];
        emac1 = prueth->emac[PRUETH_MAC1];

        prueth->hsr_members &= ~BIT(emac->port_id);
        if (prueth->is_hsr_offload_mode) {
                prueth->is_hsr_offload_mode = false;
                emac0->port_vlan = 0;
                emac1->port_vlan = 0;
                prueth->hsr_dev = NULL;
                ret = prueth_emac_restart(prueth);
                if (ret) {
                        dev_err(prueth->dev, "Failed to restart the firmwares, aborting the process");
                        return;
                }
                netdev_dbg(ndev, "Disabling HSR Offload mode\n");
        }
}

/* netdev notifier */
static int prueth_netdevice_event(struct notifier_block *unused,
                                  unsigned long event, void *ptr)
{
        struct netlink_ext_ack *extack = netdev_notifier_info_to_extack(ptr);
        struct net_device *ndev = netdev_notifier_info_to_dev(ptr);
        struct netdev_notifier_changeupper_info *info;
        struct prueth_emac *emac = netdev_priv(ndev);
        struct prueth *prueth = emac->prueth;
        enum hsr_version hsr_ndev_version;
        int ret = NOTIFY_DONE;

        if (ndev->netdev_ops != &emac_netdev_ops)
                return NOTIFY_DONE;

        switch (event) {
        case NETDEV_CHANGEUPPER:
                info = ptr;

                if ((ndev->features & NETIF_PRUETH_HSR_OFFLOAD_FEATURES) &&
                    is_hsr_master(info->upper_dev)) {
                        hsr_get_version(info->upper_dev, &hsr_ndev_version);
                        if (hsr_ndev_version != HSR_V1 && hsr_ndev_version != PRP_V1)
                                return -EOPNOTSUPP;
                        prueth->hsr_prp_version = hsr_ndev_version;

                        if (info->linking) {
                                if (!prueth->hsr_dev) {
                                        prueth->hsr_dev = info->upper_dev;
                                        icssg_class_set_host_mac_addr(prueth->miig_rt,
                                                                      prueth->hsr_dev->dev_addr);
                                } else {
                                        if (prueth->hsr_dev != info->upper_dev) {
                                                netdev_dbg(ndev, "Both interfaces must be linked to same upper device\n");
                                                return -EOPNOTSUPP;
                                        }
                                }
                                prueth_hsr_port_link(ndev);
                        } else {
                                prueth_hsr_port_unlink(ndev);
                        }
                }

                if (netif_is_bridge_master(info->upper_dev)) {
                        if (info->linking)
                                ret = prueth_netdevice_port_link(ndev, info->upper_dev, extack);
                        else
                                prueth_netdevice_port_unlink(ndev);
                }
                break;
        default:
                return NOTIFY_DONE;
        }

        return notifier_from_errno(ret);
}

static int prueth_register_notifiers(struct prueth *prueth)
{
        int ret = 0;

        prueth->prueth_netdevice_nb.notifier_call = &prueth_netdevice_event;
        ret = register_netdevice_notifier(&prueth->prueth_netdevice_nb);
        if (ret) {
                dev_err(prueth->dev, "can't register netdevice notifier\n");
                return ret;
        }

        ret = prueth_switchdev_register_notifiers(prueth);
        if (ret)
                unregister_netdevice_notifier(&prueth->prueth_netdevice_nb);

        return ret;
}

static void prueth_unregister_notifiers(struct prueth *prueth)
{
        prueth_switchdev_unregister_notifiers(prueth);
        unregister_netdevice_notifier(&prueth->prueth_netdevice_nb);
}

static void icssg_read_firmware_names(struct device_node *np,
                                      struct icssg_firmwares *fw)
{
        int i;

        for (i = 0; i < PRUETH_NUM_MACS; i++) {
                of_property_read_string_index(np, "firmware-name", i * 3 + 0,
                                              &fw[i].pru);
                of_property_read_string_index(np, "firmware-name", i * 3 + 1,
                                              &fw[i].rtu);
                of_property_read_string_index(np, "firmware-name", i * 3 + 2,
                                              &fw[i].txpru);
        }
}

/* icssg_firmware_name_replace - Replace a substring in firmware name
 * @dev: device pointer for memory allocation
 * @src: source firmware name string
 * @from: substring to replace
 * @to: replacement substring
 *
 * Return: a newly allocated string with the replacement, or the original
 * string if replacement is not possible.
 */
static const char *icssg_firmware_name_replace(struct device *dev,
                                               const char *src,
                                               const char *from,
                                               const char *to)
{
        size_t prefix, from_len, to_len, total;
        const char *p = strstr(src, from);
        char *buf;

        if (!p)
                return src; /* fallback: no replacement, use original */

        prefix = p - src;
        from_len = strlen(from);
        to_len = strlen(to);
        total = strlen(src) - from_len + to_len + 1;

        buf = devm_kzalloc(dev, total, GFP_KERNEL);
        if (!buf)
                return src; /* fallback: allocation failed, use original */

        strscpy(buf, src, prefix + 1);
        strscpy(buf + prefix, to, to_len + 1);
        strscpy(buf + prefix + to_len, p + from_len, total - prefix - to_len);

        return buf;
}

/**
 * icssg_mode_firmware_names - Generate firmware names for a specific mode
 * @dev: device pointer for logging and context
 * @src: source array of firmware name structures
 * @dst: destination array to store updated firmware name structures
 * @from: substring in firmware names to be replaced
 * @to: substring to replace @from in firmware names
 *
 * Iterates over all MACs and replaces occurrences of the @from substring
 * with @to in the firmware names (pru, rtu, txpru) for each MAC. The
 * updated firmware names are stored in the @dst array.
 */
static void icssg_mode_firmware_names(struct device *dev,
                                      struct icssg_firmwares *src,
                                      struct icssg_firmwares *dst,
                                      const char *from, const char *to)
{
        int i;

        for (i = 0; i < PRUETH_NUM_MACS; i++) {
                dst[i].pru = icssg_firmware_name_replace(dev, src[i].pru,
                                                         from, to);
                dst[i].rtu = icssg_firmware_name_replace(dev, src[i].rtu,
                                                         from, to);
                dst[i].txpru = icssg_firmware_name_replace(dev, src[i].txpru,
                                                           from, to);
        }
}

static int prueth_probe(struct platform_device *pdev)
{
        struct device_node *eth_node, *eth_ports_node;
        struct device_node  *eth0_node = NULL;
        struct device_node  *eth1_node = NULL;
        struct genpool_data_align gp_data = {
                .align = SZ_64K,
        };
        struct device *dev = &pdev->dev;
        struct device_node *np;
        struct prueth *prueth;
        struct pruss *pruss;
        u32 msmc_ram_size;
        int i, ret;

        np = dev->of_node;

        BUILD_BUG_ON_MSG((sizeof(struct prueth_swdata) > PRUETH_NAV_SW_DATA_SIZE),
                         "insufficient SW_DATA size");

        prueth = devm_kzalloc(dev, sizeof(*prueth), GFP_KERNEL);
        if (!prueth)
                return -ENOMEM;

        dev_set_drvdata(dev, prueth);
        prueth->pdev = pdev;
        prueth->pdata = *(const struct prueth_pdata *)device_get_match_data(dev);

        prueth->dev = dev;
        eth_ports_node = of_get_child_by_name(np, "ethernet-ports");
        if (!eth_ports_node)
                return -ENOENT;

        for_each_child_of_node(eth_ports_node, eth_node) {
                u32 reg;

                if (strcmp(eth_node->name, "port"))
                        continue;
                ret = of_property_read_u32(eth_node, "reg", &reg);
                if (ret < 0) {
                        dev_err(dev, "%pOF error reading port_id %d\n",
                                eth_node, ret);
                }

                of_node_get(eth_node);

                if (reg == 0) {
                        eth0_node = eth_node;
                        if (!of_device_is_available(eth0_node)) {
                                of_node_put(eth0_node);
                                eth0_node = NULL;
                        }
                } else if (reg == 1) {
                        eth1_node = eth_node;
                        if (!of_device_is_available(eth1_node)) {
                                of_node_put(eth1_node);
                                eth1_node = NULL;
                        }
                } else {
                        dev_err(dev, "port reg should be 0 or 1\n");
                }
        }

        of_node_put(eth_ports_node);

        /* At least one node must be present and available else we fail */
        if (!eth0_node && !eth1_node) {
                dev_err(dev, "neither port0 nor port1 node available\n");
                return -ENODEV;
        }

        if (eth0_node == eth1_node) {
                dev_err(dev, "port0 and port1 can't have same reg\n");
                of_node_put(eth0_node);
                return -ENODEV;
        }

        prueth->eth_node[PRUETH_MAC0] = eth0_node;
        prueth->eth_node[PRUETH_MAC1] = eth1_node;

        prueth->miig_rt = syscon_regmap_lookup_by_phandle(np, "ti,mii-g-rt");
        if (IS_ERR(prueth->miig_rt)) {
                dev_err(dev, "couldn't get ti,mii-g-rt syscon regmap\n");
                return -ENODEV;
        }

        prueth->mii_rt = syscon_regmap_lookup_by_phandle(np, "ti,mii-rt");
        if (IS_ERR(prueth->mii_rt)) {
                dev_err(dev, "couldn't get ti,mii-rt syscon regmap\n");
                return -ENODEV;
        }

        prueth->pa_stats = syscon_regmap_lookup_by_phandle(np, "ti,pa-stats");
        if (IS_ERR(prueth->pa_stats)) {
                dev_err(dev, "couldn't get ti,pa-stats syscon regmap\n");
                prueth->pa_stats = NULL;
        }

        if (eth0_node || eth1_node) {
                ret = prueth_get_cores(prueth, ICSS_SLICE0, false);
                if (ret)
                        goto put_cores;
                ret = prueth_get_cores(prueth, ICSS_SLICE1, false);
                if (ret)
                        goto put_cores;
        }

        pruss = pruss_get(eth0_node ?
                          prueth->pru[ICSS_SLICE0] : prueth->pru[ICSS_SLICE1]);
        if (IS_ERR(pruss)) {
                ret = PTR_ERR(pruss);
                dev_err(dev, "unable to get pruss handle\n");
                goto put_cores;
        }

        prueth->pruss = pruss;

        ret = pruss_request_mem_region(pruss, PRUSS_MEM_SHRD_RAM2,
                                       &prueth->shram);
        if (ret) {
                dev_err(dev, "unable to get PRUSS SHRD RAM2: %d\n", ret);
                goto put_pruss;
        }

        prueth->sram_pool = of_gen_pool_get(np, "sram", 0);
        if (!prueth->sram_pool) {
                dev_err(dev, "unable to get SRAM pool\n");
                ret = -ENODEV;

                goto put_mem;
        }

        prueth->is_switchmode_supported = prueth->pdata.switch_mode;
        if (prueth->pdata.banked_ms_ram) {
                /* Reserve 2 MSMC RAM banks for buffers to avoid arbitration */
                msmc_ram_size = (2 * MSMC_RAM_BANK_SIZE);
        } else {
                msmc_ram_size = PRUETH_EMAC_TOTAL_BUF_SIZE;
                if (prueth->is_switchmode_supported)
                        msmc_ram_size = PRUETH_SW_TOTAL_BUF_SIZE;
        }

        /* NOTE: FW bug needs buffer base to be 64KB aligned */
        prueth->msmcram.va =
                (void __iomem *)gen_pool_alloc_algo(prueth->sram_pool,
                                                    msmc_ram_size,
                                                    gen_pool_first_fit_align,
                                                    &gp_data);

        if (!prueth->msmcram.va) {
                ret = -ENOMEM;
                dev_err(dev, "unable to allocate MSMC resource\n");
                goto put_mem;
        }
        prueth->msmcram.pa = gen_pool_virt_to_phys(prueth->sram_pool,
                                                   (unsigned long)prueth->msmcram.va);
        prueth->msmcram.size = msmc_ram_size;
        memset_io(prueth->msmcram.va, 0, msmc_ram_size);
        dev_dbg(dev, "sram: pa %llx va %p size %zx\n", prueth->msmcram.pa,
                prueth->msmcram.va, prueth->msmcram.size);

        prueth->iep0 = icss_iep_get_idx(np, 0);
        if (IS_ERR(prueth->iep0)) {
                ret = dev_err_probe(dev, PTR_ERR(prueth->iep0), "iep0 get failed\n");
                prueth->iep0 = NULL;
                goto free_pool;
        }

        prueth->iep1 = icss_iep_get_idx(np, 1);
        if (IS_ERR(prueth->iep1)) {
                ret = dev_err_probe(dev, PTR_ERR(prueth->iep1), "iep1 get failed\n");
                goto put_iep0;
        }

        if (prueth->pdata.quirk_10m_link_issue) {
                /* Enable IEP1 for FW in 64bit mode as W/A for 10M FD link detect issue under TX
                 * traffic.
                 */
                icss_iep_init_fw(prueth->iep1);
        }

        /* Read EMAC firmware names from device tree */
        icssg_read_firmware_names(np, prueth->icssg_emac_firmwares);

        /* Generate other mode firmware names based on EMAC firmware names */
        icssg_mode_firmware_names(dev, prueth->icssg_emac_firmwares,
                                  prueth->icssg_switch_firmwares, "eth", "sw");
        icssg_mode_firmware_names(dev, prueth->icssg_emac_firmwares,
                                  prueth->icssg_hsr_firmwares, "eth", "hsr");
        icssg_mode_firmware_names(dev, prueth->icssg_emac_firmwares,
                                  prueth->icssg_prp_firmwares, "eth", "prp");

        spin_lock_init(&prueth->vtbl_lock);
        spin_lock_init(&prueth->stats_lock);
        /* setup netdev interfaces */
        if (eth0_node) {
                ret = prueth_netdev_init(prueth, eth0_node);
                if (ret) {
                        dev_err_probe(dev, ret, "netdev init %s failed\n",
                                      eth0_node->name);
                        goto exit_iep;
                }

                prueth->emac[PRUETH_MAC0]->half_duplex =
                        of_property_read_bool(eth0_node, "ti,half-duplex-capable");

                prueth->emac[PRUETH_MAC0]->iep = prueth->iep0;
        }

        if (eth1_node) {
                ret = prueth_netdev_init(prueth, eth1_node);
                if (ret) {
                        dev_err_probe(dev, ret, "netdev init %s failed\n",
                                      eth1_node->name);
                        goto netdev_exit;
                }

                prueth->emac[PRUETH_MAC1]->half_duplex =
                        of_property_read_bool(eth1_node, "ti,half-duplex-capable");

                prueth->emac[PRUETH_MAC1]->iep = prueth->iep0;
        }

        /* register the network devices */
        if (eth0_node) {
                ret = register_netdev(prueth->emac[PRUETH_MAC0]->ndev);
                if (ret) {
                        dev_err(dev, "can't register netdev for port MII0");
                        goto netdev_exit;
                }

                prueth->registered_netdevs[PRUETH_MAC0] = prueth->emac[PRUETH_MAC0]->ndev;

                ret = emac_phy_connect(prueth->emac[PRUETH_MAC0]);
                if (ret) {
                        dev_err(dev,
                                "can't connect to MII0 PHY, error -%d", ret);
                        goto netdev_unregister;
                }
                phy_attached_info(prueth->emac[PRUETH_MAC0]->ndev->phydev);
        }

        if (eth1_node) {
                ret = register_netdev(prueth->emac[PRUETH_MAC1]->ndev);
                if (ret) {
                        dev_err(dev, "can't register netdev for port MII1");
                        goto netdev_unregister;
                }

                prueth->registered_netdevs[PRUETH_MAC1] = prueth->emac[PRUETH_MAC1]->ndev;
                ret = emac_phy_connect(prueth->emac[PRUETH_MAC1]);
                if (ret) {
                        dev_err(dev,
                                "can't connect to MII1 PHY, error %d", ret);
                        goto netdev_unregister;
                }
                phy_attached_info(prueth->emac[PRUETH_MAC1]->ndev->phydev);
        }

        if (prueth->is_switchmode_supported) {
                ret = prueth_register_notifiers(prueth);
                if (ret)
                        goto netdev_unregister;

                sprintf(prueth->switch_id, "%s", dev_name(dev));
        }

        dev_info(dev, "TI PRU ethernet driver initialized: %s EMAC mode\n",
                 (!eth0_node || !eth1_node) ? "single" : "dual");

        if (eth1_node)
                of_node_put(eth1_node);
        if (eth0_node)
                of_node_put(eth0_node);
        return 0;

netdev_unregister:
        for (i = 0; i < PRUETH_NUM_MACS; i++) {
                if (!prueth->registered_netdevs[i])
                        continue;
                if (prueth->emac[i]->ndev->phydev) {
                        phy_disconnect(prueth->emac[i]->ndev->phydev);
                        prueth->emac[i]->ndev->phydev = NULL;
                }
                unregister_netdev(prueth->registered_netdevs[i]);
                disable_work_sync(&prueth->emac[i]->rx_mode_work);
        }

netdev_exit:
        for (i = 0; i < PRUETH_NUM_MACS; i++) {
                eth_node = prueth->eth_node[i];
                if (!eth_node)
                        continue;

                prueth_netdev_exit(prueth, eth_node);
        }

exit_iep:
        if (prueth->pdata.quirk_10m_link_issue)
                icss_iep_exit_fw(prueth->iep1);
        icss_iep_put(prueth->iep1);

put_iep0:
        icss_iep_put(prueth->iep0);
        prueth->iep0 = NULL;
        prueth->iep1 = NULL;

free_pool:
        gen_pool_free(prueth->sram_pool,
                      (unsigned long)prueth->msmcram.va,
                      prueth->msmcram.size);

put_mem:
        pruss_release_mem_region(prueth->pruss, &prueth->shram);

put_pruss:
        pruss_put(prueth->pruss);

put_cores:
        if (eth0_node || eth1_node) {
                prueth_put_cores(prueth, ICSS_SLICE0);
                of_node_put(eth0_node);

                prueth_put_cores(prueth, ICSS_SLICE1);
                of_node_put(eth1_node);
        }

        return ret;
}

static void prueth_remove(struct platform_device *pdev)
{
        struct prueth *prueth = platform_get_drvdata(pdev);
        struct device_node *eth_node;
        int i;

        prueth_unregister_notifiers(prueth);

        for (i = 0; i < PRUETH_NUM_MACS; i++) {
                if (!prueth->registered_netdevs[i])
                        continue;
                phy_stop(prueth->emac[i]->ndev->phydev);
                phy_disconnect(prueth->emac[i]->ndev->phydev);
                prueth->emac[i]->ndev->phydev = NULL;
                unregister_netdev(prueth->registered_netdevs[i]);
                disable_work_sync(&prueth->emac[i]->rx_mode_work);
        }

        for (i = 0; i < PRUETH_NUM_MACS; i++) {
                eth_node = prueth->eth_node[i];
                if (!eth_node)
                        continue;

                prueth_netdev_exit(prueth, eth_node);
        }

        if (prueth->pdata.quirk_10m_link_issue)
                icss_iep_exit_fw(prueth->iep1);

        icss_iep_put(prueth->iep1);
        icss_iep_put(prueth->iep0);

        gen_pool_free(prueth->sram_pool,
                (unsigned long)prueth->msmcram.va,
                prueth->msmcram.size);

        pruss_release_mem_region(prueth->pruss, &prueth->shram);

        pruss_put(prueth->pruss);

        if (prueth->eth_node[PRUETH_MAC1])
                prueth_put_cores(prueth, ICSS_SLICE1);

        if (prueth->eth_node[PRUETH_MAC0])
                prueth_put_cores(prueth, ICSS_SLICE0);
}

static const struct prueth_pdata am654_icssg_pdata = {
        .fdqring_mode = K3_RINGACC_RING_MODE_MESSAGE,
        .quirk_10m_link_issue = 1,
        .switch_mode = 1,
        .banked_ms_ram = 0,
};

static const struct prueth_pdata am64x_icssg_pdata = {
        .fdqring_mode = K3_RINGACC_RING_MODE_RING,
        .quirk_10m_link_issue = 1,
        .switch_mode = 1,
        .banked_ms_ram = 1,
};

static const struct of_device_id prueth_dt_match[] = {
        { .compatible = "ti,am654-icssg-prueth", .data = &am654_icssg_pdata },
        { .compatible = "ti,am642-icssg-prueth", .data = &am64x_icssg_pdata },
        { /* sentinel */ }
};
MODULE_DEVICE_TABLE(of, prueth_dt_match);

static struct platform_driver prueth_driver = {
        .probe = prueth_probe,
        .remove = prueth_remove,
        .driver = {
                .name = "icssg-prueth",
                .of_match_table = prueth_dt_match,
                .pm = &prueth_dev_pm_ops,
        },
};
module_platform_driver(prueth_driver);

MODULE_AUTHOR("Roger Quadros <rogerq@ti.com>");
MODULE_AUTHOR("Md Danish Anwar <danishanwar@ti.com>");
MODULE_DESCRIPTION("PRUSS ICSSG Ethernet Driver");
MODULE_LICENSE("GPL");