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); 61dd0bed16SAtul Gupta static DEFINE_MUTEX(device_mutex); 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; 646d88207fSIlya Lesokhin 65f66de3eeSBoris Pismenny static void update_sk_prot(struct sock *sk, struct tls_context *ctx) 666d88207fSIlya Lesokhin { 67c113187dSBoris Pismenny int ip_ver = sk->sk_family == AF_INET6 ? TLSV6 : TLSV4; 68c113187dSBoris Pismenny 69f66de3eeSBoris Pismenny sk->sk_prot = &tls_prots[ip_ver][ctx->tx_conf][ctx->rx_conf]; 706d88207fSIlya Lesokhin } 713c4d7559SDave Watson 723c4d7559SDave Watson int wait_on_pending_writer(struct sock *sk, long *timeo) 733c4d7559SDave Watson { 743c4d7559SDave Watson int rc = 0; 753c4d7559SDave Watson DEFINE_WAIT_FUNC(wait, woken_wake_function); 763c4d7559SDave Watson 773c4d7559SDave Watson add_wait_queue(sk_sleep(sk), &wait); 783c4d7559SDave Watson while (1) { 793c4d7559SDave Watson if (!*timeo) { 803c4d7559SDave Watson rc = -EAGAIN; 813c4d7559SDave Watson break; 823c4d7559SDave Watson } 833c4d7559SDave Watson 843c4d7559SDave Watson if (signal_pending(current)) { 853c4d7559SDave Watson rc = sock_intr_errno(*timeo); 863c4d7559SDave Watson break; 873c4d7559SDave Watson } 883c4d7559SDave Watson 893c4d7559SDave Watson if (sk_wait_event(sk, timeo, !sk->sk_write_pending, &wait)) 903c4d7559SDave Watson break; 913c4d7559SDave Watson } 923c4d7559SDave Watson remove_wait_queue(sk_sleep(sk), &wait); 933c4d7559SDave Watson return rc; 943c4d7559SDave Watson } 953c4d7559SDave Watson 963c4d7559SDave Watson int tls_push_sg(struct sock *sk, 973c4d7559SDave Watson struct tls_context *ctx, 983c4d7559SDave Watson struct scatterlist *sg, 993c4d7559SDave Watson u16 first_offset, 1003c4d7559SDave Watson int flags) 1013c4d7559SDave Watson { 1023c4d7559SDave Watson int sendpage_flags = flags | MSG_SENDPAGE_NOTLAST; 1033c4d7559SDave Watson int ret = 0; 1043c4d7559SDave Watson struct page *p; 1053c4d7559SDave Watson size_t size; 1063c4d7559SDave Watson int offset = first_offset; 1073c4d7559SDave Watson 1083c4d7559SDave Watson size = sg->length - offset; 1093c4d7559SDave Watson offset += sg->offset; 1103c4d7559SDave Watson 111c212d2c7SDave Watson ctx->in_tcp_sendpages = true; 1123c4d7559SDave Watson while (1) { 1133c4d7559SDave Watson if (sg_is_last(sg)) 1143c4d7559SDave Watson sendpage_flags = flags; 1153c4d7559SDave Watson 1163c4d7559SDave Watson /* is sending application-limited? */ 1173c4d7559SDave Watson tcp_rate_check_app_limited(sk); 1183c4d7559SDave Watson p = sg_page(sg); 1193c4d7559SDave Watson retry: 1203c4d7559SDave Watson ret = do_tcp_sendpages(sk, p, offset, size, sendpage_flags); 1213c4d7559SDave Watson 1223c4d7559SDave Watson if (ret != size) { 1233c4d7559SDave Watson if (ret > 0) { 1243c4d7559SDave Watson offset += ret; 1253c4d7559SDave Watson size -= ret; 1263c4d7559SDave Watson goto retry; 1273c4d7559SDave Watson } 1283c4d7559SDave Watson 1293c4d7559SDave Watson offset -= sg->offset; 1303c4d7559SDave Watson ctx->partially_sent_offset = offset; 1313c4d7559SDave Watson ctx->partially_sent_record = (void *)sg; 132080324c3SAndre Tomt ctx->in_tcp_sendpages = false; 1333c4d7559SDave Watson return ret; 1343c4d7559SDave Watson } 1353c4d7559SDave Watson 1363c4d7559SDave Watson put_page(p); 1373c4d7559SDave Watson sk_mem_uncharge(sk, sg->length); 1383c4d7559SDave Watson sg = sg_next(sg); 1393c4d7559SDave Watson if (!sg) 1403c4d7559SDave Watson break; 1413c4d7559SDave Watson 1423c4d7559SDave Watson offset = sg->offset; 1433c4d7559SDave Watson size = sg->length; 1443c4d7559SDave Watson } 1453c4d7559SDave Watson 146c212d2c7SDave Watson ctx->in_tcp_sendpages = false; 147c212d2c7SDave Watson ctx->sk_write_space(sk); 1483c4d7559SDave Watson 1493c4d7559SDave Watson return 0; 1503c4d7559SDave Watson } 1513c4d7559SDave Watson 1523c4d7559SDave Watson static int tls_handle_open_record(struct sock *sk, int flags) 1533c4d7559SDave Watson { 1543c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 1553c4d7559SDave Watson 1563c4d7559SDave Watson if (tls_is_pending_open_record(ctx)) 1573c4d7559SDave Watson return ctx->push_pending_record(sk, flags); 1583c4d7559SDave Watson 1593c4d7559SDave Watson return 0; 1603c4d7559SDave Watson } 1613c4d7559SDave Watson 1623c4d7559SDave Watson int tls_proccess_cmsg(struct sock *sk, struct msghdr *msg, 1633c4d7559SDave Watson unsigned char *record_type) 1643c4d7559SDave Watson { 1653c4d7559SDave Watson struct cmsghdr *cmsg; 1663c4d7559SDave Watson int rc = -EINVAL; 1673c4d7559SDave Watson 1683c4d7559SDave Watson for_each_cmsghdr(cmsg, msg) { 1693c4d7559SDave Watson if (!CMSG_OK(msg, cmsg)) 1703c4d7559SDave Watson return -EINVAL; 1713c4d7559SDave Watson if (cmsg->cmsg_level != SOL_TLS) 1723c4d7559SDave Watson continue; 1733c4d7559SDave Watson 1743c4d7559SDave Watson switch (cmsg->cmsg_type) { 1753c4d7559SDave Watson case TLS_SET_RECORD_TYPE: 1763c4d7559SDave Watson if (cmsg->cmsg_len < CMSG_LEN(sizeof(*record_type))) 1773c4d7559SDave Watson return -EINVAL; 1783c4d7559SDave Watson 1793c4d7559SDave Watson if (msg->msg_flags & MSG_MORE) 1803c4d7559SDave Watson return -EINVAL; 1813c4d7559SDave Watson 1823c4d7559SDave Watson rc = tls_handle_open_record(sk, msg->msg_flags); 1833c4d7559SDave Watson if (rc) 1843c4d7559SDave Watson return rc; 1853c4d7559SDave Watson 1863c4d7559SDave Watson *record_type = *(unsigned char *)CMSG_DATA(cmsg); 1873c4d7559SDave Watson rc = 0; 1883c4d7559SDave Watson break; 1893c4d7559SDave Watson default: 1903c4d7559SDave Watson return -EINVAL; 1913c4d7559SDave Watson } 1923c4d7559SDave Watson } 1933c4d7559SDave Watson 1943c4d7559SDave Watson return rc; 1953c4d7559SDave Watson } 1963c4d7559SDave Watson 197a42055e8SVakul Garg int tls_push_partial_record(struct sock *sk, struct tls_context *ctx, 198a42055e8SVakul Garg int flags) 1993c4d7559SDave Watson { 2003c4d7559SDave Watson struct scatterlist *sg; 2013c4d7559SDave Watson u16 offset; 2023c4d7559SDave Watson 2033c4d7559SDave Watson sg = ctx->partially_sent_record; 2043c4d7559SDave Watson offset = ctx->partially_sent_offset; 2053c4d7559SDave Watson 2063c4d7559SDave Watson ctx->partially_sent_record = NULL; 2073c4d7559SDave Watson return tls_push_sg(sk, ctx, sg, offset, flags); 2083c4d7559SDave Watson } 2093c4d7559SDave Watson 210a42055e8SVakul Garg int tls_push_pending_closed_record(struct sock *sk, 211a42055e8SVakul Garg struct tls_context *tls_ctx, 212a42055e8SVakul Garg int flags, long *timeo) 213a42055e8SVakul Garg { 214a42055e8SVakul Garg struct tls_sw_context_tx *ctx = tls_sw_ctx_tx(tls_ctx); 215a42055e8SVakul Garg 216a42055e8SVakul Garg if (tls_is_partially_sent_record(tls_ctx) || 2179932a29aSVakul Garg !list_empty(&ctx->tx_list)) 218a42055e8SVakul Garg return tls_tx_records(sk, flags); 219a42055e8SVakul Garg else 220a42055e8SVakul Garg return tls_ctx->push_pending_record(sk, flags); 221a42055e8SVakul Garg } 222a42055e8SVakul Garg 2233c4d7559SDave Watson static void tls_write_space(struct sock *sk) 2243c4d7559SDave Watson { 2253c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 226a42055e8SVakul Garg struct tls_sw_context_tx *tx_ctx = tls_sw_ctx_tx(ctx); 2273c4d7559SDave Watson 22867db7cd2SJohn Fastabend /* If in_tcp_sendpages call lower protocol write space handler 22967db7cd2SJohn Fastabend * to ensure we wake up any waiting operations there. For example 23067db7cd2SJohn Fastabend * if do_tcp_sendpages where to call sk_wait_event. 23167db7cd2SJohn Fastabend */ 23267db7cd2SJohn Fastabend if (ctx->in_tcp_sendpages) { 23367db7cd2SJohn Fastabend ctx->sk_write_space(sk); 234c212d2c7SDave Watson return; 23567db7cd2SJohn Fastabend } 236c212d2c7SDave Watson 237a42055e8SVakul Garg /* Schedule the transmission if tx list is ready */ 2389932a29aSVakul Garg if (is_tx_ready(tx_ctx) && !sk->sk_write_pending) { 239a42055e8SVakul Garg /* Schedule the transmission */ 240a42055e8SVakul Garg if (!test_and_set_bit(BIT_TX_SCHEDULED, &tx_ctx->tx_bitmask)) 241a42055e8SVakul Garg schedule_delayed_work(&tx_ctx->tx_work.work, 0); 2423c4d7559SDave Watson } 2433c4d7559SDave Watson 2443c4d7559SDave Watson ctx->sk_write_space(sk); 2453c4d7559SDave Watson } 2463c4d7559SDave Watson 24786029d10SSabrina Dubroca static void tls_ctx_free(struct tls_context *ctx) 24886029d10SSabrina Dubroca { 24986029d10SSabrina Dubroca if (!ctx) 25086029d10SSabrina Dubroca return; 25186029d10SSabrina Dubroca 25286029d10SSabrina Dubroca memzero_explicit(&ctx->crypto_send, sizeof(ctx->crypto_send)); 25386029d10SSabrina Dubroca memzero_explicit(&ctx->crypto_recv, sizeof(ctx->crypto_recv)); 25486029d10SSabrina Dubroca kfree(ctx); 25586029d10SSabrina Dubroca } 25686029d10SSabrina Dubroca 2573c4d7559SDave Watson static void tls_sk_proto_close(struct sock *sk, long timeout) 2583c4d7559SDave Watson { 2593c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 2603c4d7559SDave Watson long timeo = sock_sndtimeo(sk, 0); 2613c4d7559SDave Watson void (*sk_proto_close)(struct sock *sk, long timeout); 26298f0a395SEric Dumazet bool free_ctx = false; 2633c4d7559SDave Watson 2643c4d7559SDave Watson lock_sock(sk); 265ff45d820SIlya Lesokhin sk_proto_close = ctx->sk_proto_close; 266ff45d820SIlya Lesokhin 267b2d6cee1SDavid S. Miller if ((ctx->tx_conf == TLS_HW_RECORD && ctx->rx_conf == TLS_HW_RECORD) || 268b2d6cee1SDavid S. Miller (ctx->tx_conf == TLS_BASE && ctx->rx_conf == TLS_BASE)) { 26998f0a395SEric Dumazet free_ctx = true; 270ff45d820SIlya Lesokhin goto skip_tx_cleanup; 271ff45d820SIlya Lesokhin } 2723c4d7559SDave Watson 2733c4d7559SDave Watson if (!tls_complete_pending_work(sk, ctx, 0, &timeo)) 2743c4d7559SDave Watson tls_handle_open_record(sk, 0); 2753c4d7559SDave Watson 276f66de3eeSBoris Pismenny /* We need these for tls_sw_fallback handling of other packets */ 277f66de3eeSBoris Pismenny if (ctx->tx_conf == TLS_SW) { 278dbe42559SDave Watson kfree(ctx->tx.rec_seq); 279dbe42559SDave Watson kfree(ctx->tx.iv); 280f66de3eeSBoris Pismenny tls_sw_free_resources_tx(sk); 281f66de3eeSBoris Pismenny } 282f66de3eeSBoris Pismenny 283f66de3eeSBoris Pismenny if (ctx->rx_conf == TLS_SW) { 284c46234ebSDave Watson kfree(ctx->rx.rec_seq); 285c46234ebSDave Watson kfree(ctx->rx.iv); 286f66de3eeSBoris Pismenny tls_sw_free_resources_rx(sk); 287c46234ebSDave Watson } 2883c4d7559SDave Watson 289e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE 2904799ac81SBoris Pismenny if (ctx->rx_conf == TLS_HW) 2914799ac81SBoris Pismenny tls_device_offload_cleanup_rx(sk); 2924799ac81SBoris Pismenny 2934799ac81SBoris Pismenny if (ctx->tx_conf != TLS_HW && ctx->rx_conf != TLS_HW) { 294e8f69799SIlya Lesokhin #else 295e8f69799SIlya Lesokhin { 296e8f69799SIlya Lesokhin #endif 29786029d10SSabrina Dubroca tls_ctx_free(ctx); 298e8f69799SIlya Lesokhin ctx = NULL; 299e8f69799SIlya Lesokhin } 300e8f69799SIlya Lesokhin 301ff45d820SIlya Lesokhin skip_tx_cleanup: 3023c4d7559SDave Watson release_sock(sk); 3033c4d7559SDave Watson sk_proto_close(sk, timeout); 304dd0bed16SAtul Gupta /* free ctx for TLS_HW_RECORD, used by tcp_set_state 305dd0bed16SAtul Gupta * for sk->sk_prot->unhash [tls_hw_unhash] 306dd0bed16SAtul Gupta */ 30798f0a395SEric Dumazet if (free_ctx) 30886029d10SSabrina Dubroca tls_ctx_free(ctx); 3093c4d7559SDave Watson } 3103c4d7559SDave Watson 3113c4d7559SDave Watson static int do_tls_getsockopt_tx(struct sock *sk, char __user *optval, 3123c4d7559SDave Watson int __user *optlen) 3133c4d7559SDave Watson { 3143c4d7559SDave Watson int rc = 0; 3153c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 3163c4d7559SDave Watson struct tls_crypto_info *crypto_info; 3173c4d7559SDave Watson int len; 3183c4d7559SDave Watson 3193c4d7559SDave Watson if (get_user(len, optlen)) 3203c4d7559SDave Watson return -EFAULT; 3213c4d7559SDave Watson 3223c4d7559SDave Watson if (!optval || (len < sizeof(*crypto_info))) { 3233c4d7559SDave Watson rc = -EINVAL; 3243c4d7559SDave Watson goto out; 3253c4d7559SDave Watson } 3263c4d7559SDave Watson 3273c4d7559SDave Watson if (!ctx) { 3283c4d7559SDave Watson rc = -EBUSY; 3293c4d7559SDave Watson goto out; 3303c4d7559SDave Watson } 3313c4d7559SDave Watson 3323c4d7559SDave Watson /* get user crypto info */ 33386029d10SSabrina Dubroca crypto_info = &ctx->crypto_send.info; 3343c4d7559SDave Watson 3353c4d7559SDave Watson if (!TLS_CRYPTO_INFO_READY(crypto_info)) { 3363c4d7559SDave Watson rc = -EBUSY; 3373c4d7559SDave Watson goto out; 3383c4d7559SDave Watson } 3393c4d7559SDave Watson 3405a3b886cSMatthias Rosenfelder if (len == sizeof(*crypto_info)) { 341ac55cd61SDan Carpenter if (copy_to_user(optval, crypto_info, sizeof(*crypto_info))) 342ac55cd61SDan Carpenter rc = -EFAULT; 3433c4d7559SDave Watson goto out; 3443c4d7559SDave Watson } 3453c4d7559SDave Watson 3463c4d7559SDave Watson switch (crypto_info->cipher_type) { 3473c4d7559SDave Watson case TLS_CIPHER_AES_GCM_128: { 3483c4d7559SDave Watson struct tls12_crypto_info_aes_gcm_128 * 3493c4d7559SDave Watson crypto_info_aes_gcm_128 = 3503c4d7559SDave Watson container_of(crypto_info, 3513c4d7559SDave Watson struct tls12_crypto_info_aes_gcm_128, 3523c4d7559SDave Watson info); 3533c4d7559SDave Watson 3543c4d7559SDave Watson if (len != sizeof(*crypto_info_aes_gcm_128)) { 3553c4d7559SDave Watson rc = -EINVAL; 3563c4d7559SDave Watson goto out; 3573c4d7559SDave Watson } 3583c4d7559SDave Watson lock_sock(sk); 359a1dfa681SBoris Pismenny memcpy(crypto_info_aes_gcm_128->iv, 360dbe42559SDave Watson ctx->tx.iv + TLS_CIPHER_AES_GCM_128_SALT_SIZE, 3613c4d7559SDave Watson TLS_CIPHER_AES_GCM_128_IV_SIZE); 362dbe42559SDave Watson memcpy(crypto_info_aes_gcm_128->rec_seq, ctx->tx.rec_seq, 363c410c196SBoris Pismenny TLS_CIPHER_AES_GCM_128_REC_SEQ_SIZE); 3643c4d7559SDave Watson release_sock(sk); 365ac55cd61SDan Carpenter if (copy_to_user(optval, 3663c4d7559SDave Watson crypto_info_aes_gcm_128, 367ac55cd61SDan Carpenter sizeof(*crypto_info_aes_gcm_128))) 368ac55cd61SDan Carpenter rc = -EFAULT; 3693c4d7559SDave Watson break; 3703c4d7559SDave Watson } 3713c4d7559SDave Watson default: 3723c4d7559SDave Watson rc = -EINVAL; 3733c4d7559SDave Watson } 3743c4d7559SDave Watson 3753c4d7559SDave Watson out: 3763c4d7559SDave Watson return rc; 3773c4d7559SDave Watson } 3783c4d7559SDave Watson 3793c4d7559SDave Watson static int do_tls_getsockopt(struct sock *sk, int optname, 3803c4d7559SDave Watson char __user *optval, int __user *optlen) 3813c4d7559SDave Watson { 3823c4d7559SDave Watson int rc = 0; 3833c4d7559SDave Watson 3843c4d7559SDave Watson switch (optname) { 3853c4d7559SDave Watson case TLS_TX: 3863c4d7559SDave Watson rc = do_tls_getsockopt_tx(sk, optval, optlen); 3873c4d7559SDave Watson break; 3883c4d7559SDave Watson default: 3893c4d7559SDave Watson rc = -ENOPROTOOPT; 3903c4d7559SDave Watson break; 3913c4d7559SDave Watson } 3923c4d7559SDave Watson return rc; 3933c4d7559SDave Watson } 3943c4d7559SDave Watson 3953c4d7559SDave Watson static int tls_getsockopt(struct sock *sk, int level, int optname, 3963c4d7559SDave Watson char __user *optval, int __user *optlen) 3973c4d7559SDave Watson { 3983c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 3993c4d7559SDave Watson 4003c4d7559SDave Watson if (level != SOL_TLS) 4013c4d7559SDave Watson return ctx->getsockopt(sk, level, optname, optval, optlen); 4023c4d7559SDave Watson 4033c4d7559SDave Watson return do_tls_getsockopt(sk, optname, optval, optlen); 4043c4d7559SDave Watson } 4053c4d7559SDave Watson 406c46234ebSDave Watson static int do_tls_setsockopt_conf(struct sock *sk, char __user *optval, 407c46234ebSDave Watson unsigned int optlen, int tx) 4083c4d7559SDave Watson { 409196c31b4SIlya Lesokhin struct tls_crypto_info *crypto_info; 4103c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 4113c4d7559SDave Watson int rc = 0; 41258371585SDave Watson int conf; 4133c4d7559SDave Watson 4143c4d7559SDave Watson if (!optval || (optlen < sizeof(*crypto_info))) { 4153c4d7559SDave Watson rc = -EINVAL; 4163c4d7559SDave Watson goto out; 4173c4d7559SDave Watson } 4183c4d7559SDave Watson 419c46234ebSDave Watson if (tx) 42086029d10SSabrina Dubroca crypto_info = &ctx->crypto_send.info; 421c46234ebSDave Watson else 42286029d10SSabrina Dubroca crypto_info = &ctx->crypto_recv.info; 423c46234ebSDave Watson 424196c31b4SIlya Lesokhin /* Currently we don't support set crypto info more than one time */ 425877d17c7SSabrina Dubroca if (TLS_CRYPTO_INFO_READY(crypto_info)) { 426877d17c7SSabrina Dubroca rc = -EBUSY; 427196c31b4SIlya Lesokhin goto out; 428877d17c7SSabrina Dubroca } 429196c31b4SIlya Lesokhin 430196c31b4SIlya Lesokhin rc = copy_from_user(crypto_info, optval, sizeof(*crypto_info)); 4313c4d7559SDave Watson if (rc) { 4323c4d7559SDave Watson rc = -EFAULT; 433257082e6SBoris Pismenny goto err_crypto_info; 4343c4d7559SDave Watson } 4353c4d7559SDave Watson 4363c4d7559SDave Watson /* check version */ 437196c31b4SIlya Lesokhin if (crypto_info->version != TLS_1_2_VERSION) { 4383c4d7559SDave Watson rc = -ENOTSUPP; 439196c31b4SIlya Lesokhin goto err_crypto_info; 4403c4d7559SDave Watson } 4413c4d7559SDave Watson 442196c31b4SIlya Lesokhin switch (crypto_info->cipher_type) { 4433c4d7559SDave Watson case TLS_CIPHER_AES_GCM_128: { 4443c4d7559SDave Watson if (optlen != sizeof(struct tls12_crypto_info_aes_gcm_128)) { 4453c4d7559SDave Watson rc = -EINVAL; 4466db959c8SSabrina Dubroca goto err_crypto_info; 4473c4d7559SDave Watson } 448196c31b4SIlya Lesokhin rc = copy_from_user(crypto_info + 1, optval + sizeof(*crypto_info), 449196c31b4SIlya Lesokhin optlen - sizeof(*crypto_info)); 4503c4d7559SDave Watson if (rc) { 4513c4d7559SDave Watson rc = -EFAULT; 4523c4d7559SDave Watson goto err_crypto_info; 4533c4d7559SDave Watson } 4543c4d7559SDave Watson break; 4553c4d7559SDave Watson } 4563c4d7559SDave Watson default: 4573c4d7559SDave Watson rc = -EINVAL; 4586db959c8SSabrina Dubroca goto err_crypto_info; 4593c4d7559SDave Watson } 4603c4d7559SDave Watson 461c46234ebSDave Watson if (tx) { 462e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE 463e8f69799SIlya Lesokhin rc = tls_set_device_offload(sk, ctx); 464e8f69799SIlya Lesokhin conf = TLS_HW; 465e8f69799SIlya Lesokhin if (rc) { 466e8f69799SIlya Lesokhin #else 467e8f69799SIlya Lesokhin { 468e8f69799SIlya Lesokhin #endif 469c46234ebSDave Watson rc = tls_set_sw_offload(sk, ctx, 1); 470f66de3eeSBoris Pismenny conf = TLS_SW; 471e8f69799SIlya Lesokhin } 472c46234ebSDave Watson } else { 4734799ac81SBoris Pismenny #ifdef CONFIG_TLS_DEVICE 4744799ac81SBoris Pismenny rc = tls_set_device_offload_rx(sk, ctx); 4754799ac81SBoris Pismenny conf = TLS_HW; 4764799ac81SBoris Pismenny if (rc) { 4774799ac81SBoris Pismenny #else 4784799ac81SBoris Pismenny { 4794799ac81SBoris Pismenny #endif 480c46234ebSDave Watson rc = tls_set_sw_offload(sk, ctx, 0); 481f66de3eeSBoris Pismenny conf = TLS_SW; 482c46234ebSDave Watson } 4834799ac81SBoris Pismenny } 484c46234ebSDave Watson 4853c4d7559SDave Watson if (rc) 4863c4d7559SDave Watson goto err_crypto_info; 4873c4d7559SDave Watson 488f66de3eeSBoris Pismenny if (tx) 489f66de3eeSBoris Pismenny ctx->tx_conf = conf; 490f66de3eeSBoris Pismenny else 491f66de3eeSBoris Pismenny ctx->rx_conf = conf; 4926d88207fSIlya Lesokhin update_sk_prot(sk, ctx); 493c46234ebSDave Watson if (tx) { 494ee181e52SIlya Lesokhin ctx->sk_write_space = sk->sk_write_space; 495ee181e52SIlya Lesokhin sk->sk_write_space = tls_write_space; 496c46234ebSDave Watson } else { 497c46234ebSDave Watson sk->sk_socket->ops = &tls_sw_proto_ops; 498c46234ebSDave Watson } 4993c4d7559SDave Watson goto out; 5003c4d7559SDave Watson 5013c4d7559SDave Watson err_crypto_info: 502c844eb46SSabrina Dubroca memzero_explicit(crypto_info, sizeof(union tls_crypto_context)); 5033c4d7559SDave Watson out: 5043c4d7559SDave Watson return rc; 5053c4d7559SDave Watson } 5063c4d7559SDave Watson 5073c4d7559SDave Watson static int do_tls_setsockopt(struct sock *sk, int optname, 5083c4d7559SDave Watson char __user *optval, unsigned int optlen) 5093c4d7559SDave Watson { 5103c4d7559SDave Watson int rc = 0; 5113c4d7559SDave Watson 5123c4d7559SDave Watson switch (optname) { 5133c4d7559SDave Watson case TLS_TX: 514c46234ebSDave Watson case TLS_RX: 5153c4d7559SDave Watson lock_sock(sk); 516c46234ebSDave Watson rc = do_tls_setsockopt_conf(sk, optval, optlen, 517c46234ebSDave Watson optname == TLS_TX); 5183c4d7559SDave Watson release_sock(sk); 5193c4d7559SDave Watson break; 5203c4d7559SDave Watson default: 5213c4d7559SDave Watson rc = -ENOPROTOOPT; 5223c4d7559SDave Watson break; 5233c4d7559SDave Watson } 5243c4d7559SDave Watson return rc; 5253c4d7559SDave Watson } 5263c4d7559SDave Watson 5273c4d7559SDave Watson static int tls_setsockopt(struct sock *sk, int level, int optname, 5283c4d7559SDave Watson char __user *optval, unsigned int optlen) 5293c4d7559SDave Watson { 5303c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 5313c4d7559SDave Watson 5323c4d7559SDave Watson if (level != SOL_TLS) 5333c4d7559SDave Watson return ctx->setsockopt(sk, level, optname, optval, optlen); 5343c4d7559SDave Watson 5353c4d7559SDave Watson return do_tls_setsockopt(sk, optname, optval, optlen); 5363c4d7559SDave Watson } 5373c4d7559SDave Watson 538dd0bed16SAtul Gupta static struct tls_context *create_ctx(struct sock *sk) 539dd0bed16SAtul Gupta { 540dd0bed16SAtul Gupta struct inet_connection_sock *icsk = inet_csk(sk); 541dd0bed16SAtul Gupta struct tls_context *ctx; 542dd0bed16SAtul Gupta 543dd0bed16SAtul Gupta ctx = kzalloc(sizeof(*ctx), GFP_KERNEL); 544dd0bed16SAtul Gupta if (!ctx) 545dd0bed16SAtul Gupta return NULL; 546dd0bed16SAtul Gupta 547dd0bed16SAtul Gupta icsk->icsk_ulp_data = ctx; 548dd0bed16SAtul Gupta return ctx; 549dd0bed16SAtul Gupta } 550dd0bed16SAtul Gupta 551dd0bed16SAtul Gupta static int tls_hw_prot(struct sock *sk) 552dd0bed16SAtul Gupta { 553dd0bed16SAtul Gupta struct tls_context *ctx; 554dd0bed16SAtul Gupta struct tls_device *dev; 555dd0bed16SAtul Gupta int rc = 0; 556dd0bed16SAtul Gupta 557dd0bed16SAtul Gupta mutex_lock(&device_mutex); 558dd0bed16SAtul Gupta list_for_each_entry(dev, &device_list, dev_list) { 559dd0bed16SAtul Gupta if (dev->feature && dev->feature(dev)) { 560dd0bed16SAtul Gupta ctx = create_ctx(sk); 561dd0bed16SAtul Gupta if (!ctx) 562dd0bed16SAtul Gupta goto out; 563dd0bed16SAtul Gupta 564dd0bed16SAtul Gupta ctx->hash = sk->sk_prot->hash; 565dd0bed16SAtul Gupta ctx->unhash = sk->sk_prot->unhash; 566dd0bed16SAtul Gupta ctx->sk_proto_close = sk->sk_prot->close; 567f66de3eeSBoris Pismenny ctx->rx_conf = TLS_HW_RECORD; 568f66de3eeSBoris Pismenny ctx->tx_conf = TLS_HW_RECORD; 569dd0bed16SAtul Gupta update_sk_prot(sk, ctx); 570dd0bed16SAtul Gupta rc = 1; 571dd0bed16SAtul Gupta break; 572dd0bed16SAtul Gupta } 573dd0bed16SAtul Gupta } 574dd0bed16SAtul Gupta out: 575dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 576dd0bed16SAtul Gupta return rc; 577dd0bed16SAtul Gupta } 578dd0bed16SAtul Gupta 579dd0bed16SAtul Gupta static void tls_hw_unhash(struct sock *sk) 580dd0bed16SAtul Gupta { 581dd0bed16SAtul Gupta struct tls_context *ctx = tls_get_ctx(sk); 582dd0bed16SAtul Gupta struct tls_device *dev; 583dd0bed16SAtul Gupta 584dd0bed16SAtul Gupta mutex_lock(&device_mutex); 585dd0bed16SAtul Gupta list_for_each_entry(dev, &device_list, dev_list) { 586dd0bed16SAtul Gupta if (dev->unhash) 587dd0bed16SAtul Gupta dev->unhash(dev, sk); 588dd0bed16SAtul Gupta } 589dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 590dd0bed16SAtul Gupta ctx->unhash(sk); 591dd0bed16SAtul Gupta } 592dd0bed16SAtul Gupta 593dd0bed16SAtul Gupta static int tls_hw_hash(struct sock *sk) 594dd0bed16SAtul Gupta { 595dd0bed16SAtul Gupta struct tls_context *ctx = tls_get_ctx(sk); 596dd0bed16SAtul Gupta struct tls_device *dev; 597dd0bed16SAtul Gupta int err; 598dd0bed16SAtul Gupta 599dd0bed16SAtul Gupta err = ctx->hash(sk); 600dd0bed16SAtul Gupta mutex_lock(&device_mutex); 601dd0bed16SAtul Gupta list_for_each_entry(dev, &device_list, dev_list) { 602dd0bed16SAtul Gupta if (dev->hash) 603dd0bed16SAtul Gupta err |= dev->hash(dev, sk); 604dd0bed16SAtul Gupta } 605dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 606dd0bed16SAtul Gupta 607dd0bed16SAtul Gupta if (err) 608dd0bed16SAtul Gupta tls_hw_unhash(sk); 609dd0bed16SAtul Gupta return err; 610dd0bed16SAtul Gupta } 611dd0bed16SAtul Gupta 612f66de3eeSBoris Pismenny static void build_protos(struct proto prot[TLS_NUM_CONFIG][TLS_NUM_CONFIG], 613f66de3eeSBoris Pismenny struct proto *base) 614c113187dSBoris Pismenny { 615f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_BASE] = *base; 616f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_BASE].setsockopt = tls_setsockopt; 617f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_BASE].getsockopt = tls_getsockopt; 618f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_BASE].close = tls_sk_proto_close; 619c113187dSBoris Pismenny 620f66de3eeSBoris Pismenny prot[TLS_SW][TLS_BASE] = prot[TLS_BASE][TLS_BASE]; 621f66de3eeSBoris Pismenny prot[TLS_SW][TLS_BASE].sendmsg = tls_sw_sendmsg; 622f66de3eeSBoris Pismenny prot[TLS_SW][TLS_BASE].sendpage = tls_sw_sendpage; 623c46234ebSDave Watson 624f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_SW] = prot[TLS_BASE][TLS_BASE]; 625f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_SW].recvmsg = tls_sw_recvmsg; 626924ad65eSJohn Fastabend prot[TLS_BASE][TLS_SW].stream_memory_read = tls_sw_stream_read; 627f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_SW].close = tls_sk_proto_close; 628c46234ebSDave Watson 629f66de3eeSBoris Pismenny prot[TLS_SW][TLS_SW] = prot[TLS_SW][TLS_BASE]; 630f66de3eeSBoris Pismenny prot[TLS_SW][TLS_SW].recvmsg = tls_sw_recvmsg; 631924ad65eSJohn Fastabend prot[TLS_SW][TLS_SW].stream_memory_read = tls_sw_stream_read; 632f66de3eeSBoris Pismenny prot[TLS_SW][TLS_SW].close = tls_sk_proto_close; 633dd0bed16SAtul Gupta 634e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE 635e8f69799SIlya Lesokhin prot[TLS_HW][TLS_BASE] = prot[TLS_BASE][TLS_BASE]; 636e8f69799SIlya Lesokhin prot[TLS_HW][TLS_BASE].sendmsg = tls_device_sendmsg; 637e8f69799SIlya Lesokhin prot[TLS_HW][TLS_BASE].sendpage = tls_device_sendpage; 638e8f69799SIlya Lesokhin 639e8f69799SIlya Lesokhin prot[TLS_HW][TLS_SW] = prot[TLS_BASE][TLS_SW]; 640e8f69799SIlya Lesokhin prot[TLS_HW][TLS_SW].sendmsg = tls_device_sendmsg; 641e8f69799SIlya Lesokhin prot[TLS_HW][TLS_SW].sendpage = tls_device_sendpage; 6424799ac81SBoris Pismenny 6434799ac81SBoris Pismenny prot[TLS_BASE][TLS_HW] = prot[TLS_BASE][TLS_SW]; 6444799ac81SBoris Pismenny 6454799ac81SBoris Pismenny prot[TLS_SW][TLS_HW] = prot[TLS_SW][TLS_SW]; 6464799ac81SBoris Pismenny 6474799ac81SBoris Pismenny prot[TLS_HW][TLS_HW] = prot[TLS_HW][TLS_SW]; 648e8f69799SIlya Lesokhin #endif 649e8f69799SIlya Lesokhin 650f66de3eeSBoris Pismenny prot[TLS_HW_RECORD][TLS_HW_RECORD] = *base; 651f66de3eeSBoris Pismenny prot[TLS_HW_RECORD][TLS_HW_RECORD].hash = tls_hw_hash; 652f66de3eeSBoris Pismenny prot[TLS_HW_RECORD][TLS_HW_RECORD].unhash = tls_hw_unhash; 653f66de3eeSBoris Pismenny prot[TLS_HW_RECORD][TLS_HW_RECORD].close = tls_sk_proto_close; 654c113187dSBoris Pismenny } 655c113187dSBoris Pismenny 6563c4d7559SDave Watson static int tls_init(struct sock *sk) 6573c4d7559SDave Watson { 658c113187dSBoris Pismenny int ip_ver = sk->sk_family == AF_INET6 ? TLSV6 : TLSV4; 6593c4d7559SDave Watson struct tls_context *ctx; 6603c4d7559SDave Watson int rc = 0; 6613c4d7559SDave Watson 662dd0bed16SAtul Gupta if (tls_hw_prot(sk)) 663dd0bed16SAtul Gupta goto out; 664dd0bed16SAtul Gupta 665d91c3e17SIlya Lesokhin /* The TLS ulp is currently supported only for TCP sockets 666d91c3e17SIlya Lesokhin * in ESTABLISHED state. 667d91c3e17SIlya Lesokhin * Supporting sockets in LISTEN state will require us 668d91c3e17SIlya Lesokhin * to modify the accept implementation to clone rather then 669d91c3e17SIlya Lesokhin * share the ulp context. 670d91c3e17SIlya Lesokhin */ 671d91c3e17SIlya Lesokhin if (sk->sk_state != TCP_ESTABLISHED) 672d91c3e17SIlya Lesokhin return -ENOTSUPP; 673d91c3e17SIlya Lesokhin 6743c4d7559SDave Watson /* allocate tls context */ 675dd0bed16SAtul Gupta ctx = create_ctx(sk); 6763c4d7559SDave Watson if (!ctx) { 6773c4d7559SDave Watson rc = -ENOMEM; 6783c4d7559SDave Watson goto out; 6793c4d7559SDave Watson } 6803c4d7559SDave Watson ctx->setsockopt = sk->sk_prot->setsockopt; 6813c4d7559SDave Watson ctx->getsockopt = sk->sk_prot->getsockopt; 682ff45d820SIlya Lesokhin ctx->sk_proto_close = sk->sk_prot->close; 6836d88207fSIlya Lesokhin 684c113187dSBoris Pismenny /* Build IPv6 TLS whenever the address of tcpv6 _prot changes */ 685c113187dSBoris Pismenny if (ip_ver == TLSV6 && 686c113187dSBoris Pismenny unlikely(sk->sk_prot != smp_load_acquire(&saved_tcpv6_prot))) { 687c113187dSBoris Pismenny mutex_lock(&tcpv6_prot_mutex); 688c113187dSBoris Pismenny if (likely(sk->sk_prot != saved_tcpv6_prot)) { 689c113187dSBoris Pismenny build_protos(tls_prots[TLSV6], sk->sk_prot); 690c113187dSBoris Pismenny smp_store_release(&saved_tcpv6_prot, sk->sk_prot); 691c113187dSBoris Pismenny } 692c113187dSBoris Pismenny mutex_unlock(&tcpv6_prot_mutex); 693c113187dSBoris Pismenny } 694c113187dSBoris Pismenny 69528cb6f1eSJohn Fastabend if (ip_ver == TLSV4 && 69628cb6f1eSJohn Fastabend unlikely(sk->sk_prot != smp_load_acquire(&saved_tcpv4_prot))) { 69728cb6f1eSJohn Fastabend mutex_lock(&tcpv4_prot_mutex); 69828cb6f1eSJohn Fastabend if (likely(sk->sk_prot != saved_tcpv4_prot)) { 69928cb6f1eSJohn Fastabend build_protos(tls_prots[TLSV4], sk->sk_prot); 70028cb6f1eSJohn Fastabend smp_store_release(&saved_tcpv4_prot, sk->sk_prot); 70128cb6f1eSJohn Fastabend } 70228cb6f1eSJohn Fastabend mutex_unlock(&tcpv4_prot_mutex); 70328cb6f1eSJohn Fastabend } 70428cb6f1eSJohn Fastabend 705f66de3eeSBoris Pismenny ctx->tx_conf = TLS_BASE; 706f66de3eeSBoris Pismenny ctx->rx_conf = TLS_BASE; 7076d88207fSIlya Lesokhin update_sk_prot(sk, ctx); 7083c4d7559SDave Watson out: 7093c4d7559SDave Watson return rc; 7103c4d7559SDave Watson } 7113c4d7559SDave Watson 712dd0bed16SAtul Gupta void tls_register_device(struct tls_device *device) 713dd0bed16SAtul Gupta { 714dd0bed16SAtul Gupta mutex_lock(&device_mutex); 715dd0bed16SAtul Gupta list_add_tail(&device->dev_list, &device_list); 716dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 717dd0bed16SAtul Gupta } 718dd0bed16SAtul Gupta EXPORT_SYMBOL(tls_register_device); 719dd0bed16SAtul Gupta 720dd0bed16SAtul Gupta void tls_unregister_device(struct tls_device *device) 721dd0bed16SAtul Gupta { 722dd0bed16SAtul Gupta mutex_lock(&device_mutex); 723dd0bed16SAtul Gupta list_del(&device->dev_list); 724dd0bed16SAtul Gupta mutex_unlock(&device_mutex); 725dd0bed16SAtul Gupta } 726dd0bed16SAtul Gupta EXPORT_SYMBOL(tls_unregister_device); 727dd0bed16SAtul Gupta 7283c4d7559SDave Watson static struct tcp_ulp_ops tcp_tls_ulp_ops __read_mostly = { 7293c4d7559SDave Watson .name = "tls", 7303c4d7559SDave Watson .owner = THIS_MODULE, 7313c4d7559SDave Watson .init = tls_init, 7323c4d7559SDave Watson }; 7333c4d7559SDave Watson 7343c4d7559SDave Watson static int __init tls_register(void) 7353c4d7559SDave Watson { 736c46234ebSDave Watson tls_sw_proto_ops = inet_stream_ops; 737c46234ebSDave Watson tls_sw_proto_ops.splice_read = tls_sw_splice_read; 738c46234ebSDave Watson 739e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE 740e8f69799SIlya Lesokhin tls_device_init(); 741e8f69799SIlya Lesokhin #endif 7423c4d7559SDave Watson tcp_register_ulp(&tcp_tls_ulp_ops); 7433c4d7559SDave Watson 7443c4d7559SDave Watson return 0; 7453c4d7559SDave Watson } 7463c4d7559SDave Watson 7473c4d7559SDave Watson static void __exit tls_unregister(void) 7483c4d7559SDave Watson { 7493c4d7559SDave Watson tcp_unregister_ulp(&tcp_tls_ulp_ops); 750e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE 751e8f69799SIlya Lesokhin tls_device_cleanup(); 752e8f69799SIlya Lesokhin #endif 7533c4d7559SDave Watson } 7543c4d7559SDave Watson 7553c4d7559SDave Watson module_init(tls_register); 7563c4d7559SDave Watson module_exit(tls_unregister); 757