return !!dp->bridge_dev;
 }
 
-static void dsa_port_set_stp_state(struct dsa_switch *ds, int port, u8 state)
+static void dsa_slave_set_state(struct net_device *dev, u8 state)
 {
-       struct dsa_port *dp = &ds->ports[port];
+       struct dsa_slave_priv *p = netdev_priv(dev);
+       struct dsa_port *dp = p->dp;
+       struct dsa_switch *ds = dp->ds;
+       int port = dp->index;
 
        if (ds->ops->port_stp_state_set)
                ds->ops->port_stp_state_set(ds, port, state);
                        goto clear_promisc;
        }
 
-       dsa_port_set_stp_state(ds, p->dp->index, stp_state);
+       dsa_slave_set_state(dev, stp_state);
 
        if (p->phy)
                phy_start(p->phy);
        if (ds->ops->port_disable)
                ds->ops->port_disable(ds, p->dp->index, p->phy);
 
-       dsa_port_set_stp_state(ds, p->dp->index, BR_STATE_DISABLED);
+       dsa_slave_set_state(dev, BR_STATE_DISABLED);
 
        return 0;
 }
        if (switchdev_trans_ph_prepare(trans))
                return ds->ops->port_stp_state_set ? 0 : -EOPNOTSUPP;
 
-       dsa_port_set_stp_state(ds, p->dp->index, attr->u.stp_state);
+       dsa_slave_set_state(dev, attr->u.stp_state);
 
        return 0;
 }
        /* Port left the bridge, put in BR_STATE_DISABLED by the bridge layer,
         * so allow it to be in BR_STATE_FORWARDING to be kept functional
         */
-       dsa_port_set_stp_state(ds, p->dp->index, BR_STATE_FORWARDING);
+       dsa_slave_set_state(dev, BR_STATE_FORWARDING);
 }
 
 static int dsa_slave_port_attr_get(struct net_device *dev,