device.c 41.3 KB
Newer Older
L
Linus Torvalds 已提交
1 2
/*
 * Copyright (c) 2004 Topspin Communications.  All rights reserved.
3
 * Copyright (c) 2005 Sun Microsystems, Inc. All rights reserved.
L
Linus Torvalds 已提交
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
 *
 * This software is available to you under a choice of one of two
 * licenses.  You may choose to be licensed under the terms of the GNU
 * General Public License (GPL) Version 2, available from the file
 * COPYING in the main directory of this source tree, or the
 * OpenIB.org BSD license below:
 *
 *     Redistribution and use in source and binary forms, with or
 *     without modification, are permitted provided that the following
 *     conditions are met:
 *
 *      - Redistributions of source code must retain the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer.
 *
 *      - Redistributions in binary form must reproduce the above
 *        copyright notice, this list of conditions and the following
 *        disclaimer in the documentation and/or other materials
 *        provided with the distribution.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <linux/module.h>
#include <linux/string.h>
#include <linux/errno.h>
37
#include <linux/kernel.h>
L
Linus Torvalds 已提交
38 39
#include <linux/slab.h>
#include <linux/init.h>
40
#include <linux/netdevice.h>
41 42
#include <linux/security.h>
#include <linux/notifier.h>
R
Roland Dreier 已提交
43
#include <rdma/rdma_netlink.h>
44 45
#include <rdma/ib_addr.h>
#include <rdma/ib_cache.h>
L
Linus Torvalds 已提交
46 47

#include "core_priv.h"
48
#include "restrack.h"
L
Linus Torvalds 已提交
49 50 51 52 53

MODULE_AUTHOR("Roland Dreier");
MODULE_DESCRIPTION("core kernel InfiniBand API");
MODULE_LICENSE("Dual BSD/GPL");

54
struct workqueue_struct *ib_comp_wq;
55
struct workqueue_struct *ib_comp_unbound_wq;
T
Tejun Heo 已提交
56 57 58
struct workqueue_struct *ib_wq;
EXPORT_SYMBOL_GPL(ib_wq);

59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
/*
 * Each of the three rwsem locks (devices, clients, client_data) protects the
 * xarray of the same name. Specifically it allows the caller to assert that
 * the MARK will/will not be changing under the lock, and for devices and
 * clients, that the value in the xarray is still a valid pointer. Change of
 * the MARK is linked to the object state, so holding the lock and testing the
 * MARK also asserts that the contained object is in a certain state.
 *
 * This is used to build a two stage register/unregister flow where objects
 * can continue to be in the xarray even though they are still in progress to
 * register/unregister.
 *
 * The xarray itself provides additional locking, and restartable iteration,
 * which is also relied on.
 *
 * Locks should not be nested, with the exception of client_data, which is
 * allowed to nest under the read side of the other two locks.
 *
 * The devices_rwsem also protects the device name list, any change or
 * assignment of device name must also hold the write side to guarantee unique
 * names.
 */

82 83 84 85 86 87 88 89
/*
 * devices contains devices that have had their names assigned. The
 * devices may not be registered. Users that care about the registration
 * status need to call ib_device_try_get() on the device to ensure it is
 * registered, and keep it registered, for the required duration.
 *
 */
static DEFINE_XARRAY_FLAGS(devices, XA_FLAGS_ALLOC);
90
static DECLARE_RWSEM(devices_rwsem);
91 92
#define DEVICE_REGISTERED XA_MARK_1

L
Linus Torvalds 已提交
93
static LIST_HEAD(client_list);
94 95
#define CLIENT_REGISTERED XA_MARK_1
static DEFINE_XARRAY_FLAGS(clients, XA_FLAGS_ALLOC);
96
static DECLARE_RWSEM(clients_rwsem);
L
Linus Torvalds 已提交
97 98

/*
99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136
 * If client_data is registered then the corresponding client must also still
 * be registered.
 */
#define CLIENT_DATA_REGISTERED XA_MARK_1
/*
 * xarray has this behavior where it won't iterate over NULL values stored in
 * allocated arrays.  So we need our own iterator to see all values stored in
 * the array. This does the same thing as xa_for_each except that it also
 * returns NULL valued entries if the array is allocating. Simplified to only
 * work on simple xarrays.
 */
static void *xan_find_marked(struct xarray *xa, unsigned long *indexp,
			     xa_mark_t filter)
{
	XA_STATE(xas, xa, *indexp);
	void *entry;

	rcu_read_lock();
	do {
		entry = xas_find_marked(&xas, ULONG_MAX, filter);
		if (xa_is_zero(entry))
			break;
	} while (xas_retry(&xas, entry));
	rcu_read_unlock();

	if (entry) {
		*indexp = xas.xa_index;
		if (xa_is_zero(entry))
			return NULL;
		return entry;
	}
	return XA_ERROR(-ENOENT);
}
#define xan_for_each_marked(xa, index, entry, filter)                          \
	for (index = 0, entry = xan_find_marked(xa, &(index), filter);         \
	     !xa_is_err(entry);                                                \
	     (index)++, entry = xan_find_marked(xa, &(index), filter))

137 138 139 140 141 142 143 144
static int ib_security_change(struct notifier_block *nb, unsigned long event,
			      void *lsm_data);
static void ib_policy_change_task(struct work_struct *work);
static DECLARE_WORK(ib_policy_change_work, ib_policy_change_task);

static struct notifier_block ibdev_lsm_nb = {
	.notifier_call = ib_security_change,
};
L
Linus Torvalds 已提交
145 146 147

static int ib_device_check_mandatory(struct ib_device *device)
{
K
Kamal Heib 已提交
148
#define IB_MANDATORY_FUNC(x) { offsetof(struct ib_device_ops, x), #x }
L
Linus Torvalds 已提交
149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167
	static const struct {
		size_t offset;
		char  *name;
	} mandatory_table[] = {
		IB_MANDATORY_FUNC(query_device),
		IB_MANDATORY_FUNC(query_port),
		IB_MANDATORY_FUNC(query_pkey),
		IB_MANDATORY_FUNC(alloc_pd),
		IB_MANDATORY_FUNC(dealloc_pd),
		IB_MANDATORY_FUNC(create_qp),
		IB_MANDATORY_FUNC(modify_qp),
		IB_MANDATORY_FUNC(destroy_qp),
		IB_MANDATORY_FUNC(post_send),
		IB_MANDATORY_FUNC(post_recv),
		IB_MANDATORY_FUNC(create_cq),
		IB_MANDATORY_FUNC(destroy_cq),
		IB_MANDATORY_FUNC(poll_cq),
		IB_MANDATORY_FUNC(req_notify_cq),
		IB_MANDATORY_FUNC(get_dma_mr),
168 169
		IB_MANDATORY_FUNC(dereg_mr),
		IB_MANDATORY_FUNC(get_port_immutable)
L
Linus Torvalds 已提交
170 171 172
	};
	int i;

173
	device->kverbs_provider = true;
174
	for (i = 0; i < ARRAY_SIZE(mandatory_table); ++i) {
K
Kamal Heib 已提交
175 176
		if (!*(void **) ((void *) &device->ops +
				 mandatory_table[i].offset)) {
177 178
			device->kverbs_provider = false;
			break;
L
Linus Torvalds 已提交
179 180 181 182 183 184
		}
	}

	return 0;
}

185
/*
186 187
 * Caller must perform ib_device_put() to return the device reference count
 * when ib_device_get_by_index() returns valid device pointer.
188 189 190 191 192
 */
struct ib_device *ib_device_get_by_index(u32 index)
{
	struct ib_device *device;

193
	down_read(&devices_rwsem);
194
	device = xa_load(&devices, index);
195
	if (device) {
196
		if (!ib_device_try_get(device))
197 198
			device = NULL;
	}
199
	up_read(&devices_rwsem);
200 201 202
	return device;
}

203 204 205 206 207 208 209
/**
 * ib_device_put - Release IB device reference
 * @device: device whose reference to be released
 *
 * ib_device_put() releases reference to the IB device to allow it to be
 * unregistered and eventually free.
 */
210 211 212 213 214
void ib_device_put(struct ib_device *device)
{
	if (refcount_dec_and_test(&device->refcount))
		complete(&device->unreg_completion);
}
215
EXPORT_SYMBOL(ib_device_put);
216

L
Linus Torvalds 已提交
217 218 219
static struct ib_device *__ib_device_get_by_name(const char *name)
{
	struct ib_device *device;
220
	unsigned long index;
L
Linus Torvalds 已提交
221

222
	xa_for_each (&devices, index, device)
223
		if (!strcmp(name, dev_name(&device->dev)))
L
Linus Torvalds 已提交
224 225 226 227 228
			return device;

	return NULL;
}

229 230
int ib_device_rename(struct ib_device *ibdev, const char *name)
{
231
	int ret;
232

233
	down_write(&devices_rwsem);
234 235 236 237 238
	if (!strcmp(name, dev_name(&ibdev->dev))) {
		ret = 0;
		goto out;
	}

239 240 241
	if (__ib_device_get_by_name(name)) {
		ret = -EEXIST;
		goto out;
242 243 244 245 246 247 248
	}

	ret = device_rename(&ibdev->dev, name);
	if (ret)
		goto out;
	strlcpy(ibdev->name, name, IB_DEVICE_NAME_MAX);
out:
249
	up_write(&devices_rwsem);
250 251 252
	return ret;
}

253
static int alloc_name(struct ib_device *ibdev, const char *name)
L
Linus Torvalds 已提交
254 255
{
	struct ib_device *device;
256
	unsigned long index;
257 258
	struct ida inuse;
	int rc;
L
Linus Torvalds 已提交
259 260
	int i;

261
	lockdep_assert_held_exclusive(&devices_rwsem);
262
	ida_init(&inuse);
263
	xa_for_each (&devices, index, device) {
264 265
		char buf[IB_DEVICE_NAME_MAX];

266
		if (sscanf(dev_name(&device->dev), name, &i) != 1)
L
Linus Torvalds 已提交
267
			continue;
268
		if (i < 0 || i >= INT_MAX)
L
Linus Torvalds 已提交
269 270
			continue;
		snprintf(buf, sizeof buf, name, i);
271 272 273 274 275 276
		if (strcmp(buf, dev_name(&device->dev)) != 0)
			continue;

		rc = ida_alloc_range(&inuse, i, i, GFP_KERNEL);
		if (rc < 0)
			goto out;
L
Linus Torvalds 已提交
277 278
	}

279 280 281
	rc = ida_alloc(&inuse, GFP_KERNEL);
	if (rc < 0)
		goto out;
L
Linus Torvalds 已提交
282

283 284 285 286
	rc = dev_set_name(&ibdev->dev, name, rc);
out:
	ida_destroy(&inuse);
	return rc;
L
Linus Torvalds 已提交
287 288
}

289 290 291 292
static void ib_device_release(struct device *device)
{
	struct ib_device *dev = container_of(device, struct ib_device, dev);

293
	WARN_ON(refcount_read(&dev->refcount));
294
	ib_cache_release_one(dev);
295 296
	ib_security_release_port_pkey_list(dev);
	kfree(dev->port_pkey_list);
297
	kfree(dev->port_immutable);
298
	xa_destroy(&dev->client_data);
299 300 301 302 303 304
	kfree(dev);
}

static int ib_device_uevent(struct device *device,
			    struct kobj_uevent_env *env)
{
305
	if (add_uevent_var(env, "NAME=%s", dev_name(device)))
306 307 308 309 310 311 312 313 314 315 316 317 318 319 320
		return -ENOMEM;

	/*
	 * It would be nice to pass the node GUID with the event...
	 */

	return 0;
}

static struct class ib_class = {
	.name    = "infiniband",
	.dev_release = ib_device_release,
	.dev_uevent = ib_device_uevent,
};

L
Linus Torvalds 已提交
321
/**
322
 * _ib_alloc_device - allocate an IB device struct
L
Linus Torvalds 已提交
323 324 325 326 327 328 329 330
 * @size:size of structure to allocate
 *
 * Low-level drivers should use ib_alloc_device() to allocate &struct
 * ib_device.  @size is the size of the structure to be allocated,
 * including any private data used by the low-level driver.
 * ib_dealloc_device() must be used to free structures allocated with
 * ib_alloc_device().
 */
331
struct ib_device *_ib_alloc_device(size_t size)
L
Linus Torvalds 已提交
332
{
333 334 335 336 337 338 339 340 341
	struct ib_device *device;

	if (WARN_ON(size < sizeof(struct ib_device)))
		return NULL;

	device = kzalloc(size, GFP_KERNEL);
	if (!device)
		return NULL;

342 343 344 345
	if (rdma_restrack_init(device)) {
		kfree(device);
		return NULL;
	}
346

347
	device->dev.class = &ib_class;
348 349
	device->groups[0] = &ib_dev_attr_group;
	device->dev.groups = device->groups;
350 351 352 353
	device_initialize(&device->dev);

	INIT_LIST_HEAD(&device->event_handler_list);
	spin_lock_init(&device->event_handler_lock);
354 355 356 357 358
	/*
	 * client_data needs to be alloc because we don't want our mark to be
	 * destroyed if the user stores NULL in the client data.
	 */
	xa_init_flags(&device->client_data, XA_FLAGS_ALLOC);
359
	init_rwsem(&device->client_data_rwsem);
360
	INIT_LIST_HEAD(&device->port_list);
361
	init_completion(&device->unreg_completion);
L
Linus Torvalds 已提交
362

363
	return device;
L
Linus Torvalds 已提交
364
}
365
EXPORT_SYMBOL(_ib_alloc_device);
L
Linus Torvalds 已提交
366 367 368 369 370 371 372 373 374

/**
 * ib_dealloc_device - free an IB device struct
 * @device:structure to free
 *
 * Free a structure allocated with ib_alloc_device().
 */
void ib_dealloc_device(struct ib_device *device)
{
375
	WARN_ON(!xa_empty(&device->client_data));
376
	WARN_ON(refcount_read(&device->refcount));
377
	rdma_restrack_clean(device);
378
	/* Balances with device_initialize */
379
	put_device(&device->dev);
L
Linus Torvalds 已提交
380 381 382
}
EXPORT_SYMBOL(ib_dealloc_device);

383 384 385 386 387 388 389 390 391 392
/*
 * add_client_context() and remove_client_context() must be safe against
 * parallel calls on the same device - registration/unregistration of both the
 * device and client can be occurring in parallel.
 *
 * The routines need to be a fence, any caller must not return until the add
 * or remove is fully completed.
 */
static int add_client_context(struct ib_device *device,
			      struct ib_client *client)
L
Linus Torvalds 已提交
393
{
394
	int ret = 0;
L
Linus Torvalds 已提交
395

396
	if (!device->kverbs_provider && !client->no_kverbs_req)
397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431
		return 0;

	down_write(&device->client_data_rwsem);
	/*
	 * Another caller to add_client_context got here first and has already
	 * completely initialized context.
	 */
	if (xa_get_mark(&device->client_data, client->client_id,
		    CLIENT_DATA_REGISTERED))
		goto out;

	ret = xa_err(xa_store(&device->client_data, client->client_id, NULL,
			      GFP_KERNEL));
	if (ret)
		goto out;
	downgrade_write(&device->client_data_rwsem);
	if (client->add)
		client->add(device);

	/* Readers shall not see a client until add has been completed */
	xa_set_mark(&device->client_data, client->client_id,
		    CLIENT_DATA_REGISTERED);
	up_read(&device->client_data_rwsem);
	return 0;

out:
	up_write(&device->client_data_rwsem);
	return ret;
}

static void remove_client_context(struct ib_device *device,
				  unsigned int client_id)
{
	struct ib_client *client;
	void *client_data;
432

433 434 435 436 437 438 439 440 441 442
	down_write(&device->client_data_rwsem);
	if (!xa_get_mark(&device->client_data, client_id,
			 CLIENT_DATA_REGISTERED)) {
		up_write(&device->client_data_rwsem);
		return;
	}
	client_data = xa_load(&device->client_data, client_id);
	xa_clear_mark(&device->client_data, client_id, CLIENT_DATA_REGISTERED);
	client = xa_load(&clients, client_id);
	downgrade_write(&device->client_data_rwsem);
L
Linus Torvalds 已提交
443

444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462
	/*
	 * Notice we cannot be holding any exclusive locks when calling the
	 * remove callback as the remove callback can recurse back into any
	 * public functions in this module and thus try for any locks those
	 * functions take.
	 *
	 * For this reason clients and drivers should not call the
	 * unregistration functions will holdling any locks.
	 *
	 * It tempting to drop the client_data_rwsem too, but this is required
	 * to ensure that unregister_client does not return until all clients
	 * are completely unregistered, which is required to avoid module
	 * unloading races.
	 */
	if (client->remove)
		client->remove(device, client_data);

	xa_erase(&device->client_data, client_id);
	up_read(&device->client_data_rwsem);
L
Linus Torvalds 已提交
463 464
}

465 466 467 468 469 470
static int verify_immutable(const struct ib_device *dev, u8 port)
{
	return WARN_ON(!rdma_cap_ib_mad(dev, port) &&
			    rdma_max_mad_size(dev, port) != 0);
}

471
static int read_port_immutable(struct ib_device *device)
472
{
473
	int ret;
474 475 476 477 478 479 480 481 482 483 484
	u8 start_port = rdma_start_port(device);
	u8 end_port = rdma_end_port(device);
	u8 port;

	/**
	 * device->port_immutable is indexed directly by the port number to make
	 * access to this data as efficient as possible.
	 *
	 * Therefore port_immutable is declared as a 1 based array with
	 * potential empty slots at the beginning.
	 */
K
Kees Cook 已提交
485 486
	device->port_immutable = kcalloc(end_port + 1,
					 sizeof(*device->port_immutable),
487 488
					 GFP_KERNEL);
	if (!device->port_immutable)
489
		return -ENOMEM;
490

491
	for (port = start_port; port <= end_port; ++port) {
K
Kamal Heib 已提交
492 493
		ret = device->ops.get_port_immutable(
			device, port, &device->port_immutable[port]);
494
		if (ret)
495
			return ret;
496

497 498
		if (verify_immutable(device, port))
			return -EINVAL;
499
	}
500
	return 0;
501 502
}

503
void ib_get_device_fw_str(struct ib_device *dev, char *str)
504
{
K
Kamal Heib 已提交
505 506
	if (dev->ops.get_dev_fw_str)
		dev->ops.get_dev_fw_str(dev, str);
507 508 509 510 511
	else
		str[0] = '\0';
}
EXPORT_SYMBOL(ib_get_device_fw_str);

512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535
static int setup_port_pkey_list(struct ib_device *device)
{
	int i;

	/**
	 * device->port_pkey_list is indexed directly by the port number,
	 * Therefore it is declared as a 1 based array with potential empty
	 * slots at the beginning.
	 */
	device->port_pkey_list = kcalloc(rdma_end_port(device) + 1,
					 sizeof(*device->port_pkey_list),
					 GFP_KERNEL);

	if (!device->port_pkey_list)
		return -ENOMEM;

	for (i = 0; i < (rdma_end_port(device) + 1); i++) {
		spin_lock_init(&device->port_pkey_list[i].list_lock);
		INIT_LIST_HEAD(&device->port_pkey_list[i].pkey_list);
	}

	return 0;
}

536 537 538
static void ib_policy_change_task(struct work_struct *work)
{
	struct ib_device *dev;
539
	unsigned long index;
540

541
	down_read(&devices_rwsem);
542
	xa_for_each_marked (&devices, index, dev, DEVICE_REGISTERED) {
543 544 545 546 547 548 549 550 551 552 553
		int i;

		for (i = rdma_start_port(dev); i <= rdma_end_port(dev); i++) {
			u64 sp;
			int ret = ib_get_cached_subnet_prefix(dev,
							      i,
							      &sp);

			WARN_ONCE(ret,
				  "ib_get_cached_subnet_prefix err: %d, this should never happen here\n",
				  ret);
554 555
			if (!ret)
				ib_security_cache_change(dev, i, sp);
556 557
		}
	}
558
	up_read(&devices_rwsem);
559 560 561 562 563 564 565 566 567
}

static int ib_security_change(struct notifier_block *nb, unsigned long event,
			      void *lsm_data)
{
	if (event != LSM_POLICY_CHANGE)
		return NOTIFY_DONE;

	schedule_work(&ib_policy_change_work);
568
	ib_mad_agent_security_change();
569 570 571 572

	return NOTIFY_OK;
}

573 574
/*
 * Assign the unique string device name and the unique device index.
575
 */
576
static int assign_name(struct ib_device *device, const char *name)
577
{
578 579
	static u32 last_id;
	int ret;
580

581
	down_write(&devices_rwsem);
582 583 584 585 586 587 588 589 590 591 592 593 594
	/* Assign a unique name to the device */
	if (strchr(name, '%'))
		ret = alloc_name(device, name);
	else
		ret = dev_set_name(&device->dev, name);
	if (ret)
		goto out;

	if (__ib_device_get_by_name(dev_name(&device->dev))) {
		ret = -ENFILE;
		goto out;
	}
	strlcpy(device->name, dev_name(&device->dev), IB_DEVICE_NAME_MAX);
595

596 597 598 599 600 601 602
	/* Cyclically allocate a user visible ID for the device */
	device->index = last_id;
	ret = xa_alloc(&devices, &device->index, INT_MAX, device, GFP_KERNEL);
	if (ret == -ENOSPC) {
		device->index = 0;
		ret = xa_alloc(&devices, &device->index, INT_MAX, device,
			       GFP_KERNEL);
603
	}
604 605 606 607 608
	if (ret)
		goto out;
	last_id = device->index + 1;

	ret = 0;
609

610
out:
611
	up_write(&devices_rwsem);
612 613 614 615 616
	return ret;
}

static void release_name(struct ib_device *device)
{
617
	down_write(&devices_rwsem);
618
	xa_erase(&devices, device->index);
619
	up_write(&devices_rwsem);
620 621
}

622
static void setup_dma_device(struct ib_device *device)
L
Linus Torvalds 已提交
623
{
624 625
	struct device *parent = device->dev.parent;

626 627 628 629 630 631 632 633
	WARN_ON_ONCE(device->dma_device);
	if (device->dev.dma_ops) {
		/*
		 * The caller provided custom DMA operations. Copy the
		 * DMA-related fields that are used by e.g. dma_alloc_coherent()
		 * into device->dev.
		 */
		device->dma_device = &device->dev;
634 635 636 637 638 639 640 641 642 643 644 645 646
		if (!device->dev.dma_mask) {
			if (parent)
				device->dev.dma_mask = parent->dma_mask;
			else
				WARN_ON_ONCE(true);
		}
		if (!device->dev.coherent_dma_mask) {
			if (parent)
				device->dev.coherent_dma_mask =
					parent->coherent_dma_mask;
			else
				WARN_ON_ONCE(true);
		}
647 648 649 650 651
	} else {
		/*
		 * The caller did not provide custom DMA operations. Use the
		 * DMA mapping operations of the parent device.
		 */
652
		WARN_ON_ONCE(!parent);
653 654
		device->dma_device = parent;
	}
655
}
L
Linus Torvalds 已提交
656

657 658 659 660 661
/*
 * setup_device() allocates memory and sets up data that requires calling the
 * device ops, this is the only reason these actions are not done during
 * ib_alloc_device. It is undone by ib_dealloc_device().
 */
662 663 664 665
static int setup_device(struct ib_device *device)
{
	struct ib_udata uhw = {.outlen = 0, .inlen = 0};
	int ret;
L
Linus Torvalds 已提交
666

667 668
	setup_dma_device(device);

669 670 671
	ret = ib_device_check_mandatory(device);
	if (ret)
		return ret;
L
Linus Torvalds 已提交
672

673
	ret = read_port_immutable(device);
674
	if (ret) {
675 676
		dev_warn(&device->dev,
			 "Couldn't create per port immutable data\n");
677 678 679 680
		return ret;
	}

	memset(&device->attrs, 0, sizeof(device->attrs));
K
Kamal Heib 已提交
681
	ret = device->ops.query_device(device, &device->attrs, &uhw);
682 683 684
	if (ret) {
		dev_warn(&device->dev,
			 "Couldn't query the device attributes\n");
685
		return ret;
686 687
	}

688 689
	ret = setup_port_pkey_list(device);
	if (ret) {
690
		dev_warn(&device->dev, "Couldn't create per port_pkey_list\n");
691
		return ret;
692
	}
693

694
	return 0;
695 696
}

697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744
static void disable_device(struct ib_device *device)
{
	struct ib_client *client;

	WARN_ON(!refcount_read(&device->refcount));

	down_write(&devices_rwsem);
	xa_clear_mark(&devices, device->index, DEVICE_REGISTERED);
	up_write(&devices_rwsem);

	down_read(&clients_rwsem);
	list_for_each_entry_reverse(client, &client_list, list)
		remove_client_context(device, client->client_id);
	up_read(&clients_rwsem);

	/* Pairs with refcount_set in enable_device */
	ib_device_put(device);
	wait_for_completion(&device->unreg_completion);
}

/*
 * An enabled device is visible to all clients and to all the public facing
 * APIs that return a device pointer.
 */
static int enable_device(struct ib_device *device)
{
	struct ib_client *client;
	unsigned long index;
	int ret;

	refcount_set(&device->refcount, 1);
	down_write(&devices_rwsem);
	xa_set_mark(&devices, device->index, DEVICE_REGISTERED);
	up_write(&devices_rwsem);

	down_read(&clients_rwsem);
	xa_for_each_marked (&clients, index, client, CLIENT_REGISTERED) {
		ret = add_client_context(device, client);
		if (ret) {
			up_read(&clients_rwsem);
			disable_device(device);
			return ret;
		}
	}
	up_read(&clients_rwsem);
	return 0;
}

745 746 747 748 749 750 751 752 753
/**
 * ib_register_device - Register an IB device with IB core
 * @device:Device to register
 *
 * Low-level drivers use ib_register_device() to register their
 * devices with the IB core.  All registered clients will receive a
 * callback for each device that is added. @device must be allocated
 * with ib_alloc_device().
 */
754
int ib_register_device(struct ib_device *device, const char *name)
755 756 757
{
	int ret;

758 759
	ret = assign_name(device, name);
	if (ret)
760
		return ret;
761 762 763

	ret = setup_device(device);
	if (ret)
764
		goto out;
765

766 767 768 769
	ret = ib_cache_setup_one(device);
	if (ret) {
		dev_warn(&device->dev,
			 "Couldn't set up InfiniBand P_Key/GID cache\n");
770
		goto out;
771 772
	}

773
	ib_device_register_rdmacg(device);
774

775 776 777 778
	ret = device_add(&device->dev);
	if (ret)
		goto cg_cleanup;

779
	ret = ib_device_register_sysfs(device);
L
Linus Torvalds 已提交
780
	if (ret) {
781 782
		dev_warn(&device->dev,
			 "Couldn't register device with driver model\n");
783
		goto dev_cleanup;
L
Linus Torvalds 已提交
784 785
	}

786 787 788
	ret = enable_device(device);
	if (ret)
		goto sysfs_cleanup;
L
Linus Torvalds 已提交
789

790 791
	return 0;

792 793
sysfs_cleanup:
	ib_device_unregister_sysfs(device);
794 795
dev_cleanup:
	device_del(&device->dev);
796 797
cg_cleanup:
	ib_device_unregister_rdmacg(device);
798
	ib_cache_cleanup_one(device);
799
out:
800
	release_name(device);
L
Linus Torvalds 已提交
801 802 803 804 805 806 807 808 809 810 811 812
	return ret;
}
EXPORT_SYMBOL(ib_register_device);

/**
 * ib_unregister_device - Unregister an IB device
 * @device:Device to unregister
 *
 * Unregister an IB device.  All clients will receive a remove callback.
 */
void ib_unregister_device(struct ib_device *device)
{
813
	disable_device(device);
814
	ib_device_unregister_sysfs(device);
815
	device_del(&device->dev);
816
	ib_device_unregister_rdmacg(device);
817
	ib_cache_cleanup_one(device);
818
	release_name(device);
L
Linus Torvalds 已提交
819 820 821
}
EXPORT_SYMBOL(ib_unregister_device);

822 823 824 825
static int assign_client_id(struct ib_client *client)
{
	int ret;

826
	down_write(&clients_rwsem);
827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844
	/*
	 * The add/remove callbacks must be called in FIFO/LIFO order. To
	 * achieve this we assign client_ids so they are sorted in
	 * registration order, and retain a linked list we can reverse iterate
	 * to get the LIFO order. The extra linked list can go away if xarray
	 * learns to reverse iterate.
	 */
	if (list_empty(&client_list))
		client->client_id = 0;
	else
		client->client_id =
			list_last_entry(&client_list, struct ib_client, list)
				->client_id;
	ret = xa_alloc(&clients, &client->client_id, INT_MAX, client,
		       GFP_KERNEL);
	if (ret)
		goto out;

845 846 847
	xa_set_mark(&clients, client->client_id, CLIENT_REGISTERED);
	list_add_tail(&client->list, &client_list);

848
out:
849
	up_write(&clients_rwsem);
850 851 852
	return ret;
}

L
Linus Torvalds 已提交
853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868
/**
 * ib_register_client - Register an IB client
 * @client:Client to register
 *
 * Upper level users of the IB drivers can use ib_register_client() to
 * register callbacks for IB device addition and removal.  When an IB
 * device is added, each registered client's add method will be called
 * (in the order the clients were registered), and when a device is
 * removed, each client's remove method will be called (in the reverse
 * order that clients were registered).  In addition, when
 * ib_register_client() is called, the client will receive an add
 * callback for all devices already registered.
 */
int ib_register_client(struct ib_client *client)
{
	struct ib_device *device;
869
	unsigned long index;
870
	int ret;
L
Linus Torvalds 已提交
871

872
	ret = assign_client_id(client);
873
	if (ret)
874
		return ret;
L
Linus Torvalds 已提交
875

876 877 878 879 880 881 882 883 884 885
	down_read(&devices_rwsem);
	xa_for_each_marked (&devices, index, device, DEVICE_REGISTERED) {
		ret = add_client_context(device, client);
		if (ret) {
			up_read(&devices_rwsem);
			ib_unregister_client(client);
			return ret;
		}
	}
	up_read(&devices_rwsem);
L
Linus Torvalds 已提交
886 887 888 889 890 891 892 893 894 895 896
	return 0;
}
EXPORT_SYMBOL(ib_register_client);

/**
 * ib_unregister_client - Unregister an IB client
 * @client:Client to unregister
 *
 * Upper level users use ib_unregister_client() to remove their client
 * registration.  When ib_unregister_client() is called, the client
 * will receive a remove callback for each IB device still registered.
897 898 899
 *
 * This is a full fence, once it returns no client callbacks will be called,
 * or are running in another thread.
L
Linus Torvalds 已提交
900 901 902 903
 */
void ib_unregister_client(struct ib_client *client)
{
	struct ib_device *device;
904
	unsigned long index;
L
Linus Torvalds 已提交
905

906
	down_write(&clients_rwsem);
907
	xa_clear_mark(&clients, client->client_id, CLIENT_REGISTERED);
908 909 910 911 912 913 914 915 916
	up_write(&clients_rwsem);
	/*
	 * Every device still known must be serialized to make sure we are
	 * done with the client callbacks before we return.
	 */
	down_read(&devices_rwsem);
	xa_for_each (&devices, index, device)
		remove_client_context(device, client->client_id);
	up_read(&devices_rwsem);
L
Linus Torvalds 已提交
917

918
	down_write(&clients_rwsem);
919 920
	list_del(&client->list);
	xa_erase(&clients, client->client_id);
921
	up_write(&clients_rwsem);
L
Linus Torvalds 已提交
922 923 924 925
}
EXPORT_SYMBOL(ib_unregister_client);

/**
926
 * ib_set_client_data - Set IB client context
L
Linus Torvalds 已提交
927 928 929 930
 * @device:Device to set context for
 * @client:Client to set context for
 * @data:Context to set
 *
931 932 933 934
 * ib_set_client_data() sets client context data that can be retrieved with
 * ib_get_client_data(). This can only be called while the client is
 * registered to the device, once the ib_client remove() callback returns this
 * cannot be called.
L
Linus Torvalds 已提交
935 936 937 938
 */
void ib_set_client_data(struct ib_device *device, struct ib_client *client,
			void *data)
{
939
	void *rc;
L
Linus Torvalds 已提交
940

941 942
	if (WARN_ON(IS_ERR(data)))
		data = NULL;
L
Linus Torvalds 已提交
943

944 945 946
	rc = xa_store(&device->client_data, client->client_id, data,
		      GFP_KERNEL);
	WARN_ON(xa_is_err(rc));
L
Linus Torvalds 已提交
947 948 949 950 951 952 953 954 955 956 957 958
}
EXPORT_SYMBOL(ib_set_client_data);

/**
 * ib_register_event_handler - Register an IB event handler
 * @event_handler:Handler to register
 *
 * ib_register_event_handler() registers an event handler that will be
 * called back when asynchronous IB events occur (as defined in
 * chapter 11 of the InfiniBand Architecture Specification).  This
 * callback may occur in interrupt context.
 */
959
void ib_register_event_handler(struct ib_event_handler *event_handler)
L
Linus Torvalds 已提交
960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976
{
	unsigned long flags;

	spin_lock_irqsave(&event_handler->device->event_handler_lock, flags);
	list_add_tail(&event_handler->list,
		      &event_handler->device->event_handler_list);
	spin_unlock_irqrestore(&event_handler->device->event_handler_lock, flags);
}
EXPORT_SYMBOL(ib_register_event_handler);

/**
 * ib_unregister_event_handler - Unregister an event handler
 * @event_handler:Handler to unregister
 *
 * Unregister an event handler registered with
 * ib_register_event_handler().
 */
977
void ib_unregister_event_handler(struct ib_event_handler *event_handler)
L
Linus Torvalds 已提交
978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021
{
	unsigned long flags;

	spin_lock_irqsave(&event_handler->device->event_handler_lock, flags);
	list_del(&event_handler->list);
	spin_unlock_irqrestore(&event_handler->device->event_handler_lock, flags);
}
EXPORT_SYMBOL(ib_unregister_event_handler);

/**
 * ib_dispatch_event - Dispatch an asynchronous event
 * @event:Event to dispatch
 *
 * Low-level drivers must call ib_dispatch_event() to dispatch the
 * event to all registered event handlers when an asynchronous event
 * occurs.
 */
void ib_dispatch_event(struct ib_event *event)
{
	unsigned long flags;
	struct ib_event_handler *handler;

	spin_lock_irqsave(&event->device->event_handler_lock, flags);

	list_for_each_entry(handler, &event->device->event_handler_list, list)
		handler->handler(handler, event);

	spin_unlock_irqrestore(&event->device->event_handler_lock, flags);
}
EXPORT_SYMBOL(ib_dispatch_event);

/**
 * ib_query_port - Query IB port attributes
 * @device:Device to query
 * @port_num:Port number to query
 * @port_attr:Port attributes
 *
 * ib_query_port() returns the attributes of a port through the
 * @port_attr pointer.
 */
int ib_query_port(struct ib_device *device,
		  u8 port_num,
		  struct ib_port_attr *port_attr)
{
1022 1023 1024
	union ib_gid gid;
	int err;

1025
	if (!rdma_is_port_valid(device, port_num))
1026 1027
		return -EINVAL;

1028
	memset(port_attr, 0, sizeof(*port_attr));
K
Kamal Heib 已提交
1029
	err = device->ops.query_port(device, port_num, port_attr);
1030 1031 1032
	if (err || port_attr->subnet_prefix)
		return err;

1033 1034 1035
	if (rdma_port_get_link_layer(device, port_num) != IB_LINK_LAYER_INFINIBAND)
		return 0;

K
Kamal Heib 已提交
1036
	err = device->ops.query_gid(device, port_num, 0, &gid);
1037 1038 1039 1040 1041
	if (err)
		return err;

	port_attr->subnet_prefix = be64_to_cpu(gid.global.subnet_prefix);
	return 0;
L
Linus Torvalds 已提交
1042 1043 1044
}
EXPORT_SYMBOL(ib_query_port);

1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069
/**
 * ib_enum_roce_netdev - enumerate all RoCE ports
 * @ib_dev : IB device we want to query
 * @filter: Should we call the callback?
 * @filter_cookie: Cookie passed to filter
 * @cb: Callback to call for each found RoCE ports
 * @cookie: Cookie passed back to the callback
 *
 * Enumerates all of the physical RoCE ports of ib_dev
 * which are related to netdevice and calls callback() on each
 * device for which filter() function returns non zero.
 */
void ib_enum_roce_netdev(struct ib_device *ib_dev,
			 roce_netdev_filter filter,
			 void *filter_cookie,
			 roce_netdev_callback cb,
			 void *cookie)
{
	u8 port;

	for (port = rdma_start_port(ib_dev); port <= rdma_end_port(ib_dev);
	     port++)
		if (rdma_protocol_roce(ib_dev, port)) {
			struct net_device *idev = NULL;

K
Kamal Heib 已提交
1070 1071
			if (ib_dev->ops.get_netdev)
				idev = ib_dev->ops.get_netdev(ib_dev, port);
1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103

			if (idev &&
			    idev->reg_state >= NETREG_UNREGISTERED) {
				dev_put(idev);
				idev = NULL;
			}

			if (filter(ib_dev, port, idev, filter_cookie))
				cb(ib_dev, port, idev, cookie);

			if (idev)
				dev_put(idev);
		}
}

/**
 * ib_enum_all_roce_netdevs - enumerate all RoCE devices
 * @filter: Should we call the callback?
 * @filter_cookie: Cookie passed to filter
 * @cb: Callback to call for each found RoCE ports
 * @cookie: Cookie passed back to the callback
 *
 * Enumerates all RoCE devices' physical ports which are related
 * to netdevices and calls callback() on each device for which
 * filter() function returns non zero.
 */
void ib_enum_all_roce_netdevs(roce_netdev_filter filter,
			      void *filter_cookie,
			      roce_netdev_callback cb,
			      void *cookie)
{
	struct ib_device *dev;
1104
	unsigned long index;
1105

1106
	down_read(&devices_rwsem);
1107
	xa_for_each_marked (&devices, index, dev, DEVICE_REGISTERED)
1108
		ib_enum_roce_netdev(dev, filter, filter_cookie, cb, cookie);
1109
	up_read(&devices_rwsem);
1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120
}

/**
 * ib_enum_all_devs - enumerate all ib_devices
 * @cb: Callback to call for each found ib_device
 *
 * Enumerates all ib_devices and calls callback() on each device.
 */
int ib_enum_all_devs(nldev_callback nldev_cb, struct sk_buff *skb,
		     struct netlink_callback *cb)
{
1121
	unsigned long index;
1122 1123 1124 1125
	struct ib_device *dev;
	unsigned int idx = 0;
	int ret = 0;

1126
	down_read(&devices_rwsem);
1127
	xa_for_each_marked (&devices, index, dev, DEVICE_REGISTERED) {
1128 1129 1130 1131 1132
		ret = nldev_cb(dev, skb, cb, idx);
		if (ret)
			break;
		idx++;
	}
1133
	up_read(&devices_rwsem);
1134
	return ret;
1135 1136
}

L
Linus Torvalds 已提交
1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148
/**
 * ib_query_pkey - Get P_Key table entry
 * @device:Device to query
 * @port_num:Port number to query
 * @index:P_Key table index to query
 * @pkey:Returned P_Key
 *
 * ib_query_pkey() fetches the specified P_Key table entry.
 */
int ib_query_pkey(struct ib_device *device,
		  u8 port_num, u16 index, u16 *pkey)
{
1149 1150 1151
	if (!rdma_is_port_valid(device, port_num))
		return -EINVAL;

K
Kamal Heib 已提交
1152
	return device->ops.query_pkey(device, port_num, index, pkey);
L
Linus Torvalds 已提交
1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168
}
EXPORT_SYMBOL(ib_query_pkey);

/**
 * ib_modify_device - Change IB device attributes
 * @device:Device to modify
 * @device_modify_mask:Mask of attributes to change
 * @device_modify:New attribute values
 *
 * ib_modify_device() changes a device's attributes as specified by
 * the @device_modify_mask and @device_modify structure.
 */
int ib_modify_device(struct ib_device *device,
		     int device_modify_mask,
		     struct ib_device_modify *device_modify)
{
K
Kamal Heib 已提交
1169
	if (!device->ops.modify_device)
1170 1171
		return -ENOSYS;

K
Kamal Heib 已提交
1172 1173
	return device->ops.modify_device(device, device_modify_mask,
					 device_modify);
L
Linus Torvalds 已提交
1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191
}
EXPORT_SYMBOL(ib_modify_device);

/**
 * ib_modify_port - Modifies the attributes for the specified port.
 * @device: The device to modify.
 * @port_num: The number of the port to modify.
 * @port_modify_mask: Mask used to specify which attributes of the port
 *   to change.
 * @port_modify: New attribute values for the port.
 *
 * ib_modify_port() changes a port's attributes as specified by the
 * @port_modify_mask and @port_modify structure.
 */
int ib_modify_port(struct ib_device *device,
		   u8 port_num, int port_modify_mask,
		   struct ib_port_modify *port_modify)
{
1192
	int rc;
1193

1194
	if (!rdma_is_port_valid(device, port_num))
1195 1196
		return -EINVAL;

K
Kamal Heib 已提交
1197 1198 1199 1200
	if (device->ops.modify_port)
		rc = device->ops.modify_port(device, port_num,
					     port_modify_mask,
					     port_modify);
1201 1202 1203
	else
		rc = rdma_protocol_roce(device, port_num) ? 0 : -ENOSYS;
	return rc;
L
Linus Torvalds 已提交
1204 1205 1206
}
EXPORT_SYMBOL(ib_modify_port);

1207 1208
/**
 * ib_find_gid - Returns the port number and GID table index where
1209
 *   a specified GID value occurs. Its searches only for IB link layer.
1210 1211 1212 1213 1214 1215 1216
 * @device: The device to query.
 * @gid: The GID value to search for.
 * @port_num: The port number of the device where the GID value was found.
 * @index: The index into the GID table where the GID was found.  This
 *   parameter may be NULL.
 */
int ib_find_gid(struct ib_device *device, union ib_gid *gid,
1217
		u8 *port_num, u16 *index)
1218 1219 1220 1221
{
	union ib_gid tmp_gid;
	int ret, port, i;

1222
	for (port = rdma_start_port(device); port <= rdma_end_port(device); ++port) {
1223
		if (!rdma_protocol_ib(device, port))
1224 1225
			continue;

1226
		for (i = 0; i < device->port_immutable[port].gid_tbl_len; ++i) {
1227
			ret = rdma_query_gid(device, port, i, &tmp_gid);
1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255
			if (ret)
				return ret;
			if (!memcmp(&tmp_gid, gid, sizeof *gid)) {
				*port_num = port;
				if (index)
					*index = i;
				return 0;
			}
		}
	}

	return -ENOENT;
}
EXPORT_SYMBOL(ib_find_gid);

/**
 * ib_find_pkey - Returns the PKey table index where a specified
 *   PKey value occurs.
 * @device: The device to query.
 * @port_num: The port number of the device to search for the PKey.
 * @pkey: The PKey value to search for.
 * @index: The index into the PKey table where the PKey was found.
 */
int ib_find_pkey(struct ib_device *device,
		 u8 port_num, u16 pkey, u16 *index)
{
	int ret, i;
	u16 tmp_pkey;
1256
	int partial_ix = -1;
1257

1258
	for (i = 0; i < device->port_immutable[port_num].pkey_tbl_len; ++i) {
1259 1260 1261
		ret = ib_query_pkey(device, port_num, i, &tmp_pkey);
		if (ret)
			return ret;
1262
		if ((pkey & 0x7fff) == (tmp_pkey & 0x7fff)) {
1263 1264 1265 1266 1267 1268 1269
			/* if there is full-member pkey take it.*/
			if (tmp_pkey & 0x8000) {
				*index = i;
				return 0;
			}
			if (partial_ix < 0)
				partial_ix = i;
1270 1271 1272
		}
	}

1273 1274 1275 1276 1277
	/*no full-member, if exists take the limited*/
	if (partial_ix >= 0) {
		*index = partial_ix;
		return 0;
	}
1278 1279 1280 1281
	return -ENOENT;
}
EXPORT_SYMBOL(ib_find_pkey);

1282 1283 1284 1285 1286 1287 1288 1289 1290
/**
 * ib_get_net_dev_by_params() - Return the appropriate net_dev
 * for a received CM request
 * @dev:	An RDMA device on which the request has been received.
 * @port:	Port number on the RDMA device.
 * @pkey:	The Pkey the request came on.
 * @gid:	A GID that the net_dev uses to communicate.
 * @addr:	Contains the IP address that the request specified as its
 *		destination.
1291
 *
1292 1293 1294 1295 1296 1297 1298 1299
 */
struct net_device *ib_get_net_dev_by_params(struct ib_device *dev,
					    u8 port,
					    u16 pkey,
					    const union ib_gid *gid,
					    const struct sockaddr *addr)
{
	struct net_device *net_dev = NULL;
1300 1301
	unsigned long index;
	void *client_data;
1302 1303 1304 1305

	if (!rdma_protocol_ib(dev, port))
		return NULL;

1306 1307 1308 1309 1310
	/*
	 * Holding the read side guarantees that the client will not become
	 * unregistered while we are calling get_net_dev_by_params()
	 */
	down_read(&dev->client_data_rwsem);
1311 1312 1313
	xan_for_each_marked (&dev->client_data, index, client_data,
			     CLIENT_DATA_REGISTERED) {
		struct ib_client *client = xa_load(&clients, index);
1314

1315
		if (!client || !client->get_net_dev_by_params)
1316 1317
			continue;

1318 1319 1320 1321
		net_dev = client->get_net_dev_by_params(dev, port, pkey, gid,
							addr, client_data);
		if (net_dev)
			break;
1322
	}
1323
	up_read(&dev->client_data_rwsem);
1324 1325 1326 1327 1328

	return net_dev;
}
EXPORT_SYMBOL(ib_get_net_dev_by_params);

K
Kamal Heib 已提交
1329 1330
void ib_set_device_ops(struct ib_device *dev, const struct ib_device_ops *ops)
{
K
Kamal Heib 已提交
1331
	struct ib_device_ops *dev_ops = &dev->ops;
K
Kamal Heib 已提交
1332 1333 1334 1335 1336 1337 1338
#define SET_DEVICE_OP(ptr, name)                                               \
	do {                                                                   \
		if (ops->name)                                                 \
			if (!((ptr)->name))				       \
				(ptr)->name = ops->name;                       \
	} while (0)

1339 1340
#define SET_OBJ_SIZE(ptr, name) SET_DEVICE_OP(ptr, size_##name)

K
Kamal Heib 已提交
1341
	SET_DEVICE_OP(dev_ops, add_gid);
1342
	SET_DEVICE_OP(dev_ops, advise_mr);
K
Kamal Heib 已提交
1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383
	SET_DEVICE_OP(dev_ops, alloc_dm);
	SET_DEVICE_OP(dev_ops, alloc_fmr);
	SET_DEVICE_OP(dev_ops, alloc_hw_stats);
	SET_DEVICE_OP(dev_ops, alloc_mr);
	SET_DEVICE_OP(dev_ops, alloc_mw);
	SET_DEVICE_OP(dev_ops, alloc_pd);
	SET_DEVICE_OP(dev_ops, alloc_rdma_netdev);
	SET_DEVICE_OP(dev_ops, alloc_ucontext);
	SET_DEVICE_OP(dev_ops, alloc_xrcd);
	SET_DEVICE_OP(dev_ops, attach_mcast);
	SET_DEVICE_OP(dev_ops, check_mr_status);
	SET_DEVICE_OP(dev_ops, create_ah);
	SET_DEVICE_OP(dev_ops, create_counters);
	SET_DEVICE_OP(dev_ops, create_cq);
	SET_DEVICE_OP(dev_ops, create_flow);
	SET_DEVICE_OP(dev_ops, create_flow_action_esp);
	SET_DEVICE_OP(dev_ops, create_qp);
	SET_DEVICE_OP(dev_ops, create_rwq_ind_table);
	SET_DEVICE_OP(dev_ops, create_srq);
	SET_DEVICE_OP(dev_ops, create_wq);
	SET_DEVICE_OP(dev_ops, dealloc_dm);
	SET_DEVICE_OP(dev_ops, dealloc_fmr);
	SET_DEVICE_OP(dev_ops, dealloc_mw);
	SET_DEVICE_OP(dev_ops, dealloc_pd);
	SET_DEVICE_OP(dev_ops, dealloc_ucontext);
	SET_DEVICE_OP(dev_ops, dealloc_xrcd);
	SET_DEVICE_OP(dev_ops, del_gid);
	SET_DEVICE_OP(dev_ops, dereg_mr);
	SET_DEVICE_OP(dev_ops, destroy_ah);
	SET_DEVICE_OP(dev_ops, destroy_counters);
	SET_DEVICE_OP(dev_ops, destroy_cq);
	SET_DEVICE_OP(dev_ops, destroy_flow);
	SET_DEVICE_OP(dev_ops, destroy_flow_action);
	SET_DEVICE_OP(dev_ops, destroy_qp);
	SET_DEVICE_OP(dev_ops, destroy_rwq_ind_table);
	SET_DEVICE_OP(dev_ops, destroy_srq);
	SET_DEVICE_OP(dev_ops, destroy_wq);
	SET_DEVICE_OP(dev_ops, detach_mcast);
	SET_DEVICE_OP(dev_ops, disassociate_ucontext);
	SET_DEVICE_OP(dev_ops, drain_rq);
	SET_DEVICE_OP(dev_ops, drain_sq);
1384
	SET_DEVICE_OP(dev_ops, fill_res_entry);
K
Kamal Heib 已提交
1385 1386 1387 1388 1389 1390 1391 1392 1393
	SET_DEVICE_OP(dev_ops, get_dev_fw_str);
	SET_DEVICE_OP(dev_ops, get_dma_mr);
	SET_DEVICE_OP(dev_ops, get_hw_stats);
	SET_DEVICE_OP(dev_ops, get_link_layer);
	SET_DEVICE_OP(dev_ops, get_netdev);
	SET_DEVICE_OP(dev_ops, get_port_immutable);
	SET_DEVICE_OP(dev_ops, get_vector_affinity);
	SET_DEVICE_OP(dev_ops, get_vf_config);
	SET_DEVICE_OP(dev_ops, get_vf_stats);
1394
	SET_DEVICE_OP(dev_ops, init_port);
K
Kamal Heib 已提交
1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429
	SET_DEVICE_OP(dev_ops, map_mr_sg);
	SET_DEVICE_OP(dev_ops, map_phys_fmr);
	SET_DEVICE_OP(dev_ops, mmap);
	SET_DEVICE_OP(dev_ops, modify_ah);
	SET_DEVICE_OP(dev_ops, modify_cq);
	SET_DEVICE_OP(dev_ops, modify_device);
	SET_DEVICE_OP(dev_ops, modify_flow_action_esp);
	SET_DEVICE_OP(dev_ops, modify_port);
	SET_DEVICE_OP(dev_ops, modify_qp);
	SET_DEVICE_OP(dev_ops, modify_srq);
	SET_DEVICE_OP(dev_ops, modify_wq);
	SET_DEVICE_OP(dev_ops, peek_cq);
	SET_DEVICE_OP(dev_ops, poll_cq);
	SET_DEVICE_OP(dev_ops, post_recv);
	SET_DEVICE_OP(dev_ops, post_send);
	SET_DEVICE_OP(dev_ops, post_srq_recv);
	SET_DEVICE_OP(dev_ops, process_mad);
	SET_DEVICE_OP(dev_ops, query_ah);
	SET_DEVICE_OP(dev_ops, query_device);
	SET_DEVICE_OP(dev_ops, query_gid);
	SET_DEVICE_OP(dev_ops, query_pkey);
	SET_DEVICE_OP(dev_ops, query_port);
	SET_DEVICE_OP(dev_ops, query_qp);
	SET_DEVICE_OP(dev_ops, query_srq);
	SET_DEVICE_OP(dev_ops, rdma_netdev_get_params);
	SET_DEVICE_OP(dev_ops, read_counters);
	SET_DEVICE_OP(dev_ops, reg_dm_mr);
	SET_DEVICE_OP(dev_ops, reg_user_mr);
	SET_DEVICE_OP(dev_ops, req_ncomp_notif);
	SET_DEVICE_OP(dev_ops, req_notify_cq);
	SET_DEVICE_OP(dev_ops, rereg_user_mr);
	SET_DEVICE_OP(dev_ops, resize_cq);
	SET_DEVICE_OP(dev_ops, set_vf_guid);
	SET_DEVICE_OP(dev_ops, set_vf_link_state);
	SET_DEVICE_OP(dev_ops, unmap_fmr);
1430 1431

	SET_OBJ_SIZE(dev_ops, ib_pd);
K
Kamal Heib 已提交
1432 1433 1434
}
EXPORT_SYMBOL(ib_set_device_ops);

1435
static const struct rdma_nl_cbs ibnl_ls_cb_table[RDMA_NL_LS_NUM_OPS] = {
1436
	[RDMA_NL_LS_OP_RESOLVE] = {
1437
		.doit = ib_nl_handle_resolve_resp,
1438 1439
		.flags = RDMA_NL_ADMIN_PERM,
	},
1440
	[RDMA_NL_LS_OP_SET_TIMEOUT] = {
1441
		.doit = ib_nl_handle_set_timeout,
1442 1443
		.flags = RDMA_NL_ADMIN_PERM,
	},
1444
	[RDMA_NL_LS_OP_IP_RESOLVE] = {
1445
		.doit = ib_nl_handle_ip_res_resp,
1446 1447
		.flags = RDMA_NL_ADMIN_PERM,
	},
1448 1449
};

L
Linus Torvalds 已提交
1450 1451 1452 1453
static int __init ib_core_init(void)
{
	int ret;

T
Tejun Heo 已提交
1454 1455 1456 1457
	ib_wq = alloc_workqueue("infiniband", 0, 0);
	if (!ib_wq)
		return -ENOMEM;

1458
	ib_comp_wq = alloc_workqueue("ib-comp-wq",
1459
			WQ_HIGHPRI | WQ_MEM_RECLAIM | WQ_SYSFS, 0);
1460 1461 1462 1463 1464
	if (!ib_comp_wq) {
		ret = -ENOMEM;
		goto err;
	}

1465 1466 1467 1468 1469 1470 1471 1472 1473
	ib_comp_unbound_wq =
		alloc_workqueue("ib-comp-unb-wq",
				WQ_UNBOUND | WQ_HIGHPRI | WQ_MEM_RECLAIM |
				WQ_SYSFS, WQ_UNBOUND_MAX_ACTIVE);
	if (!ib_comp_unbound_wq) {
		ret = -ENOMEM;
		goto err_comp;
	}

1474
	ret = class_register(&ib_class);
1475
	if (ret) {
P
Parav Pandit 已提交
1476
		pr_warn("Couldn't create InfiniBand device class\n");
1477
		goto err_comp_unbound;
1478
	}
L
Linus Torvalds 已提交
1479

1480
	ret = rdma_nl_init();
R
Roland Dreier 已提交
1481
	if (ret) {
1482
		pr_warn("Couldn't init IB netlink interface: err %d\n", ret);
R
Roland Dreier 已提交
1483 1484 1485
		goto err_sysfs;
	}

1486 1487 1488 1489 1490 1491
	ret = addr_init();
	if (ret) {
		pr_warn("Could't init IB address resolution\n");
		goto err_ibnl;
	}

1492 1493 1494 1495 1496 1497
	ret = ib_mad_init();
	if (ret) {
		pr_warn("Couldn't init IB MAD\n");
		goto err_addr;
	}

1498 1499 1500 1501 1502 1503
	ret = ib_sa_init();
	if (ret) {
		pr_warn("Couldn't init SA\n");
		goto err_mad;
	}

1504 1505 1506
	ret = register_lsm_notifier(&ibdev_lsm_nb);
	if (ret) {
		pr_warn("Couldn't register LSM notifier. ret %d\n", ret);
1507
		goto err_sa;
1508 1509
	}

1510
	nldev_init();
1511
	rdma_nl_register(RDMA_NL_LS, ibnl_ls_cb_table);
1512
	roce_gid_mgmt_init();
L
Linus Torvalds 已提交
1513

1514 1515
	return 0;

1516 1517
err_sa:
	ib_sa_cleanup();
1518 1519
err_mad:
	ib_mad_cleanup();
1520 1521
err_addr:
	addr_cleanup();
1522
err_ibnl:
1523
	rdma_nl_exit();
1524
err_sysfs:
1525
	class_unregister(&ib_class);
1526 1527
err_comp_unbound:
	destroy_workqueue(ib_comp_unbound_wq);
1528 1529
err_comp:
	destroy_workqueue(ib_comp_wq);
1530 1531
err:
	destroy_workqueue(ib_wq);
L
Linus Torvalds 已提交
1532 1533 1534 1535 1536
	return ret;
}

static void __exit ib_core_cleanup(void)
{
1537
	roce_gid_mgmt_cleanup();
1538
	nldev_exit();
1539 1540
	rdma_nl_unregister(RDMA_NL_LS);
	unregister_lsm_notifier(&ibdev_lsm_nb);
1541
	ib_sa_cleanup();
1542
	ib_mad_cleanup();
1543
	addr_cleanup();
1544
	rdma_nl_exit();
1545
	class_unregister(&ib_class);
1546
	destroy_workqueue(ib_comp_unbound_wq);
1547
	destroy_workqueue(ib_comp_wq);
1548
	/* Make sure that any pending umem accounting work is done. */
T
Tejun Heo 已提交
1549
	destroy_workqueue(ib_wq);
1550
	WARN_ON(!xa_empty(&clients));
1551
	WARN_ON(!xa_empty(&devices));
L
Linus Torvalds 已提交
1552 1553
}

1554 1555
MODULE_ALIAS_RDMA_NETLINK(RDMA_NL_LS, 4);

1556
subsys_initcall(ib_core_init);
L
Linus Torvalds 已提交
1557
module_exit(ib_core_cleanup);