#include "ocelot.h"
 #include "ocelot_vcap.h"
 
+const struct devlink_ops ocelot_devlink_ops = {
+};
+
+int ocelot_port_devlink_init(struct ocelot *ocelot, int port,
+                            enum devlink_port_flavour flavour)
+{
+       struct devlink_port *dlp = &ocelot->devlink_ports[port];
+       int id_len = sizeof(ocelot->base_mac);
+       struct devlink *dl = ocelot->devlink;
+       struct devlink_port_attrs attrs = {};
+
+       memcpy(attrs.switch_id.id, &ocelot->base_mac, id_len);
+       attrs.switch_id.id_len = id_len;
+       attrs.phys.port_number = port;
+       attrs.flavour = flavour;
+
+       devlink_port_attrs_set(dlp, &attrs);
+
+       return devlink_port_register(dl, dlp, port);
+}
+
+void ocelot_port_devlink_teardown(struct ocelot *ocelot, int port)
+{
+       struct devlink_port *dlp = &ocelot->devlink_ports[port];
+
+       devlink_port_unregister(dlp);
+}
+
+static struct devlink_port *ocelot_get_devlink_port(struct net_device *dev)
+{
+       struct ocelot_port_private *priv = netdev_priv(dev);
+       struct ocelot *ocelot = priv->port.ocelot;
+       int port = priv->chip_port;
+
+       return &ocelot->devlink_ports[port];
+}
+
 int ocelot_setup_tc_cls_flower(struct ocelot_port_private *priv,
                               struct flow_cls_offload *f,
                               bool ingress)
        __dev_mc_sync(dev, ocelot_mc_sync, ocelot_mc_unsync);
 }
 
-static int ocelot_port_get_phys_port_name(struct net_device *dev,
-                                         char *buf, size_t len)
-{
-       struct ocelot_port_private *priv = netdev_priv(dev);
-       int port = priv->chip_port;
-       int ret;
-
-       ret = snprintf(buf, len, "p%d", port);
-       if (ret >= len)
-               return -EINVAL;
-
-       return 0;
-}
-
 static int ocelot_port_set_mac_address(struct net_device *dev, void *p)
 {
        struct ocelot_port_private *priv = netdev_priv(dev);
        return 0;
 }
 
-static int ocelot_get_port_parent_id(struct net_device *dev,
-                                    struct netdev_phys_item_id *ppid)
-{
-       struct ocelot_port_private *priv = netdev_priv(dev);
-       struct ocelot *ocelot = priv->port.ocelot;
-
-       ppid->id_len = sizeof(ocelot->base_mac);
-       memcpy(&ppid->id, &ocelot->base_mac, ppid->id_len);
-
-       return 0;
-}
-
 static int ocelot_ioctl(struct net_device *dev, struct ifreq *ifr, int cmd)
 {
        struct ocelot_port_private *priv = netdev_priv(dev);
        .ndo_stop                       = ocelot_port_stop,
        .ndo_start_xmit                 = ocelot_port_xmit,
        .ndo_set_rx_mode                = ocelot_set_rx_mode,
-       .ndo_get_phys_port_name         = ocelot_port_get_phys_port_name,
        .ndo_set_mac_address            = ocelot_port_set_mac_address,
        .ndo_get_stats64                = ocelot_get_stats64,
        .ndo_fdb_add                    = ocelot_port_fdb_add,
        .ndo_vlan_rx_add_vid            = ocelot_vlan_rx_add_vid,
        .ndo_vlan_rx_kill_vid           = ocelot_vlan_rx_kill_vid,
        .ndo_set_features               = ocelot_set_features,
-       .ndo_get_port_parent_id         = ocelot_get_port_parent_id,
        .ndo_setup_tc                   = ocelot_setup_tc,
        .ndo_do_ioctl                   = ocelot_ioctl,
+       .ndo_get_devlink_port           = ocelot_get_devlink_port,
 };
 
 struct net_device *ocelot_port_to_netdev(struct ocelot *ocelot, int port)
 
        .enable         = ocelot_ptp_enable,
 };
 
+static void mscc_ocelot_teardown_devlink_ports(struct ocelot *ocelot)
+{
+       int port;
+
+       for (port = 0; port < ocelot->num_phys_ports; port++)
+               ocelot_port_devlink_teardown(ocelot, port);
+}
+
 static void mscc_ocelot_release_ports(struct ocelot *ocelot)
 {
        int port;
 {
        struct ocelot *ocelot = platform_get_drvdata(pdev);
        struct device_node *portnp;
-       int err;
+       bool *registered_ports;
+       int port, err;
+       u32 reg;
 
        ocelot->ports = devm_kcalloc(ocelot->dev, ocelot->num_phys_ports,
                                     sizeof(struct ocelot_port *), GFP_KERNEL);
        if (!ocelot->ports)
                return -ENOMEM;
 
+       ocelot->devlink_ports = devm_kcalloc(ocelot->dev,
+                                            ocelot->num_phys_ports,
+                                            sizeof(*ocelot->devlink_ports),
+                                            GFP_KERNEL);
+       if (!ocelot->devlink_ports)
+               return -ENOMEM;
+
+       registered_ports = kcalloc(ocelot->num_phys_ports, sizeof(bool),
+                                  GFP_KERNEL);
+       if (!registered_ports)
+               return -ENOMEM;
+
        for_each_available_child_of_node(ports, portnp) {
                struct ocelot_port_private *priv;
                struct ocelot_port *ocelot_port;
                struct device_node *phy_node;
+               struct devlink_port *dlp;
                phy_interface_t phy_mode;
                struct phy_device *phy;
                struct regmap *target;
                struct resource *res;
                struct phy *serdes;
                char res_name[8];
-               u32 port;
 
-               if (of_property_read_u32(portnp, "reg", &port))
+               if (of_property_read_u32(portnp, "reg", ®))
                        continue;
 
+               port = reg;
+
                snprintf(res_name, sizeof(res_name), "port%d", port);
 
                res = platform_get_resource_byname(pdev, IORESOURCE_MEM,
                if (!phy)
                        continue;
 
+               err = ocelot_port_devlink_init(ocelot, port,
+                                              DEVLINK_PORT_FLAVOUR_PHYSICAL);
+               if (err) {
+                       of_node_put(portnp);
+                       goto out_teardown;
+               }
+
                err = ocelot_probe_port(ocelot, port, target, phy);
                if (err) {
                        of_node_put(portnp);
-                       return err;
+                       goto out_teardown;
                }
 
+               registered_ports[port] = true;
+
                ocelot_port = ocelot->ports[port];
                priv = container_of(ocelot_port, struct ocelot_port_private,
                                    port);
+               dlp = &ocelot->devlink_ports[port];
+               devlink_port_type_eth_set(dlp, priv->dev);
 
                of_get_phy_mode(portnp, &phy_mode);
 
                                "invalid phy mode for port%d, (Q)SGMII only\n",
                                port);
                        of_node_put(portnp);
-                       return -EINVAL;
+                       err = -EINVAL;
+                       goto out_teardown;
                }
 
                serdes = devm_of_phy_get(ocelot->dev, portnp, NULL);
                                        port);
 
                        of_node_put(portnp);
-                       return err;
+                       goto out_teardown;
                }
 
                priv->serdes = serdes;
        }
 
+       /* Initialize unused devlink ports at the end */
+       for (port = 0; port < ocelot->num_phys_ports; port++) {
+               if (registered_ports[port])
+                       continue;
+
+               err = ocelot_port_devlink_init(ocelot, port,
+                                              DEVLINK_PORT_FLAVOUR_UNUSED);
+               if (err) {
+                       while (port-- >= 0) {
+                               if (!registered_ports[port])
+                                       continue;
+                               ocelot_port_devlink_teardown(ocelot, port);
+                       }
+
+                       goto out_teardown;
+               }
+       }
+
+       kfree(registered_ports);
+
        return 0;
+
+out_teardown:
+       /* Unregister the network interfaces */
+       mscc_ocelot_release_ports(ocelot);
+       /* Tear down devlink ports for the registered network interfaces */
+       for (port = 0; port < ocelot->num_phys_ports; port++) {
+               if (!registered_ports[port])
+                       continue;
+
+               ocelot_port_devlink_teardown(ocelot, port);
+       }
+       kfree(registered_ports);
+       return err;
 }
 
 static int mscc_ocelot_probe(struct platform_device *pdev)
        struct device_node *np = pdev->dev.of_node;
        int err, irq_xtr, irq_ptp_rdy;
        struct device_node *ports;
+       struct devlink *devlink;
        struct ocelot *ocelot;
        struct regmap *hsio;
        unsigned int i;
        if (!np && !pdev->dev.platform_data)
                return -ENODEV;
 
-       ocelot = devm_kzalloc(&pdev->dev, sizeof(*ocelot), GFP_KERNEL);
-       if (!ocelot)
+       devlink = devlink_alloc(&ocelot_devlink_ops, sizeof(*ocelot));
+       if (!devlink)
                return -ENOMEM;
 
+       ocelot = devlink_priv(devlink);
+       ocelot->devlink = priv_to_devlink(ocelot);
        platform_set_drvdata(pdev, ocelot);
        ocelot->dev = &pdev->dev;
 
                                ocelot->targets[io_target[i].id] = NULL;
                                continue;
                        }
-                       return PTR_ERR(target);
+                       err = PTR_ERR(target);
+                       goto out_free_devlink;
                }
 
                ocelot->targets[io_target[i].id] = target;
        hsio = syscon_regmap_lookup_by_compatible("mscc,ocelot-hsio");
        if (IS_ERR(hsio)) {
                dev_err(&pdev->dev, "missing hsio syscon\n");
-               return PTR_ERR(hsio);
+               err = PTR_ERR(hsio);
+               goto out_free_devlink;
        }
 
        ocelot->targets[HSIO] = hsio;
 
        err = ocelot_chip_init(ocelot, &ocelot_ops);
        if (err)
-               return err;
+               goto out_free_devlink;
 
        irq_xtr = platform_get_irq_byname(pdev, "xtr");
        if (irq_xtr < 0)
-               return -ENODEV;
+               goto out_free_devlink;
 
        err = devm_request_threaded_irq(&pdev->dev, irq_xtr, NULL,
                                        ocelot_xtr_irq_handler, IRQF_ONESHOT,
                                        "frame extraction", ocelot);
        if (err)
-               return err;
+               goto out_free_devlink;
 
        irq_ptp_rdy = platform_get_irq_byname(pdev, "ptp_rdy");
        if (irq_ptp_rdy > 0 && ocelot->targets[PTP]) {
                                                IRQF_ONESHOT, "ptp ready",
                                                ocelot);
                if (err)
-                       return err;
+                       goto out_free_devlink;
 
                /* Both the PTP interrupt and the PTP bank are available */
                ocelot->ptp = 1;
        ports = of_get_child_by_name(np, "ethernet-ports");
        if (!ports) {
                dev_err(ocelot->dev, "no ethernet-ports child node found\n");
-               return -ENODEV;
+               err = -ENODEV;
+               goto out_free_devlink;
        }
 
        ocelot->num_phys_ports = of_get_child_count(ports);
        if (err)
                goto out_put_ports;
 
-       err = mscc_ocelot_init_ports(pdev, ports);
+       err = devlink_register(devlink, ocelot->dev);
        if (err)
                goto out_ocelot_deinit;
 
+       err = mscc_ocelot_init_ports(pdev, ports);
+       if (err)
+               goto out_ocelot_devlink_unregister;
+
        if (ocelot->ptp) {
                err = ocelot_init_timestamp(ocelot, &ocelot_ptp_clock_info);
                if (err) {
 
        return 0;
 
+out_ocelot_devlink_unregister:
+       devlink_unregister(devlink);
 out_ocelot_deinit:
        ocelot_deinit(ocelot);
 out_put_ports:
        of_node_put(ports);
+out_free_devlink:
+       devlink_free(devlink);
        return err;
 }
 
 
        ocelot_deinit_timestamp(ocelot);
        mscc_ocelot_release_ports(ocelot);
+       mscc_ocelot_teardown_devlink_ports(ocelot);
+       devlink_unregister(ocelot->devlink);
        ocelot_deinit(ocelot);
        unregister_switchdev_blocking_notifier(&ocelot_switchdev_blocking_nb);
        unregister_switchdev_notifier(&ocelot_switchdev_nb);
        unregister_netdevice_notifier(&ocelot_netdevice_nb);
+       devlink_free(ocelot->devlink);
 
        return 0;
 }