1 files changed, 14 insertions, 3 deletions

diff --git a/drivers/net/dsa/mv88e6123.c b/drivers/net/dsa/mv88e6123.c
index ede4c6f0b31a..bcab3ef22448 100644
--- a/drivers/net/dsa/mv88e6123.c
+++ b/drivers/net/dsa/mv88e6123.c
@@ -30,10 +30,20 @@ static const struct mv88e6xxx_switch_id mv88e6123_table[] = {
 };
 
 static char *mv88e6123_probe(struct device *dsa_dev, struct device *host_dev,
-                             int sw_addr)
+                             int sw_addr, void **priv)
 {
-        return mv88e6xxx_lookup_name(host_dev, sw_addr, mv88e6123_table,
+        struct mv88e6xxx_priv_state *ps;
+        char *name;
+
+        name = mv88e6xxx_lookup_name(host_dev, sw_addr, mv88e6123_table,
                                      ARRAY_SIZE(mv88e6123_table));
+        if (name) {
+                ps = devm_kzalloc(dsa_dev, sizeof(*ps), GFP_KERNEL);
+                if (!ps)
+                        return NULL;
+                *priv = ps;
+        }
+        return name;
 }
 
 static int mv88e6123_setup_global(struct dsa_switch *ds)
@@ -74,6 +84,8 @@ static int mv88e6123_setup(struct dsa_switch *ds)
         struct mv88e6xxx_priv_state *ps = ds_to_priv(ds);
         int ret;
 
+        ps->ds = ds;
+
         ret = mv88e6xxx_setup_common(ds);
         if (ret < 0)
                 return ret;
@@ -103,7 +115,6 @@ static int mv88e6123_setup(struct dsa_switch *ds)
 
 struct dsa_switch_driver mv88e6123_switch_driver = {
         .tag_protocol           = DSA_TAG_PROTO_EDSA,
-        .priv_size              = sizeof(struct mv88e6xxx_priv_state),
         .probe                  = mv88e6123_probe,
         .setup                  = mv88e6123_setup,
         .set_addr               = mv88e6xxx_set_addr_indirect,