dsa2.c 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831
  1. /*
  2. * net/dsa/dsa2.c - Hardware switch handling, binding version 2
  3. * Copyright (c) 2008-2009 Marvell Semiconductor
  4. * Copyright (c) 2013 Florian Fainelli <florian@openwrt.org>
  5. * Copyright (c) 2016 Andrew Lunn <andrew@lunn.ch>
  6. *
  7. * This program is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU General Public License as published by
  9. * the Free Software Foundation; either version 2 of the License, or
  10. * (at your option) any later version.
  11. */
  12. #include <linux/device.h>
  13. #include <linux/err.h>
  14. #include <linux/list.h>
  15. #include <linux/netdevice.h>
  16. #include <linux/slab.h>
  17. #include <linux/rtnetlink.h>
  18. #include <linux/of.h>
  19. #include <linux/of_net.h>
  20. #include "dsa_priv.h"
  21. static LIST_HEAD(dsa_tree_list);
  22. static DEFINE_MUTEX(dsa2_mutex);
  23. static const struct devlink_ops dsa_devlink_ops = {
  24. };
  25. static struct dsa_switch_tree *dsa_tree_find(int index)
  26. {
  27. struct dsa_switch_tree *dst;
  28. list_for_each_entry(dst, &dsa_tree_list, list)
  29. if (dst->index == index)
  30. return dst;
  31. return NULL;
  32. }
  33. static struct dsa_switch_tree *dsa_tree_alloc(int index)
  34. {
  35. struct dsa_switch_tree *dst;
  36. dst = kzalloc(sizeof(*dst), GFP_KERNEL);
  37. if (!dst)
  38. return NULL;
  39. dst->index = index;
  40. INIT_LIST_HEAD(&dst->list);
  41. list_add_tail(&dsa_tree_list, &dst->list);
  42. kref_init(&dst->refcount);
  43. return dst;
  44. }
  45. static void dsa_tree_free(struct dsa_switch_tree *dst)
  46. {
  47. list_del(&dst->list);
  48. kfree(dst);
  49. }
  50. static struct dsa_switch_tree *dsa_tree_get(struct dsa_switch_tree *dst)
  51. {
  52. if (dst)
  53. kref_get(&dst->refcount);
  54. return dst;
  55. }
  56. static struct dsa_switch_tree *dsa_tree_touch(int index)
  57. {
  58. struct dsa_switch_tree *dst;
  59. dst = dsa_tree_find(index);
  60. if (dst)
  61. return dsa_tree_get(dst);
  62. else
  63. return dsa_tree_alloc(index);
  64. }
  65. static void dsa_tree_release(struct kref *ref)
  66. {
  67. struct dsa_switch_tree *dst;
  68. dst = container_of(ref, struct dsa_switch_tree, refcount);
  69. dsa_tree_free(dst);
  70. }
  71. static void dsa_tree_put(struct dsa_switch_tree *dst)
  72. {
  73. if (dst)
  74. kref_put(&dst->refcount, dsa_tree_release);
  75. }
  76. static bool dsa_port_is_dsa(struct dsa_port *port)
  77. {
  78. return port->type == DSA_PORT_TYPE_DSA;
  79. }
  80. static bool dsa_port_is_cpu(struct dsa_port *port)
  81. {
  82. return port->type == DSA_PORT_TYPE_CPU;
  83. }
  84. static bool dsa_port_is_user(struct dsa_port *dp)
  85. {
  86. return dp->type == DSA_PORT_TYPE_USER;
  87. }
  88. static struct dsa_port *dsa_tree_find_port_by_node(struct dsa_switch_tree *dst,
  89. struct device_node *dn)
  90. {
  91. struct dsa_switch *ds;
  92. struct dsa_port *dp;
  93. int device, port;
  94. for (device = 0; device < DSA_MAX_SWITCHES; device++) {
  95. ds = dst->ds[device];
  96. if (!ds)
  97. continue;
  98. for (port = 0; port < ds->num_ports; port++) {
  99. dp = &ds->ports[port];
  100. if (dp->dn == dn)
  101. return dp;
  102. }
  103. }
  104. return NULL;
  105. }
  106. static bool dsa_port_setup_routing_table(struct dsa_port *dp)
  107. {
  108. struct dsa_switch *ds = dp->ds;
  109. struct dsa_switch_tree *dst = ds->dst;
  110. struct device_node *dn = dp->dn;
  111. struct of_phandle_iterator it;
  112. struct dsa_port *link_dp;
  113. int err;
  114. of_for_each_phandle(&it, err, dn, "link", NULL, 0) {
  115. link_dp = dsa_tree_find_port_by_node(dst, it.node);
  116. if (!link_dp) {
  117. of_node_put(it.node);
  118. return false;
  119. }
  120. ds->rtable[link_dp->ds->index] = dp->index;
  121. }
  122. return true;
  123. }
  124. static bool dsa_switch_setup_routing_table(struct dsa_switch *ds)
  125. {
  126. bool complete = true;
  127. struct dsa_port *dp;
  128. int i;
  129. for (i = 0; i < DSA_MAX_SWITCHES; i++)
  130. ds->rtable[i] = DSA_RTABLE_NONE;
  131. for (i = 0; i < ds->num_ports; i++) {
  132. dp = &ds->ports[i];
  133. if (dsa_port_is_dsa(dp)) {
  134. complete = dsa_port_setup_routing_table(dp);
  135. if (!complete)
  136. break;
  137. }
  138. }
  139. return complete;
  140. }
  141. static bool dsa_tree_setup_routing_table(struct dsa_switch_tree *dst)
  142. {
  143. struct dsa_switch *ds;
  144. bool complete = true;
  145. int device;
  146. for (device = 0; device < DSA_MAX_SWITCHES; device++) {
  147. ds = dst->ds[device];
  148. if (!ds)
  149. continue;
  150. complete = dsa_switch_setup_routing_table(ds);
  151. if (!complete)
  152. break;
  153. }
  154. return complete;
  155. }
  156. static struct dsa_port *dsa_tree_find_first_cpu(struct dsa_switch_tree *dst)
  157. {
  158. struct dsa_switch *ds;
  159. struct dsa_port *dp;
  160. int device, port;
  161. for (device = 0; device < DSA_MAX_SWITCHES; device++) {
  162. ds = dst->ds[device];
  163. if (!ds)
  164. continue;
  165. for (port = 0; port < ds->num_ports; port++) {
  166. dp = &ds->ports[port];
  167. if (dsa_port_is_cpu(dp))
  168. return dp;
  169. }
  170. }
  171. return NULL;
  172. }
  173. static int dsa_tree_setup_default_cpu(struct dsa_switch_tree *dst)
  174. {
  175. struct dsa_switch *ds;
  176. struct dsa_port *dp;
  177. int device, port;
  178. /* DSA currently only supports a single CPU port */
  179. dst->cpu_dp = dsa_tree_find_first_cpu(dst);
  180. if (!dst->cpu_dp) {
  181. pr_warn("Tree has no master device\n");
  182. return -EINVAL;
  183. }
  184. /* Assign the default CPU port to all ports of the fabric */
  185. for (device = 0; device < DSA_MAX_SWITCHES; device++) {
  186. ds = dst->ds[device];
  187. if (!ds)
  188. continue;
  189. for (port = 0; port < ds->num_ports; port++) {
  190. dp = &ds->ports[port];
  191. if (dsa_port_is_user(dp) || dsa_port_is_dsa(dp))
  192. dp->cpu_dp = dst->cpu_dp;
  193. }
  194. }
  195. return 0;
  196. }
  197. static void dsa_tree_teardown_default_cpu(struct dsa_switch_tree *dst)
  198. {
  199. /* DSA currently only supports a single CPU port */
  200. dst->cpu_dp = NULL;
  201. }
  202. static int dsa_port_setup(struct dsa_port *dp)
  203. {
  204. struct dsa_switch *ds = dp->ds;
  205. int err = 0;
  206. memset(&dp->devlink_port, 0, sizeof(dp->devlink_port));
  207. if (dp->type != DSA_PORT_TYPE_UNUSED)
  208. err = devlink_port_register(ds->devlink, &dp->devlink_port,
  209. dp->index);
  210. if (err)
  211. return err;
  212. switch (dp->type) {
  213. case DSA_PORT_TYPE_UNUSED:
  214. break;
  215. case DSA_PORT_TYPE_CPU:
  216. /* dp->index is used now as port_number. However
  217. * CPU ports should have separate numbering
  218. * independent from front panel port numbers.
  219. */
  220. devlink_port_attrs_set(&dp->devlink_port,
  221. DEVLINK_PORT_FLAVOUR_CPU,
  222. dp->index, false, 0);
  223. err = dsa_port_link_register_of(dp);
  224. if (err) {
  225. dev_err(ds->dev, "failed to setup link for port %d.%d\n",
  226. ds->index, dp->index);
  227. return err;
  228. }
  229. break;
  230. case DSA_PORT_TYPE_DSA:
  231. /* dp->index is used now as port_number. However
  232. * DSA ports should have separate numbering
  233. * independent from front panel port numbers.
  234. */
  235. devlink_port_attrs_set(&dp->devlink_port,
  236. DEVLINK_PORT_FLAVOUR_DSA,
  237. dp->index, false, 0);
  238. err = dsa_port_link_register_of(dp);
  239. if (err) {
  240. dev_err(ds->dev, "failed to setup link for port %d.%d\n",
  241. ds->index, dp->index);
  242. return err;
  243. }
  244. break;
  245. case DSA_PORT_TYPE_USER:
  246. devlink_port_attrs_set(&dp->devlink_port,
  247. DEVLINK_PORT_FLAVOUR_PHYSICAL,
  248. dp->index, false, 0);
  249. err = dsa_slave_create(dp);
  250. if (err)
  251. dev_err(ds->dev, "failed to create slave for port %d.%d\n",
  252. ds->index, dp->index);
  253. else
  254. devlink_port_type_eth_set(&dp->devlink_port, dp->slave);
  255. break;
  256. }
  257. return 0;
  258. }
  259. static void dsa_port_teardown(struct dsa_port *dp)
  260. {
  261. if (dp->type != DSA_PORT_TYPE_UNUSED)
  262. devlink_port_unregister(&dp->devlink_port);
  263. switch (dp->type) {
  264. case DSA_PORT_TYPE_UNUSED:
  265. break;
  266. case DSA_PORT_TYPE_CPU:
  267. case DSA_PORT_TYPE_DSA:
  268. dsa_port_link_unregister_of(dp);
  269. break;
  270. case DSA_PORT_TYPE_USER:
  271. if (dp->slave) {
  272. dsa_slave_destroy(dp->slave);
  273. dp->slave = NULL;
  274. }
  275. break;
  276. }
  277. }
  278. static int dsa_switch_setup(struct dsa_switch *ds)
  279. {
  280. int err;
  281. /* Initialize ds->phys_mii_mask before registering the slave MDIO bus
  282. * driver and before ops->setup() has run, since the switch drivers and
  283. * the slave MDIO bus driver rely on these values for probing PHY
  284. * devices or not
  285. */
  286. ds->phys_mii_mask |= dsa_user_ports(ds);
  287. /* Add the switch to devlink before calling setup, so that setup can
  288. * add dpipe tables
  289. */
  290. ds->devlink = devlink_alloc(&dsa_devlink_ops, 0);
  291. if (!ds->devlink)
  292. return -ENOMEM;
  293. err = devlink_register(ds->devlink, ds->dev);
  294. if (err)
  295. return err;
  296. err = ds->ops->setup(ds);
  297. if (err < 0)
  298. return err;
  299. err = dsa_switch_register_notifier(ds);
  300. if (err)
  301. return err;
  302. if (!ds->slave_mii_bus && ds->ops->phy_read) {
  303. ds->slave_mii_bus = devm_mdiobus_alloc(ds->dev);
  304. if (!ds->slave_mii_bus)
  305. return -ENOMEM;
  306. dsa_slave_mii_bus_init(ds);
  307. err = mdiobus_register(ds->slave_mii_bus);
  308. if (err < 0)
  309. return err;
  310. }
  311. return 0;
  312. }
  313. static void dsa_switch_teardown(struct dsa_switch *ds)
  314. {
  315. if (ds->slave_mii_bus && ds->ops->phy_read)
  316. mdiobus_unregister(ds->slave_mii_bus);
  317. dsa_switch_unregister_notifier(ds);
  318. if (ds->devlink) {
  319. devlink_unregister(ds->devlink);
  320. devlink_free(ds->devlink);
  321. ds->devlink = NULL;
  322. }
  323. }
  324. static int dsa_tree_setup_switches(struct dsa_switch_tree *dst)
  325. {
  326. struct dsa_switch *ds;
  327. struct dsa_port *dp;
  328. int device, port;
  329. int err;
  330. for (device = 0; device < DSA_MAX_SWITCHES; device++) {
  331. ds = dst->ds[device];
  332. if (!ds)
  333. continue;
  334. err = dsa_switch_setup(ds);
  335. if (err)
  336. return err;
  337. for (port = 0; port < ds->num_ports; port++) {
  338. dp = &ds->ports[port];
  339. err = dsa_port_setup(dp);
  340. if (err)
  341. return err;
  342. }
  343. }
  344. return 0;
  345. }
  346. static void dsa_tree_teardown_switches(struct dsa_switch_tree *dst)
  347. {
  348. struct dsa_switch *ds;
  349. struct dsa_port *dp;
  350. int device, port;
  351. for (device = 0; device < DSA_MAX_SWITCHES; device++) {
  352. ds = dst->ds[device];
  353. if (!ds)
  354. continue;
  355. for (port = 0; port < ds->num_ports; port++) {
  356. dp = &ds->ports[port];
  357. dsa_port_teardown(dp);
  358. }
  359. dsa_switch_teardown(ds);
  360. }
  361. }
  362. static int dsa_tree_setup_master(struct dsa_switch_tree *dst)
  363. {
  364. struct dsa_port *cpu_dp = dst->cpu_dp;
  365. struct net_device *master = cpu_dp->master;
  366. /* DSA currently supports a single pair of CPU port and master device */
  367. return dsa_master_setup(master, cpu_dp);
  368. }
  369. static void dsa_tree_teardown_master(struct dsa_switch_tree *dst)
  370. {
  371. struct dsa_port *cpu_dp = dst->cpu_dp;
  372. struct net_device *master = cpu_dp->master;
  373. return dsa_master_teardown(master);
  374. }
  375. static int dsa_tree_setup(struct dsa_switch_tree *dst)
  376. {
  377. bool complete;
  378. int err;
  379. if (dst->setup) {
  380. pr_err("DSA: tree %d already setup! Disjoint trees?\n",
  381. dst->index);
  382. return -EEXIST;
  383. }
  384. complete = dsa_tree_setup_routing_table(dst);
  385. if (!complete)
  386. return 0;
  387. err = dsa_tree_setup_default_cpu(dst);
  388. if (err)
  389. return err;
  390. err = dsa_tree_setup_switches(dst);
  391. if (err)
  392. return err;
  393. err = dsa_tree_setup_master(dst);
  394. if (err)
  395. return err;
  396. dst->setup = true;
  397. pr_info("DSA: tree %d setup\n", dst->index);
  398. return 0;
  399. }
  400. static void dsa_tree_teardown(struct dsa_switch_tree *dst)
  401. {
  402. if (!dst->setup)
  403. return;
  404. dsa_tree_teardown_master(dst);
  405. dsa_tree_teardown_switches(dst);
  406. dsa_tree_teardown_default_cpu(dst);
  407. pr_info("DSA: tree %d torn down\n", dst->index);
  408. dst->setup = false;
  409. }
  410. static void dsa_tree_remove_switch(struct dsa_switch_tree *dst,
  411. unsigned int index)
  412. {
  413. dsa_tree_teardown(dst);
  414. dst->ds[index] = NULL;
  415. dsa_tree_put(dst);
  416. }
  417. static int dsa_tree_add_switch(struct dsa_switch_tree *dst,
  418. struct dsa_switch *ds)
  419. {
  420. unsigned int index = ds->index;
  421. int err;
  422. if (dst->ds[index])
  423. return -EBUSY;
  424. dsa_tree_get(dst);
  425. dst->ds[index] = ds;
  426. err = dsa_tree_setup(dst);
  427. if (err)
  428. dsa_tree_remove_switch(dst, index);
  429. return err;
  430. }
  431. static int dsa_port_parse_user(struct dsa_port *dp, const char *name)
  432. {
  433. if (!name)
  434. name = "eth%d";
  435. dp->type = DSA_PORT_TYPE_USER;
  436. dp->name = name;
  437. return 0;
  438. }
  439. static int dsa_port_parse_dsa(struct dsa_port *dp)
  440. {
  441. dp->type = DSA_PORT_TYPE_DSA;
  442. return 0;
  443. }
  444. static int dsa_port_parse_cpu(struct dsa_port *dp, struct net_device *master)
  445. {
  446. struct dsa_switch *ds = dp->ds;
  447. struct dsa_switch_tree *dst = ds->dst;
  448. const struct dsa_device_ops *tag_ops;
  449. enum dsa_tag_protocol tag_protocol;
  450. tag_protocol = ds->ops->get_tag_protocol(ds, dp->index);
  451. tag_ops = dsa_resolve_tag_protocol(tag_protocol);
  452. if (IS_ERR(tag_ops)) {
  453. dev_warn(ds->dev, "No tagger for this switch\n");
  454. return PTR_ERR(tag_ops);
  455. }
  456. dp->type = DSA_PORT_TYPE_CPU;
  457. dp->rcv = tag_ops->rcv;
  458. dp->tag_ops = tag_ops;
  459. dp->master = master;
  460. dp->dst = dst;
  461. return 0;
  462. }
  463. static int dsa_port_parse_of(struct dsa_port *dp, struct device_node *dn)
  464. {
  465. struct device_node *ethernet = of_parse_phandle(dn, "ethernet", 0);
  466. const char *name = of_get_property(dn, "label", NULL);
  467. bool link = of_property_read_bool(dn, "link");
  468. dp->dn = dn;
  469. if (ethernet) {
  470. struct net_device *master;
  471. master = of_find_net_device_by_node(ethernet);
  472. if (!master)
  473. return -EPROBE_DEFER;
  474. return dsa_port_parse_cpu(dp, master);
  475. }
  476. if (link)
  477. return dsa_port_parse_dsa(dp);
  478. return dsa_port_parse_user(dp, name);
  479. }
  480. static int dsa_switch_parse_ports_of(struct dsa_switch *ds,
  481. struct device_node *dn)
  482. {
  483. struct device_node *ports, *port;
  484. struct dsa_port *dp;
  485. u32 reg;
  486. int err;
  487. ports = of_get_child_by_name(dn, "ports");
  488. if (!ports) {
  489. dev_err(ds->dev, "no ports child node found\n");
  490. return -EINVAL;
  491. }
  492. for_each_available_child_of_node(ports, port) {
  493. err = of_property_read_u32(port, "reg", &reg);
  494. if (err)
  495. return err;
  496. if (reg >= ds->num_ports)
  497. return -EINVAL;
  498. dp = &ds->ports[reg];
  499. err = dsa_port_parse_of(dp, port);
  500. if (err)
  501. return err;
  502. }
  503. return 0;
  504. }
  505. static int dsa_switch_parse_member_of(struct dsa_switch *ds,
  506. struct device_node *dn)
  507. {
  508. u32 m[2] = { 0, 0 };
  509. int sz;
  510. /* Don't error out if this optional property isn't found */
  511. sz = of_property_read_variable_u32_array(dn, "dsa,member", m, 2, 2);
  512. if (sz < 0 && sz != -EINVAL)
  513. return sz;
  514. ds->index = m[1];
  515. if (ds->index >= DSA_MAX_SWITCHES)
  516. return -EINVAL;
  517. ds->dst = dsa_tree_touch(m[0]);
  518. if (!ds->dst)
  519. return -ENOMEM;
  520. return 0;
  521. }
  522. static int dsa_switch_parse_of(struct dsa_switch *ds, struct device_node *dn)
  523. {
  524. int err;
  525. err = dsa_switch_parse_member_of(ds, dn);
  526. if (err)
  527. return err;
  528. return dsa_switch_parse_ports_of(ds, dn);
  529. }
  530. static int dsa_port_parse(struct dsa_port *dp, const char *name,
  531. struct device *dev)
  532. {
  533. if (!strcmp(name, "cpu")) {
  534. struct net_device *master;
  535. master = dsa_dev_to_net_device(dev);
  536. if (!master)
  537. return -EPROBE_DEFER;
  538. dev_put(master);
  539. return dsa_port_parse_cpu(dp, master);
  540. }
  541. if (!strcmp(name, "dsa"))
  542. return dsa_port_parse_dsa(dp);
  543. return dsa_port_parse_user(dp, name);
  544. }
  545. static int dsa_switch_parse_ports(struct dsa_switch *ds,
  546. struct dsa_chip_data *cd)
  547. {
  548. bool valid_name_found = false;
  549. struct dsa_port *dp;
  550. struct device *dev;
  551. const char *name;
  552. unsigned int i;
  553. int err;
  554. for (i = 0; i < DSA_MAX_PORTS; i++) {
  555. name = cd->port_names[i];
  556. dev = cd->netdev[i];
  557. dp = &ds->ports[i];
  558. if (!name)
  559. continue;
  560. err = dsa_port_parse(dp, name, dev);
  561. if (err)
  562. return err;
  563. valid_name_found = true;
  564. }
  565. if (!valid_name_found && i == DSA_MAX_PORTS)
  566. return -EINVAL;
  567. return 0;
  568. }
  569. static int dsa_switch_parse(struct dsa_switch *ds, struct dsa_chip_data *cd)
  570. {
  571. ds->cd = cd;
  572. /* We don't support interconnected switches nor multiple trees via
  573. * platform data, so this is the unique switch of the tree.
  574. */
  575. ds->index = 0;
  576. ds->dst = dsa_tree_touch(0);
  577. if (!ds->dst)
  578. return -ENOMEM;
  579. return dsa_switch_parse_ports(ds, cd);
  580. }
  581. static int dsa_switch_add(struct dsa_switch *ds)
  582. {
  583. struct dsa_switch_tree *dst = ds->dst;
  584. return dsa_tree_add_switch(dst, ds);
  585. }
  586. static int dsa_switch_probe(struct dsa_switch *ds)
  587. {
  588. struct dsa_chip_data *pdata = ds->dev->platform_data;
  589. struct device_node *np = ds->dev->of_node;
  590. int err;
  591. if (np)
  592. err = dsa_switch_parse_of(ds, np);
  593. else if (pdata)
  594. err = dsa_switch_parse(ds, pdata);
  595. else
  596. err = -ENODEV;
  597. if (err)
  598. return err;
  599. return dsa_switch_add(ds);
  600. }
  601. struct dsa_switch *dsa_switch_alloc(struct device *dev, size_t n)
  602. {
  603. size_t size = sizeof(struct dsa_switch) + n * sizeof(struct dsa_port);
  604. struct dsa_switch *ds;
  605. int i;
  606. ds = devm_kzalloc(dev, size, GFP_KERNEL);
  607. if (!ds)
  608. return NULL;
  609. /* We avoid allocating memory outside dsa_switch
  610. * if it is not needed.
  611. */
  612. if (n <= sizeof(ds->_bitmap) * 8) {
  613. ds->bitmap = &ds->_bitmap;
  614. } else {
  615. ds->bitmap = devm_kcalloc(dev,
  616. BITS_TO_LONGS(n),
  617. sizeof(unsigned long),
  618. GFP_KERNEL);
  619. if (unlikely(!ds->bitmap))
  620. return NULL;
  621. }
  622. ds->dev = dev;
  623. ds->num_ports = n;
  624. for (i = 0; i < ds->num_ports; ++i) {
  625. ds->ports[i].index = i;
  626. ds->ports[i].ds = ds;
  627. }
  628. return ds;
  629. }
  630. EXPORT_SYMBOL_GPL(dsa_switch_alloc);
  631. int dsa_register_switch(struct dsa_switch *ds)
  632. {
  633. int err;
  634. mutex_lock(&dsa2_mutex);
  635. err = dsa_switch_probe(ds);
  636. dsa_tree_put(ds->dst);
  637. mutex_unlock(&dsa2_mutex);
  638. return err;
  639. }
  640. EXPORT_SYMBOL_GPL(dsa_register_switch);
  641. static void dsa_switch_remove(struct dsa_switch *ds)
  642. {
  643. struct dsa_switch_tree *dst = ds->dst;
  644. unsigned int index = ds->index;
  645. dsa_tree_remove_switch(dst, index);
  646. }
  647. void dsa_unregister_switch(struct dsa_switch *ds)
  648. {
  649. mutex_lock(&dsa2_mutex);
  650. dsa_switch_remove(ds);
  651. mutex_unlock(&dsa2_mutex);
  652. }
  653. EXPORT_SYMBOL_GPL(dsa_unregister_switch);