dsa2.c 16.4 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * net/dsa/dsa2.c - Hardware switch handling, binding version 2
 * Copyright (c) 2008-2009 Marvell Semiconductor
 * Copyright (c) 2013 Florian Fainelli <florian@openwrt.org>
 * Copyright (c) 2016 Andrew Lunn <andrew@lunn.ch>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 */

#include <linux/device.h>
#include <linux/err.h>
#include <linux/list.h>
16
#include <linux/netdevice.h>
17 18 19 20
#include <linux/slab.h>
#include <linux/rtnetlink.h>
#include <linux/of.h>
#include <linux/of_net.h>
21
#include <net/devlink.h>
22

23 24
#include "dsa_priv.h"

25
static LIST_HEAD(dsa_tree_list);
26 27
static DEFINE_MUTEX(dsa2_mutex);

28 29 30
static const struct devlink_ops dsa_devlink_ops = {
};

31
static struct dsa_switch_tree *dsa_tree_find(int index)
32 33 34
{
	struct dsa_switch_tree *dst;

35
	list_for_each_entry(dst, &dsa_tree_list, list)
36
		if (dst->index == index)
37
			return dst;
38

39 40 41
	return NULL;
}

42
static struct dsa_switch_tree *dsa_tree_alloc(int index)
43 44 45 46 47 48
{
	struct dsa_switch_tree *dst;

	dst = kzalloc(sizeof(*dst), GFP_KERNEL);
	if (!dst)
		return NULL;
49

50
	dst->index = index;
51

52
	INIT_LIST_HEAD(&dst->list);
53
	list_add_tail(&dsa_tree_list, &dst->list);
54

55 56 57 58 59
	kref_init(&dst->refcount);

	return dst;
}

60 61 62 63 64 65
static void dsa_tree_free(struct dsa_switch_tree *dst)
{
	list_del(&dst->list);
	kfree(dst);
}

66
static struct dsa_switch_tree *dsa_tree_get(struct dsa_switch_tree *dst)
67
{
68 69
	if (dst)
		kref_get(&dst->refcount);
70 71 72 73

	return dst;
}

74
static struct dsa_switch_tree *dsa_tree_touch(int index)
75
{
76 77 78 79 80 81 82
	struct dsa_switch_tree *dst;

	dst = dsa_tree_find(index);
	if (dst)
		return dsa_tree_get(dst);
	else
		return dsa_tree_alloc(index);
83 84 85 86 87 88 89 90 91 92 93 94 95
}

static void dsa_tree_release(struct kref *ref)
{
	struct dsa_switch_tree *dst;

	dst = container_of(ref, struct dsa_switch_tree, refcount);

	dsa_tree_free(dst);
}

static void dsa_tree_put(struct dsa_switch_tree *dst)
{
96 97
	if (dst)
		kref_put(&dst->refcount, dsa_tree_release);
98 99
}

100
static bool dsa_port_is_dsa(struct dsa_port *port)
101
{
102
	return port->type == DSA_PORT_TYPE_DSA;
103 104 105 106
}

static bool dsa_port_is_cpu(struct dsa_port *port)
{
107
	return port->type == DSA_PORT_TYPE_CPU;
108 109
}

110 111 112 113 114
static bool dsa_port_is_user(struct dsa_port *dp)
{
	return dp->type == DSA_PORT_TYPE_USER;
}

115 116
static struct dsa_port *dsa_tree_find_port_by_node(struct dsa_switch_tree *dst,
						   struct device_node *dn)
117 118
{
	struct dsa_switch *ds;
119 120
	struct dsa_port *dp;
	int device, port;
121

122 123
	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
124 125 126
		if (!ds)
			continue;

127 128 129 130 131 132
		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

			if (dp->dn == dn)
				return dp;
		}
133 134 135 136 137
	}

	return NULL;
}

V
Vivien Didelot 已提交
138
static bool dsa_port_setup_routing_table(struct dsa_port *dp)
139
{
V
Vivien Didelot 已提交
140 141 142
	struct dsa_switch *ds = dp->ds;
	struct dsa_switch_tree *dst = ds->dst;
	struct device_node *dn = dp->dn;
V
Vivien Didelot 已提交
143
	struct of_phandle_iterator it;
144
	struct dsa_port *link_dp;
V
Vivien Didelot 已提交
145
	int err;
146

V
Vivien Didelot 已提交
147 148 149 150
	of_for_each_phandle(&it, err, dn, "link", NULL, 0) {
		link_dp = dsa_tree_find_port_by_node(dst, it.node);
		if (!link_dp) {
			of_node_put(it.node);
V
Vivien Didelot 已提交
151
			return false;
V
Vivien Didelot 已提交
152
		}
153

V
Vivien Didelot 已提交
154
		ds->rtable[link_dp->ds->index] = dp->index;
155 156
	}

V
Vivien Didelot 已提交
157
	return true;
158 159
}

V
Vivien Didelot 已提交
160
static bool dsa_switch_setup_routing_table(struct dsa_switch *ds)
161
{
V
Vivien Didelot 已提交
162 163 164
	bool complete = true;
	struct dsa_port *dp;
	int i;
165

V
Vivien Didelot 已提交
166 167
	for (i = 0; i < DSA_MAX_SWITCHES; i++)
		ds->rtable[i] = DSA_RTABLE_NONE;
168

V
Vivien Didelot 已提交
169 170
	for (i = 0; i < ds->num_ports; i++) {
		dp = &ds->ports[i];
171

V
Vivien Didelot 已提交
172 173 174 175 176
		if (dsa_port_is_dsa(dp)) {
			complete = dsa_port_setup_routing_table(dp);
			if (!complete)
				break;
		}
177 178
	}

V
Vivien Didelot 已提交
179
	return complete;
180 181
}

V
Vivien Didelot 已提交
182
static bool dsa_tree_setup_routing_table(struct dsa_switch_tree *dst)
183 184
{
	struct dsa_switch *ds;
V
Vivien Didelot 已提交
185 186
	bool complete = true;
	int device;
187

V
Vivien Didelot 已提交
188 189
	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
190 191 192
		if (!ds)
			continue;

V
Vivien Didelot 已提交
193 194 195
		complete = dsa_switch_setup_routing_table(ds);
		if (!complete)
			break;
196 197
	}

V
Vivien Didelot 已提交
198
	return complete;
199 200
}

201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244
static struct dsa_port *dsa_tree_find_first_cpu(struct dsa_switch_tree *dst)
{
	struct dsa_switch *ds;
	struct dsa_port *dp;
	int device, port;

	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
		if (!ds)
			continue;

		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

			if (dsa_port_is_cpu(dp))
				return dp;
		}
	}

	return NULL;
}

static int dsa_tree_setup_default_cpu(struct dsa_switch_tree *dst)
{
	struct dsa_switch *ds;
	struct dsa_port *dp;
	int device, port;

	/* DSA currently only supports a single CPU port */
	dst->cpu_dp = dsa_tree_find_first_cpu(dst);
	if (!dst->cpu_dp) {
		pr_warn("Tree has no master device\n");
		return -EINVAL;
	}

	/* Assign the default CPU port to all ports of the fabric */
	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
		if (!ds)
			continue;

		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

245
			if (dsa_port_is_user(dp) || dsa_port_is_dsa(dp))
246 247 248 249 250 251 252 253 254 255 256 257 258
				dp->cpu_dp = dst->cpu_dp;
		}
	}

	return 0;
}

static void dsa_tree_teardown_default_cpu(struct dsa_switch_tree *dst)
{
	/* DSA currently only supports a single CPU port */
	dst->cpu_dp = NULL;
}

259
static int dsa_port_setup(struct dsa_port *dp)
260
{
261
	enum devlink_port_flavour flavour;
262
	struct dsa_switch *ds = dp->ds;
263 264 265 266
	int err;

	if (dp->type == DSA_PORT_TYPE_UNUSED)
		return 0;
267

268
	memset(&dp->devlink_port, 0, sizeof(dp->devlink_port));
269

270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290
	switch (dp->type) {
	case DSA_PORT_TYPE_CPU:
		flavour = DEVLINK_PORT_FLAVOUR_CPU;
		break;
	case DSA_PORT_TYPE_DSA:
		flavour = DEVLINK_PORT_FLAVOUR_DSA;
		break;
	case DSA_PORT_TYPE_USER: /* fall-through */
	default:
		flavour = DEVLINK_PORT_FLAVOUR_PHYSICAL;
		break;
	}

	/* dp->index is used now as port_number. However
	 * CPU and DSA ports should have separate numbering
	 * independent from front panel port numbers.
	 */
	devlink_port_attrs_set(&dp->devlink_port, flavour,
			       dp->index, false, 0);
	err = devlink_port_register(ds->devlink, &dp->devlink_port,
				    dp->index);
291
	if (err)
292 293
		return err;

294 295 296 297
	switch (dp->type) {
	case DSA_PORT_TYPE_UNUSED:
		break;
	case DSA_PORT_TYPE_CPU:
298 299 300 301 302 303 304
		err = dsa_port_link_register_of(dp);
		if (err) {
			dev_err(ds->dev, "failed to setup link for port %d.%d\n",
				ds->index, dp->index);
			return err;
		}
		break;
305
	case DSA_PORT_TYPE_DSA:
306
		err = dsa_port_link_register_of(dp);
307
		if (err) {
308
			dev_err(ds->dev, "failed to setup link for port %d.%d\n",
309 310 311 312 313 314 315 316 317 318 319 320
				ds->index, dp->index);
			return err;
		}
		break;
	case DSA_PORT_TYPE_USER:
		err = dsa_slave_create(dp);
		if (err)
			dev_err(ds->dev, "failed to create slave for port %d.%d\n",
				ds->index, dp->index);
		else
			devlink_port_type_eth_set(&dp->devlink_port, dp->slave);
		break;
321 322 323 324 325
	}

	return 0;
}

326
static void dsa_port_teardown(struct dsa_port *dp)
327
{
328 329
	if (dp->type != DSA_PORT_TYPE_UNUSED)
		devlink_port_unregister(&dp->devlink_port);
330 331 332 333 334 335

	switch (dp->type) {
	case DSA_PORT_TYPE_UNUSED:
		break;
	case DSA_PORT_TYPE_CPU:
	case DSA_PORT_TYPE_DSA:
336
		dsa_port_link_unregister_of(dp);
337 338 339 340 341 342 343
		break;
	case DSA_PORT_TYPE_USER:
		if (dp->slave) {
			dsa_slave_destroy(dp->slave);
			dp->slave = NULL;
		}
		break;
344 345 346
	}
}

347
static int dsa_switch_setup(struct dsa_switch *ds)
348 349 350
{
	int err;

351
	/* Initialize ds->phys_mii_mask before registering the slave MDIO bus
352
	 * driver and before ops->setup() has run, since the switch drivers and
353 354 355
	 * the slave MDIO bus driver rely on these values for probing PHY
	 * devices or not
	 */
356
	ds->phys_mii_mask |= dsa_user_ports(ds);
357

358 359 360 361 362 363 364 365 366 367 368
	/* Add the switch to devlink before calling setup, so that setup can
	 * add dpipe tables
	 */
	ds->devlink = devlink_alloc(&dsa_devlink_ops, 0);
	if (!ds->devlink)
		return -ENOMEM;

	err = devlink_register(ds->devlink, ds->dev);
	if (err)
		return err;

369
	err = ds->ops->setup(ds);
370 371 372
	if (err < 0)
		return err;

V
Vivien Didelot 已提交
373 374 375 376
	err = dsa_switch_register_notifier(ds);
	if (err)
		return err;

377
	if (!ds->slave_mii_bus && ds->ops->phy_read) {
378 379 380 381 382 383 384 385 386 387 388
		ds->slave_mii_bus = devm_mdiobus_alloc(ds->dev);
		if (!ds->slave_mii_bus)
			return -ENOMEM;

		dsa_slave_mii_bus_init(ds);

		err = mdiobus_register(ds->slave_mii_bus);
		if (err < 0)
			return err;
	}

389 390 391
	return 0;
}

392
static void dsa_switch_teardown(struct dsa_switch *ds)
393
{
394
	if (ds->slave_mii_bus && ds->ops->phy_read)
395
		mdiobus_unregister(ds->slave_mii_bus);
V
Vivien Didelot 已提交
396 397

	dsa_switch_unregister_notifier(ds);
398 399 400 401 402 403 404

	if (ds->devlink) {
		devlink_unregister(ds->devlink);
		devlink_free(ds->devlink);
		ds->devlink = NULL;
	}

405 406
}

407 408 409
static int dsa_tree_setup_switches(struct dsa_switch_tree *dst)
{
	struct dsa_switch *ds;
410 411
	struct dsa_port *dp;
	int device, port;
412 413 414 415 416 417 418 419 420 421
	int err;

	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
		if (!ds)
			continue;

		err = dsa_switch_setup(ds);
		if (err)
			return err;
422 423 424 425 426 427 428 429

		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

			err = dsa_port_setup(dp);
			if (err)
				return err;
		}
430 431 432 433 434 435 436 437
	}

	return 0;
}

static void dsa_tree_teardown_switches(struct dsa_switch_tree *dst)
{
	struct dsa_switch *ds;
438 439
	struct dsa_port *dp;
	int device, port;
440 441 442 443 444 445

	for (device = 0; device < DSA_MAX_SWITCHES; device++) {
		ds = dst->ds[device];
		if (!ds)
			continue;

446 447 448 449 450 451
		for (port = 0; port < ds->num_ports; port++) {
			dp = &ds->ports[port];

			dsa_port_teardown(dp);
		}

452 453 454 455
		dsa_switch_teardown(ds);
	}
}

456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472
static int dsa_tree_setup_master(struct dsa_switch_tree *dst)
{
	struct dsa_port *cpu_dp = dst->cpu_dp;
	struct net_device *master = cpu_dp->master;

	/* DSA currently supports a single pair of CPU port and master device */
	return dsa_master_setup(master, cpu_dp);
}

static void dsa_tree_teardown_master(struct dsa_switch_tree *dst)
{
	struct dsa_port *cpu_dp = dst->cpu_dp;
	struct net_device *master = cpu_dp->master;

	return dsa_master_teardown(master);
}

V
Vivien Didelot 已提交
473
static int dsa_tree_setup(struct dsa_switch_tree *dst)
474
{
V
Vivien Didelot 已提交
475
	bool complete;
476 477
	int err;

V
Vivien Didelot 已提交
478 479 480 481 482 483
	if (dst->setup) {
		pr_err("DSA: tree %d already setup! Disjoint trees?\n",
		       dst->index);
		return -EEXIST;
	}

V
Vivien Didelot 已提交
484 485 486 487
	complete = dsa_tree_setup_routing_table(dst);
	if (!complete)
		return 0;

488 489 490 491
	err = dsa_tree_setup_default_cpu(dst);
	if (err)
		return err;

492 493 494
	err = dsa_tree_setup_switches(dst);
	if (err)
		return err;
495

496
	err = dsa_tree_setup_master(dst);
497 498 499
	if (err)
		return err;

V
Vivien Didelot 已提交
500 501 502
	dst->setup = true;

	pr_info("DSA: tree %d setup\n", dst->index);
503 504 505 506

	return 0;
}

V
Vivien Didelot 已提交
507
static void dsa_tree_teardown(struct dsa_switch_tree *dst)
508
{
V
Vivien Didelot 已提交
509
	if (!dst->setup)
510 511
		return;

512
	dsa_tree_teardown_master(dst);
513

514
	dsa_tree_teardown_switches(dst);
515

516
	dsa_tree_teardown_default_cpu(dst);
517

V
Vivien Didelot 已提交
518 519 520
	pr_info("DSA: tree %d torn down\n", dst->index);

	dst->setup = false;
521 522
}

523 524 525
static void dsa_tree_remove_switch(struct dsa_switch_tree *dst,
				   unsigned int index)
{
526 527
	dsa_tree_teardown(dst);

528 529 530 531 532 533 534 535
	dst->ds[index] = NULL;
	dsa_tree_put(dst);
}

static int dsa_tree_add_switch(struct dsa_switch_tree *dst,
			       struct dsa_switch *ds)
{
	unsigned int index = ds->index;
536
	int err;
537 538 539 540 541 542 543

	if (dst->ds[index])
		return -EBUSY;

	dsa_tree_get(dst);
	dst->ds[index] = ds;

544 545 546 547 548
	err = dsa_tree_setup(dst);
	if (err)
		dsa_tree_remove_switch(dst, index);

	return err;
549 550
}

551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570
static int dsa_port_parse_user(struct dsa_port *dp, const char *name)
{
	if (!name)
		name = "eth%d";

	dp->type = DSA_PORT_TYPE_USER;
	dp->name = name;

	return 0;
}

static int dsa_port_parse_dsa(struct dsa_port *dp)
{
	dp->type = DSA_PORT_TYPE_DSA;

	return 0;
}

static int dsa_port_parse_cpu(struct dsa_port *dp, struct net_device *master)
{
571 572 573 574 575
	struct dsa_switch *ds = dp->ds;
	struct dsa_switch_tree *dst = ds->dst;
	const struct dsa_device_ops *tag_ops;
	enum dsa_tag_protocol tag_protocol;

576
	tag_protocol = ds->ops->get_tag_protocol(ds, dp->index);
577 578 579 580 581 582
	tag_ops = dsa_resolve_tag_protocol(tag_protocol);
	if (IS_ERR(tag_ops)) {
		dev_warn(ds->dev, "No tagger for this switch\n");
		return PTR_ERR(tag_ops);
	}

583
	dp->type = DSA_PORT_TYPE_CPU;
584 585
	dp->rcv = tag_ops->rcv;
	dp->tag_ops = tag_ops;
586
	dp->master = master;
587
	dp->dst = dst;
588 589 590 591

	return 0;
}

592 593
static int dsa_port_parse_of(struct dsa_port *dp, struct device_node *dn)
{
594
	struct device_node *ethernet = of_parse_phandle(dn, "ethernet", 0);
595
	const char *name = of_get_property(dn, "label", NULL);
596
	bool link = of_property_read_bool(dn, "link");
597

598 599
	dp->dn = dn;

600
	if (ethernet) {
601 602 603 604 605 606
		struct net_device *master;

		master = of_find_net_device_by_node(ethernet);
		if (!master)
			return -EPROBE_DEFER;

607
		return dsa_port_parse_cpu(dp, master);
608 609
	}

610 611
	if (link)
		return dsa_port_parse_dsa(dp);
612

613
	return dsa_port_parse_user(dp, name);
614 615
}

V
Vivien Didelot 已提交
616 617
static int dsa_switch_parse_ports_of(struct dsa_switch *ds,
				     struct device_node *dn)
618
{
619
	struct device_node *ports, *port;
620
	struct dsa_port *dp;
621
	int err = 0;
622
	u32 reg;
623 624 625 626 627 628

	ports = of_get_child_by_name(dn, "ports");
	if (!ports) {
		dev_err(ds->dev, "no ports child node found\n");
		return -EINVAL;
	}
629 630 631 632

	for_each_available_child_of_node(ports, port) {
		err = of_property_read_u32(port, "reg", &reg);
		if (err)
633
			goto out_put_node;
634

635 636 637 638
		if (reg >= ds->num_ports) {
			err = -EINVAL;
			goto out_put_node;
		}
639

640 641 642 643
		dp = &ds->ports[reg];

		err = dsa_port_parse_of(dp, port);
		if (err)
644
			goto out_put_node;
645 646
	}

647 648 649
out_put_node:
	of_node_put(ports);
	return err;
650 651
}

V
Vivien Didelot 已提交
652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684
static int dsa_switch_parse_member_of(struct dsa_switch *ds,
				      struct device_node *dn)
{
	u32 m[2] = { 0, 0 };
	int sz;

	/* Don't error out if this optional property isn't found */
	sz = of_property_read_variable_u32_array(dn, "dsa,member", m, 2, 2);
	if (sz < 0 && sz != -EINVAL)
		return sz;

	ds->index = m[1];
	if (ds->index >= DSA_MAX_SWITCHES)
		return -EINVAL;

	ds->dst = dsa_tree_touch(m[0]);
	if (!ds->dst)
		return -ENOMEM;

	return 0;
}

static int dsa_switch_parse_of(struct dsa_switch *ds, struct device_node *dn)
{
	int err;

	err = dsa_switch_parse_member_of(ds, dn);
	if (err)
		return err;

	return dsa_switch_parse_ports_of(ds, dn);
}

685 686 687
static int dsa_port_parse(struct dsa_port *dp, const char *name,
			  struct device *dev)
{
688
	if (!strcmp(name, "cpu")) {
689 690 691 692 693 694 695 696
		struct net_device *master;

		master = dsa_dev_to_net_device(dev);
		if (!master)
			return -EPROBE_DEFER;

		dev_put(master);

697
		return dsa_port_parse_cpu(dp, master);
698 699
	}

700 701
	if (!strcmp(name, "dsa"))
		return dsa_port_parse_dsa(dp);
702

703
	return dsa_port_parse_user(dp, name);
704 705
}

V
Vivien Didelot 已提交
706 707
static int dsa_switch_parse_ports(struct dsa_switch *ds,
				  struct dsa_chip_data *cd)
708 709
{
	bool valid_name_found = false;
710 711 712
	struct dsa_port *dp;
	struct device *dev;
	const char *name;
713
	unsigned int i;
714
	int err;
715 716

	for (i = 0; i < DSA_MAX_PORTS; i++) {
717 718 719 720 721
		name = cd->port_names[i];
		dev = cd->netdev[i];
		dp = &ds->ports[i];

		if (!name)
722 723
			continue;

724 725 726 727
		err = dsa_port_parse(dp, name, dev);
		if (err)
			return err;

728 729 730 731 732 733 734 735 736
		valid_name_found = true;
	}

	if (!valid_name_found && i == DSA_MAX_PORTS)
		return -EINVAL;

	return 0;
}

V
Vivien Didelot 已提交
737
static int dsa_switch_parse(struct dsa_switch *ds, struct dsa_chip_data *cd)
738
{
V
Vivien Didelot 已提交
739
	ds->cd = cd;
740

V
Vivien Didelot 已提交
741 742 743 744 745 746 747
	/* We don't support interconnected switches nor multiple trees via
	 * platform data, so this is the unique switch of the tree.
	 */
	ds->index = 0;
	ds->dst = dsa_tree_touch(0);
	if (!ds->dst)
		return -ENOMEM;
748

V
Vivien Didelot 已提交
749
	return dsa_switch_parse_ports(ds, cd);
750 751
}

752 753 754 755 756 757 758
static int dsa_switch_add(struct dsa_switch *ds)
{
	struct dsa_switch_tree *dst = ds->dst;

	return dsa_tree_add_switch(dst, ds);
}

759
static int dsa_switch_probe(struct dsa_switch *ds)
760
{
761 762
	struct dsa_chip_data *pdata = ds->dev->platform_data;
	struct device_node *np = ds->dev->of_node;
V
Vivien Didelot 已提交
763
	int err;
764

V
Vivien Didelot 已提交
765 766 767 768 769 770
	if (np)
		err = dsa_switch_parse_of(ds, np);
	else if (pdata)
		err = dsa_switch_parse(ds, pdata);
	else
		err = -ENODEV;
771

V
Vivien Didelot 已提交
772 773
	if (err)
		return err;
774

775
	return dsa_switch_add(ds);
776 777
}

778 779 780
struct dsa_switch *dsa_switch_alloc(struct device *dev, size_t n)
{
	struct dsa_switch *ds;
781
	int i;
782

783
	ds = devm_kzalloc(dev, struct_size(ds, ports, n), GFP_KERNEL);
784 785 786
	if (!ds)
		return NULL;

S
Salvatore Mesoraca 已提交
787 788 789 790 791 792 793 794 795 796 797 798 799 800
	/* We avoid allocating memory outside dsa_switch
	 * if it is not needed.
	 */
	if (n <= sizeof(ds->_bitmap) * 8) {
		ds->bitmap = &ds->_bitmap;
	} else {
		ds->bitmap = devm_kcalloc(dev,
					  BITS_TO_LONGS(n),
					  sizeof(unsigned long),
					  GFP_KERNEL);
		if (unlikely(!ds->bitmap))
			return NULL;
	}

801 802 803
	ds->dev = dev;
	ds->num_ports = n;

804 805 806 807 808
	for (i = 0; i < ds->num_ports; ++i) {
		ds->ports[i].index = i;
		ds->ports[i].ds = ds;
	}

809 810 811 812
	return ds;
}
EXPORT_SYMBOL_GPL(dsa_switch_alloc);

813
int dsa_register_switch(struct dsa_switch *ds)
814 815 816 817
{
	int err;

	mutex_lock(&dsa2_mutex);
818
	err = dsa_switch_probe(ds);
819
	dsa_tree_put(ds->dst);
820 821 822 823 824 825
	mutex_unlock(&dsa2_mutex);

	return err;
}
EXPORT_SYMBOL_GPL(dsa_register_switch);

826
static void dsa_switch_remove(struct dsa_switch *ds)
827 828
{
	struct dsa_switch_tree *dst = ds->dst;
829
	unsigned int index = ds->index;
830

831
	dsa_tree_remove_switch(dst, index);
832 833 834 835 836
}

void dsa_unregister_switch(struct dsa_switch *ds)
{
	mutex_lock(&dsa2_mutex);
837
	dsa_switch_remove(ds);
838 839 840
	mutex_unlock(&dsa2_mutex);
}
EXPORT_SYMBOL_GPL(dsa_unregister_switch);