return err;
 }
 
+static int mv88e6xxx_g2_trunk_mask_write(struct mv88e6xxx_chip *chip, int num,
+                                        bool hask, u16 mask)
+{
+       const u16 port_mask = BIT(chip->info->num_ports) - 1;
+       u16 val = (num << 12) | (mask & port_mask);
+
+       if (hask)
+               val |= GLOBAL2_TRUNK_MASK_HASK;
+
+       return mv88e6xxx_update(chip, REG_GLOBAL2, GLOBAL2_TRUNK_MASK, val);
+}
+
+static int mv88e6xxx_g2_trunk_mapping_write(struct mv88e6xxx_chip *chip, int id,
+                                           u16 map)
+{
+       const u16 port_mask = BIT(chip->info->num_ports) - 1;
+       u16 val = (id << 11) | (map & port_mask);
+
+       return mv88e6xxx_update(chip, REG_GLOBAL2, GLOBAL2_TRUNK_MAPPING, val);
+}
+
+static int mv88e6xxx_g2_clear_trunk(struct mv88e6xxx_chip *chip)
+{
+       const u16 port_mask = BIT(chip->info->num_ports) - 1;
+       int i, err;
+
+       /* Clear all eight possible Trunk Mask vectors */
+       for (i = 0; i < 8; ++i) {
+               err = mv88e6xxx_g2_trunk_mask_write(chip, i, false, port_mask);
+               if (err)
+                       return err;
+       }
+
+       /* Clear all sixteen possible Trunk ID routing vectors */
+       for (i = 0; i < 16; ++i) {
+               err = mv88e6xxx_g2_trunk_mapping_write(chip, i, 0);
+               if (err)
+                       return err;
+       }
+
+       return 0;
+}
+
 static int mv88e6xxx_g2_setup(struct mv88e6xxx_chip *chip)
 {
        int err;
        if (err)
                return err;
 
-       /* Clear all trunk masks. */
-       for (i = 0; i < 8; i++) {
-               err = _mv88e6xxx_reg_write(chip, REG_GLOBAL2,
-                                          GLOBAL2_TRUNK_MASK,
-                                          0x8000 |
-                                          (i << GLOBAL2_TRUNK_MASK_NUM_SHIFT) |
-                                          ((1 << chip->info->num_ports) - 1));
-               if (err)
-                       return err;
-       }
-
-       /* Clear all trunk mappings. */
-       for (i = 0; i < 16; i++) {
-               err = _mv88e6xxx_reg_write(
-                       chip, REG_GLOBAL2,
-                       GLOBAL2_TRUNK_MAPPING,
-                       GLOBAL2_TRUNK_MAPPING_UPDATE |
-                       (i << GLOBAL2_TRUNK_MAPPING_ID_SHIFT));
-               if (err)
-                       return err;
-       }
+       /* Clear all trunk masks and mapping. */
+       err = mv88e6xxx_g2_clear_trunk(chip);
+       if (err)
+               return err;
 
        if (mv88e6xxx_6352_family(chip) || mv88e6xxx_6351_family(chip) ||
            mv88e6xxx_6165_family(chip) || mv88e6xxx_6097_family(chip) ||