af_smc.c 54.2 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
	int rc;
1192

1193 1194 1195
	rc = smc_conn_create(new_smc, ini);
	if (rc)
		return rc;
1196 1197

	/* Create send and receive buffers */
1198 1199
	rc = smc_buf_create(new_smc, true);
	if (rc) {
1200
		if (ini->first_contact_local)
1201 1202 1203
			smc_lgr_cleanup_early(&new_smc->conn);
		else
			smc_conn_free(&new_smc->conn);
1204 1205
		return (rc == -ENOSPC) ? SMC_CLC_DECL_MAX_DMB :
					 SMC_CLC_DECL_MEM;
1206 1207 1208 1209 1210
	}

	return 0;
}

1211
/* listen worker: register buffers */
1212
static int smc_listen_rdma_reg(struct smc_sock *new_smc, bool local_first)
1213
{
1214
	struct smc_connection *conn = &new_smc->conn;
1215

1216
	if (!local_first) {
1217
		if (smcr_lgr_reg_rmbs(conn->lnk, conn->rmb_desc))
1218
			return SMC_CLC_DECL_ERR_REGRMB;
1219
	}
1220
	smc_rmb_sync_sg_for_device(&new_smc->conn);
1221

1222 1223 1224 1225
	return 0;
}

/* listen worker: finish RDMA setup */
1226 1227
static int smc_listen_rdma_finish(struct smc_sock *new_smc,
				  struct smc_clc_msg_accept_confirm *cclc,
1228
				  bool local_first)
1229
{
1230
	struct smc_link *link = new_smc->conn.lnk;
1231
	int reason_code = 0;
1232

1233
	if (local_first)
1234
		smc_link_save_peer_info(link, cclc);
1235

1236 1237
	if (smc_rmb_rtoken_handling(&new_smc->conn, link, cclc))
		return SMC_CLC_DECL_ERR_RTOK;
1238

1239
	if (local_first) {
1240 1241
		if (smc_ib_ready_link(link))
			return SMC_CLC_DECL_ERR_RDYLNK;
U
Ursula Braun 已提交
1242
		/* QP confirmation over RoCE fabric */
1243
		smc_llc_flow_initiate(link->lgr, SMC_LLC_FLOW_ADD_LINK);
1244
		reason_code = smcr_serv_conf_first_link(new_smc);
1245
		smc_llc_flow_stop(link->lgr, &link->lgr->llc_flow_lcl);
1246
	}
1247
	return reason_code;
1248
}
U
Ursula Braun 已提交
1249

1250 1251 1252 1253 1254 1255 1256
/* 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;
1257
	struct smc_clc_msg_proposal_area *buf;
1258
	struct smc_clc_msg_proposal *pclc;
1259
	struct smc_init_info ini = {0};
1260
	bool ism_supported = false;
1261 1262
	int rc = 0;

1263 1264 1265
	if (new_smc->listen_smc->sk.sk_state != SMC_LISTEN)
		return smc_listen_out_err(new_smc);

1266 1267 1268
	if (new_smc->use_fallback) {
		smc_listen_out_connected(new_smc);
		return;
1269 1270
	}

1271 1272
	/* check if peer is smc capable */
	if (!tcp_sk(newclcsock->sk)->syn_smc) {
1273
		smc_switch_to_fallback(new_smc);
1274
		new_smc->fallback_rsn = SMC_CLC_DECL_PEERNOSMC;
1275 1276 1277
		smc_listen_out_connected(new_smc);
		return;
	}
1278

1279 1280 1281
	/* do inband token exchange -
	 * wait for and receive SMC Proposal CLC message
	 */
1282 1283 1284 1285 1286 1287 1288
	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),
1289
			      SMC_CLC_PROPOSAL, CLC_WAIT_TIME);
1290 1291
	if (rc)
		goto out_decl;
1292

1293 1294
	/* IPSec connections opt out of SMC-R optimizations */
	if (using_ipsec(new_smc)) {
1295 1296
		rc = SMC_CLC_DECL_IPSEC;
		goto out_decl;
1297 1298
	}

1299 1300
	/* check for matching IP prefix and subnet length */
	rc = smc_listen_prfx_check(new_smc, pclc);
1301 1302
	if (rc)
		goto out_decl;
1303

K
Karsten Graul 已提交
1304 1305
	/* get vlan id from IP device */
	if (smc_vlan_by_tcpsk(new_smc->clcsock, &ini)) {
1306 1307
		rc = SMC_CLC_DECL_GETVLANERR;
		goto out_decl;
K
Karsten Graul 已提交
1308 1309
	}

1310
	mutex_lock(&smc_server_lgr_pending);
1311 1312 1313 1314
	smc_close_init(new_smc);
	smc_rx_init(new_smc);
	smc_tx_init(new_smc);

1315
	/* check if ISM is available */
1316
	if (pclc->hdr.path == SMC_TYPE_D || pclc->hdr.path == SMC_TYPE_B) {
1317 1318
		struct smc_clc_msg_smcd *pclc_smcd = smc_get_clc_msg_smcd(pclc);

1319
		ini.is_smcd = true; /* prepare ISM check */
1320
		ini.ism_peer_gid = pclc_smcd->gid;
1321 1322
		rc = smc_find_ism_device(new_smc, &ini);
		if (!rc)
1323
			rc = smc_listen_ism_init(new_smc, pclc, &ini);
1324 1325 1326 1327 1328 1329 1330 1331
		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 */
1332 1333
		/* prepare RDMA check */
		ini.is_smcd = false;
1334
		ini.ism_dev = NULL;
1335
		ini.ib_lcl = &pclc->lcl;
1336 1337 1338 1339 1340 1341 1342 1343
		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;
		}
1344
		rc = smc_listen_rdma_init(new_smc, &ini);
1345 1346
		if (rc)
			goto out_unlock;
1347
		rc = smc_listen_rdma_reg(new_smc, ini.first_contact_local);
1348 1349
		if (rc)
			goto out_unlock;
1350 1351 1352
	}

	/* send SMC Accept CLC message */
1353
	rc = smc_clc_send_accept(new_smc, ini.first_contact_local);
1354 1355
	if (rc)
		goto out_unlock;
1356

1357 1358
	/* SMC-D does not need this lock any more */
	if (ism_supported)
1359
		mutex_unlock(&smc_server_lgr_pending);
1360

1361
	/* receive SMC Confirm CLC message */
1362 1363 1364
	rc = smc_clc_wait_msg(new_smc, &cclc, sizeof(cclc),
			      SMC_CLC_CONFIRM, CLC_WAIT_TIME);
	if (rc) {
1365
		if (!ism_supported)
1366 1367
			goto out_unlock;
		goto out_decl;
1368 1369 1370
	}

	/* finish worker */
1371
	kfree(buf);
1372
	if (!ism_supported) {
1373
		rc = smc_listen_rdma_finish(new_smc, &cclc,
1374
					    ini.first_contact_local);
1375
		if (rc)
1376 1377
			goto out_unlock;
		mutex_unlock(&smc_server_lgr_pending);
1378
	}
1379 1380
	smc_conn_save_peer_info(new_smc, &cclc);
	smc_listen_out_connected(new_smc);
1381 1382 1383 1384 1385
	return;

out_unlock:
	mutex_unlock(&smc_server_lgr_pending);
out_decl:
1386
	smc_listen_decline(new_smc, rc, ini.first_contact_local);
1387
	kfree(buf);
1388 1389 1390 1391 1392 1393
}

static void smc_tcp_listen_work(struct work_struct *work)
{
	struct smc_sock *lsmc = container_of(work, struct smc_sock,
					     tcp_listen_work);
1394
	struct sock *lsk = &lsmc->sk;
1395 1396 1397
	struct smc_sock *new_smc;
	int rc = 0;

1398 1399
	lock_sock(lsk);
	while (lsk->sk_state == SMC_LISTEN) {
1400
		rc = smc_clcsock_accept(lsmc, &new_smc);
1401
		if (rc) /* clcsock accept queue empty or error */
1402 1403 1404 1405 1406
			goto out;
		if (!new_smc)
			continue;

		new_smc->listen_smc = lsmc;
1407
		new_smc->use_fallback = lsmc->use_fallback;
1408
		new_smc->fallback_rsn = lsmc->fallback_rsn;
1409
		sock_hold(lsk); /* sock_put in smc_listen_work */
1410 1411
		INIT_WORK(&new_smc->smc_listen_work, smc_listen_work);
		smc_copy_sock_settings_to_smc(new_smc);
1412 1413
		new_smc->sk.sk_sndbuf = lsmc->sk.sk_sndbuf;
		new_smc->sk.sk_rcvbuf = lsmc->sk.sk_rcvbuf;
1414 1415 1416
		sock_hold(&new_smc->sk); /* sock_put in passive closing */
		if (!schedule_work(&new_smc->smc_listen_work))
			sock_put(&new_smc->sk);
1417 1418 1419
	}

out:
1420
	release_sock(lsk);
1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437
	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);
	}
1438 1439
}

U
Ursula Braun 已提交
1440 1441 1442 1443 1444 1445 1446 1447 1448 1449
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;
1450 1451
	if ((sk->sk_state != SMC_INIT && sk->sk_state != SMC_LISTEN) ||
	    smc->connect_nonblock)
U
Ursula Braun 已提交
1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462
		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);
1463 1464
	if (!smc->use_fallback)
		tcp_sk(smc->clcsock->sk)->syn_smc = 1;
U
Ursula Braun 已提交
1465

1466 1467 1468 1469 1470 1471 1472
	/* 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 已提交
1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485
	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,
1486
		      int flags, bool kern)
U
Ursula Braun 已提交
1487
{
1488 1489
	struct sock *sk = sock->sk, *nsk;
	DECLARE_WAITQUEUE(wait, current);
U
Ursula Braun 已提交
1490
	struct smc_sock *lsmc;
1491 1492
	long timeo;
	int rc = 0;
U
Ursula Braun 已提交
1493 1494

	lsmc = smc_sk(sk);
1495
	sock_hold(sk); /* sock_put below */
U
Ursula Braun 已提交
1496 1497 1498 1499
	lock_sock(sk);

	if (lsmc->sk.sk_state != SMC_LISTEN) {
		rc = -EINVAL;
1500
		release_sock(sk);
U
Ursula Braun 已提交
1501 1502 1503
		goto out;
	}

1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524
	/* 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 已提交
1525

1526 1527
	if (!rc)
		rc = sock_error(nsk);
1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544
	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);
1545
			smc_rx_wait(smc_sk(nsk), &timeo, smc_rx_data_available);
1546 1547 1548
			release_sock(nsk);
		}
	}
U
Ursula Braun 已提交
1549 1550

out:
1551
	sock_put(sk); /* sock_hold above */
U
Ursula Braun 已提交
1552 1553 1554 1555
	return rc;
}

static int smc_getname(struct socket *sock, struct sockaddr *addr,
1556
		       int peer)
U
Ursula Braun 已提交
1557 1558 1559
{
	struct smc_sock *smc;

1560 1561
	if (peer && (sock->sk->sk_state != SMC_ACTIVE) &&
	    (sock->sk->sk_state != SMC_APPCLOSEWAIT1))
U
Ursula Braun 已提交
1562 1563 1564 1565
		return -ENOTCONN;

	smc = smc_sk(sock->sk);

1566
	return smc->clcsock->ops->getname(smc->clcsock, addr, peer);
U
Ursula Braun 已提交
1567 1568 1569 1570 1571 1572 1573 1574 1575 1576
}

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);
1577 1578 1579
	if ((sk->sk_state != SMC_ACTIVE) &&
	    (sk->sk_state != SMC_APPCLOSEWAIT1) &&
	    (sk->sk_state != SMC_INIT))
U
Ursula Braun 已提交
1580
		goto out;
1581 1582

	if (msg->msg_flags & MSG_FASTOPEN) {
1583
		if (sk->sk_state == SMC_INIT && !smc->connect_nonblock) {
1584
			smc_switch_to_fallback(smc);
1585
			smc->fallback_rsn = SMC_CLC_DECL_OPTUNSUPP;
1586 1587 1588 1589 1590 1591
		} else {
			rc = -EINVAL;
			goto out;
		}
	}

U
Ursula Braun 已提交
1592 1593 1594
	if (smc->use_fallback)
		rc = smc->clcsock->ops->sendmsg(smc->clcsock, msg, len);
	else
U
Ursula Braun 已提交
1595
		rc = smc_tx_sendmsg(smc, msg, len);
U
Ursula Braun 已提交
1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609
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);
1610 1611 1612 1613 1614
	if (sk->sk_state == SMC_CLOSED && (sk->sk_shutdown & RCV_SHUTDOWN)) {
		/* socket was connected before, no more data to read */
		rc = 0;
		goto out;
	}
1615 1616 1617
	if ((sk->sk_state == SMC_INIT) ||
	    (sk->sk_state == SMC_LISTEN) ||
	    (sk->sk_state == SMC_CLOSED))
U
Ursula Braun 已提交
1618 1619
		goto out;

1620 1621 1622 1623 1624
	if (sk->sk_state == SMC_PEERFINCLOSEWAIT) {
		rc = 0;
		goto out;
	}

S
Stefan Raspl 已提交
1625
	if (smc->use_fallback) {
U
Ursula Braun 已提交
1626
		rc = smc->clcsock->ops->recvmsg(smc->clcsock, msg, len, flags);
S
Stefan Raspl 已提交
1627 1628 1629 1630
	} else {
		msg->msg_namelen = 0;
		rc = smc_rx_recvmsg(smc, msg, NULL, len, flags);
	}
1631

U
Ursula Braun 已提交
1632 1633 1634 1635 1636
out:
	release_sock(sk);
	return rc;
}

A
Al Viro 已提交
1637
static __poll_t smc_accept_poll(struct sock *parent)
1638
{
U
Ursula Braun 已提交
1639
	struct smc_sock *isk = smc_sk(parent);
A
Al Viro 已提交
1640
	__poll_t mask = 0;
1641

U
Ursula Braun 已提交
1642 1643
	spin_lock(&isk->accept_q_lock);
	if (!list_empty(&isk->accept_q))
1644
		mask = EPOLLIN | EPOLLRDNORM;
U
Ursula Braun 已提交
1645
	spin_unlock(&isk->accept_q_lock);
1646

U
Ursula Braun 已提交
1647
	return mask;
1648 1649
}

1650 1651
static __poll_t smc_poll(struct file *file, struct socket *sock,
			     poll_table *wait)
U
Ursula Braun 已提交
1652 1653 1654
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
1655
	__poll_t mask = 0;
U
Ursula Braun 已提交
1656

U
Ursula Braun 已提交
1657
	if (!sk)
1658
		return EPOLLNVAL;
U
Ursula Braun 已提交
1659

U
Ursula Braun 已提交
1660
	smc = smc_sk(sock->sk);
1661
	if (smc->use_fallback) {
1662
		/* delegate to CLC child sock */
1663
		mask = smc->clcsock->ops->poll(file, smc->clcsock, wait);
1664
		sk->sk_err = smc->clcsock->sk->sk_err;
U
Ursula Braun 已提交
1665
	} else {
1666
		if (sk->sk_state != SMC_CLOSED)
1667
			sock_poll_wait(file, sock, wait);
1668
		if (sk->sk_err)
1669
			mask |= EPOLLERR;
1670 1671
		if ((sk->sk_shutdown == SHUTDOWN_MASK) ||
		    (sk->sk_state == SMC_CLOSED))
1672
			mask |= EPOLLHUP;
U
Ursula Braun 已提交
1673 1674
		if (sk->sk_state == SMC_LISTEN) {
			/* woken up by sk_data_ready in smc_listen_work() */
1675 1676 1677 1678 1679
			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 已提交
1680
		} else {
1681 1682
			if ((sk->sk_state != SMC_INIT &&
			     atomic_read(&smc->conn.sndbuf_space)) ||
U
Ursula Braun 已提交
1683
			    sk->sk_shutdown & SEND_SHUTDOWN) {
1684
				mask |= EPOLLOUT | EPOLLWRNORM;
U
Ursula Braun 已提交
1685 1686 1687 1688 1689
			} else {
				sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
				set_bit(SOCK_NOSPACE, &sk->sk_socket->flags);
			}
			if (atomic_read(&smc->conn.bytes_to_rcv))
1690
				mask |= EPOLLIN | EPOLLRDNORM;
U
Ursula Braun 已提交
1691
			if (sk->sk_shutdown & RCV_SHUTDOWN)
1692
				mask |= EPOLLIN | EPOLLRDNORM | EPOLLRDHUP;
U
Ursula Braun 已提交
1693
			if (sk->sk_state == SMC_APPCLOSEWAIT1)
1694
				mask |= EPOLLIN;
1695 1696
			if (smc->conn.urg_state == SMC_URG_VALID)
				mask |= EPOLLPRI;
U
Ursula Braun 已提交
1697
		}
U
Ursula Braun 已提交
1698 1699 1700 1701 1702 1703 1704 1705 1706 1707
	}

	return mask;
}

static int smc_shutdown(struct socket *sock, int how)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -EINVAL;
1708
	int rc1 = 0;
U
Ursula Braun 已提交
1709 1710 1711 1712

	smc = smc_sk(sk);

	if ((how < SHUT_RD) || (how > SHUT_RDWR))
1713
		return rc;
U
Ursula Braun 已提交
1714 1715 1716 1717

	lock_sock(sk);

	rc = -ENOTCONN;
1718
	if ((sk->sk_state != SMC_ACTIVE) &&
1719 1720 1721 1722 1723
	    (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 已提交
1724 1725 1726 1727 1728 1729
		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;
1730
		goto out;
U
Ursula Braun 已提交
1731
	}
1732 1733 1734 1735 1736 1737 1738 1739
	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:
1740 1741
		rc = 0;
		/* nothing more to do because peer is not involved */
1742 1743
		break;
	}
1744 1745
	if (smc->clcsock)
		rc1 = kernel_sock_shutdown(smc->clcsock, how);
1746 1747
	/* map sock_shutdown_cmd constants to sk_shutdown value range */
	sk->sk_shutdown |= how + 1;
U
Ursula Braun 已提交
1748 1749 1750

out:
	release_sock(sk);
1751
	return rc ? rc : rc1;
U
Ursula Braun 已提交
1752 1753 1754
}

static int smc_setsockopt(struct socket *sock, int level, int optname,
1755
			  sockptr_t optval, unsigned int optlen)
U
Ursula Braun 已提交
1756 1757 1758
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
1759
	int val, rc;
U
Ursula Braun 已提交
1760 1761 1762 1763 1764 1765

	smc = smc_sk(sk);

	/* generic setsockopts reaching us here always apply to the
	 * CLC socket
	 */
1766 1767 1768 1769 1770
	if (unlikely(!smc->clcsock->ops->setsockopt))
		rc = -EOPNOTSUPP;
	else
		rc = smc->clcsock->ops->setsockopt(smc->clcsock, level, optname,
						   optval, optlen);
1771 1772 1773 1774 1775
	if (smc->clcsock->sk->sk_err) {
		sk->sk_err = smc->clcsock->sk->sk_err;
		sk->sk_error_report(sk);
	}

1776
	if (optlen < sizeof(int))
1777
		return -EINVAL;
1778
	if (copy_from_sockptr(&val, optval, sizeof(int)))
1779
		return -EFAULT;
1780

1781
	lock_sock(sk);
1782 1783
	if (rc || smc->use_fallback)
		goto out;
1784 1785 1786 1787 1788 1789 1790
	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 */
1791
		if (sk->sk_state == SMC_INIT && !smc->connect_nonblock) {
1792
			smc_switch_to_fallback(smc);
1793
			smc->fallback_rsn = SMC_CLC_DECL_OPTUNSUPP;
1794
		} else {
1795
			rc = -EINVAL;
1796 1797
		}
		break;
1798
	case TCP_NODELAY:
1799 1800 1801
		if (sk->sk_state != SMC_INIT &&
		    sk->sk_state != SMC_LISTEN &&
		    sk->sk_state != SMC_CLOSED) {
1802
			if (val)
1803 1804 1805 1806 1807
				mod_delayed_work(system_wq, &smc->conn.tx_work,
						 0);
		}
		break;
	case TCP_CORK:
1808 1809 1810
		if (sk->sk_state != SMC_INIT &&
		    sk->sk_state != SMC_LISTEN &&
		    sk->sk_state != SMC_CLOSED) {
1811
			if (!val)
1812 1813 1814 1815
				mod_delayed_work(system_wq, &smc->conn.tx_work,
						 0);
		}
		break;
1816 1817 1818
	case TCP_DEFER_ACCEPT:
		smc->sockopt_defer_accept = val;
		break;
1819 1820 1821
	default:
		break;
	}
1822
out:
1823 1824 1825
	release_sock(sk);

	return rc;
U
Ursula Braun 已提交
1826 1827 1828 1829 1830 1831 1832 1833 1834
}

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 */
1835 1836
	if (unlikely(!smc->clcsock->ops->getsockopt))
		return -EOPNOTSUPP;
U
Ursula Braun 已提交
1837 1838 1839 1840 1841 1842 1843
	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 已提交
1844 1845
	union smc_host_cursor cons, urg;
	struct smc_connection *conn;
U
Ursula Braun 已提交
1846
	struct smc_sock *smc;
1847
	int answ;
U
Ursula Braun 已提交
1848 1849

	smc = smc_sk(sock->sk);
S
Stefan Raspl 已提交
1850
	conn = &smc->conn;
1851
	lock_sock(&smc->sk);
1852
	if (smc->use_fallback) {
1853 1854
		if (!smc->clcsock) {
			release_sock(&smc->sk);
1855
			return -EBADF;
1856 1857 1858 1859
		}
		answ = smc->clcsock->ops->ioctl(smc->clcsock, cmd, arg);
		release_sock(&smc->sk);
		return answ;
1860 1861 1862
	}
	switch (cmd) {
	case SIOCINQ: /* same as FIONREAD */
1863 1864
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
1865
			return -EINVAL;
1866
		}
1867 1868 1869 1870 1871
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED)
			answ = 0;
		else
			answ = atomic_read(&smc->conn.bytes_to_rcv);
1872 1873 1874
		break;
	case SIOCOUTQ:
		/* output queue size (not send + not acked) */
1875 1876
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
1877
			return -EINVAL;
1878
		}
1879 1880 1881 1882 1883
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED)
			answ = 0;
		else
			answ = smc->conn.sndbuf_desc->len -
1884 1885 1886 1887
					atomic_read(&smc->conn.sndbuf_space);
		break;
	case SIOCOUTQNSD:
		/* output queue size (not send only) */
1888 1889
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
1890
			return -EINVAL;
1891
		}
1892 1893 1894 1895 1896
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED)
			answ = 0;
		else
			answ = smc_tx_prepared_sends(&smc->conn);
1897
		break;
S
Stefan Raspl 已提交
1898
	case SIOCATMARK:
1899 1900
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
S
Stefan Raspl 已提交
1901
			return -EINVAL;
1902
		}
S
Stefan Raspl 已提交
1903 1904 1905 1906
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED) {
			answ = 0;
		} else {
1907 1908
			smc_curs_copy(&cons, &conn->local_tx_ctrl.cons, conn);
			smc_curs_copy(&urg, &conn->urg_curs, conn);
S
Stefan Raspl 已提交
1909 1910 1911 1912
			answ = smc_curs_diff(conn->rmb_desc->len,
					     &cons, &urg) == 1;
		}
		break;
1913
	default:
1914
		release_sock(&smc->sk);
1915 1916
		return -ENOIOCTLCMD;
	}
1917
	release_sock(&smc->sk);
1918 1919

	return put_user(answ, (int __user *)arg);
U
Ursula Braun 已提交
1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930
}

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 已提交
1931 1932
	if (sk->sk_state != SMC_ACTIVE) {
		release_sock(sk);
U
Ursula Braun 已提交
1933
		goto out;
S
Stefan Raspl 已提交
1934 1935
	}
	release_sock(sk);
U
Ursula Braun 已提交
1936 1937 1938 1939 1940 1941 1942 1943 1944 1945
	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 已提交
1946 1947 1948 1949 1950 1951
/* 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 已提交
1952 1953
static ssize_t smc_splice_read(struct socket *sock, loff_t *ppos,
			       struct pipe_inode_info *pipe, size_t len,
S
Stefan Raspl 已提交
1954
			       unsigned int flags)
U
Ursula Braun 已提交
1955 1956 1957 1958 1959 1960 1961
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -ENOTCONN;

	smc = smc_sk(sk);
	lock_sock(sk);
1962 1963 1964 1965 1966
	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 已提交
1967 1968 1969 1970 1971 1972 1973
	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 已提交
1974
		goto out;
S
Stefan Raspl 已提交
1975 1976
	}

U
Ursula Braun 已提交
1977 1978 1979 1980
	if (smc->use_fallback) {
		rc = smc->clcsock->ops->splice_read(smc->clcsock, ppos,
						    pipe, len, flags);
	} else {
S
Stefan Raspl 已提交
1981 1982 1983 1984 1985 1986 1987 1988 1989
		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 已提交
1990 1991 1992
	}
out:
	release_sock(sk);
S
Stefan Raspl 已提交
1993

U
Ursula Braun 已提交
1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006
	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,
2007
	.poll		= smc_poll,
U
Ursula Braun 已提交
2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022
	.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)
{
2023
	int family = (protocol == SMCPROTO_SMC6) ? PF_INET6 : PF_INET;
U
Ursula Braun 已提交
2024 2025 2026 2027 2028 2029 2030 2031 2032
	struct smc_sock *smc;
	struct sock *sk;
	int rc;

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

	rc = -EPROTONOSUPPORT;
2033
	if (protocol != SMCPROTO_SMC && protocol != SMCPROTO_SMC6)
U
Ursula Braun 已提交
2034 2035 2036 2037
		goto out;

	rc = -ENOBUFS;
	sock->ops = &smc_sock_ops;
2038
	sk = smc_sock_alloc(net, sock, protocol);
U
Ursula Braun 已提交
2039 2040 2041 2042 2043
	if (!sk)
		goto out;

	/* create internal TCP socket for CLC handshake and fallback */
	smc = smc_sk(sk);
2044
	smc->use_fallback = false; /* assume rdma capability first */
2045
	smc->fallback_rsn = 0;
2046 2047
	rc = sock_create_kern(net, family, SOCK_STREAM, IPPROTO_TCP,
			      &smc->clcsock);
2048
	if (rc) {
U
Ursula Braun 已提交
2049
		sk_common_release(sk);
2050 2051
		goto out;
	}
U
Ursula Braun 已提交
2052 2053
	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 已提交
2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064

out:
	return rc;
}

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

2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083
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 已提交
2084 2085 2086 2087
static int __init smc_init(void)
{
	int rc;

2088 2089 2090 2091
	rc = register_pernet_subsys(&smc_net_ops);
	if (rc)
		return rc;

2092 2093
	rc = smc_pnet_init();
	if (rc)
Y
YueHaibing 已提交
2094
		goto out_pernet_subsys;
2095

2096 2097 2098 2099 2100 2101
	rc = smc_core_init();
	if (rc) {
		pr_err("%s: smc_core_init fails with %d\n", __func__, rc);
		goto out_pnet;
	}

U
Ursula Braun 已提交
2102 2103 2104
	rc = smc_llc_init();
	if (rc) {
		pr_err("%s: smc_llc_init fails with %d\n", __func__, rc);
2105
		goto out_core;
U
Ursula Braun 已提交
2106 2107
	}

2108 2109 2110
	rc = smc_cdc_init();
	if (rc) {
		pr_err("%s: smc_cdc_init fails with %d\n", __func__, rc);
2111
		goto out_core;
2112 2113
	}

U
Ursula Braun 已提交
2114 2115
	rc = proto_register(&smc_proto, 1);
	if (rc) {
2116
		pr_err("%s: proto_register(v4) fails with %d\n", __func__, rc);
2117
		goto out_core;
U
Ursula Braun 已提交
2118 2119
	}

2120 2121 2122 2123 2124 2125
	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 已提交
2126 2127 2128
	rc = sock_register(&smc_sock_family_ops);
	if (rc) {
		pr_err("%s: sock_register fails with %d\n", __func__, rc);
2129
		goto out_proto6;
U
Ursula Braun 已提交
2130
	}
2131
	INIT_HLIST_HEAD(&smc_v4_hashinfo.ht);
2132
	INIT_HLIST_HEAD(&smc_v6_hashinfo.ht);
U
Ursula Braun 已提交
2133

2134 2135 2136 2137 2138 2139
	rc = smc_ib_register_client();
	if (rc) {
		pr_err("%s: ib_register fails with %d\n", __func__, rc);
		goto out_sock;
	}

U
Ursula Braun 已提交
2140
	static_branch_enable(&tcp_have_smc);
U
Ursula Braun 已提交
2141 2142
	return 0;

2143 2144
out_sock:
	sock_unregister(PF_SMC);
2145 2146
out_proto6:
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
2147 2148
out_proto:
	proto_unregister(&smc_proto);
2149 2150
out_core:
	smc_core_exit();
2151 2152
out_pnet:
	smc_pnet_exit();
Y
YueHaibing 已提交
2153 2154 2155
out_pernet_subsys:
	unregister_pernet_subsys(&smc_net_ops);

U
Ursula Braun 已提交
2156 2157 2158 2159 2160
	return rc;
}

static void __exit smc_exit(void)
{
U
Ursula Braun 已提交
2161
	static_branch_disable(&tcp_have_smc);
U
Ursula Braun 已提交
2162
	sock_unregister(PF_SMC);
2163 2164
	smc_core_exit();
	smc_ib_unregister_client();
2165
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
2166
	proto_unregister(&smc_proto);
2167
	smc_pnet_exit();
2168
	unregister_pernet_subsys(&smc_net_ops);
2169
	rcu_barrier();
U
Ursula Braun 已提交
2170 2171 2172 2173 2174 2175 2176 2177 2178
}

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