af_smc.c 54.7 KB
Newer Older
1
// SPDX-License-Identifier: GPL-2.0-only
U
Ursula Braun 已提交
2 3 4 5 6 7 8 9
/*
 *  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
 *
10 11 12
 *  Initial restrictions:
 *    - support for alternate links postponed
 *
13
 *  Copyright IBM Corp. 2016, 2018
U
Ursula Braun 已提交
14 15 16 17 18 19 20 21 22 23
 *
 *  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>
24
#include <linux/workqueue.h>
25
#include <linux/in.h>
26
#include <linux/sched/signal.h>
27
#include <linux/if_vlan.h>
28
#include <linux/rcupdate_wait.h>
29

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

35 36 37 38
#include <net/net_namespace.h>
#include <net/netns/generic.h>
#include "smc_netns.h"

U
Ursula Braun 已提交
39
#include "smc.h"
40
#include "smc_clc.h"
U
Ursula Braun 已提交
41
#include "smc_llc.h"
42
#include "smc_cdc.h"
43
#include "smc_core.h"
44
#include "smc_ib.h"
45
#include "smc_ism.h"
46
#include "smc_pnet.h"
U
Ursula Braun 已提交
47
#include "smc_tx.h"
U
Ursula Braun 已提交
48
#include "smc_rx.h"
49
#include "smc_close.h"
U
Ursula Braun 已提交
50

51 52 53 54 55
static DEFINE_MUTEX(smc_server_lgr_pending);	/* serialize link group
						 * creation on server
						 */
static DEFINE_MUTEX(smc_client_lgr_pending);	/* serialize link group
						 * creation on client
56 57
						 */

58
static void smc_tcp_listen_work(struct work_struct *);
59
static void smc_connect_work(struct work_struct *);
60

U
Ursula Braun 已提交
61 62 63 64 65 66 67
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);
}

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

72 73 74 75
static struct smc_hashinfo smc_v6_hashinfo = {
	.lock = __RW_LOCK_UNLOCKED(smc_v6_hashinfo.lock),
};

76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
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 已提交
104 105 106
	.name		= "SMC",
	.owner		= THIS_MODULE,
	.keepalive	= smc_set_keepalive,
107 108
	.hash		= smc_hash_sk,
	.unhash		= smc_unhash_sk,
U
Ursula Braun 已提交
109
	.obj_size	= sizeof(struct smc_sock),
110
	.h.smc_hash	= &smc_v4_hashinfo,
111
	.slab_flags	= SLAB_TYPESAFE_BY_RCU,
U
Ursula Braun 已提交
112
};
113
EXPORT_SYMBOL_GPL(smc_proto);
U
Ursula Braun 已提交
114

115 116 117 118 119 120 121 122 123 124 125 126
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);

127 128
static void smc_restore_fallback_changes(struct smc_sock *smc)
{
129 130 131 132
	if (smc->clcsock->file) { /* non-accepted sockets have no file yet */
		smc->clcsock->file->private_data = smc->sk.sk_socket;
		smc->clcsock->file = NULL;
	}
133 134
}

135
static int __smc_release(struct smc_sock *smc)
U
Ursula Braun 已提交
136
{
137
	struct sock *sk = &smc->sk;
138
	int rc = 0;
U
Ursula Braun 已提交
139

140
	if (!smc->use_fallback) {
141 142 143
		rc = smc_close_active(smc);
		sock_set_flag(sk, SOCK_DEAD);
		sk->sk_shutdown |= SHUTDOWN_MASK;
144 145 146 147
	} else {
		if (sk->sk_state != SMC_LISTEN && sk->sk_state != SMC_INIT)
			sock_put(sk); /* passive closing */
		if (sk->sk_state == SMC_LISTEN) {
148 149 150
			/* wake up clcsock accept */
			rc = kernel_sock_shutdown(smc->clcsock, SHUT_RDWR);
		}
151 152
		sk->sk_state = SMC_CLOSED;
		sk->sk_state_change(sk);
153
		smc_restore_fallback_changes(smc);
154
	}
U
Ursula Braun 已提交
155

156 157 158 159
	sk->sk_prot->unhash(sk);

	if (sk->sk_state == SMC_CLOSED) {
		if (smc->clcsock) {
160 161 162
			release_sock(sk);
			smc_clcsock_release(smc);
			lock_sock(sk);
163 164 165 166 167
		}
		if (!smc->use_fallback)
			smc_conn_free(&smc->conn);
	}

168 169 170 171 172 173 174 175 176 177 178 179
	return rc;
}

static int smc_release(struct socket *sock)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = 0;

	if (!sk)
		goto out;

U
Ursula Braun 已提交
180
	sock_hold(sk); /* sock_put below */
181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197
	smc = smc_sk(sk);

	/* cleanup for a dangling non-blocking connect */
	if (smc->connect_nonblock && sk->sk_state == SMC_INIT)
		tcp_abort(smc->clcsock->sk, ECONNABORTED);
	flush_work(&smc->connect_work);

	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);

	rc = __smc_release(smc);

U
Ursula Braun 已提交
198 199 200 201 202
	/* detach socket */
	sock_orphan(sk);
	sock->sk = NULL;
	release_sock(sk);

U
Ursula Braun 已提交
203
	sock_put(sk); /* sock_hold above */
204
	sock_put(sk); /* final sock_put */
U
Ursula Braun 已提交
205
out:
206
	return rc;
U
Ursula Braun 已提交
207 208 209 210 211 212 213 214 215 216 217 218
}

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);
}

219 220
static struct sock *smc_sock_alloc(struct net *net, struct socket *sock,
				   int protocol)
U
Ursula Braun 已提交
221 222
{
	struct smc_sock *smc;
223
	struct proto *prot;
U
Ursula Braun 已提交
224 225
	struct sock *sk;

226 227
	prot = (protocol == SMCPROTO_SMC6) ? &smc_proto6 : &smc_proto;
	sk = sk_alloc(net, PF_SMC, GFP_KERNEL, prot, 0);
U
Ursula Braun 已提交
228 229 230 231 232 233
	if (!sk)
		return NULL;

	sock_init_data(sock, sk); /* sets sk_refcnt to 1 */
	sk->sk_state = SMC_INIT;
	sk->sk_destruct = smc_destruct;
234
	sk->sk_protocol = protocol;
U
Ursula Braun 已提交
235
	smc = smc_sk(sk);
236
	INIT_WORK(&smc->tcp_listen_work, smc_tcp_listen_work);
237
	INIT_WORK(&smc->connect_work, smc_connect_work);
238
	INIT_DELAYED_WORK(&smc->conn.tx_work, smc_tx_work);
239 240
	INIT_LIST_HEAD(&smc->accept_q);
	spin_lock_init(&smc->accept_q_lock);
241
	spin_lock_init(&smc->conn.send_lock);
242
	sk->sk_prot->hash(sk);
243
	sk_refcnt_debug_inc(sk);
244
	mutex_init(&smc->clcsock_release_lock);
U
Ursula Braun 已提交
245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264

	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;
265 266 267 268
	if (addr->sin_family != AF_INET &&
	    addr->sin_family != AF_INET6 &&
	    addr->sin_family != AF_UNSPEC)
		goto out;
U
Ursula Braun 已提交
269
	/* accept AF_UNSPEC (mapped to AF_INET) only if s_addr is INADDR_ANY */
270 271
	if (addr->sin_family == AF_UNSPEC &&
	    addr->sin_addr.s_addr != htonl(INADDR_ANY))
U
Ursula Braun 已提交
272 273 274 275 276 277
		goto out;

	lock_sock(sk);

	/* Check if socket is already active */
	rc = -EINVAL;
278
	if (sk->sk_state != SMC_INIT || smc->connect_nonblock)
U
Ursula Braun 已提交
279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321
		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) | \
D
Deepa Dinamani 已提交
322 323
			     (1UL << SOCK_FILTER_LOCKED) | \
			     (1UL << SOCK_TSTAMP_NEW))
U
Ursula Braun 已提交
324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341
/* 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);
}

342
/* register the new rmb on all links */
343
static int smcr_lgr_reg_rmbs(struct smc_link *link,
344 345
			     struct smc_buf_desc *rmb_desc)
{
346 347
	struct smc_link_group *lgr = link->lgr;
	int i, rc = 0;
348

349 350 351 352 353 354 355
	rc = smc_llc_flow_initiate(lgr, SMC_LLC_FLOW_RKEY);
	if (rc)
		return rc;
	/* protect against parallel smc_llc_cli_rkey_exchange() and
	 * parallel smcr_link_reg_rmb()
	 */
	mutex_lock(&lgr->llc_conf_mutex);
356
	for (i = 0; i < SMC_LINKS_PER_LGR_MAX; i++) {
357
		if (!smc_link_active(&lgr->lnk[i]))
358
			continue;
359
		rc = smcr_link_reg_rmb(&lgr->lnk[i], rmb_desc);
360
		if (rc)
361
			goto out;
362
	}
363 364 365 366 367 368 369 370 371

	/* exchange confirm_rkey msg with peer */
	rc = smc_llc_do_confirm_rkey(link, rmb_desc);
	if (rc) {
		rc = -EFAULT;
		goto out;
	}
	rmb_desc->is_conf_rkey = true;
out:
372 373
	mutex_unlock(&lgr->llc_conf_mutex);
	smc_llc_flow_stop(lgr, &lgr->llc_flow_lcl);
374
	return rc;
K
Karsten Graul 已提交
375 376
}

377
static int smcr_clnt_conf_first_link(struct smc_sock *smc)
U
Ursula Braun 已提交
378
{
379
	struct smc_link *link = smc->conn.lnk;
380
	struct smc_llc_qentry *qentry;
U
Ursula Braun 已提交
381 382 383
	int rc;

	/* receive CONFIRM LINK request from server over RoCE fabric */
384 385 386
	qentry = smc_llc_wait(link->lgr, NULL, SMC_LLC_WAIT_TIME,
			      SMC_LLC_CONFIRM_LINK);
	if (!qentry) {
U
Ursula Braun 已提交
387 388 389
		struct smc_clc_msg_decline dclc;

		rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc),
390
				      SMC_CLC_DECLINE, CLC_WAIT_TIME_SHORT);
391
		return rc == -EAGAIN ? SMC_CLC_DECL_TIMEOUT_CL : rc;
U
Ursula Braun 已提交
392
	}
393
	smc_llc_save_peer_uid(qentry);
394 395 396
	rc = smc_llc_eval_conf_link(qentry, SMC_LLC_REQ);
	smc_llc_flow_qentry_del(&link->lgr->llc_flow_lcl);
	if (rc)
397 398
		return SMC_CLC_DECL_RMBE_EC;

U
Ursula Braun 已提交
399 400
	rc = smc_ib_modify_qp_rts(link);
	if (rc)
401
		return SMC_CLC_DECL_ERR_RDYLNK;
U
Ursula Braun 已提交
402 403

	smc_wr_remember_qp_attr(link);
404

405
	if (smcr_link_reg_rmb(link, smc->conn.rmb_desc))
406
		return SMC_CLC_DECL_ERR_REGRMB;
407

408 409 410
	/* confirm_rkey is implicit on 1st contact */
	smc->conn.rmb_desc->is_conf_rkey = true;

U
Ursula Braun 已提交
411
	/* send CONFIRM LINK response over RoCE fabric */
412
	rc = smc_llc_send_confirm_link(link, SMC_LLC_RESP);
U
Ursula Braun 已提交
413
	if (rc < 0)
414
		return SMC_CLC_DECL_TIMEOUT_CL;
U
Ursula Braun 已提交
415

416
	smc_llc_link_active(link);
417
	smcr_lgr_set_type(link->lgr, SMC_LGR_SINGLE);
418 419 420 421 422

	/* optional 2nd link, receive ADD LINK request from server */
	qentry = smc_llc_wait(link->lgr, NULL, SMC_LLC_WAIT_TIME,
			      SMC_LLC_ADD_LINK);
	if (!qentry) {
423 424 425
		struct smc_clc_msg_decline dclc;

		rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc),
426
				      SMC_CLC_DECLINE, CLC_WAIT_TIME_SHORT);
427 428 429
		if (rc == -EAGAIN)
			rc = 0; /* no DECLINE received, go with one link */
		return rc;
430
	}
431
	smc_llc_flow_qentry_clr(&link->lgr->llc_flow_lcl);
432
	smc_llc_cli_add_link(link, qentry);
433
	return 0;
U
Ursula Braun 已提交
434 435
}

436 437
static void smcr_conn_save_peer_info(struct smc_sock *smc,
				     struct smc_clc_msg_accept_confirm *clc)
438
{
439
	int bufsize = smc_uncompress_bufsize(clc->r0.rmbe_size);
440

441 442
	smc->conn.peer_rmbe_idx = clc->r0.rmbe_idx;
	smc->conn.local_tx_ctrl.token = ntohl(clc->r0.rmbe_alert_token);
443
	smc->conn.peer_rmbe_size = bufsize;
U
Ursula Braun 已提交
444
	atomic_set(&smc->conn.peer_rmbe_space, smc->conn.peer_rmbe_size);
445
	smc->conn.tx_off = bufsize * (smc->conn.peer_rmbe_idx - 1);
446 447
}

448 449 450
static void smcd_conn_save_peer_info(struct smc_sock *smc,
				     struct smc_clc_msg_accept_confirm *clc)
{
451
	int bufsize = smc_uncompress_bufsize(clc->d0.dmbe_size);
452

453 454
	smc->conn.peer_rmbe_idx = clc->d0.dmbe_idx;
	smc->conn.peer_token = clc->d0.token;
455 456 457 458 459 460 461 462 463 464 465 466 467 468 469
	/* msg header takes up space in the buffer */
	smc->conn.peer_rmbe_size = bufsize - sizeof(struct smcd_cdc_msg);
	atomic_set(&smc->conn.peer_rmbe_space, smc->conn.peer_rmbe_size);
	smc->conn.tx_off = bufsize * smc->conn.peer_rmbe_idx;
}

static void smc_conn_save_peer_info(struct smc_sock *smc,
				    struct smc_clc_msg_accept_confirm *clc)
{
	if (smc->conn.lgr->is_smcd)
		smcd_conn_save_peer_info(smc, clc);
	else
		smcr_conn_save_peer_info(smc, clc);
}

470 471 472
static void smc_link_save_peer_info(struct smc_link *link,
				    struct smc_clc_msg_accept_confirm *clc)
{
473 474 475 476 477
	link->peer_qpn = ntoh24(clc->r0.qpn);
	memcpy(link->peer_gid, clc->r0.lcl.gid, SMC_GID_SIZE);
	memcpy(link->peer_mac, clc->r0.lcl.mac, sizeof(link->peer_mac));
	link->peer_psn = ntoh24(clc->r0.psn);
	link->peer_mtu = clc->r0.qp_mtu;
478 479
}

480 481 482 483 484 485
static void smc_switch_to_fallback(struct smc_sock *smc)
{
	smc->use_fallback = true;
	if (smc->sk.sk_socket && smc->sk.sk_socket->file) {
		smc->clcsock->file = smc->sk.sk_socket->file;
		smc->clcsock->file->private_data = smc->clcsock;
486 487
		smc->clcsock->wq.fasync_list =
			smc->sk.sk_socket->wq.fasync_list;
488 489 490
	}
}

491
/* fall back during connect */
492
static int smc_connect_fallback(struct smc_sock *smc, int reason_code)
493
{
494
	smc_switch_to_fallback(smc);
495
	smc->fallback_rsn = reason_code;
496
	smc_copy_sock_settings_to_clc(smc);
497
	smc->connect_nonblock = 0;
498 499 500 501
	if (smc->sk.sk_state == SMC_INIT)
		smc->sk.sk_state = SMC_ACTIVE;
	return 0;
}
502

503 504 505 506
/* decline and fall back during connect */
static int smc_connect_decline_fallback(struct smc_sock *smc, int reason_code)
{
	int rc;
507

508 509 510
	if (reason_code < 0) { /* error, fallback is not possible */
		if (smc->sk.sk_state == SMC_INIT)
			sock_put(&smc->sk); /* passive closing */
511
		return reason_code;
512
	}
513
	if (reason_code != SMC_CLC_DECL_PEERDECL) {
514
		rc = smc_clc_send_decline(smc, reason_code);
515 516 517
		if (rc < 0) {
			if (smc->sk.sk_state == SMC_INIT)
				sock_put(&smc->sk); /* passive closing */
518
			return rc;
519
		}
U
Ursula Braun 已提交
520
	}
521
	return smc_connect_fallback(smc, reason_code);
522
}
U
Ursula Braun 已提交
523

524 525
/* abort connecting */
static int smc_connect_abort(struct smc_sock *smc, int reason_code,
526
			     int local_first)
527
{
528 529
	bool is_smcd = smc->conn.lgr->is_smcd;

530
	if (local_first)
531 532 533 534
		smc_lgr_cleanup_early(&smc->conn);
	else
		smc_conn_free(&smc->conn);
	if (is_smcd)
535 536 537 538 539
		/* there is only one lgr role for SMC-D; use server lock */
		mutex_unlock(&smc_server_lgr_pending);
	else
		mutex_unlock(&smc_client_lgr_pending);

540
	smc->connect_nonblock = 0;
541 542 543 544 545
	return reason_code;
}

/* check if there is a rdma device available for this connection. */
/* called for connect and listen */
546
static int smc_find_rdma_device(struct smc_sock *smc, struct smc_init_info *ini)
547
{
548 549 550 551
	/* 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
	 */
552
	smc_pnet_find_roce_resource(smc->clcsock->sk, ini);
553 554
	if (!ini->ib_dev)
		return SMC_CLC_DECL_NOSMCRDEV;
555
	return 0;
556 557
}

558 559
/* check if there is an ISM device available for this connection. */
/* called for connect and listen */
560
static int smc_find_ism_device(struct smc_sock *smc, struct smc_init_info *ini)
561 562
{
	/* Find ISM device with same PNETID as connecting interface  */
563 564
	smc_pnet_find_ism_resource(smc->clcsock->sk, ini);
	if (!ini->ism_dev)
565
		return SMC_CLC_DECL_NOSMCDDEV;
566 567 568 569 570
	return 0;
}

/* Check for VLAN ID and register it on ISM device just for CLC handshake */
static int smc_connect_ism_vlan_setup(struct smc_sock *smc,
571
				      struct smc_init_info *ini)
572
{
573
	if (ini->vlan_id && smc_ism_get_vlan(ini->ism_dev, ini->vlan_id))
574
		return SMC_CLC_DECL_ISMVLANERR;
575 576 577 578 579 580 581
	return 0;
}

/* cleanup temporary VLAN ID registration used for CLC handshake. If ISM is
 * used, the VLAN ID will be registered again during the connection setup.
 */
static int smc_connect_ism_vlan_cleanup(struct smc_sock *smc, bool is_smcd,
582
					struct smc_init_info *ini)
583 584 585
{
	if (!is_smcd)
		return 0;
586
	if (ini->vlan_id && smc_ism_put_vlan(ini->ism_dev, ini->vlan_id))
587 588 589 590
		return SMC_CLC_DECL_CNFERR;
	return 0;
}

591
/* CLC handshake during connect */
592
static int smc_connect_clc(struct smc_sock *smc, int smc_type,
593
			   struct smc_clc_msg_accept_confirm *aclc,
594
			   struct smc_init_info *ini)
595 596
{
	int rc = 0;
597 598

	/* do inband token exchange */
599
	rc = smc_clc_send_proposal(smc, smc_type, ini);
600 601
	if (rc)
		return rc;
602
	/* receive SMC Accept CLC message */
603 604
	return smc_clc_wait_msg(smc, aclc, sizeof(*aclc), SMC_CLC_ACCEPT,
				CLC_WAIT_TIME);
605 606 607 608 609
}

/* setup for RDMA connection of client */
static int smc_connect_rdma(struct smc_sock *smc,
			    struct smc_clc_msg_accept_confirm *aclc,
610
			    struct smc_init_info *ini)
611
{
612
	int i, reason_code = 0;
613
	struct smc_link *link;
614

615
	ini->is_smcd = false;
616 617
	ini->ib_lcl = &aclc->r0.lcl;
	ini->ib_clcqpn = ntoh24(aclc->r0.qpn);
618
	ini->first_contact_peer = aclc->hdr.flag;
619

620
	mutex_lock(&smc_client_lgr_pending);
621 622
	reason_code = smc_conn_create(smc, ini);
	if (reason_code) {
623 624
		mutex_unlock(&smc_client_lgr_pending);
		return reason_code;
625
	}
626

627
	smc_conn_save_peer_info(smc, aclc);
U
Ursula Braun 已提交
628

629
	if (ini->first_contact_local) {
630 631 632 633 634 635 636
		link = smc->conn.lnk;
	} else {
		/* set link that was assigned by server */
		link = NULL;
		for (i = 0; i < SMC_LINKS_PER_LGR_MAX; i++) {
			struct smc_link *l = &smc->conn.lgr->lnk[i];

637 638 639 640 641
			if (l->peer_qpn == ntoh24(aclc->r0.qpn) &&
			    !memcmp(l->peer_gid, &aclc->r0.lcl.gid,
				    SMC_GID_SIZE) &&
			    !memcmp(l->peer_mac, &aclc->r0.lcl.mac,
				    sizeof(l->peer_mac))) {
642 643 644 645 646 647
				link = l;
				break;
			}
		}
		if (!link)
			return smc_connect_abort(smc, SMC_CLC_DECL_NOSRVLINK,
648
						 ini->first_contact_local);
649 650 651
		smc->conn.lnk = link;
	}

652
	/* create send buffer and rmb */
653
	if (smc_buf_create(smc, false))
654
		return smc_connect_abort(smc, SMC_CLC_DECL_MEM,
655
					 ini->first_contact_local);
U
Ursula Braun 已提交
656

657
	if (ini->first_contact_local)
658
		smc_link_save_peer_info(link, aclc);
659

660
	if (smc_rmb_rtoken_handling(&smc->conn, link, aclc))
661
		return smc_connect_abort(smc, SMC_CLC_DECL_ERR_RTOK,
662
					 ini->first_contact_local);
663

664 665 666
	smc_close_init(smc);
	smc_rx_init(smc);

667
	if (ini->first_contact_local) {
668
		if (smc_ib_ready_link(link))
669
			return smc_connect_abort(smc, SMC_CLC_DECL_ERR_RDYLNK,
670
						 ini->first_contact_local);
671
	} else {
672
		if (smcr_lgr_reg_rmbs(link, smc->conn.rmb_desc))
673
			return smc_connect_abort(smc, SMC_CLC_DECL_ERR_REGRMB,
674
						 ini->first_contact_local);
675
	}
676
	smc_rmb_sync_sg_for_device(&smc->conn);
677

678 679
	reason_code = smc_clc_send_confirm(smc);
	if (reason_code)
680
		return smc_connect_abort(smc, reason_code,
681
					 ini->first_contact_local);
682 683

	smc_tx_init(smc);
684

685
	if (ini->first_contact_local) {
U
Ursula Braun 已提交
686
		/* QP confirmation over RoCE fabric */
687
		smc_llc_flow_initiate(link->lgr, SMC_LLC_FLOW_ADD_LINK);
688
		reason_code = smcr_clnt_conf_first_link(smc);
689
		smc_llc_flow_stop(link->lgr, &link->lgr->llc_flow_lcl);
690 691
		if (reason_code)
			return smc_connect_abort(smc, reason_code,
692
						 ini->first_contact_local);
U
Ursula Braun 已提交
693
	}
694
	mutex_unlock(&smc_client_lgr_pending);
U
Ursula Braun 已提交
695

696
	smc_copy_sock_settings_to_clc(smc);
697
	smc->connect_nonblock = 0;
698 699
	if (smc->sk.sk_state == SMC_INIT)
		smc->sk.sk_state = SMC_ACTIVE;
700

701 702
	return 0;
}
703

704 705 706
/* setup for ISM connection of client */
static int smc_connect_ism(struct smc_sock *smc,
			   struct smc_clc_msg_accept_confirm *aclc,
707
			   struct smc_init_info *ini)
708 709 710
{
	int rc = 0;

711
	ini->is_smcd = true;
712
	ini->ism_peer_gid = aclc->d0.gid;
713
	ini->first_contact_peer = aclc->hdr.flag;
714

715 716
	/* there is only one lgr role for SMC-D; use server lock */
	mutex_lock(&smc_server_lgr_pending);
717 718
	rc = smc_conn_create(smc, ini);
	if (rc) {
719
		mutex_unlock(&smc_server_lgr_pending);
720
		return rc;
721
	}
722 723

	/* Create send and receive buffers */
724 725 726 727 728
	rc = smc_buf_create(smc, true);
	if (rc)
		return smc_connect_abort(smc, (rc == -ENOSPC) ?
					      SMC_CLC_DECL_MAX_DMB :
					      SMC_CLC_DECL_MEM,
729
					 ini->first_contact_local);
730 731 732 733 734 735 736 737

	smc_conn_save_peer_info(smc, aclc);
	smc_close_init(smc);
	smc_rx_init(smc);
	smc_tx_init(smc);

	rc = smc_clc_send_confirm(smc);
	if (rc)
738
		return smc_connect_abort(smc, rc, ini->first_contact_local);
739
	mutex_unlock(&smc_server_lgr_pending);
740 741

	smc_copy_sock_settings_to_clc(smc);
742
	smc->connect_nonblock = 0;
743 744 745 746 747 748
	if (smc->sk.sk_state == SMC_INIT)
		smc->sk.sk_state = SMC_ACTIVE;

	return 0;
}

749 750 751
/* perform steps before actually connecting */
static int __smc_connect(struct smc_sock *smc)
{
752
	bool ism_supported = false, rdma_supported = false;
753
	struct smc_clc_msg_accept_confirm aclc;
754
	struct smc_init_info ini = {0};
755
	int smc_type;
756
	int rc = 0;
757

758
	if (smc->use_fallback)
759
		return smc_connect_fallback(smc, smc->fallback_rsn);
760 761 762

	/* if peer has not signalled SMC-capability, fall back */
	if (!tcp_sk(smc->clcsock->sk)->syn_smc)
763
		return smc_connect_fallback(smc, SMC_CLC_DECL_PEERNOSMC);
764 765 766 767 768

	/* IPSec connections opt out of SMC-R optimizations */
	if (using_ipsec(smc))
		return smc_connect_decline_fallback(smc, SMC_CLC_DECL_IPSEC);

K
Karsten Graul 已提交
769
	/* get vlan id from IP device */
770
	if (smc_vlan_by_tcpsk(smc->clcsock, &ini))
K
Karsten Graul 已提交
771 772
		return smc_connect_decline_fallback(smc,
						    SMC_CLC_DECL_GETVLANERR);
773 774

	/* check if there is an ism device available */
775
	if (!smc_find_ism_device(smc, &ini) &&
776
	    !smc_connect_ism_vlan_setup(smc, &ini)) {
777 778 779 780 781 782
		/* ISM is supported for this connection */
		ism_supported = true;
		smc_type = SMC_TYPE_D;
	}

	/* check if there is a rdma device available */
783
	if (!smc_find_rdma_device(smc, &ini)) {
784 785 786 787 788 789 790 791 792 793
		/* RDMA is supported for this connection */
		rdma_supported = true;
		if (ism_supported)
			smc_type = SMC_TYPE_B; /* both */
		else
			smc_type = SMC_TYPE_R; /* only RDMA */
	}

	/* if neither ISM nor RDMA are supported, fallback */
	if (!rdma_supported && !ism_supported)
794
		return smc_connect_decline_fallback(smc, SMC_CLC_DECL_NOSMCDEV);
795 796

	/* perform CLC handshake */
797
	rc = smc_connect_clc(smc, smc_type, &aclc, &ini);
798
	if (rc) {
799
		smc_connect_ism_vlan_cleanup(smc, ism_supported, &ini);
800
		return smc_connect_decline_fallback(smc, rc);
801
	}
802

803 804
	/* depending on previous steps, connect using rdma or ism */
	if (rdma_supported && aclc.hdr.path == SMC_TYPE_R)
805
		rc = smc_connect_rdma(smc, &aclc, &ini);
806
	else if (ism_supported && aclc.hdr.path == SMC_TYPE_D)
807
		rc = smc_connect_ism(smc, &aclc, &ini);
808
	else
809
		rc = SMC_CLC_DECL_MODEUNSUPP;
810
	if (rc) {
811
		smc_connect_ism_vlan_cleanup(smc, ism_supported, &ini);
812
		return smc_connect_decline_fallback(smc, rc);
813
	}
814

815
	smc_connect_ism_vlan_cleanup(smc, ism_supported, &ini);
816
	return 0;
817 818
}

819 820 821 822
static void smc_connect_work(struct work_struct *work)
{
	struct smc_sock *smc = container_of(work, struct smc_sock,
					    connect_work);
823 824
	long timeo = smc->sk.sk_sndtimeo;
	int rc = 0;
825

826 827 828
	if (!timeo)
		timeo = MAX_SCHEDULE_TIMEOUT;
	lock_sock(smc->clcsock->sk);
829 830
	if (smc->clcsock->sk->sk_err) {
		smc->sk.sk_err = smc->clcsock->sk->sk_err;
831 832 833 834 835 836 837
	} else if ((1 << smc->clcsock->sk->sk_state) &
					(TCPF_SYN_SENT | TCP_SYN_RECV)) {
		rc = sk_stream_wait_connect(smc->clcsock->sk, &timeo);
		if ((rc == -EPIPE) &&
		    ((1 << smc->clcsock->sk->sk_state) &
					(TCPF_ESTABLISHED | TCPF_CLOSE_WAIT)))
			rc = 0;
838
	}
839 840 841 842 843 844 845 846
	release_sock(smc->clcsock->sk);
	lock_sock(&smc->sk);
	if (rc != 0 || smc->sk.sk_err) {
		smc->sk.sk_state = SMC_CLOSED;
		if (rc == -EPIPE || rc == -EAGAIN)
			smc->sk.sk_err = EPIPE;
		else if (signal_pending(current))
			smc->sk.sk_err = -sock_intr_errno(timeo);
847
		sock_put(&smc->sk); /* passive closing */
848 849 850 851 852 853 854 855
		goto out;
	}

	rc = __smc_connect(smc);
	if (rc < 0)
		smc->sk.sk_err = -rc;

out:
856 857 858 859 860 861 862 863
	if (!sock_flag(&smc->sk, SOCK_DEAD)) {
		if (smc->sk.sk_err) {
			smc->sk.sk_state_change(&smc->sk);
		} else { /* allow polling before and after fallback decision */
			smc->clcsock->sk->sk_write_space(smc->clcsock->sk);
			smc->sk.sk_write_space(&smc->sk);
		}
	}
864 865 866
	release_sock(&smc->sk);
}

U
Ursula Braun 已提交
867 868 869 870 871 872 873 874 875 876 877 878
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;
879
	if (addr->sa_family != AF_INET && addr->sa_family != AF_INET6)
U
Ursula Braun 已提交
880 881 882 883 884 885 886 887 888 889 890 891 892 893 894
		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 已提交
895
	tcp_sk(smc->clcsock->sk)->syn_smc = 1;
896 897 898 899 900 901 902
	if (smc->connect_nonblock) {
		rc = -EALREADY;
		goto out;
	}
	rc = kernel_connect(smc->clcsock, addr, alen, flags);
	if (rc && rc != -EINPROGRESS)
		goto out;
903 904

	sock_hold(&smc->sk); /* sock put in passive closing */
905 906
	if (smc->use_fallback)
		goto out;
907
	if (flags & O_NONBLOCK) {
908 909
		if (schedule_work(&smc->connect_work))
			smc->connect_nonblock = 1;
910 911 912 913 914 915 916 917
		rc = -EINPROGRESS;
	} else {
		rc = __smc_connect(smc);
		if (rc < 0)
			goto out;
		else
			rc = 0; /* success cases including fallback */
	}
U
Ursula Braun 已提交
918 919 920 921 922 923 924 925 926

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

static int smc_clcsock_accept(struct smc_sock *lsmc, struct smc_sock **new_smc)
{
927 928
	struct socket *new_clcsock = NULL;
	struct sock *lsk = &lsmc->sk;
U
Ursula Braun 已提交
929
	struct sock *new_sk;
930
	int rc = -EINVAL;
U
Ursula Braun 已提交
931

932
	release_sock(lsk);
933
	new_sk = smc_sock_alloc(sock_net(lsk), NULL, lsk->sk_protocol);
U
Ursula Braun 已提交
934 935
	if (!new_sk) {
		rc = -ENOMEM;
936
		lsk->sk_err = ENOMEM;
U
Ursula Braun 已提交
937
		*new_smc = NULL;
938
		lock_sock(lsk);
U
Ursula Braun 已提交
939 940 941 942
		goto out;
	}
	*new_smc = smc_sk(new_sk);

943 944
	mutex_lock(&lsmc->clcsock_release_lock);
	if (lsmc->clcsock)
945
		rc = kernel_accept(lsmc->clcsock, &new_clcsock, SOCK_NONBLOCK);
946
	mutex_unlock(&lsmc->clcsock_release_lock);
947
	lock_sock(lsk);
948
	if  (rc < 0 && rc != -EAGAIN)
949
		lsk->sk_err = -rc;
950
	if (rc < 0 || lsk->sk_state == SMC_CLOSED) {
951
		new_sk->sk_prot->unhash(new_sk);
952 953 954 955
		if (new_clcsock)
			sock_release(new_clcsock);
		new_sk->sk_state = SMC_CLOSED;
		sock_set_flag(new_sk, SOCK_DEAD);
956
		sock_put(new_sk); /* final */
U
Ursula Braun 已提交
957 958 959 960
		*new_smc = NULL;
		goto out;
	}

961 962 963 964
	/* new clcsock has inherited the smc listen-specific sk_data_ready
	 * function; switch it back to the original sk_data_ready function
	 */
	new_clcsock->sk->sk_data_ready = lsmc->clcsk_data_ready;
U
Ursula Braun 已提交
965 966 967 968 969
	(*new_smc)->clcsock = new_clcsock;
out:
	return rc;
}

970 971 972 973 974 975 976
/* 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);

977
	sock_hold(sk); /* sock_put in smc_accept_unlink () */
978 979 980 981 982 983 984 985 986 987 988 989 990 991 992
	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);
993
	sock_put(sk); /* sock_hold in smc_accept_enqueue */
994 995 996 997 998
}

/* remove a sock from the accept queue to bind it to a new socket created
 * for a socket accept call from user space
 */
999 1000
struct sock *smc_accept_dequeue(struct sock *parent,
				struct socket *new_sock)
1001 1002 1003 1004 1005 1006 1007 1008 1009
{
	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) {
1010
			new_sk->sk_prot->unhash(new_sk);
1011 1012 1013 1014
			if (isk->clcsock) {
				sock_release(isk->clcsock);
				isk->clcsock = NULL;
			}
1015
			sock_put(new_sk); /* final */
1016 1017
			continue;
		}
1018
		if (new_sock) {
1019
			sock_graft(new_sk, new_sock);
1020 1021 1022 1023 1024
			if (isk->use_fallback) {
				smc_sk(new_sk)->clcsock->file = new_sock->file;
				isk->clcsock->file->private_data = isk->clcsock;
			}
		}
1025 1026 1027 1028 1029 1030
		return new_sk;
	}
	return NULL;
}

/* clean up for a created but never accepted sock */
1031
void smc_close_non_accepted(struct sock *sk)
1032 1033 1034
{
	struct smc_sock *smc = smc_sk(sk);

U
Ursula Braun 已提交
1035
	sock_hold(sk); /* sock_put below */
1036 1037 1038 1039
	lock_sock(sk);
	if (!sk->sk_lingertime)
		/* wait for peer closing */
		sk->sk_lingertime = SMC_MAX_STREAM_WAIT_TIMEOUT;
1040
	__smc_release(smc);
1041
	release_sock(sk);
U
Ursula Braun 已提交
1042
	sock_put(sk); /* sock_hold above */
1043
	sock_put(sk); /* final sock_put */
1044 1045
}

1046
static int smcr_serv_conf_first_link(struct smc_sock *smc)
U
Ursula Braun 已提交
1047
{
1048
	struct smc_link *link = smc->conn.lnk;
1049
	struct smc_llc_qentry *qentry;
U
Ursula Braun 已提交
1050 1051
	int rc;

1052
	if (smcr_link_reg_rmb(link, smc->conn.rmb_desc))
1053
		return SMC_CLC_DECL_ERR_REGRMB;
1054

U
Ursula Braun 已提交
1055
	/* send CONFIRM LINK request to client over the RoCE fabric */
1056
	rc = smc_llc_send_confirm_link(link, SMC_LLC_REQ);
U
Ursula Braun 已提交
1057
	if (rc < 0)
1058
		return SMC_CLC_DECL_TIMEOUT_CL;
U
Ursula Braun 已提交
1059 1060

	/* receive CONFIRM LINK response from client over the RoCE fabric */
1061 1062 1063
	qentry = smc_llc_wait(link->lgr, link, SMC_LLC_WAIT_TIME,
			      SMC_LLC_CONFIRM_LINK);
	if (!qentry) {
U
Ursula Braun 已提交
1064 1065 1066
		struct smc_clc_msg_decline dclc;

		rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc),
1067
				      SMC_CLC_DECLINE, CLC_WAIT_TIME_SHORT);
1068
		return rc == -EAGAIN ? SMC_CLC_DECL_TIMEOUT_CL : rc;
U
Ursula Braun 已提交
1069
	}
1070
	smc_llc_save_peer_uid(qentry);
1071 1072 1073
	rc = smc_llc_eval_conf_link(qentry, SMC_LLC_RESP);
	smc_llc_flow_qentry_del(&link->lgr->llc_flow_lcl);
	if (rc)
1074 1075
		return SMC_CLC_DECL_RMBE_EC;

1076 1077
	/* confirm_rkey is implicit on 1st contact */
	smc->conn.rmb_desc->is_conf_rkey = true;
1078

1079
	smc_llc_link_active(link);
1080
	smcr_lgr_set_type(link->lgr, SMC_LGR_SINGLE);
1081

1082
	/* initial contact - try to establish second link */
1083
	smc_llc_srv_add_link(link);
1084
	return 0;
U
Ursula Braun 已提交
1085 1086
}

1087 1088
/* listen worker: finish */
static void smc_listen_out(struct smc_sock *new_smc)
1089 1090 1091 1092
{
	struct smc_sock *lsmc = new_smc->listen_smc;
	struct sock *newsmcsk = &new_smc->sk;

1093
	if (lsmc->sk.sk_state == SMC_LISTEN) {
1094
		lock_sock_nested(&lsmc->sk, SINGLE_DEPTH_NESTING);
1095
		smc_accept_enqueue(&lsmc->sk, newsmcsk);
1096
		release_sock(&lsmc->sk);
1097 1098
	} else { /* no longer listening */
		smc_close_non_accepted(newsmcsk);
U
Ursula Braun 已提交
1099 1100
	}

1101 1102 1103 1104
	/* Wake up accept */
	lsmc->sk.sk_data_ready(&lsmc->sk);
	sock_put(&lsmc->sk); /* sock_hold in smc_tcp_listen_work */
}
1105

1106 1107 1108 1109
/* listen worker: finish in state connected */
static void smc_listen_out_connected(struct smc_sock *new_smc)
{
	struct sock *newsmcsk = &new_smc->sk;
1110

1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131
	sk_refcnt_debug_inc(newsmcsk);
	if (newsmcsk->sk_state == SMC_INIT)
		newsmcsk->sk_state = SMC_ACTIVE;

	smc_listen_out(new_smc);
}

/* listen worker: finish in error state */
static void smc_listen_out_err(struct smc_sock *new_smc)
{
	struct sock *newsmcsk = &new_smc->sk;

	if (newsmcsk->sk_state == SMC_INIT)
		sock_put(&new_smc->sk); /* passive closing */
	newsmcsk->sk_state = SMC_CLOSED;

	smc_listen_out(new_smc);
}

/* listen worker: decline and fall back if possible */
static void smc_listen_decline(struct smc_sock *new_smc, int reason_code,
1132
			       bool local_first)
1133 1134
{
	/* RDMA setup failed, switch back to TCP */
1135
	if (local_first)
1136 1137 1138
		smc_lgr_cleanup_early(&new_smc->conn);
	else
		smc_conn_free(&new_smc->conn);
1139 1140 1141 1142
	if (reason_code < 0) { /* error, no fallback possible */
		smc_listen_out_err(new_smc);
		return;
	}
1143
	smc_switch_to_fallback(new_smc);
1144 1145
	new_smc->fallback_rsn = reason_code;
	if (reason_code && reason_code != SMC_CLC_DECL_PEERDECL) {
1146 1147 1148 1149
		if (smc_clc_send_decline(new_smc, reason_code) < 0) {
			smc_listen_out_err(new_smc);
			return;
		}
1150
	}
1151 1152 1153 1154
	smc_listen_out_connected(new_smc);
}

/* listen worker: check prefixes */
1155
static int smc_listen_prfx_check(struct smc_sock *new_smc,
1156 1157 1158 1159
				 struct smc_clc_msg_proposal *pclc)
{
	struct smc_clc_msg_proposal_prefix *pclc_prfx;
	struct socket *newclcsock = new_smc->clcsock;
1160

1161
	pclc_prfx = smc_clc_proposal_get_prefix(pclc);
1162
	if (smc_clc_prfx_match(newclcsock, pclc_prfx))
1163
		return SMC_CLC_DECL_DIFFPREFIX;
1164

1165 1166
	return 0;
}
1167

1168 1169
/* listen worker: initialize connection and buffers */
static int smc_listen_rdma_init(struct smc_sock *new_smc,
1170
				struct smc_init_info *ini)
1171
{
1172 1173
	int rc;

1174
	/* allocate connection / link group */
1175 1176 1177
	rc = smc_conn_create(new_smc, ini);
	if (rc)
		return rc;
1178

1179
	/* create send buffer and rmb */
1180
	if (smc_buf_create(new_smc, false))
1181
		return SMC_CLC_DECL_MEM;
1182

1183 1184 1185
	return 0;
}

1186 1187 1188
/* listen worker: initialize connection and buffers for SMC-D */
static int smc_listen_ism_init(struct smc_sock *new_smc,
			       struct smc_clc_msg_proposal *pclc,
1189
			       struct smc_init_info *ini)
1190 1191
{
	struct smc_clc_msg_smcd *pclc_smcd;
1192
	int rc;
1193 1194

	pclc_smcd = smc_get_clc_msg_smcd(pclc);
1195
	ini->ism_peer_gid = pclc_smcd->gid;
1196 1197 1198
	rc = smc_conn_create(new_smc, ini);
	if (rc)
		return rc;
1199 1200 1201 1202 1203

	/* Check if peer can be reached via ISM device */
	if (smc_ism_cantalk(new_smc->conn.lgr->peer_gid,
			    new_smc->conn.lgr->vlan_id,
			    new_smc->conn.lgr->smcd)) {
1204
		if (ini->first_contact_local)
1205 1206 1207
			smc_lgr_cleanup_early(&new_smc->conn);
		else
			smc_conn_free(&new_smc->conn);
1208
		return SMC_CLC_DECL_SMCDNOTALK;
1209 1210 1211
	}

	/* Create send and receive buffers */
1212 1213
	rc = smc_buf_create(new_smc, true);
	if (rc) {
1214
		if (ini->first_contact_local)
1215 1216 1217
			smc_lgr_cleanup_early(&new_smc->conn);
		else
			smc_conn_free(&new_smc->conn);
1218 1219
		return (rc == -ENOSPC) ? SMC_CLC_DECL_MAX_DMB :
					 SMC_CLC_DECL_MEM;
1220 1221 1222 1223 1224
	}

	return 0;
}

1225
/* listen worker: register buffers */
1226
static int smc_listen_rdma_reg(struct smc_sock *new_smc, bool local_first)
1227
{
1228
	struct smc_connection *conn = &new_smc->conn;
1229

1230
	if (!local_first) {
1231
		if (smcr_lgr_reg_rmbs(conn->lnk, conn->rmb_desc))
1232
			return SMC_CLC_DECL_ERR_REGRMB;
1233
	}
1234
	smc_rmb_sync_sg_for_device(&new_smc->conn);
1235

1236 1237 1238 1239
	return 0;
}

/* listen worker: finish RDMA setup */
1240 1241
static int smc_listen_rdma_finish(struct smc_sock *new_smc,
				  struct smc_clc_msg_accept_confirm *cclc,
1242
				  bool local_first)
1243
{
1244
	struct smc_link *link = new_smc->conn.lnk;
1245
	int reason_code = 0;
1246

1247
	if (local_first)
1248
		smc_link_save_peer_info(link, cclc);
1249

1250
	if (smc_rmb_rtoken_handling(&new_smc->conn, link, cclc)) {
1251
		reason_code = SMC_CLC_DECL_ERR_RTOK;
1252
		goto decline;
1253 1254
	}

1255
	if (local_first) {
1256
		if (smc_ib_ready_link(link)) {
1257
			reason_code = SMC_CLC_DECL_ERR_RDYLNK;
1258
			goto decline;
1259
		}
U
Ursula Braun 已提交
1260
		/* QP confirmation over RoCE fabric */
1261
		smc_llc_flow_initiate(link->lgr, SMC_LLC_FLOW_ADD_LINK);
1262
		reason_code = smcr_serv_conf_first_link(new_smc);
1263
		smc_llc_flow_stop(link->lgr, &link->lgr->llc_flow_lcl);
1264 1265
		if (reason_code)
			goto decline;
1266
	}
1267
	return 0;
1268

1269
decline:
1270
	smc_listen_decline(new_smc, reason_code, local_first);
1271
	return reason_code;
1272
}
U
Ursula Braun 已提交
1273

1274 1275 1276 1277 1278 1279 1280
/* 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);
	struct socket *newclcsock = new_smc->clcsock;
	struct smc_clc_msg_accept_confirm cclc;
1281
	struct smc_clc_msg_proposal_area *buf;
1282
	struct smc_clc_msg_proposal *pclc;
1283
	struct smc_init_info ini = {0};
1284
	bool ism_supported = false;
1285 1286
	int rc = 0;

1287 1288 1289
	if (new_smc->listen_smc->sk.sk_state != SMC_LISTEN)
		return smc_listen_out_err(new_smc);

1290 1291 1292
	if (new_smc->use_fallback) {
		smc_listen_out_connected(new_smc);
		return;
1293 1294
	}

1295 1296
	/* check if peer is smc capable */
	if (!tcp_sk(newclcsock->sk)->syn_smc) {
1297
		smc_switch_to_fallback(new_smc);
1298
		new_smc->fallback_rsn = SMC_CLC_DECL_PEERNOSMC;
1299 1300 1301
		smc_listen_out_connected(new_smc);
		return;
	}
1302

1303 1304 1305
	/* do inband token exchange -
	 * wait for and receive SMC Proposal CLC message
	 */
1306 1307 1308 1309 1310 1311 1312
	buf = kzalloc(sizeof(*buf), GFP_KERNEL);
	if (!buf) {
		rc = SMC_CLC_DECL_MEM;
		goto out_decl;
	}
	pclc = (struct smc_clc_msg_proposal *)buf;
	rc = smc_clc_wait_msg(new_smc, pclc, sizeof(*buf),
1313
			      SMC_CLC_PROPOSAL, CLC_WAIT_TIME);
1314 1315
	if (rc)
		goto out_decl;
1316

1317 1318
	/* IPSec connections opt out of SMC-R optimizations */
	if (using_ipsec(new_smc)) {
1319 1320
		rc = SMC_CLC_DECL_IPSEC;
		goto out_decl;
1321 1322
	}

1323 1324
	/* check for matching IP prefix and subnet length */
	rc = smc_listen_prfx_check(new_smc, pclc);
1325 1326
	if (rc)
		goto out_decl;
1327

K
Karsten Graul 已提交
1328 1329
	/* get vlan id from IP device */
	if (smc_vlan_by_tcpsk(new_smc->clcsock, &ini)) {
1330 1331
		rc = SMC_CLC_DECL_GETVLANERR;
		goto out_decl;
K
Karsten Graul 已提交
1332 1333
	}

1334
	mutex_lock(&smc_server_lgr_pending);
1335 1336 1337 1338
	smc_close_init(new_smc);
	smc_rx_init(new_smc);
	smc_tx_init(new_smc);

1339
	/* check if ISM is available */
1340 1341 1342 1343
	if (pclc->hdr.path == SMC_TYPE_D || pclc->hdr.path == SMC_TYPE_B) {
		ini.is_smcd = true; /* prepare ISM check */
		rc = smc_find_ism_device(new_smc, &ini);
		if (!rc)
1344
			rc = smc_listen_ism_init(new_smc, pclc, &ini);
1345 1346 1347 1348 1349 1350 1351 1352
		if (!rc)
			ism_supported = true;
		else if (pclc->hdr.path == SMC_TYPE_D)
			goto out_unlock; /* skip RDMA and decline */
	}

	/* check if RDMA is available */
	if (!ism_supported) { /* SMC_TYPE_R or SMC_TYPE_B */
1353 1354
		/* prepare RDMA check */
		ini.is_smcd = false;
1355
		ini.ism_dev = NULL;
1356
		ini.ib_lcl = &pclc->lcl;
1357 1358 1359 1360 1361 1362 1363 1364
		rc = smc_find_rdma_device(new_smc, &ini);
		if (rc) {
			/* no RDMA device found */
			if (pclc->hdr.path == SMC_TYPE_B)
				/* neither ISM nor RDMA device found */
				rc = SMC_CLC_DECL_NOSMCDEV;
			goto out_unlock;
		}
1365
		rc = smc_listen_rdma_init(new_smc, &ini);
1366 1367
		if (rc)
			goto out_unlock;
1368
		rc = smc_listen_rdma_reg(new_smc, ini.first_contact_local);
1369 1370
		if (rc)
			goto out_unlock;
1371 1372 1373
	}

	/* send SMC Accept CLC message */
1374
	rc = smc_clc_send_accept(new_smc, ini.first_contact_local);
1375 1376
	if (rc)
		goto out_unlock;
1377

1378 1379
	/* SMC-D does not need this lock any more */
	if (ism_supported)
1380
		mutex_unlock(&smc_server_lgr_pending);
1381

1382
	/* receive SMC Confirm CLC message */
1383 1384 1385
	rc = smc_clc_wait_msg(new_smc, &cclc, sizeof(cclc),
			      SMC_CLC_CONFIRM, CLC_WAIT_TIME);
	if (rc) {
1386
		if (!ism_supported)
1387 1388
			goto out_unlock;
		goto out_decl;
1389 1390 1391
	}

	/* finish worker */
1392
	kfree(buf);
1393
	if (!ism_supported) {
1394
		rc = smc_listen_rdma_finish(new_smc, &cclc,
1395
					    ini.first_contact_local);
1396
		mutex_unlock(&smc_server_lgr_pending);
1397
		if (rc)
1398 1399
			return;
	}
1400 1401
	smc_conn_save_peer_info(new_smc, &cclc);
	smc_listen_out_connected(new_smc);
1402 1403 1404 1405 1406
	return;

out_unlock:
	mutex_unlock(&smc_server_lgr_pending);
out_decl:
1407
	smc_listen_decline(new_smc, rc, ini.first_contact_local);
1408
	kfree(buf);
1409 1410 1411 1412 1413 1414
}

static void smc_tcp_listen_work(struct work_struct *work)
{
	struct smc_sock *lsmc = container_of(work, struct smc_sock,
					     tcp_listen_work);
1415
	struct sock *lsk = &lsmc->sk;
1416 1417 1418
	struct smc_sock *new_smc;
	int rc = 0;

1419 1420
	lock_sock(lsk);
	while (lsk->sk_state == SMC_LISTEN) {
1421
		rc = smc_clcsock_accept(lsmc, &new_smc);
1422
		if (rc) /* clcsock accept queue empty or error */
1423 1424 1425 1426 1427
			goto out;
		if (!new_smc)
			continue;

		new_smc->listen_smc = lsmc;
1428
		new_smc->use_fallback = lsmc->use_fallback;
1429
		new_smc->fallback_rsn = lsmc->fallback_rsn;
1430
		sock_hold(lsk); /* sock_put in smc_listen_work */
1431 1432
		INIT_WORK(&new_smc->smc_listen_work, smc_listen_work);
		smc_copy_sock_settings_to_smc(new_smc);
1433 1434
		new_smc->sk.sk_sndbuf = lsmc->sk.sk_sndbuf;
		new_smc->sk.sk_rcvbuf = lsmc->sk.sk_rcvbuf;
1435 1436 1437
		sock_hold(&new_smc->sk); /* sock_put in passive closing */
		if (!schedule_work(&new_smc->smc_listen_work))
			sock_put(&new_smc->sk);
1438 1439 1440
	}

out:
1441
	release_sock(lsk);
1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458
	sock_put(&lsmc->sk); /* sock_hold in smc_clcsock_data_ready() */
}

static void smc_clcsock_data_ready(struct sock *listen_clcsock)
{
	struct smc_sock *lsmc;

	lsmc = (struct smc_sock *)
	       ((uintptr_t)listen_clcsock->sk_user_data & ~SK_USER_DATA_NOCOPY);
	if (!lsmc)
		return;
	lsmc->clcsk_data_ready(listen_clcsock);
	if (lsmc->sk.sk_state == SMC_LISTEN) {
		sock_hold(&lsmc->sk); /* sock_put in smc_tcp_listen_work() */
		if (!schedule_work(&lsmc->tcp_listen_work))
			sock_put(&lsmc->sk);
	}
1459 1460
}

U
Ursula Braun 已提交
1461 1462 1463 1464 1465 1466 1467 1468 1469 1470
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;
1471 1472
	if ((sk->sk_state != SMC_INIT && sk->sk_state != SMC_LISTEN) ||
	    smc->connect_nonblock)
U
Ursula Braun 已提交
1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483
		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);
1484 1485
	if (!smc->use_fallback)
		tcp_sk(smc->clcsock->sk)->syn_smc = 1;
U
Ursula Braun 已提交
1486

1487 1488 1489 1490 1491 1492 1493
	/* save original sk_data_ready function and establish
	 * smc-specific sk_data_ready function
	 */
	smc->clcsk_data_ready = smc->clcsock->sk->sk_data_ready;
	smc->clcsock->sk->sk_data_ready = smc_clcsock_data_ready;
	smc->clcsock->sk->sk_user_data =
		(void *)((uintptr_t)smc | SK_USER_DATA_NOCOPY);
U
Ursula Braun 已提交
1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506
	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;

out:
	release_sock(sk);
	return rc;
}

static int smc_accept(struct socket *sock, struct socket *new_sock,
1507
		      int flags, bool kern)
U
Ursula Braun 已提交
1508
{
1509 1510
	struct sock *sk = sock->sk, *nsk;
	DECLARE_WAITQUEUE(wait, current);
U
Ursula Braun 已提交
1511
	struct smc_sock *lsmc;
1512 1513
	long timeo;
	int rc = 0;
U
Ursula Braun 已提交
1514 1515

	lsmc = smc_sk(sk);
1516
	sock_hold(sk); /* sock_put below */
U
Ursula Braun 已提交
1517 1518 1519 1520
	lock_sock(sk);

	if (lsmc->sk.sk_state != SMC_LISTEN) {
		rc = -EINVAL;
1521
		release_sock(sk);
U
Ursula Braun 已提交
1522 1523 1524
		goto out;
	}

1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545
	/* 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 已提交
1546

1547 1548
	if (!rc)
		rc = sock_error(nsk);
1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
	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);
1566
			smc_rx_wait(smc_sk(nsk), &timeo, smc_rx_data_available);
1567 1568 1569
			release_sock(nsk);
		}
	}
U
Ursula Braun 已提交
1570 1571

out:
1572
	sock_put(sk); /* sock_hold above */
U
Ursula Braun 已提交
1573 1574 1575 1576
	return rc;
}

static int smc_getname(struct socket *sock, struct sockaddr *addr,
1577
		       int peer)
U
Ursula Braun 已提交
1578 1579 1580
{
	struct smc_sock *smc;

1581 1582
	if (peer && (sock->sk->sk_state != SMC_ACTIVE) &&
	    (sock->sk->sk_state != SMC_APPCLOSEWAIT1))
U
Ursula Braun 已提交
1583 1584 1585 1586
		return -ENOTCONN;

	smc = smc_sk(sock->sk);

1587
	return smc->clcsock->ops->getname(smc->clcsock, addr, peer);
U
Ursula Braun 已提交
1588 1589 1590 1591 1592 1593 1594 1595 1596 1597
}

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);
1598 1599 1600
	if ((sk->sk_state != SMC_ACTIVE) &&
	    (sk->sk_state != SMC_APPCLOSEWAIT1) &&
	    (sk->sk_state != SMC_INIT))
U
Ursula Braun 已提交
1601
		goto out;
1602 1603

	if (msg->msg_flags & MSG_FASTOPEN) {
1604
		if (sk->sk_state == SMC_INIT && !smc->connect_nonblock) {
1605
			smc_switch_to_fallback(smc);
1606
			smc->fallback_rsn = SMC_CLC_DECL_OPTUNSUPP;
1607 1608 1609 1610 1611 1612
		} else {
			rc = -EINVAL;
			goto out;
		}
	}

U
Ursula Braun 已提交
1613 1614 1615
	if (smc->use_fallback)
		rc = smc->clcsock->ops->sendmsg(smc->clcsock, msg, len);
	else
U
Ursula Braun 已提交
1616
		rc = smc_tx_sendmsg(smc, msg, len);
U
Ursula Braun 已提交
1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630
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);
1631 1632 1633 1634 1635
	if (sk->sk_state == SMC_CLOSED && (sk->sk_shutdown & RCV_SHUTDOWN)) {
		/* socket was connected before, no more data to read */
		rc = 0;
		goto out;
	}
1636 1637 1638
	if ((sk->sk_state == SMC_INIT) ||
	    (sk->sk_state == SMC_LISTEN) ||
	    (sk->sk_state == SMC_CLOSED))
U
Ursula Braun 已提交
1639 1640
		goto out;

1641 1642 1643 1644 1645
	if (sk->sk_state == SMC_PEERFINCLOSEWAIT) {
		rc = 0;
		goto out;
	}

S
Stefan Raspl 已提交
1646
	if (smc->use_fallback) {
U
Ursula Braun 已提交
1647
		rc = smc->clcsock->ops->recvmsg(smc->clcsock, msg, len, flags);
S
Stefan Raspl 已提交
1648 1649 1650 1651
	} else {
		msg->msg_namelen = 0;
		rc = smc_rx_recvmsg(smc, msg, NULL, len, flags);
	}
1652

U
Ursula Braun 已提交
1653 1654 1655 1656 1657
out:
	release_sock(sk);
	return rc;
}

A
Al Viro 已提交
1658
static __poll_t smc_accept_poll(struct sock *parent)
1659
{
U
Ursula Braun 已提交
1660
	struct smc_sock *isk = smc_sk(parent);
A
Al Viro 已提交
1661
	__poll_t mask = 0;
1662

U
Ursula Braun 已提交
1663 1664
	spin_lock(&isk->accept_q_lock);
	if (!list_empty(&isk->accept_q))
1665
		mask = EPOLLIN | EPOLLRDNORM;
U
Ursula Braun 已提交
1666
	spin_unlock(&isk->accept_q_lock);
1667

U
Ursula Braun 已提交
1668
	return mask;
1669 1670
}

1671 1672
static __poll_t smc_poll(struct file *file, struct socket *sock,
			     poll_table *wait)
U
Ursula Braun 已提交
1673 1674 1675
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
1676
	__poll_t mask = 0;
U
Ursula Braun 已提交
1677

U
Ursula Braun 已提交
1678
	if (!sk)
1679
		return EPOLLNVAL;
U
Ursula Braun 已提交
1680

U
Ursula Braun 已提交
1681
	smc = smc_sk(sock->sk);
1682
	if (smc->use_fallback) {
1683
		/* delegate to CLC child sock */
1684
		mask = smc->clcsock->ops->poll(file, smc->clcsock, wait);
1685
		sk->sk_err = smc->clcsock->sk->sk_err;
U
Ursula Braun 已提交
1686
	} else {
1687
		if (sk->sk_state != SMC_CLOSED)
1688
			sock_poll_wait(file, sock, wait);
1689
		if (sk->sk_err)
1690
			mask |= EPOLLERR;
1691 1692
		if ((sk->sk_shutdown == SHUTDOWN_MASK) ||
		    (sk->sk_state == SMC_CLOSED))
1693
			mask |= EPOLLHUP;
U
Ursula Braun 已提交
1694 1695
		if (sk->sk_state == SMC_LISTEN) {
			/* woken up by sk_data_ready in smc_listen_work() */
1696 1697 1698 1699 1700
			mask |= smc_accept_poll(sk);
		} else if (smc->use_fallback) { /* as result of connect_work()*/
			mask |= smc->clcsock->ops->poll(file, smc->clcsock,
							   wait);
			sk->sk_err = smc->clcsock->sk->sk_err;
U
Ursula Braun 已提交
1701
		} else {
1702 1703
			if ((sk->sk_state != SMC_INIT &&
			     atomic_read(&smc->conn.sndbuf_space)) ||
U
Ursula Braun 已提交
1704
			    sk->sk_shutdown & SEND_SHUTDOWN) {
1705
				mask |= EPOLLOUT | EPOLLWRNORM;
U
Ursula Braun 已提交
1706 1707 1708 1709 1710
			} else {
				sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
				set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
			}
			if (atomic_read(&smc->conn.bytes_to_rcv))
1711
				mask |= EPOLLIN | EPOLLRDNORM;
U
Ursula Braun 已提交
1712
			if (sk->sk_shutdown & RCV_SHUTDOWN)
1713
				mask |= EPOLLIN | EPOLLRDNORM | EPOLLRDHUP;
U
Ursula Braun 已提交
1714
			if (sk->sk_state == SMC_APPCLOSEWAIT1)
1715
				mask |= EPOLLIN;
1716 1717
			if (smc->conn.urg_state == SMC_URG_VALID)
				mask |= EPOLLPRI;
U
Ursula Braun 已提交
1718
		}
U
Ursula Braun 已提交
1719 1720 1721 1722 1723 1724 1725 1726 1727 1728
	}

	return mask;
}

static int smc_shutdown(struct socket *sock, int how)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -EINVAL;
1729
	int rc1 = 0;
U
Ursula Braun 已提交
1730 1731 1732 1733

	smc = smc_sk(sk);

	if ((how < SHUT_RD) || (how > SHUT_RDWR))
1734
		return rc;
U
Ursula Braun 已提交
1735 1736 1737 1738

	lock_sock(sk);

	rc = -ENOTCONN;
1739
	if ((sk->sk_state != SMC_ACTIVE) &&
1740 1741 1742 1743 1744
	    (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 已提交
1745 1746 1747 1748 1749 1750
		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;
1751
		goto out;
U
Ursula Braun 已提交
1752
	}
1753 1754 1755 1756 1757 1758 1759 1760
	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:
1761 1762
		rc = 0;
		/* nothing more to do because peer is not involved */
1763 1764
		break;
	}
1765 1766
	if (smc->clcsock)
		rc1 = kernel_sock_shutdown(smc->clcsock, how);
1767 1768
	/* map sock_shutdown_cmd constants to sk_shutdown value range */
	sk->sk_shutdown |= how + 1;
U
Ursula Braun 已提交
1769 1770 1771

out:
	release_sock(sk);
1772
	return rc ? rc : rc1;
U
Ursula Braun 已提交
1773 1774 1775
}

static int smc_setsockopt(struct socket *sock, int level, int optname,
1776
			  sockptr_t optval, unsigned int optlen)
U
Ursula Braun 已提交
1777 1778 1779
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
1780
	int val, rc;
U
Ursula Braun 已提交
1781 1782 1783 1784 1785 1786

	smc = smc_sk(sk);

	/* generic setsockopts reaching us here always apply to the
	 * CLC socket
	 */
1787 1788 1789 1790 1791
	if (unlikely(!smc->clcsock->ops->setsockopt))
		rc = -EOPNOTSUPP;
	else
		rc = smc->clcsock->ops->setsockopt(smc->clcsock, level, optname,
						   optval, optlen);
1792 1793 1794 1795 1796
	if (smc->clcsock->sk->sk_err) {
		sk->sk_err = smc->clcsock->sk->sk_err;
		sk->sk_error_report(sk);
	}

1797
	if (optlen < sizeof(int))
1798
		return -EINVAL;
1799
	if (copy_from_sockptr(&val, optval, sizeof(int)))
1800
		return -EFAULT;
1801

1802
	lock_sock(sk);
1803 1804
	if (rc || smc->use_fallback)
		goto out;
1805 1806 1807 1808 1809 1810 1811
	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 */
1812
		if (sk->sk_state == SMC_INIT && !smc->connect_nonblock) {
1813
			smc_switch_to_fallback(smc);
1814
			smc->fallback_rsn = SMC_CLC_DECL_OPTUNSUPP;
1815
		} else {
1816
			rc = -EINVAL;
1817 1818
		}
		break;
1819
	case TCP_NODELAY:
1820 1821 1822
		if (sk->sk_state != SMC_INIT &&
		    sk->sk_state != SMC_LISTEN &&
		    sk->sk_state != SMC_CLOSED) {
1823
			if (val)
1824 1825 1826 1827 1828
				mod_delayed_work(system_wq, &smc->conn.tx_work,
						 0);
		}
		break;
	case TCP_CORK:
1829 1830 1831
		if (sk->sk_state != SMC_INIT &&
		    sk->sk_state != SMC_LISTEN &&
		    sk->sk_state != SMC_CLOSED) {
1832
			if (!val)
1833 1834 1835 1836
				mod_delayed_work(system_wq, &smc->conn.tx_work,
						 0);
		}
		break;
1837 1838 1839
	case TCP_DEFER_ACCEPT:
		smc->sockopt_defer_accept = val;
		break;
1840 1841 1842
	default:
		break;
	}
1843
out:
1844 1845 1846
	release_sock(sk);

	return rc;
U
Ursula Braun 已提交
1847 1848 1849 1850 1851 1852 1853 1854 1855
}

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 */
1856 1857
	if (unlikely(!smc->clcsock->ops->getsockopt))
		return -EOPNOTSUPP;
U
Ursula Braun 已提交
1858 1859 1860 1861 1862 1863 1864
	return smc->clcsock->ops->getsockopt(smc->clcsock, level, optname,
					     optval, optlen);
}

static int smc_ioctl(struct socket *sock, unsigned int cmd,
		     unsigned long arg)
{
S
Stefan Raspl 已提交
1865 1866
	union smc_host_cursor cons, urg;
	struct smc_connection *conn;
U
Ursula Braun 已提交
1867
	struct smc_sock *smc;
1868
	int answ;
U
Ursula Braun 已提交
1869 1870

	smc = smc_sk(sock->sk);
S
Stefan Raspl 已提交
1871
	conn = &smc->conn;
1872
	lock_sock(&smc->sk);
1873
	if (smc->use_fallback) {
1874 1875
		if (!smc->clcsock) {
			release_sock(&smc->sk);
1876
			return -EBADF;
1877 1878 1879 1880
		}
		answ = smc->clcsock->ops->ioctl(smc->clcsock, cmd, arg);
		release_sock(&smc->sk);
		return answ;
1881 1882 1883
	}
	switch (cmd) {
	case SIOCINQ: /* same as FIONREAD */
1884 1885
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
1886
			return -EINVAL;
1887
		}
1888 1889 1890 1891 1892
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED)
			answ = 0;
		else
			answ = atomic_read(&smc->conn.bytes_to_rcv);
1893 1894 1895
		break;
	case SIOCOUTQ:
		/* output queue size (not send + not acked) */
1896 1897
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
1898
			return -EINVAL;
1899
		}
1900 1901 1902 1903 1904
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED)
			answ = 0;
		else
			answ = smc->conn.sndbuf_desc->len -
1905 1906 1907 1908
					atomic_read(&smc->conn.sndbuf_space);
		break;
	case SIOCOUTQNSD:
		/* output queue size (not send only) */
1909 1910
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
1911
			return -EINVAL;
1912
		}
1913 1914 1915 1916 1917
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED)
			answ = 0;
		else
			answ = smc_tx_prepared_sends(&smc->conn);
1918
		break;
S
Stefan Raspl 已提交
1919
	case SIOCATMARK:
1920 1921
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
S
Stefan Raspl 已提交
1922
			return -EINVAL;
1923
		}
S
Stefan Raspl 已提交
1924 1925 1926 1927
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED) {
			answ = 0;
		} else {
1928 1929
			smc_curs_copy(&cons, &conn->local_tx_ctrl.cons, conn);
			smc_curs_copy(&urg, &conn->urg_curs, conn);
S
Stefan Raspl 已提交
1930 1931 1932 1933
			answ = smc_curs_diff(conn->rmb_desc->len,
					     &cons, &urg) == 1;
		}
		break;
1934
	default:
1935
		release_sock(&smc->sk);
1936 1937
		return -ENOIOCTLCMD;
	}
1938
	release_sock(&smc->sk);
1939 1940

	return put_user(answ, (int __user *)arg);
U
Ursula Braun 已提交
1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951
}

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 已提交
1952 1953
	if (sk->sk_state != SMC_ACTIVE) {
		release_sock(sk);
U
Ursula Braun 已提交
1954
		goto out;
S
Stefan Raspl 已提交
1955 1956
	}
	release_sock(sk);
U
Ursula Braun 已提交
1957 1958 1959 1960 1961 1962 1963 1964 1965 1966
	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 已提交
1967 1968 1969 1970 1971 1972
/* 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 已提交
1973 1974
static ssize_t smc_splice_read(struct socket *sock, loff_t *ppos,
			       struct pipe_inode_info *pipe, size_t len,
S
Stefan Raspl 已提交
1975
			       unsigned int flags)
U
Ursula Braun 已提交
1976 1977 1978 1979 1980 1981 1982
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -ENOTCONN;

	smc = smc_sk(sk);
	lock_sock(sk);
1983 1984 1985 1986 1987
	if (sk->sk_state == SMC_CLOSED && (sk->sk_shutdown & RCV_SHUTDOWN)) {
		/* socket was connected before, no more data to read */
		rc = 0;
		goto out;
	}
S
Stefan Raspl 已提交
1988 1989 1990 1991 1992 1993 1994
	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 已提交
1995
		goto out;
S
Stefan Raspl 已提交
1996 1997
	}

U
Ursula Braun 已提交
1998 1999 2000 2001
	if (smc->use_fallback) {
		rc = smc->clcsock->ops->splice_read(smc->clcsock, ppos,
						    pipe, len, flags);
	} else {
S
Stefan Raspl 已提交
2002 2003 2004 2005 2006 2007 2008 2009 2010
		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 已提交
2011 2012 2013
	}
out:
	release_sock(sk);
S
Stefan Raspl 已提交
2014

U
Ursula Braun 已提交
2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027
	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,
2028
	.poll		= smc_poll,
U
Ursula Braun 已提交
2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043
	.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)
{
2044
	int family = (protocol == SMCPROTO_SMC6) ? PF_INET6 : PF_INET;
U
Ursula Braun 已提交
2045 2046 2047 2048 2049 2050 2051 2052 2053
	struct smc_sock *smc;
	struct sock *sk;
	int rc;

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

	rc = -EPROTONOSUPPORT;
2054
	if (protocol != SMCPROTO_SMC && protocol != SMCPROTO_SMC6)
U
Ursula Braun 已提交
2055 2056 2057 2058
		goto out;

	rc = -ENOBUFS;
	sock->ops = &smc_sock_ops;
2059
	sk = smc_sock_alloc(net, sock, protocol);
U
Ursula Braun 已提交
2060 2061 2062 2063 2064
	if (!sk)
		goto out;

	/* create internal TCP socket for CLC handshake and fallback */
	smc = smc_sk(sk);
2065
	smc->use_fallback = false; /* assume rdma capability first */
2066
	smc->fallback_rsn = 0;
2067 2068
	rc = sock_create_kern(net, family, SOCK_STREAM, IPPROTO_TCP,
			      &smc->clcsock);
2069
	if (rc) {
U
Ursula Braun 已提交
2070
		sk_common_release(sk);
2071 2072
		goto out;
	}
U
Ursula Braun 已提交
2073 2074
	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 已提交
2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085

out:
	return rc;
}

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

2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104
unsigned int smc_net_id;

static __net_init int smc_net_init(struct net *net)
{
	return smc_pnet_net_init(net);
}

static void __net_exit smc_net_exit(struct net *net)
{
	smc_pnet_net_exit(net);
}

static struct pernet_operations smc_net_ops = {
	.init = smc_net_init,
	.exit = smc_net_exit,
	.id   = &smc_net_id,
	.size = sizeof(struct smc_net),
};

U
Ursula Braun 已提交
2105 2106 2107 2108
static int __init smc_init(void)
{
	int rc;

2109 2110 2111 2112
	rc = register_pernet_subsys(&smc_net_ops);
	if (rc)
		return rc;

2113 2114
	rc = smc_pnet_init();
	if (rc)
Y
YueHaibing 已提交
2115
		goto out_pernet_subsys;
2116

2117 2118 2119 2120 2121 2122
	rc = smc_core_init();
	if (rc) {
		pr_err("%s: smc_core_init fails with %d\n", __func__, rc);
		goto out_pnet;
	}

U
Ursula Braun 已提交
2123 2124 2125
	rc = smc_llc_init();
	if (rc) {
		pr_err("%s: smc_llc_init fails with %d\n", __func__, rc);
2126
		goto out_core;
U
Ursula Braun 已提交
2127 2128
	}

2129 2130 2131
	rc = smc_cdc_init();
	if (rc) {
		pr_err("%s: smc_cdc_init fails with %d\n", __func__, rc);
2132
		goto out_core;
2133 2134
	}

U
Ursula Braun 已提交
2135 2136
	rc = proto_register(&smc_proto, 1);
	if (rc) {
2137
		pr_err("%s: proto_register(v4) fails with %d\n", __func__, rc);
2138
		goto out_core;
U
Ursula Braun 已提交
2139 2140
	}

2141 2142 2143 2144 2145 2146
	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 已提交
2147 2148 2149
	rc = sock_register(&smc_sock_family_ops);
	if (rc) {
		pr_err("%s: sock_register fails with %d\n", __func__, rc);
2150
		goto out_proto6;
U
Ursula Braun 已提交
2151
	}
2152
	INIT_HLIST_HEAD(&smc_v4_hashinfo.ht);
2153
	INIT_HLIST_HEAD(&smc_v6_hashinfo.ht);
U
Ursula Braun 已提交
2154

2155 2156 2157 2158 2159 2160
	rc = smc_ib_register_client();
	if (rc) {
		pr_err("%s: ib_register fails with %d\n", __func__, rc);
		goto out_sock;
	}

U
Ursula Braun 已提交
2161
	static_branch_enable(&tcp_have_smc);
U
Ursula Braun 已提交
2162 2163
	return 0;

2164 2165
out_sock:
	sock_unregister(PF_SMC);
2166 2167
out_proto6:
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
2168 2169
out_proto:
	proto_unregister(&smc_proto);
2170 2171
out_core:
	smc_core_exit();
2172 2173
out_pnet:
	smc_pnet_exit();
Y
YueHaibing 已提交
2174 2175 2176
out_pernet_subsys:
	unregister_pernet_subsys(&smc_net_ops);

U
Ursula Braun 已提交
2177 2178 2179 2180 2181
	return rc;
}

static void __exit smc_exit(void)
{
U
Ursula Braun 已提交
2182
	static_branch_disable(&tcp_have_smc);
U
Ursula Braun 已提交
2183
	sock_unregister(PF_SMC);
2184 2185
	smc_core_exit();
	smc_ib_unregister_client();
2186
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
2187
	proto_unregister(&smc_proto);
2188
	smc_pnet_exit();
2189
	unregister_pernet_subsys(&smc_net_ops);
2190
	rcu_barrier();
U
Ursula Braun 已提交
2191 2192 2193 2194 2195 2196 2197 2198 2199
}

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);