Linux Audio

Check our new training course

Loading...
v3.1
   1/*
   2 * Copyright (c) 2005 Topspin Communications.  All rights reserved.
   3 * Copyright (c) 2005 Intel Corporation.  All rights reserved.
   4 *
   5 * This software is available to you under a choice of one of two
   6 * licenses.  You may choose to be licensed under the terms of the GNU
   7 * General Public License (GPL) Version 2, available from the file
   8 * COPYING in the main directory of this source tree, or the
   9 * OpenIB.org BSD license below:
  10 *
  11 *     Redistribution and use in source and binary forms, with or
  12 *     without modification, are permitted provided that the following
  13 *     conditions are met:
  14 *
  15 *      - Redistributions of source code must retain the above
  16 *	copyright notice, this list of conditions and the following
  17 *	disclaimer.
  18 *
  19 *      - Redistributions in binary form must reproduce the above
  20 *	copyright notice, this list of conditions and the following
  21 *	disclaimer in the documentation and/or other materials
  22 *	provided with the distribution.
  23 *
  24 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  25 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  26 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  27 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  28 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  29 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  30 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  31 * SOFTWARE.
  32 */
  33
  34#include <linux/completion.h>
  35#include <linux/init.h>
  36#include <linux/fs.h>
  37#include <linux/module.h>
  38#include <linux/device.h>
  39#include <linux/err.h>
  40#include <linux/poll.h>
  41#include <linux/sched.h>
  42#include <linux/file.h>
  43#include <linux/mount.h>
  44#include <linux/cdev.h>
  45#include <linux/idr.h>
  46#include <linux/mutex.h>
  47#include <linux/slab.h>
  48
  49#include <asm/uaccess.h>
  50
 
  51#include <rdma/ib_cm.h>
  52#include <rdma/ib_user_cm.h>
  53#include <rdma/ib_marshall.h>
  54
  55MODULE_AUTHOR("Libor Michalek");
  56MODULE_DESCRIPTION("InfiniBand userspace Connection Manager access");
  57MODULE_LICENSE("Dual BSD/GPL");
  58
  59struct ib_ucm_device {
  60	int			devnum;
  61	struct cdev		cdev;
  62	struct device		dev;
  63	struct ib_device	*ib_dev;
  64};
  65
  66struct ib_ucm_file {
  67	struct mutex file_mutex;
  68	struct file *filp;
  69	struct ib_ucm_device *device;
  70
  71	struct list_head  ctxs;
  72	struct list_head  events;
  73	wait_queue_head_t poll_wait;
  74};
  75
  76struct ib_ucm_context {
  77	int                 id;
  78	struct completion   comp;
  79	atomic_t            ref;
  80	int		    events_reported;
  81
  82	struct ib_ucm_file *file;
  83	struct ib_cm_id    *cm_id;
  84	__u64		   uid;
  85
  86	struct list_head    events;    /* list of pending events. */
  87	struct list_head    file_list; /* member in file ctx list */
  88};
  89
  90struct ib_ucm_event {
  91	struct ib_ucm_context *ctx;
  92	struct list_head file_list; /* member in file event list */
  93	struct list_head ctx_list;  /* member in ctx event list */
  94
  95	struct ib_cm_id *cm_id;
  96	struct ib_ucm_event_resp resp;
  97	void *data;
  98	void *info;
  99	int data_len;
 100	int info_len;
 101};
 102
 103enum {
 104	IB_UCM_MAJOR = 231,
 105	IB_UCM_BASE_MINOR = 224,
 106	IB_UCM_MAX_DEVICES = 32
 107};
 108
 109/* ib_cm and ib_user_cm modules share /sys/class/infiniband_cm */
 110extern struct class cm_class;
 111
 112#define IB_UCM_BASE_DEV MKDEV(IB_UCM_MAJOR, IB_UCM_BASE_MINOR)
 113
 114static void ib_ucm_add_one(struct ib_device *device);
 115static void ib_ucm_remove_one(struct ib_device *device);
 116
 117static struct ib_client ucm_client = {
 118	.name   = "ucm",
 119	.add    = ib_ucm_add_one,
 120	.remove = ib_ucm_remove_one
 121};
 122
 123static DEFINE_MUTEX(ctx_id_mutex);
 124static DEFINE_IDR(ctx_id_table);
 125static DECLARE_BITMAP(dev_map, IB_UCM_MAX_DEVICES);
 126
 127static struct ib_ucm_context *ib_ucm_ctx_get(struct ib_ucm_file *file, int id)
 128{
 129	struct ib_ucm_context *ctx;
 130
 131	mutex_lock(&ctx_id_mutex);
 132	ctx = idr_find(&ctx_id_table, id);
 133	if (!ctx)
 134		ctx = ERR_PTR(-ENOENT);
 135	else if (ctx->file != file)
 136		ctx = ERR_PTR(-EINVAL);
 137	else
 138		atomic_inc(&ctx->ref);
 139	mutex_unlock(&ctx_id_mutex);
 140
 141	return ctx;
 142}
 143
 144static void ib_ucm_ctx_put(struct ib_ucm_context *ctx)
 145{
 146	if (atomic_dec_and_test(&ctx->ref))
 147		complete(&ctx->comp);
 148}
 149
 150static inline int ib_ucm_new_cm_id(int event)
 151{
 152	return event == IB_CM_REQ_RECEIVED || event == IB_CM_SIDR_REQ_RECEIVED;
 153}
 154
 155static void ib_ucm_cleanup_events(struct ib_ucm_context *ctx)
 156{
 157	struct ib_ucm_event *uevent;
 158
 159	mutex_lock(&ctx->file->file_mutex);
 160	list_del(&ctx->file_list);
 161	while (!list_empty(&ctx->events)) {
 162
 163		uevent = list_entry(ctx->events.next,
 164				    struct ib_ucm_event, ctx_list);
 165		list_del(&uevent->file_list);
 166		list_del(&uevent->ctx_list);
 167		mutex_unlock(&ctx->file->file_mutex);
 168
 169		/* clear incoming connections. */
 170		if (ib_ucm_new_cm_id(uevent->resp.event))
 171			ib_destroy_cm_id(uevent->cm_id);
 172
 173		kfree(uevent);
 174		mutex_lock(&ctx->file->file_mutex);
 175	}
 176	mutex_unlock(&ctx->file->file_mutex);
 177}
 178
 179static struct ib_ucm_context *ib_ucm_ctx_alloc(struct ib_ucm_file *file)
 180{
 181	struct ib_ucm_context *ctx;
 182	int result;
 183
 184	ctx = kzalloc(sizeof *ctx, GFP_KERNEL);
 185	if (!ctx)
 186		return NULL;
 187
 188	atomic_set(&ctx->ref, 1);
 189	init_completion(&ctx->comp);
 190	ctx->file = file;
 191	INIT_LIST_HEAD(&ctx->events);
 192
 193	do {
 194		result = idr_pre_get(&ctx_id_table, GFP_KERNEL);
 195		if (!result)
 196			goto error;
 197
 198		mutex_lock(&ctx_id_mutex);
 199		result = idr_get_new(&ctx_id_table, ctx, &ctx->id);
 200		mutex_unlock(&ctx_id_mutex);
 201	} while (result == -EAGAIN);
 202
 203	if (result)
 204		goto error;
 205
 206	list_add_tail(&ctx->file_list, &file->ctxs);
 207	return ctx;
 208
 209error:
 210	kfree(ctx);
 211	return NULL;
 212}
 213
 214static void ib_ucm_event_req_get(struct ib_ucm_req_event_resp *ureq,
 215				 struct ib_cm_req_event_param *kreq)
 216{
 217	ureq->remote_ca_guid             = kreq->remote_ca_guid;
 218	ureq->remote_qkey                = kreq->remote_qkey;
 219	ureq->remote_qpn                 = kreq->remote_qpn;
 220	ureq->qp_type                    = kreq->qp_type;
 221	ureq->starting_psn               = kreq->starting_psn;
 222	ureq->responder_resources        = kreq->responder_resources;
 223	ureq->initiator_depth            = kreq->initiator_depth;
 224	ureq->local_cm_response_timeout  = kreq->local_cm_response_timeout;
 225	ureq->flow_control               = kreq->flow_control;
 226	ureq->remote_cm_response_timeout = kreq->remote_cm_response_timeout;
 227	ureq->retry_count                = kreq->retry_count;
 228	ureq->rnr_retry_count            = kreq->rnr_retry_count;
 229	ureq->srq                        = kreq->srq;
 230	ureq->port			 = kreq->port;
 231
 232	ib_copy_path_rec_to_user(&ureq->primary_path, kreq->primary_path);
 233	if (kreq->alternate_path)
 234		ib_copy_path_rec_to_user(&ureq->alternate_path,
 235					 kreq->alternate_path);
 236}
 237
 238static void ib_ucm_event_rep_get(struct ib_ucm_rep_event_resp *urep,
 239				 struct ib_cm_rep_event_param *krep)
 240{
 241	urep->remote_ca_guid      = krep->remote_ca_guid;
 242	urep->remote_qkey         = krep->remote_qkey;
 243	urep->remote_qpn          = krep->remote_qpn;
 244	urep->starting_psn        = krep->starting_psn;
 245	urep->responder_resources = krep->responder_resources;
 246	urep->initiator_depth     = krep->initiator_depth;
 247	urep->target_ack_delay    = krep->target_ack_delay;
 248	urep->failover_accepted   = krep->failover_accepted;
 249	urep->flow_control        = krep->flow_control;
 250	urep->rnr_retry_count     = krep->rnr_retry_count;
 251	urep->srq                 = krep->srq;
 252}
 253
 254static void ib_ucm_event_sidr_rep_get(struct ib_ucm_sidr_rep_event_resp *urep,
 255				      struct ib_cm_sidr_rep_event_param *krep)
 256{
 257	urep->status = krep->status;
 258	urep->qkey   = krep->qkey;
 259	urep->qpn    = krep->qpn;
 260};
 261
 262static int ib_ucm_event_process(struct ib_cm_event *evt,
 263				struct ib_ucm_event *uvt)
 264{
 265	void *info = NULL;
 266
 267	switch (evt->event) {
 268	case IB_CM_REQ_RECEIVED:
 269		ib_ucm_event_req_get(&uvt->resp.u.req_resp,
 270				     &evt->param.req_rcvd);
 271		uvt->data_len      = IB_CM_REQ_PRIVATE_DATA_SIZE;
 272		uvt->resp.present  = IB_UCM_PRES_PRIMARY;
 273		uvt->resp.present |= (evt->param.req_rcvd.alternate_path ?
 274				      IB_UCM_PRES_ALTERNATE : 0);
 275		break;
 276	case IB_CM_REP_RECEIVED:
 277		ib_ucm_event_rep_get(&uvt->resp.u.rep_resp,
 278				     &evt->param.rep_rcvd);
 279		uvt->data_len = IB_CM_REP_PRIVATE_DATA_SIZE;
 280		break;
 281	case IB_CM_RTU_RECEIVED:
 282		uvt->data_len = IB_CM_RTU_PRIVATE_DATA_SIZE;
 283		uvt->resp.u.send_status = evt->param.send_status;
 284		break;
 285	case IB_CM_DREQ_RECEIVED:
 286		uvt->data_len = IB_CM_DREQ_PRIVATE_DATA_SIZE;
 287		uvt->resp.u.send_status = evt->param.send_status;
 288		break;
 289	case IB_CM_DREP_RECEIVED:
 290		uvt->data_len = IB_CM_DREP_PRIVATE_DATA_SIZE;
 291		uvt->resp.u.send_status = evt->param.send_status;
 292		break;
 293	case IB_CM_MRA_RECEIVED:
 294		uvt->resp.u.mra_resp.timeout =
 295					evt->param.mra_rcvd.service_timeout;
 296		uvt->data_len = IB_CM_MRA_PRIVATE_DATA_SIZE;
 297		break;
 298	case IB_CM_REJ_RECEIVED:
 299		uvt->resp.u.rej_resp.reason = evt->param.rej_rcvd.reason;
 300		uvt->data_len = IB_CM_REJ_PRIVATE_DATA_SIZE;
 301		uvt->info_len = evt->param.rej_rcvd.ari_length;
 302		info	      = evt->param.rej_rcvd.ari;
 303		break;
 304	case IB_CM_LAP_RECEIVED:
 305		ib_copy_path_rec_to_user(&uvt->resp.u.lap_resp.path,
 306					 evt->param.lap_rcvd.alternate_path);
 307		uvt->data_len = IB_CM_LAP_PRIVATE_DATA_SIZE;
 308		uvt->resp.present = IB_UCM_PRES_ALTERNATE;
 309		break;
 310	case IB_CM_APR_RECEIVED:
 311		uvt->resp.u.apr_resp.status = evt->param.apr_rcvd.ap_status;
 312		uvt->data_len = IB_CM_APR_PRIVATE_DATA_SIZE;
 313		uvt->info_len = evt->param.apr_rcvd.info_len;
 314		info	      = evt->param.apr_rcvd.apr_info;
 315		break;
 316	case IB_CM_SIDR_REQ_RECEIVED:
 317		uvt->resp.u.sidr_req_resp.pkey =
 318					evt->param.sidr_req_rcvd.pkey;
 319		uvt->resp.u.sidr_req_resp.port =
 320					evt->param.sidr_req_rcvd.port;
 321		uvt->data_len = IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE;
 322		break;
 323	case IB_CM_SIDR_REP_RECEIVED:
 324		ib_ucm_event_sidr_rep_get(&uvt->resp.u.sidr_rep_resp,
 325					  &evt->param.sidr_rep_rcvd);
 326		uvt->data_len = IB_CM_SIDR_REP_PRIVATE_DATA_SIZE;
 327		uvt->info_len = evt->param.sidr_rep_rcvd.info_len;
 328		info	      = evt->param.sidr_rep_rcvd.info;
 329		break;
 330	default:
 331		uvt->resp.u.send_status = evt->param.send_status;
 332		break;
 333	}
 334
 335	if (uvt->data_len) {
 336		uvt->data = kmemdup(evt->private_data, uvt->data_len, GFP_KERNEL);
 337		if (!uvt->data)
 338			goto err1;
 339
 340		uvt->resp.present |= IB_UCM_PRES_DATA;
 341	}
 342
 343	if (uvt->info_len) {
 344		uvt->info = kmemdup(info, uvt->info_len, GFP_KERNEL);
 345		if (!uvt->info)
 346			goto err2;
 347
 348		uvt->resp.present |= IB_UCM_PRES_INFO;
 349	}
 350	return 0;
 351
 352err2:
 353	kfree(uvt->data);
 354err1:
 355	return -ENOMEM;
 356}
 357
 358static int ib_ucm_event_handler(struct ib_cm_id *cm_id,
 359				struct ib_cm_event *event)
 360{
 361	struct ib_ucm_event *uevent;
 362	struct ib_ucm_context *ctx;
 363	int result = 0;
 364
 365	ctx = cm_id->context;
 366
 367	uevent = kzalloc(sizeof *uevent, GFP_KERNEL);
 368	if (!uevent)
 369		goto err1;
 370
 371	uevent->ctx = ctx;
 372	uevent->cm_id = cm_id;
 373	uevent->resp.uid = ctx->uid;
 374	uevent->resp.id = ctx->id;
 375	uevent->resp.event = event->event;
 376
 377	result = ib_ucm_event_process(event, uevent);
 378	if (result)
 379		goto err2;
 380
 381	mutex_lock(&ctx->file->file_mutex);
 382	list_add_tail(&uevent->file_list, &ctx->file->events);
 383	list_add_tail(&uevent->ctx_list, &ctx->events);
 384	wake_up_interruptible(&ctx->file->poll_wait);
 385	mutex_unlock(&ctx->file->file_mutex);
 386	return 0;
 387
 388err2:
 389	kfree(uevent);
 390err1:
 391	/* Destroy new cm_id's */
 392	return ib_ucm_new_cm_id(event->event);
 393}
 394
 395static ssize_t ib_ucm_event(struct ib_ucm_file *file,
 396			    const char __user *inbuf,
 397			    int in_len, int out_len)
 398{
 399	struct ib_ucm_context *ctx;
 400	struct ib_ucm_event_get cmd;
 401	struct ib_ucm_event *uevent;
 402	int result = 0;
 403	DEFINE_WAIT(wait);
 404
 405	if (out_len < sizeof(struct ib_ucm_event_resp))
 406		return -ENOSPC;
 407
 408	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 409		return -EFAULT;
 410
 411	mutex_lock(&file->file_mutex);
 412	while (list_empty(&file->events)) {
 413		mutex_unlock(&file->file_mutex);
 414
 415		if (file->filp->f_flags & O_NONBLOCK)
 416			return -EAGAIN;
 417
 418		if (wait_event_interruptible(file->poll_wait,
 419					     !list_empty(&file->events)))
 420			return -ERESTARTSYS;
 421
 422		mutex_lock(&file->file_mutex);
 423	}
 424
 425	uevent = list_entry(file->events.next, struct ib_ucm_event, file_list);
 426
 427	if (ib_ucm_new_cm_id(uevent->resp.event)) {
 428		ctx = ib_ucm_ctx_alloc(file);
 429		if (!ctx) {
 430			result = -ENOMEM;
 431			goto done;
 432		}
 433
 434		ctx->cm_id = uevent->cm_id;
 435		ctx->cm_id->context = ctx;
 436		uevent->resp.id = ctx->id;
 437	}
 438
 439	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 440			 &uevent->resp, sizeof(uevent->resp))) {
 441		result = -EFAULT;
 442		goto done;
 443	}
 444
 445	if (uevent->data) {
 446		if (cmd.data_len < uevent->data_len) {
 447			result = -ENOMEM;
 448			goto done;
 449		}
 450		if (copy_to_user((void __user *)(unsigned long)cmd.data,
 451				 uevent->data, uevent->data_len)) {
 452			result = -EFAULT;
 453			goto done;
 454		}
 455	}
 456
 457	if (uevent->info) {
 458		if (cmd.info_len < uevent->info_len) {
 459			result = -ENOMEM;
 460			goto done;
 461		}
 462		if (copy_to_user((void __user *)(unsigned long)cmd.info,
 463				 uevent->info, uevent->info_len)) {
 464			result = -EFAULT;
 465			goto done;
 466		}
 467	}
 468
 469	list_del(&uevent->file_list);
 470	list_del(&uevent->ctx_list);
 471	uevent->ctx->events_reported++;
 472
 473	kfree(uevent->data);
 474	kfree(uevent->info);
 475	kfree(uevent);
 476done:
 477	mutex_unlock(&file->file_mutex);
 478	return result;
 479}
 480
 481static ssize_t ib_ucm_create_id(struct ib_ucm_file *file,
 482				const char __user *inbuf,
 483				int in_len, int out_len)
 484{
 485	struct ib_ucm_create_id cmd;
 486	struct ib_ucm_create_id_resp resp;
 487	struct ib_ucm_context *ctx;
 488	int result;
 489
 490	if (out_len < sizeof(resp))
 491		return -ENOSPC;
 492
 493	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 494		return -EFAULT;
 495
 496	mutex_lock(&file->file_mutex);
 497	ctx = ib_ucm_ctx_alloc(file);
 498	mutex_unlock(&file->file_mutex);
 499	if (!ctx)
 500		return -ENOMEM;
 501
 502	ctx->uid = cmd.uid;
 503	ctx->cm_id = ib_create_cm_id(file->device->ib_dev,
 504				     ib_ucm_event_handler, ctx);
 505	if (IS_ERR(ctx->cm_id)) {
 506		result = PTR_ERR(ctx->cm_id);
 507		goto err1;
 508	}
 509
 510	resp.id = ctx->id;
 511	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 512			 &resp, sizeof(resp))) {
 513		result = -EFAULT;
 514		goto err2;
 515	}
 516	return 0;
 517
 518err2:
 519	ib_destroy_cm_id(ctx->cm_id);
 520err1:
 521	mutex_lock(&ctx_id_mutex);
 522	idr_remove(&ctx_id_table, ctx->id);
 523	mutex_unlock(&ctx_id_mutex);
 524	kfree(ctx);
 525	return result;
 526}
 527
 528static ssize_t ib_ucm_destroy_id(struct ib_ucm_file *file,
 529				 const char __user *inbuf,
 530				 int in_len, int out_len)
 531{
 532	struct ib_ucm_destroy_id cmd;
 533	struct ib_ucm_destroy_id_resp resp;
 534	struct ib_ucm_context *ctx;
 535	int result = 0;
 536
 537	if (out_len < sizeof(resp))
 538		return -ENOSPC;
 539
 540	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 541		return -EFAULT;
 542
 543	mutex_lock(&ctx_id_mutex);
 544	ctx = idr_find(&ctx_id_table, cmd.id);
 545	if (!ctx)
 546		ctx = ERR_PTR(-ENOENT);
 547	else if (ctx->file != file)
 548		ctx = ERR_PTR(-EINVAL);
 549	else
 550		idr_remove(&ctx_id_table, ctx->id);
 551	mutex_unlock(&ctx_id_mutex);
 552
 553	if (IS_ERR(ctx))
 554		return PTR_ERR(ctx);
 555
 556	ib_ucm_ctx_put(ctx);
 557	wait_for_completion(&ctx->comp);
 558
 559	/* No new events will be generated after destroying the cm_id. */
 560	ib_destroy_cm_id(ctx->cm_id);
 561	/* Cleanup events not yet reported to the user. */
 562	ib_ucm_cleanup_events(ctx);
 563
 564	resp.events_reported = ctx->events_reported;
 565	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 566			 &resp, sizeof(resp)))
 567		result = -EFAULT;
 568
 569	kfree(ctx);
 570	return result;
 571}
 572
 573static ssize_t ib_ucm_attr_id(struct ib_ucm_file *file,
 574			      const char __user *inbuf,
 575			      int in_len, int out_len)
 576{
 577	struct ib_ucm_attr_id_resp resp;
 578	struct ib_ucm_attr_id cmd;
 579	struct ib_ucm_context *ctx;
 580	int result = 0;
 581
 582	if (out_len < sizeof(resp))
 583		return -ENOSPC;
 584
 585	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 586		return -EFAULT;
 587
 588	ctx = ib_ucm_ctx_get(file, cmd.id);
 589	if (IS_ERR(ctx))
 590		return PTR_ERR(ctx);
 591
 592	resp.service_id   = ctx->cm_id->service_id;
 593	resp.service_mask = ctx->cm_id->service_mask;
 594	resp.local_id     = ctx->cm_id->local_id;
 595	resp.remote_id    = ctx->cm_id->remote_id;
 596
 597	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 598			 &resp, sizeof(resp)))
 599		result = -EFAULT;
 600
 601	ib_ucm_ctx_put(ctx);
 602	return result;
 603}
 604
 605static ssize_t ib_ucm_init_qp_attr(struct ib_ucm_file *file,
 606				   const char __user *inbuf,
 607				   int in_len, int out_len)
 608{
 609	struct ib_uverbs_qp_attr resp;
 610	struct ib_ucm_init_qp_attr cmd;
 611	struct ib_ucm_context *ctx;
 612	struct ib_qp_attr qp_attr;
 613	int result = 0;
 614
 615	if (out_len < sizeof(resp))
 616		return -ENOSPC;
 617
 618	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 619		return -EFAULT;
 620
 621	ctx = ib_ucm_ctx_get(file, cmd.id);
 622	if (IS_ERR(ctx))
 623		return PTR_ERR(ctx);
 624
 625	resp.qp_attr_mask = 0;
 626	memset(&qp_attr, 0, sizeof qp_attr);
 627	qp_attr.qp_state = cmd.qp_state;
 628	result = ib_cm_init_qp_attr(ctx->cm_id, &qp_attr, &resp.qp_attr_mask);
 629	if (result)
 630		goto out;
 631
 632	ib_copy_qp_attr_to_user(&resp, &qp_attr);
 633
 634	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 635			 &resp, sizeof(resp)))
 636		result = -EFAULT;
 637
 638out:
 639	ib_ucm_ctx_put(ctx);
 640	return result;
 641}
 642
 643static int ucm_validate_listen(__be64 service_id, __be64 service_mask)
 644{
 645	service_id &= service_mask;
 646
 647	if (((service_id & IB_CMA_SERVICE_ID_MASK) == IB_CMA_SERVICE_ID) ||
 648	    ((service_id & IB_SDP_SERVICE_ID_MASK) == IB_SDP_SERVICE_ID))
 649		return -EINVAL;
 650
 651	return 0;
 652}
 653
 654static ssize_t ib_ucm_listen(struct ib_ucm_file *file,
 655			     const char __user *inbuf,
 656			     int in_len, int out_len)
 657{
 658	struct ib_ucm_listen cmd;
 659	struct ib_ucm_context *ctx;
 660	int result;
 661
 662	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 663		return -EFAULT;
 664
 665	ctx = ib_ucm_ctx_get(file, cmd.id);
 666	if (IS_ERR(ctx))
 667		return PTR_ERR(ctx);
 668
 669	result = ucm_validate_listen(cmd.service_id, cmd.service_mask);
 670	if (result)
 671		goto out;
 672
 673	result = ib_cm_listen(ctx->cm_id, cmd.service_id, cmd.service_mask,
 674			      NULL);
 675out:
 676	ib_ucm_ctx_put(ctx);
 677	return result;
 678}
 679
 680static ssize_t ib_ucm_notify(struct ib_ucm_file *file,
 681			     const char __user *inbuf,
 682			     int in_len, int out_len)
 683{
 684	struct ib_ucm_notify cmd;
 685	struct ib_ucm_context *ctx;
 686	int result;
 687
 688	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 689		return -EFAULT;
 690
 691	ctx = ib_ucm_ctx_get(file, cmd.id);
 692	if (IS_ERR(ctx))
 693		return PTR_ERR(ctx);
 694
 695	result = ib_cm_notify(ctx->cm_id, (enum ib_event_type) cmd.event);
 696	ib_ucm_ctx_put(ctx);
 697	return result;
 698}
 699
 700static int ib_ucm_alloc_data(const void **dest, u64 src, u32 len)
 701{
 702	void *data;
 703
 704	*dest = NULL;
 705
 706	if (!len)
 707		return 0;
 708
 709	data = memdup_user((void __user *)(unsigned long)src, len);
 710	if (IS_ERR(data))
 711		return PTR_ERR(data);
 712
 713	*dest = data;
 714	return 0;
 715}
 716
 717static int ib_ucm_path_get(struct ib_sa_path_rec **path, u64 src)
 718{
 719	struct ib_user_path_rec upath;
 720	struct ib_sa_path_rec  *sa_path;
 721
 722	*path = NULL;
 723
 724	if (!src)
 725		return 0;
 726
 727	sa_path = kmalloc(sizeof(*sa_path), GFP_KERNEL);
 728	if (!sa_path)
 729		return -ENOMEM;
 730
 731	if (copy_from_user(&upath, (void __user *)(unsigned long)src,
 732			   sizeof(upath))) {
 733
 734		kfree(sa_path);
 735		return -EFAULT;
 736	}
 737
 738	ib_copy_path_rec_from_user(sa_path, &upath);
 739	*path = sa_path;
 740	return 0;
 741}
 742
 743static ssize_t ib_ucm_send_req(struct ib_ucm_file *file,
 744			       const char __user *inbuf,
 745			       int in_len, int out_len)
 746{
 747	struct ib_cm_req_param param;
 748	struct ib_ucm_context *ctx;
 749	struct ib_ucm_req cmd;
 750	int result;
 751
 752	param.private_data   = NULL;
 753	param.primary_path   = NULL;
 754	param.alternate_path = NULL;
 755
 756	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 757		return -EFAULT;
 758
 759	result = ib_ucm_alloc_data(&param.private_data, cmd.data, cmd.len);
 760	if (result)
 761		goto done;
 762
 763	result = ib_ucm_path_get(&param.primary_path, cmd.primary_path);
 764	if (result)
 765		goto done;
 766
 767	result = ib_ucm_path_get(&param.alternate_path, cmd.alternate_path);
 768	if (result)
 769		goto done;
 770
 771	param.private_data_len           = cmd.len;
 772	param.service_id                 = cmd.sid;
 773	param.qp_num                     = cmd.qpn;
 774	param.qp_type                    = cmd.qp_type;
 775	param.starting_psn               = cmd.psn;
 776	param.peer_to_peer               = cmd.peer_to_peer;
 777	param.responder_resources        = cmd.responder_resources;
 778	param.initiator_depth            = cmd.initiator_depth;
 779	param.remote_cm_response_timeout = cmd.remote_cm_response_timeout;
 780	param.flow_control               = cmd.flow_control;
 781	param.local_cm_response_timeout  = cmd.local_cm_response_timeout;
 782	param.retry_count                = cmd.retry_count;
 783	param.rnr_retry_count            = cmd.rnr_retry_count;
 784	param.max_cm_retries             = cmd.max_cm_retries;
 785	param.srq                        = cmd.srq;
 786
 787	ctx = ib_ucm_ctx_get(file, cmd.id);
 788	if (!IS_ERR(ctx)) {
 789		result = ib_send_cm_req(ctx->cm_id, &param);
 790		ib_ucm_ctx_put(ctx);
 791	} else
 792		result = PTR_ERR(ctx);
 793
 794done:
 795	kfree(param.private_data);
 796	kfree(param.primary_path);
 797	kfree(param.alternate_path);
 798	return result;
 799}
 800
 801static ssize_t ib_ucm_send_rep(struct ib_ucm_file *file,
 802			       const char __user *inbuf,
 803			       int in_len, int out_len)
 804{
 805	struct ib_cm_rep_param param;
 806	struct ib_ucm_context *ctx;
 807	struct ib_ucm_rep cmd;
 808	int result;
 809
 810	param.private_data = NULL;
 811
 812	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 813		return -EFAULT;
 814
 815	result = ib_ucm_alloc_data(&param.private_data, cmd.data, cmd.len);
 816	if (result)
 817		return result;
 818
 819	param.qp_num              = cmd.qpn;
 820	param.starting_psn        = cmd.psn;
 821	param.private_data_len    = cmd.len;
 822	param.responder_resources = cmd.responder_resources;
 823	param.initiator_depth     = cmd.initiator_depth;
 824	param.failover_accepted   = cmd.failover_accepted;
 825	param.flow_control        = cmd.flow_control;
 826	param.rnr_retry_count     = cmd.rnr_retry_count;
 827	param.srq                 = cmd.srq;
 828
 829	ctx = ib_ucm_ctx_get(file, cmd.id);
 830	if (!IS_ERR(ctx)) {
 831		ctx->uid = cmd.uid;
 832		result = ib_send_cm_rep(ctx->cm_id, &param);
 833		ib_ucm_ctx_put(ctx);
 834	} else
 835		result = PTR_ERR(ctx);
 836
 837	kfree(param.private_data);
 838	return result;
 839}
 840
 841static ssize_t ib_ucm_send_private_data(struct ib_ucm_file *file,
 842					const char __user *inbuf, int in_len,
 843					int (*func)(struct ib_cm_id *cm_id,
 844						    const void *private_data,
 845						    u8 private_data_len))
 846{
 847	struct ib_ucm_private_data cmd;
 848	struct ib_ucm_context *ctx;
 849	const void *private_data = NULL;
 850	int result;
 851
 852	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 853		return -EFAULT;
 854
 855	result = ib_ucm_alloc_data(&private_data, cmd.data, cmd.len);
 856	if (result)
 857		return result;
 858
 859	ctx = ib_ucm_ctx_get(file, cmd.id);
 860	if (!IS_ERR(ctx)) {
 861		result = func(ctx->cm_id, private_data, cmd.len);
 862		ib_ucm_ctx_put(ctx);
 863	} else
 864		result = PTR_ERR(ctx);
 865
 866	kfree(private_data);
 867	return result;
 868}
 869
 870static ssize_t ib_ucm_send_rtu(struct ib_ucm_file *file,
 871			       const char __user *inbuf,
 872			       int in_len, int out_len)
 873{
 874	return ib_ucm_send_private_data(file, inbuf, in_len, ib_send_cm_rtu);
 875}
 876
 877static ssize_t ib_ucm_send_dreq(struct ib_ucm_file *file,
 878				const char __user *inbuf,
 879				int in_len, int out_len)
 880{
 881	return ib_ucm_send_private_data(file, inbuf, in_len, ib_send_cm_dreq);
 882}
 883
 884static ssize_t ib_ucm_send_drep(struct ib_ucm_file *file,
 885				const char __user *inbuf,
 886				int in_len, int out_len)
 887{
 888	return ib_ucm_send_private_data(file, inbuf, in_len, ib_send_cm_drep);
 889}
 890
 891static ssize_t ib_ucm_send_info(struct ib_ucm_file *file,
 892				const char __user *inbuf, int in_len,
 893				int (*func)(struct ib_cm_id *cm_id,
 894					    int status,
 895					    const void *info,
 896					    u8 info_len,
 897					    const void *data,
 898					    u8 data_len))
 899{
 900	struct ib_ucm_context *ctx;
 901	struct ib_ucm_info cmd;
 902	const void *data = NULL;
 903	const void *info = NULL;
 904	int result;
 905
 906	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 907		return -EFAULT;
 908
 909	result = ib_ucm_alloc_data(&data, cmd.data, cmd.data_len);
 910	if (result)
 911		goto done;
 912
 913	result = ib_ucm_alloc_data(&info, cmd.info, cmd.info_len);
 914	if (result)
 915		goto done;
 916
 917	ctx = ib_ucm_ctx_get(file, cmd.id);
 918	if (!IS_ERR(ctx)) {
 919		result = func(ctx->cm_id, cmd.status, info, cmd.info_len,
 920			      data, cmd.data_len);
 921		ib_ucm_ctx_put(ctx);
 922	} else
 923		result = PTR_ERR(ctx);
 924
 925done:
 926	kfree(data);
 927	kfree(info);
 928	return result;
 929}
 930
 931static ssize_t ib_ucm_send_rej(struct ib_ucm_file *file,
 932			       const char __user *inbuf,
 933			       int in_len, int out_len)
 934{
 935	return ib_ucm_send_info(file, inbuf, in_len, (void *)ib_send_cm_rej);
 936}
 937
 938static ssize_t ib_ucm_send_apr(struct ib_ucm_file *file,
 939			       const char __user *inbuf,
 940			       int in_len, int out_len)
 941{
 942	return ib_ucm_send_info(file, inbuf, in_len, (void *)ib_send_cm_apr);
 943}
 944
 945static ssize_t ib_ucm_send_mra(struct ib_ucm_file *file,
 946			       const char __user *inbuf,
 947			       int in_len, int out_len)
 948{
 949	struct ib_ucm_context *ctx;
 950	struct ib_ucm_mra cmd;
 951	const void *data = NULL;
 952	int result;
 953
 954	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 955		return -EFAULT;
 956
 957	result = ib_ucm_alloc_data(&data, cmd.data, cmd.len);
 958	if (result)
 959		return result;
 960
 961	ctx = ib_ucm_ctx_get(file, cmd.id);
 962	if (!IS_ERR(ctx)) {
 963		result = ib_send_cm_mra(ctx->cm_id, cmd.timeout, data, cmd.len);
 964		ib_ucm_ctx_put(ctx);
 965	} else
 966		result = PTR_ERR(ctx);
 967
 968	kfree(data);
 969	return result;
 970}
 971
 972static ssize_t ib_ucm_send_lap(struct ib_ucm_file *file,
 973			       const char __user *inbuf,
 974			       int in_len, int out_len)
 975{
 976	struct ib_ucm_context *ctx;
 977	struct ib_sa_path_rec *path = NULL;
 978	struct ib_ucm_lap cmd;
 979	const void *data = NULL;
 980	int result;
 981
 982	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 983		return -EFAULT;
 984
 985	result = ib_ucm_alloc_data(&data, cmd.data, cmd.len);
 986	if (result)
 987		goto done;
 988
 989	result = ib_ucm_path_get(&path, cmd.path);
 990	if (result)
 991		goto done;
 992
 993	ctx = ib_ucm_ctx_get(file, cmd.id);
 994	if (!IS_ERR(ctx)) {
 995		result = ib_send_cm_lap(ctx->cm_id, path, data, cmd.len);
 996		ib_ucm_ctx_put(ctx);
 997	} else
 998		result = PTR_ERR(ctx);
 999
1000done:
1001	kfree(data);
1002	kfree(path);
1003	return result;
1004}
1005
1006static ssize_t ib_ucm_send_sidr_req(struct ib_ucm_file *file,
1007				    const char __user *inbuf,
1008				    int in_len, int out_len)
1009{
1010	struct ib_cm_sidr_req_param param;
1011	struct ib_ucm_context *ctx;
1012	struct ib_ucm_sidr_req cmd;
1013	int result;
1014
1015	param.private_data = NULL;
1016	param.path = NULL;
1017
1018	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
1019		return -EFAULT;
1020
1021	result = ib_ucm_alloc_data(&param.private_data, cmd.data, cmd.len);
1022	if (result)
1023		goto done;
1024
1025	result = ib_ucm_path_get(&param.path, cmd.path);
1026	if (result)
1027		goto done;
1028
1029	param.private_data_len = cmd.len;
1030	param.service_id       = cmd.sid;
1031	param.timeout_ms       = cmd.timeout;
1032	param.max_cm_retries   = cmd.max_cm_retries;
1033
1034	ctx = ib_ucm_ctx_get(file, cmd.id);
1035	if (!IS_ERR(ctx)) {
1036		result = ib_send_cm_sidr_req(ctx->cm_id, &param);
1037		ib_ucm_ctx_put(ctx);
1038	} else
1039		result = PTR_ERR(ctx);
1040
1041done:
1042	kfree(param.private_data);
1043	kfree(param.path);
1044	return result;
1045}
1046
1047static ssize_t ib_ucm_send_sidr_rep(struct ib_ucm_file *file,
1048				    const char __user *inbuf,
1049				    int in_len, int out_len)
1050{
1051	struct ib_cm_sidr_rep_param param;
1052	struct ib_ucm_sidr_rep cmd;
1053	struct ib_ucm_context *ctx;
1054	int result;
1055
1056	param.info = NULL;
1057
1058	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
1059		return -EFAULT;
1060
1061	result = ib_ucm_alloc_data(&param.private_data,
1062				   cmd.data, cmd.data_len);
1063	if (result)
1064		goto done;
1065
1066	result = ib_ucm_alloc_data(&param.info, cmd.info, cmd.info_len);
1067	if (result)
1068		goto done;
1069
1070	param.qp_num		= cmd.qpn;
1071	param.qkey		= cmd.qkey;
1072	param.status		= cmd.status;
1073	param.info_length	= cmd.info_len;
1074	param.private_data_len	= cmd.data_len;
1075
1076	ctx = ib_ucm_ctx_get(file, cmd.id);
1077	if (!IS_ERR(ctx)) {
1078		result = ib_send_cm_sidr_rep(ctx->cm_id, &param);
1079		ib_ucm_ctx_put(ctx);
1080	} else
1081		result = PTR_ERR(ctx);
1082
1083done:
1084	kfree(param.private_data);
1085	kfree(param.info);
1086	return result;
1087}
1088
1089static ssize_t (*ucm_cmd_table[])(struct ib_ucm_file *file,
1090				  const char __user *inbuf,
1091				  int in_len, int out_len) = {
1092	[IB_USER_CM_CMD_CREATE_ID]     = ib_ucm_create_id,
1093	[IB_USER_CM_CMD_DESTROY_ID]    = ib_ucm_destroy_id,
1094	[IB_USER_CM_CMD_ATTR_ID]       = ib_ucm_attr_id,
1095	[IB_USER_CM_CMD_LISTEN]        = ib_ucm_listen,
1096	[IB_USER_CM_CMD_NOTIFY]        = ib_ucm_notify,
1097	[IB_USER_CM_CMD_SEND_REQ]      = ib_ucm_send_req,
1098	[IB_USER_CM_CMD_SEND_REP]      = ib_ucm_send_rep,
1099	[IB_USER_CM_CMD_SEND_RTU]      = ib_ucm_send_rtu,
1100	[IB_USER_CM_CMD_SEND_DREQ]     = ib_ucm_send_dreq,
1101	[IB_USER_CM_CMD_SEND_DREP]     = ib_ucm_send_drep,
1102	[IB_USER_CM_CMD_SEND_REJ]      = ib_ucm_send_rej,
1103	[IB_USER_CM_CMD_SEND_MRA]      = ib_ucm_send_mra,
1104	[IB_USER_CM_CMD_SEND_LAP]      = ib_ucm_send_lap,
1105	[IB_USER_CM_CMD_SEND_APR]      = ib_ucm_send_apr,
1106	[IB_USER_CM_CMD_SEND_SIDR_REQ] = ib_ucm_send_sidr_req,
1107	[IB_USER_CM_CMD_SEND_SIDR_REP] = ib_ucm_send_sidr_rep,
1108	[IB_USER_CM_CMD_EVENT]	       = ib_ucm_event,
1109	[IB_USER_CM_CMD_INIT_QP_ATTR]  = ib_ucm_init_qp_attr,
1110};
1111
1112static ssize_t ib_ucm_write(struct file *filp, const char __user *buf,
1113			    size_t len, loff_t *pos)
1114{
1115	struct ib_ucm_file *file = filp->private_data;
1116	struct ib_ucm_cmd_hdr hdr;
1117	ssize_t result;
1118
 
 
 
1119	if (len < sizeof(hdr))
1120		return -EINVAL;
1121
1122	if (copy_from_user(&hdr, buf, sizeof(hdr)))
1123		return -EFAULT;
1124
1125	if (hdr.cmd < 0 || hdr.cmd >= ARRAY_SIZE(ucm_cmd_table))
1126		return -EINVAL;
1127
1128	if (hdr.in + sizeof(hdr) > len)
1129		return -EINVAL;
1130
1131	result = ucm_cmd_table[hdr.cmd](file, buf + sizeof(hdr),
1132					hdr.in, hdr.out);
1133	if (!result)
1134		result = len;
1135
1136	return result;
1137}
1138
1139static unsigned int ib_ucm_poll(struct file *filp,
1140				struct poll_table_struct *wait)
1141{
1142	struct ib_ucm_file *file = filp->private_data;
1143	unsigned int mask = 0;
1144
1145	poll_wait(filp, &file->poll_wait, wait);
1146
1147	if (!list_empty(&file->events))
1148		mask = POLLIN | POLLRDNORM;
1149
1150	return mask;
1151}
1152
1153/*
1154 * ib_ucm_open() does not need the BKL:
1155 *
1156 *  - no global state is referred to;
1157 *  - there is no ioctl method to race against;
1158 *  - no further module initialization is required for open to work
1159 *    after the device is registered.
1160 */
1161static int ib_ucm_open(struct inode *inode, struct file *filp)
1162{
1163	struct ib_ucm_file *file;
1164
1165	file = kmalloc(sizeof(*file), GFP_KERNEL);
1166	if (!file)
1167		return -ENOMEM;
1168
1169	INIT_LIST_HEAD(&file->events);
1170	INIT_LIST_HEAD(&file->ctxs);
1171	init_waitqueue_head(&file->poll_wait);
1172
1173	mutex_init(&file->file_mutex);
1174
1175	filp->private_data = file;
1176	file->filp = filp;
1177	file->device = container_of(inode->i_cdev, struct ib_ucm_device, cdev);
1178
1179	return nonseekable_open(inode, filp);
1180}
1181
1182static int ib_ucm_close(struct inode *inode, struct file *filp)
1183{
1184	struct ib_ucm_file *file = filp->private_data;
1185	struct ib_ucm_context *ctx;
1186
1187	mutex_lock(&file->file_mutex);
1188	while (!list_empty(&file->ctxs)) {
1189		ctx = list_entry(file->ctxs.next,
1190				 struct ib_ucm_context, file_list);
1191		mutex_unlock(&file->file_mutex);
1192
1193		mutex_lock(&ctx_id_mutex);
1194		idr_remove(&ctx_id_table, ctx->id);
1195		mutex_unlock(&ctx_id_mutex);
1196
1197		ib_destroy_cm_id(ctx->cm_id);
1198		ib_ucm_cleanup_events(ctx);
1199		kfree(ctx);
1200
1201		mutex_lock(&file->file_mutex);
1202	}
1203	mutex_unlock(&file->file_mutex);
1204	kfree(file);
1205	return 0;
1206}
1207
 
1208static void ib_ucm_release_dev(struct device *dev)
1209{
1210	struct ib_ucm_device *ucm_dev;
1211
1212	ucm_dev = container_of(dev, struct ib_ucm_device, dev);
1213	cdev_del(&ucm_dev->cdev);
1214	if (ucm_dev->devnum < IB_UCM_MAX_DEVICES)
1215		clear_bit(ucm_dev->devnum, dev_map);
1216	else
1217		clear_bit(ucm_dev->devnum - IB_UCM_MAX_DEVICES, dev_map);
1218	kfree(ucm_dev);
1219}
1220
1221static const struct file_operations ucm_fops = {
1222	.owner	 = THIS_MODULE,
1223	.open	 = ib_ucm_open,
1224	.release = ib_ucm_close,
1225	.write	 = ib_ucm_write,
1226	.poll    = ib_ucm_poll,
1227	.llseek	 = no_llseek,
1228};
1229
1230static ssize_t show_ibdev(struct device *dev, struct device_attribute *attr,
1231			  char *buf)
1232{
1233	struct ib_ucm_device *ucm_dev;
1234
1235	ucm_dev = container_of(dev, struct ib_ucm_device, dev);
1236	return sprintf(buf, "%s\n", ucm_dev->ib_dev->name);
1237}
1238static DEVICE_ATTR(ibdev, S_IRUGO, show_ibdev, NULL);
1239
1240static dev_t overflow_maj;
1241static DECLARE_BITMAP(overflow_map, IB_UCM_MAX_DEVICES);
1242static int find_overflow_devnum(void)
1243{
1244	int ret;
1245
1246	if (!overflow_maj) {
1247		ret = alloc_chrdev_region(&overflow_maj, 0, IB_UCM_MAX_DEVICES,
1248					  "infiniband_cm");
1249		if (ret) {
1250			printk(KERN_ERR "ucm: couldn't register dynamic device number\n");
1251			return ret;
1252		}
1253	}
1254
1255	ret = find_first_zero_bit(overflow_map, IB_UCM_MAX_DEVICES);
1256	if (ret >= IB_UCM_MAX_DEVICES)
1257		return -1;
1258
1259	return ret;
1260}
1261
1262static void ib_ucm_add_one(struct ib_device *device)
1263{
1264	int devnum;
1265	dev_t base;
1266	struct ib_ucm_device *ucm_dev;
1267
1268	if (!device->alloc_ucontext ||
1269	    rdma_node_get_transport(device->node_type) != RDMA_TRANSPORT_IB)
1270		return;
1271
1272	ucm_dev = kzalloc(sizeof *ucm_dev, GFP_KERNEL);
1273	if (!ucm_dev)
1274		return;
1275
1276	ucm_dev->ib_dev = device;
1277
1278	devnum = find_first_zero_bit(dev_map, IB_UCM_MAX_DEVICES);
1279	if (devnum >= IB_UCM_MAX_DEVICES) {
1280		devnum = find_overflow_devnum();
1281		if (devnum < 0)
1282			goto err;
1283
1284		ucm_dev->devnum = devnum + IB_UCM_MAX_DEVICES;
1285		base = devnum + overflow_maj;
1286		set_bit(devnum, overflow_map);
1287	} else {
1288		ucm_dev->devnum = devnum;
1289		base = devnum + IB_UCM_BASE_DEV;
1290		set_bit(devnum, dev_map);
1291	}
1292
1293	cdev_init(&ucm_dev->cdev, &ucm_fops);
1294	ucm_dev->cdev.owner = THIS_MODULE;
1295	kobject_set_name(&ucm_dev->cdev.kobj, "ucm%d", ucm_dev->devnum);
1296	if (cdev_add(&ucm_dev->cdev, base, 1))
1297		goto err;
1298
1299	ucm_dev->dev.class = &cm_class;
1300	ucm_dev->dev.parent = device->dma_device;
1301	ucm_dev->dev.devt = ucm_dev->cdev.dev;
1302	ucm_dev->dev.release = ib_ucm_release_dev;
1303	dev_set_name(&ucm_dev->dev, "ucm%d", ucm_dev->devnum);
1304	if (device_register(&ucm_dev->dev))
1305		goto err_cdev;
1306
1307	if (device_create_file(&ucm_dev->dev, &dev_attr_ibdev))
1308		goto err_dev;
1309
1310	ib_set_client_data(device, &ucm_client, ucm_dev);
1311	return;
1312
1313err_dev:
1314	device_unregister(&ucm_dev->dev);
1315err_cdev:
1316	cdev_del(&ucm_dev->cdev);
1317	if (ucm_dev->devnum < IB_UCM_MAX_DEVICES)
1318		clear_bit(devnum, dev_map);
1319	else
1320		clear_bit(devnum, overflow_map);
1321err:
1322	kfree(ucm_dev);
1323	return;
1324}
1325
1326static void ib_ucm_remove_one(struct ib_device *device)
1327{
1328	struct ib_ucm_device *ucm_dev = ib_get_client_data(device, &ucm_client);
1329
1330	if (!ucm_dev)
1331		return;
1332
1333	device_unregister(&ucm_dev->dev);
1334}
1335
1336static CLASS_ATTR_STRING(abi_version, S_IRUGO,
1337			 __stringify(IB_USER_CM_ABI_VERSION));
1338
1339static int __init ib_ucm_init(void)
1340{
1341	int ret;
1342
1343	ret = register_chrdev_region(IB_UCM_BASE_DEV, IB_UCM_MAX_DEVICES,
1344				     "infiniband_cm");
1345	if (ret) {
1346		printk(KERN_ERR "ucm: couldn't register device number\n");
1347		goto error1;
1348	}
1349
1350	ret = class_create_file(&cm_class, &class_attr_abi_version.attr);
1351	if (ret) {
1352		printk(KERN_ERR "ucm: couldn't create abi_version attribute\n");
1353		goto error2;
1354	}
1355
1356	ret = ib_register_client(&ucm_client);
1357	if (ret) {
1358		printk(KERN_ERR "ucm: couldn't register client\n");
1359		goto error3;
1360	}
1361	return 0;
1362
1363error3:
1364	class_remove_file(&cm_class, &class_attr_abi_version.attr);
1365error2:
1366	unregister_chrdev_region(IB_UCM_BASE_DEV, IB_UCM_MAX_DEVICES);
1367error1:
1368	return ret;
1369}
1370
1371static void __exit ib_ucm_cleanup(void)
1372{
1373	ib_unregister_client(&ucm_client);
1374	class_remove_file(&cm_class, &class_attr_abi_version.attr);
1375	unregister_chrdev_region(IB_UCM_BASE_DEV, IB_UCM_MAX_DEVICES);
1376	if (overflow_maj)
1377		unregister_chrdev_region(overflow_maj, IB_UCM_MAX_DEVICES);
1378	idr_destroy(&ctx_id_table);
1379}
1380
1381module_init(ib_ucm_init);
1382module_exit(ib_ucm_cleanup);
v4.6
   1/*
   2 * Copyright (c) 2005 Topspin Communications.  All rights reserved.
   3 * Copyright (c) 2005 Intel Corporation.  All rights reserved.
   4 *
   5 * This software is available to you under a choice of one of two
   6 * licenses.  You may choose to be licensed under the terms of the GNU
   7 * General Public License (GPL) Version 2, available from the file
   8 * COPYING in the main directory of this source tree, or the
   9 * OpenIB.org BSD license below:
  10 *
  11 *     Redistribution and use in source and binary forms, with or
  12 *     without modification, are permitted provided that the following
  13 *     conditions are met:
  14 *
  15 *      - Redistributions of source code must retain the above
  16 *	copyright notice, this list of conditions and the following
  17 *	disclaimer.
  18 *
  19 *      - Redistributions in binary form must reproduce the above
  20 *	copyright notice, this list of conditions and the following
  21 *	disclaimer in the documentation and/or other materials
  22 *	provided with the distribution.
  23 *
  24 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  25 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  26 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  27 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  28 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  29 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  30 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  31 * SOFTWARE.
  32 */
  33
  34#include <linux/completion.h>
  35#include <linux/init.h>
  36#include <linux/fs.h>
  37#include <linux/module.h>
  38#include <linux/device.h>
  39#include <linux/err.h>
  40#include <linux/poll.h>
  41#include <linux/sched.h>
  42#include <linux/file.h>
  43#include <linux/mount.h>
  44#include <linux/cdev.h>
  45#include <linux/idr.h>
  46#include <linux/mutex.h>
  47#include <linux/slab.h>
  48
  49#include <asm/uaccess.h>
  50
  51#include <rdma/ib.h>
  52#include <rdma/ib_cm.h>
  53#include <rdma/ib_user_cm.h>
  54#include <rdma/ib_marshall.h>
  55
  56MODULE_AUTHOR("Libor Michalek");
  57MODULE_DESCRIPTION("InfiniBand userspace Connection Manager access");
  58MODULE_LICENSE("Dual BSD/GPL");
  59
  60struct ib_ucm_device {
  61	int			devnum;
  62	struct cdev		cdev;
  63	struct device		dev;
  64	struct ib_device	*ib_dev;
  65};
  66
  67struct ib_ucm_file {
  68	struct mutex file_mutex;
  69	struct file *filp;
  70	struct ib_ucm_device *device;
  71
  72	struct list_head  ctxs;
  73	struct list_head  events;
  74	wait_queue_head_t poll_wait;
  75};
  76
  77struct ib_ucm_context {
  78	int                 id;
  79	struct completion   comp;
  80	atomic_t            ref;
  81	int		    events_reported;
  82
  83	struct ib_ucm_file *file;
  84	struct ib_cm_id    *cm_id;
  85	__u64		   uid;
  86
  87	struct list_head    events;    /* list of pending events. */
  88	struct list_head    file_list; /* member in file ctx list */
  89};
  90
  91struct ib_ucm_event {
  92	struct ib_ucm_context *ctx;
  93	struct list_head file_list; /* member in file event list */
  94	struct list_head ctx_list;  /* member in ctx event list */
  95
  96	struct ib_cm_id *cm_id;
  97	struct ib_ucm_event_resp resp;
  98	void *data;
  99	void *info;
 100	int data_len;
 101	int info_len;
 102};
 103
 104enum {
 105	IB_UCM_MAJOR = 231,
 106	IB_UCM_BASE_MINOR = 224,
 107	IB_UCM_MAX_DEVICES = 32
 108};
 109
 
 
 
 110#define IB_UCM_BASE_DEV MKDEV(IB_UCM_MAJOR, IB_UCM_BASE_MINOR)
 111
 112static void ib_ucm_add_one(struct ib_device *device);
 113static void ib_ucm_remove_one(struct ib_device *device, void *client_data);
 114
 115static struct ib_client ucm_client = {
 116	.name   = "ucm",
 117	.add    = ib_ucm_add_one,
 118	.remove = ib_ucm_remove_one
 119};
 120
 121static DEFINE_MUTEX(ctx_id_mutex);
 122static DEFINE_IDR(ctx_id_table);
 123static DECLARE_BITMAP(dev_map, IB_UCM_MAX_DEVICES);
 124
 125static struct ib_ucm_context *ib_ucm_ctx_get(struct ib_ucm_file *file, int id)
 126{
 127	struct ib_ucm_context *ctx;
 128
 129	mutex_lock(&ctx_id_mutex);
 130	ctx = idr_find(&ctx_id_table, id);
 131	if (!ctx)
 132		ctx = ERR_PTR(-ENOENT);
 133	else if (ctx->file != file)
 134		ctx = ERR_PTR(-EINVAL);
 135	else
 136		atomic_inc(&ctx->ref);
 137	mutex_unlock(&ctx_id_mutex);
 138
 139	return ctx;
 140}
 141
 142static void ib_ucm_ctx_put(struct ib_ucm_context *ctx)
 143{
 144	if (atomic_dec_and_test(&ctx->ref))
 145		complete(&ctx->comp);
 146}
 147
 148static inline int ib_ucm_new_cm_id(int event)
 149{
 150	return event == IB_CM_REQ_RECEIVED || event == IB_CM_SIDR_REQ_RECEIVED;
 151}
 152
 153static void ib_ucm_cleanup_events(struct ib_ucm_context *ctx)
 154{
 155	struct ib_ucm_event *uevent;
 156
 157	mutex_lock(&ctx->file->file_mutex);
 158	list_del(&ctx->file_list);
 159	while (!list_empty(&ctx->events)) {
 160
 161		uevent = list_entry(ctx->events.next,
 162				    struct ib_ucm_event, ctx_list);
 163		list_del(&uevent->file_list);
 164		list_del(&uevent->ctx_list);
 165		mutex_unlock(&ctx->file->file_mutex);
 166
 167		/* clear incoming connections. */
 168		if (ib_ucm_new_cm_id(uevent->resp.event))
 169			ib_destroy_cm_id(uevent->cm_id);
 170
 171		kfree(uevent);
 172		mutex_lock(&ctx->file->file_mutex);
 173	}
 174	mutex_unlock(&ctx->file->file_mutex);
 175}
 176
 177static struct ib_ucm_context *ib_ucm_ctx_alloc(struct ib_ucm_file *file)
 178{
 179	struct ib_ucm_context *ctx;
 
 180
 181	ctx = kzalloc(sizeof *ctx, GFP_KERNEL);
 182	if (!ctx)
 183		return NULL;
 184
 185	atomic_set(&ctx->ref, 1);
 186	init_completion(&ctx->comp);
 187	ctx->file = file;
 188	INIT_LIST_HEAD(&ctx->events);
 189
 190	mutex_lock(&ctx_id_mutex);
 191	ctx->id = idr_alloc(&ctx_id_table, ctx, 0, 0, GFP_KERNEL);
 192	mutex_unlock(&ctx_id_mutex);
 193	if (ctx->id < 0)
 
 
 
 
 
 
 
 194		goto error;
 195
 196	list_add_tail(&ctx->file_list, &file->ctxs);
 197	return ctx;
 198
 199error:
 200	kfree(ctx);
 201	return NULL;
 202}
 203
 204static void ib_ucm_event_req_get(struct ib_ucm_req_event_resp *ureq,
 205				 struct ib_cm_req_event_param *kreq)
 206{
 207	ureq->remote_ca_guid             = kreq->remote_ca_guid;
 208	ureq->remote_qkey                = kreq->remote_qkey;
 209	ureq->remote_qpn                 = kreq->remote_qpn;
 210	ureq->qp_type                    = kreq->qp_type;
 211	ureq->starting_psn               = kreq->starting_psn;
 212	ureq->responder_resources        = kreq->responder_resources;
 213	ureq->initiator_depth            = kreq->initiator_depth;
 214	ureq->local_cm_response_timeout  = kreq->local_cm_response_timeout;
 215	ureq->flow_control               = kreq->flow_control;
 216	ureq->remote_cm_response_timeout = kreq->remote_cm_response_timeout;
 217	ureq->retry_count                = kreq->retry_count;
 218	ureq->rnr_retry_count            = kreq->rnr_retry_count;
 219	ureq->srq                        = kreq->srq;
 220	ureq->port			 = kreq->port;
 221
 222	ib_copy_path_rec_to_user(&ureq->primary_path, kreq->primary_path);
 223	if (kreq->alternate_path)
 224		ib_copy_path_rec_to_user(&ureq->alternate_path,
 225					 kreq->alternate_path);
 226}
 227
 228static void ib_ucm_event_rep_get(struct ib_ucm_rep_event_resp *urep,
 229				 struct ib_cm_rep_event_param *krep)
 230{
 231	urep->remote_ca_guid      = krep->remote_ca_guid;
 232	urep->remote_qkey         = krep->remote_qkey;
 233	urep->remote_qpn          = krep->remote_qpn;
 234	urep->starting_psn        = krep->starting_psn;
 235	urep->responder_resources = krep->responder_resources;
 236	urep->initiator_depth     = krep->initiator_depth;
 237	urep->target_ack_delay    = krep->target_ack_delay;
 238	urep->failover_accepted   = krep->failover_accepted;
 239	urep->flow_control        = krep->flow_control;
 240	urep->rnr_retry_count     = krep->rnr_retry_count;
 241	urep->srq                 = krep->srq;
 242}
 243
 244static void ib_ucm_event_sidr_rep_get(struct ib_ucm_sidr_rep_event_resp *urep,
 245				      struct ib_cm_sidr_rep_event_param *krep)
 246{
 247	urep->status = krep->status;
 248	urep->qkey   = krep->qkey;
 249	urep->qpn    = krep->qpn;
 250};
 251
 252static int ib_ucm_event_process(struct ib_cm_event *evt,
 253				struct ib_ucm_event *uvt)
 254{
 255	void *info = NULL;
 256
 257	switch (evt->event) {
 258	case IB_CM_REQ_RECEIVED:
 259		ib_ucm_event_req_get(&uvt->resp.u.req_resp,
 260				     &evt->param.req_rcvd);
 261		uvt->data_len      = IB_CM_REQ_PRIVATE_DATA_SIZE;
 262		uvt->resp.present  = IB_UCM_PRES_PRIMARY;
 263		uvt->resp.present |= (evt->param.req_rcvd.alternate_path ?
 264				      IB_UCM_PRES_ALTERNATE : 0);
 265		break;
 266	case IB_CM_REP_RECEIVED:
 267		ib_ucm_event_rep_get(&uvt->resp.u.rep_resp,
 268				     &evt->param.rep_rcvd);
 269		uvt->data_len = IB_CM_REP_PRIVATE_DATA_SIZE;
 270		break;
 271	case IB_CM_RTU_RECEIVED:
 272		uvt->data_len = IB_CM_RTU_PRIVATE_DATA_SIZE;
 273		uvt->resp.u.send_status = evt->param.send_status;
 274		break;
 275	case IB_CM_DREQ_RECEIVED:
 276		uvt->data_len = IB_CM_DREQ_PRIVATE_DATA_SIZE;
 277		uvt->resp.u.send_status = evt->param.send_status;
 278		break;
 279	case IB_CM_DREP_RECEIVED:
 280		uvt->data_len = IB_CM_DREP_PRIVATE_DATA_SIZE;
 281		uvt->resp.u.send_status = evt->param.send_status;
 282		break;
 283	case IB_CM_MRA_RECEIVED:
 284		uvt->resp.u.mra_resp.timeout =
 285					evt->param.mra_rcvd.service_timeout;
 286		uvt->data_len = IB_CM_MRA_PRIVATE_DATA_SIZE;
 287		break;
 288	case IB_CM_REJ_RECEIVED:
 289		uvt->resp.u.rej_resp.reason = evt->param.rej_rcvd.reason;
 290		uvt->data_len = IB_CM_REJ_PRIVATE_DATA_SIZE;
 291		uvt->info_len = evt->param.rej_rcvd.ari_length;
 292		info	      = evt->param.rej_rcvd.ari;
 293		break;
 294	case IB_CM_LAP_RECEIVED:
 295		ib_copy_path_rec_to_user(&uvt->resp.u.lap_resp.path,
 296					 evt->param.lap_rcvd.alternate_path);
 297		uvt->data_len = IB_CM_LAP_PRIVATE_DATA_SIZE;
 298		uvt->resp.present = IB_UCM_PRES_ALTERNATE;
 299		break;
 300	case IB_CM_APR_RECEIVED:
 301		uvt->resp.u.apr_resp.status = evt->param.apr_rcvd.ap_status;
 302		uvt->data_len = IB_CM_APR_PRIVATE_DATA_SIZE;
 303		uvt->info_len = evt->param.apr_rcvd.info_len;
 304		info	      = evt->param.apr_rcvd.apr_info;
 305		break;
 306	case IB_CM_SIDR_REQ_RECEIVED:
 307		uvt->resp.u.sidr_req_resp.pkey =
 308					evt->param.sidr_req_rcvd.pkey;
 309		uvt->resp.u.sidr_req_resp.port =
 310					evt->param.sidr_req_rcvd.port;
 311		uvt->data_len = IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE;
 312		break;
 313	case IB_CM_SIDR_REP_RECEIVED:
 314		ib_ucm_event_sidr_rep_get(&uvt->resp.u.sidr_rep_resp,
 315					  &evt->param.sidr_rep_rcvd);
 316		uvt->data_len = IB_CM_SIDR_REP_PRIVATE_DATA_SIZE;
 317		uvt->info_len = evt->param.sidr_rep_rcvd.info_len;
 318		info	      = evt->param.sidr_rep_rcvd.info;
 319		break;
 320	default:
 321		uvt->resp.u.send_status = evt->param.send_status;
 322		break;
 323	}
 324
 325	if (uvt->data_len) {
 326		uvt->data = kmemdup(evt->private_data, uvt->data_len, GFP_KERNEL);
 327		if (!uvt->data)
 328			goto err1;
 329
 330		uvt->resp.present |= IB_UCM_PRES_DATA;
 331	}
 332
 333	if (uvt->info_len) {
 334		uvt->info = kmemdup(info, uvt->info_len, GFP_KERNEL);
 335		if (!uvt->info)
 336			goto err2;
 337
 338		uvt->resp.present |= IB_UCM_PRES_INFO;
 339	}
 340	return 0;
 341
 342err2:
 343	kfree(uvt->data);
 344err1:
 345	return -ENOMEM;
 346}
 347
 348static int ib_ucm_event_handler(struct ib_cm_id *cm_id,
 349				struct ib_cm_event *event)
 350{
 351	struct ib_ucm_event *uevent;
 352	struct ib_ucm_context *ctx;
 353	int result = 0;
 354
 355	ctx = cm_id->context;
 356
 357	uevent = kzalloc(sizeof *uevent, GFP_KERNEL);
 358	if (!uevent)
 359		goto err1;
 360
 361	uevent->ctx = ctx;
 362	uevent->cm_id = cm_id;
 363	uevent->resp.uid = ctx->uid;
 364	uevent->resp.id = ctx->id;
 365	uevent->resp.event = event->event;
 366
 367	result = ib_ucm_event_process(event, uevent);
 368	if (result)
 369		goto err2;
 370
 371	mutex_lock(&ctx->file->file_mutex);
 372	list_add_tail(&uevent->file_list, &ctx->file->events);
 373	list_add_tail(&uevent->ctx_list, &ctx->events);
 374	wake_up_interruptible(&ctx->file->poll_wait);
 375	mutex_unlock(&ctx->file->file_mutex);
 376	return 0;
 377
 378err2:
 379	kfree(uevent);
 380err1:
 381	/* Destroy new cm_id's */
 382	return ib_ucm_new_cm_id(event->event);
 383}
 384
 385static ssize_t ib_ucm_event(struct ib_ucm_file *file,
 386			    const char __user *inbuf,
 387			    int in_len, int out_len)
 388{
 389	struct ib_ucm_context *ctx;
 390	struct ib_ucm_event_get cmd;
 391	struct ib_ucm_event *uevent;
 392	int result = 0;
 
 393
 394	if (out_len < sizeof(struct ib_ucm_event_resp))
 395		return -ENOSPC;
 396
 397	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 398		return -EFAULT;
 399
 400	mutex_lock(&file->file_mutex);
 401	while (list_empty(&file->events)) {
 402		mutex_unlock(&file->file_mutex);
 403
 404		if (file->filp->f_flags & O_NONBLOCK)
 405			return -EAGAIN;
 406
 407		if (wait_event_interruptible(file->poll_wait,
 408					     !list_empty(&file->events)))
 409			return -ERESTARTSYS;
 410
 411		mutex_lock(&file->file_mutex);
 412	}
 413
 414	uevent = list_entry(file->events.next, struct ib_ucm_event, file_list);
 415
 416	if (ib_ucm_new_cm_id(uevent->resp.event)) {
 417		ctx = ib_ucm_ctx_alloc(file);
 418		if (!ctx) {
 419			result = -ENOMEM;
 420			goto done;
 421		}
 422
 423		ctx->cm_id = uevent->cm_id;
 424		ctx->cm_id->context = ctx;
 425		uevent->resp.id = ctx->id;
 426	}
 427
 428	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 429			 &uevent->resp, sizeof(uevent->resp))) {
 430		result = -EFAULT;
 431		goto done;
 432	}
 433
 434	if (uevent->data) {
 435		if (cmd.data_len < uevent->data_len) {
 436			result = -ENOMEM;
 437			goto done;
 438		}
 439		if (copy_to_user((void __user *)(unsigned long)cmd.data,
 440				 uevent->data, uevent->data_len)) {
 441			result = -EFAULT;
 442			goto done;
 443		}
 444	}
 445
 446	if (uevent->info) {
 447		if (cmd.info_len < uevent->info_len) {
 448			result = -ENOMEM;
 449			goto done;
 450		}
 451		if (copy_to_user((void __user *)(unsigned long)cmd.info,
 452				 uevent->info, uevent->info_len)) {
 453			result = -EFAULT;
 454			goto done;
 455		}
 456	}
 457
 458	list_del(&uevent->file_list);
 459	list_del(&uevent->ctx_list);
 460	uevent->ctx->events_reported++;
 461
 462	kfree(uevent->data);
 463	kfree(uevent->info);
 464	kfree(uevent);
 465done:
 466	mutex_unlock(&file->file_mutex);
 467	return result;
 468}
 469
 470static ssize_t ib_ucm_create_id(struct ib_ucm_file *file,
 471				const char __user *inbuf,
 472				int in_len, int out_len)
 473{
 474	struct ib_ucm_create_id cmd;
 475	struct ib_ucm_create_id_resp resp;
 476	struct ib_ucm_context *ctx;
 477	int result;
 478
 479	if (out_len < sizeof(resp))
 480		return -ENOSPC;
 481
 482	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 483		return -EFAULT;
 484
 485	mutex_lock(&file->file_mutex);
 486	ctx = ib_ucm_ctx_alloc(file);
 487	mutex_unlock(&file->file_mutex);
 488	if (!ctx)
 489		return -ENOMEM;
 490
 491	ctx->uid = cmd.uid;
 492	ctx->cm_id = ib_create_cm_id(file->device->ib_dev,
 493				     ib_ucm_event_handler, ctx);
 494	if (IS_ERR(ctx->cm_id)) {
 495		result = PTR_ERR(ctx->cm_id);
 496		goto err1;
 497	}
 498
 499	resp.id = ctx->id;
 500	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 501			 &resp, sizeof(resp))) {
 502		result = -EFAULT;
 503		goto err2;
 504	}
 505	return 0;
 506
 507err2:
 508	ib_destroy_cm_id(ctx->cm_id);
 509err1:
 510	mutex_lock(&ctx_id_mutex);
 511	idr_remove(&ctx_id_table, ctx->id);
 512	mutex_unlock(&ctx_id_mutex);
 513	kfree(ctx);
 514	return result;
 515}
 516
 517static ssize_t ib_ucm_destroy_id(struct ib_ucm_file *file,
 518				 const char __user *inbuf,
 519				 int in_len, int out_len)
 520{
 521	struct ib_ucm_destroy_id cmd;
 522	struct ib_ucm_destroy_id_resp resp;
 523	struct ib_ucm_context *ctx;
 524	int result = 0;
 525
 526	if (out_len < sizeof(resp))
 527		return -ENOSPC;
 528
 529	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 530		return -EFAULT;
 531
 532	mutex_lock(&ctx_id_mutex);
 533	ctx = idr_find(&ctx_id_table, cmd.id);
 534	if (!ctx)
 535		ctx = ERR_PTR(-ENOENT);
 536	else if (ctx->file != file)
 537		ctx = ERR_PTR(-EINVAL);
 538	else
 539		idr_remove(&ctx_id_table, ctx->id);
 540	mutex_unlock(&ctx_id_mutex);
 541
 542	if (IS_ERR(ctx))
 543		return PTR_ERR(ctx);
 544
 545	ib_ucm_ctx_put(ctx);
 546	wait_for_completion(&ctx->comp);
 547
 548	/* No new events will be generated after destroying the cm_id. */
 549	ib_destroy_cm_id(ctx->cm_id);
 550	/* Cleanup events not yet reported to the user. */
 551	ib_ucm_cleanup_events(ctx);
 552
 553	resp.events_reported = ctx->events_reported;
 554	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 555			 &resp, sizeof(resp)))
 556		result = -EFAULT;
 557
 558	kfree(ctx);
 559	return result;
 560}
 561
 562static ssize_t ib_ucm_attr_id(struct ib_ucm_file *file,
 563			      const char __user *inbuf,
 564			      int in_len, int out_len)
 565{
 566	struct ib_ucm_attr_id_resp resp;
 567	struct ib_ucm_attr_id cmd;
 568	struct ib_ucm_context *ctx;
 569	int result = 0;
 570
 571	if (out_len < sizeof(resp))
 572		return -ENOSPC;
 573
 574	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 575		return -EFAULT;
 576
 577	ctx = ib_ucm_ctx_get(file, cmd.id);
 578	if (IS_ERR(ctx))
 579		return PTR_ERR(ctx);
 580
 581	resp.service_id   = ctx->cm_id->service_id;
 582	resp.service_mask = ctx->cm_id->service_mask;
 583	resp.local_id     = ctx->cm_id->local_id;
 584	resp.remote_id    = ctx->cm_id->remote_id;
 585
 586	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 587			 &resp, sizeof(resp)))
 588		result = -EFAULT;
 589
 590	ib_ucm_ctx_put(ctx);
 591	return result;
 592}
 593
 594static ssize_t ib_ucm_init_qp_attr(struct ib_ucm_file *file,
 595				   const char __user *inbuf,
 596				   int in_len, int out_len)
 597{
 598	struct ib_uverbs_qp_attr resp;
 599	struct ib_ucm_init_qp_attr cmd;
 600	struct ib_ucm_context *ctx;
 601	struct ib_qp_attr qp_attr;
 602	int result = 0;
 603
 604	if (out_len < sizeof(resp))
 605		return -ENOSPC;
 606
 607	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 608		return -EFAULT;
 609
 610	ctx = ib_ucm_ctx_get(file, cmd.id);
 611	if (IS_ERR(ctx))
 612		return PTR_ERR(ctx);
 613
 614	resp.qp_attr_mask = 0;
 615	memset(&qp_attr, 0, sizeof qp_attr);
 616	qp_attr.qp_state = cmd.qp_state;
 617	result = ib_cm_init_qp_attr(ctx->cm_id, &qp_attr, &resp.qp_attr_mask);
 618	if (result)
 619		goto out;
 620
 621	ib_copy_qp_attr_to_user(&resp, &qp_attr);
 622
 623	if (copy_to_user((void __user *)(unsigned long)cmd.response,
 624			 &resp, sizeof(resp)))
 625		result = -EFAULT;
 626
 627out:
 628	ib_ucm_ctx_put(ctx);
 629	return result;
 630}
 631
 632static int ucm_validate_listen(__be64 service_id, __be64 service_mask)
 633{
 634	service_id &= service_mask;
 635
 636	if (((service_id & IB_CMA_SERVICE_ID_MASK) == IB_CMA_SERVICE_ID) ||
 637	    ((service_id & IB_SDP_SERVICE_ID_MASK) == IB_SDP_SERVICE_ID))
 638		return -EINVAL;
 639
 640	return 0;
 641}
 642
 643static ssize_t ib_ucm_listen(struct ib_ucm_file *file,
 644			     const char __user *inbuf,
 645			     int in_len, int out_len)
 646{
 647	struct ib_ucm_listen cmd;
 648	struct ib_ucm_context *ctx;
 649	int result;
 650
 651	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 652		return -EFAULT;
 653
 654	ctx = ib_ucm_ctx_get(file, cmd.id);
 655	if (IS_ERR(ctx))
 656		return PTR_ERR(ctx);
 657
 658	result = ucm_validate_listen(cmd.service_id, cmd.service_mask);
 659	if (result)
 660		goto out;
 661
 662	result = ib_cm_listen(ctx->cm_id, cmd.service_id, cmd.service_mask);
 
 663out:
 664	ib_ucm_ctx_put(ctx);
 665	return result;
 666}
 667
 668static ssize_t ib_ucm_notify(struct ib_ucm_file *file,
 669			     const char __user *inbuf,
 670			     int in_len, int out_len)
 671{
 672	struct ib_ucm_notify cmd;
 673	struct ib_ucm_context *ctx;
 674	int result;
 675
 676	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 677		return -EFAULT;
 678
 679	ctx = ib_ucm_ctx_get(file, cmd.id);
 680	if (IS_ERR(ctx))
 681		return PTR_ERR(ctx);
 682
 683	result = ib_cm_notify(ctx->cm_id, (enum ib_event_type) cmd.event);
 684	ib_ucm_ctx_put(ctx);
 685	return result;
 686}
 687
 688static int ib_ucm_alloc_data(const void **dest, u64 src, u32 len)
 689{
 690	void *data;
 691
 692	*dest = NULL;
 693
 694	if (!len)
 695		return 0;
 696
 697	data = memdup_user((void __user *)(unsigned long)src, len);
 698	if (IS_ERR(data))
 699		return PTR_ERR(data);
 700
 701	*dest = data;
 702	return 0;
 703}
 704
 705static int ib_ucm_path_get(struct ib_sa_path_rec **path, u64 src)
 706{
 707	struct ib_user_path_rec upath;
 708	struct ib_sa_path_rec  *sa_path;
 709
 710	*path = NULL;
 711
 712	if (!src)
 713		return 0;
 714
 715	sa_path = kmalloc(sizeof(*sa_path), GFP_KERNEL);
 716	if (!sa_path)
 717		return -ENOMEM;
 718
 719	if (copy_from_user(&upath, (void __user *)(unsigned long)src,
 720			   sizeof(upath))) {
 721
 722		kfree(sa_path);
 723		return -EFAULT;
 724	}
 725
 726	ib_copy_path_rec_from_user(sa_path, &upath);
 727	*path = sa_path;
 728	return 0;
 729}
 730
 731static ssize_t ib_ucm_send_req(struct ib_ucm_file *file,
 732			       const char __user *inbuf,
 733			       int in_len, int out_len)
 734{
 735	struct ib_cm_req_param param;
 736	struct ib_ucm_context *ctx;
 737	struct ib_ucm_req cmd;
 738	int result;
 739
 740	param.private_data   = NULL;
 741	param.primary_path   = NULL;
 742	param.alternate_path = NULL;
 743
 744	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 745		return -EFAULT;
 746
 747	result = ib_ucm_alloc_data(&param.private_data, cmd.data, cmd.len);
 748	if (result)
 749		goto done;
 750
 751	result = ib_ucm_path_get(&param.primary_path, cmd.primary_path);
 752	if (result)
 753		goto done;
 754
 755	result = ib_ucm_path_get(&param.alternate_path, cmd.alternate_path);
 756	if (result)
 757		goto done;
 758
 759	param.private_data_len           = cmd.len;
 760	param.service_id                 = cmd.sid;
 761	param.qp_num                     = cmd.qpn;
 762	param.qp_type                    = cmd.qp_type;
 763	param.starting_psn               = cmd.psn;
 764	param.peer_to_peer               = cmd.peer_to_peer;
 765	param.responder_resources        = cmd.responder_resources;
 766	param.initiator_depth            = cmd.initiator_depth;
 767	param.remote_cm_response_timeout = cmd.remote_cm_response_timeout;
 768	param.flow_control               = cmd.flow_control;
 769	param.local_cm_response_timeout  = cmd.local_cm_response_timeout;
 770	param.retry_count                = cmd.retry_count;
 771	param.rnr_retry_count            = cmd.rnr_retry_count;
 772	param.max_cm_retries             = cmd.max_cm_retries;
 773	param.srq                        = cmd.srq;
 774
 775	ctx = ib_ucm_ctx_get(file, cmd.id);
 776	if (!IS_ERR(ctx)) {
 777		result = ib_send_cm_req(ctx->cm_id, &param);
 778		ib_ucm_ctx_put(ctx);
 779	} else
 780		result = PTR_ERR(ctx);
 781
 782done:
 783	kfree(param.private_data);
 784	kfree(param.primary_path);
 785	kfree(param.alternate_path);
 786	return result;
 787}
 788
 789static ssize_t ib_ucm_send_rep(struct ib_ucm_file *file,
 790			       const char __user *inbuf,
 791			       int in_len, int out_len)
 792{
 793	struct ib_cm_rep_param param;
 794	struct ib_ucm_context *ctx;
 795	struct ib_ucm_rep cmd;
 796	int result;
 797
 798	param.private_data = NULL;
 799
 800	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 801		return -EFAULT;
 802
 803	result = ib_ucm_alloc_data(&param.private_data, cmd.data, cmd.len);
 804	if (result)
 805		return result;
 806
 807	param.qp_num              = cmd.qpn;
 808	param.starting_psn        = cmd.psn;
 809	param.private_data_len    = cmd.len;
 810	param.responder_resources = cmd.responder_resources;
 811	param.initiator_depth     = cmd.initiator_depth;
 812	param.failover_accepted   = cmd.failover_accepted;
 813	param.flow_control        = cmd.flow_control;
 814	param.rnr_retry_count     = cmd.rnr_retry_count;
 815	param.srq                 = cmd.srq;
 816
 817	ctx = ib_ucm_ctx_get(file, cmd.id);
 818	if (!IS_ERR(ctx)) {
 819		ctx->uid = cmd.uid;
 820		result = ib_send_cm_rep(ctx->cm_id, &param);
 821		ib_ucm_ctx_put(ctx);
 822	} else
 823		result = PTR_ERR(ctx);
 824
 825	kfree(param.private_data);
 826	return result;
 827}
 828
 829static ssize_t ib_ucm_send_private_data(struct ib_ucm_file *file,
 830					const char __user *inbuf, int in_len,
 831					int (*func)(struct ib_cm_id *cm_id,
 832						    const void *private_data,
 833						    u8 private_data_len))
 834{
 835	struct ib_ucm_private_data cmd;
 836	struct ib_ucm_context *ctx;
 837	const void *private_data = NULL;
 838	int result;
 839
 840	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 841		return -EFAULT;
 842
 843	result = ib_ucm_alloc_data(&private_data, cmd.data, cmd.len);
 844	if (result)
 845		return result;
 846
 847	ctx = ib_ucm_ctx_get(file, cmd.id);
 848	if (!IS_ERR(ctx)) {
 849		result = func(ctx->cm_id, private_data, cmd.len);
 850		ib_ucm_ctx_put(ctx);
 851	} else
 852		result = PTR_ERR(ctx);
 853
 854	kfree(private_data);
 855	return result;
 856}
 857
 858static ssize_t ib_ucm_send_rtu(struct ib_ucm_file *file,
 859			       const char __user *inbuf,
 860			       int in_len, int out_len)
 861{
 862	return ib_ucm_send_private_data(file, inbuf, in_len, ib_send_cm_rtu);
 863}
 864
 865static ssize_t ib_ucm_send_dreq(struct ib_ucm_file *file,
 866				const char __user *inbuf,
 867				int in_len, int out_len)
 868{
 869	return ib_ucm_send_private_data(file, inbuf, in_len, ib_send_cm_dreq);
 870}
 871
 872static ssize_t ib_ucm_send_drep(struct ib_ucm_file *file,
 873				const char __user *inbuf,
 874				int in_len, int out_len)
 875{
 876	return ib_ucm_send_private_data(file, inbuf, in_len, ib_send_cm_drep);
 877}
 878
 879static ssize_t ib_ucm_send_info(struct ib_ucm_file *file,
 880				const char __user *inbuf, int in_len,
 881				int (*func)(struct ib_cm_id *cm_id,
 882					    int status,
 883					    const void *info,
 884					    u8 info_len,
 885					    const void *data,
 886					    u8 data_len))
 887{
 888	struct ib_ucm_context *ctx;
 889	struct ib_ucm_info cmd;
 890	const void *data = NULL;
 891	const void *info = NULL;
 892	int result;
 893
 894	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 895		return -EFAULT;
 896
 897	result = ib_ucm_alloc_data(&data, cmd.data, cmd.data_len);
 898	if (result)
 899		goto done;
 900
 901	result = ib_ucm_alloc_data(&info, cmd.info, cmd.info_len);
 902	if (result)
 903		goto done;
 904
 905	ctx = ib_ucm_ctx_get(file, cmd.id);
 906	if (!IS_ERR(ctx)) {
 907		result = func(ctx->cm_id, cmd.status, info, cmd.info_len,
 908			      data, cmd.data_len);
 909		ib_ucm_ctx_put(ctx);
 910	} else
 911		result = PTR_ERR(ctx);
 912
 913done:
 914	kfree(data);
 915	kfree(info);
 916	return result;
 917}
 918
 919static ssize_t ib_ucm_send_rej(struct ib_ucm_file *file,
 920			       const char __user *inbuf,
 921			       int in_len, int out_len)
 922{
 923	return ib_ucm_send_info(file, inbuf, in_len, (void *)ib_send_cm_rej);
 924}
 925
 926static ssize_t ib_ucm_send_apr(struct ib_ucm_file *file,
 927			       const char __user *inbuf,
 928			       int in_len, int out_len)
 929{
 930	return ib_ucm_send_info(file, inbuf, in_len, (void *)ib_send_cm_apr);
 931}
 932
 933static ssize_t ib_ucm_send_mra(struct ib_ucm_file *file,
 934			       const char __user *inbuf,
 935			       int in_len, int out_len)
 936{
 937	struct ib_ucm_context *ctx;
 938	struct ib_ucm_mra cmd;
 939	const void *data = NULL;
 940	int result;
 941
 942	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 943		return -EFAULT;
 944
 945	result = ib_ucm_alloc_data(&data, cmd.data, cmd.len);
 946	if (result)
 947		return result;
 948
 949	ctx = ib_ucm_ctx_get(file, cmd.id);
 950	if (!IS_ERR(ctx)) {
 951		result = ib_send_cm_mra(ctx->cm_id, cmd.timeout, data, cmd.len);
 952		ib_ucm_ctx_put(ctx);
 953	} else
 954		result = PTR_ERR(ctx);
 955
 956	kfree(data);
 957	return result;
 958}
 959
 960static ssize_t ib_ucm_send_lap(struct ib_ucm_file *file,
 961			       const char __user *inbuf,
 962			       int in_len, int out_len)
 963{
 964	struct ib_ucm_context *ctx;
 965	struct ib_sa_path_rec *path = NULL;
 966	struct ib_ucm_lap cmd;
 967	const void *data = NULL;
 968	int result;
 969
 970	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
 971		return -EFAULT;
 972
 973	result = ib_ucm_alloc_data(&data, cmd.data, cmd.len);
 974	if (result)
 975		goto done;
 976
 977	result = ib_ucm_path_get(&path, cmd.path);
 978	if (result)
 979		goto done;
 980
 981	ctx = ib_ucm_ctx_get(file, cmd.id);
 982	if (!IS_ERR(ctx)) {
 983		result = ib_send_cm_lap(ctx->cm_id, path, data, cmd.len);
 984		ib_ucm_ctx_put(ctx);
 985	} else
 986		result = PTR_ERR(ctx);
 987
 988done:
 989	kfree(data);
 990	kfree(path);
 991	return result;
 992}
 993
 994static ssize_t ib_ucm_send_sidr_req(struct ib_ucm_file *file,
 995				    const char __user *inbuf,
 996				    int in_len, int out_len)
 997{
 998	struct ib_cm_sidr_req_param param;
 999	struct ib_ucm_context *ctx;
1000	struct ib_ucm_sidr_req cmd;
1001	int result;
1002
1003	param.private_data = NULL;
1004	param.path = NULL;
1005
1006	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
1007		return -EFAULT;
1008
1009	result = ib_ucm_alloc_data(&param.private_data, cmd.data, cmd.len);
1010	if (result)
1011		goto done;
1012
1013	result = ib_ucm_path_get(&param.path, cmd.path);
1014	if (result)
1015		goto done;
1016
1017	param.private_data_len = cmd.len;
1018	param.service_id       = cmd.sid;
1019	param.timeout_ms       = cmd.timeout;
1020	param.max_cm_retries   = cmd.max_cm_retries;
1021
1022	ctx = ib_ucm_ctx_get(file, cmd.id);
1023	if (!IS_ERR(ctx)) {
1024		result = ib_send_cm_sidr_req(ctx->cm_id, &param);
1025		ib_ucm_ctx_put(ctx);
1026	} else
1027		result = PTR_ERR(ctx);
1028
1029done:
1030	kfree(param.private_data);
1031	kfree(param.path);
1032	return result;
1033}
1034
1035static ssize_t ib_ucm_send_sidr_rep(struct ib_ucm_file *file,
1036				    const char __user *inbuf,
1037				    int in_len, int out_len)
1038{
1039	struct ib_cm_sidr_rep_param param;
1040	struct ib_ucm_sidr_rep cmd;
1041	struct ib_ucm_context *ctx;
1042	int result;
1043
1044	param.info = NULL;
1045
1046	if (copy_from_user(&cmd, inbuf, sizeof(cmd)))
1047		return -EFAULT;
1048
1049	result = ib_ucm_alloc_data(&param.private_data,
1050				   cmd.data, cmd.data_len);
1051	if (result)
1052		goto done;
1053
1054	result = ib_ucm_alloc_data(&param.info, cmd.info, cmd.info_len);
1055	if (result)
1056		goto done;
1057
1058	param.qp_num		= cmd.qpn;
1059	param.qkey		= cmd.qkey;
1060	param.status		= cmd.status;
1061	param.info_length	= cmd.info_len;
1062	param.private_data_len	= cmd.data_len;
1063
1064	ctx = ib_ucm_ctx_get(file, cmd.id);
1065	if (!IS_ERR(ctx)) {
1066		result = ib_send_cm_sidr_rep(ctx->cm_id, &param);
1067		ib_ucm_ctx_put(ctx);
1068	} else
1069		result = PTR_ERR(ctx);
1070
1071done:
1072	kfree(param.private_data);
1073	kfree(param.info);
1074	return result;
1075}
1076
1077static ssize_t (*ucm_cmd_table[])(struct ib_ucm_file *file,
1078				  const char __user *inbuf,
1079				  int in_len, int out_len) = {
1080	[IB_USER_CM_CMD_CREATE_ID]     = ib_ucm_create_id,
1081	[IB_USER_CM_CMD_DESTROY_ID]    = ib_ucm_destroy_id,
1082	[IB_USER_CM_CMD_ATTR_ID]       = ib_ucm_attr_id,
1083	[IB_USER_CM_CMD_LISTEN]        = ib_ucm_listen,
1084	[IB_USER_CM_CMD_NOTIFY]        = ib_ucm_notify,
1085	[IB_USER_CM_CMD_SEND_REQ]      = ib_ucm_send_req,
1086	[IB_USER_CM_CMD_SEND_REP]      = ib_ucm_send_rep,
1087	[IB_USER_CM_CMD_SEND_RTU]      = ib_ucm_send_rtu,
1088	[IB_USER_CM_CMD_SEND_DREQ]     = ib_ucm_send_dreq,
1089	[IB_USER_CM_CMD_SEND_DREP]     = ib_ucm_send_drep,
1090	[IB_USER_CM_CMD_SEND_REJ]      = ib_ucm_send_rej,
1091	[IB_USER_CM_CMD_SEND_MRA]      = ib_ucm_send_mra,
1092	[IB_USER_CM_CMD_SEND_LAP]      = ib_ucm_send_lap,
1093	[IB_USER_CM_CMD_SEND_APR]      = ib_ucm_send_apr,
1094	[IB_USER_CM_CMD_SEND_SIDR_REQ] = ib_ucm_send_sidr_req,
1095	[IB_USER_CM_CMD_SEND_SIDR_REP] = ib_ucm_send_sidr_rep,
1096	[IB_USER_CM_CMD_EVENT]	       = ib_ucm_event,
1097	[IB_USER_CM_CMD_INIT_QP_ATTR]  = ib_ucm_init_qp_attr,
1098};
1099
1100static ssize_t ib_ucm_write(struct file *filp, const char __user *buf,
1101			    size_t len, loff_t *pos)
1102{
1103	struct ib_ucm_file *file = filp->private_data;
1104	struct ib_ucm_cmd_hdr hdr;
1105	ssize_t result;
1106
1107	if (WARN_ON_ONCE(!ib_safe_file_access(filp)))
1108		return -EACCES;
1109
1110	if (len < sizeof(hdr))
1111		return -EINVAL;
1112
1113	if (copy_from_user(&hdr, buf, sizeof(hdr)))
1114		return -EFAULT;
1115
1116	if (hdr.cmd >= ARRAY_SIZE(ucm_cmd_table))
1117		return -EINVAL;
1118
1119	if (hdr.in + sizeof(hdr) > len)
1120		return -EINVAL;
1121
1122	result = ucm_cmd_table[hdr.cmd](file, buf + sizeof(hdr),
1123					hdr.in, hdr.out);
1124	if (!result)
1125		result = len;
1126
1127	return result;
1128}
1129
1130static unsigned int ib_ucm_poll(struct file *filp,
1131				struct poll_table_struct *wait)
1132{
1133	struct ib_ucm_file *file = filp->private_data;
1134	unsigned int mask = 0;
1135
1136	poll_wait(filp, &file->poll_wait, wait);
1137
1138	if (!list_empty(&file->events))
1139		mask = POLLIN | POLLRDNORM;
1140
1141	return mask;
1142}
1143
1144/*
1145 * ib_ucm_open() does not need the BKL:
1146 *
1147 *  - no global state is referred to;
1148 *  - there is no ioctl method to race against;
1149 *  - no further module initialization is required for open to work
1150 *    after the device is registered.
1151 */
1152static int ib_ucm_open(struct inode *inode, struct file *filp)
1153{
1154	struct ib_ucm_file *file;
1155
1156	file = kmalloc(sizeof(*file), GFP_KERNEL);
1157	if (!file)
1158		return -ENOMEM;
1159
1160	INIT_LIST_HEAD(&file->events);
1161	INIT_LIST_HEAD(&file->ctxs);
1162	init_waitqueue_head(&file->poll_wait);
1163
1164	mutex_init(&file->file_mutex);
1165
1166	filp->private_data = file;
1167	file->filp = filp;
1168	file->device = container_of(inode->i_cdev, struct ib_ucm_device, cdev);
1169
1170	return nonseekable_open(inode, filp);
1171}
1172
1173static int ib_ucm_close(struct inode *inode, struct file *filp)
1174{
1175	struct ib_ucm_file *file = filp->private_data;
1176	struct ib_ucm_context *ctx;
1177
1178	mutex_lock(&file->file_mutex);
1179	while (!list_empty(&file->ctxs)) {
1180		ctx = list_entry(file->ctxs.next,
1181				 struct ib_ucm_context, file_list);
1182		mutex_unlock(&file->file_mutex);
1183
1184		mutex_lock(&ctx_id_mutex);
1185		idr_remove(&ctx_id_table, ctx->id);
1186		mutex_unlock(&ctx_id_mutex);
1187
1188		ib_destroy_cm_id(ctx->cm_id);
1189		ib_ucm_cleanup_events(ctx);
1190		kfree(ctx);
1191
1192		mutex_lock(&file->file_mutex);
1193	}
1194	mutex_unlock(&file->file_mutex);
1195	kfree(file);
1196	return 0;
1197}
1198
1199static DECLARE_BITMAP(overflow_map, IB_UCM_MAX_DEVICES);
1200static void ib_ucm_release_dev(struct device *dev)
1201{
1202	struct ib_ucm_device *ucm_dev;
1203
1204	ucm_dev = container_of(dev, struct ib_ucm_device, dev);
1205	cdev_del(&ucm_dev->cdev);
1206	if (ucm_dev->devnum < IB_UCM_MAX_DEVICES)
1207		clear_bit(ucm_dev->devnum, dev_map);
1208	else
1209		clear_bit(ucm_dev->devnum - IB_UCM_MAX_DEVICES, overflow_map);
1210	kfree(ucm_dev);
1211}
1212
1213static const struct file_operations ucm_fops = {
1214	.owner	 = THIS_MODULE,
1215	.open	 = ib_ucm_open,
1216	.release = ib_ucm_close,
1217	.write	 = ib_ucm_write,
1218	.poll    = ib_ucm_poll,
1219	.llseek	 = no_llseek,
1220};
1221
1222static ssize_t show_ibdev(struct device *dev, struct device_attribute *attr,
1223			  char *buf)
1224{
1225	struct ib_ucm_device *ucm_dev;
1226
1227	ucm_dev = container_of(dev, struct ib_ucm_device, dev);
1228	return sprintf(buf, "%s\n", ucm_dev->ib_dev->name);
1229}
1230static DEVICE_ATTR(ibdev, S_IRUGO, show_ibdev, NULL);
1231
1232static dev_t overflow_maj;
 
1233static int find_overflow_devnum(void)
1234{
1235	int ret;
1236
1237	if (!overflow_maj) {
1238		ret = alloc_chrdev_region(&overflow_maj, 0, IB_UCM_MAX_DEVICES,
1239					  "infiniband_cm");
1240		if (ret) {
1241			pr_err("ucm: couldn't register dynamic device number\n");
1242			return ret;
1243		}
1244	}
1245
1246	ret = find_first_zero_bit(overflow_map, IB_UCM_MAX_DEVICES);
1247	if (ret >= IB_UCM_MAX_DEVICES)
1248		return -1;
1249
1250	return ret;
1251}
1252
1253static void ib_ucm_add_one(struct ib_device *device)
1254{
1255	int devnum;
1256	dev_t base;
1257	struct ib_ucm_device *ucm_dev;
1258
1259	if (!device->alloc_ucontext || !rdma_cap_ib_cm(device, 1))
 
1260		return;
1261
1262	ucm_dev = kzalloc(sizeof *ucm_dev, GFP_KERNEL);
1263	if (!ucm_dev)
1264		return;
1265
1266	ucm_dev->ib_dev = device;
1267
1268	devnum = find_first_zero_bit(dev_map, IB_UCM_MAX_DEVICES);
1269	if (devnum >= IB_UCM_MAX_DEVICES) {
1270		devnum = find_overflow_devnum();
1271		if (devnum < 0)
1272			goto err;
1273
1274		ucm_dev->devnum = devnum + IB_UCM_MAX_DEVICES;
1275		base = devnum + overflow_maj;
1276		set_bit(devnum, overflow_map);
1277	} else {
1278		ucm_dev->devnum = devnum;
1279		base = devnum + IB_UCM_BASE_DEV;
1280		set_bit(devnum, dev_map);
1281	}
1282
1283	cdev_init(&ucm_dev->cdev, &ucm_fops);
1284	ucm_dev->cdev.owner = THIS_MODULE;
1285	kobject_set_name(&ucm_dev->cdev.kobj, "ucm%d", ucm_dev->devnum);
1286	if (cdev_add(&ucm_dev->cdev, base, 1))
1287		goto err;
1288
1289	ucm_dev->dev.class = &cm_class;
1290	ucm_dev->dev.parent = device->dma_device;
1291	ucm_dev->dev.devt = ucm_dev->cdev.dev;
1292	ucm_dev->dev.release = ib_ucm_release_dev;
1293	dev_set_name(&ucm_dev->dev, "ucm%d", ucm_dev->devnum);
1294	if (device_register(&ucm_dev->dev))
1295		goto err_cdev;
1296
1297	if (device_create_file(&ucm_dev->dev, &dev_attr_ibdev))
1298		goto err_dev;
1299
1300	ib_set_client_data(device, &ucm_client, ucm_dev);
1301	return;
1302
1303err_dev:
1304	device_unregister(&ucm_dev->dev);
1305err_cdev:
1306	cdev_del(&ucm_dev->cdev);
1307	if (ucm_dev->devnum < IB_UCM_MAX_DEVICES)
1308		clear_bit(devnum, dev_map);
1309	else
1310		clear_bit(devnum, overflow_map);
1311err:
1312	kfree(ucm_dev);
1313	return;
1314}
1315
1316static void ib_ucm_remove_one(struct ib_device *device, void *client_data)
1317{
1318	struct ib_ucm_device *ucm_dev = client_data;
1319
1320	if (!ucm_dev)
1321		return;
1322
1323	device_unregister(&ucm_dev->dev);
1324}
1325
1326static CLASS_ATTR_STRING(abi_version, S_IRUGO,
1327			 __stringify(IB_USER_CM_ABI_VERSION));
1328
1329static int __init ib_ucm_init(void)
1330{
1331	int ret;
1332
1333	ret = register_chrdev_region(IB_UCM_BASE_DEV, IB_UCM_MAX_DEVICES,
1334				     "infiniband_cm");
1335	if (ret) {
1336		pr_err("ucm: couldn't register device number\n");
1337		goto error1;
1338	}
1339
1340	ret = class_create_file(&cm_class, &class_attr_abi_version.attr);
1341	if (ret) {
1342		pr_err("ucm: couldn't create abi_version attribute\n");
1343		goto error2;
1344	}
1345
1346	ret = ib_register_client(&ucm_client);
1347	if (ret) {
1348		pr_err("ucm: couldn't register client\n");
1349		goto error3;
1350	}
1351	return 0;
1352
1353error3:
1354	class_remove_file(&cm_class, &class_attr_abi_version.attr);
1355error2:
1356	unregister_chrdev_region(IB_UCM_BASE_DEV, IB_UCM_MAX_DEVICES);
1357error1:
1358	return ret;
1359}
1360
1361static void __exit ib_ucm_cleanup(void)
1362{
1363	ib_unregister_client(&ucm_client);
1364	class_remove_file(&cm_class, &class_attr_abi_version.attr);
1365	unregister_chrdev_region(IB_UCM_BASE_DEV, IB_UCM_MAX_DEVICES);
1366	if (overflow_maj)
1367		unregister_chrdev_region(overflow_maj, IB_UCM_MAX_DEVICES);
1368	idr_destroy(&ctx_id_table);
1369}
1370
1371module_init(ib_ucm_init);
1372module_exit(ib_ucm_cleanup);