af_smc.c 40.2 KB
Newer Older
U
Ursula Braun 已提交
1 2 3 4 5 6 7 8
/*
 *  Shared Memory Communications over RDMA (SMC-R) and RoCE
 *
 *  AF_SMC protocol family socket handler keeping the AF_INET sock address type
 *  applies to SOCK_STREAM sockets only
 *  offers an alternative communication option for TCP-protocol sockets
 *  applicable with RoCE-cards only
 *
9 10 11 12 13
 *  Initial restrictions:
 *    - support for alternate links postponed
 *    - partial support for non-blocking sockets only
 *    - support for urgent data postponed
 *
14
 *  Copyright IBM Corp. 2016, 2018
U
Ursula Braun 已提交
15 16 17 18 19 20 21 22 23 24
 *
 *  Author(s):  Ursula Braun <ubraun@linux.vnet.ibm.com>
 *              based on prototype from Frank Blaschka
 */

#define KMSG_COMPONENT "smc"
#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt

#include <linux/module.h>
#include <linux/socket.h>
25
#include <linux/workqueue.h>
26
#include <linux/in.h>
27 28
#include <linux/sched/signal.h>

U
Ursula Braun 已提交
29
#include <net/sock.h>
30
#include <net/tcp.h>
31
#include <net/smc.h>
32
#include <asm/ioctls.h>
U
Ursula Braun 已提交
33 34

#include "smc.h"
35
#include "smc_clc.h"
U
Ursula Braun 已提交
36
#include "smc_llc.h"
37
#include "smc_cdc.h"
38
#include "smc_core.h"
39
#include "smc_ib.h"
40
#include "smc_pnet.h"
U
Ursula Braun 已提交
41
#include "smc_tx.h"
U
Ursula Braun 已提交
42
#include "smc_rx.h"
43
#include "smc_close.h"
U
Ursula Braun 已提交
44

45 46 47 48 49 50 51 52 53
static DEFINE_MUTEX(smc_create_lgr_pending);	/* serialize link group
						 * creation
						 */

struct smc_lgr_list smc_lgr_list = {		/* established link groups */
	.lock = __SPIN_LOCK_UNLOCKED(smc_lgr_list.lock),
	.list = LIST_HEAD_INIT(smc_lgr_list.list),
};

54 55
static void smc_tcp_listen_work(struct work_struct *);

U
Ursula Braun 已提交
56 57 58 59 60 61 62
static void smc_set_keepalive(struct sock *sk, int val)
{
	struct smc_sock *smc = smc_sk(sk);

	smc->clcsock->sk->sk_prot->keepalive(smc->clcsock->sk, val);
}

63 64 65 66
static struct smc_hashinfo smc_v4_hashinfo = {
	.lock = __RW_LOCK_UNLOCKED(smc_v4_hashinfo.lock),
};

67 68 69 70
static struct smc_hashinfo smc_v6_hashinfo = {
	.lock = __RW_LOCK_UNLOCKED(smc_v6_hashinfo.lock),
};

71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
int smc_hash_sk(struct sock *sk)
{
	struct smc_hashinfo *h = sk->sk_prot->h.smc_hash;
	struct hlist_head *head;

	head = &h->ht;

	write_lock_bh(&h->lock);
	sk_add_node(sk, head);
	sock_prot_inuse_add(sock_net(sk), sk->sk_prot, 1);
	write_unlock_bh(&h->lock);

	return 0;
}
EXPORT_SYMBOL_GPL(smc_hash_sk);

void smc_unhash_sk(struct sock *sk)
{
	struct smc_hashinfo *h = sk->sk_prot->h.smc_hash;

	write_lock_bh(&h->lock);
	if (sk_del_node_init(sk))
		sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
	write_unlock_bh(&h->lock);
}
EXPORT_SYMBOL_GPL(smc_unhash_sk);

struct proto smc_proto = {
U
Ursula Braun 已提交
99 100 101
	.name		= "SMC",
	.owner		= THIS_MODULE,
	.keepalive	= smc_set_keepalive,
102 103
	.hash		= smc_hash_sk,
	.unhash		= smc_unhash_sk,
U
Ursula Braun 已提交
104
	.obj_size	= sizeof(struct smc_sock),
105
	.h.smc_hash	= &smc_v4_hashinfo,
106
	.slab_flags	= SLAB_TYPESAFE_BY_RCU,
U
Ursula Braun 已提交
107
};
108
EXPORT_SYMBOL_GPL(smc_proto);
U
Ursula Braun 已提交
109

110 111 112 113 114 115 116 117 118 119 120 121
struct proto smc_proto6 = {
	.name		= "SMC6",
	.owner		= THIS_MODULE,
	.keepalive	= smc_set_keepalive,
	.hash		= smc_hash_sk,
	.unhash		= smc_unhash_sk,
	.obj_size	= sizeof(struct smc_sock),
	.h.smc_hash	= &smc_v6_hashinfo,
	.slab_flags	= SLAB_TYPESAFE_BY_RCU,
};
EXPORT_SYMBOL_GPL(smc_proto6);

U
Ursula Braun 已提交
122 123 124 125
static int smc_release(struct socket *sock)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
126
	int rc = 0;
U
Ursula Braun 已提交
127 128 129 130 131

	if (!sk)
		goto out;

	smc = smc_sk(sk);
132 133 134 135 136 137 138
	if (sk->sk_state == SMC_LISTEN)
		/* smc_close_non_accepted() is called and acquires
		 * sock lock for child sockets again
		 */
		lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
	else
		lock_sock(sk);
U
Ursula Braun 已提交
139

140
	if (!smc->use_fallback) {
141 142 143 144
		rc = smc_close_active(smc);
		sock_set_flag(sk, SOCK_DEAD);
		sk->sk_shutdown |= SHUTDOWN_MASK;
	}
U
Ursula Braun 已提交
145 146 147 148
	if (smc->clcsock) {
		sock_release(smc->clcsock);
		smc->clcsock = NULL;
	}
149 150 151 152 153
	if (smc->use_fallback) {
		sock_put(sk); /* passive closing */
		sk->sk_state = SMC_CLOSED;
		sk->sk_state_change(sk);
	}
U
Ursula Braun 已提交
154 155 156 157

	/* detach socket */
	sock_orphan(sk);
	sock->sk = NULL;
158
	if (!smc->use_fallback && sk->sk_state == SMC_CLOSED)
159
		smc_conn_free(&smc->conn);
U
Ursula Braun 已提交
160 161
	release_sock(sk);

162 163
	sk->sk_prot->unhash(sk);
	sock_put(sk); /* final sock_put */
U
Ursula Braun 已提交
164
out:
165
	return rc;
U
Ursula Braun 已提交
166 167 168 169 170 171 172 173 174 175 176 177
}

static void smc_destruct(struct sock *sk)
{
	if (sk->sk_state != SMC_CLOSED)
		return;
	if (!sock_flag(sk, SOCK_DEAD))
		return;

	sk_refcnt_debug_dec(sk);
}

178 179
static struct sock *smc_sock_alloc(struct net *net, struct socket *sock,
				   int protocol)
U
Ursula Braun 已提交
180 181
{
	struct smc_sock *smc;
182
	struct proto *prot;
U
Ursula Braun 已提交
183 184
	struct sock *sk;

185 186
	prot = (protocol == SMCPROTO_SMC6) ? &smc_proto6 : &smc_proto;
	sk = sk_alloc(net, PF_SMC, GFP_KERNEL, prot, 0);
U
Ursula Braun 已提交
187 188 189 190 191 192
	if (!sk)
		return NULL;

	sock_init_data(sock, sk); /* sets sk_refcnt to 1 */
	sk->sk_state = SMC_INIT;
	sk->sk_destruct = smc_destruct;
193
	sk->sk_protocol = protocol;
U
Ursula Braun 已提交
194
	smc = smc_sk(sk);
195
	INIT_WORK(&smc->tcp_listen_work, smc_tcp_listen_work);
196
	INIT_DELAYED_WORK(&smc->conn.tx_work, smc_tx_work);
197 198
	INIT_LIST_HEAD(&smc->accept_q);
	spin_lock_init(&smc->accept_q_lock);
199
	spin_lock_init(&smc->conn.send_lock);
200
	sk->sk_prot->hash(sk);
201
	sk_refcnt_debug_inc(sk);
U
Ursula Braun 已提交
202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221

	return sk;
}

static int smc_bind(struct socket *sock, struct sockaddr *uaddr,
		    int addr_len)
{
	struct sockaddr_in *addr = (struct sockaddr_in *)uaddr;
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc;

	smc = smc_sk(sk);

	/* replicate tests from inet_bind(), to be safe wrt. future changes */
	rc = -EINVAL;
	if (addr_len < sizeof(struct sockaddr_in))
		goto out;

	rc = -EAFNOSUPPORT;
222 223 224 225
	if (addr->sin_family != AF_INET &&
	    addr->sin_family != AF_INET6 &&
	    addr->sin_family != AF_UNSPEC)
		goto out;
U
Ursula Braun 已提交
226
	/* accept AF_UNSPEC (mapped to AF_INET) only if s_addr is INADDR_ANY */
227 228
	if (addr->sin_family == AF_UNSPEC &&
	    addr->sin_addr.s_addr != htonl(INADDR_ANY))
U
Ursula Braun 已提交
229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297
		goto out;

	lock_sock(sk);

	/* Check if socket is already active */
	rc = -EINVAL;
	if (sk->sk_state != SMC_INIT)
		goto out_rel;

	smc->clcsock->sk->sk_reuse = sk->sk_reuse;
	rc = kernel_bind(smc->clcsock, uaddr, addr_len);

out_rel:
	release_sock(sk);
out:
	return rc;
}

static void smc_copy_sock_settings(struct sock *nsk, struct sock *osk,
				   unsigned long mask)
{
	/* options we don't get control via setsockopt for */
	nsk->sk_type = osk->sk_type;
	nsk->sk_sndbuf = osk->sk_sndbuf;
	nsk->sk_rcvbuf = osk->sk_rcvbuf;
	nsk->sk_sndtimeo = osk->sk_sndtimeo;
	nsk->sk_rcvtimeo = osk->sk_rcvtimeo;
	nsk->sk_mark = osk->sk_mark;
	nsk->sk_priority = osk->sk_priority;
	nsk->sk_rcvlowat = osk->sk_rcvlowat;
	nsk->sk_bound_dev_if = osk->sk_bound_dev_if;
	nsk->sk_err = osk->sk_err;

	nsk->sk_flags &= ~mask;
	nsk->sk_flags |= osk->sk_flags & mask;
}

#define SK_FLAGS_SMC_TO_CLC ((1UL << SOCK_URGINLINE) | \
			     (1UL << SOCK_KEEPOPEN) | \
			     (1UL << SOCK_LINGER) | \
			     (1UL << SOCK_BROADCAST) | \
			     (1UL << SOCK_TIMESTAMP) | \
			     (1UL << SOCK_DBG) | \
			     (1UL << SOCK_RCVTSTAMP) | \
			     (1UL << SOCK_RCVTSTAMPNS) | \
			     (1UL << SOCK_LOCALROUTE) | \
			     (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE) | \
			     (1UL << SOCK_RXQ_OVFL) | \
			     (1UL << SOCK_WIFI_STATUS) | \
			     (1UL << SOCK_NOFCS) | \
			     (1UL << SOCK_FILTER_LOCKED))
/* copy only relevant settings and flags of SOL_SOCKET level from smc to
 * clc socket (since smc is not called for these options from net/core)
 */
static void smc_copy_sock_settings_to_clc(struct smc_sock *smc)
{
	smc_copy_sock_settings(smc->clcsock->sk, &smc->sk, SK_FLAGS_SMC_TO_CLC);
}

#define SK_FLAGS_CLC_TO_SMC ((1UL << SOCK_URGINLINE) | \
			     (1UL << SOCK_KEEPOPEN) | \
			     (1UL << SOCK_LINGER) | \
			     (1UL << SOCK_DBG))
/* copy only settings and flags relevant for smc from clc to smc socket */
static void smc_copy_sock_settings_to_smc(struct smc_sock *smc)
{
	smc_copy_sock_settings(&smc->sk, smc->clcsock->sk, SK_FLAGS_CLC_TO_SMC);
}

298 299 300
/* register a new rmb, optionally send confirm_rkey msg to register with peer */
static int smc_reg_rmb(struct smc_link *link, struct smc_buf_desc *rmb_desc,
		       bool conf_rkey)
K
Karsten Graul 已提交
301 302
{
	/* register memory region for new rmb */
303 304
	if (smc_wr_reg_send(link, rmb_desc->mr_rx[SMC_SINGLE_LINK])) {
		rmb_desc->regerr = 1;
K
Karsten Graul 已提交
305
		return -EFAULT;
306
	}
307 308 309 310 311 312 313
	if (!conf_rkey)
		return 0;
	/* exchange confirm_rkey msg with peer */
	if (smc_llc_do_confirm_rkey(link, rmb_desc)) {
		rmb_desc->regerr = 1;
		return -EFAULT;
	}
K
Karsten Graul 已提交
314 315 316
	return 0;
}

317
static int smc_clnt_conf_first_link(struct smc_sock *smc)
U
Ursula Braun 已提交
318
{
319
	struct net *net = sock_net(smc->clcsock->sk);
U
Ursula Braun 已提交
320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337
	struct smc_link_group *lgr = smc->conn.lgr;
	struct smc_link *link;
	int rest;
	int rc;

	link = &lgr->lnk[SMC_SINGLE_LINK];
	/* receive CONFIRM LINK request from server over RoCE fabric */
	rest = wait_for_completion_interruptible_timeout(
		&link->llc_confirm,
		SMC_LLC_WAIT_FIRST_TIME);
	if (rest <= 0) {
		struct smc_clc_msg_decline dclc;

		rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc),
				      SMC_CLC_DECLINE);
		return rc;
	}

338 339 340
	if (link->llc_confirm_rc)
		return SMC_CLC_DECL_RMBE_EC;

U
Ursula Braun 已提交
341 342 343 344 345
	rc = smc_ib_modify_qp_rts(link);
	if (rc)
		return SMC_CLC_DECL_INTERR;

	smc_wr_remember_qp_attr(link);
346

347
	if (smc_reg_rmb(link, smc->conn.rmb_desc, false))
348 349
		return SMC_CLC_DECL_INTERR;

U
Ursula Braun 已提交
350 351 352
	/* send CONFIRM LINK response over RoCE fabric */
	rc = smc_llc_send_confirm_link(link,
				       link->smcibdev->mac[link->ibport - 1],
353 354
				       &link->smcibdev->gid[link->ibport - 1],
				       SMC_LLC_RESP);
U
Ursula Braun 已提交
355 356 357
	if (rc < 0)
		return SMC_CLC_DECL_TCL;

358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376
	/* receive ADD LINK request from server over RoCE fabric */
	rest = wait_for_completion_interruptible_timeout(&link->llc_add,
							 SMC_LLC_WAIT_TIME);
	if (rest <= 0) {
		struct smc_clc_msg_decline dclc;

		rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc),
				      SMC_CLC_DECLINE);
		return rc;
	}

	/* send add link reject message, only one link supported for now */
	rc = smc_llc_send_add_link(link,
				   link->smcibdev->mac[link->ibport - 1],
				   &link->smcibdev->gid[link->ibport - 1],
				   SMC_LLC_RESP);
	if (rc < 0)
		return SMC_CLC_DECL_TCL;

377
	smc_llc_link_active(link, net->ipv4.sysctl_tcp_keepalive_time);
378

379
	return 0;
U
Ursula Braun 已提交
380 381
}

382 383 384 385
static void smc_conn_save_peer_info(struct smc_sock *smc,
				    struct smc_clc_msg_accept_confirm *clc)
{
	smc->conn.peer_conn_idx = clc->conn_idx;
386
	smc->conn.local_tx_ctrl.token = ntohl(clc->rmbe_alert_token);
U
Ursula Braun 已提交
387 388
	smc->conn.peer_rmbe_size = smc_uncompress_bufsize(clc->rmbe_size);
	atomic_set(&smc->conn.peer_rmbe_space, smc->conn.peer_rmbe_size);
389 390 391 392 393 394 395 396 397 398 399 400
}

static void smc_link_save_peer_info(struct smc_link *link,
				    struct smc_clc_msg_accept_confirm *clc)
{
	link->peer_qpn = ntoh24(clc->qpn);
	memcpy(link->peer_gid, clc->lcl.gid, SMC_GID_SIZE);
	memcpy(link->peer_mac, clc->lcl.mac, sizeof(link->peer_mac));
	link->peer_psn = ntoh24(clc->psn);
	link->peer_mtu = clc->qp_mtu;
}

401 402 403 404
/* setup for RDMA connection of client */
static int smc_connect_rdma(struct smc_sock *smc)
{
	struct smc_clc_msg_accept_confirm aclc;
405
	int local_contact = SMC_FIRST_CONTACT;
406
	struct smc_ib_device *smcibdev;
407 408
	struct smc_link *link;
	u8 srv_first_contact;
409 410 411 412
	int reason_code = 0;
	int rc = 0;
	u8 ibport;

413 414
	sock_hold(&smc->sk); /* sock put in passive closing */

415 416 417
	if (smc->use_fallback)
		goto out_connected;

U
Ursula Braun 已提交
418 419 420 421 422 423
	if (!tcp_sk(smc->clcsock->sk)->syn_smc) {
		/* peer has not signalled SMC-capability */
		smc->use_fallback = true;
		goto out_connected;
	}

424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457
	/* IPSec connections opt out of SMC-R optimizations */
	if (using_ipsec(smc)) {
		reason_code = SMC_CLC_DECL_IPSEC;
		goto decline_rdma;
	}

	/* PNET table look up: search active ib_device and port
	 * within same PNETID that also contains the ethernet device
	 * used for the internal TCP socket
	 */
	smc_pnet_find_roce_resource(smc->clcsock->sk, &smcibdev, &ibport);
	if (!smcibdev) {
		reason_code = SMC_CLC_DECL_CNFERR; /* configuration error */
		goto decline_rdma;
	}

	/* do inband token exchange */
	reason_code = smc_clc_send_proposal(smc, smcibdev, ibport);
	if (reason_code < 0) {
		rc = reason_code;
		goto out_err;
	}
	if (reason_code > 0) /* configuration error */
		goto decline_rdma;
	/* receive SMC Accept CLC message */
	reason_code = smc_clc_wait_msg(smc, &aclc, sizeof(aclc),
				       SMC_CLC_ACCEPT);
	if (reason_code < 0) {
		rc = reason_code;
		goto out_err;
	}
	if (reason_code > 0)
		goto decline_rdma;

458 459
	srv_first_contact = aclc.hdr.flag;
	mutex_lock(&smc_create_lgr_pending);
460 461
	local_contact = smc_conn_create(smc, smcibdev, ibport, &aclc.lcl,
					srv_first_contact);
462 463 464 465 466 467
	if (local_contact < 0) {
		rc = local_contact;
		if (rc == -ENOMEM)
			reason_code = SMC_CLC_DECL_MEM;/* insufficient memory*/
		else if (rc == -ENOLINK)
			reason_code = SMC_CLC_DECL_SYNCERR; /* synchr. error */
468 469
		else
			reason_code = SMC_CLC_DECL_INTERR; /* other error */
470 471 472
		goto decline_rdma_unlock;
	}
	link = &smc->conn.lgr->lnk[SMC_SINGLE_LINK];
473

474
	smc_conn_save_peer_info(smc, &aclc);
U
Ursula Braun 已提交
475

476 477
	/* create send buffer and rmb */
	rc = smc_buf_create(smc);
U
Ursula Braun 已提交
478 479 480 481 482
	if (rc) {
		reason_code = SMC_CLC_DECL_MEM;
		goto decline_rdma_unlock;
	}

483 484
	if (local_contact == SMC_FIRST_CONTACT)
		smc_link_save_peer_info(link, &aclc);
485 486 487 488 489 490 491

	rc = smc_rmb_rtoken_handling(&smc->conn, &aclc);
	if (rc) {
		reason_code = SMC_CLC_DECL_INTERR;
		goto decline_rdma_unlock;
	}

492 493 494
	smc_close_init(smc);
	smc_rx_init(smc);

495 496 497 498 499 500
	if (local_contact == SMC_FIRST_CONTACT) {
		rc = smc_ib_ready_link(link);
		if (rc) {
			reason_code = SMC_CLC_DECL_INTERR;
			goto decline_rdma_unlock;
		}
501
	} else {
K
Karsten Graul 已提交
502
		if (!smc->conn.rmb_desc->reused) {
503
			if (smc_reg_rmb(link, smc->conn.rmb_desc, true)) {
504 505 506 507
				reason_code = SMC_CLC_DECL_INTERR;
				goto decline_rdma_unlock;
			}
		}
508
	}
509
	smc_rmb_sync_sg_for_device(&smc->conn);
510 511 512

	rc = smc_clc_send_confirm(smc);
	if (rc)
513
		goto out_err_unlock;
514

U
Ursula Braun 已提交
515 516
	if (local_contact == SMC_FIRST_CONTACT) {
		/* QP confirmation over RoCE fabric */
517
		reason_code = smc_clnt_conf_first_link(smc);
U
Ursula Braun 已提交
518 519 520 521 522 523 524
		if (reason_code < 0) {
			rc = reason_code;
			goto out_err_unlock;
		}
		if (reason_code > 0)
			goto decline_rdma_unlock;
	}
525

526
	mutex_unlock(&smc_create_lgr_pending);
U
Ursula Braun 已提交
527 528
	smc_tx_init(smc);

529 530
out_connected:
	smc_copy_sock_settings_to_clc(smc);
531 532
	if (smc->sk.sk_state == SMC_INIT)
		smc->sk.sk_state = SMC_ACTIVE;
533

534
	return rc ? rc : local_contact;
535

536
decline_rdma_unlock:
537 538
	if (local_contact == SMC_FIRST_CONTACT)
		smc_lgr_forget(smc->conn.lgr);
539 540
	mutex_unlock(&smc_create_lgr_pending);
	smc_conn_free(&smc->conn);
541 542 543 544
decline_rdma:
	/* RDMA setup failed, switch back to TCP */
	smc->use_fallback = true;
	if (reason_code && (reason_code != SMC_CLC_DECL_REPLY)) {
545
		rc = smc_clc_send_decline(smc, reason_code);
546
		if (rc < 0)
547 548 549 550
			goto out_err;
	}
	goto out_connected;

551
out_err_unlock:
552 553
	if (local_contact == SMC_FIRST_CONTACT)
		smc_lgr_forget(smc->conn.lgr);
554 555
	mutex_unlock(&smc_create_lgr_pending);
	smc_conn_free(&smc->conn);
556
out_err:
557 558
	if (smc->sk.sk_state == SMC_INIT)
		sock_put(&smc->sk); /* passive closing */
559 560 561
	return rc;
}

U
Ursula Braun 已提交
562 563 564 565 566 567 568 569 570 571 572 573
static int smc_connect(struct socket *sock, struct sockaddr *addr,
		       int alen, int flags)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -EINVAL;

	smc = smc_sk(sk);

	/* separate smc parameter checking to be safe */
	if (alen < sizeof(addr->sa_family))
		goto out_err;
574
	if (addr->sa_family != AF_INET && addr->sa_family != AF_INET6)
U
Ursula Braun 已提交
575 576 577 578 579 580 581 582 583 584 585 586 587 588 589
		goto out_err;

	lock_sock(sk);
	switch (sk->sk_state) {
	default:
		goto out;
	case SMC_ACTIVE:
		rc = -EISCONN;
		goto out;
	case SMC_INIT:
		rc = 0;
		break;
	}

	smc_copy_sock_settings_to_clc(smc);
U
Ursula Braun 已提交
590
	tcp_sk(smc->clcsock->sk)->syn_smc = 1;
U
Ursula Braun 已提交
591 592 593 594
	rc = kernel_connect(smc->clcsock, addr, alen, flags);
	if (rc)
		goto out;

595 596 597 598 599 600
	/* setup RDMA connection */
	rc = smc_connect_rdma(smc);
	if (rc < 0)
		goto out;
	else
		rc = 0; /* success cases including fallback */
U
Ursula Braun 已提交
601 602 603 604 605 606 607 608 609

out:
	release_sock(sk);
out_err:
	return rc;
}

static int smc_clcsock_accept(struct smc_sock *lsmc, struct smc_sock **new_smc)
{
610 611
	struct socket *new_clcsock = NULL;
	struct sock *lsk = &lsmc->sk;
U
Ursula Braun 已提交
612 613 614
	struct sock *new_sk;
	int rc;

615
	release_sock(lsk);
616
	new_sk = smc_sock_alloc(sock_net(lsk), NULL, lsk->sk_protocol);
U
Ursula Braun 已提交
617 618
	if (!new_sk) {
		rc = -ENOMEM;
619
		lsk->sk_err = ENOMEM;
U
Ursula Braun 已提交
620
		*new_smc = NULL;
621
		lock_sock(lsk);
U
Ursula Braun 已提交
622 623 624 625 626
		goto out;
	}
	*new_smc = smc_sk(new_sk);

	rc = kernel_accept(lsmc->clcsock, &new_clcsock, 0);
627
	lock_sock(lsk);
628
	if  (rc < 0)
629
		lsk->sk_err = -rc;
630
	if (rc < 0 || lsk->sk_state == SMC_CLOSED) {
631 632 633 634
		if (new_clcsock)
			sock_release(new_clcsock);
		new_sk->sk_state = SMC_CLOSED;
		sock_set_flag(new_sk, SOCK_DEAD);
635
		new_sk->sk_prot->unhash(new_sk);
636
		sock_put(new_sk); /* final */
U
Ursula Braun 已提交
637 638 639 640 641 642 643 644 645
		*new_smc = NULL;
		goto out;
	}

	(*new_smc)->clcsock = new_clcsock;
out:
	return rc;
}

646 647 648 649 650 651 652
/* add a just created sock to the accept queue of the listen sock as
 * candidate for a following socket accept call from user space
 */
static void smc_accept_enqueue(struct sock *parent, struct sock *sk)
{
	struct smc_sock *par = smc_sk(parent);

653
	sock_hold(sk); /* sock_put in smc_accept_unlink () */
654 655 656 657 658 659 660 661 662 663 664 665 666 667 668
	spin_lock(&par->accept_q_lock);
	list_add_tail(&smc_sk(sk)->accept_q, &par->accept_q);
	spin_unlock(&par->accept_q_lock);
	sk_acceptq_added(parent);
}

/* remove a socket from the accept queue of its parental listening socket */
static void smc_accept_unlink(struct sock *sk)
{
	struct smc_sock *par = smc_sk(sk)->listen_smc;

	spin_lock(&par->accept_q_lock);
	list_del_init(&smc_sk(sk)->accept_q);
	spin_unlock(&par->accept_q_lock);
	sk_acceptq_removed(&smc_sk(sk)->listen_smc->sk);
669
	sock_put(sk); /* sock_hold in smc_accept_enqueue */
670 671 672 673 674
}

/* remove a sock from the accept queue to bind it to a new socket created
 * for a socket accept call from user space
 */
675 676
struct sock *smc_accept_dequeue(struct sock *parent,
				struct socket *new_sock)
677 678 679 680 681 682 683 684 685
{
	struct smc_sock *isk, *n;
	struct sock *new_sk;

	list_for_each_entry_safe(isk, n, &smc_sk(parent)->accept_q, accept_q) {
		new_sk = (struct sock *)isk;

		smc_accept_unlink(new_sk);
		if (new_sk->sk_state == SMC_CLOSED) {
686 687 688 689
			if (isk->clcsock) {
				sock_release(isk->clcsock);
				isk->clcsock = NULL;
			}
690
			new_sk->sk_prot->unhash(new_sk);
691
			sock_put(new_sk); /* final */
692 693 694 695 696 697 698 699 700 701
			continue;
		}
		if (new_sock)
			sock_graft(new_sk, new_sock);
		return new_sk;
	}
	return NULL;
}

/* clean up for a created but never accepted sock */
702
void smc_close_non_accepted(struct sock *sk)
703 704 705
{
	struct smc_sock *smc = smc_sk(sk);

706 707 708 709
	lock_sock(sk);
	if (!sk->sk_lingertime)
		/* wait for peer closing */
		sk->sk_lingertime = SMC_MAX_STREAM_WAIT_TIMEOUT;
710
	if (!smc->use_fallback) {
711
		smc_close_active(smc);
712 713 714
		sock_set_flag(sk, SOCK_DEAD);
		sk->sk_shutdown |= SHUTDOWN_MASK;
	}
715 716 717 718 719 720 721
	if (smc->clcsock) {
		struct socket *tcp;

		tcp = smc->clcsock;
		smc->clcsock = NULL;
		sock_release(tcp);
	}
722
	if (smc->use_fallback) {
723 724 725 726 727
		sock_put(sk); /* passive closing */
		sk->sk_state = SMC_CLOSED;
	} else {
		if (sk->sk_state == SMC_CLOSED)
			smc_conn_free(&smc->conn);
728 729
	}
	release_sock(sk);
730 731
	sk->sk_prot->unhash(sk);
	sock_put(sk); /* final sock_put */
732 733
}

U
Ursula Braun 已提交
734 735
static int smc_serv_conf_first_link(struct smc_sock *smc)
{
736
	struct net *net = sock_net(smc->clcsock->sk);
U
Ursula Braun 已提交
737 738 739 740 741 742
	struct smc_link_group *lgr = smc->conn.lgr;
	struct smc_link *link;
	int rest;
	int rc;

	link = &lgr->lnk[SMC_SINGLE_LINK];
743

744
	if (smc_reg_rmb(link, smc->conn.rmb_desc, false))
745 746
		return SMC_CLC_DECL_INTERR;

U
Ursula Braun 已提交
747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763
	/* send CONFIRM LINK request to client over the RoCE fabric */
	rc = smc_llc_send_confirm_link(link,
				       link->smcibdev->mac[link->ibport - 1],
				       &link->smcibdev->gid[link->ibport - 1],
				       SMC_LLC_REQ);
	if (rc < 0)
		return SMC_CLC_DECL_TCL;

	/* receive CONFIRM LINK response from client over the RoCE fabric */
	rest = wait_for_completion_interruptible_timeout(
		&link->llc_confirm_resp,
		SMC_LLC_WAIT_FIRST_TIME);
	if (rest <= 0) {
		struct smc_clc_msg_decline dclc;

		rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc),
				      SMC_CLC_DECLINE);
764
		return rc;
U
Ursula Braun 已提交
765 766
	}

767 768 769
	if (link->llc_confirm_resp_rc)
		return SMC_CLC_DECL_RMBE_EC;

770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788
	/* send ADD LINK request to client over the RoCE fabric */
	rc = smc_llc_send_add_link(link,
				   link->smcibdev->mac[link->ibport - 1],
				   &link->smcibdev->gid[link->ibport - 1],
				   SMC_LLC_REQ);
	if (rc < 0)
		return SMC_CLC_DECL_TCL;

	/* receive ADD LINK response from client over the RoCE fabric */
	rest = wait_for_completion_interruptible_timeout(&link->llc_add_resp,
							 SMC_LLC_WAIT_TIME);
	if (rest <= 0) {
		struct smc_clc_msg_decline dclc;

		rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc),
				      SMC_CLC_DECLINE);
		return rc;
	}

789
	smc_llc_link_active(link, net->ipv4.sysctl_tcp_keepalive_time);
790

791
	return 0;
U
Ursula Braun 已提交
792 793
}

794 795 796 797 798
/* setup for RDMA connection of server */
static void smc_listen_work(struct work_struct *work)
{
	struct smc_sock *new_smc = container_of(work, struct smc_sock,
						smc_listen_work);
799
	struct smc_clc_msg_proposal_prefix *pclc_prfx;
800 801 802
	struct socket *newclcsock = new_smc->clcsock;
	struct smc_sock *lsmc = new_smc->listen_smc;
	struct smc_clc_msg_accept_confirm cclc;
803
	int local_contact = SMC_REUSE_CONTACT;
804
	struct sock *newsmcsk = &new_smc->sk;
805
	struct smc_clc_msg_proposal *pclc;
806
	struct smc_ib_device *smcibdev;
807
	u8 buf[SMC_CLC_MAX_LEN];
808
	struct smc_link *link;
809
	int reason_code = 0;
810
	int rc = 0;
811 812
	u8 ibport;

813 814 815
	if (new_smc->use_fallback)
		goto out_connected;

U
Ursula Braun 已提交
816 817 818 819 820 821
	/* check if peer is smc capable */
	if (!tcp_sk(newclcsock->sk)->syn_smc) {
		new_smc->use_fallback = true;
		goto out_connected;
	}

822 823 824
	/* do inband token exchange -
	 *wait for and receive SMC Proposal CLC message
	 */
825
	reason_code = smc_clc_wait_msg(new_smc, &buf, sizeof(buf),
826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847
				       SMC_CLC_PROPOSAL);
	if (reason_code < 0)
		goto out_err;
	if (reason_code > 0)
		goto decline_rdma;

	/* IPSec connections opt out of SMC-R optimizations */
	if (using_ipsec(new_smc)) {
		reason_code = SMC_CLC_DECL_IPSEC;
		goto decline_rdma;
	}

	/* PNET table look up: search active ib_device and port
	 * within same PNETID that also contains the ethernet device
	 * used for the internal TCP socket
	 */
	smc_pnet_find_roce_resource(newclcsock->sk, &smcibdev, &ibport);
	if (!smcibdev) {
		reason_code = SMC_CLC_DECL_CNFERR; /* configuration error */
		goto decline_rdma;
	}

848 849
	pclc = (struct smc_clc_msg_proposal *)&buf;
	pclc_prfx = smc_clc_proposal_get_prefix(pclc);
850 851 852

	rc = smc_clc_prfx_match(newclcsock, pclc_prfx);
	if (rc) {
853 854 855 856
		reason_code = SMC_CLC_DECL_CNFERR; /* configuration error */
		goto decline_rdma;
	}

857 858
	/* allocate connection / link group */
	mutex_lock(&smc_create_lgr_pending);
859 860
	local_contact = smc_conn_create(new_smc, smcibdev, ibport, &pclc->lcl,
					0);
861 862 863 864
	if (local_contact < 0) {
		rc = local_contact;
		if (rc == -ENOMEM)
			reason_code = SMC_CLC_DECL_MEM;/* insufficient memory*/
865
		goto decline_rdma_unlock;
866 867
	}
	link = &new_smc->conn.lgr->lnk[SMC_SINGLE_LINK];
868

869 870
	/* create send buffer and rmb */
	rc = smc_buf_create(new_smc);
U
Ursula Braun 已提交
871 872
	if (rc) {
		reason_code = SMC_CLC_DECL_MEM;
873
		goto decline_rdma_unlock;
U
Ursula Braun 已提交
874
	}
875

876 877 878
	smc_close_init(new_smc);
	smc_rx_init(new_smc);

879
	if (local_contact != SMC_FIRST_CONTACT) {
K
Karsten Graul 已提交
880
		if (!new_smc->conn.rmb_desc->reused) {
881
			if (smc_reg_rmb(link, new_smc->conn.rmb_desc, true)) {
882
				reason_code = SMC_CLC_DECL_INTERR;
883
				goto decline_rdma_unlock;
884 885 886
			}
		}
	}
887
	smc_rmb_sync_sg_for_device(&new_smc->conn);
888

889
	rc = smc_clc_send_accept(new_smc, local_contact);
890
	if (rc)
891
		goto out_err_unlock;
892 893 894 895 896

	/* receive SMC Confirm CLC message */
	reason_code = smc_clc_wait_msg(new_smc, &cclc, sizeof(cclc),
				       SMC_CLC_CONFIRM);
	if (reason_code < 0)
897
		goto out_err_unlock;
898
	if (reason_code > 0)
899
		goto decline_rdma_unlock;
900 901 902
	smc_conn_save_peer_info(new_smc, &cclc);
	if (local_contact == SMC_FIRST_CONTACT)
		smc_link_save_peer_info(link, &cclc);
903

904 905 906
	rc = smc_rmb_rtoken_handling(&new_smc->conn, &cclc);
	if (rc) {
		reason_code = SMC_CLC_DECL_INTERR;
907
		goto decline_rdma_unlock;
908 909 910 911 912 913
	}

	if (local_contact == SMC_FIRST_CONTACT) {
		rc = smc_ib_ready_link(link);
		if (rc) {
			reason_code = SMC_CLC_DECL_INTERR;
914
			goto decline_rdma_unlock;
915
		}
U
Ursula Braun 已提交
916 917
		/* QP confirmation over RoCE fabric */
		reason_code = smc_serv_conf_first_link(new_smc);
918
		if (reason_code < 0)
U
Ursula Braun 已提交
919
			/* peer is not aware of a problem */
920
			goto out_err_unlock;
U
Ursula Braun 已提交
921
		if (reason_code > 0)
922
			goto decline_rdma_unlock;
923
	}
924

U
Ursula Braun 已提交
925
	smc_tx_init(new_smc);
926
	mutex_unlock(&smc_create_lgr_pending);
U
Ursula Braun 已提交
927

928 929
out_connected:
	sk_refcnt_debug_inc(newsmcsk);
930 931
	if (newsmcsk->sk_state == SMC_INIT)
		newsmcsk->sk_state = SMC_ACTIVE;
932
enqueue:
933
	lock_sock_nested(&lsmc->sk, SINGLE_DEPTH_NESTING);
934 935 936 937 938 939 940 941 942 943 944 945
	if (lsmc->sk.sk_state == SMC_LISTEN) {
		smc_accept_enqueue(&lsmc->sk, newsmcsk);
	} else { /* no longer listening */
		smc_close_non_accepted(newsmcsk);
	}
	release_sock(&lsmc->sk);

	/* Wake up accept */
	lsmc->sk.sk_data_ready(&lsmc->sk);
	sock_put(&lsmc->sk); /* sock_hold in smc_tcp_listen_work */
	return;

946
decline_rdma_unlock:
947 948
	if (local_contact == SMC_FIRST_CONTACT)
		smc_lgr_forget(new_smc->conn.lgr);
949
	mutex_unlock(&smc_create_lgr_pending);
950 951
decline_rdma:
	/* RDMA setup failed, switch back to TCP */
952
	smc_conn_free(&new_smc->conn);
953 954
	new_smc->use_fallback = true;
	if (reason_code && (reason_code != SMC_CLC_DECL_REPLY)) {
955
		if (smc_clc_send_decline(new_smc, reason_code) < 0)
956 957 958 959
			goto out_err;
	}
	goto out_connected;

960
out_err_unlock:
961 962
	if (local_contact == SMC_FIRST_CONTACT)
		smc_lgr_forget(new_smc->conn.lgr);
963
	mutex_unlock(&smc_create_lgr_pending);
964
out_err:
965 966
	if (newsmcsk->sk_state == SMC_INIT)
		sock_put(&new_smc->sk); /* passive closing */
967
	newsmcsk->sk_state = SMC_CLOSED;
968
	smc_conn_free(&new_smc->conn);
969 970 971 972 973 974 975
	goto enqueue; /* queue new sock with sk_err set */
}

static void smc_tcp_listen_work(struct work_struct *work)
{
	struct smc_sock *lsmc = container_of(work, struct smc_sock,
					     tcp_listen_work);
976
	struct sock *lsk = &lsmc->sk;
977 978 979
	struct smc_sock *new_smc;
	int rc = 0;

980 981
	lock_sock(lsk);
	while (lsk->sk_state == SMC_LISTEN) {
982 983 984 985 986 987 988
		rc = smc_clcsock_accept(lsmc, &new_smc);
		if (rc)
			goto out;
		if (!new_smc)
			continue;

		new_smc->listen_smc = lsmc;
989
		new_smc->use_fallback = lsmc->use_fallback;
990
		sock_hold(lsk); /* sock_put in smc_listen_work */
991 992
		INIT_WORK(&new_smc->smc_listen_work, smc_listen_work);
		smc_copy_sock_settings_to_smc(new_smc);
993 994 995
		sock_hold(&new_smc->sk); /* sock_put in passive closing */
		if (!schedule_work(&new_smc->smc_listen_work))
			sock_put(&new_smc->sk);
996 997 998
	}

out:
999
	release_sock(lsk);
1000
	sock_put(&lsmc->sk); /* sock_hold in smc_listen */
1001 1002
}

U
Ursula Braun 已提交
1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024
static int smc_listen(struct socket *sock, int backlog)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc;

	smc = smc_sk(sk);
	lock_sock(sk);

	rc = -EINVAL;
	if ((sk->sk_state != SMC_INIT) && (sk->sk_state != SMC_LISTEN))
		goto out;

	rc = 0;
	if (sk->sk_state == SMC_LISTEN) {
		sk->sk_max_ack_backlog = backlog;
		goto out;
	}
	/* some socket options are handled in core, so we could not apply
	 * them to the clc socket -- copy smc socket options to clc socket
	 */
	smc_copy_sock_settings_to_clc(smc);
1025 1026
	if (!smc->use_fallback)
		tcp_sk(smc->clcsock->sk)->syn_smc = 1;
U
Ursula Braun 已提交
1027 1028 1029 1030 1031 1032 1033

	rc = kernel_listen(smc->clcsock, backlog);
	if (rc)
		goto out;
	sk->sk_max_ack_backlog = backlog;
	sk->sk_ack_backlog = 0;
	sk->sk_state = SMC_LISTEN;
1034
	INIT_WORK(&smc->tcp_listen_work, smc_tcp_listen_work);
1035 1036 1037
	sock_hold(sk); /* sock_hold in tcp_listen_worker */
	if (!schedule_work(&smc->tcp_listen_work))
		sock_put(sk);
U
Ursula Braun 已提交
1038 1039 1040 1041 1042 1043 1044

out:
	release_sock(sk);
	return rc;
}

static int smc_accept(struct socket *sock, struct socket *new_sock,
1045
		      int flags, bool kern)
U
Ursula Braun 已提交
1046
{
1047 1048
	struct sock *sk = sock->sk, *nsk;
	DECLARE_WAITQUEUE(wait, current);
U
Ursula Braun 已提交
1049
	struct smc_sock *lsmc;
1050 1051
	long timeo;
	int rc = 0;
U
Ursula Braun 已提交
1052 1053

	lsmc = smc_sk(sk);
1054
	sock_hold(sk); /* sock_put below */
U
Ursula Braun 已提交
1055 1056 1057 1058
	lock_sock(sk);

	if (lsmc->sk.sk_state != SMC_LISTEN) {
		rc = -EINVAL;
1059
		release_sock(sk);
U
Ursula Braun 已提交
1060 1061 1062
		goto out;
	}

1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083
	/* Wait for an incoming connection */
	timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
	add_wait_queue_exclusive(sk_sleep(sk), &wait);
	while (!(nsk = smc_accept_dequeue(sk, new_sock))) {
		set_current_state(TASK_INTERRUPTIBLE);
		if (!timeo) {
			rc = -EAGAIN;
			break;
		}
		release_sock(sk);
		timeo = schedule_timeout(timeo);
		/* wakeup by sk_data_ready in smc_listen_work() */
		sched_annotate_sleep();
		lock_sock(sk);
		if (signal_pending(current)) {
			rc = sock_intr_errno(timeo);
			break;
		}
	}
	set_current_state(TASK_RUNNING);
	remove_wait_queue(sk_sleep(sk), &wait);
U
Ursula Braun 已提交
1084

1085 1086
	if (!rc)
		rc = sock_error(nsk);
1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103
	release_sock(sk);
	if (rc)
		goto out;

	if (lsmc->sockopt_defer_accept && !(flags & O_NONBLOCK)) {
		/* wait till data arrives on the socket */
		timeo = msecs_to_jiffies(lsmc->sockopt_defer_accept *
								MSEC_PER_SEC);
		if (smc_sk(nsk)->use_fallback) {
			struct sock *clcsk = smc_sk(nsk)->clcsock->sk;

			lock_sock(clcsk);
			if (skb_queue_empty(&clcsk->sk_receive_queue))
				sk_wait_data(clcsk, &timeo, NULL);
			release_sock(clcsk);
		} else if (!atomic_read(&smc_sk(nsk)->conn.bytes_to_rcv)) {
			lock_sock(nsk);
1104
			smc_rx_wait(smc_sk(nsk), &timeo, smc_rx_data_available);
1105 1106 1107
			release_sock(nsk);
		}
	}
U
Ursula Braun 已提交
1108 1109

out:
1110
	sock_put(sk); /* sock_hold above */
U
Ursula Braun 已提交
1111 1112 1113 1114
	return rc;
}

static int smc_getname(struct socket *sock, struct sockaddr *addr,
1115
		       int peer)
U
Ursula Braun 已提交
1116 1117 1118
{
	struct smc_sock *smc;

1119 1120
	if (peer && (sock->sk->sk_state != SMC_ACTIVE) &&
	    (sock->sk->sk_state != SMC_APPCLOSEWAIT1))
U
Ursula Braun 已提交
1121 1122 1123 1124
		return -ENOTCONN;

	smc = smc_sk(sock->sk);

1125
	return smc->clcsock->ops->getname(smc->clcsock, addr, peer);
U
Ursula Braun 已提交
1126 1127 1128 1129 1130 1131 1132 1133 1134 1135
}

static int smc_sendmsg(struct socket *sock, struct msghdr *msg, size_t len)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -EPIPE;

	smc = smc_sk(sk);
	lock_sock(sk);
1136 1137 1138
	if ((sk->sk_state != SMC_ACTIVE) &&
	    (sk->sk_state != SMC_APPCLOSEWAIT1) &&
	    (sk->sk_state != SMC_INIT))
U
Ursula Braun 已提交
1139
		goto out;
1140 1141 1142 1143 1144 1145 1146 1147 1148 1149

	if (msg->msg_flags & MSG_FASTOPEN) {
		if (sk->sk_state == SMC_INIT) {
			smc->use_fallback = true;
		} else {
			rc = -EINVAL;
			goto out;
		}
	}

U
Ursula Braun 已提交
1150 1151 1152
	if (smc->use_fallback)
		rc = smc->clcsock->ops->sendmsg(smc->clcsock, msg, len);
	else
U
Ursula Braun 已提交
1153
		rc = smc_tx_sendmsg(smc, msg, len);
U
Ursula Braun 已提交
1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167
out:
	release_sock(sk);
	return rc;
}

static int smc_recvmsg(struct socket *sock, struct msghdr *msg, size_t len,
		       int flags)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -ENOTCONN;

	smc = smc_sk(sk);
	lock_sock(sk);
1168 1169 1170
	if ((sk->sk_state == SMC_INIT) ||
	    (sk->sk_state == SMC_LISTEN) ||
	    (sk->sk_state == SMC_CLOSED))
U
Ursula Braun 已提交
1171 1172
		goto out;

1173 1174 1175 1176 1177
	if (sk->sk_state == SMC_PEERFINCLOSEWAIT) {
		rc = 0;
		goto out;
	}

S
Stefan Raspl 已提交
1178
	if (smc->use_fallback) {
U
Ursula Braun 已提交
1179
		rc = smc->clcsock->ops->recvmsg(smc->clcsock, msg, len, flags);
S
Stefan Raspl 已提交
1180 1181 1182 1183
	} else {
		msg->msg_namelen = 0;
		rc = smc_rx_recvmsg(smc, msg, NULL, len, flags);
	}
1184

U
Ursula Braun 已提交
1185 1186 1187 1188 1189
out:
	release_sock(sk);
	return rc;
}

A
Al Viro 已提交
1190
static __poll_t smc_accept_poll(struct sock *parent)
1191
{
U
Ursula Braun 已提交
1192
	struct smc_sock *isk = smc_sk(parent);
A
Al Viro 已提交
1193
	__poll_t mask = 0;
1194

U
Ursula Braun 已提交
1195 1196
	spin_lock(&isk->accept_q_lock);
	if (!list_empty(&isk->accept_q))
1197
		mask = EPOLLIN | EPOLLRDNORM;
U
Ursula Braun 已提交
1198
	spin_unlock(&isk->accept_q_lock);
1199

U
Ursula Braun 已提交
1200
	return mask;
1201 1202
}

A
Al Viro 已提交
1203
static __poll_t smc_poll(struct file *file, struct socket *sock,
U
Ursula Braun 已提交
1204 1205 1206
			     poll_table *wait)
{
	struct sock *sk = sock->sk;
1207
	__poll_t mask = 0;
U
Ursula Braun 已提交
1208
	struct smc_sock *smc;
1209
	int rc;
U
Ursula Braun 已提交
1210

U
Ursula Braun 已提交
1211
	if (!sk)
1212
		return EPOLLNVAL;
U
Ursula Braun 已提交
1213

U
Ursula Braun 已提交
1214
	smc = smc_sk(sock->sk);
U
Ursula Braun 已提交
1215 1216
	sock_hold(sk);
	lock_sock(sk);
1217 1218
	if ((sk->sk_state == SMC_INIT) || smc->use_fallback) {
		/* delegate to CLC child sock */
U
Ursula Braun 已提交
1219
		release_sock(sk);
U
Ursula Braun 已提交
1220 1221
		mask = smc->clcsock->ops->poll(file, smc->clcsock, wait);
		lock_sock(sk);
1222 1223 1224 1225 1226 1227 1228 1229
		sk->sk_err = smc->clcsock->sk->sk_err;
		if (sk->sk_err) {
			mask |= EPOLLERR;
		} else {
			/* if non-blocking connect finished ... */
			if (sk->sk_state == SMC_INIT &&
			    mask & EPOLLOUT &&
			    smc->clcsock->sk->sk_state != TCP_CLOSE) {
1230 1231
				rc = smc_connect_rdma(smc);
				if (rc < 0)
1232
					mask |= EPOLLERR;
U
Ursula Braun 已提交
1233
				/* success cases including fallback */
1234
				mask |= EPOLLOUT | EPOLLWRNORM;
1235
			}
U
Ursula Braun 已提交
1236 1237
		}
	} else {
U
Ursula Braun 已提交
1238 1239 1240 1241 1242
		if (sk->sk_state != SMC_CLOSED) {
			release_sock(sk);
			sock_poll_wait(file, sk_sleep(sk), wait);
			lock_sock(sk);
		}
1243
		if (sk->sk_err)
1244
			mask |= EPOLLERR;
1245 1246
		if ((sk->sk_shutdown == SHUTDOWN_MASK) ||
		    (sk->sk_state == SMC_CLOSED))
1247
			mask |= EPOLLHUP;
U
Ursula Braun 已提交
1248 1249 1250 1251 1252 1253
		if (sk->sk_state == SMC_LISTEN) {
			/* woken up by sk_data_ready in smc_listen_work() */
			mask = smc_accept_poll(sk);
		} else {
			if (atomic_read(&smc->conn.sndbuf_space) ||
			    sk->sk_shutdown & SEND_SHUTDOWN) {
1254
				mask |= EPOLLOUT | EPOLLWRNORM;
U
Ursula Braun 已提交
1255 1256 1257 1258 1259
			} else {
				sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
				set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
			}
			if (atomic_read(&smc->conn.bytes_to_rcv))
1260
				mask |= EPOLLIN | EPOLLRDNORM;
U
Ursula Braun 已提交
1261
			if (sk->sk_shutdown & RCV_SHUTDOWN)
1262
				mask |= EPOLLIN | EPOLLRDNORM | EPOLLRDHUP;
U
Ursula Braun 已提交
1263
			if (sk->sk_state == SMC_APPCLOSEWAIT1)
1264
				mask |= EPOLLIN;
U
Ursula Braun 已提交
1265
		}
1266

U
Ursula Braun 已提交
1267
	}
U
Ursula Braun 已提交
1268 1269
	release_sock(sk);
	sock_put(sk);
U
Ursula Braun 已提交
1270 1271 1272 1273 1274 1275 1276 1277 1278

	return mask;
}

static int smc_shutdown(struct socket *sock, int how)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -EINVAL;
1279
	int rc1 = 0;
U
Ursula Braun 已提交
1280 1281 1282 1283

	smc = smc_sk(sk);

	if ((how < SHUT_RD) || (how > SHUT_RDWR))
1284
		return rc;
U
Ursula Braun 已提交
1285 1286 1287 1288

	lock_sock(sk);

	rc = -ENOTCONN;
1289 1290 1291 1292 1293 1294 1295
	if ((sk->sk_state != SMC_LISTEN) &&
	    (sk->sk_state != SMC_ACTIVE) &&
	    (sk->sk_state != SMC_PEERCLOSEWAIT1) &&
	    (sk->sk_state != SMC_PEERCLOSEWAIT2) &&
	    (sk->sk_state != SMC_APPCLOSEWAIT1) &&
	    (sk->sk_state != SMC_APPCLOSEWAIT2) &&
	    (sk->sk_state != SMC_APPFINCLOSEWAIT))
U
Ursula Braun 已提交
1296 1297 1298 1299 1300 1301
		goto out;
	if (smc->use_fallback) {
		rc = kernel_sock_shutdown(smc->clcsock, how);
		sk->sk_shutdown = smc->clcsock->sk->sk_shutdown;
		if (sk->sk_shutdown == SHUTDOWN_MASK)
			sk->sk_state = SMC_CLOSED;
1302
		goto out;
U
Ursula Braun 已提交
1303
	}
1304 1305 1306 1307 1308 1309 1310 1311
	switch (how) {
	case SHUT_RDWR:		/* shutdown in both directions */
		rc = smc_close_active(smc);
		break;
	case SHUT_WR:
		rc = smc_close_shutdown_write(smc);
		break;
	case SHUT_RD:
1312 1313
		rc = 0;
		/* nothing more to do because peer is not involved */
1314 1315
		break;
	}
1316 1317
	if (smc->clcsock)
		rc1 = kernel_sock_shutdown(smc->clcsock, how);
1318 1319
	/* map sock_shutdown_cmd constants to sk_shutdown value range */
	sk->sk_shutdown |= how + 1;
U
Ursula Braun 已提交
1320 1321 1322

out:
	release_sock(sk);
1323
	return rc ? rc : rc1;
U
Ursula Braun 已提交
1324 1325 1326 1327 1328 1329 1330
}

static int smc_setsockopt(struct socket *sock, int level, int optname,
			  char __user *optval, unsigned int optlen)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
1331
	int val, rc;
U
Ursula Braun 已提交
1332 1333 1334 1335 1336 1337

	smc = smc_sk(sk);

	/* generic setsockopts reaching us here always apply to the
	 * CLC socket
	 */
1338 1339 1340 1341 1342 1343 1344 1345 1346
	rc = smc->clcsock->ops->setsockopt(smc->clcsock, level, optname,
					   optval, optlen);
	if (smc->clcsock->sk->sk_err) {
		sk->sk_err = smc->clcsock->sk->sk_err;
		sk->sk_error_report(sk);
	}
	if (rc)
		return rc;

1347 1348 1349 1350
	if (optlen < sizeof(int))
		return rc;
	get_user(val, (int __user *)optval);

1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365
	lock_sock(sk);
	switch (optname) {
	case TCP_ULP:
	case TCP_FASTOPEN:
	case TCP_FASTOPEN_CONNECT:
	case TCP_FASTOPEN_KEY:
	case TCP_FASTOPEN_NO_COOKIE:
		/* option not supported by SMC */
		if (sk->sk_state == SMC_INIT) {
			smc->use_fallback = true;
		} else {
			if (!smc->use_fallback)
				rc = -EINVAL;
		}
		break;
1366 1367
	case TCP_NODELAY:
		if (sk->sk_state != SMC_INIT && sk->sk_state != SMC_LISTEN) {
1368
			if (val && !smc->use_fallback)
1369 1370 1371 1372 1373 1374
				mod_delayed_work(system_wq, &smc->conn.tx_work,
						 0);
		}
		break;
	case TCP_CORK:
		if (sk->sk_state != SMC_INIT && sk->sk_state != SMC_LISTEN) {
1375
			if (!val && !smc->use_fallback)
1376 1377 1378 1379
				mod_delayed_work(system_wq, &smc->conn.tx_work,
						 0);
		}
		break;
1380 1381 1382
	case TCP_DEFER_ACCEPT:
		smc->sockopt_defer_accept = val;
		break;
1383 1384 1385 1386 1387 1388
	default:
		break;
	}
	release_sock(sk);

	return rc;
U
Ursula Braun 已提交
1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405
}

static int smc_getsockopt(struct socket *sock, int level, int optname,
			  char __user *optval, int __user *optlen)
{
	struct smc_sock *smc;

	smc = smc_sk(sock->sk);
	/* socket options apply to the CLC socket */
	return smc->clcsock->ops->getsockopt(smc->clcsock, level, optname,
					     optval, optlen);
}

static int smc_ioctl(struct socket *sock, unsigned int cmd,
		     unsigned long arg)
{
	struct smc_sock *smc;
1406
	int answ;
U
Ursula Braun 已提交
1407 1408

	smc = smc_sk(sock->sk);
1409 1410 1411
	if (smc->use_fallback) {
		if (!smc->clcsock)
			return -EBADF;
U
Ursula Braun 已提交
1412
		return smc->clcsock->ops->ioctl(smc->clcsock, cmd, arg);
1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423
	}
	switch (cmd) {
	case SIOCINQ: /* same as FIONREAD */
		if (smc->sk.sk_state == SMC_LISTEN)
			return -EINVAL;
		answ = atomic_read(&smc->conn.bytes_to_rcv);
		break;
	case SIOCOUTQ:
		/* output queue size (not send + not acked) */
		if (smc->sk.sk_state == SMC_LISTEN)
			return -EINVAL;
1424
		answ = smc->conn.sndbuf_desc->len -
1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437
					atomic_read(&smc->conn.sndbuf_space);
		break;
	case SIOCOUTQNSD:
		/* output queue size (not send only) */
		if (smc->sk.sk_state == SMC_LISTEN)
			return -EINVAL;
		answ = smc_tx_prepared_sends(&smc->conn);
		break;
	default:
		return -ENOIOCTLCMD;
	}

	return put_user(answ, (int __user *)arg);
U
Ursula Braun 已提交
1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448
}

static ssize_t smc_sendpage(struct socket *sock, struct page *page,
			    int offset, size_t size, int flags)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -EPIPE;

	smc = smc_sk(sk);
	lock_sock(sk);
S
Stefan Raspl 已提交
1449 1450
	if (sk->sk_state != SMC_ACTIVE) {
		release_sock(sk);
U
Ursula Braun 已提交
1451
		goto out;
S
Stefan Raspl 已提交
1452 1453
	}
	release_sock(sk);
U
Ursula Braun 已提交
1454 1455 1456 1457 1458 1459 1460 1461 1462 1463
	if (smc->use_fallback)
		rc = kernel_sendpage(smc->clcsock, page, offset,
				     size, flags);
	else
		rc = sock_no_sendpage(sock, page, offset, size, flags);

out:
	return rc;
}

S
Stefan Raspl 已提交
1464 1465 1466 1467 1468 1469
/* Map the affected portions of the rmbe into an spd, note the number of bytes
 * to splice in conn->splice_pending, and press 'go'. Delays consumer cursor
 * updates till whenever a respective page has been fully processed.
 * Note that subsequent recv() calls have to wait till all splice() processing
 * completed.
 */
U
Ursula Braun 已提交
1470 1471
static ssize_t smc_splice_read(struct socket *sock, loff_t *ppos,
			       struct pipe_inode_info *pipe, size_t len,
S
Stefan Raspl 已提交
1472
			       unsigned int flags)
U
Ursula Braun 已提交
1473 1474 1475 1476 1477 1478 1479
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -ENOTCONN;

	smc = smc_sk(sk);
	lock_sock(sk);
S
Stefan Raspl 已提交
1480 1481 1482 1483 1484 1485 1486 1487

	if (sk->sk_state == SMC_INIT ||
	    sk->sk_state == SMC_LISTEN ||
	    sk->sk_state == SMC_CLOSED)
		goto out;

	if (sk->sk_state == SMC_PEERFINCLOSEWAIT) {
		rc = 0;
U
Ursula Braun 已提交
1488
		goto out;
S
Stefan Raspl 已提交
1489 1490
	}

U
Ursula Braun 已提交
1491 1492 1493 1494
	if (smc->use_fallback) {
		rc = smc->clcsock->ops->splice_read(smc->clcsock, ppos,
						    pipe, len, flags);
	} else {
S
Stefan Raspl 已提交
1495 1496 1497 1498 1499 1500 1501 1502 1503
		if (*ppos) {
			rc = -ESPIPE;
			goto out;
		}
		if (flags & SPLICE_F_NONBLOCK)
			flags = MSG_DONTWAIT;
		else
			flags = 0;
		rc = smc_rx_recvmsg(smc, NULL, pipe, len, flags);
U
Ursula Braun 已提交
1504 1505 1506
	}
out:
	release_sock(sk);
S
Stefan Raspl 已提交
1507

U
Ursula Braun 已提交
1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536
	return rc;
}

/* must look like tcp */
static const struct proto_ops smc_sock_ops = {
	.family		= PF_SMC,
	.owner		= THIS_MODULE,
	.release	= smc_release,
	.bind		= smc_bind,
	.connect	= smc_connect,
	.socketpair	= sock_no_socketpair,
	.accept		= smc_accept,
	.getname	= smc_getname,
	.poll		= smc_poll,
	.ioctl		= smc_ioctl,
	.listen		= smc_listen,
	.shutdown	= smc_shutdown,
	.setsockopt	= smc_setsockopt,
	.getsockopt	= smc_getsockopt,
	.sendmsg	= smc_sendmsg,
	.recvmsg	= smc_recvmsg,
	.mmap		= sock_no_mmap,
	.sendpage	= smc_sendpage,
	.splice_read	= smc_splice_read,
};

static int smc_create(struct net *net, struct socket *sock, int protocol,
		      int kern)
{
1537
	int family = (protocol == SMCPROTO_SMC6) ? PF_INET6 : PF_INET;
U
Ursula Braun 已提交
1538 1539 1540 1541 1542 1543 1544 1545 1546
	struct smc_sock *smc;
	struct sock *sk;
	int rc;

	rc = -ESOCKTNOSUPPORT;
	if (sock->type != SOCK_STREAM)
		goto out;

	rc = -EPROTONOSUPPORT;
1547
	if (protocol != SMCPROTO_SMC && protocol != SMCPROTO_SMC6)
U
Ursula Braun 已提交
1548 1549 1550 1551
		goto out;

	rc = -ENOBUFS;
	sock->ops = &smc_sock_ops;
1552
	sk = smc_sock_alloc(net, sock, protocol);
U
Ursula Braun 已提交
1553 1554 1555 1556 1557
	if (!sk)
		goto out;

	/* create internal TCP socket for CLC handshake and fallback */
	smc = smc_sk(sk);
1558
	smc->use_fallback = false; /* assume rdma capability first */
1559 1560
	rc = sock_create_kern(net, family, SOCK_STREAM, IPPROTO_TCP,
			      &smc->clcsock);
1561
	if (rc) {
U
Ursula Braun 已提交
1562
		sk_common_release(sk);
1563 1564
		goto out;
	}
U
Ursula Braun 已提交
1565 1566
	smc->sk.sk_sndbuf = max(smc->clcsock->sk->sk_sndbuf, SMC_BUF_MIN_SIZE);
	smc->sk.sk_rcvbuf = max(smc->clcsock->sk->sk_rcvbuf, SMC_BUF_MIN_SIZE);
U
Ursula Braun 已提交
1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581

out:
	return rc;
}

static const struct net_proto_family smc_sock_family_ops = {
	.family	= PF_SMC,
	.owner	= THIS_MODULE,
	.create	= smc_create,
};

static int __init smc_init(void)
{
	int rc;

1582 1583 1584 1585
	rc = smc_pnet_init();
	if (rc)
		return rc;

U
Ursula Braun 已提交
1586 1587 1588 1589 1590 1591
	rc = smc_llc_init();
	if (rc) {
		pr_err("%s: smc_llc_init fails with %d\n", __func__, rc);
		goto out_pnet;
	}

1592 1593 1594 1595 1596 1597
	rc = smc_cdc_init();
	if (rc) {
		pr_err("%s: smc_cdc_init fails with %d\n", __func__, rc);
		goto out_pnet;
	}

U
Ursula Braun 已提交
1598 1599
	rc = proto_register(&smc_proto, 1);
	if (rc) {
1600
		pr_err("%s: proto_register(v4) fails with %d\n", __func__, rc);
1601
		goto out_pnet;
U
Ursula Braun 已提交
1602 1603
	}

1604 1605 1606 1607 1608 1609
	rc = proto_register(&smc_proto6, 1);
	if (rc) {
		pr_err("%s: proto_register(v6) fails with %d\n", __func__, rc);
		goto out_proto;
	}

U
Ursula Braun 已提交
1610 1611 1612
	rc = sock_register(&smc_sock_family_ops);
	if (rc) {
		pr_err("%s: sock_register fails with %d\n", __func__, rc);
1613
		goto out_proto6;
U
Ursula Braun 已提交
1614
	}
1615
	INIT_HLIST_HEAD(&smc_v4_hashinfo.ht);
1616
	INIT_HLIST_HEAD(&smc_v6_hashinfo.ht);
U
Ursula Braun 已提交
1617

1618 1619 1620 1621 1622 1623
	rc = smc_ib_register_client();
	if (rc) {
		pr_err("%s: ib_register fails with %d\n", __func__, rc);
		goto out_sock;
	}

U
Ursula Braun 已提交
1624
	static_branch_enable(&tcp_have_smc);
U
Ursula Braun 已提交
1625 1626
	return 0;

1627 1628
out_sock:
	sock_unregister(PF_SMC);
1629 1630
out_proto6:
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
1631 1632
out_proto:
	proto_unregister(&smc_proto);
1633 1634
out_pnet:
	smc_pnet_exit();
U
Ursula Braun 已提交
1635 1636 1637 1638 1639
	return rc;
}

static void __exit smc_exit(void)
{
1640 1641 1642 1643 1644 1645 1646 1647 1648
	struct smc_link_group *lgr, *lg;
	LIST_HEAD(lgr_freeing_list);

	spin_lock_bh(&smc_lgr_list.lock);
	if (!list_empty(&smc_lgr_list.list))
		list_splice_init(&smc_lgr_list.list, &lgr_freeing_list);
	spin_unlock_bh(&smc_lgr_list.lock);
	list_for_each_entry_safe(lgr, lg, &lgr_freeing_list, list) {
		list_del_init(&lgr->list);
1649
		smc_llc_link_inactive(&lgr->lnk[SMC_SINGLE_LINK]);
1650
		cancel_delayed_work_sync(&lgr->free_work);
1651 1652
		smc_lgr_free(lgr); /* free link group */
	}
U
Ursula Braun 已提交
1653
	static_branch_disable(&tcp_have_smc);
1654
	smc_ib_unregister_client();
U
Ursula Braun 已提交
1655
	sock_unregister(PF_SMC);
1656
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
1657
	proto_unregister(&smc_proto);
1658
	smc_pnet_exit();
U
Ursula Braun 已提交
1659 1660 1661 1662 1663 1664 1665 1666 1667
}

module_init(smc_init);
module_exit(smc_exit);

MODULE_AUTHOR("Ursula Braun <ubraun@linux.vnet.ibm.com>");
MODULE_DESCRIPTION("smc socket address family");
MODULE_LICENSE("GPL");
MODULE_ALIAS_NETPROTO(PF_SMC);