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"); 483c4d7559SDave Watson 496d88207fSIlya Lesokhin enum { 50c113187dSBoris Pismenny TLSV4, 51c113187dSBoris Pismenny TLSV6, 52c113187dSBoris Pismenny TLS_NUM_PROTS, 53c113187dSBoris Pismenny }; 54c113187dSBoris Pismenny 55c113187dSBoris Pismenny enum { 5658371585SDave Watson TLS_BASE, 576d88207fSIlya Lesokhin TLS_SW_TX, 58c46234ebSDave Watson TLS_SW_RX, 59c46234ebSDave Watson TLS_SW_RXTX, 60dd0bed16SAtul Gupta TLS_HW_RECORD, 616d88207fSIlya Lesokhin TLS_NUM_CONFIG, 626d88207fSIlya Lesokhin }; 636d88207fSIlya Lesokhin 64c113187dSBoris Pismenny static struct proto *saved_tcpv6_prot; 65c113187dSBoris Pismenny static DEFINE_MUTEX(tcpv6_prot_mutex); 66dd0bed16SAtul Gupta static LIST_HEAD(device_list); 67dd0bed16SAtul Gupta static DEFINE_MUTEX(device_mutex); 68c113187dSBoris Pismenny static struct proto tls_prots[TLS_NUM_PROTS][TLS_NUM_CONFIG]; 69c46234ebSDave Watson static struct proto_ops tls_sw_proto_ops; 706d88207fSIlya Lesokhin 716d88207fSIlya Lesokhin static inline void update_sk_prot(struct sock *sk, struct tls_context *ctx) 726d88207fSIlya Lesokhin { 73c113187dSBoris Pismenny int ip_ver = sk->sk_family == AF_INET6 ? TLSV6 : TLSV4; 74c113187dSBoris Pismenny 7558371585SDave Watson sk->sk_prot = &tls_prots[ip_ver][ctx->conf]; 766d88207fSIlya Lesokhin } 773c4d7559SDave Watson 783c4d7559SDave Watson int wait_on_pending_writer(struct sock *sk, long *timeo) 793c4d7559SDave Watson { 803c4d7559SDave Watson int rc = 0; 813c4d7559SDave Watson DEFINE_WAIT_FUNC(wait, woken_wake_function); 823c4d7559SDave Watson 833c4d7559SDave Watson add_wait_queue(sk_sleep(sk), &wait); 843c4d7559SDave Watson while (1) { 853c4d7559SDave Watson if (!*timeo) { 863c4d7559SDave Watson rc = -EAGAIN; 873c4d7559SDave Watson break; 883c4d7559SDave Watson } 893c4d7559SDave Watson 903c4d7559SDave Watson if (signal_pending(current)) { 913c4d7559SDave Watson rc = sock_intr_errno(*timeo); 923c4d7559SDave Watson break; 933c4d7559SDave Watson } 943c4d7559SDave Watson 953c4d7559SDave Watson if (sk_wait_event(sk, timeo, !sk->sk_write_pending, &wait)) 963c4d7559SDave Watson break; 973c4d7559SDave Watson } 983c4d7559SDave Watson remove_wait_queue(sk_sleep(sk), &wait); 993c4d7559SDave Watson return rc; 1003c4d7559SDave Watson } 1013c4d7559SDave Watson 1023c4d7559SDave Watson int tls_push_sg(struct sock *sk, 1033c4d7559SDave Watson struct tls_context *ctx, 1043c4d7559SDave Watson struct scatterlist *sg, 1053c4d7559SDave Watson u16 first_offset, 1063c4d7559SDave Watson int flags) 1073c4d7559SDave Watson { 1083c4d7559SDave Watson int sendpage_flags = flags | MSG_SENDPAGE_NOTLAST; 1093c4d7559SDave Watson int ret = 0; 1103c4d7559SDave Watson struct page *p; 1113c4d7559SDave Watson size_t size; 1123c4d7559SDave Watson int offset = first_offset; 1133c4d7559SDave Watson 1143c4d7559SDave Watson size = sg->length - offset; 1153c4d7559SDave Watson offset += sg->offset; 1163c4d7559SDave Watson 117c212d2c7SDave Watson ctx->in_tcp_sendpages = true; 1183c4d7559SDave Watson while (1) { 1193c4d7559SDave Watson if (sg_is_last(sg)) 1203c4d7559SDave Watson sendpage_flags = flags; 1213c4d7559SDave Watson 1223c4d7559SDave Watson /* is sending application-limited? */ 1233c4d7559SDave Watson tcp_rate_check_app_limited(sk); 1243c4d7559SDave Watson p = sg_page(sg); 1253c4d7559SDave Watson retry: 1263c4d7559SDave Watson ret = do_tcp_sendpages(sk, p, offset, size, sendpage_flags); 1273c4d7559SDave Watson 1283c4d7559SDave Watson if (ret != size) { 1293c4d7559SDave Watson if (ret > 0) { 1303c4d7559SDave Watson offset += ret; 1313c4d7559SDave Watson size -= ret; 1323c4d7559SDave Watson goto retry; 1333c4d7559SDave Watson } 1343c4d7559SDave Watson 1353c4d7559SDave Watson offset -= sg->offset; 1363c4d7559SDave Watson ctx->partially_sent_offset = offset; 1373c4d7559SDave Watson ctx->partially_sent_record = (void *)sg; 1383c4d7559SDave Watson return ret; 1393c4d7559SDave Watson } 1403c4d7559SDave Watson 1413c4d7559SDave Watson put_page(p); 1423c4d7559SDave Watson sk_mem_uncharge(sk, sg->length); 1433c4d7559SDave Watson sg = sg_next(sg); 1443c4d7559SDave Watson if (!sg) 1453c4d7559SDave Watson break; 1463c4d7559SDave Watson 1473c4d7559SDave Watson offset = sg->offset; 1483c4d7559SDave Watson size = sg->length; 1493c4d7559SDave Watson } 1503c4d7559SDave Watson 1513c4d7559SDave Watson clear_bit(TLS_PENDING_CLOSED_RECORD, &ctx->flags); 152c212d2c7SDave Watson ctx->in_tcp_sendpages = false; 153c212d2c7SDave Watson ctx->sk_write_space(sk); 1543c4d7559SDave Watson 1553c4d7559SDave Watson return 0; 1563c4d7559SDave Watson } 1573c4d7559SDave Watson 1583c4d7559SDave Watson static int tls_handle_open_record(struct sock *sk, int flags) 1593c4d7559SDave Watson { 1603c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 1613c4d7559SDave Watson 1623c4d7559SDave Watson if (tls_is_pending_open_record(ctx)) 1633c4d7559SDave Watson return ctx->push_pending_record(sk, flags); 1643c4d7559SDave Watson 1653c4d7559SDave Watson return 0; 1663c4d7559SDave Watson } 1673c4d7559SDave Watson 1683c4d7559SDave Watson int tls_proccess_cmsg(struct sock *sk, struct msghdr *msg, 1693c4d7559SDave Watson unsigned char *record_type) 1703c4d7559SDave Watson { 1713c4d7559SDave Watson struct cmsghdr *cmsg; 1723c4d7559SDave Watson int rc = -EINVAL; 1733c4d7559SDave Watson 1743c4d7559SDave Watson for_each_cmsghdr(cmsg, msg) { 1753c4d7559SDave Watson if (!CMSG_OK(msg, cmsg)) 1763c4d7559SDave Watson return -EINVAL; 1773c4d7559SDave Watson if (cmsg->cmsg_level != SOL_TLS) 1783c4d7559SDave Watson continue; 1793c4d7559SDave Watson 1803c4d7559SDave Watson switch (cmsg->cmsg_type) { 1813c4d7559SDave Watson case TLS_SET_RECORD_TYPE: 1823c4d7559SDave Watson if (cmsg->cmsg_len < CMSG_LEN(sizeof(*record_type))) 1833c4d7559SDave Watson return -EINVAL; 1843c4d7559SDave Watson 1853c4d7559SDave Watson if (msg->msg_flags & MSG_MORE) 1863c4d7559SDave Watson return -EINVAL; 1873c4d7559SDave Watson 1883c4d7559SDave Watson rc = tls_handle_open_record(sk, msg->msg_flags); 1893c4d7559SDave Watson if (rc) 1903c4d7559SDave Watson return rc; 1913c4d7559SDave Watson 1923c4d7559SDave Watson *record_type = *(unsigned char *)CMSG_DATA(cmsg); 1933c4d7559SDave Watson rc = 0; 1943c4d7559SDave Watson break; 1953c4d7559SDave Watson default: 1963c4d7559SDave Watson return -EINVAL; 1973c4d7559SDave Watson } 1983c4d7559SDave Watson } 1993c4d7559SDave Watson 2003c4d7559SDave Watson return rc; 2013c4d7559SDave Watson } 2023c4d7559SDave Watson 2033c4d7559SDave Watson int tls_push_pending_closed_record(struct sock *sk, struct tls_context *ctx, 2043c4d7559SDave Watson int flags, long *timeo) 2053c4d7559SDave Watson { 2063c4d7559SDave Watson struct scatterlist *sg; 2073c4d7559SDave Watson u16 offset; 2083c4d7559SDave Watson 2093c4d7559SDave Watson if (!tls_is_partially_sent_record(ctx)) 2103c4d7559SDave Watson return ctx->push_pending_record(sk, flags); 2113c4d7559SDave Watson 2123c4d7559SDave Watson sg = ctx->partially_sent_record; 2133c4d7559SDave Watson offset = ctx->partially_sent_offset; 2143c4d7559SDave Watson 2153c4d7559SDave Watson ctx->partially_sent_record = NULL; 2163c4d7559SDave Watson return tls_push_sg(sk, ctx, sg, offset, flags); 2173c4d7559SDave Watson } 2183c4d7559SDave Watson 2193c4d7559SDave Watson static void tls_write_space(struct sock *sk) 2203c4d7559SDave Watson { 2213c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 2223c4d7559SDave Watson 223c212d2c7SDave Watson /* We are already sending pages, ignore notification */ 224c212d2c7SDave Watson if (ctx->in_tcp_sendpages) 225c212d2c7SDave Watson return; 226c212d2c7SDave Watson 2273c4d7559SDave Watson if (!sk->sk_write_pending && tls_is_pending_closed_record(ctx)) { 2283c4d7559SDave Watson gfp_t sk_allocation = sk->sk_allocation; 2293c4d7559SDave Watson int rc; 2303c4d7559SDave Watson long timeo = 0; 2313c4d7559SDave Watson 2323c4d7559SDave Watson sk->sk_allocation = GFP_ATOMIC; 2333c4d7559SDave Watson rc = tls_push_pending_closed_record(sk, ctx, 2343c4d7559SDave Watson MSG_DONTWAIT | 2353c4d7559SDave Watson MSG_NOSIGNAL, 2363c4d7559SDave Watson &timeo); 2373c4d7559SDave Watson sk->sk_allocation = sk_allocation; 2383c4d7559SDave Watson 2393c4d7559SDave Watson if (rc < 0) 2403c4d7559SDave Watson return; 2413c4d7559SDave Watson } 2423c4d7559SDave Watson 2433c4d7559SDave Watson ctx->sk_write_space(sk); 2443c4d7559SDave Watson } 2453c4d7559SDave Watson 2463c4d7559SDave Watson static void tls_sk_proto_close(struct sock *sk, long timeout) 2473c4d7559SDave Watson { 2483c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 2493c4d7559SDave Watson long timeo = sock_sndtimeo(sk, 0); 2503c4d7559SDave Watson void (*sk_proto_close)(struct sock *sk, long timeout); 2513c4d7559SDave Watson 2523c4d7559SDave Watson lock_sock(sk); 253ff45d820SIlya Lesokhin sk_proto_close = ctx->sk_proto_close; 254ff45d820SIlya Lesokhin 255dd0bed16SAtul Gupta if (ctx->conf == TLS_HW_RECORD) 256dd0bed16SAtul Gupta goto skip_tx_cleanup; 257dd0bed16SAtul Gupta 25858371585SDave Watson if (ctx->conf == TLS_BASE) { 259ff45d820SIlya Lesokhin kfree(ctx); 260dd0bed16SAtul Gupta ctx = NULL; 261ff45d820SIlya Lesokhin goto skip_tx_cleanup; 262ff45d820SIlya Lesokhin } 2633c4d7559SDave Watson 2643c4d7559SDave Watson if (!tls_complete_pending_work(sk, ctx, 0, &timeo)) 2653c4d7559SDave Watson tls_handle_open_record(sk, 0); 2663c4d7559SDave Watson 2673c4d7559SDave Watson if (ctx->partially_sent_record) { 2683c4d7559SDave Watson struct scatterlist *sg = ctx->partially_sent_record; 2693c4d7559SDave Watson 2703c4d7559SDave Watson while (1) { 2713c4d7559SDave Watson put_page(sg_page(sg)); 2723c4d7559SDave Watson sk_mem_uncharge(sk, sg->length); 2733c4d7559SDave Watson 2743c4d7559SDave Watson if (sg_is_last(sg)) 2753c4d7559SDave Watson break; 2763c4d7559SDave Watson sg++; 2773c4d7559SDave Watson } 2783c4d7559SDave Watson } 279ff45d820SIlya Lesokhin 280dbe42559SDave Watson kfree(ctx->tx.rec_seq); 281dbe42559SDave Watson kfree(ctx->tx.iv); 282c46234ebSDave Watson kfree(ctx->rx.rec_seq); 283c46234ebSDave Watson kfree(ctx->rx.iv); 2843c4d7559SDave Watson 285c46234ebSDave Watson if (ctx->conf == TLS_SW_TX || 286c46234ebSDave Watson ctx->conf == TLS_SW_RX || 287c46234ebSDave Watson ctx->conf == TLS_SW_RXTX) { 288c46234ebSDave Watson tls_sw_free_resources(sk); 289c46234ebSDave Watson } 2903c4d7559SDave Watson 291ff45d820SIlya Lesokhin skip_tx_cleanup: 2923c4d7559SDave Watson release_sock(sk); 2933c4d7559SDave Watson sk_proto_close(sk, timeout); 294dd0bed16SAtul Gupta /* free ctx for TLS_HW_RECORD, used by tcp_set_state 295dd0bed16SAtul Gupta * for sk->sk_prot->unhash [tls_hw_unhash] 296dd0bed16SAtul Gupta */ 297dd0bed16SAtul Gupta if (ctx && ctx->conf == TLS_HW_RECORD) 298dd0bed16SAtul Gupta kfree(ctx); 2993c4d7559SDave Watson } 3003c4d7559SDave Watson 3013c4d7559SDave Watson static int do_tls_getsockopt_tx(struct sock *sk, char __user *optval, 3023c4d7559SDave Watson int __user *optlen) 3033c4d7559SDave Watson { 3043c4d7559SDave Watson int rc = 0; 3053c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 3063c4d7559SDave Watson struct tls_crypto_info *crypto_info; 3073c4d7559SDave Watson int len; 3083c4d7559SDave Watson 3093c4d7559SDave Watson if (get_user(len, optlen)) 3103c4d7559SDave Watson return -EFAULT; 3113c4d7559SDave Watson 3123c4d7559SDave Watson if (!optval || (len < sizeof(*crypto_info))) { 3133c4d7559SDave Watson rc = -EINVAL; 3143c4d7559SDave Watson goto out; 3153c4d7559SDave Watson } 3163c4d7559SDave Watson 3173c4d7559SDave Watson if (!ctx) { 3183c4d7559SDave Watson rc = -EBUSY; 3193c4d7559SDave Watson goto out; 3203c4d7559SDave Watson } 3213c4d7559SDave Watson 3223c4d7559SDave Watson /* get user crypto info */ 3233c4d7559SDave Watson crypto_info = &ctx->crypto_send; 3243c4d7559SDave Watson 3253c4d7559SDave Watson if (!TLS_CRYPTO_INFO_READY(crypto_info)) { 3263c4d7559SDave Watson rc = -EBUSY; 3273c4d7559SDave Watson goto out; 3283c4d7559SDave Watson } 3293c4d7559SDave Watson 3305a3b886cSMatthias Rosenfelder if (len == sizeof(*crypto_info)) { 331ac55cd61SDan Carpenter if (copy_to_user(optval, crypto_info, sizeof(*crypto_info))) 332ac55cd61SDan Carpenter rc = -EFAULT; 3333c4d7559SDave Watson goto out; 3343c4d7559SDave Watson } 3353c4d7559SDave Watson 3363c4d7559SDave Watson switch (crypto_info->cipher_type) { 3373c4d7559SDave Watson case TLS_CIPHER_AES_GCM_128: { 3383c4d7559SDave Watson struct tls12_crypto_info_aes_gcm_128 * 3393c4d7559SDave Watson crypto_info_aes_gcm_128 = 3403c4d7559SDave Watson container_of(crypto_info, 3413c4d7559SDave Watson struct tls12_crypto_info_aes_gcm_128, 3423c4d7559SDave Watson info); 3433c4d7559SDave Watson 3443c4d7559SDave Watson if (len != sizeof(*crypto_info_aes_gcm_128)) { 3453c4d7559SDave Watson rc = -EINVAL; 3463c4d7559SDave Watson goto out; 3473c4d7559SDave Watson } 3483c4d7559SDave Watson lock_sock(sk); 349a1dfa681SBoris Pismenny memcpy(crypto_info_aes_gcm_128->iv, 350dbe42559SDave Watson ctx->tx.iv + TLS_CIPHER_AES_GCM_128_SALT_SIZE, 3513c4d7559SDave Watson TLS_CIPHER_AES_GCM_128_IV_SIZE); 352dbe42559SDave Watson memcpy(crypto_info_aes_gcm_128->rec_seq, ctx->tx.rec_seq, 353c410c196SBoris Pismenny TLS_CIPHER_AES_GCM_128_REC_SEQ_SIZE); 3543c4d7559SDave Watson release_sock(sk); 355ac55cd61SDan Carpenter if (copy_to_user(optval, 3563c4d7559SDave Watson crypto_info_aes_gcm_128, 357ac55cd61SDan Carpenter sizeof(*crypto_info_aes_gcm_128))) 358ac55cd61SDan Carpenter rc = -EFAULT; 3593c4d7559SDave Watson break; 3603c4d7559SDave Watson } 3613c4d7559SDave Watson default: 3623c4d7559SDave Watson rc = -EINVAL; 3633c4d7559SDave Watson } 3643c4d7559SDave Watson 3653c4d7559SDave Watson out: 3663c4d7559SDave Watson return rc; 3673c4d7559SDave Watson } 3683c4d7559SDave Watson 3693c4d7559SDave Watson static int do_tls_getsockopt(struct sock *sk, int optname, 3703c4d7559SDave Watson char __user *optval, int __user *optlen) 3713c4d7559SDave Watson { 3723c4d7559SDave Watson int rc = 0; 3733c4d7559SDave Watson 3743c4d7559SDave Watson switch (optname) { 3753c4d7559SDave Watson case TLS_TX: 3763c4d7559SDave Watson rc = do_tls_getsockopt_tx(sk, optval, optlen); 3773c4d7559SDave Watson break; 3783c4d7559SDave Watson default: 3793c4d7559SDave Watson rc = -ENOPROTOOPT; 3803c4d7559SDave Watson break; 3813c4d7559SDave Watson } 3823c4d7559SDave Watson return rc; 3833c4d7559SDave Watson } 3843c4d7559SDave Watson 3853c4d7559SDave Watson static int tls_getsockopt(struct sock *sk, int level, int optname, 3863c4d7559SDave Watson char __user *optval, int __user *optlen) 3873c4d7559SDave Watson { 3883c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 3893c4d7559SDave Watson 3903c4d7559SDave Watson if (level != SOL_TLS) 3913c4d7559SDave Watson return ctx->getsockopt(sk, level, optname, optval, optlen); 3923c4d7559SDave Watson 3933c4d7559SDave Watson return do_tls_getsockopt(sk, optname, optval, optlen); 3943c4d7559SDave Watson } 3953c4d7559SDave Watson 396c46234ebSDave Watson static int do_tls_setsockopt_conf(struct sock *sk, char __user *optval, 397c46234ebSDave Watson unsigned int optlen, int tx) 3983c4d7559SDave Watson { 399196c31b4SIlya Lesokhin struct tls_crypto_info *crypto_info; 4003c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 4013c4d7559SDave Watson int rc = 0; 40258371585SDave Watson int conf; 4033c4d7559SDave Watson 4043c4d7559SDave Watson if (!optval || (optlen < sizeof(*crypto_info))) { 4053c4d7559SDave Watson rc = -EINVAL; 4063c4d7559SDave Watson goto out; 4073c4d7559SDave Watson } 4083c4d7559SDave Watson 409c46234ebSDave Watson if (tx) 410196c31b4SIlya Lesokhin crypto_info = &ctx->crypto_send; 411c46234ebSDave Watson else 412c46234ebSDave Watson crypto_info = &ctx->crypto_recv; 413c46234ebSDave Watson 414196c31b4SIlya Lesokhin /* Currently we don't support set crypto info more than one time */ 415877d17c7SSabrina Dubroca if (TLS_CRYPTO_INFO_READY(crypto_info)) { 416877d17c7SSabrina Dubroca rc = -EBUSY; 417196c31b4SIlya Lesokhin goto out; 418877d17c7SSabrina Dubroca } 419196c31b4SIlya Lesokhin 420196c31b4SIlya Lesokhin rc = copy_from_user(crypto_info, optval, sizeof(*crypto_info)); 4213c4d7559SDave Watson if (rc) { 4223c4d7559SDave Watson rc = -EFAULT; 423257082e6SBoris Pismenny goto err_crypto_info; 4243c4d7559SDave Watson } 4253c4d7559SDave Watson 4263c4d7559SDave Watson /* check version */ 427196c31b4SIlya Lesokhin if (crypto_info->version != TLS_1_2_VERSION) { 4283c4d7559SDave Watson rc = -ENOTSUPP; 429196c31b4SIlya Lesokhin goto err_crypto_info; 4303c4d7559SDave Watson } 4313c4d7559SDave Watson 432196c31b4SIlya Lesokhin switch (crypto_info->cipher_type) { 4333c4d7559SDave Watson case TLS_CIPHER_AES_GCM_128: { 4343c4d7559SDave Watson if (optlen != sizeof(struct tls12_crypto_info_aes_gcm_128)) { 4353c4d7559SDave Watson rc = -EINVAL; 4366db959c8SSabrina Dubroca goto err_crypto_info; 4373c4d7559SDave Watson } 438196c31b4SIlya Lesokhin rc = copy_from_user(crypto_info + 1, optval + sizeof(*crypto_info), 439196c31b4SIlya Lesokhin optlen - sizeof(*crypto_info)); 4403c4d7559SDave Watson if (rc) { 4413c4d7559SDave Watson rc = -EFAULT; 4423c4d7559SDave Watson goto err_crypto_info; 4433c4d7559SDave Watson } 4443c4d7559SDave Watson break; 4453c4d7559SDave Watson } 4463c4d7559SDave Watson default: 4473c4d7559SDave Watson rc = -EINVAL; 4486db959c8SSabrina Dubroca goto err_crypto_info; 4493c4d7559SDave Watson } 4503c4d7559SDave Watson 4513c4d7559SDave Watson /* currently SW is default, we will have ethtool in future */ 452c46234ebSDave Watson if (tx) { 453c46234ebSDave Watson rc = tls_set_sw_offload(sk, ctx, 1); 454c46234ebSDave Watson if (ctx->conf == TLS_SW_RX) 455c46234ebSDave Watson conf = TLS_SW_RXTX; 456c46234ebSDave Watson else 45758371585SDave Watson conf = TLS_SW_TX; 458c46234ebSDave Watson } else { 459c46234ebSDave Watson rc = tls_set_sw_offload(sk, ctx, 0); 460c46234ebSDave Watson if (ctx->conf == TLS_SW_TX) 461c46234ebSDave Watson conf = TLS_SW_RXTX; 462c46234ebSDave Watson else 463c46234ebSDave Watson conf = TLS_SW_RX; 464c46234ebSDave Watson } 465c46234ebSDave Watson 4663c4d7559SDave Watson if (rc) 4673c4d7559SDave Watson goto err_crypto_info; 4683c4d7559SDave Watson 46958371585SDave Watson ctx->conf = conf; 4706d88207fSIlya Lesokhin update_sk_prot(sk, ctx); 471c46234ebSDave Watson if (tx) { 472ee181e52SIlya Lesokhin ctx->sk_write_space = sk->sk_write_space; 473ee181e52SIlya Lesokhin sk->sk_write_space = tls_write_space; 474c46234ebSDave Watson } else { 475c46234ebSDave Watson sk->sk_socket->ops = &tls_sw_proto_ops; 476c46234ebSDave Watson } 4773c4d7559SDave Watson goto out; 4783c4d7559SDave Watson 4793c4d7559SDave Watson err_crypto_info: 4803c4d7559SDave Watson memset(crypto_info, 0, sizeof(*crypto_info)); 4813c4d7559SDave Watson out: 4823c4d7559SDave Watson return rc; 4833c4d7559SDave Watson } 4843c4d7559SDave Watson 4853c4d7559SDave Watson static int do_tls_setsockopt(struct sock *sk, int optname, 4863c4d7559SDave Watson char __user *optval, unsigned int optlen) 4873c4d7559SDave Watson { 4883c4d7559SDave Watson int rc = 0; 4893c4d7559SDave Watson 4903c4d7559SDave Watson switch (optname) { 4913c4d7559SDave Watson case TLS_TX: 492c46234ebSDave Watson case TLS_RX: 4933c4d7559SDave Watson lock_sock(sk); 494c46234ebSDave Watson rc = do_tls_setsockopt_conf(sk, optval, optlen, 495c46234ebSDave Watson optname == TLS_TX); 4963c4d7559SDave Watson release_sock(sk); 4973c4d7559SDave Watson break; 4983c4d7559SDave Watson default: 4993c4d7559SDave Watson rc = -ENOPROTOOPT; 5003c4d7559SDave Watson break; 5013c4d7559SDave Watson } 5023c4d7559SDave Watson return rc; 5033c4d7559SDave Watson } 5043c4d7559SDave Watson 5053c4d7559SDave Watson static int tls_setsockopt(struct sock *sk, int level, int optname, 5063c4d7559SDave Watson char __user *optval, unsigned int optlen) 5073c4d7559SDave Watson { 5083c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 5093c4d7559SDave Watson 5103c4d7559SDave Watson if (level != SOL_TLS) 5113c4d7559SDave Watson return ctx->setsockopt(sk, level, optname, optval, optlen); 5123c4d7559SDave Watson 5133c4d7559SDave Watson return do_tls_setsockopt(sk, optname, optval, optlen); 5143c4d7559SDave Watson } 5153c4d7559SDave Watson 516dd0bed16SAtul Gupta static struct tls_context *create_ctx(struct sock *sk) 517dd0bed16SAtul Gupta { 518dd0bed16SAtul Gupta struct inet_connection_sock *icsk = inet_csk(sk); 519dd0bed16SAtul Gupta struct tls_context *ctx; 520dd0bed16SAtul Gupta 521dd0bed16SAtul Gupta ctx = kzalloc(sizeof(*ctx), GFP_KERNEL); 522dd0bed16SAtul Gupta if (!ctx) 523dd0bed16SAtul Gupta return NULL; 524dd0bed16SAtul Gupta 525dd0bed16SAtul Gupta icsk->icsk_ulp_data = ctx; 526dd0bed16SAtul Gupta return ctx; 527dd0bed16SAtul Gupta } 528dd0bed16SAtul Gupta 529dd0bed16SAtul Gupta static int tls_hw_prot(struct sock *sk) 530dd0bed16SAtul Gupta { 531dd0bed16SAtul Gupta struct tls_context *ctx; 532dd0bed16SAtul Gupta struct tls_device *dev; 533dd0bed16SAtul Gupta int rc = 0; 534dd0bed16SAtul Gupta 535dd0bed16SAtul Gupta mutex_lock(&device_mutex); 536dd0bed16SAtul Gupta list_for_each_entry(dev, &device_list, dev_list) { 537dd0bed16SAtul Gupta if (dev->feature && dev->feature(dev)) { 538dd0bed16SAtul Gupta ctx = create_ctx(sk); 539dd0bed16SAtul Gupta if (!ctx) 540dd0bed16SAtul Gupta goto out; 541dd0bed16SAtul Gupta 542dd0bed16SAtul Gupta ctx->hash = sk->sk_prot->hash; 543dd0bed16SAtul Gupta ctx->unhash = sk->sk_prot->unhash; 544dd0bed16SAtul Gupta ctx->sk_proto_close = sk->sk_prot->close; 545dd0bed16SAtul Gupta ctx->conf = TLS_HW_RECORD; 546dd0bed16SAtul Gupta update_sk_prot(sk, ctx); 547dd0bed16SAtul Gupta rc = 1; 548dd0bed16SAtul Gupta break; 549dd0bed16SAtul Gupta } 550dd0bed16SAtul Gupta } 551dd0bed16SAtul Gupta out: 552dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 553dd0bed16SAtul Gupta return rc; 554dd0bed16SAtul Gupta } 555dd0bed16SAtul Gupta 556dd0bed16SAtul Gupta static void tls_hw_unhash(struct sock *sk) 557dd0bed16SAtul Gupta { 558dd0bed16SAtul Gupta struct tls_context *ctx = tls_get_ctx(sk); 559dd0bed16SAtul Gupta struct tls_device *dev; 560dd0bed16SAtul Gupta 561dd0bed16SAtul Gupta mutex_lock(&device_mutex); 562dd0bed16SAtul Gupta list_for_each_entry(dev, &device_list, dev_list) { 563dd0bed16SAtul Gupta if (dev->unhash) 564dd0bed16SAtul Gupta dev->unhash(dev, sk); 565dd0bed16SAtul Gupta } 566dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 567dd0bed16SAtul Gupta ctx->unhash(sk); 568dd0bed16SAtul Gupta } 569dd0bed16SAtul Gupta 570dd0bed16SAtul Gupta static int tls_hw_hash(struct sock *sk) 571dd0bed16SAtul Gupta { 572dd0bed16SAtul Gupta struct tls_context *ctx = tls_get_ctx(sk); 573dd0bed16SAtul Gupta struct tls_device *dev; 574dd0bed16SAtul Gupta int err; 575dd0bed16SAtul Gupta 576dd0bed16SAtul Gupta err = ctx->hash(sk); 577dd0bed16SAtul Gupta mutex_lock(&device_mutex); 578dd0bed16SAtul Gupta list_for_each_entry(dev, &device_list, dev_list) { 579dd0bed16SAtul Gupta if (dev->hash) 580dd0bed16SAtul Gupta err |= dev->hash(dev, sk); 581dd0bed16SAtul Gupta } 582dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 583dd0bed16SAtul Gupta 584dd0bed16SAtul Gupta if (err) 585dd0bed16SAtul Gupta tls_hw_unhash(sk); 586dd0bed16SAtul Gupta return err; 587dd0bed16SAtul Gupta } 588dd0bed16SAtul Gupta 589c113187dSBoris Pismenny static void build_protos(struct proto *prot, struct proto *base) 590c113187dSBoris Pismenny { 59158371585SDave Watson prot[TLS_BASE] = *base; 59258371585SDave Watson prot[TLS_BASE].setsockopt = tls_setsockopt; 59358371585SDave Watson prot[TLS_BASE].getsockopt = tls_getsockopt; 59458371585SDave Watson prot[TLS_BASE].close = tls_sk_proto_close; 595c113187dSBoris Pismenny 59658371585SDave Watson prot[TLS_SW_TX] = prot[TLS_BASE]; 597c113187dSBoris Pismenny prot[TLS_SW_TX].sendmsg = tls_sw_sendmsg; 598c113187dSBoris Pismenny prot[TLS_SW_TX].sendpage = tls_sw_sendpage; 599c46234ebSDave Watson 600c46234ebSDave Watson prot[TLS_SW_RX] = prot[TLS_BASE]; 601c46234ebSDave Watson prot[TLS_SW_RX].recvmsg = tls_sw_recvmsg; 602c46234ebSDave Watson prot[TLS_SW_RX].close = tls_sk_proto_close; 603c46234ebSDave Watson 604c46234ebSDave Watson prot[TLS_SW_RXTX] = prot[TLS_SW_TX]; 605c46234ebSDave Watson prot[TLS_SW_RXTX].recvmsg = tls_sw_recvmsg; 606c46234ebSDave Watson prot[TLS_SW_RXTX].close = tls_sk_proto_close; 607dd0bed16SAtul Gupta 608dd0bed16SAtul Gupta prot[TLS_HW_RECORD] = *base; 609dd0bed16SAtul Gupta prot[TLS_HW_RECORD].hash = tls_hw_hash; 610dd0bed16SAtul Gupta prot[TLS_HW_RECORD].unhash = tls_hw_unhash; 611dd0bed16SAtul Gupta prot[TLS_HW_RECORD].close = tls_sk_proto_close; 612c113187dSBoris Pismenny } 613c113187dSBoris Pismenny 6143c4d7559SDave Watson static int tls_init(struct sock *sk) 6153c4d7559SDave Watson { 616c113187dSBoris Pismenny int ip_ver = sk->sk_family == AF_INET6 ? TLSV6 : TLSV4; 6173c4d7559SDave Watson struct tls_context *ctx; 6183c4d7559SDave Watson int rc = 0; 6193c4d7559SDave Watson 620dd0bed16SAtul Gupta if (tls_hw_prot(sk)) 621dd0bed16SAtul Gupta goto out; 622dd0bed16SAtul Gupta 623d91c3e17SIlya Lesokhin /* The TLS ulp is currently supported only for TCP sockets 624d91c3e17SIlya Lesokhin * in ESTABLISHED state. 625d91c3e17SIlya Lesokhin * Supporting sockets in LISTEN state will require us 626d91c3e17SIlya Lesokhin * to modify the accept implementation to clone rather then 627d91c3e17SIlya Lesokhin * share the ulp context. 628d91c3e17SIlya Lesokhin */ 629d91c3e17SIlya Lesokhin if (sk->sk_state != TCP_ESTABLISHED) 630d91c3e17SIlya Lesokhin return -ENOTSUPP; 631d91c3e17SIlya Lesokhin 6323c4d7559SDave Watson /* allocate tls context */ 633dd0bed16SAtul Gupta ctx = create_ctx(sk); 6343c4d7559SDave Watson if (!ctx) { 6353c4d7559SDave Watson rc = -ENOMEM; 6363c4d7559SDave Watson goto out; 6373c4d7559SDave Watson } 6383c4d7559SDave Watson ctx->setsockopt = sk->sk_prot->setsockopt; 6393c4d7559SDave Watson ctx->getsockopt = sk->sk_prot->getsockopt; 640ff45d820SIlya Lesokhin ctx->sk_proto_close = sk->sk_prot->close; 6416d88207fSIlya Lesokhin 642c113187dSBoris Pismenny /* Build IPv6 TLS whenever the address of tcpv6_prot changes */ 643c113187dSBoris Pismenny if (ip_ver == TLSV6 && 644c113187dSBoris Pismenny unlikely(sk->sk_prot != smp_load_acquire(&saved_tcpv6_prot))) { 645c113187dSBoris Pismenny mutex_lock(&tcpv6_prot_mutex); 646c113187dSBoris Pismenny if (likely(sk->sk_prot != saved_tcpv6_prot)) { 647c113187dSBoris Pismenny build_protos(tls_prots[TLSV6], sk->sk_prot); 648c113187dSBoris Pismenny smp_store_release(&saved_tcpv6_prot, sk->sk_prot); 649c113187dSBoris Pismenny } 650c113187dSBoris Pismenny mutex_unlock(&tcpv6_prot_mutex); 651c113187dSBoris Pismenny } 652c113187dSBoris Pismenny 65358371585SDave Watson ctx->conf = TLS_BASE; 6546d88207fSIlya Lesokhin update_sk_prot(sk, ctx); 6553c4d7559SDave Watson out: 6563c4d7559SDave Watson return rc; 6573c4d7559SDave Watson } 6583c4d7559SDave Watson 659dd0bed16SAtul Gupta void tls_register_device(struct tls_device *device) 660dd0bed16SAtul Gupta { 661dd0bed16SAtul Gupta mutex_lock(&device_mutex); 662dd0bed16SAtul Gupta list_add_tail(&device->dev_list, &device_list); 663dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 664dd0bed16SAtul Gupta } 665dd0bed16SAtul Gupta EXPORT_SYMBOL(tls_register_device); 666dd0bed16SAtul Gupta 667dd0bed16SAtul Gupta void tls_unregister_device(struct tls_device *device) 668dd0bed16SAtul Gupta { 669dd0bed16SAtul Gupta mutex_lock(&device_mutex); 670dd0bed16SAtul Gupta list_del(&device->dev_list); 671dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 672dd0bed16SAtul Gupta } 673dd0bed16SAtul Gupta EXPORT_SYMBOL(tls_unregister_device); 674dd0bed16SAtul Gupta 6753c4d7559SDave Watson static struct tcp_ulp_ops tcp_tls_ulp_ops __read_mostly = { 6763c4d7559SDave Watson .name = "tls", 677b11a632cSJohn Fastabend .uid = TCP_ULP_TLS, 678b11a632cSJohn Fastabend .user_visible = true, 6793c4d7559SDave Watson .owner = THIS_MODULE, 6803c4d7559SDave Watson .init = tls_init, 6813c4d7559SDave Watson }; 6823c4d7559SDave Watson 6833c4d7559SDave Watson static int __init tls_register(void) 6843c4d7559SDave Watson { 685c113187dSBoris Pismenny build_protos(tls_prots[TLSV4], &tcp_prot); 6863c4d7559SDave Watson 687c46234ebSDave Watson tls_sw_proto_ops = inet_stream_ops; 688c46234ebSDave Watson tls_sw_proto_ops.poll = tls_sw_poll; 689c46234ebSDave Watson tls_sw_proto_ops.splice_read = tls_sw_splice_read; 690c46234ebSDave Watson 6913c4d7559SDave Watson tcp_register_ulp(&tcp_tls_ulp_ops); 6923c4d7559SDave Watson 6933c4d7559SDave Watson return 0; 6943c4d7559SDave Watson } 6953c4d7559SDave Watson 6963c4d7559SDave Watson static void __exit tls_unregister(void) 6973c4d7559SDave Watson { 6983c4d7559SDave Watson tcp_unregister_ulp(&tcp_tls_ulp_ops); 6993c4d7559SDave Watson } 7003c4d7559SDave Watson 7013c4d7559SDave Watson module_init(tls_register); 7023c4d7559SDave Watson module_exit(tls_unregister); 703