af_smc.c 53.8 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 440
	int bufsize = smc_uncompress_bufsize(clc->rmbe_size);

441
	smc->conn.peer_rmbe_idx = clc->rmbe_idx;
442
	smc->conn.local_tx_ctrl.token = ntohl(clc->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 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469
static void smcd_conn_save_peer_info(struct smc_sock *smc,
				     struct smc_clc_msg_accept_confirm *clc)
{
	int bufsize = smc_uncompress_bufsize(clc->dmbe_size);

	smc->conn.peer_rmbe_idx = clc->dmbe_idx;
	smc->conn.peer_token = clc->token;
	/* 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 473 474 475 476 477 478 479
static void smc_link_save_peer_info(struct smc_link *link,
				    struct smc_clc_msg_accept_confirm *clc)
{
	link->peer_qpn = ntoh24(clc->qpn);
	memcpy(link->peer_gid, clc->lcl.gid, SMC_GID_SIZE);
	memcpy(link->peer_mac, clc->lcl.mac, sizeof(link->peer_mac));
	link->peer_psn = ntoh24(clc->psn);
	link->peer_mtu = clc->qp_mtu;
}

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 526 527
/* abort connecting */
static int smc_connect_abort(struct smc_sock *smc, int reason_code,
			     int local_contact)
{
528 529
	bool is_smcd = smc->conn.lgr->is_smcd;

530
	if (local_contact == SMC_FIRST_CONTACT)
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 616 617 618 619
	ini->is_smcd = false;
	ini->ib_lcl = &aclc->lcl;
	ini->ib_clcqpn = ntoh24(aclc->qpn);
	ini->srv_first_contact = aclc->hdr.flag;

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 630 631 632 633 634 635 636
	if (ini->cln_first_contact == SMC_FIRST_CONTACT) {
		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
			if (l->peer_qpn == ntoh24(aclc->qpn) &&
			    !memcmp(l->peer_gid, &aclc->lcl.gid, SMC_GID_SIZE) &&
			    !memcmp(l->peer_mac, &aclc->lcl.mac, sizeof(l->peer_mac))) {
640 641 642 643 644 645 646 647 648 649
				link = l;
				break;
			}
		}
		if (!link)
			return smc_connect_abort(smc, SMC_CLC_DECL_NOSRVLINK,
						 ini->cln_first_contact);
		smc->conn.lnk = link;
	}

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

655
	if (ini->cln_first_contact == SMC_FIRST_CONTACT)
656
		smc_link_save_peer_info(link, aclc);
657

658
	if (smc_rmb_rtoken_handling(&smc->conn, link, aclc))
659
		return smc_connect_abort(smc, SMC_CLC_DECL_ERR_RTOK,
660
					 ini->cln_first_contact);
661

662 663 664
	smc_close_init(smc);
	smc_rx_init(smc);

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

676 677
	reason_code = smc_clc_send_confirm(smc);
	if (reason_code)
678 679
		return smc_connect_abort(smc, reason_code,
					 ini->cln_first_contact);
680 681

	smc_tx_init(smc);
682

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

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

699 700
	return 0;
}
701

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

709 710 711 712
	ini->is_smcd = true;
	ini->ism_gid = aclc->gid;
	ini->srv_first_contact = aclc->hdr.flag;

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

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

	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)
736
		return smc_connect_abort(smc, rc, ini->cln_first_contact);
737
	mutex_unlock(&smc_server_lgr_pending);
738 739

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

	return 0;
}

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

756
	if (smc->use_fallback)
757
		return smc_connect_fallback(smc, smc->fallback_rsn);
758 759 760

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

	/* 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 已提交
767
	/* get vlan id from IP device */
768
	if (smc_vlan_by_tcpsk(smc->clcsock, &ini))
K
Karsten Graul 已提交
769 770
		return smc_connect_decline_fallback(smc,
						    SMC_CLC_DECL_GETVLANERR);
771 772

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

	/* check if there is a rdma device available */
781
	if (!smc_find_rdma_device(smc, &ini)) {
782 783 784 785 786 787 788 789 790 791
		/* 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)
792
		return smc_connect_decline_fallback(smc, SMC_CLC_DECL_NOSMCDEV);
793 794

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

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

813
	smc_connect_ism_vlan_cleanup(smc, ism_supported, &ini);
814
	return 0;
815 816
}

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

824 825 826
	if (!timeo)
		timeo = MAX_SCHEDULE_TIMEOUT;
	lock_sock(smc->clcsock->sk);
827 828
	if (smc->clcsock->sk->sk_err) {
		smc->sk.sk_err = smc->clcsock->sk->sk_err;
829 830 831 832 833 834 835
	} 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;
836
	}
837 838 839 840 841 842 843 844
	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);
845
		sock_put(&smc->sk); /* passive closing */
846 847 848 849 850 851 852 853
		goto out;
	}

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

out:
854 855 856 857 858 859 860 861
	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);
		}
	}
862 863 864
	release_sock(&smc->sk);
}

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

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

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

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

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

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

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

964 965 966 967 968 969 970
/* 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);

971
	sock_hold(sk); /* sock_put in smc_accept_unlink () */
972 973 974 975 976 977 978 979 980 981 982 983 984 985 986
	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);
987
	sock_put(sk); /* sock_hold in smc_accept_enqueue */
988 989 990 991 992
}

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

/* clean up for a created but never accepted sock */
1025
void smc_close_non_accepted(struct sock *sk)
1026 1027 1028
{
	struct smc_sock *smc = smc_sk(sk);

U
Ursula Braun 已提交
1029
	sock_hold(sk); /* sock_put below */
1030 1031 1032 1033
	lock_sock(sk);
	if (!sk->sk_lingertime)
		/* wait for peer closing */
		sk->sk_lingertime = SMC_MAX_STREAM_WAIT_TIMEOUT;
1034
	__smc_release(smc);
1035
	release_sock(sk);
U
Ursula Braun 已提交
1036
	sock_put(sk); /* sock_hold above */
1037
	sock_put(sk); /* final sock_put */
1038 1039
}

1040
static int smcr_serv_conf_first_link(struct smc_sock *smc)
U
Ursula Braun 已提交
1041
{
1042
	struct smc_link *link = smc->conn.lnk;
1043
	struct smc_llc_qentry *qentry;
U
Ursula Braun 已提交
1044 1045
	int rc;

1046
	if (smcr_link_reg_rmb(link, smc->conn.rmb_desc))
1047
		return SMC_CLC_DECL_ERR_REGRMB;
1048

U
Ursula Braun 已提交
1049
	/* send CONFIRM LINK request to client over the RoCE fabric */
1050
	rc = smc_llc_send_confirm_link(link, SMC_LLC_REQ);
U
Ursula Braun 已提交
1051
	if (rc < 0)
1052
		return SMC_CLC_DECL_TIMEOUT_CL;
U
Ursula Braun 已提交
1053 1054

	/* receive CONFIRM LINK response from client over the RoCE fabric */
1055 1056 1057
	qentry = smc_llc_wait(link->lgr, link, SMC_LLC_WAIT_TIME,
			      SMC_LLC_CONFIRM_LINK);
	if (!qentry) {
U
Ursula Braun 已提交
1058 1059 1060
		struct smc_clc_msg_decline dclc;

		rc = smc_clc_wait_msg(smc, &dclc, sizeof(dclc),
1061
				      SMC_CLC_DECLINE, CLC_WAIT_TIME_SHORT);
1062
		return rc == -EAGAIN ? SMC_CLC_DECL_TIMEOUT_CL : rc;
U
Ursula Braun 已提交
1063
	}
1064
	smc_llc_save_peer_uid(qentry);
1065 1066 1067
	rc = smc_llc_eval_conf_link(qentry, SMC_LLC_RESP);
	smc_llc_flow_qentry_del(&link->lgr->llc_flow_lcl);
	if (rc)
1068 1069
		return SMC_CLC_DECL_RMBE_EC;

1070 1071
	/* confirm_rkey is implicit on 1st contact */
	smc->conn.rmb_desc->is_conf_rkey = true;
1072

1073
	smc_llc_link_active(link);
1074
	smcr_lgr_set_type(link->lgr, SMC_LGR_SINGLE);
1075

1076
	/* initial contact - try to establish second link */
1077
	smc_llc_srv_add_link(link);
1078
	return 0;
U
Ursula Braun 已提交
1079 1080
}

1081 1082
/* listen worker: finish */
static void smc_listen_out(struct smc_sock *new_smc)
1083 1084 1085 1086
{
	struct smc_sock *lsmc = new_smc->listen_smc;
	struct sock *newsmcsk = &new_smc->sk;

1087
	if (lsmc->sk.sk_state == SMC_LISTEN) {
1088
		lock_sock_nested(&lsmc->sk, SINGLE_DEPTH_NESTING);
1089
		smc_accept_enqueue(&lsmc->sk, newsmcsk);
1090
		release_sock(&lsmc->sk);
1091 1092
	} else { /* no longer listening */
		smc_close_non_accepted(newsmcsk);
U
Ursula Braun 已提交
1093 1094
	}

1095 1096 1097 1098
	/* Wake up accept */
	lsmc->sk.sk_data_ready(&lsmc->sk);
	sock_put(&lsmc->sk); /* sock_hold in smc_tcp_listen_work */
}
1099

1100 1101 1102 1103
/* listen worker: finish in state connected */
static void smc_listen_out_connected(struct smc_sock *new_smc)
{
	struct sock *newsmcsk = &new_smc->sk;
1104

1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129
	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,
			       int local_contact)
{
	/* RDMA setup failed, switch back to TCP */
	if (local_contact == SMC_FIRST_CONTACT)
1130 1131 1132
		smc_lgr_cleanup_early(&new_smc->conn);
	else
		smc_conn_free(&new_smc->conn);
1133 1134 1135 1136
	if (reason_code < 0) { /* error, no fallback possible */
		smc_listen_out_err(new_smc);
		return;
	}
1137
	smc_switch_to_fallback(new_smc);
1138 1139
	new_smc->fallback_rsn = reason_code;
	if (reason_code && reason_code != SMC_CLC_DECL_PEERDECL) {
1140 1141 1142 1143
		if (smc_clc_send_decline(new_smc, reason_code) < 0) {
			smc_listen_out_err(new_smc);
			return;
		}
1144
	}
1145 1146 1147 1148
	smc_listen_out_connected(new_smc);
}

/* listen worker: check prefixes */
1149
static int smc_listen_prfx_check(struct smc_sock *new_smc,
1150 1151 1152 1153
				 struct smc_clc_msg_proposal *pclc)
{
	struct smc_clc_msg_proposal_prefix *pclc_prfx;
	struct socket *newclcsock = new_smc->clcsock;
1154

1155
	pclc_prfx = smc_clc_proposal_get_prefix(pclc);
1156
	if (smc_clc_prfx_match(newclcsock, pclc_prfx))
1157
		return SMC_CLC_DECL_DIFFPREFIX;
1158

1159 1160
	return 0;
}
1161

1162 1163
/* listen worker: initialize connection and buffers */
static int smc_listen_rdma_init(struct smc_sock *new_smc,
1164
				struct smc_init_info *ini)
1165
{
1166 1167
	int rc;

1168
	/* allocate connection / link group */
1169 1170 1171
	rc = smc_conn_create(new_smc, ini);
	if (rc)
		return rc;
1172

1173
	/* create send buffer and rmb */
1174
	if (smc_buf_create(new_smc, false))
1175
		return SMC_CLC_DECL_MEM;
1176

1177 1178 1179
	return 0;
}

1180 1181 1182
/* 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,
1183
			       struct smc_init_info *ini)
1184 1185
{
	struct smc_clc_msg_smcd *pclc_smcd;
1186
	int rc;
1187 1188

	pclc_smcd = smc_get_clc_msg_smcd(pclc);
1189
	ini->ism_gid = pclc_smcd->gid;
1190 1191 1192
	rc = smc_conn_create(new_smc, ini);
	if (rc)
		return rc;
1193 1194 1195 1196 1197

	/* Check if peer can be reached via ISM device */
	if (smc_ism_cantalk(new_smc->conn.lgr->peer_gid,
			    new_smc->conn.lgr->vlan_id,
			    new_smc->conn.lgr->smcd)) {
1198
		if (ini->cln_first_contact == SMC_FIRST_CONTACT)
1199 1200 1201
			smc_lgr_cleanup_early(&new_smc->conn);
		else
			smc_conn_free(&new_smc->conn);
1202
		return SMC_CLC_DECL_SMCDNOTALK;
1203 1204 1205
	}

	/* Create send and receive buffers */
1206 1207
	rc = smc_buf_create(new_smc, true);
	if (rc) {
1208
		if (ini->cln_first_contact == SMC_FIRST_CONTACT)
1209 1210 1211
			smc_lgr_cleanup_early(&new_smc->conn);
		else
			smc_conn_free(&new_smc->conn);
1212 1213
		return (rc == -ENOSPC) ? SMC_CLC_DECL_MAX_DMB :
					 SMC_CLC_DECL_MEM;
1214 1215 1216 1217 1218
	}

	return 0;
}

1219 1220 1221
/* listen worker: register buffers */
static int smc_listen_rdma_reg(struct smc_sock *new_smc, int local_contact)
{
1222
	struct smc_connection *conn = &new_smc->conn;
1223

1224
	if (local_contact != SMC_FIRST_CONTACT) {
1225
		if (smcr_lgr_reg_rmbs(conn->lnk, conn->rmb_desc))
1226
			return SMC_CLC_DECL_ERR_REGRMB;
1227
	}
1228
	smc_rmb_sync_sg_for_device(&new_smc->conn);
1229

1230 1231 1232 1233
	return 0;
}

/* listen worker: finish RDMA setup */
1234 1235 1236
static int smc_listen_rdma_finish(struct smc_sock *new_smc,
				  struct smc_clc_msg_accept_confirm *cclc,
				  int local_contact)
1237
{
1238
	struct smc_link *link = new_smc->conn.lnk;
1239
	int reason_code = 0;
1240

1241
	if (local_contact == SMC_FIRST_CONTACT)
1242
		smc_link_save_peer_info(link, cclc);
1243

1244
	if (smc_rmb_rtoken_handling(&new_smc->conn, link, cclc)) {
1245
		reason_code = SMC_CLC_DECL_ERR_RTOK;
1246
		goto decline;
1247 1248 1249
	}

	if (local_contact == SMC_FIRST_CONTACT) {
1250
		if (smc_ib_ready_link(link)) {
1251
			reason_code = SMC_CLC_DECL_ERR_RDYLNK;
1252
			goto decline;
1253
		}
U
Ursula Braun 已提交
1254
		/* QP confirmation over RoCE fabric */
1255
		smc_llc_flow_initiate(link->lgr, SMC_LLC_FLOW_ADD_LINK);
1256
		reason_code = smcr_serv_conf_first_link(new_smc);
1257
		smc_llc_flow_stop(link->lgr, &link->lgr->llc_flow_lcl);
1258 1259
		if (reason_code)
			goto decline;
1260
	}
1261
	return 0;
1262

1263 1264
decline:
	smc_listen_decline(new_smc, reason_code, local_contact);
1265
	return reason_code;
1266
}
U
Ursula Braun 已提交
1267

1268 1269 1270 1271 1272 1273 1274 1275
/* 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;
	struct smc_clc_msg_proposal *pclc;
1276
	struct smc_init_info ini = {0};
1277
	bool ism_supported = false;
1278 1279 1280
	u8 buf[SMC_CLC_MAX_LEN];
	int rc = 0;

1281 1282 1283
	if (new_smc->listen_smc->sk.sk_state != SMC_LISTEN)
		return smc_listen_out_err(new_smc);

1284 1285 1286
	if (new_smc->use_fallback) {
		smc_listen_out_connected(new_smc);
		return;
1287 1288
	}

1289 1290
	/* check if peer is smc capable */
	if (!tcp_sk(newclcsock->sk)->syn_smc) {
1291
		smc_switch_to_fallback(new_smc);
1292
		new_smc->fallback_rsn = SMC_CLC_DECL_PEERNOSMC;
1293 1294 1295
		smc_listen_out_connected(new_smc);
		return;
	}
1296

1297 1298 1299 1300
	/* do inband token exchange -
	 * wait for and receive SMC Proposal CLC message
	 */
	pclc = (struct smc_clc_msg_proposal *)&buf;
1301 1302
	rc = smc_clc_wait_msg(new_smc, pclc, SMC_CLC_MAX_LEN,
			      SMC_CLC_PROPOSAL, CLC_WAIT_TIME);
1303 1304
	if (rc)
		goto out_decl;
1305

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

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

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

1323
	mutex_lock(&smc_server_lgr_pending);
1324 1325 1326 1327
	smc_close_init(new_smc);
	smc_rx_init(new_smc);
	smc_tx_init(new_smc);

1328
	/* check if ISM is available */
1329 1330 1331 1332
	if (pclc->hdr.path == SMC_TYPE_D || pclc->hdr.path == SMC_TYPE_B) {
		ini.is_smcd = true; /* prepare ISM check */
		rc = smc_find_ism_device(new_smc, &ini);
		if (!rc)
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.cln_first_contact);
1358 1359
		if (rc)
			goto out_unlock;
1360 1361 1362
	}

	/* send SMC Accept CLC message */
1363
	rc = smc_clc_send_accept(new_smc, ini.cln_first_contact);
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
	if (!ism_supported) {
1382 1383
		rc = smc_listen_rdma_finish(new_smc, &cclc,
					    ini.cln_first_contact);
1384
		mutex_unlock(&smc_server_lgr_pending);
1385
		if (rc)
1386 1387
			return;
	}
1388 1389
	smc_conn_save_peer_info(new_smc, &cclc);
	smc_listen_out_connected(new_smc);
1390 1391 1392 1393 1394
	return;

out_unlock:
	mutex_unlock(&smc_server_lgr_pending);
out_decl:
1395
	smc_listen_decline(new_smc, rc, ini.cln_first_contact);
1396 1397 1398 1399 1400 1401
}

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

1406 1407
	lock_sock(lsk);
	while (lsk->sk_state == SMC_LISTEN) {
1408 1409 1410 1411 1412 1413 1414
		rc = smc_clcsock_accept(lsmc, &new_smc);
		if (rc)
			goto out;
		if (!new_smc)
			continue;

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

out:
1428
	release_sock(lsk);
1429
	sock_put(&lsmc->sk); /* sock_hold in smc_listen */
1430 1431
}

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

	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;
1464 1465 1466
	sock_hold(sk); /* sock_hold in tcp_listen_worker */
	if (!schedule_work(&smc->tcp_listen_work))
		sock_put(sk);
U
Ursula Braun 已提交
1467 1468 1469 1470 1471 1472 1473

out:
	release_sock(sk);
	return rc;
}

static int smc_accept(struct socket *sock, struct socket *new_sock,
1474
		      int flags, bool kern)
U
Ursula Braun 已提交
1475
{
1476 1477
	struct sock *sk = sock->sk, *nsk;
	DECLARE_WAITQUEUE(wait, current);
U
Ursula Braun 已提交
1478
	struct smc_sock *lsmc;
1479 1480
	long timeo;
	int rc = 0;
U
Ursula Braun 已提交
1481 1482

	lsmc = smc_sk(sk);
1483
	sock_hold(sk); /* sock_put below */
U
Ursula Braun 已提交
1484 1485 1486 1487
	lock_sock(sk);

	if (lsmc->sk.sk_state != SMC_LISTEN) {
		rc = -EINVAL;
1488
		release_sock(sk);
U
Ursula Braun 已提交
1489 1490 1491
		goto out;
	}

1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512
	/* 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 已提交
1513

1514 1515
	if (!rc)
		rc = sock_error(nsk);
1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532
	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);
1533
			smc_rx_wait(smc_sk(nsk), &timeo, smc_rx_data_available);
1534 1535 1536
			release_sock(nsk);
		}
	}
U
Ursula Braun 已提交
1537 1538

out:
1539
	sock_put(sk); /* sock_hold above */
U
Ursula Braun 已提交
1540 1541 1542 1543
	return rc;
}

static int smc_getname(struct socket *sock, struct sockaddr *addr,
1544
		       int peer)
U
Ursula Braun 已提交
1545 1546 1547
{
	struct smc_sock *smc;

1548 1549
	if (peer && (sock->sk->sk_state != SMC_ACTIVE) &&
	    (sock->sk->sk_state != SMC_APPCLOSEWAIT1))
U
Ursula Braun 已提交
1550 1551 1552 1553
		return -ENOTCONN;

	smc = smc_sk(sock->sk);

1554
	return smc->clcsock->ops->getname(smc->clcsock, addr, peer);
U
Ursula Braun 已提交
1555 1556 1557 1558 1559 1560 1561 1562 1563 1564
}

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);
1565 1566 1567
	if ((sk->sk_state != SMC_ACTIVE) &&
	    (sk->sk_state != SMC_APPCLOSEWAIT1) &&
	    (sk->sk_state != SMC_INIT))
U
Ursula Braun 已提交
1568
		goto out;
1569 1570

	if (msg->msg_flags & MSG_FASTOPEN) {
1571
		if (sk->sk_state == SMC_INIT && !smc->connect_nonblock) {
1572
			smc_switch_to_fallback(smc);
1573
			smc->fallback_rsn = SMC_CLC_DECL_OPTUNSUPP;
1574 1575 1576 1577 1578 1579
		} else {
			rc = -EINVAL;
			goto out;
		}
	}

U
Ursula Braun 已提交
1580 1581 1582
	if (smc->use_fallback)
		rc = smc->clcsock->ops->sendmsg(smc->clcsock, msg, len);
	else
U
Ursula Braun 已提交
1583
		rc = smc_tx_sendmsg(smc, msg, len);
U
Ursula Braun 已提交
1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597
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);
1598 1599 1600 1601 1602
	if (sk->sk_state == SMC_CLOSED && (sk->sk_shutdown & RCV_SHUTDOWN)) {
		/* socket was connected before, no more data to read */
		rc = 0;
		goto out;
	}
1603 1604 1605
	if ((sk->sk_state == SMC_INIT) ||
	    (sk->sk_state == SMC_LISTEN) ||
	    (sk->sk_state == SMC_CLOSED))
U
Ursula Braun 已提交
1606 1607
		goto out;

1608 1609 1610 1611 1612
	if (sk->sk_state == SMC_PEERFINCLOSEWAIT) {
		rc = 0;
		goto out;
	}

S
Stefan Raspl 已提交
1613
	if (smc->use_fallback) {
U
Ursula Braun 已提交
1614
		rc = smc->clcsock->ops->recvmsg(smc->clcsock, msg, len, flags);
S
Stefan Raspl 已提交
1615 1616 1617 1618
	} else {
		msg->msg_namelen = 0;
		rc = smc_rx_recvmsg(smc, msg, NULL, len, flags);
	}
1619

U
Ursula Braun 已提交
1620 1621 1622 1623 1624
out:
	release_sock(sk);
	return rc;
}

A
Al Viro 已提交
1625
static __poll_t smc_accept_poll(struct sock *parent)
1626
{
U
Ursula Braun 已提交
1627
	struct smc_sock *isk = smc_sk(parent);
A
Al Viro 已提交
1628
	__poll_t mask = 0;
1629

U
Ursula Braun 已提交
1630 1631
	spin_lock(&isk->accept_q_lock);
	if (!list_empty(&isk->accept_q))
1632
		mask = EPOLLIN | EPOLLRDNORM;
U
Ursula Braun 已提交
1633
	spin_unlock(&isk->accept_q_lock);
1634

U
Ursula Braun 已提交
1635
	return mask;
1636 1637
}

1638 1639
static __poll_t smc_poll(struct file *file, struct socket *sock,
			     poll_table *wait)
U
Ursula Braun 已提交
1640 1641 1642
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
1643
	__poll_t mask = 0;
U
Ursula Braun 已提交
1644

U
Ursula Braun 已提交
1645
	if (!sk)
1646
		return EPOLLNVAL;
U
Ursula Braun 已提交
1647

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

	return mask;
}

static int smc_shutdown(struct socket *sock, int how)
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -EINVAL;
1696
	int rc1 = 0;
U
Ursula Braun 已提交
1697 1698 1699 1700

	smc = smc_sk(sk);

	if ((how < SHUT_RD) || (how > SHUT_RDWR))
1701
		return rc;
U
Ursula Braun 已提交
1702 1703 1704 1705

	lock_sock(sk);

	rc = -ENOTCONN;
1706
	if ((sk->sk_state != SMC_ACTIVE) &&
1707 1708 1709 1710 1711
	    (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 已提交
1712 1713 1714 1715 1716 1717
		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;
1718
		goto out;
U
Ursula Braun 已提交
1719
	}
1720 1721 1722 1723 1724 1725 1726 1727
	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:
1728 1729
		rc = 0;
		/* nothing more to do because peer is not involved */
1730 1731
		break;
	}
1732 1733
	if (smc->clcsock)
		rc1 = kernel_sock_shutdown(smc->clcsock, how);
1734 1735
	/* map sock_shutdown_cmd constants to sk_shutdown value range */
	sk->sk_shutdown |= how + 1;
U
Ursula Braun 已提交
1736 1737 1738

out:
	release_sock(sk);
1739
	return rc ? rc : rc1;
U
Ursula Braun 已提交
1740 1741 1742
}

static int smc_setsockopt(struct socket *sock, int level, int optname,
1743
			  sockptr_t optval, unsigned int optlen)
U
Ursula Braun 已提交
1744 1745 1746
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
1747
	int val, rc;
U
Ursula Braun 已提交
1748 1749 1750 1751 1752 1753

	smc = smc_sk(sk);

	/* generic setsockopts reaching us here always apply to the
	 * CLC socket
	 */
1754 1755 1756 1757 1758
	if (unlikely(!smc->clcsock->ops->setsockopt))
		rc = -EOPNOTSUPP;
	else
		rc = smc->clcsock->ops->setsockopt(smc->clcsock, level, optname,
						   optval, optlen);
1759 1760 1761 1762 1763
	if (smc->clcsock->sk->sk_err) {
		sk->sk_err = smc->clcsock->sk->sk_err;
		sk->sk_error_report(sk);
	}

1764
	if (optlen < sizeof(int))
1765
		return -EINVAL;
1766
	if (copy_from_sockptr(&val, optval, sizeof(int)))
1767
		return -EFAULT;
1768

1769
	lock_sock(sk);
1770 1771
	if (rc || smc->use_fallback)
		goto out;
1772 1773 1774 1775 1776 1777 1778
	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 */
1779
		if (sk->sk_state == SMC_INIT && !smc->connect_nonblock) {
1780
			smc_switch_to_fallback(smc);
1781
			smc->fallback_rsn = SMC_CLC_DECL_OPTUNSUPP;
1782
		} else {
1783
			rc = -EINVAL;
1784 1785
		}
		break;
1786
	case TCP_NODELAY:
1787 1788 1789
		if (sk->sk_state != SMC_INIT &&
		    sk->sk_state != SMC_LISTEN &&
		    sk->sk_state != SMC_CLOSED) {
1790
			if (val)
1791 1792 1793 1794 1795
				mod_delayed_work(system_wq, &smc->conn.tx_work,
						 0);
		}
		break;
	case TCP_CORK:
1796 1797 1798
		if (sk->sk_state != SMC_INIT &&
		    sk->sk_state != SMC_LISTEN &&
		    sk->sk_state != SMC_CLOSED) {
1799
			if (!val)
1800 1801 1802 1803
				mod_delayed_work(system_wq, &smc->conn.tx_work,
						 0);
		}
		break;
1804 1805 1806
	case TCP_DEFER_ACCEPT:
		smc->sockopt_defer_accept = val;
		break;
1807 1808 1809
	default:
		break;
	}
1810
out:
1811 1812 1813
	release_sock(sk);

	return rc;
U
Ursula Braun 已提交
1814 1815 1816 1817 1818 1819 1820 1821 1822
}

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 */
1823 1824
	if (unlikely(!smc->clcsock->ops->getsockopt))
		return -EOPNOTSUPP;
U
Ursula Braun 已提交
1825 1826 1827 1828 1829 1830 1831
	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 已提交
1832 1833
	union smc_host_cursor cons, urg;
	struct smc_connection *conn;
U
Ursula Braun 已提交
1834
	struct smc_sock *smc;
1835
	int answ;
U
Ursula Braun 已提交
1836 1837

	smc = smc_sk(sock->sk);
S
Stefan Raspl 已提交
1838
	conn = &smc->conn;
1839
	lock_sock(&smc->sk);
1840
	if (smc->use_fallback) {
1841 1842
		if (!smc->clcsock) {
			release_sock(&smc->sk);
1843
			return -EBADF;
1844 1845 1846 1847
		}
		answ = smc->clcsock->ops->ioctl(smc->clcsock, cmd, arg);
		release_sock(&smc->sk);
		return answ;
1848 1849 1850
	}
	switch (cmd) {
	case SIOCINQ: /* same as FIONREAD */
1851 1852
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
1853
			return -EINVAL;
1854
		}
1855 1856 1857 1858 1859
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED)
			answ = 0;
		else
			answ = atomic_read(&smc->conn.bytes_to_rcv);
1860 1861 1862
		break;
	case SIOCOUTQ:
		/* output queue size (not send + not acked) */
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 = smc->conn.sndbuf_desc->len -
1872 1873 1874 1875
					atomic_read(&smc->conn.sndbuf_space);
		break;
	case SIOCOUTQNSD:
		/* output queue size (not send only) */
1876 1877
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
1878
			return -EINVAL;
1879
		}
1880 1881 1882 1883 1884
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED)
			answ = 0;
		else
			answ = smc_tx_prepared_sends(&smc->conn);
1885
		break;
S
Stefan Raspl 已提交
1886
	case SIOCATMARK:
1887 1888
		if (smc->sk.sk_state == SMC_LISTEN) {
			release_sock(&smc->sk);
S
Stefan Raspl 已提交
1889
			return -EINVAL;
1890
		}
S
Stefan Raspl 已提交
1891 1892 1893 1894
		if (smc->sk.sk_state == SMC_INIT ||
		    smc->sk.sk_state == SMC_CLOSED) {
			answ = 0;
		} else {
1895 1896
			smc_curs_copy(&cons, &conn->local_tx_ctrl.cons, conn);
			smc_curs_copy(&urg, &conn->urg_curs, conn);
S
Stefan Raspl 已提交
1897 1898 1899 1900
			answ = smc_curs_diff(conn->rmb_desc->len,
					     &cons, &urg) == 1;
		}
		break;
1901
	default:
1902
		release_sock(&smc->sk);
1903 1904
		return -ENOIOCTLCMD;
	}
1905
	release_sock(&smc->sk);
1906 1907

	return put_user(answ, (int __user *)arg);
U
Ursula Braun 已提交
1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918
}

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 已提交
1919 1920
	if (sk->sk_state != SMC_ACTIVE) {
		release_sock(sk);
U
Ursula Braun 已提交
1921
		goto out;
S
Stefan Raspl 已提交
1922 1923
	}
	release_sock(sk);
U
Ursula Braun 已提交
1924 1925 1926 1927 1928 1929 1930 1931 1932 1933
	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 已提交
1934 1935 1936 1937 1938 1939
/* 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 已提交
1940 1941
static ssize_t smc_splice_read(struct socket *sock, loff_t *ppos,
			       struct pipe_inode_info *pipe, size_t len,
S
Stefan Raspl 已提交
1942
			       unsigned int flags)
U
Ursula Braun 已提交
1943 1944 1945 1946 1947 1948 1949
{
	struct sock *sk = sock->sk;
	struct smc_sock *smc;
	int rc = -ENOTCONN;

	smc = smc_sk(sk);
	lock_sock(sk);
1950 1951 1952 1953 1954
	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 已提交
1955 1956 1957 1958 1959 1960 1961
	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 已提交
1962
		goto out;
S
Stefan Raspl 已提交
1963 1964
	}

U
Ursula Braun 已提交
1965 1966 1967 1968
	if (smc->use_fallback) {
		rc = smc->clcsock->ops->splice_read(smc->clcsock, ppos,
						    pipe, len, flags);
	} else {
S
Stefan Raspl 已提交
1969 1970 1971 1972 1973 1974 1975 1976 1977
		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 已提交
1978 1979 1980
	}
out:
	release_sock(sk);
S
Stefan Raspl 已提交
1981

U
Ursula Braun 已提交
1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994
	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,
1995
	.poll		= smc_poll,
U
Ursula Braun 已提交
1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010
	.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)
{
2011
	int family = (protocol == SMCPROTO_SMC6) ? PF_INET6 : PF_INET;
U
Ursula Braun 已提交
2012 2013 2014 2015 2016 2017 2018 2019 2020
	struct smc_sock *smc;
	struct sock *sk;
	int rc;

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

	rc = -EPROTONOSUPPORT;
2021
	if (protocol != SMCPROTO_SMC && protocol != SMCPROTO_SMC6)
U
Ursula Braun 已提交
2022 2023 2024 2025
		goto out;

	rc = -ENOBUFS;
	sock->ops = &smc_sock_ops;
2026
	sk = smc_sock_alloc(net, sock, protocol);
U
Ursula Braun 已提交
2027 2028 2029 2030 2031
	if (!sk)
		goto out;

	/* create internal TCP socket for CLC handshake and fallback */
	smc = smc_sk(sk);
2032
	smc->use_fallback = false; /* assume rdma capability first */
2033
	smc->fallback_rsn = 0;
2034 2035
	rc = sock_create_kern(net, family, SOCK_STREAM, IPPROTO_TCP,
			      &smc->clcsock);
2036
	if (rc) {
U
Ursula Braun 已提交
2037
		sk_common_release(sk);
2038 2039
		goto out;
	}
U
Ursula Braun 已提交
2040 2041
	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 已提交
2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052

out:
	return rc;
}

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

2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071
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 已提交
2072 2073 2074 2075
static int __init smc_init(void)
{
	int rc;

2076 2077 2078 2079
	rc = register_pernet_subsys(&smc_net_ops);
	if (rc)
		return rc;

2080 2081
	rc = smc_pnet_init();
	if (rc)
Y
YueHaibing 已提交
2082
		goto out_pernet_subsys;
2083

2084 2085 2086 2087 2088 2089
	rc = smc_core_init();
	if (rc) {
		pr_err("%s: smc_core_init fails with %d\n", __func__, rc);
		goto out_pnet;
	}

U
Ursula Braun 已提交
2090 2091 2092
	rc = smc_llc_init();
	if (rc) {
		pr_err("%s: smc_llc_init fails with %d\n", __func__, rc);
2093
		goto out_core;
U
Ursula Braun 已提交
2094 2095
	}

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

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

2108 2109 2110 2111 2112 2113
	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 已提交
2114 2115 2116
	rc = sock_register(&smc_sock_family_ops);
	if (rc) {
		pr_err("%s: sock_register fails with %d\n", __func__, rc);
2117
		goto out_proto6;
U
Ursula Braun 已提交
2118
	}
2119
	INIT_HLIST_HEAD(&smc_v4_hashinfo.ht);
2120
	INIT_HLIST_HEAD(&smc_v6_hashinfo.ht);
U
Ursula Braun 已提交
2121

2122 2123 2124 2125 2126 2127
	rc = smc_ib_register_client();
	if (rc) {
		pr_err("%s: ib_register fails with %d\n", __func__, rc);
		goto out_sock;
	}

U
Ursula Braun 已提交
2128
	static_branch_enable(&tcp_have_smc);
U
Ursula Braun 已提交
2129 2130
	return 0;

2131 2132
out_sock:
	sock_unregister(PF_SMC);
2133 2134
out_proto6:
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
2135 2136
out_proto:
	proto_unregister(&smc_proto);
2137 2138
out_core:
	smc_core_exit();
2139 2140
out_pnet:
	smc_pnet_exit();
Y
YueHaibing 已提交
2141 2142 2143
out_pernet_subsys:
	unregister_pernet_subsys(&smc_net_ops);

U
Ursula Braun 已提交
2144 2145 2146 2147 2148
	return rc;
}

static void __exit smc_exit(void)
{
U
Ursula Braun 已提交
2149
	static_branch_disable(&tcp_have_smc);
U
Ursula Braun 已提交
2150
	sock_unregister(PF_SMC);
2151 2152
	smc_core_exit();
	smc_ib_unregister_client();
2153
	proto_unregister(&smc_proto6);
U
Ursula Braun 已提交
2154
	proto_unregister(&smc_proto);
2155
	smc_pnet_exit();
2156
	unregister_pernet_subsys(&smc_net_ops);
2157
	rcu_barrier();
U
Ursula Braun 已提交
2158 2159 2160 2161 2162 2163 2164 2165 2166
}

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