return 0;
 }
 
+static int
+dsa_switch_vlan_prepare_bitmap(struct dsa_switch *ds,
+                              const struct switchdev_obj_port_vlan *vlan,
+                              const unsigned long *bitmap)
+{
+       int port, err;
+
+       if (!ds->ops->port_vlan_prepare || !ds->ops->port_vlan_add)
+               return -EOPNOTSUPP;
+
+       for_each_set_bit(port, bitmap, ds->num_ports) {
+               err = ds->ops->port_vlan_prepare(ds, port, vlan);
+               if (err)
+                       return err;
+       }
+
+       return 0;
+}
+
+static void
+dsa_switch_vlan_add_bitmap(struct dsa_switch *ds,
+                          const struct switchdev_obj_port_vlan *vlan,
+                          const unsigned long *bitmap)
+{
+       int port;
+
+       for_each_set_bit(port, bitmap, ds->num_ports)
+               ds->ops->port_vlan_add(ds, port, vlan);
+}
+
 static int dsa_switch_vlan_add(struct dsa_switch *ds,
                               struct dsa_notifier_vlan_info *info)
 {
        const struct switchdev_obj_port_vlan *vlan = info->vlan;
        struct switchdev_trans *trans = info->trans;
        DECLARE_BITMAP(members, ds->num_ports);
-       int port, err;
+       int port;
 
        /* Build a mask of VLAN members */
        bitmap_zero(members, ds->num_ports);
                if (dsa_is_cpu_port(ds, port) || dsa_is_dsa_port(ds, port))
                        set_bit(port, members);
 
-       if (switchdev_trans_ph_prepare(trans)) {
-               if (!ds->ops->port_vlan_prepare || !ds->ops->port_vlan_add)
-                       return -EOPNOTSUPP;
+       if (switchdev_trans_ph_prepare(trans))
+               return dsa_switch_vlan_prepare_bitmap(ds, vlan, members);
 
-               for_each_set_bit(port, members, ds->num_ports) {
-                       err = ds->ops->port_vlan_prepare(ds, port, vlan);
-                       if (err)
-                               return err;
-               }
-
-               return 0;
-       }
-
-       for_each_set_bit(port, members, ds->num_ports)
-               ds->ops->port_vlan_add(ds, port, vlan);
+       dsa_switch_vlan_add_bitmap(ds, vlan, members);
 
        return 0;
 }