rdma_transport.c 7.6 KB
Newer Older
A
Andy Grover 已提交
1
/*
2
 * Copyright (c) 2009, 2018 Oracle and/or its affiliates. All rights reserved.
A
Andy Grover 已提交
3 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
 *
 * 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.
 *
 */
33
#include <linux/module.h>
A
Andy Grover 已提交
34 35
#include <rdma/rdma_cm.h>

36
#include "rds_single_path.h"
A
Andy Grover 已提交
37
#include "rdma_transport.h"
38
#include "ib.h"
A
Andy Grover 已提交
39

K
Ka-Cheong Poon 已提交
40
/* Global IPv4 and IPv6 RDS RDMA listener cm_id */
41
static struct rdma_cm_id *rds_rdma_listen_id;
K
Ka-Cheong Poon 已提交
42
static struct rdma_cm_id *rds6_rdma_listen_id;
A
Andy Grover 已提交
43

44 45 46
static int rds_rdma_cm_event_handler_cmn(struct rdma_cm_id *cm_id,
					 struct rdma_cm_event *event,
					 bool isv6)
A
Andy Grover 已提交
47 48 49 50 51 52
{
	/* this can be null in the listening path */
	struct rds_connection *conn = cm_id->context;
	struct rds_transport *trans;
	int ret = 0;

53
	rdsdebug("conn %p id %p handling event %u (%s)\n", conn, cm_id,
54
		 event->event, rdma_event_msg(event->event));
A
Andy Grover 已提交
55

56
	if (cm_id->device->node_type == RDMA_NODE_IB_CA)
A
Andy Grover 已提交
57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
		trans = &rds_ib_transport;

	/* Prevent shutdown from tearing down the connection
	 * while we're executing. */
	if (conn) {
		mutex_lock(&conn->c_cm_lock);

		/* If the connection is being shut down, bail out
		 * right away. We return 0 so cm_id doesn't get
		 * destroyed prematurely */
		if (rds_conn_state(conn) == RDS_CONN_DISCONNECTING) {
			/* Reject incoming connections while we're tearing
			 * down an existing one. */
			if (event->event == RDMA_CM_EVENT_CONNECT_REQUEST)
				ret = 1;
			goto out;
		}
	}

	switch (event->event) {
	case RDMA_CM_EVENT_CONNECT_REQUEST:
78
		ret = trans->cm_handle_connect(cm_id, event, isv6);
A
Andy Grover 已提交
79 80 81 82 83 84 85 86 87
		break;

	case RDMA_CM_EVENT_ADDR_RESOLVED:
		/* XXX do we need to clean up if this fails? */
		ret = rdma_resolve_route(cm_id,
					 RDS_RDMA_RESOLVE_TIMEOUT_MS);
		break;

	case RDMA_CM_EVENT_ROUTE_RESOLVED:
88 89 90 91 92 93 94 95
		/* Connection could have been dropped so make sure the
		 * cm_id is valid before proceeding
		 */
		if (conn) {
			struct rds_ib_connection *ibic;

			ibic = conn->c_transport_data;
			if (ibic && ibic->i_cm_id == cm_id)
96
				ret = trans->cm_initiate_connect(cm_id, isv6);
97 98 99
			else
				rds_conn_drop(conn);
		}
A
Andy Grover 已提交
100 101 102 103 104 105
		break;

	case RDMA_CM_EVENT_ESTABLISHED:
		trans->cm_connect_complete(conn, event);
		break;

106 107 108 109
	case RDMA_CM_EVENT_REJECTED:
		rdsdebug("Connection rejected: %s\n",
			 rdma_reject_msg(cm_id, event->status));
		/* FALLTHROUGH */
A
Andy Grover 已提交
110 111 112 113 114 115 116 117 118 119 120
	case RDMA_CM_EVENT_ADDR_ERROR:
	case RDMA_CM_EVENT_ROUTE_ERROR:
	case RDMA_CM_EVENT_CONNECT_ERROR:
	case RDMA_CM_EVENT_UNREACHABLE:
	case RDMA_CM_EVENT_DEVICE_REMOVAL:
	case RDMA_CM_EVENT_ADDR_CHANGE:
		if (conn)
			rds_conn_drop(conn);
		break;

	case RDMA_CM_EVENT_DISCONNECTED:
121
		rdsdebug("DISCONNECT event - dropping connection "
122
			 "%pI6c->%pI6c\n", &conn->c_laddr,
A
Andy Grover 已提交
123 124 125 126
			 &conn->c_faddr);
		rds_conn_drop(conn);
		break;

127 128
	case RDMA_CM_EVENT_TIMEWAIT_EXIT:
		if (conn) {
129
			pr_info("RDS: RDMA_CM_EVENT_TIMEWAIT_EXIT event: dropping connection %pI6c->%pI6c\n",
130 131 132 133 134
				&conn->c_laddr, &conn->c_faddr);
			rds_conn_drop(conn);
		}
		break;

A
Andy Grover 已提交
135 136
	default:
		/* things like device disconnect? */
137
		printk(KERN_ERR "RDS: unknown event %u (%s)!\n",
138
		       event->event, rdma_event_msg(event->event));
A
Andy Grover 已提交
139 140 141 142 143 144 145
		break;
	}

out:
	if (conn)
		mutex_unlock(&conn->c_cm_lock);

146
	rdsdebug("id %p event %u (%s) handling ret %d\n", cm_id, event->event,
147
		 rdma_event_msg(event->event), ret);
A
Andy Grover 已提交
148 149 150 151

	return ret;
}

152 153 154 155 156 157
int rds_rdma_cm_event_handler(struct rdma_cm_id *cm_id,
			      struct rdma_cm_event *event)
{
	return rds_rdma_cm_event_handler_cmn(cm_id, event, false);
}

K
Ka-Cheong Poon 已提交
158 159 160 161 162 163
int rds6_rdma_cm_event_handler(struct rdma_cm_id *cm_id,
			       struct rdma_cm_event *event)
{
	return rds_rdma_cm_event_handler_cmn(cm_id, event, true);
}

164 165 166
static int rds_rdma_listen_init_common(rdma_cm_event_handler handler,
				       struct sockaddr *sa,
				       struct rdma_cm_id **ret_cm_id)
A
Andy Grover 已提交
167 168 169 170
{
	struct rdma_cm_id *cm_id;
	int ret;

171
	cm_id = rdma_create_id(&init_net, handler, NULL,
172
			       RDMA_PS_TCP, IB_QPT_RC);
A
Andy Grover 已提交
173 174
	if (IS_ERR(cm_id)) {
		ret = PTR_ERR(cm_id);
175
		printk(KERN_ERR "RDS/RDMA: failed to setup listener, "
A
Andy Grover 已提交
176
		       "rdma_create_id() returned %d\n", ret);
177
		return ret;
A
Andy Grover 已提交
178 179 180 181 182 183
	}

	/*
	 * XXX I bet this binds the cm_id to a device.  If we want to support
	 * fail-over we'll have to take this into consideration.
	 */
184
	ret = rdma_bind_addr(cm_id, sa);
A
Andy Grover 已提交
185
	if (ret) {
186
		printk(KERN_ERR "RDS/RDMA: failed to setup listener, "
A
Andy Grover 已提交
187 188 189 190 191 192
		       "rdma_bind_addr() returned %d\n", ret);
		goto out;
	}

	ret = rdma_listen(cm_id, 128);
	if (ret) {
193
		printk(KERN_ERR "RDS/RDMA: failed to setup listener, "
A
Andy Grover 已提交
194 195 196 197 198 199
		       "rdma_listen() returned %d\n", ret);
		goto out;
	}

	rdsdebug("cm %p listening on port %u\n", cm_id, RDS_PORT);

200
	*ret_cm_id = cm_id;
A
Andy Grover 已提交
201 202 203 204 205 206 207
	cm_id = NULL;
out:
	if (cm_id)
		rdma_destroy_id(cm_id);
	return ret;
}

208 209 210 211 212 213 214 215 216
/* Initialize the RDS RDMA listeners.  We create two listeners for
 * compatibility reason.  The one on RDS_PORT is used for IPv4
 * requests only.  The one on RDS_CM_PORT is used for IPv6 requests
 * only.  So only IPv6 enabled RDS module will communicate using this
 * port.
 */
static int rds_rdma_listen_init(void)
{
	int ret;
K
Ka-Cheong Poon 已提交
217
	struct sockaddr_in6 sin6;
218 219 220 221 222 223 224 225
	struct sockaddr_in sin;

	sin.sin_family = PF_INET;
	sin.sin_addr.s_addr = htonl(INADDR_ANY);
	sin.sin_port = htons(RDS_PORT);
	ret = rds_rdma_listen_init_common(rds_rdma_cm_event_handler,
					  (struct sockaddr *)&sin,
					  &rds_rdma_listen_id);
K
Ka-Cheong Poon 已提交
226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
	if (ret != 0)
		return ret;

	sin6.sin6_family = PF_INET6;
	sin6.sin6_addr = in6addr_any;
	sin6.sin6_port = htons(RDS_CM_PORT);
	sin6.sin6_scope_id = 0;
	sin6.sin6_flowinfo = 0;
	ret = rds_rdma_listen_init_common(rds6_rdma_cm_event_handler,
					  (struct sockaddr *)&sin6,
					  &rds6_rdma_listen_id);
	/* Keep going even when IPv6 is not enabled in the system. */
	if (ret != 0)
		rdsdebug("Cannot set up IPv6 RDMA listener\n");
	return 0;
241 242
}

A
Andy Grover 已提交
243 244
static void rds_rdma_listen_stop(void)
{
245 246 247 248
	if (rds_rdma_listen_id) {
		rdsdebug("cm %p\n", rds_rdma_listen_id);
		rdma_destroy_id(rds_rdma_listen_id);
		rds_rdma_listen_id = NULL;
A
Andy Grover 已提交
249
	}
K
Ka-Cheong Poon 已提交
250 251 252 253 254
	if (rds6_rdma_listen_id) {
		rdsdebug("cm %p\n", rds6_rdma_listen_id);
		rdma_destroy_id(rds6_rdma_listen_id);
		rds6_rdma_listen_id = NULL;
	}
A
Andy Grover 已提交
255 256
}

257
static int rds_rdma_init(void)
A
Andy Grover 已提交
258 259 260
{
	int ret;

261
	ret = rds_ib_init();
A
Andy Grover 已提交
262 263 264
	if (ret)
		goto out;

265
	ret = rds_rdma_listen_init();
A
Andy Grover 已提交
266
	if (ret)
267
		rds_ib_exit();
A
Andy Grover 已提交
268 269 270
out:
	return ret;
}
271
module_init(rds_rdma_init);
A
Andy Grover 已提交
272

273
static void rds_rdma_exit(void)
A
Andy Grover 已提交
274 275 276 277 278
{
	/* stop listening first to ensure no new connections are attempted */
	rds_rdma_listen_stop();
	rds_ib_exit();
}
279 280 281
module_exit(rds_rdma_exit);

MODULE_AUTHOR("Oracle Corporation <rds-devel@oss.oracle.com>");
282
MODULE_DESCRIPTION("RDS: IB transport");
283
MODULE_LICENSE("Dual BSD/GPL");