// SPDX-License-Identifier: GPL-2.0-or-later

#include <net/switchdev.h>

#include "br_private_mrp.h"

static enum br_mrp_hw_support
br_mrp_switchdev_port_obj(struct net_bridge *br,
                          const struct switchdev_obj *obj, bool add)
{
        int err;

        if (add)
                err = switchdev_port_obj_add(br->dev, obj, NULL);
        else
                err = switchdev_port_obj_del(br->dev, obj);

        /* In case of success just return and notify the SW that doesn't need
         * to do anything
         */
        if (!err)
                return BR_MRP_HW;

        if (err != -EOPNOTSUPP)
                return BR_MRP_NONE;

        /* Continue with SW backup */
        return BR_MRP_SW;
}

int br_mrp_switchdev_add(struct net_bridge *br, struct br_mrp *mrp)
{
        struct switchdev_obj_mrp mrp_obj = {
                .obj.orig_dev = br->dev,
                .obj.id = SWITCHDEV_OBJ_ID_MRP,
                .p_port = rtnl_dereference(mrp->p_port)->dev,
                .s_port = rtnl_dereference(mrp->s_port)->dev,
                .ring_id = mrp->ring_id,
                .prio = mrp->prio,
        };

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return 0;

        return switchdev_port_obj_add(br->dev, &mrp_obj.obj, NULL);
}

int br_mrp_switchdev_del(struct net_bridge *br, struct br_mrp *mrp)
{
        struct switchdev_obj_mrp mrp_obj = {
                .obj.orig_dev = br->dev,
                .obj.id = SWITCHDEV_OBJ_ID_MRP,
                .p_port = NULL,
                .s_port = NULL,
                .ring_id = mrp->ring_id,
        };

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return 0;

        return switchdev_port_obj_del(br->dev, &mrp_obj.obj);
}

enum br_mrp_hw_support
br_mrp_switchdev_set_ring_role(struct net_bridge *br, struct br_mrp *mrp,
                               enum br_mrp_ring_role_type role)
{
        struct switchdev_obj_ring_role_mrp mrp_role = {
                .obj.orig_dev = br->dev,
                .obj.id = SWITCHDEV_OBJ_ID_RING_ROLE_MRP,
                .ring_role = role,
                .ring_id = mrp->ring_id,
                .sw_backup = false,
        };
        enum br_mrp_hw_support support;
        int err;

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return BR_MRP_SW;

        support = br_mrp_switchdev_port_obj(br, &mrp_role.obj,
                                            role != BR_MRP_RING_ROLE_DISABLED);
        if (support != BR_MRP_SW)
                return support;

        /* If the driver can't configure to run completely the protocol in HW,
         * then try again to configure the HW so the SW can run the protocol.
         */
        mrp_role.sw_backup = true;
        if (role != BR_MRP_RING_ROLE_DISABLED)
                err = switchdev_port_obj_add(br->dev, &mrp_role.obj, NULL);
        else
                err = switchdev_port_obj_del(br->dev, &mrp_role.obj);

        if (!err)
                return BR_MRP_SW;

        return BR_MRP_NONE;
}

enum br_mrp_hw_support
br_mrp_switchdev_send_ring_test(struct net_bridge *br, struct br_mrp *mrp,
                                u32 interval, u8 max_miss, u32 period,
                                bool monitor)
{
        struct switchdev_obj_ring_test_mrp test = {
                .obj.orig_dev = br->dev,
                .obj.id = SWITCHDEV_OBJ_ID_RING_TEST_MRP,
                .interval = interval,
                .max_miss = max_miss,
                .ring_id = mrp->ring_id,
                .period = period,
                .monitor = monitor,
        };

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return BR_MRP_SW;

        return br_mrp_switchdev_port_obj(br, &test.obj, interval != 0);
}

int br_mrp_switchdev_set_ring_state(struct net_bridge *br,
                                    struct br_mrp *mrp,
                                    enum br_mrp_ring_state_type state)
{
        struct switchdev_obj_ring_state_mrp mrp_state = {
                .obj.orig_dev = br->dev,
                .obj.id = SWITCHDEV_OBJ_ID_RING_STATE_MRP,
                .ring_state = state,
                .ring_id = mrp->ring_id,
        };

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return 0;

        return switchdev_port_obj_add(br->dev, &mrp_state.obj, NULL);
}

enum br_mrp_hw_support
br_mrp_switchdev_set_in_role(struct net_bridge *br, struct br_mrp *mrp,
                             u16 in_id, u32 ring_id,
                             enum br_mrp_in_role_type role)
{
        struct switchdev_obj_in_role_mrp mrp_role = {
                .obj.orig_dev = br->dev,
                .obj.id = SWITCHDEV_OBJ_ID_IN_ROLE_MRP,
                .in_role = role,
                .in_id = mrp->in_id,
                .ring_id = mrp->ring_id,
                .i_port = rtnl_dereference(mrp->i_port)->dev,
                .sw_backup = false,
        };
        enum br_mrp_hw_support support;
        int err;

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return BR_MRP_SW;

        support = br_mrp_switchdev_port_obj(br, &mrp_role.obj,
                                            role != BR_MRP_IN_ROLE_DISABLED);
        if (support != BR_MRP_NONE)
                return support;

        /* If the driver can't configure to run completely the protocol in HW,
         * then try again to configure the HW so the SW can run the protocol.
         */
        mrp_role.sw_backup = true;
        if (role != BR_MRP_IN_ROLE_DISABLED)
                err = switchdev_port_obj_add(br->dev, &mrp_role.obj, NULL);
        else
                err = switchdev_port_obj_del(br->dev, &mrp_role.obj);

        if (!err)
                return BR_MRP_SW;

        return BR_MRP_NONE;
}

int br_mrp_switchdev_set_in_state(struct net_bridge *br, struct br_mrp *mrp,
                                  enum br_mrp_in_state_type state)
{
        struct switchdev_obj_in_state_mrp mrp_state = {
                .obj.orig_dev = br->dev,
                .obj.id = SWITCHDEV_OBJ_ID_IN_STATE_MRP,
                .in_state = state,
                .in_id = mrp->in_id,
        };

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return 0;

        return switchdev_port_obj_add(br->dev, &mrp_state.obj, NULL);
}

enum br_mrp_hw_support
br_mrp_switchdev_send_in_test(struct net_bridge *br, struct br_mrp *mrp,
                              u32 interval, u8 max_miss, u32 period)
{
        struct switchdev_obj_in_test_mrp test = {
                .obj.orig_dev = br->dev,
                .obj.id = SWITCHDEV_OBJ_ID_IN_TEST_MRP,
                .interval = interval,
                .max_miss = max_miss,
                .in_id = mrp->in_id,
                .period = period,
        };

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return BR_MRP_SW;

        return br_mrp_switchdev_port_obj(br, &test.obj, interval != 0);
}

int br_mrp_port_switchdev_set_state(struct net_bridge_port *p, u32 state)
{
        struct switchdev_attr attr = {
                .orig_dev = p->dev,
                .id = SWITCHDEV_ATTR_ID_PORT_STP_STATE,
                .u.stp_state = state,
        };

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return 0;

        return switchdev_port_attr_set(p->dev, &attr, NULL);
}

int br_mrp_port_switchdev_set_role(struct net_bridge_port *p,
                                   enum br_mrp_port_role_type role)
{
        struct switchdev_attr attr = {
                .orig_dev = p->dev,
                .id = SWITCHDEV_ATTR_ID_MRP_PORT_ROLE,
                .u.mrp_port_role = role,
        };

        if (!IS_ENABLED(CONFIG_NET_SWITCHDEV))
                return 0;

        return switchdev_port_attr_set(p->dev, &attr, NULL);
}