af_smc.c 54.4 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
	if (smc_rmb_rtoken_handling(&new_smc->conn, link, cclc)) {
1237
		reason_code = SMC_CLC_DECL_ERR_RTOK;
1238
		goto decline;
1239 1240
	}

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

1255
decline:
1256
	smc_listen_decline(new_smc, reason_code, local_first);
1257
	return reason_code;
1258
}
U
Ursula Braun 已提交
1259

1260 1261 1262 1263 1264 1265 1266
/* 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;
1267
	struct smc_clc_msg_proposal_area *buf;
1268
	struct smc_clc_msg_proposal *pclc;
1269
	struct smc_init_info ini = {0};
1270
	bool ism_supported = false;
1271 1272
	int rc = 0;

1273 1274 1275
	if (new_smc->listen_smc->sk.sk_state != SMC_LISTEN)
		return smc_listen_out_err(new_smc);

1276 1277 1278
	if (new_smc->use_fallback) {
		smc_listen_out_connected(new_smc);
		return;
1279 1280
	}

1281 1282
	/* check if peer is smc capable */
	if (!tcp_sk(newclcsock->sk)->syn_smc) {
1283
		smc_switch_to_fallback(new_smc);
1284
		new_smc->fallback_rsn = SMC_CLC_DECL_PEERNOSMC;
1285 1286 1287
		smc_listen_out_connected(new_smc);
		return;
	}
1288

1289 1290 1291
	/* do inband token exchange -
	 * wait for and receive SMC Proposal CLC message
	 */
1292 1293 1294 1295 1296 1297 1298
	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),
1299
			      SMC_CLC_PROPOSAL, CLC_WAIT_TIME);
1300 1301
	if (rc)
		goto out_decl;
1302

1303 1304
	/* IPSec connections opt out of SMC-R optimizations */
	if (using_ipsec(new_smc)) {
1305 1306
		rc = SMC_CLC_DECL_IPSEC;
		goto out_decl;
1307 1308
	}

1309 1310
	/* check for matching IP prefix and subnet length */
	rc = smc_listen_prfx_check(new_smc, pclc);
1311 1312
	if (rc)
		goto out_decl;
1313

K
Karsten Graul 已提交
1314 1315
	/* get vlan id from IP device */
	if (smc_vlan_by_tcpsk(new_smc->clcsock, &ini)) {
1316 1317
		rc = SMC_CLC_DECL_GETVLANERR;
		goto out_decl;
K
Karsten Graul 已提交
1318 1319
	}

1320
	mutex_lock(&smc_server_lgr_pending);
1321 1322 1323 1324
	smc_close_init(new_smc);
	smc_rx_init(new_smc);
	smc_tx_init(new_smc);

1325
	/* check if ISM is available */
1326
	if (pclc->hdr.path == SMC_TYPE_D || pclc->hdr.path == SMC_TYPE_B) {
1327 1328
		struct smc_clc_msg_smcd *pclc_smcd = smc_get_clc_msg_smcd(pclc);

1329
		ini.is_smcd = true; /* prepare ISM check */
1330
		ini.ism_peer_gid = pclc_smcd->gid;
1331 1332
		rc = smc_find_ism_device(new_smc, &ini);
		if (!rc)
1333
			rc = smc_listen_ism_init(new_smc, pclc, &ini);
1334 1335 1336 1337 1338 1339 1340 1341
		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 */
1342 1343
		/* prepare RDMA check */
		ini.is_smcd = false;
1344
		ini.ism_dev = NULL;
1345
		ini.ib_lcl = &pclc->lcl;
1346 1347 1348 1349 1350 1351 1352 1353
		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;
		}
1354
		rc = smc_listen_rdma_init(new_smc, &ini);
1355 1356
		if (rc)
			goto out_unlock;
1357
		rc = smc_listen_rdma_reg(new_smc, ini.first_contact_local);
1358 1359
		if (rc)
			goto out_unlock;
1360 1361 1362
	}

	/* send SMC Accept CLC message */
1363
	rc = smc_clc_send_accept(new_smc, ini.first_contact_local);
1364 1365
	if (rc)
		goto out_unlock;
1366

1367 1368
	/* SMC-D does not need this lock any more */
	if (ism_supported)
1369
		mutex_unlock(&smc_server_lgr_pending);
1370

1371
	/* receive SMC Confirm CLC message */
1372 1373 1374
	rc = smc_clc_wait_msg(new_smc, &cclc, sizeof(cclc),
			      SMC_CLC_CONFIRM, CLC_WAIT_TIME);
	if (rc) {
1375
		if (!ism_supported)
1376 1377
			goto out_unlock;
		goto out_decl;
1378 1379 1380
	}

	/* finish worker */
1381
	kfree(buf);
1382
	if (!ism_supported) {
1383
		rc = smc_listen_rdma_finish(new_smc, &cclc,
1384
					    ini.first_contact_local);
1385
		mutex_unlock(&smc_server_lgr_pending);
1386
		if (rc)
1387 1388
			return;
	}
1389 1390
	smc_conn_save_peer_info(new_smc, &cclc);
	smc_listen_out_connected(new_smc);
1391 1392 1393 1394 1395
	return;

out_unlock:
	mutex_unlock(&smc_server_lgr_pending);
out_decl:
1396
	smc_listen_decline(new_smc, rc, ini.first_contact_local);
1397
	kfree(buf);
1398 1399 1400 1401 1402 1403
}

static void smc_tcp_listen_work(struct work_struct *work)
{
	struct smc_sock *lsmc = container_of(work, struct smc_sock,
					     tcp_listen_work);
1404
	struct sock *lsk = &lsmc->sk;
1405 1406 1407
	struct smc_sock *new_smc;
	int rc = 0;

1408 1409
	lock_sock(lsk);
	while (lsk->sk_state == SMC_LISTEN) {
1410
		rc = smc_clcsock_accept(lsmc, &new_smc);
1411
		if (rc) /* clcsock accept queue empty or error */
1412 1413 1414 1415 1416
			goto out;
		if (!new_smc)
			continue;

		new_smc->listen_smc = lsmc;
1417
		new_smc->use_fallback = lsmc->use_fallback;
1418
		new_smc->fallback_rsn = lsmc->fallback_rsn;
1419
		sock_hold(lsk); /* sock_put in smc_listen_work */
1420 1421
		INIT_WORK(&new_smc->smc_listen_work, smc_listen_work);
		smc_copy_sock_settings_to_smc(new_smc);
1422 1423
		new_smc->sk.sk_sndbuf = lsmc->sk.sk_sndbuf;
		new_smc->sk.sk_rcvbuf = lsmc->sk.sk_rcvbuf;
1424 1425 1426
		sock_hold(&new_smc->sk); /* sock_put in passive closing */
		if (!schedule_work(&new_smc->smc_listen_work))
			sock_put(&new_smc->sk);
1427 1428 1429
	}

out:
1430
	release_sock(lsk);
1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
	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);
	}
1448 1449
}

U
Ursula Braun 已提交
1450 1451 1452 1453 1454 1455 1456 1457 1458 1459
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;
1460 1461
	if ((sk->sk_state != SMC_INIT && sk->sk_state != SMC_LISTEN) ||
	    smc->connect_nonblock)
U
Ursula Braun 已提交
1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472
		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);
1473 1474
	if (!smc->use_fallback)
		tcp_sk(smc->clcsock->sk)->syn_smc = 1;
U
Ursula Braun 已提交
1475

1476 1477 1478 1479 1480 1481 1482
	/* 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 已提交
1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495
	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,
1496
		      int flags, bool kern)
U
Ursula Braun 已提交
1497
{
1498 1499
	struct sock *sk = sock->sk, *nsk;
	DECLARE_WAITQUEUE(wait, current);
U
Ursula Braun 已提交
1500
	struct smc_sock *lsmc;
1501 1502
	long timeo;
	int rc = 0;
U
Ursula Braun 已提交
1503 1504

	lsmc = smc_sk(sk);
1505
	sock_hold(sk); /* sock_put below */
U
Ursula Braun 已提交
1506 1507 1508 1509
	lock_sock(sk);

	if (lsmc->sk.sk_state != SMC_LISTEN) {
		rc = -EINVAL;
1510
		release_sock(sk);
U
Ursula Braun 已提交
1511 1512 1513
		goto out;
	}

1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534
	/* 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 已提交
1535

1536 1537
	if (!rc)
		rc = sock_error(nsk);
1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554
	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);
1555
			smc_rx_wait(smc_sk(nsk), &timeo, smc_rx_data_available);
1556 1557 1558
			release_sock(nsk);
		}
	}
U
Ursula Braun 已提交
1559 1560

out:
1561
	sock_put(sk); /* sock_hold above */
U
Ursula Braun 已提交
1562 1563 1564 1565
	return rc;
}

static int smc_getname(struct socket *sock, struct sockaddr *addr,
1566
		       int peer)
U
Ursula Braun 已提交
1567 1568 1569
{
	struct smc_sock *smc;

1570 1571
	if (peer && (sock->sk->sk_state != SMC_ACTIVE) &&
	    (sock->sk->sk_state != SMC_APPCLOSEWAIT1))
U
Ursula Braun 已提交
1572 1573 1574 1575
		return -ENOTCONN;

	smc = smc_sk(sock->sk);

1576
	return smc->clcsock->ops->getname(smc->clcsock, addr, peer);
U
Ursula Braun 已提交
1577 1578 1579 1580 1581 1582 1583 1584 1585 1586
}

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);
1587 1588 1589
	if ((sk->sk_state != SMC_ACTIVE) &&
	    (sk->sk_state != SMC_APPCLOSEWAIT1) &&
	    (sk->sk_state != SMC_INIT))
U
Ursula Braun 已提交
1590
		goto out;
1591 1592

	if (msg->msg_flags & MSG_FASTOPEN) {
1593
		if (sk->sk_state == SMC_INIT && !smc->connect_nonblock) {
1594
			smc_switch_to_fallback(smc);
1595
			smc->fallback_rsn = SMC_CLC_DECL_OPTUNSUPP;
1596 1597 1598 1599 1600 1601
		} else {
			rc = -EINVAL;
			goto out;
		}
	}

U
Ursula Braun 已提交
1602 1603 1604
	if (smc->use_fallback)
		rc = smc->clcsock->ops->sendmsg(smc->clcsock, msg, len);
	else
U
Ursula Braun 已提交
1605
		rc = smc_tx_sendmsg(smc, msg, len);
U
Ursula Braun 已提交
1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619
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);
1620 1621 1622 1623 1624
	if (sk->sk_state == SMC_CLOSED && (sk->sk_shutdown & RCV_SHUTDOWN)) {
		/* socket was connected before, no more data to read */
		rc = 0;
		goto out;
	}
1625 1626 1627
	if ((sk->sk_state == SMC_INIT) ||
	    (sk->sk_state == SMC_LISTEN) ||
	    (sk->sk_state == SMC_CLOSED))
U
Ursula Braun 已提交
1628 1629
		goto out;

1630 1631 1632 1633 1634
	if (sk->sk_state == SMC_PEERFINCLOSEWAIT) {
		rc = 0;
		goto out;
	}

S
Stefan Raspl 已提交
1635
	if (smc->use_fallback) {
U
Ursula Braun 已提交
1636
		rc = smc->clcsock->ops->recvmsg(smc->clcsock, msg, len, flags);
S
Stefan Raspl 已提交
1637 1638 1639 1640
	} else {
		msg->msg_namelen = 0;
		rc = smc_rx_recvmsg(smc, msg, NULL, len, flags);
	}
1641

U
Ursula Braun 已提交
1642 1643 1644 1645 1646
out:
	release_sock(sk);
	return rc;
}

A
Al Viro 已提交
1647
static __poll_t smc_accept_poll(struct sock *parent)
1648
{
U
Ursula Braun 已提交
1649
	struct smc_sock *isk = smc_sk(parent);
A
Al Viro 已提交
1650
	__poll_t mask = 0;
1651

U
Ursula Braun 已提交
1652 1653
	spin_lock(&isk->accept_q_lock);
	if (!list_empty(&isk->accept_q))
1654
		mask = EPOLLIN | EPOLLRDNORM;
U
Ursula Braun 已提交
1655
	spin_unlock(&isk->accept_q_lock);
1656

U
Ursula Braun 已提交
1657
	return mask;
1658 1659
}

1660 1661
static __poll_t smc_poll(struct file *file, struct socket *sock,
			     poll_table *wait)
U
Ursula Braun 已提交
1662 1663 1664
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
1665
	__poll_t mask = 0;
U
Ursula Braun 已提交
1666

U
Ursula Braun 已提交
1667
	if (!sk)
1668
		return EPOLLNVAL;
U
Ursula Braun 已提交
1669

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

	return mask;
}

static int smc_shutdown(struct socket *sock, int how)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -EINVAL;
1718
	int rc1 = 0;
U
Ursula Braun 已提交
1719 1720 1721 1722

	smc = smc_sk(sk);

	if ((how < SHUT_RD) || (how > SHUT_RDWR))
1723
		return rc;
U
Ursula Braun 已提交
1724 1725 1726 1727

	lock_sock(sk);

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

out:
	release_sock(sk);
1761
	return rc ? rc : rc1;
U
Ursula Braun 已提交
1762 1763 1764
}

static int smc_setsockopt(struct socket *sock, int level, int optname,
1765
			  sockptr_t optval, unsigned int optlen)
U
Ursula Braun 已提交
1766 1767 1768
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
1769
	int val, rc;
U
Ursula Braun 已提交
1770 1771 1772 1773 1774 1775

	smc = smc_sk(sk);

	/* generic setsockopts reaching us here always apply to the
	 * CLC socket
	 */
1776 1777 1778 1779 1780
	if (unlikely(!smc->clcsock->ops->setsockopt))
		rc = -EOPNOTSUPP;
	else
		rc = smc->clcsock->ops->setsockopt(smc->clcsock, level, optname,
						   optval, optlen);
1781 1782 1783 1784 1785
	if (smc->clcsock->sk->sk_err) {
		sk->sk_err = smc->clcsock->sk->sk_err;
		sk->sk_error_report(sk);
	}

1786
	if (optlen < sizeof(int))
1787
		return -EINVAL;
1788
	if (copy_from_sockptr(&val, optval, sizeof(int)))
1789
		return -EFAULT;
1790

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

	return rc;
U
Ursula Braun 已提交
1836 1837 1838 1839 1840 1841 1842 1843 1844
}

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 */
1845 1846
	if (unlikely(!smc->clcsock->ops->getsockopt))
		return -EOPNOTSUPP;
U
Ursula Braun 已提交
1847 1848 1849 1850 1851 1852 1853
	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 已提交
1854 1855
	union smc_host_cursor cons, urg;
	struct smc_connection *conn;
U
Ursula Braun 已提交
1856
	struct smc_sock *smc;
1857
	int answ;
U
Ursula Braun 已提交
1858 1859

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

	return put_user(answ, (int __user *)arg);
U
Ursula Braun 已提交
1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940
}

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 已提交
1941 1942
	if (sk->sk_state != SMC_ACTIVE) {
		release_sock(sk);
U
Ursula Braun 已提交
1943
		goto out;
S
Stefan Raspl 已提交
1944 1945
	}
	release_sock(sk);
U
Ursula Braun 已提交
1946 1947 1948 1949 1950 1951 1952 1953 1954 1955
	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 已提交
1956 1957 1958 1959 1960 1961
/* 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 已提交
1962 1963
static ssize_t smc_splice_read(struct socket *sock, loff_t *ppos,
			       struct pipe_inode_info *pipe, size_t len,
S
Stefan Raspl 已提交
1964
			       unsigned int flags)
U
Ursula Braun 已提交
1965 1966 1967 1968 1969 1970 1971
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -ENOTCONN;

	smc = smc_sk(sk);
	lock_sock(sk);
1972 1973 1974 1975 1976
	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 已提交
1977 1978 1979 1980 1981 1982 1983
	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 已提交
1984
		goto out;
S
Stefan Raspl 已提交
1985 1986
	}

U
Ursula Braun 已提交
1987 1988 1989 1990
	if (smc->use_fallback) {
		rc = smc->clcsock->ops->splice_read(smc->clcsock, ppos,
						    pipe, len, flags);
	} else {
S
Stefan Raspl 已提交
1991 1992 1993 1994 1995 1996 1997 1998 1999
		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 已提交
2000 2001 2002
	}
out:
	release_sock(sk);
S
Stefan Raspl 已提交
2003

U
Ursula Braun 已提交
2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016
	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,
2017
	.poll		= smc_poll,
U
Ursula Braun 已提交
2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032
	.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)
{
2033
	int family = (protocol == SMCPROTO_SMC6) ? PF_INET6 : PF_INET;
U
Ursula Braun 已提交
2034 2035 2036 2037 2038 2039 2040 2041 2042
	struct smc_sock *smc;
	struct sock *sk;
	int rc;

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

	rc = -EPROTONOSUPPORT;
2043
	if (protocol != SMCPROTO_SMC && protocol != SMCPROTO_SMC6)
U
Ursula Braun 已提交
2044 2045 2046 2047
		goto out;

	rc = -ENOBUFS;
	sock->ops = &smc_sock_ops;
2048
	sk = smc_sock_alloc(net, sock, protocol);
U
Ursula Braun 已提交
2049 2050 2051 2052 2053
	if (!sk)
		goto out;

	/* create internal TCP socket for CLC handshake and fallback */
	smc = smc_sk(sk);
2054
	smc->use_fallback = false; /* assume rdma capability first */
2055
	smc->fallback_rsn = 0;
2056 2057
	rc = sock_create_kern(net, family, SOCK_STREAM, IPPROTO_TCP,
			      &smc->clcsock);
2058
	if (rc) {
U
Ursula Braun 已提交
2059
		sk_common_release(sk);
2060 2061
		goto out;
	}
U
Ursula Braun 已提交
2062 2063
	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 已提交
2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074

out:
	return rc;
}

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

2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093
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 已提交
2094 2095 2096 2097
static int __init smc_init(void)
{
	int rc;

2098 2099 2100 2101
	rc = register_pernet_subsys(&smc_net_ops);
	if (rc)
		return rc;

2102 2103
	rc = smc_pnet_init();
	if (rc)
Y
YueHaibing 已提交
2104
		goto out_pernet_subsys;
2105

2106 2107 2108 2109 2110 2111
	rc = smc_core_init();
	if (rc) {
		pr_err("%s: smc_core_init fails with %d\n", __func__, rc);
		goto out_pnet;
	}

U
Ursula Braun 已提交
2112 2113 2114
	rc = smc_llc_init();
	if (rc) {
		pr_err("%s: smc_llc_init fails with %d\n", __func__, rc);
2115
		goto out_core;
U
Ursula Braun 已提交
2116 2117
	}

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

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

2130 2131 2132 2133 2134 2135
	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 已提交
2136 2137 2138
	rc = sock_register(&smc_sock_family_ops);
	if (rc) {
		pr_err("%s: sock_register fails with %d\n", __func__, rc);
2139
		goto out_proto6;
U
Ursula Braun 已提交
2140
	}
2141
	INIT_HLIST_HEAD(&smc_v4_hashinfo.ht);
2142
	INIT_HLIST_HEAD(&smc_v6_hashinfo.ht);
U
Ursula Braun 已提交
2143

2144 2145 2146 2147 2148 2149
	rc = smc_ib_register_client();
	if (rc) {
		pr_err("%s: ib_register fails with %d\n", __func__, rc);
		goto out_sock;
	}

U
Ursula Braun 已提交
2150
	static_branch_enable(&tcp_have_smc);
U
Ursula Braun 已提交
2151 2152
	return 0;

2153 2154
out_sock:
	sock_unregister(PF_SMC);
2155 2156
out_proto6:
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
2157 2158
out_proto:
	proto_unregister(&smc_proto);
2159 2160
out_core:
	smc_core_exit();
2161 2162
out_pnet:
	smc_pnet_exit();
Y
YueHaibing 已提交
2163 2164 2165
out_pernet_subsys:
	unregister_pernet_subsys(&smc_net_ops);

U
Ursula Braun 已提交
2166 2167 2168 2169 2170
	return rc;
}

static void __exit smc_exit(void)
{
U
Ursula Braun 已提交
2171
	static_branch_disable(&tcp_have_smc);
U
Ursula Braun 已提交
2172
	sock_unregister(PF_SMC);
2173 2174
	smc_core_exit();
	smc_ib_unregister_client();
2175
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
2176
	proto_unregister(&smc_proto);
2177
	smc_pnet_exit();
2178
	unregister_pernet_subsys(&smc_net_ops);
2179
	rcu_barrier();
U
Ursula Braun 已提交
2180 2181 2182 2183 2184 2185 2186 2187 2188
}

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