#include "nxp-c45-tja11xx.h"
 
-#define PHY_ID_MASK                    GENMASK(31, 4)
 /* Same id: TJA1103, TJA1104 */
 #define PHY_ID_TJA_1103                        0x001BB010
 /* Same id: TJA1120, TJA1121 */
        return macsec_ability;
 }
 
-static int tja1103_match_phy_device(struct phy_device *phydev,
-                                   const struct phy_driver *phydrv)
+static int tja11xx_no_macsec_match_phy_device(struct phy_device *phydev,
+                                             const struct phy_driver *phydrv)
 {
-       return phy_id_compare(phydev->phy_id, PHY_ID_TJA_1103, PHY_ID_MASK) &&
-              !nxp_c45_macsec_ability(phydev);
-}
+       if (!phy_id_compare(phydev->phy_id, phydrv->phy_id,
+                           phydrv->phy_id_mask))
+               return 0;
 
-static int tja1104_match_phy_device(struct phy_device *phydev,
-                                   const struct phy_driver *phydrv)
-{
-       return phy_id_compare(phydev->phy_id, PHY_ID_TJA_1103, PHY_ID_MASK) &&
-              nxp_c45_macsec_ability(phydev);
+       return !nxp_c45_macsec_ability(phydev);
 }
 
-static int tja1120_match_phy_device(struct phy_device *phydev,
-                                   const struct phy_driver *phydrv)
+static int tja11xx_macsec_match_phy_device(struct phy_device *phydev,
+                                          const struct phy_driver *phydrv)
 {
-       return phy_id_compare(phydev->phy_id, PHY_ID_TJA_1120, PHY_ID_MASK) &&
-              !nxp_c45_macsec_ability(phydev);
-}
+       if (!phy_id_compare(phydev->phy_id, phydrv->phy_id,
+                           phydrv->phy_id_mask))
+               return 0;
 
-static int tja1121_match_phy_device(struct phy_device *phydev,
-                                   const struct phy_driver *phydrv)
-{
-       return phy_id_compare(phydev->phy_id, PHY_ID_TJA_1120, PHY_ID_MASK) &&
-              nxp_c45_macsec_ability(phydev);
+       return nxp_c45_macsec_ability(phydev);
 }
 
 static const struct nxp_c45_regmap tja1120_regmap = {
 
 static struct phy_driver nxp_c45_driver[] = {
        {
+               PHY_ID_MATCH_MODEL(PHY_ID_TJA_1103),
                .name                   = "NXP C45 TJA1103",
                .get_features           = nxp_c45_get_features,
                .driver_data            = &tja1103_phy_data,
                .get_sqi                = nxp_c45_get_sqi,
                .get_sqi_max            = nxp_c45_get_sqi_max,
                .remove                 = nxp_c45_remove,
-               .match_phy_device       = tja1103_match_phy_device,
+               .match_phy_device       = tja11xx_no_macsec_match_phy_device,
        },
        {
+               PHY_ID_MATCH_MODEL(PHY_ID_TJA_1103),
                .name                   = "NXP C45 TJA1104",
                .get_features           = nxp_c45_get_features,
                .driver_data            = &tja1103_phy_data,
                .get_sqi                = nxp_c45_get_sqi,
                .get_sqi_max            = nxp_c45_get_sqi_max,
                .remove                 = nxp_c45_remove,
-               .match_phy_device       = tja1104_match_phy_device,
+               .match_phy_device       = tja11xx_macsec_match_phy_device,
        },
        {
+               PHY_ID_MATCH_MODEL(PHY_ID_TJA_1120),
                .name                   = "NXP C45 TJA1120",
                .get_features           = nxp_c45_get_features,
                .driver_data            = &tja1120_phy_data,
                .get_sqi                = nxp_c45_get_sqi,
                .get_sqi_max            = nxp_c45_get_sqi_max,
                .remove                 = nxp_c45_remove,
-               .match_phy_device       = tja1120_match_phy_device,
+               .match_phy_device       = tja11xx_no_macsec_match_phy_device,
        },
        {
+               PHY_ID_MATCH_MODEL(PHY_ID_TJA_1120),
                .name                   = "NXP C45 TJA1121",
                .get_features           = nxp_c45_get_features,
                .driver_data            = &tja1120_phy_data,
                .get_sqi                = nxp_c45_get_sqi,
                .get_sqi_max            = nxp_c45_get_sqi_max,
                .remove                 = nxp_c45_remove,
-               .match_phy_device       = tja1121_match_phy_device,
+               .match_phy_device       = tja11xx_macsec_match_phy_device,
        },
 };