xref: /openbmc/linux/net/tls/tls_main.c (revision 76f7164d)
13c4d7559SDave Watson /*
23c4d7559SDave Watson  * Copyright (c) 2016-2017, Mellanox Technologies. All rights reserved.
33c4d7559SDave Watson  * Copyright (c) 2016-2017, Dave Watson <davejwatson@fb.com>. All rights reserved.
43c4d7559SDave Watson  *
53c4d7559SDave Watson  * This software is available to you under a choice of one of two
63c4d7559SDave Watson  * licenses.  You may choose to be licensed under the terms of the GNU
73c4d7559SDave Watson  * General Public License (GPL) Version 2, available from the file
83c4d7559SDave Watson  * COPYING in the main directory of this source tree, or the
93c4d7559SDave Watson  * OpenIB.org BSD license below:
103c4d7559SDave Watson  *
113c4d7559SDave Watson  *     Redistribution and use in source and binary forms, with or
123c4d7559SDave Watson  *     without modification, are permitted provided that the following
133c4d7559SDave Watson  *     conditions are met:
143c4d7559SDave Watson  *
153c4d7559SDave Watson  *      - Redistributions of source code must retain the above
163c4d7559SDave Watson  *        copyright notice, this list of conditions and the following
173c4d7559SDave Watson  *        disclaimer.
183c4d7559SDave Watson  *
193c4d7559SDave Watson  *      - Redistributions in binary form must reproduce the above
203c4d7559SDave Watson  *        copyright notice, this list of conditions and the following
213c4d7559SDave Watson  *        disclaimer in the documentation and/or other materials
223c4d7559SDave Watson  *        provided with the distribution.
233c4d7559SDave Watson  *
243c4d7559SDave Watson  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
253c4d7559SDave Watson  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
263c4d7559SDave Watson  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
273c4d7559SDave Watson  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
283c4d7559SDave Watson  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
293c4d7559SDave Watson  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
303c4d7559SDave Watson  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
313c4d7559SDave Watson  * SOFTWARE.
323c4d7559SDave Watson  */
333c4d7559SDave Watson 
343c4d7559SDave Watson #include <linux/module.h>
353c4d7559SDave Watson 
363c4d7559SDave Watson #include <net/tcp.h>
373c4d7559SDave Watson #include <net/inet_common.h>
383c4d7559SDave Watson #include <linux/highmem.h>
393c4d7559SDave Watson #include <linux/netdevice.h>
403c4d7559SDave Watson #include <linux/sched/signal.h>
41dd0bed16SAtul Gupta #include <linux/inetdevice.h>
423c4d7559SDave Watson 
433c4d7559SDave Watson #include <net/tls.h>
443c4d7559SDave Watson 
453c4d7559SDave Watson MODULE_AUTHOR("Mellanox Technologies");
463c4d7559SDave Watson MODULE_DESCRIPTION("Transport Layer Security Support");
473c4d7559SDave Watson MODULE_LICENSE("Dual BSD/GPL");
48037b0b86SDaniel Borkmann MODULE_ALIAS_TCP_ULP("tls");
493c4d7559SDave Watson 
506d88207fSIlya Lesokhin enum {
51c113187dSBoris Pismenny 	TLSV4,
52c113187dSBoris Pismenny 	TLSV6,
53c113187dSBoris Pismenny 	TLS_NUM_PROTS,
54c113187dSBoris Pismenny };
556d88207fSIlya Lesokhin 
56c113187dSBoris Pismenny static struct proto *saved_tcpv6_prot;
57c113187dSBoris Pismenny static DEFINE_MUTEX(tcpv6_prot_mutex);
5828cb6f1eSJohn Fastabend static struct proto *saved_tcpv4_prot;
5928cb6f1eSJohn Fastabend static DEFINE_MUTEX(tcpv4_prot_mutex);
60dd0bed16SAtul Gupta static LIST_HEAD(device_list);
61df9d4a17SAtul Gupta static DEFINE_SPINLOCK(device_spinlock);
62f66de3eeSBoris Pismenny static struct proto tls_prots[TLS_NUM_PROTS][TLS_NUM_CONFIG][TLS_NUM_CONFIG];
63c46234ebSDave Watson static struct proto_ops tls_sw_proto_ops;
6463a6b3feSAtul Gupta static void build_protos(struct proto prot[TLS_NUM_CONFIG][TLS_NUM_CONFIG],
6563a6b3feSAtul Gupta 			 struct proto *base);
666d88207fSIlya Lesokhin 
67f66de3eeSBoris Pismenny static void update_sk_prot(struct sock *sk, struct tls_context *ctx)
686d88207fSIlya Lesokhin {
69c113187dSBoris Pismenny 	int ip_ver = sk->sk_family == AF_INET6 ? TLSV6 : TLSV4;
70c113187dSBoris Pismenny 
71f66de3eeSBoris Pismenny 	sk->sk_prot = &tls_prots[ip_ver][ctx->tx_conf][ctx->rx_conf];
726d88207fSIlya Lesokhin }
733c4d7559SDave Watson 
743c4d7559SDave Watson int wait_on_pending_writer(struct sock *sk, long *timeo)
753c4d7559SDave Watson {
763c4d7559SDave Watson 	int rc = 0;
773c4d7559SDave Watson 	DEFINE_WAIT_FUNC(wait, woken_wake_function);
783c4d7559SDave Watson 
793c4d7559SDave Watson 	add_wait_queue(sk_sleep(sk), &wait);
803c4d7559SDave Watson 	while (1) {
813c4d7559SDave Watson 		if (!*timeo) {
823c4d7559SDave Watson 			rc = -EAGAIN;
833c4d7559SDave Watson 			break;
843c4d7559SDave Watson 		}
853c4d7559SDave Watson 
863c4d7559SDave Watson 		if (signal_pending(current)) {
873c4d7559SDave Watson 			rc = sock_intr_errno(*timeo);
883c4d7559SDave Watson 			break;
893c4d7559SDave Watson 		}
903c4d7559SDave Watson 
913c4d7559SDave Watson 		if (sk_wait_event(sk, timeo, !sk->sk_write_pending, &wait))
923c4d7559SDave Watson 			break;
933c4d7559SDave Watson 	}
943c4d7559SDave Watson 	remove_wait_queue(sk_sleep(sk), &wait);
953c4d7559SDave Watson 	return rc;
963c4d7559SDave Watson }
973c4d7559SDave Watson 
983c4d7559SDave Watson int tls_push_sg(struct sock *sk,
993c4d7559SDave Watson 		struct tls_context *ctx,
1003c4d7559SDave Watson 		struct scatterlist *sg,
1013c4d7559SDave Watson 		u16 first_offset,
1023c4d7559SDave Watson 		int flags)
1033c4d7559SDave Watson {
1043c4d7559SDave Watson 	int sendpage_flags = flags | MSG_SENDPAGE_NOTLAST;
1053c4d7559SDave Watson 	int ret = 0;
1063c4d7559SDave Watson 	struct page *p;
1073c4d7559SDave Watson 	size_t size;
1083c4d7559SDave Watson 	int offset = first_offset;
1093c4d7559SDave Watson 
1103c4d7559SDave Watson 	size = sg->length - offset;
1113c4d7559SDave Watson 	offset += sg->offset;
1123c4d7559SDave Watson 
113c212d2c7SDave Watson 	ctx->in_tcp_sendpages = true;
1143c4d7559SDave Watson 	while (1) {
1153c4d7559SDave Watson 		if (sg_is_last(sg))
1163c4d7559SDave Watson 			sendpage_flags = flags;
1173c4d7559SDave Watson 
1183c4d7559SDave Watson 		/* is sending application-limited? */
1193c4d7559SDave Watson 		tcp_rate_check_app_limited(sk);
1203c4d7559SDave Watson 		p = sg_page(sg);
1213c4d7559SDave Watson retry:
1223c4d7559SDave Watson 		ret = do_tcp_sendpages(sk, p, offset, size, sendpage_flags);
1233c4d7559SDave Watson 
1243c4d7559SDave Watson 		if (ret != size) {
1253c4d7559SDave Watson 			if (ret > 0) {
1263c4d7559SDave Watson 				offset += ret;
1273c4d7559SDave Watson 				size -= ret;
1283c4d7559SDave Watson 				goto retry;
1293c4d7559SDave Watson 			}
1303c4d7559SDave Watson 
1313c4d7559SDave Watson 			offset -= sg->offset;
1323c4d7559SDave Watson 			ctx->partially_sent_offset = offset;
1333c4d7559SDave Watson 			ctx->partially_sent_record = (void *)sg;
134080324c3SAndre Tomt 			ctx->in_tcp_sendpages = false;
1353c4d7559SDave Watson 			return ret;
1363c4d7559SDave Watson 		}
1373c4d7559SDave Watson 
1383c4d7559SDave Watson 		put_page(p);
1393c4d7559SDave Watson 		sk_mem_uncharge(sk, sg->length);
1403c4d7559SDave Watson 		sg = sg_next(sg);
1413c4d7559SDave Watson 		if (!sg)
1423c4d7559SDave Watson 			break;
1433c4d7559SDave Watson 
1443c4d7559SDave Watson 		offset = sg->offset;
1453c4d7559SDave Watson 		size = sg->length;
1463c4d7559SDave Watson 	}
1473c4d7559SDave Watson 
148c212d2c7SDave Watson 	ctx->in_tcp_sendpages = false;
149c212d2c7SDave Watson 	ctx->sk_write_space(sk);
1503c4d7559SDave Watson 
1513c4d7559SDave Watson 	return 0;
1523c4d7559SDave Watson }
1533c4d7559SDave Watson 
1543c4d7559SDave Watson static int tls_handle_open_record(struct sock *sk, int flags)
1553c4d7559SDave Watson {
1563c4d7559SDave Watson 	struct tls_context *ctx = tls_get_ctx(sk);
1573c4d7559SDave Watson 
1583c4d7559SDave Watson 	if (tls_is_pending_open_record(ctx))
1593c4d7559SDave Watson 		return ctx->push_pending_record(sk, flags);
1603c4d7559SDave Watson 
1613c4d7559SDave Watson 	return 0;
1623c4d7559SDave Watson }
1633c4d7559SDave Watson 
1643c4d7559SDave Watson int tls_proccess_cmsg(struct sock *sk, struct msghdr *msg,
1653c4d7559SDave Watson 		      unsigned char *record_type)
1663c4d7559SDave Watson {
1673c4d7559SDave Watson 	struct cmsghdr *cmsg;
1683c4d7559SDave Watson 	int rc = -EINVAL;
1693c4d7559SDave Watson 
1703c4d7559SDave Watson 	for_each_cmsghdr(cmsg, msg) {
1713c4d7559SDave Watson 		if (!CMSG_OK(msg, cmsg))
1723c4d7559SDave Watson 			return -EINVAL;
1733c4d7559SDave Watson 		if (cmsg->cmsg_level != SOL_TLS)
1743c4d7559SDave Watson 			continue;
1753c4d7559SDave Watson 
1763c4d7559SDave Watson 		switch (cmsg->cmsg_type) {
1773c4d7559SDave Watson 		case TLS_SET_RECORD_TYPE:
1783c4d7559SDave Watson 			if (cmsg->cmsg_len < CMSG_LEN(sizeof(*record_type)))
1793c4d7559SDave Watson 				return -EINVAL;
1803c4d7559SDave Watson 
1813c4d7559SDave Watson 			if (msg->msg_flags & MSG_MORE)
1823c4d7559SDave Watson 				return -EINVAL;
1833c4d7559SDave Watson 
1843c4d7559SDave Watson 			rc = tls_handle_open_record(sk, msg->msg_flags);
1853c4d7559SDave Watson 			if (rc)
1863c4d7559SDave Watson 				return rc;
1873c4d7559SDave Watson 
1883c4d7559SDave Watson 			*record_type = *(unsigned char *)CMSG_DATA(cmsg);
1893c4d7559SDave Watson 			rc = 0;
1903c4d7559SDave Watson 			break;
1913c4d7559SDave Watson 		default:
1923c4d7559SDave Watson 			return -EINVAL;
1933c4d7559SDave Watson 		}
1943c4d7559SDave Watson 	}
1953c4d7559SDave Watson 
1963c4d7559SDave Watson 	return rc;
1973c4d7559SDave Watson }
1983c4d7559SDave Watson 
199a42055e8SVakul Garg int tls_push_partial_record(struct sock *sk, struct tls_context *ctx,
200a42055e8SVakul Garg 			    int flags)
2013c4d7559SDave Watson {
2023c4d7559SDave Watson 	struct scatterlist *sg;
2033c4d7559SDave Watson 	u16 offset;
2043c4d7559SDave Watson 
2053c4d7559SDave Watson 	sg = ctx->partially_sent_record;
2063c4d7559SDave Watson 	offset = ctx->partially_sent_offset;
2073c4d7559SDave Watson 
2083c4d7559SDave Watson 	ctx->partially_sent_record = NULL;
2093c4d7559SDave Watson 	return tls_push_sg(sk, ctx, sg, offset, flags);
2103c4d7559SDave Watson }
2113c4d7559SDave Watson 
212a42055e8SVakul Garg int tls_push_pending_closed_record(struct sock *sk,
213a42055e8SVakul Garg 				   struct tls_context *tls_ctx,
214a42055e8SVakul Garg 				   int flags, long *timeo)
215a42055e8SVakul Garg {
216a42055e8SVakul Garg 	struct tls_sw_context_tx *ctx = tls_sw_ctx_tx(tls_ctx);
217a42055e8SVakul Garg 
218a42055e8SVakul Garg 	if (tls_is_partially_sent_record(tls_ctx) ||
2199932a29aSVakul Garg 	    !list_empty(&ctx->tx_list))
220a42055e8SVakul Garg 		return tls_tx_records(sk, flags);
221a42055e8SVakul Garg 	else
222a42055e8SVakul Garg 		return tls_ctx->push_pending_record(sk, flags);
223a42055e8SVakul Garg }
224a42055e8SVakul Garg 
2253c4d7559SDave Watson static void tls_write_space(struct sock *sk)
2263c4d7559SDave Watson {
2273c4d7559SDave Watson 	struct tls_context *ctx = tls_get_ctx(sk);
228a42055e8SVakul Garg 	struct tls_sw_context_tx *tx_ctx = tls_sw_ctx_tx(ctx);
2293c4d7559SDave Watson 
23067db7cd2SJohn Fastabend 	/* If in_tcp_sendpages call lower protocol write space handler
23167db7cd2SJohn Fastabend 	 * to ensure we wake up any waiting operations there. For example
23267db7cd2SJohn Fastabend 	 * if do_tcp_sendpages where to call sk_wait_event.
23367db7cd2SJohn Fastabend 	 */
23467db7cd2SJohn Fastabend 	if (ctx->in_tcp_sendpages) {
23567db7cd2SJohn Fastabend 		ctx->sk_write_space(sk);
236c212d2c7SDave Watson 		return;
23767db7cd2SJohn Fastabend 	}
238c212d2c7SDave Watson 
239a42055e8SVakul Garg 	/* Schedule the transmission if tx list is ready */
2409932a29aSVakul Garg 	if (is_tx_ready(tx_ctx) && !sk->sk_write_pending) {
241a42055e8SVakul Garg 		/* Schedule the transmission */
242a42055e8SVakul Garg 		if (!test_and_set_bit(BIT_TX_SCHEDULED, &tx_ctx->tx_bitmask))
243a42055e8SVakul Garg 			schedule_delayed_work(&tx_ctx->tx_work.work, 0);
2443c4d7559SDave Watson 	}
2453c4d7559SDave Watson 
2463c4d7559SDave Watson 	ctx->sk_write_space(sk);
2473c4d7559SDave Watson }
2483c4d7559SDave Watson 
24986029d10SSabrina Dubroca static void tls_ctx_free(struct tls_context *ctx)
25086029d10SSabrina Dubroca {
25186029d10SSabrina Dubroca 	if (!ctx)
25286029d10SSabrina Dubroca 		return;
25386029d10SSabrina Dubroca 
25486029d10SSabrina Dubroca 	memzero_explicit(&ctx->crypto_send, sizeof(ctx->crypto_send));
25586029d10SSabrina Dubroca 	memzero_explicit(&ctx->crypto_recv, sizeof(ctx->crypto_recv));
25686029d10SSabrina Dubroca 	kfree(ctx);
25786029d10SSabrina Dubroca }
25886029d10SSabrina Dubroca 
2593c4d7559SDave Watson static void tls_sk_proto_close(struct sock *sk, long timeout)
2603c4d7559SDave Watson {
2613c4d7559SDave Watson 	struct tls_context *ctx = tls_get_ctx(sk);
2623c4d7559SDave Watson 	long timeo = sock_sndtimeo(sk, 0);
2633c4d7559SDave Watson 	void (*sk_proto_close)(struct sock *sk, long timeout);
26498f0a395SEric Dumazet 	bool free_ctx = false;
2653c4d7559SDave Watson 
2663c4d7559SDave Watson 	lock_sock(sk);
267ff45d820SIlya Lesokhin 	sk_proto_close = ctx->sk_proto_close;
268ff45d820SIlya Lesokhin 
26976f7164dSAtul Gupta 	if (ctx->tx_conf == TLS_HW_RECORD && ctx->rx_conf == TLS_HW_RECORD)
27076f7164dSAtul Gupta 		goto skip_tx_cleanup;
27176f7164dSAtul Gupta 
27276f7164dSAtul Gupta 	if (ctx->tx_conf == TLS_BASE && ctx->rx_conf == TLS_BASE) {
27398f0a395SEric Dumazet 		free_ctx = true;
274ff45d820SIlya Lesokhin 		goto skip_tx_cleanup;
275ff45d820SIlya Lesokhin 	}
2763c4d7559SDave Watson 
2773c4d7559SDave Watson 	if (!tls_complete_pending_work(sk, ctx, 0, &timeo))
2783c4d7559SDave Watson 		tls_handle_open_record(sk, 0);
2793c4d7559SDave Watson 
280f66de3eeSBoris Pismenny 	/* We need these for tls_sw_fallback handling of other packets */
281f66de3eeSBoris Pismenny 	if (ctx->tx_conf == TLS_SW) {
282dbe42559SDave Watson 		kfree(ctx->tx.rec_seq);
283dbe42559SDave Watson 		kfree(ctx->tx.iv);
284f66de3eeSBoris Pismenny 		tls_sw_free_resources_tx(sk);
285f66de3eeSBoris Pismenny 	}
286f66de3eeSBoris Pismenny 
287f66de3eeSBoris Pismenny 	if (ctx->rx_conf == TLS_SW) {
288c46234ebSDave Watson 		kfree(ctx->rx.rec_seq);
289c46234ebSDave Watson 		kfree(ctx->rx.iv);
290f66de3eeSBoris Pismenny 		tls_sw_free_resources_rx(sk);
291c46234ebSDave Watson 	}
2923c4d7559SDave Watson 
293e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE
2944799ac81SBoris Pismenny 	if (ctx->rx_conf == TLS_HW)
2954799ac81SBoris Pismenny 		tls_device_offload_cleanup_rx(sk);
2964799ac81SBoris Pismenny 
2974799ac81SBoris Pismenny 	if (ctx->tx_conf != TLS_HW && ctx->rx_conf != TLS_HW) {
298e8f69799SIlya Lesokhin #else
299e8f69799SIlya Lesokhin 	{
300e8f69799SIlya Lesokhin #endif
30186029d10SSabrina Dubroca 		tls_ctx_free(ctx);
302e8f69799SIlya Lesokhin 		ctx = NULL;
303e8f69799SIlya Lesokhin 	}
304e8f69799SIlya Lesokhin 
305ff45d820SIlya Lesokhin skip_tx_cleanup:
3063c4d7559SDave Watson 	release_sock(sk);
3073c4d7559SDave Watson 	sk_proto_close(sk, timeout);
308dd0bed16SAtul Gupta 	/* free ctx for TLS_HW_RECORD, used by tcp_set_state
309dd0bed16SAtul Gupta 	 * for sk->sk_prot->unhash [tls_hw_unhash]
310dd0bed16SAtul Gupta 	 */
31198f0a395SEric Dumazet 	if (free_ctx)
31286029d10SSabrina Dubroca 		tls_ctx_free(ctx);
3133c4d7559SDave Watson }
3143c4d7559SDave Watson 
3153c4d7559SDave Watson static int do_tls_getsockopt_tx(struct sock *sk, char __user *optval,
3163c4d7559SDave Watson 				int __user *optlen)
3173c4d7559SDave Watson {
3183c4d7559SDave Watson 	int rc = 0;
3193c4d7559SDave Watson 	struct tls_context *ctx = tls_get_ctx(sk);
3203c4d7559SDave Watson 	struct tls_crypto_info *crypto_info;
3213c4d7559SDave Watson 	int len;
3223c4d7559SDave Watson 
3233c4d7559SDave Watson 	if (get_user(len, optlen))
3243c4d7559SDave Watson 		return -EFAULT;
3253c4d7559SDave Watson 
3263c4d7559SDave Watson 	if (!optval || (len < sizeof(*crypto_info))) {
3273c4d7559SDave Watson 		rc = -EINVAL;
3283c4d7559SDave Watson 		goto out;
3293c4d7559SDave Watson 	}
3303c4d7559SDave Watson 
3313c4d7559SDave Watson 	if (!ctx) {
3323c4d7559SDave Watson 		rc = -EBUSY;
3333c4d7559SDave Watson 		goto out;
3343c4d7559SDave Watson 	}
3353c4d7559SDave Watson 
3363c4d7559SDave Watson 	/* get user crypto info */
33786029d10SSabrina Dubroca 	crypto_info = &ctx->crypto_send.info;
3383c4d7559SDave Watson 
3393c4d7559SDave Watson 	if (!TLS_CRYPTO_INFO_READY(crypto_info)) {
3403c4d7559SDave Watson 		rc = -EBUSY;
3413c4d7559SDave Watson 		goto out;
3423c4d7559SDave Watson 	}
3433c4d7559SDave Watson 
3445a3b886cSMatthias Rosenfelder 	if (len == sizeof(*crypto_info)) {
345ac55cd61SDan Carpenter 		if (copy_to_user(optval, crypto_info, sizeof(*crypto_info)))
346ac55cd61SDan Carpenter 			rc = -EFAULT;
3473c4d7559SDave Watson 		goto out;
3483c4d7559SDave Watson 	}
3493c4d7559SDave Watson 
3503c4d7559SDave Watson 	switch (crypto_info->cipher_type) {
3513c4d7559SDave Watson 	case TLS_CIPHER_AES_GCM_128: {
3523c4d7559SDave Watson 		struct tls12_crypto_info_aes_gcm_128 *
3533c4d7559SDave Watson 		  crypto_info_aes_gcm_128 =
3543c4d7559SDave Watson 		  container_of(crypto_info,
3553c4d7559SDave Watson 			       struct tls12_crypto_info_aes_gcm_128,
3563c4d7559SDave Watson 			       info);
3573c4d7559SDave Watson 
3583c4d7559SDave Watson 		if (len != sizeof(*crypto_info_aes_gcm_128)) {
3593c4d7559SDave Watson 			rc = -EINVAL;
3603c4d7559SDave Watson 			goto out;
3613c4d7559SDave Watson 		}
3623c4d7559SDave Watson 		lock_sock(sk);
363a1dfa681SBoris Pismenny 		memcpy(crypto_info_aes_gcm_128->iv,
364dbe42559SDave Watson 		       ctx->tx.iv + TLS_CIPHER_AES_GCM_128_SALT_SIZE,
3653c4d7559SDave Watson 		       TLS_CIPHER_AES_GCM_128_IV_SIZE);
366dbe42559SDave Watson 		memcpy(crypto_info_aes_gcm_128->rec_seq, ctx->tx.rec_seq,
367c410c196SBoris Pismenny 		       TLS_CIPHER_AES_GCM_128_REC_SEQ_SIZE);
3683c4d7559SDave Watson 		release_sock(sk);
369ac55cd61SDan Carpenter 		if (copy_to_user(optval,
3703c4d7559SDave Watson 				 crypto_info_aes_gcm_128,
371ac55cd61SDan Carpenter 				 sizeof(*crypto_info_aes_gcm_128)))
372ac55cd61SDan Carpenter 			rc = -EFAULT;
3733c4d7559SDave Watson 		break;
3743c4d7559SDave Watson 	}
3753c4d7559SDave Watson 	default:
3763c4d7559SDave Watson 		rc = -EINVAL;
3773c4d7559SDave Watson 	}
3783c4d7559SDave Watson 
3793c4d7559SDave Watson out:
3803c4d7559SDave Watson 	return rc;
3813c4d7559SDave Watson }
3823c4d7559SDave Watson 
3833c4d7559SDave Watson static int do_tls_getsockopt(struct sock *sk, int optname,
3843c4d7559SDave Watson 			     char __user *optval, int __user *optlen)
3853c4d7559SDave Watson {
3863c4d7559SDave Watson 	int rc = 0;
3873c4d7559SDave Watson 
3883c4d7559SDave Watson 	switch (optname) {
3893c4d7559SDave Watson 	case TLS_TX:
3903c4d7559SDave Watson 		rc = do_tls_getsockopt_tx(sk, optval, optlen);
3913c4d7559SDave Watson 		break;
3923c4d7559SDave Watson 	default:
3933c4d7559SDave Watson 		rc = -ENOPROTOOPT;
3943c4d7559SDave Watson 		break;
3953c4d7559SDave Watson 	}
3963c4d7559SDave Watson 	return rc;
3973c4d7559SDave Watson }
3983c4d7559SDave Watson 
3993c4d7559SDave Watson static int tls_getsockopt(struct sock *sk, int level, int optname,
4003c4d7559SDave Watson 			  char __user *optval, int __user *optlen)
4013c4d7559SDave Watson {
4023c4d7559SDave Watson 	struct tls_context *ctx = tls_get_ctx(sk);
4033c4d7559SDave Watson 
4043c4d7559SDave Watson 	if (level != SOL_TLS)
4053c4d7559SDave Watson 		return ctx->getsockopt(sk, level, optname, optval, optlen);
4063c4d7559SDave Watson 
4073c4d7559SDave Watson 	return do_tls_getsockopt(sk, optname, optval, optlen);
4083c4d7559SDave Watson }
4093c4d7559SDave Watson 
410c46234ebSDave Watson static int do_tls_setsockopt_conf(struct sock *sk, char __user *optval,
411c46234ebSDave Watson 				  unsigned int optlen, int tx)
4123c4d7559SDave Watson {
413196c31b4SIlya Lesokhin 	struct tls_crypto_info *crypto_info;
4143c4d7559SDave Watson 	struct tls_context *ctx = tls_get_ctx(sk);
4153c4d7559SDave Watson 	int rc = 0;
41658371585SDave Watson 	int conf;
4173c4d7559SDave Watson 
4183c4d7559SDave Watson 	if (!optval || (optlen < sizeof(*crypto_info))) {
4193c4d7559SDave Watson 		rc = -EINVAL;
4203c4d7559SDave Watson 		goto out;
4213c4d7559SDave Watson 	}
4223c4d7559SDave Watson 
423c46234ebSDave Watson 	if (tx)
42486029d10SSabrina Dubroca 		crypto_info = &ctx->crypto_send.info;
425c46234ebSDave Watson 	else
42686029d10SSabrina Dubroca 		crypto_info = &ctx->crypto_recv.info;
427c46234ebSDave Watson 
428196c31b4SIlya Lesokhin 	/* Currently we don't support set crypto info more than one time */
429877d17c7SSabrina Dubroca 	if (TLS_CRYPTO_INFO_READY(crypto_info)) {
430877d17c7SSabrina Dubroca 		rc = -EBUSY;
431196c31b4SIlya Lesokhin 		goto out;
432877d17c7SSabrina Dubroca 	}
433196c31b4SIlya Lesokhin 
434196c31b4SIlya Lesokhin 	rc = copy_from_user(crypto_info, optval, sizeof(*crypto_info));
4353c4d7559SDave Watson 	if (rc) {
4363c4d7559SDave Watson 		rc = -EFAULT;
437257082e6SBoris Pismenny 		goto err_crypto_info;
4383c4d7559SDave Watson 	}
4393c4d7559SDave Watson 
4403c4d7559SDave Watson 	/* check version */
441196c31b4SIlya Lesokhin 	if (crypto_info->version != TLS_1_2_VERSION) {
4423c4d7559SDave Watson 		rc = -ENOTSUPP;
443196c31b4SIlya Lesokhin 		goto err_crypto_info;
4443c4d7559SDave Watson 	}
4453c4d7559SDave Watson 
446196c31b4SIlya Lesokhin 	switch (crypto_info->cipher_type) {
4473c4d7559SDave Watson 	case TLS_CIPHER_AES_GCM_128: {
4483c4d7559SDave Watson 		if (optlen != sizeof(struct tls12_crypto_info_aes_gcm_128)) {
4493c4d7559SDave Watson 			rc = -EINVAL;
4506db959c8SSabrina Dubroca 			goto err_crypto_info;
4513c4d7559SDave Watson 		}
452196c31b4SIlya Lesokhin 		rc = copy_from_user(crypto_info + 1, optval + sizeof(*crypto_info),
453196c31b4SIlya Lesokhin 				    optlen - sizeof(*crypto_info));
4543c4d7559SDave Watson 		if (rc) {
4553c4d7559SDave Watson 			rc = -EFAULT;
4563c4d7559SDave Watson 			goto err_crypto_info;
4573c4d7559SDave Watson 		}
4583c4d7559SDave Watson 		break;
4593c4d7559SDave Watson 	}
4603c4d7559SDave Watson 	default:
4613c4d7559SDave Watson 		rc = -EINVAL;
4626db959c8SSabrina Dubroca 		goto err_crypto_info;
4633c4d7559SDave Watson 	}
4643c4d7559SDave Watson 
465c46234ebSDave Watson 	if (tx) {
466e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE
467e8f69799SIlya Lesokhin 		rc = tls_set_device_offload(sk, ctx);
468e8f69799SIlya Lesokhin 		conf = TLS_HW;
469e8f69799SIlya Lesokhin 		if (rc) {
470e8f69799SIlya Lesokhin #else
471e8f69799SIlya Lesokhin 		{
472e8f69799SIlya Lesokhin #endif
473c46234ebSDave Watson 			rc = tls_set_sw_offload(sk, ctx, 1);
474f66de3eeSBoris Pismenny 			conf = TLS_SW;
475e8f69799SIlya Lesokhin 		}
476c46234ebSDave Watson 	} else {
4774799ac81SBoris Pismenny #ifdef CONFIG_TLS_DEVICE
4784799ac81SBoris Pismenny 		rc = tls_set_device_offload_rx(sk, ctx);
4794799ac81SBoris Pismenny 		conf = TLS_HW;
4804799ac81SBoris Pismenny 		if (rc) {
4814799ac81SBoris Pismenny #else
4824799ac81SBoris Pismenny 		{
4834799ac81SBoris Pismenny #endif
484c46234ebSDave Watson 			rc = tls_set_sw_offload(sk, ctx, 0);
485f66de3eeSBoris Pismenny 			conf = TLS_SW;
486c46234ebSDave Watson 		}
4874799ac81SBoris Pismenny 	}
488c46234ebSDave Watson 
4893c4d7559SDave Watson 	if (rc)
4903c4d7559SDave Watson 		goto err_crypto_info;
4913c4d7559SDave Watson 
492f66de3eeSBoris Pismenny 	if (tx)
493f66de3eeSBoris Pismenny 		ctx->tx_conf = conf;
494f66de3eeSBoris Pismenny 	else
495f66de3eeSBoris Pismenny 		ctx->rx_conf = conf;
4966d88207fSIlya Lesokhin 	update_sk_prot(sk, ctx);
497c46234ebSDave Watson 	if (tx) {
498ee181e52SIlya Lesokhin 		ctx->sk_write_space = sk->sk_write_space;
499ee181e52SIlya Lesokhin 		sk->sk_write_space = tls_write_space;
500c46234ebSDave Watson 	} else {
501c46234ebSDave Watson 		sk->sk_socket->ops = &tls_sw_proto_ops;
502c46234ebSDave Watson 	}
5033c4d7559SDave Watson 	goto out;
5043c4d7559SDave Watson 
5053c4d7559SDave Watson err_crypto_info:
506c844eb46SSabrina Dubroca 	memzero_explicit(crypto_info, sizeof(union tls_crypto_context));
5073c4d7559SDave Watson out:
5083c4d7559SDave Watson 	return rc;
5093c4d7559SDave Watson }
5103c4d7559SDave Watson 
5113c4d7559SDave Watson static int do_tls_setsockopt(struct sock *sk, int optname,
5123c4d7559SDave Watson 			     char __user *optval, unsigned int optlen)
5133c4d7559SDave Watson {
5143c4d7559SDave Watson 	int rc = 0;
5153c4d7559SDave Watson 
5163c4d7559SDave Watson 	switch (optname) {
5173c4d7559SDave Watson 	case TLS_TX:
518c46234ebSDave Watson 	case TLS_RX:
5193c4d7559SDave Watson 		lock_sock(sk);
520c46234ebSDave Watson 		rc = do_tls_setsockopt_conf(sk, optval, optlen,
521c46234ebSDave Watson 					    optname == TLS_TX);
5223c4d7559SDave Watson 		release_sock(sk);
5233c4d7559SDave Watson 		break;
5243c4d7559SDave Watson 	default:
5253c4d7559SDave Watson 		rc = -ENOPROTOOPT;
5263c4d7559SDave Watson 		break;
5273c4d7559SDave Watson 	}
5283c4d7559SDave Watson 	return rc;
5293c4d7559SDave Watson }
5303c4d7559SDave Watson 
5313c4d7559SDave Watson static int tls_setsockopt(struct sock *sk, int level, int optname,
5323c4d7559SDave Watson 			  char __user *optval, unsigned int optlen)
5333c4d7559SDave Watson {
5343c4d7559SDave Watson 	struct tls_context *ctx = tls_get_ctx(sk);
5353c4d7559SDave Watson 
5363c4d7559SDave Watson 	if (level != SOL_TLS)
5373c4d7559SDave Watson 		return ctx->setsockopt(sk, level, optname, optval, optlen);
5383c4d7559SDave Watson 
5393c4d7559SDave Watson 	return do_tls_setsockopt(sk, optname, optval, optlen);
5403c4d7559SDave Watson }
5413c4d7559SDave Watson 
542dd0bed16SAtul Gupta static struct tls_context *create_ctx(struct sock *sk)
543dd0bed16SAtul Gupta {
544dd0bed16SAtul Gupta 	struct inet_connection_sock *icsk = inet_csk(sk);
545dd0bed16SAtul Gupta 	struct tls_context *ctx;
546dd0bed16SAtul Gupta 
547c6ec179aSGanesh Goudar 	ctx = kzalloc(sizeof(*ctx), GFP_ATOMIC);
548dd0bed16SAtul Gupta 	if (!ctx)
549dd0bed16SAtul Gupta 		return NULL;
550dd0bed16SAtul Gupta 
551dd0bed16SAtul Gupta 	icsk->icsk_ulp_data = ctx;
5526c0563e4SAtul Gupta 	ctx->setsockopt = sk->sk_prot->setsockopt;
5536c0563e4SAtul Gupta 	ctx->getsockopt = sk->sk_prot->getsockopt;
5546c0563e4SAtul Gupta 	ctx->sk_proto_close = sk->sk_prot->close;
555dd0bed16SAtul Gupta 	return ctx;
556dd0bed16SAtul Gupta }
557dd0bed16SAtul Gupta 
55863a6b3feSAtul Gupta static void tls_build_proto(struct sock *sk)
55963a6b3feSAtul Gupta {
56063a6b3feSAtul Gupta 	int ip_ver = sk->sk_family == AF_INET6 ? TLSV6 : TLSV4;
56163a6b3feSAtul Gupta 
56263a6b3feSAtul Gupta 	/* Build IPv6 TLS whenever the address of tcpv6 _prot changes */
56363a6b3feSAtul Gupta 	if (ip_ver == TLSV6 &&
56463a6b3feSAtul Gupta 	    unlikely(sk->sk_prot != smp_load_acquire(&saved_tcpv6_prot))) {
56563a6b3feSAtul Gupta 		mutex_lock(&tcpv6_prot_mutex);
56663a6b3feSAtul Gupta 		if (likely(sk->sk_prot != saved_tcpv6_prot)) {
56763a6b3feSAtul Gupta 			build_protos(tls_prots[TLSV6], sk->sk_prot);
56863a6b3feSAtul Gupta 			smp_store_release(&saved_tcpv6_prot, sk->sk_prot);
56963a6b3feSAtul Gupta 		}
57063a6b3feSAtul Gupta 		mutex_unlock(&tcpv6_prot_mutex);
57163a6b3feSAtul Gupta 	}
57263a6b3feSAtul Gupta 
57363a6b3feSAtul Gupta 	if (ip_ver == TLSV4 &&
57463a6b3feSAtul Gupta 	    unlikely(sk->sk_prot != smp_load_acquire(&saved_tcpv4_prot))) {
57563a6b3feSAtul Gupta 		mutex_lock(&tcpv4_prot_mutex);
57663a6b3feSAtul Gupta 		if (likely(sk->sk_prot != saved_tcpv4_prot)) {
57763a6b3feSAtul Gupta 			build_protos(tls_prots[TLSV4], sk->sk_prot);
57863a6b3feSAtul Gupta 			smp_store_release(&saved_tcpv4_prot, sk->sk_prot);
57963a6b3feSAtul Gupta 		}
58063a6b3feSAtul Gupta 		mutex_unlock(&tcpv4_prot_mutex);
58163a6b3feSAtul Gupta 	}
58263a6b3feSAtul Gupta }
58363a6b3feSAtul Gupta 
58476f7164dSAtul Gupta static void tls_hw_sk_destruct(struct sock *sk)
58576f7164dSAtul Gupta {
58676f7164dSAtul Gupta 	struct tls_context *ctx = tls_get_ctx(sk);
58776f7164dSAtul Gupta 	struct inet_connection_sock *icsk = inet_csk(sk);
58876f7164dSAtul Gupta 
58976f7164dSAtul Gupta 	ctx->sk_destruct(sk);
59076f7164dSAtul Gupta 	/* Free ctx */
59176f7164dSAtul Gupta 	kfree(ctx);
59276f7164dSAtul Gupta 	icsk->icsk_ulp_data = NULL;
59376f7164dSAtul Gupta }
59476f7164dSAtul Gupta 
595dd0bed16SAtul Gupta static int tls_hw_prot(struct sock *sk)
596dd0bed16SAtul Gupta {
597dd0bed16SAtul Gupta 	struct tls_context *ctx;
598dd0bed16SAtul Gupta 	struct tls_device *dev;
599dd0bed16SAtul Gupta 	int rc = 0;
600dd0bed16SAtul Gupta 
601df9d4a17SAtul Gupta 	spin_lock_bh(&device_spinlock);
602dd0bed16SAtul Gupta 	list_for_each_entry(dev, &device_list, dev_list) {
603dd0bed16SAtul Gupta 		if (dev->feature && dev->feature(dev)) {
604dd0bed16SAtul Gupta 			ctx = create_ctx(sk);
605dd0bed16SAtul Gupta 			if (!ctx)
606dd0bed16SAtul Gupta 				goto out;
607dd0bed16SAtul Gupta 
60863a6b3feSAtul Gupta 			spin_unlock_bh(&device_spinlock);
60963a6b3feSAtul Gupta 			tls_build_proto(sk);
610dd0bed16SAtul Gupta 			ctx->hash = sk->sk_prot->hash;
611dd0bed16SAtul Gupta 			ctx->unhash = sk->sk_prot->unhash;
612dd0bed16SAtul Gupta 			ctx->sk_proto_close = sk->sk_prot->close;
61376f7164dSAtul Gupta 			ctx->sk_destruct = sk->sk_destruct;
61476f7164dSAtul Gupta 			sk->sk_destruct = tls_hw_sk_destruct;
615f66de3eeSBoris Pismenny 			ctx->rx_conf = TLS_HW_RECORD;
616f66de3eeSBoris Pismenny 			ctx->tx_conf = TLS_HW_RECORD;
617dd0bed16SAtul Gupta 			update_sk_prot(sk, ctx);
61863a6b3feSAtul Gupta 			spin_lock_bh(&device_spinlock);
619dd0bed16SAtul Gupta 			rc = 1;
620dd0bed16SAtul Gupta 			break;
621dd0bed16SAtul Gupta 		}
622dd0bed16SAtul Gupta 	}
623dd0bed16SAtul Gupta out:
624df9d4a17SAtul Gupta 	spin_unlock_bh(&device_spinlock);
625dd0bed16SAtul Gupta 	return rc;
626dd0bed16SAtul Gupta }
627dd0bed16SAtul Gupta 
628dd0bed16SAtul Gupta static void tls_hw_unhash(struct sock *sk)
629dd0bed16SAtul Gupta {
630dd0bed16SAtul Gupta 	struct tls_context *ctx = tls_get_ctx(sk);
631dd0bed16SAtul Gupta 	struct tls_device *dev;
632dd0bed16SAtul Gupta 
633df9d4a17SAtul Gupta 	spin_lock_bh(&device_spinlock);
634dd0bed16SAtul Gupta 	list_for_each_entry(dev, &device_list, dev_list) {
635df9d4a17SAtul Gupta 		if (dev->unhash) {
636df9d4a17SAtul Gupta 			kref_get(&dev->kref);
637df9d4a17SAtul Gupta 			spin_unlock_bh(&device_spinlock);
638dd0bed16SAtul Gupta 			dev->unhash(dev, sk);
639df9d4a17SAtul Gupta 			kref_put(&dev->kref, dev->release);
640df9d4a17SAtul Gupta 			spin_lock_bh(&device_spinlock);
641dd0bed16SAtul Gupta 		}
642df9d4a17SAtul Gupta 	}
643df9d4a17SAtul Gupta 	spin_unlock_bh(&device_spinlock);
644dd0bed16SAtul Gupta 	ctx->unhash(sk);
645dd0bed16SAtul Gupta }
646dd0bed16SAtul Gupta 
647dd0bed16SAtul Gupta static int tls_hw_hash(struct sock *sk)
648dd0bed16SAtul Gupta {
649dd0bed16SAtul Gupta 	struct tls_context *ctx = tls_get_ctx(sk);
650dd0bed16SAtul Gupta 	struct tls_device *dev;
651dd0bed16SAtul Gupta 	int err;
652dd0bed16SAtul Gupta 
653dd0bed16SAtul Gupta 	err = ctx->hash(sk);
654df9d4a17SAtul Gupta 	spin_lock_bh(&device_spinlock);
655dd0bed16SAtul Gupta 	list_for_each_entry(dev, &device_list, dev_list) {
656df9d4a17SAtul Gupta 		if (dev->hash) {
657df9d4a17SAtul Gupta 			kref_get(&dev->kref);
658df9d4a17SAtul Gupta 			spin_unlock_bh(&device_spinlock);
659dd0bed16SAtul Gupta 			err |= dev->hash(dev, sk);
660df9d4a17SAtul Gupta 			kref_put(&dev->kref, dev->release);
661df9d4a17SAtul Gupta 			spin_lock_bh(&device_spinlock);
662dd0bed16SAtul Gupta 		}
663df9d4a17SAtul Gupta 	}
664df9d4a17SAtul Gupta 	spin_unlock_bh(&device_spinlock);
665dd0bed16SAtul Gupta 
666dd0bed16SAtul Gupta 	if (err)
667dd0bed16SAtul Gupta 		tls_hw_unhash(sk);
668dd0bed16SAtul Gupta 	return err;
669dd0bed16SAtul Gupta }
670dd0bed16SAtul Gupta 
671f66de3eeSBoris Pismenny static void build_protos(struct proto prot[TLS_NUM_CONFIG][TLS_NUM_CONFIG],
672f66de3eeSBoris Pismenny 			 struct proto *base)
673c113187dSBoris Pismenny {
674f66de3eeSBoris Pismenny 	prot[TLS_BASE][TLS_BASE] = *base;
675f66de3eeSBoris Pismenny 	prot[TLS_BASE][TLS_BASE].setsockopt	= tls_setsockopt;
676f66de3eeSBoris Pismenny 	prot[TLS_BASE][TLS_BASE].getsockopt	= tls_getsockopt;
677f66de3eeSBoris Pismenny 	prot[TLS_BASE][TLS_BASE].close		= tls_sk_proto_close;
678c113187dSBoris Pismenny 
679f66de3eeSBoris Pismenny 	prot[TLS_SW][TLS_BASE] = prot[TLS_BASE][TLS_BASE];
680f66de3eeSBoris Pismenny 	prot[TLS_SW][TLS_BASE].sendmsg		= tls_sw_sendmsg;
681f66de3eeSBoris Pismenny 	prot[TLS_SW][TLS_BASE].sendpage		= tls_sw_sendpage;
682c46234ebSDave Watson 
683f66de3eeSBoris Pismenny 	prot[TLS_BASE][TLS_SW] = prot[TLS_BASE][TLS_BASE];
684f66de3eeSBoris Pismenny 	prot[TLS_BASE][TLS_SW].recvmsg		  = tls_sw_recvmsg;
685924ad65eSJohn Fastabend 	prot[TLS_BASE][TLS_SW].stream_memory_read = tls_sw_stream_read;
686f66de3eeSBoris Pismenny 	prot[TLS_BASE][TLS_SW].close		  = tls_sk_proto_close;
687c46234ebSDave Watson 
688f66de3eeSBoris Pismenny 	prot[TLS_SW][TLS_SW] = prot[TLS_SW][TLS_BASE];
689f66de3eeSBoris Pismenny 	prot[TLS_SW][TLS_SW].recvmsg		= tls_sw_recvmsg;
690924ad65eSJohn Fastabend 	prot[TLS_SW][TLS_SW].stream_memory_read	= tls_sw_stream_read;
691f66de3eeSBoris Pismenny 	prot[TLS_SW][TLS_SW].close		= tls_sk_proto_close;
692dd0bed16SAtul Gupta 
693e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE
694e8f69799SIlya Lesokhin 	prot[TLS_HW][TLS_BASE] = prot[TLS_BASE][TLS_BASE];
695e8f69799SIlya Lesokhin 	prot[TLS_HW][TLS_BASE].sendmsg		= tls_device_sendmsg;
696e8f69799SIlya Lesokhin 	prot[TLS_HW][TLS_BASE].sendpage		= tls_device_sendpage;
697e8f69799SIlya Lesokhin 
698e8f69799SIlya Lesokhin 	prot[TLS_HW][TLS_SW] = prot[TLS_BASE][TLS_SW];
699e8f69799SIlya Lesokhin 	prot[TLS_HW][TLS_SW].sendmsg		= tls_device_sendmsg;
700e8f69799SIlya Lesokhin 	prot[TLS_HW][TLS_SW].sendpage		= tls_device_sendpage;
7014799ac81SBoris Pismenny 
7024799ac81SBoris Pismenny 	prot[TLS_BASE][TLS_HW] = prot[TLS_BASE][TLS_SW];
7034799ac81SBoris Pismenny 
7044799ac81SBoris Pismenny 	prot[TLS_SW][TLS_HW] = prot[TLS_SW][TLS_SW];
7054799ac81SBoris Pismenny 
7064799ac81SBoris Pismenny 	prot[TLS_HW][TLS_HW] = prot[TLS_HW][TLS_SW];
707e8f69799SIlya Lesokhin #endif
708e8f69799SIlya Lesokhin 
709f66de3eeSBoris Pismenny 	prot[TLS_HW_RECORD][TLS_HW_RECORD] = *base;
710f66de3eeSBoris Pismenny 	prot[TLS_HW_RECORD][TLS_HW_RECORD].hash		= tls_hw_hash;
711f66de3eeSBoris Pismenny 	prot[TLS_HW_RECORD][TLS_HW_RECORD].unhash	= tls_hw_unhash;
712f66de3eeSBoris Pismenny 	prot[TLS_HW_RECORD][TLS_HW_RECORD].close	= tls_sk_proto_close;
713c113187dSBoris Pismenny }
714c113187dSBoris Pismenny 
7153c4d7559SDave Watson static int tls_init(struct sock *sk)
7163c4d7559SDave Watson {
7173c4d7559SDave Watson 	struct tls_context *ctx;
7183c4d7559SDave Watson 	int rc = 0;
7193c4d7559SDave Watson 
720dd0bed16SAtul Gupta 	if (tls_hw_prot(sk))
721dd0bed16SAtul Gupta 		goto out;
722dd0bed16SAtul Gupta 
723d91c3e17SIlya Lesokhin 	/* The TLS ulp is currently supported only for TCP sockets
724d91c3e17SIlya Lesokhin 	 * in ESTABLISHED state.
725d91c3e17SIlya Lesokhin 	 * Supporting sockets in LISTEN state will require us
726d91c3e17SIlya Lesokhin 	 * to modify the accept implementation to clone rather then
727d91c3e17SIlya Lesokhin 	 * share the ulp context.
728d91c3e17SIlya Lesokhin 	 */
729d91c3e17SIlya Lesokhin 	if (sk->sk_state != TCP_ESTABLISHED)
730d91c3e17SIlya Lesokhin 		return -ENOTSUPP;
731d91c3e17SIlya Lesokhin 
7323c4d7559SDave Watson 	/* allocate tls context */
733dd0bed16SAtul Gupta 	ctx = create_ctx(sk);
7343c4d7559SDave Watson 	if (!ctx) {
7353c4d7559SDave Watson 		rc = -ENOMEM;
7363c4d7559SDave Watson 		goto out;
7373c4d7559SDave Watson 	}
7386d88207fSIlya Lesokhin 
73963a6b3feSAtul Gupta 	tls_build_proto(sk);
740f66de3eeSBoris Pismenny 	ctx->tx_conf = TLS_BASE;
741f66de3eeSBoris Pismenny 	ctx->rx_conf = TLS_BASE;
7426d88207fSIlya Lesokhin 	update_sk_prot(sk, ctx);
7433c4d7559SDave Watson out:
7443c4d7559SDave Watson 	return rc;
7453c4d7559SDave Watson }
7463c4d7559SDave Watson 
747dd0bed16SAtul Gupta void tls_register_device(struct tls_device *device)
748dd0bed16SAtul Gupta {
749df9d4a17SAtul Gupta 	spin_lock_bh(&device_spinlock);
750dd0bed16SAtul Gupta 	list_add_tail(&device->dev_list, &device_list);
751df9d4a17SAtul Gupta 	spin_unlock_bh(&device_spinlock);
752dd0bed16SAtul Gupta }
753dd0bed16SAtul Gupta EXPORT_SYMBOL(tls_register_device);
754dd0bed16SAtul Gupta 
755dd0bed16SAtul Gupta void tls_unregister_device(struct tls_device *device)
756dd0bed16SAtul Gupta {
757df9d4a17SAtul Gupta 	spin_lock_bh(&device_spinlock);
758dd0bed16SAtul Gupta 	list_del(&device->dev_list);
759df9d4a17SAtul Gupta 	spin_unlock_bh(&device_spinlock);
760dd0bed16SAtul Gupta }
761dd0bed16SAtul Gupta EXPORT_SYMBOL(tls_unregister_device);
762dd0bed16SAtul Gupta 
7633c4d7559SDave Watson static struct tcp_ulp_ops tcp_tls_ulp_ops __read_mostly = {
7643c4d7559SDave Watson 	.name			= "tls",
7653c4d7559SDave Watson 	.owner			= THIS_MODULE,
7663c4d7559SDave Watson 	.init			= tls_init,
7673c4d7559SDave Watson };
7683c4d7559SDave Watson 
7693c4d7559SDave Watson static int __init tls_register(void)
7703c4d7559SDave Watson {
771c46234ebSDave Watson 	tls_sw_proto_ops = inet_stream_ops;
772c46234ebSDave Watson 	tls_sw_proto_ops.splice_read = tls_sw_splice_read;
773c46234ebSDave Watson 
774e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE
775e8f69799SIlya Lesokhin 	tls_device_init();
776e8f69799SIlya Lesokhin #endif
7773c4d7559SDave Watson 	tcp_register_ulp(&tcp_tls_ulp_ops);
7783c4d7559SDave Watson 
7793c4d7559SDave Watson 	return 0;
7803c4d7559SDave Watson }
7813c4d7559SDave Watson 
7823c4d7559SDave Watson static void __exit tls_unregister(void)
7833c4d7559SDave Watson {
7843c4d7559SDave Watson 	tcp_unregister_ulp(&tcp_tls_ulp_ops);
785e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE
786e8f69799SIlya Lesokhin 	tls_device_cleanup();
787e8f69799SIlya Lesokhin #endif
7883c4d7559SDave Watson }
7893c4d7559SDave Watson 
7903c4d7559SDave Watson module_init(tls_register);
7913c4d7559SDave Watson module_exit(tls_unregister);
792