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> 4226811cc9SDavide Caratti #include <linux/inet_diag.h> 433c4d7559SDave Watson 44d26b698dSJakub Kicinski #include <net/snmp.h> 453c4d7559SDave Watson #include <net/tls.h> 4625a3cd81SJakub Kicinski #include <net/tls_toe.h> 473c4d7559SDave Watson 4858790314SJakub Kicinski #include "tls.h" 4958790314SJakub Kicinski 503c4d7559SDave Watson MODULE_AUTHOR("Mellanox Technologies"); 513c4d7559SDave Watson MODULE_DESCRIPTION("Transport Layer Security Support"); 523c4d7559SDave Watson MODULE_LICENSE("Dual BSD/GPL"); 53037b0b86SDaniel Borkmann MODULE_ALIAS_TCP_ULP("tls"); 543c4d7559SDave Watson 556d88207fSIlya Lesokhin enum { 56c113187dSBoris Pismenny TLSV4, 57c113187dSBoris Pismenny TLSV6, 58c113187dSBoris Pismenny TLS_NUM_PROTS, 59c113187dSBoris Pismenny }; 606d88207fSIlya Lesokhin 612d2c5ea2STariq Toukan #define CIPHER_SIZE_DESC(cipher) [cipher] = { \ 622d2c5ea2STariq Toukan .iv = cipher ## _IV_SIZE, \ 632d2c5ea2STariq Toukan .key = cipher ## _KEY_SIZE, \ 642d2c5ea2STariq Toukan .salt = cipher ## _SALT_SIZE, \ 652d2c5ea2STariq Toukan .tag = cipher ## _TAG_SIZE, \ 662d2c5ea2STariq Toukan .rec_seq = cipher ## _REC_SEQ_SIZE, \ 672d2c5ea2STariq Toukan } 682d2c5ea2STariq Toukan 692d2c5ea2STariq Toukan const struct tls_cipher_size_desc tls_cipher_size_desc[] = { 702d2c5ea2STariq Toukan CIPHER_SIZE_DESC(TLS_CIPHER_AES_GCM_128), 712d2c5ea2STariq Toukan CIPHER_SIZE_DESC(TLS_CIPHER_AES_GCM_256), 722d2c5ea2STariq Toukan CIPHER_SIZE_DESC(TLS_CIPHER_AES_CCM_128), 732d2c5ea2STariq Toukan CIPHER_SIZE_DESC(TLS_CIPHER_CHACHA20_POLY1305), 742d2c5ea2STariq Toukan CIPHER_SIZE_DESC(TLS_CIPHER_SM4_GCM), 752d2c5ea2STariq Toukan CIPHER_SIZE_DESC(TLS_CIPHER_SM4_CCM), 762d2c5ea2STariq Toukan }; 772d2c5ea2STariq Toukan 78f691a25cSArnd Bergmann static const struct proto *saved_tcpv6_prot; 79c113187dSBoris Pismenny static DEFINE_MUTEX(tcpv6_prot_mutex); 80f691a25cSArnd Bergmann static const struct proto *saved_tcpv4_prot; 8128cb6f1eSJohn Fastabend static DEFINE_MUTEX(tcpv4_prot_mutex); 82f66de3eeSBoris Pismenny static struct proto tls_prots[TLS_NUM_PROTS][TLS_NUM_CONFIG][TLS_NUM_CONFIG]; 83f3911f73SJakub Kicinski static struct proto_ops tls_proto_ops[TLS_NUM_PROTS][TLS_NUM_CONFIG][TLS_NUM_CONFIG]; 8463a6b3feSAtul Gupta static void build_protos(struct proto prot[TLS_NUM_CONFIG][TLS_NUM_CONFIG], 85f13fe3e6SJakub Sitnicki const struct proto *base); 866d88207fSIlya Lesokhin 8708700dabSJakub Kicinski void update_sk_prot(struct sock *sk, struct tls_context *ctx) 886d88207fSIlya Lesokhin { 89c113187dSBoris Pismenny int ip_ver = sk->sk_family == AF_INET6 ? TLSV6 : TLSV4; 90c113187dSBoris Pismenny 91d5bee737SJakub Sitnicki WRITE_ONCE(sk->sk_prot, 92d5bee737SJakub Sitnicki &tls_prots[ip_ver][ctx->tx_conf][ctx->rx_conf]); 93f3911f73SJakub Kicinski WRITE_ONCE(sk->sk_socket->ops, 94f3911f73SJakub Kicinski &tls_proto_ops[ip_ver][ctx->tx_conf][ctx->rx_conf]); 956d88207fSIlya Lesokhin } 963c4d7559SDave Watson 973c4d7559SDave Watson int wait_on_pending_writer(struct sock *sk, long *timeo) 983c4d7559SDave Watson { 993c4d7559SDave Watson int rc = 0; 1003c4d7559SDave Watson DEFINE_WAIT_FUNC(wait, woken_wake_function); 1013c4d7559SDave Watson 1023c4d7559SDave Watson add_wait_queue(sk_sleep(sk), &wait); 1033c4d7559SDave Watson while (1) { 1043c4d7559SDave Watson if (!*timeo) { 1053c4d7559SDave Watson rc = -EAGAIN; 1063c4d7559SDave Watson break; 1073c4d7559SDave Watson } 1083c4d7559SDave Watson 1093c4d7559SDave Watson if (signal_pending(current)) { 1103c4d7559SDave Watson rc = sock_intr_errno(*timeo); 1113c4d7559SDave Watson break; 1123c4d7559SDave Watson } 1133c4d7559SDave Watson 114d0ac89f6SEric Dumazet if (sk_wait_event(sk, timeo, 115d0ac89f6SEric Dumazet !READ_ONCE(sk->sk_write_pending), &wait)) 1163c4d7559SDave Watson break; 1173c4d7559SDave Watson } 1183c4d7559SDave Watson remove_wait_queue(sk_sleep(sk), &wait); 1193c4d7559SDave Watson return rc; 1203c4d7559SDave Watson } 1213c4d7559SDave Watson 1223c4d7559SDave Watson int tls_push_sg(struct sock *sk, 1233c4d7559SDave Watson struct tls_context *ctx, 1243c4d7559SDave Watson struct scatterlist *sg, 1253c4d7559SDave Watson u16 first_offset, 1263c4d7559SDave Watson int flags) 1273c4d7559SDave Watson { 128e117dcfdSDavid Howells struct bio_vec bvec; 129e117dcfdSDavid Howells struct msghdr msg = { 130b848b26cSDavid Howells .msg_flags = MSG_SPLICE_PAGES | flags, 131e117dcfdSDavid Howells }; 1323c4d7559SDave Watson int ret = 0; 1333c4d7559SDave Watson struct page *p; 1343c4d7559SDave Watson size_t size; 1353c4d7559SDave Watson int offset = first_offset; 1363c4d7559SDave Watson 1373c4d7559SDave Watson size = sg->length - offset; 1383c4d7559SDave Watson offset += sg->offset; 1393c4d7559SDave Watson 140e117dcfdSDavid Howells ctx->splicing_pages = true; 1413c4d7559SDave Watson while (1) { 1423c4d7559SDave Watson if (sg_is_last(sg)) 143e117dcfdSDavid Howells msg.msg_flags = flags; 1443c4d7559SDave Watson 1453c4d7559SDave Watson /* is sending application-limited? */ 1463c4d7559SDave Watson tcp_rate_check_app_limited(sk); 1473c4d7559SDave Watson p = sg_page(sg); 1483c4d7559SDave Watson retry: 149e117dcfdSDavid Howells bvec_set_page(&bvec, p, size, offset); 150e117dcfdSDavid Howells iov_iter_bvec(&msg.msg_iter, ITER_SOURCE, &bvec, 1, size); 151e117dcfdSDavid Howells 152e117dcfdSDavid Howells ret = tcp_sendmsg_locked(sk, &msg, size); 1533c4d7559SDave Watson 1543c4d7559SDave Watson if (ret != size) { 1553c4d7559SDave Watson if (ret > 0) { 1563c4d7559SDave Watson offset += ret; 1573c4d7559SDave Watson size -= ret; 1583c4d7559SDave Watson goto retry; 1593c4d7559SDave Watson } 1603c4d7559SDave Watson 1613c4d7559SDave Watson offset -= sg->offset; 1623c4d7559SDave Watson ctx->partially_sent_offset = offset; 1633c4d7559SDave Watson ctx->partially_sent_record = (void *)sg; 164e117dcfdSDavid Howells ctx->splicing_pages = false; 1653c4d7559SDave Watson return ret; 1663c4d7559SDave Watson } 1673c4d7559SDave Watson 1683c4d7559SDave Watson put_page(p); 1693c4d7559SDave Watson sk_mem_uncharge(sk, sg->length); 1703c4d7559SDave Watson sg = sg_next(sg); 1713c4d7559SDave Watson if (!sg) 1723c4d7559SDave Watson break; 1733c4d7559SDave Watson 1743c4d7559SDave Watson offset = sg->offset; 1753c4d7559SDave Watson size = sg->length; 1763c4d7559SDave Watson } 1773c4d7559SDave Watson 178e117dcfdSDavid Howells ctx->splicing_pages = false; 1793c4d7559SDave Watson 1803c4d7559SDave Watson return 0; 1813c4d7559SDave Watson } 1823c4d7559SDave Watson 1833c4d7559SDave Watson static int tls_handle_open_record(struct sock *sk, int flags) 1843c4d7559SDave Watson { 1853c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 1863c4d7559SDave Watson 1873c4d7559SDave Watson if (tls_is_pending_open_record(ctx)) 1883c4d7559SDave Watson return ctx->push_pending_record(sk, flags); 1893c4d7559SDave Watson 1903c4d7559SDave Watson return 0; 1913c4d7559SDave Watson } 1923c4d7559SDave Watson 19358790314SJakub Kicinski int tls_process_cmsg(struct sock *sk, struct msghdr *msg, 1943c4d7559SDave Watson unsigned char *record_type) 1953c4d7559SDave Watson { 1963c4d7559SDave Watson struct cmsghdr *cmsg; 1973c4d7559SDave Watson int rc = -EINVAL; 1983c4d7559SDave Watson 1993c4d7559SDave Watson for_each_cmsghdr(cmsg, msg) { 2003c4d7559SDave Watson if (!CMSG_OK(msg, cmsg)) 2013c4d7559SDave Watson return -EINVAL; 2023c4d7559SDave Watson if (cmsg->cmsg_level != SOL_TLS) 2033c4d7559SDave Watson continue; 2043c4d7559SDave Watson 2053c4d7559SDave Watson switch (cmsg->cmsg_type) { 2063c4d7559SDave Watson case TLS_SET_RECORD_TYPE: 2073c4d7559SDave Watson if (cmsg->cmsg_len < CMSG_LEN(sizeof(*record_type))) 2083c4d7559SDave Watson return -EINVAL; 2093c4d7559SDave Watson 2103c4d7559SDave Watson if (msg->msg_flags & MSG_MORE) 2113c4d7559SDave Watson return -EINVAL; 2123c4d7559SDave Watson 2133c4d7559SDave Watson rc = tls_handle_open_record(sk, msg->msg_flags); 2143c4d7559SDave Watson if (rc) 2153c4d7559SDave Watson return rc; 2163c4d7559SDave Watson 2173c4d7559SDave Watson *record_type = *(unsigned char *)CMSG_DATA(cmsg); 2183c4d7559SDave Watson rc = 0; 2193c4d7559SDave Watson break; 2203c4d7559SDave Watson default: 2213c4d7559SDave Watson return -EINVAL; 2223c4d7559SDave Watson } 2233c4d7559SDave Watson } 2243c4d7559SDave Watson 2253c4d7559SDave Watson return rc; 2263c4d7559SDave Watson } 2273c4d7559SDave Watson 228a42055e8SVakul Garg int tls_push_partial_record(struct sock *sk, struct tls_context *ctx, 229a42055e8SVakul Garg int flags) 2303c4d7559SDave Watson { 2313c4d7559SDave Watson struct scatterlist *sg; 2323c4d7559SDave Watson u16 offset; 2333c4d7559SDave Watson 2343c4d7559SDave Watson sg = ctx->partially_sent_record; 2353c4d7559SDave Watson offset = ctx->partially_sent_offset; 2363c4d7559SDave Watson 2373c4d7559SDave Watson ctx->partially_sent_record = NULL; 2383c4d7559SDave Watson return tls_push_sg(sk, ctx, sg, offset, flags); 2393c4d7559SDave Watson } 2403c4d7559SDave Watson 241c5daa6ccSJakub Kicinski void tls_free_partial_record(struct sock *sk, struct tls_context *ctx) 24235b71a34SJakub Kicinski { 24335b71a34SJakub Kicinski struct scatterlist *sg; 24435b71a34SJakub Kicinski 245c5daa6ccSJakub Kicinski for (sg = ctx->partially_sent_record; sg; sg = sg_next(sg)) { 24635b71a34SJakub Kicinski put_page(sg_page(sg)); 24735b71a34SJakub Kicinski sk_mem_uncharge(sk, sg->length); 24835b71a34SJakub Kicinski } 24935b71a34SJakub Kicinski ctx->partially_sent_record = NULL; 25035b71a34SJakub Kicinski } 25135b71a34SJakub Kicinski 2523c4d7559SDave Watson static void tls_write_space(struct sock *sk) 2533c4d7559SDave Watson { 2543c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 2553c4d7559SDave Watson 256e117dcfdSDavid Howells /* If splicing_pages call lower protocol write space handler 25767db7cd2SJohn Fastabend * to ensure we wake up any waiting operations there. For example 258e117dcfdSDavid Howells * if splicing pages where to call sk_wait_event. 25967db7cd2SJohn Fastabend */ 260e117dcfdSDavid Howells if (ctx->splicing_pages) { 26167db7cd2SJohn Fastabend ctx->sk_write_space(sk); 262c212d2c7SDave Watson return; 26367db7cd2SJohn Fastabend } 264c212d2c7SDave Watson 2657463d3a2SBoris Pismenny #ifdef CONFIG_TLS_DEVICE 2667463d3a2SBoris Pismenny if (ctx->tx_conf == TLS_HW) 2677463d3a2SBoris Pismenny tls_device_write_space(sk, ctx); 2687463d3a2SBoris Pismenny else 2697463d3a2SBoris Pismenny #endif 2707463d3a2SBoris Pismenny tls_sw_write_space(sk, ctx); 2714504ab0eSVakul Garg 2724504ab0eSVakul Garg ctx->sk_write_space(sk); 2733c4d7559SDave Watson } 2743c4d7559SDave Watson 27515a7dea7SJakub Kicinski /** 27615a7dea7SJakub Kicinski * tls_ctx_free() - free TLS ULP context 27715a7dea7SJakub Kicinski * @sk: socket to with @ctx is attached 27815a7dea7SJakub Kicinski * @ctx: TLS context structure 27915a7dea7SJakub Kicinski * 28015a7dea7SJakub Kicinski * Free TLS context. If @sk is %NULL caller guarantees that the socket 28115a7dea7SJakub Kicinski * to which @ctx was attached has no outstanding references. 28215a7dea7SJakub Kicinski */ 28315a7dea7SJakub Kicinski void tls_ctx_free(struct sock *sk, struct tls_context *ctx) 28486029d10SSabrina Dubroca { 28586029d10SSabrina Dubroca if (!ctx) 28686029d10SSabrina Dubroca return; 28786029d10SSabrina Dubroca 28886029d10SSabrina Dubroca memzero_explicit(&ctx->crypto_send, sizeof(ctx->crypto_send)); 28986029d10SSabrina Dubroca memzero_explicit(&ctx->crypto_recv, sizeof(ctx->crypto_recv)); 29079ffe608SJakub Kicinski mutex_destroy(&ctx->tx_lock); 29115a7dea7SJakub Kicinski 29215a7dea7SJakub Kicinski if (sk) 29315a7dea7SJakub Kicinski kfree_rcu(ctx, rcu); 29415a7dea7SJakub Kicinski else 29586029d10SSabrina Dubroca kfree(ctx); 29686029d10SSabrina Dubroca } 29786029d10SSabrina Dubroca 298313ab004SJohn Fastabend static void tls_sk_proto_cleanup(struct sock *sk, 299313ab004SJohn Fastabend struct tls_context *ctx, long timeo) 3003c4d7559SDave Watson { 3019354544cSDirk van der Merwe if (unlikely(sk->sk_write_pending) && 3029354544cSDirk van der Merwe !wait_on_pending_writer(sk, &timeo)) 3033c4d7559SDave Watson tls_handle_open_record(sk, 0); 3043c4d7559SDave Watson 305f66de3eeSBoris Pismenny /* We need these for tls_sw_fallback handling of other packets */ 306f66de3eeSBoris Pismenny if (ctx->tx_conf == TLS_SW) { 307dbe42559SDave Watson kfree(ctx->tx.rec_seq); 308dbe42559SDave Watson kfree(ctx->tx.iv); 309313ab004SJohn Fastabend tls_sw_release_resources_tx(sk); 310b32fd3ccSJakub Kicinski TLS_DEC_STATS(sock_net(sk), LINUX_MIB_TLSCURRTXSW); 31135b71a34SJakub Kicinski } else if (ctx->tx_conf == TLS_HW) { 31235b71a34SJakub Kicinski tls_device_free_resources_tx(sk); 313b32fd3ccSJakub Kicinski TLS_DEC_STATS(sock_net(sk), LINUX_MIB_TLSCURRTXDEVICE); 314f66de3eeSBoris Pismenny } 315f66de3eeSBoris Pismenny 316b32fd3ccSJakub Kicinski if (ctx->rx_conf == TLS_SW) { 317313ab004SJohn Fastabend tls_sw_release_resources_rx(sk); 318b32fd3ccSJakub Kicinski TLS_DEC_STATS(sock_net(sk), LINUX_MIB_TLSCURRRXSW); 319b32fd3ccSJakub Kicinski } else if (ctx->rx_conf == TLS_HW) { 3204799ac81SBoris Pismenny tls_device_offload_cleanup_rx(sk); 321b32fd3ccSJakub Kicinski TLS_DEC_STATS(sock_net(sk), LINUX_MIB_TLSCURRRXDEVICE); 322b32fd3ccSJakub Kicinski } 323e8f69799SIlya Lesokhin } 324e8f69799SIlya Lesokhin 325313ab004SJohn Fastabend static void tls_sk_proto_close(struct sock *sk, long timeout) 326313ab004SJohn Fastabend { 32795fa1454SJohn Fastabend struct inet_connection_sock *icsk = inet_csk(sk); 328313ab004SJohn Fastabend struct tls_context *ctx = tls_get_ctx(sk); 329313ab004SJohn Fastabend long timeo = sock_sndtimeo(sk, 0); 330313ab004SJohn Fastabend bool free_ctx; 331313ab004SJohn Fastabend 332313ab004SJohn Fastabend if (ctx->tx_conf == TLS_SW) 333313ab004SJohn Fastabend tls_sw_cancel_work_tx(ctx); 334313ab004SJohn Fastabend 335313ab004SJohn Fastabend lock_sock(sk); 336313ab004SJohn Fastabend free_ctx = ctx->tx_conf != TLS_HW && ctx->rx_conf != TLS_HW; 337313ab004SJohn Fastabend 338313ab004SJohn Fastabend if (ctx->tx_conf != TLS_BASE || ctx->rx_conf != TLS_BASE) 339313ab004SJohn Fastabend tls_sk_proto_cleanup(sk, ctx, timeo); 340313ab004SJohn Fastabend 34195fa1454SJohn Fastabend write_lock_bh(&sk->sk_callback_lock); 34295fa1454SJohn Fastabend if (free_ctx) 34315a7dea7SJakub Kicinski rcu_assign_pointer(icsk->icsk_ulp_data, NULL); 344d5bee737SJakub Sitnicki WRITE_ONCE(sk->sk_prot, ctx->sk_proto); 345d85f0177SJohn Fastabend if (sk->sk_write_space == tls_write_space) 34657c722e9SJakub Kicinski sk->sk_write_space = ctx->sk_write_space; 34795fa1454SJohn Fastabend write_unlock_bh(&sk->sk_callback_lock); 3483c4d7559SDave Watson release_sock(sk); 349313ab004SJohn Fastabend if (ctx->tx_conf == TLS_SW) 350313ab004SJohn Fastabend tls_sw_free_ctx_tx(ctx); 351313ab004SJohn Fastabend if (ctx->rx_conf == TLS_SW || ctx->rx_conf == TLS_HW) 352313ab004SJohn Fastabend tls_sw_strparser_done(ctx); 353313ab004SJohn Fastabend if (ctx->rx_conf == TLS_SW) 354313ab004SJohn Fastabend tls_sw_free_ctx_rx(ctx); 355be7bbea1SJakub Kicinski ctx->sk_proto->close(sk, timeout); 356313ab004SJohn Fastabend 35798f0a395SEric Dumazet if (free_ctx) 35815a7dea7SJakub Kicinski tls_ctx_free(sk, ctx); 3593c4d7559SDave Watson } 3603c4d7559SDave Watson 361121dca78SJakub Kicinski static __poll_t tls_sk_poll(struct file *file, struct socket *sock, 362121dca78SJakub Kicinski struct poll_table_struct *wait) 363121dca78SJakub Kicinski { 364121dca78SJakub Kicinski struct tls_sw_context_rx *ctx; 365121dca78SJakub Kicinski struct tls_context *tls_ctx; 366121dca78SJakub Kicinski struct sock *sk = sock->sk; 367121dca78SJakub Kicinski struct sk_psock *psock; 368121dca78SJakub Kicinski __poll_t mask = 0; 369121dca78SJakub Kicinski u8 shutdown; 370121dca78SJakub Kicinski int state; 371121dca78SJakub Kicinski 372121dca78SJakub Kicinski mask = tcp_poll(file, sock, wait); 373121dca78SJakub Kicinski 374121dca78SJakub Kicinski state = inet_sk_state_load(sk); 375121dca78SJakub Kicinski shutdown = READ_ONCE(sk->sk_shutdown); 376121dca78SJakub Kicinski if (unlikely(state != TCP_ESTABLISHED || shutdown & RCV_SHUTDOWN)) 377121dca78SJakub Kicinski return mask; 378121dca78SJakub Kicinski 379121dca78SJakub Kicinski tls_ctx = tls_get_ctx(sk); 380121dca78SJakub Kicinski ctx = tls_sw_ctx_rx(tls_ctx); 381121dca78SJakub Kicinski psock = sk_psock_get(sk); 382121dca78SJakub Kicinski 383121dca78SJakub Kicinski if (skb_queue_empty_lockless(&ctx->rx_list) && 384121dca78SJakub Kicinski !tls_strp_msg_ready(ctx) && 385121dca78SJakub Kicinski sk_psock_queue_empty(psock)) 386121dca78SJakub Kicinski mask &= ~(EPOLLIN | EPOLLRDNORM); 387121dca78SJakub Kicinski 388121dca78SJakub Kicinski if (psock) 389121dca78SJakub Kicinski sk_psock_put(sk, psock); 390121dca78SJakub Kicinski 391121dca78SJakub Kicinski return mask; 392121dca78SJakub Kicinski } 393121dca78SJakub Kicinski 394ffa81fa4SYutaro Hayakawa static int do_tls_getsockopt_conf(struct sock *sk, char __user *optval, 395ffa81fa4SYutaro Hayakawa int __user *optlen, int tx) 3963c4d7559SDave Watson { 3973c4d7559SDave Watson int rc = 0; 3983c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 3993c4d7559SDave Watson struct tls_crypto_info *crypto_info; 400ffa81fa4SYutaro Hayakawa struct cipher_context *cctx; 4013c4d7559SDave Watson int len; 4023c4d7559SDave Watson 4033c4d7559SDave Watson if (get_user(len, optlen)) 4043c4d7559SDave Watson return -EFAULT; 4053c4d7559SDave Watson 4063c4d7559SDave Watson if (!optval || (len < sizeof(*crypto_info))) { 4073c4d7559SDave Watson rc = -EINVAL; 4083c4d7559SDave Watson goto out; 4093c4d7559SDave Watson } 4103c4d7559SDave Watson 4113c4d7559SDave Watson if (!ctx) { 4123c4d7559SDave Watson rc = -EBUSY; 4133c4d7559SDave Watson goto out; 4143c4d7559SDave Watson } 4153c4d7559SDave Watson 4163c4d7559SDave Watson /* get user crypto info */ 417ffa81fa4SYutaro Hayakawa if (tx) { 41886029d10SSabrina Dubroca crypto_info = &ctx->crypto_send.info; 419ffa81fa4SYutaro Hayakawa cctx = &ctx->tx; 420ffa81fa4SYutaro Hayakawa } else { 421ffa81fa4SYutaro Hayakawa crypto_info = &ctx->crypto_recv.info; 422ffa81fa4SYutaro Hayakawa cctx = &ctx->rx; 423ffa81fa4SYutaro Hayakawa } 4243c4d7559SDave Watson 4253c4d7559SDave Watson if (!TLS_CRYPTO_INFO_READY(crypto_info)) { 4263c4d7559SDave Watson rc = -EBUSY; 4273c4d7559SDave Watson goto out; 4283c4d7559SDave Watson } 4293c4d7559SDave Watson 4305a3b886cSMatthias Rosenfelder if (len == sizeof(*crypto_info)) { 431ac55cd61SDan Carpenter if (copy_to_user(optval, crypto_info, sizeof(*crypto_info))) 432ac55cd61SDan Carpenter rc = -EFAULT; 4333c4d7559SDave Watson goto out; 4343c4d7559SDave Watson } 4353c4d7559SDave Watson 4363c4d7559SDave Watson switch (crypto_info->cipher_type) { 4373c4d7559SDave Watson case TLS_CIPHER_AES_GCM_128: { 4383c4d7559SDave Watson struct tls12_crypto_info_aes_gcm_128 * 4393c4d7559SDave Watson crypto_info_aes_gcm_128 = 4403c4d7559SDave Watson container_of(crypto_info, 4413c4d7559SDave Watson struct tls12_crypto_info_aes_gcm_128, 4423c4d7559SDave Watson info); 4433c4d7559SDave Watson 4443c4d7559SDave Watson if (len != sizeof(*crypto_info_aes_gcm_128)) { 4453c4d7559SDave Watson rc = -EINVAL; 4463c4d7559SDave Watson goto out; 4473c4d7559SDave Watson } 448a1dfa681SBoris Pismenny memcpy(crypto_info_aes_gcm_128->iv, 449ffa81fa4SYutaro Hayakawa cctx->iv + TLS_CIPHER_AES_GCM_128_SALT_SIZE, 4503c4d7559SDave Watson TLS_CIPHER_AES_GCM_128_IV_SIZE); 451ffa81fa4SYutaro Hayakawa memcpy(crypto_info_aes_gcm_128->rec_seq, cctx->rec_seq, 452c410c196SBoris Pismenny TLS_CIPHER_AES_GCM_128_REC_SEQ_SIZE); 453ac55cd61SDan Carpenter if (copy_to_user(optval, 4543c4d7559SDave Watson crypto_info_aes_gcm_128, 455ac55cd61SDan Carpenter sizeof(*crypto_info_aes_gcm_128))) 456ac55cd61SDan Carpenter rc = -EFAULT; 4573c4d7559SDave Watson break; 4583c4d7559SDave Watson } 459fb99bce7SDave Watson case TLS_CIPHER_AES_GCM_256: { 460fb99bce7SDave Watson struct tls12_crypto_info_aes_gcm_256 * 461fb99bce7SDave Watson crypto_info_aes_gcm_256 = 462fb99bce7SDave Watson container_of(crypto_info, 463fb99bce7SDave Watson struct tls12_crypto_info_aes_gcm_256, 464fb99bce7SDave Watson info); 465fb99bce7SDave Watson 466fb99bce7SDave Watson if (len != sizeof(*crypto_info_aes_gcm_256)) { 467fb99bce7SDave Watson rc = -EINVAL; 468fb99bce7SDave Watson goto out; 469fb99bce7SDave Watson } 470fb99bce7SDave Watson memcpy(crypto_info_aes_gcm_256->iv, 471ffa81fa4SYutaro Hayakawa cctx->iv + TLS_CIPHER_AES_GCM_256_SALT_SIZE, 472fb99bce7SDave Watson TLS_CIPHER_AES_GCM_256_IV_SIZE); 473ffa81fa4SYutaro Hayakawa memcpy(crypto_info_aes_gcm_256->rec_seq, cctx->rec_seq, 474fb99bce7SDave Watson TLS_CIPHER_AES_GCM_256_REC_SEQ_SIZE); 475fb99bce7SDave Watson if (copy_to_user(optval, 476fb99bce7SDave Watson crypto_info_aes_gcm_256, 477fb99bce7SDave Watson sizeof(*crypto_info_aes_gcm_256))) 478fb99bce7SDave Watson rc = -EFAULT; 479fb99bce7SDave Watson break; 480fb99bce7SDave Watson } 4813fb59a5dSTianjia Zhang case TLS_CIPHER_AES_CCM_128: { 4823fb59a5dSTianjia Zhang struct tls12_crypto_info_aes_ccm_128 *aes_ccm_128 = 4833fb59a5dSTianjia Zhang container_of(crypto_info, 4843fb59a5dSTianjia Zhang struct tls12_crypto_info_aes_ccm_128, info); 4853fb59a5dSTianjia Zhang 4863fb59a5dSTianjia Zhang if (len != sizeof(*aes_ccm_128)) { 4873fb59a5dSTianjia Zhang rc = -EINVAL; 4883fb59a5dSTianjia Zhang goto out; 4893fb59a5dSTianjia Zhang } 4903fb59a5dSTianjia Zhang memcpy(aes_ccm_128->iv, 4913fb59a5dSTianjia Zhang cctx->iv + TLS_CIPHER_AES_CCM_128_SALT_SIZE, 4923fb59a5dSTianjia Zhang TLS_CIPHER_AES_CCM_128_IV_SIZE); 4933fb59a5dSTianjia Zhang memcpy(aes_ccm_128->rec_seq, cctx->rec_seq, 4943fb59a5dSTianjia Zhang TLS_CIPHER_AES_CCM_128_REC_SEQ_SIZE); 4953fb59a5dSTianjia Zhang if (copy_to_user(optval, aes_ccm_128, sizeof(*aes_ccm_128))) 4963fb59a5dSTianjia Zhang rc = -EFAULT; 4973fb59a5dSTianjia Zhang break; 4983fb59a5dSTianjia Zhang } 4993fb59a5dSTianjia Zhang case TLS_CIPHER_CHACHA20_POLY1305: { 5003fb59a5dSTianjia Zhang struct tls12_crypto_info_chacha20_poly1305 *chacha20_poly1305 = 5013fb59a5dSTianjia Zhang container_of(crypto_info, 5023fb59a5dSTianjia Zhang struct tls12_crypto_info_chacha20_poly1305, 5033fb59a5dSTianjia Zhang info); 5043fb59a5dSTianjia Zhang 5053fb59a5dSTianjia Zhang if (len != sizeof(*chacha20_poly1305)) { 5063fb59a5dSTianjia Zhang rc = -EINVAL; 5073fb59a5dSTianjia Zhang goto out; 5083fb59a5dSTianjia Zhang } 5093fb59a5dSTianjia Zhang memcpy(chacha20_poly1305->iv, 5103fb59a5dSTianjia Zhang cctx->iv + TLS_CIPHER_CHACHA20_POLY1305_SALT_SIZE, 5113fb59a5dSTianjia Zhang TLS_CIPHER_CHACHA20_POLY1305_IV_SIZE); 5123fb59a5dSTianjia Zhang memcpy(chacha20_poly1305->rec_seq, cctx->rec_seq, 5133fb59a5dSTianjia Zhang TLS_CIPHER_CHACHA20_POLY1305_REC_SEQ_SIZE); 5143fb59a5dSTianjia Zhang if (copy_to_user(optval, chacha20_poly1305, 5153fb59a5dSTianjia Zhang sizeof(*chacha20_poly1305))) 5163fb59a5dSTianjia Zhang rc = -EFAULT; 5173fb59a5dSTianjia Zhang break; 5183fb59a5dSTianjia Zhang } 519227b9644STianjia Zhang case TLS_CIPHER_SM4_GCM: { 520227b9644STianjia Zhang struct tls12_crypto_info_sm4_gcm *sm4_gcm_info = 521227b9644STianjia Zhang container_of(crypto_info, 522227b9644STianjia Zhang struct tls12_crypto_info_sm4_gcm, info); 523227b9644STianjia Zhang 524227b9644STianjia Zhang if (len != sizeof(*sm4_gcm_info)) { 525227b9644STianjia Zhang rc = -EINVAL; 526227b9644STianjia Zhang goto out; 527227b9644STianjia Zhang } 528227b9644STianjia Zhang memcpy(sm4_gcm_info->iv, 529227b9644STianjia Zhang cctx->iv + TLS_CIPHER_SM4_GCM_SALT_SIZE, 530227b9644STianjia Zhang TLS_CIPHER_SM4_GCM_IV_SIZE); 531227b9644STianjia Zhang memcpy(sm4_gcm_info->rec_seq, cctx->rec_seq, 532227b9644STianjia Zhang TLS_CIPHER_SM4_GCM_REC_SEQ_SIZE); 533227b9644STianjia Zhang if (copy_to_user(optval, sm4_gcm_info, sizeof(*sm4_gcm_info))) 534227b9644STianjia Zhang rc = -EFAULT; 535227b9644STianjia Zhang break; 536227b9644STianjia Zhang } 537227b9644STianjia Zhang case TLS_CIPHER_SM4_CCM: { 538227b9644STianjia Zhang struct tls12_crypto_info_sm4_ccm *sm4_ccm_info = 539227b9644STianjia Zhang container_of(crypto_info, 540227b9644STianjia Zhang struct tls12_crypto_info_sm4_ccm, info); 541227b9644STianjia Zhang 542227b9644STianjia Zhang if (len != sizeof(*sm4_ccm_info)) { 543227b9644STianjia Zhang rc = -EINVAL; 544227b9644STianjia Zhang goto out; 545227b9644STianjia Zhang } 546227b9644STianjia Zhang memcpy(sm4_ccm_info->iv, 547227b9644STianjia Zhang cctx->iv + TLS_CIPHER_SM4_CCM_SALT_SIZE, 548227b9644STianjia Zhang TLS_CIPHER_SM4_CCM_IV_SIZE); 549227b9644STianjia Zhang memcpy(sm4_ccm_info->rec_seq, cctx->rec_seq, 550227b9644STianjia Zhang TLS_CIPHER_SM4_CCM_REC_SEQ_SIZE); 551227b9644STianjia Zhang if (copy_to_user(optval, sm4_ccm_info, sizeof(*sm4_ccm_info))) 552227b9644STianjia Zhang rc = -EFAULT; 553227b9644STianjia Zhang break; 554227b9644STianjia Zhang } 55562e56ef5STaehee Yoo case TLS_CIPHER_ARIA_GCM_128: { 55662e56ef5STaehee Yoo struct tls12_crypto_info_aria_gcm_128 * 55762e56ef5STaehee Yoo crypto_info_aria_gcm_128 = 55862e56ef5STaehee Yoo container_of(crypto_info, 55962e56ef5STaehee Yoo struct tls12_crypto_info_aria_gcm_128, 56062e56ef5STaehee Yoo info); 56162e56ef5STaehee Yoo 56262e56ef5STaehee Yoo if (len != sizeof(*crypto_info_aria_gcm_128)) { 56362e56ef5STaehee Yoo rc = -EINVAL; 56462e56ef5STaehee Yoo goto out; 56562e56ef5STaehee Yoo } 56662e56ef5STaehee Yoo memcpy(crypto_info_aria_gcm_128->iv, 56762e56ef5STaehee Yoo cctx->iv + TLS_CIPHER_ARIA_GCM_128_SALT_SIZE, 56862e56ef5STaehee Yoo TLS_CIPHER_ARIA_GCM_128_IV_SIZE); 56962e56ef5STaehee Yoo memcpy(crypto_info_aria_gcm_128->rec_seq, cctx->rec_seq, 57062e56ef5STaehee Yoo TLS_CIPHER_ARIA_GCM_128_REC_SEQ_SIZE); 57162e56ef5STaehee Yoo if (copy_to_user(optval, 57262e56ef5STaehee Yoo crypto_info_aria_gcm_128, 57362e56ef5STaehee Yoo sizeof(*crypto_info_aria_gcm_128))) 57462e56ef5STaehee Yoo rc = -EFAULT; 57562e56ef5STaehee Yoo break; 57662e56ef5STaehee Yoo } 57762e56ef5STaehee Yoo case TLS_CIPHER_ARIA_GCM_256: { 57862e56ef5STaehee Yoo struct tls12_crypto_info_aria_gcm_256 * 57962e56ef5STaehee Yoo crypto_info_aria_gcm_256 = 58062e56ef5STaehee Yoo container_of(crypto_info, 58162e56ef5STaehee Yoo struct tls12_crypto_info_aria_gcm_256, 58262e56ef5STaehee Yoo info); 58362e56ef5STaehee Yoo 58462e56ef5STaehee Yoo if (len != sizeof(*crypto_info_aria_gcm_256)) { 58562e56ef5STaehee Yoo rc = -EINVAL; 58662e56ef5STaehee Yoo goto out; 58762e56ef5STaehee Yoo } 58862e56ef5STaehee Yoo memcpy(crypto_info_aria_gcm_256->iv, 58962e56ef5STaehee Yoo cctx->iv + TLS_CIPHER_ARIA_GCM_256_SALT_SIZE, 59062e56ef5STaehee Yoo TLS_CIPHER_ARIA_GCM_256_IV_SIZE); 59162e56ef5STaehee Yoo memcpy(crypto_info_aria_gcm_256->rec_seq, cctx->rec_seq, 59262e56ef5STaehee Yoo TLS_CIPHER_ARIA_GCM_256_REC_SEQ_SIZE); 59362e56ef5STaehee Yoo if (copy_to_user(optval, 59462e56ef5STaehee Yoo crypto_info_aria_gcm_256, 59562e56ef5STaehee Yoo sizeof(*crypto_info_aria_gcm_256))) 59662e56ef5STaehee Yoo rc = -EFAULT; 59762e56ef5STaehee Yoo break; 59862e56ef5STaehee Yoo } 5993c4d7559SDave Watson default: 6003c4d7559SDave Watson rc = -EINVAL; 6013c4d7559SDave Watson } 6023c4d7559SDave Watson 6033c4d7559SDave Watson out: 6043c4d7559SDave Watson return rc; 6053c4d7559SDave Watson } 6063c4d7559SDave Watson 607c1318b39SBoris Pismenny static int do_tls_getsockopt_tx_zc(struct sock *sk, char __user *optval, 608c1318b39SBoris Pismenny int __user *optlen) 609c1318b39SBoris Pismenny { 610c1318b39SBoris Pismenny struct tls_context *ctx = tls_get_ctx(sk); 611c1318b39SBoris Pismenny unsigned int value; 612c1318b39SBoris Pismenny int len; 613c1318b39SBoris Pismenny 614c1318b39SBoris Pismenny if (get_user(len, optlen)) 615c1318b39SBoris Pismenny return -EFAULT; 616c1318b39SBoris Pismenny 617c1318b39SBoris Pismenny if (len != sizeof(value)) 618c1318b39SBoris Pismenny return -EINVAL; 619c1318b39SBoris Pismenny 620c1318b39SBoris Pismenny value = ctx->zerocopy_sendfile; 621c1318b39SBoris Pismenny if (copy_to_user(optval, &value, sizeof(value))) 622c1318b39SBoris Pismenny return -EFAULT; 623c1318b39SBoris Pismenny 624c1318b39SBoris Pismenny return 0; 625c1318b39SBoris Pismenny } 626c1318b39SBoris Pismenny 62788527790SJakub Kicinski static int do_tls_getsockopt_no_pad(struct sock *sk, char __user *optval, 62888527790SJakub Kicinski int __user *optlen) 62988527790SJakub Kicinski { 63088527790SJakub Kicinski struct tls_context *ctx = tls_get_ctx(sk); 63157128e98SJakub Kicinski int value, len; 63288527790SJakub Kicinski 63388527790SJakub Kicinski if (ctx->prot_info.version != TLS_1_3_VERSION) 63488527790SJakub Kicinski return -EINVAL; 63588527790SJakub Kicinski 63688527790SJakub Kicinski if (get_user(len, optlen)) 63788527790SJakub Kicinski return -EFAULT; 63888527790SJakub Kicinski if (len < sizeof(value)) 63988527790SJakub Kicinski return -EINVAL; 64088527790SJakub Kicinski 64157128e98SJakub Kicinski value = -EINVAL; 64288527790SJakub Kicinski if (ctx->rx_conf == TLS_SW || ctx->rx_conf == TLS_HW) 64388527790SJakub Kicinski value = ctx->rx_no_pad; 64457128e98SJakub Kicinski if (value < 0) 64557128e98SJakub Kicinski return value; 64688527790SJakub Kicinski 64788527790SJakub Kicinski if (put_user(sizeof(value), optlen)) 64888527790SJakub Kicinski return -EFAULT; 64988527790SJakub Kicinski if (copy_to_user(optval, &value, sizeof(value))) 65088527790SJakub Kicinski return -EFAULT; 65188527790SJakub Kicinski 65288527790SJakub Kicinski return 0; 65388527790SJakub Kicinski } 65488527790SJakub Kicinski 6553c4d7559SDave Watson static int do_tls_getsockopt(struct sock *sk, int optname, 6563c4d7559SDave Watson char __user *optval, int __user *optlen) 6573c4d7559SDave Watson { 6583c4d7559SDave Watson int rc = 0; 6593c4d7559SDave Watson 66049c47cc2SHangyu Hua lock_sock(sk); 66149c47cc2SHangyu Hua 6623c4d7559SDave Watson switch (optname) { 6633c4d7559SDave Watson case TLS_TX: 664ffa81fa4SYutaro Hayakawa case TLS_RX: 665ffa81fa4SYutaro Hayakawa rc = do_tls_getsockopt_conf(sk, optval, optlen, 666ffa81fa4SYutaro Hayakawa optname == TLS_TX); 6673c4d7559SDave Watson break; 668b489a6e5SMaxim Mikityanskiy case TLS_TX_ZEROCOPY_RO: 669c1318b39SBoris Pismenny rc = do_tls_getsockopt_tx_zc(sk, optval, optlen); 670c1318b39SBoris Pismenny break; 67188527790SJakub Kicinski case TLS_RX_EXPECT_NO_PAD: 67288527790SJakub Kicinski rc = do_tls_getsockopt_no_pad(sk, optval, optlen); 67388527790SJakub Kicinski break; 6743c4d7559SDave Watson default: 6753c4d7559SDave Watson rc = -ENOPROTOOPT; 6763c4d7559SDave Watson break; 6773c4d7559SDave Watson } 67849c47cc2SHangyu Hua 67949c47cc2SHangyu Hua release_sock(sk); 68049c47cc2SHangyu Hua 6813c4d7559SDave Watson return rc; 6823c4d7559SDave Watson } 6833c4d7559SDave Watson 6843c4d7559SDave Watson static int tls_getsockopt(struct sock *sk, int level, int optname, 6853c4d7559SDave Watson char __user *optval, int __user *optlen) 6863c4d7559SDave Watson { 6873c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 6883c4d7559SDave Watson 6893c4d7559SDave Watson if (level != SOL_TLS) 690be7bbea1SJakub Kicinski return ctx->sk_proto->getsockopt(sk, level, 691be7bbea1SJakub Kicinski optname, optval, optlen); 6923c4d7559SDave Watson 6933c4d7559SDave Watson return do_tls_getsockopt(sk, optname, optval, optlen); 6943c4d7559SDave Watson } 6953c4d7559SDave Watson 696a7b75c5aSChristoph Hellwig static int do_tls_setsockopt_conf(struct sock *sk, sockptr_t optval, 697c46234ebSDave Watson unsigned int optlen, int tx) 6983c4d7559SDave Watson { 699196c31b4SIlya Lesokhin struct tls_crypto_info *crypto_info; 7004509de14SVakul Garg struct tls_crypto_info *alt_crypto_info; 7013c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 702fb99bce7SDave Watson size_t optsize; 7033c4d7559SDave Watson int rc = 0; 70458371585SDave Watson int conf; 7053c4d7559SDave Watson 7061ddcbfbfSZiyang Xuan if (sockptr_is_null(optval) || (optlen < sizeof(*crypto_info))) 7071ddcbfbfSZiyang Xuan return -EINVAL; 7083c4d7559SDave Watson 7094509de14SVakul Garg if (tx) { 71086029d10SSabrina Dubroca crypto_info = &ctx->crypto_send.info; 7114509de14SVakul Garg alt_crypto_info = &ctx->crypto_recv.info; 7124509de14SVakul Garg } else { 71386029d10SSabrina Dubroca crypto_info = &ctx->crypto_recv.info; 7144509de14SVakul Garg alt_crypto_info = &ctx->crypto_send.info; 7154509de14SVakul Garg } 716c46234ebSDave Watson 717196c31b4SIlya Lesokhin /* Currently we don't support set crypto info more than one time */ 7181ddcbfbfSZiyang Xuan if (TLS_CRYPTO_INFO_READY(crypto_info)) 7191ddcbfbfSZiyang Xuan return -EBUSY; 720196c31b4SIlya Lesokhin 721a7b75c5aSChristoph Hellwig rc = copy_from_sockptr(crypto_info, optval, sizeof(*crypto_info)); 7223c4d7559SDave Watson if (rc) { 7233c4d7559SDave Watson rc = -EFAULT; 724257082e6SBoris Pismenny goto err_crypto_info; 7253c4d7559SDave Watson } 7263c4d7559SDave Watson 7273c4d7559SDave Watson /* check version */ 728130b392cSDave Watson if (crypto_info->version != TLS_1_2_VERSION && 729130b392cSDave Watson crypto_info->version != TLS_1_3_VERSION) { 7304a5cdc60SValentin Vidic rc = -EINVAL; 731196c31b4SIlya Lesokhin goto err_crypto_info; 7323c4d7559SDave Watson } 7333c4d7559SDave Watson 7344509de14SVakul Garg /* Ensure that TLS version and ciphers are same in both directions */ 7354509de14SVakul Garg if (TLS_CRYPTO_INFO_READY(alt_crypto_info)) { 7364509de14SVakul Garg if (alt_crypto_info->version != crypto_info->version || 7374509de14SVakul Garg alt_crypto_info->cipher_type != crypto_info->cipher_type) { 7384509de14SVakul Garg rc = -EINVAL; 7394509de14SVakul Garg goto err_crypto_info; 7404509de14SVakul Garg } 7414509de14SVakul Garg } 7424509de14SVakul Garg 743196c31b4SIlya Lesokhin switch (crypto_info->cipher_type) { 744fb99bce7SDave Watson case TLS_CIPHER_AES_GCM_128: 745f295b3aeSVakul Garg optsize = sizeof(struct tls12_crypto_info_aes_gcm_128); 746f295b3aeSVakul Garg break; 747fb99bce7SDave Watson case TLS_CIPHER_AES_GCM_256: { 748f295b3aeSVakul Garg optsize = sizeof(struct tls12_crypto_info_aes_gcm_256); 749f295b3aeSVakul Garg break; 750f295b3aeSVakul Garg } 751f295b3aeSVakul Garg case TLS_CIPHER_AES_CCM_128: 752f295b3aeSVakul Garg optsize = sizeof(struct tls12_crypto_info_aes_ccm_128); 753f295b3aeSVakul Garg break; 75474ea6106SVadim Fedorenko case TLS_CIPHER_CHACHA20_POLY1305: 75574ea6106SVadim Fedorenko optsize = sizeof(struct tls12_crypto_info_chacha20_poly1305); 75674ea6106SVadim Fedorenko break; 757227b9644STianjia Zhang case TLS_CIPHER_SM4_GCM: 758227b9644STianjia Zhang optsize = sizeof(struct tls12_crypto_info_sm4_gcm); 759227b9644STianjia Zhang break; 760227b9644STianjia Zhang case TLS_CIPHER_SM4_CCM: 761227b9644STianjia Zhang optsize = sizeof(struct tls12_crypto_info_sm4_ccm); 762227b9644STianjia Zhang break; 76362e56ef5STaehee Yoo case TLS_CIPHER_ARIA_GCM_128: 76462e56ef5STaehee Yoo if (crypto_info->version != TLS_1_2_VERSION) { 76562e56ef5STaehee Yoo rc = -EINVAL; 76662e56ef5STaehee Yoo goto err_crypto_info; 76762e56ef5STaehee Yoo } 76862e56ef5STaehee Yoo optsize = sizeof(struct tls12_crypto_info_aria_gcm_128); 76962e56ef5STaehee Yoo break; 77062e56ef5STaehee Yoo case TLS_CIPHER_ARIA_GCM_256: 77162e56ef5STaehee Yoo if (crypto_info->version != TLS_1_2_VERSION) { 77262e56ef5STaehee Yoo rc = -EINVAL; 77362e56ef5STaehee Yoo goto err_crypto_info; 77462e56ef5STaehee Yoo } 77562e56ef5STaehee Yoo optsize = sizeof(struct tls12_crypto_info_aria_gcm_256); 77662e56ef5STaehee Yoo break; 777f295b3aeSVakul Garg default: 778f295b3aeSVakul Garg rc = -EINVAL; 779f295b3aeSVakul Garg goto err_crypto_info; 780f295b3aeSVakul Garg } 781f295b3aeSVakul Garg 782fb99bce7SDave Watson if (optlen != optsize) { 7833c4d7559SDave Watson rc = -EINVAL; 7846db959c8SSabrina Dubroca goto err_crypto_info; 7853c4d7559SDave Watson } 786f295b3aeSVakul Garg 787d3c48151SChristoph Hellwig rc = copy_from_sockptr_offset(crypto_info + 1, optval, 788d3c48151SChristoph Hellwig sizeof(*crypto_info), 789196c31b4SIlya Lesokhin optlen - sizeof(*crypto_info)); 7903c4d7559SDave Watson if (rc) { 7913c4d7559SDave Watson rc = -EFAULT; 7923c4d7559SDave Watson goto err_crypto_info; 7933c4d7559SDave Watson } 7943c4d7559SDave Watson 795c46234ebSDave Watson if (tx) { 796e8f69799SIlya Lesokhin rc = tls_set_device_offload(sk, ctx); 797e8f69799SIlya Lesokhin conf = TLS_HW; 798b32fd3ccSJakub Kicinski if (!rc) { 799b32fd3ccSJakub Kicinski TLS_INC_STATS(sock_net(sk), LINUX_MIB_TLSTXDEVICE); 800b32fd3ccSJakub Kicinski TLS_INC_STATS(sock_net(sk), LINUX_MIB_TLSCURRTXDEVICE); 801b32fd3ccSJakub Kicinski } else { 802c46234ebSDave Watson rc = tls_set_sw_offload(sk, ctx, 1); 803318892acSJakub Kicinski if (rc) 804318892acSJakub Kicinski goto err_crypto_info; 805b32fd3ccSJakub Kicinski TLS_INC_STATS(sock_net(sk), LINUX_MIB_TLSTXSW); 806b32fd3ccSJakub Kicinski TLS_INC_STATS(sock_net(sk), LINUX_MIB_TLSCURRTXSW); 807f66de3eeSBoris Pismenny conf = TLS_SW; 808e8f69799SIlya Lesokhin } 809c46234ebSDave Watson } else { 8104799ac81SBoris Pismenny rc = tls_set_device_offload_rx(sk, ctx); 8114799ac81SBoris Pismenny conf = TLS_HW; 812b32fd3ccSJakub Kicinski if (!rc) { 813b32fd3ccSJakub Kicinski TLS_INC_STATS(sock_net(sk), LINUX_MIB_TLSRXDEVICE); 814b32fd3ccSJakub Kicinski TLS_INC_STATS(sock_net(sk), LINUX_MIB_TLSCURRRXDEVICE); 815b32fd3ccSJakub Kicinski } else { 816c46234ebSDave Watson rc = tls_set_sw_offload(sk, ctx, 0); 817318892acSJakub Kicinski if (rc) 818318892acSJakub Kicinski goto err_crypto_info; 819b32fd3ccSJakub Kicinski TLS_INC_STATS(sock_net(sk), LINUX_MIB_TLSRXSW); 820b32fd3ccSJakub Kicinski TLS_INC_STATS(sock_net(sk), LINUX_MIB_TLSCURRRXSW); 821f66de3eeSBoris Pismenny conf = TLS_SW; 822c46234ebSDave Watson } 823313ab004SJohn Fastabend tls_sw_strparser_arm(sk, ctx); 8244799ac81SBoris Pismenny } 825c46234ebSDave Watson 826f66de3eeSBoris Pismenny if (tx) 827f66de3eeSBoris Pismenny ctx->tx_conf = conf; 828f66de3eeSBoris Pismenny else 829f66de3eeSBoris Pismenny ctx->rx_conf = conf; 8306d88207fSIlya Lesokhin update_sk_prot(sk, ctx); 831c46234ebSDave Watson if (tx) { 832ee181e52SIlya Lesokhin ctx->sk_write_space = sk->sk_write_space; 833ee181e52SIlya Lesokhin sk->sk_write_space = tls_write_space; 83484c61fe1SJakub Kicinski } else { 83584c61fe1SJakub Kicinski struct tls_sw_context_rx *rx_ctx = tls_sw_ctx_rx(ctx); 83684c61fe1SJakub Kicinski 83784c61fe1SJakub Kicinski tls_strp_check_rcv(&rx_ctx->strp); 838c46234ebSDave Watson } 8391ddcbfbfSZiyang Xuan return 0; 8403c4d7559SDave Watson 8413c4d7559SDave Watson err_crypto_info: 842c844eb46SSabrina Dubroca memzero_explicit(crypto_info, sizeof(union tls_crypto_context)); 8433c4d7559SDave Watson return rc; 8443c4d7559SDave Watson } 8453c4d7559SDave Watson 846c1318b39SBoris Pismenny static int do_tls_setsockopt_tx_zc(struct sock *sk, sockptr_t optval, 847c1318b39SBoris Pismenny unsigned int optlen) 848c1318b39SBoris Pismenny { 849c1318b39SBoris Pismenny struct tls_context *ctx = tls_get_ctx(sk); 850c1318b39SBoris Pismenny unsigned int value; 851c1318b39SBoris Pismenny 852c1318b39SBoris Pismenny if (sockptr_is_null(optval) || optlen != sizeof(value)) 853c1318b39SBoris Pismenny return -EINVAL; 854c1318b39SBoris Pismenny 855c1318b39SBoris Pismenny if (copy_from_sockptr(&value, optval, sizeof(value))) 856c1318b39SBoris Pismenny return -EFAULT; 857c1318b39SBoris Pismenny 858c1318b39SBoris Pismenny if (value > 1) 859c1318b39SBoris Pismenny return -EINVAL; 860c1318b39SBoris Pismenny 861c1318b39SBoris Pismenny ctx->zerocopy_sendfile = value; 862c1318b39SBoris Pismenny 863c1318b39SBoris Pismenny return 0; 864c1318b39SBoris Pismenny } 865c1318b39SBoris Pismenny 86688527790SJakub Kicinski static int do_tls_setsockopt_no_pad(struct sock *sk, sockptr_t optval, 86788527790SJakub Kicinski unsigned int optlen) 86888527790SJakub Kicinski { 86988527790SJakub Kicinski struct tls_context *ctx = tls_get_ctx(sk); 87088527790SJakub Kicinski u32 val; 87188527790SJakub Kicinski int rc; 87288527790SJakub Kicinski 87388527790SJakub Kicinski if (ctx->prot_info.version != TLS_1_3_VERSION || 87488527790SJakub Kicinski sockptr_is_null(optval) || optlen < sizeof(val)) 87588527790SJakub Kicinski return -EINVAL; 87688527790SJakub Kicinski 87788527790SJakub Kicinski rc = copy_from_sockptr(&val, optval, sizeof(val)); 87888527790SJakub Kicinski if (rc) 87988527790SJakub Kicinski return -EFAULT; 88088527790SJakub Kicinski if (val > 1) 88188527790SJakub Kicinski return -EINVAL; 88288527790SJakub Kicinski rc = check_zeroed_sockptr(optval, sizeof(val), optlen - sizeof(val)); 88388527790SJakub Kicinski if (rc < 1) 88488527790SJakub Kicinski return rc == 0 ? -EINVAL : rc; 88588527790SJakub Kicinski 88688527790SJakub Kicinski lock_sock(sk); 88788527790SJakub Kicinski rc = -EINVAL; 88888527790SJakub Kicinski if (ctx->rx_conf == TLS_SW || ctx->rx_conf == TLS_HW) { 88988527790SJakub Kicinski ctx->rx_no_pad = val; 89088527790SJakub Kicinski tls_update_rx_zc_capable(ctx); 89188527790SJakub Kicinski rc = 0; 89288527790SJakub Kicinski } 89388527790SJakub Kicinski release_sock(sk); 89488527790SJakub Kicinski 89588527790SJakub Kicinski return rc; 89688527790SJakub Kicinski } 89788527790SJakub Kicinski 898a7b75c5aSChristoph Hellwig static int do_tls_setsockopt(struct sock *sk, int optname, sockptr_t optval, 899a7b75c5aSChristoph Hellwig unsigned int optlen) 9003c4d7559SDave Watson { 9013c4d7559SDave Watson int rc = 0; 9023c4d7559SDave Watson 9033c4d7559SDave Watson switch (optname) { 9043c4d7559SDave Watson case TLS_TX: 905c46234ebSDave Watson case TLS_RX: 9063c4d7559SDave Watson lock_sock(sk); 907c46234ebSDave Watson rc = do_tls_setsockopt_conf(sk, optval, optlen, 908c46234ebSDave Watson optname == TLS_TX); 9093c4d7559SDave Watson release_sock(sk); 9103c4d7559SDave Watson break; 911b489a6e5SMaxim Mikityanskiy case TLS_TX_ZEROCOPY_RO: 912c1318b39SBoris Pismenny lock_sock(sk); 913c1318b39SBoris Pismenny rc = do_tls_setsockopt_tx_zc(sk, optval, optlen); 914c1318b39SBoris Pismenny release_sock(sk); 915c1318b39SBoris Pismenny break; 91688527790SJakub Kicinski case TLS_RX_EXPECT_NO_PAD: 91788527790SJakub Kicinski rc = do_tls_setsockopt_no_pad(sk, optval, optlen); 91888527790SJakub Kicinski break; 9193c4d7559SDave Watson default: 9203c4d7559SDave Watson rc = -ENOPROTOOPT; 9213c4d7559SDave Watson break; 9223c4d7559SDave Watson } 9233c4d7559SDave Watson return rc; 9243c4d7559SDave Watson } 9253c4d7559SDave Watson 9263c4d7559SDave Watson static int tls_setsockopt(struct sock *sk, int level, int optname, 927a7b75c5aSChristoph Hellwig sockptr_t optval, unsigned int optlen) 9283c4d7559SDave Watson { 9293c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 9303c4d7559SDave Watson 9313c4d7559SDave Watson if (level != SOL_TLS) 932be7bbea1SJakub Kicinski return ctx->sk_proto->setsockopt(sk, level, optname, optval, 933be7bbea1SJakub Kicinski optlen); 9343c4d7559SDave Watson 9353c4d7559SDave Watson return do_tls_setsockopt(sk, optname, optval, optlen); 9363c4d7559SDave Watson } 9373c4d7559SDave Watson 93808700dabSJakub Kicinski struct tls_context *tls_ctx_create(struct sock *sk) 939dd0bed16SAtul Gupta { 940dd0bed16SAtul Gupta struct inet_connection_sock *icsk = inet_csk(sk); 941dd0bed16SAtul Gupta struct tls_context *ctx; 942dd0bed16SAtul Gupta 943c6ec179aSGanesh Goudar ctx = kzalloc(sizeof(*ctx), GFP_ATOMIC); 944dd0bed16SAtul Gupta if (!ctx) 945dd0bed16SAtul Gupta return NULL; 946dd0bed16SAtul Gupta 94779ffe608SJakub Kicinski mutex_init(&ctx->tx_lock); 94815a7dea7SJakub Kicinski rcu_assign_pointer(icsk->icsk_ulp_data, ctx); 949d5bee737SJakub Sitnicki ctx->sk_proto = READ_ONCE(sk->sk_prot); 950c55dcdd4SMaxim Mikityanskiy ctx->sk = sk; 951dd0bed16SAtul Gupta return ctx; 952dd0bed16SAtul Gupta } 953dd0bed16SAtul Gupta 954f3911f73SJakub Kicinski static void build_proto_ops(struct proto_ops ops[TLS_NUM_CONFIG][TLS_NUM_CONFIG], 955f3911f73SJakub Kicinski const struct proto_ops *base) 956f3911f73SJakub Kicinski { 957f3911f73SJakub Kicinski ops[TLS_BASE][TLS_BASE] = *base; 958f3911f73SJakub Kicinski 959f3911f73SJakub Kicinski ops[TLS_SW ][TLS_BASE] = ops[TLS_BASE][TLS_BASE]; 960df720d28SDavid Howells ops[TLS_SW ][TLS_BASE].splice_eof = tls_sw_splice_eof; 961f3911f73SJakub Kicinski 962f3911f73SJakub Kicinski ops[TLS_BASE][TLS_SW ] = ops[TLS_BASE][TLS_BASE]; 963f3911f73SJakub Kicinski ops[TLS_BASE][TLS_SW ].splice_read = tls_sw_splice_read; 964121dca78SJakub Kicinski ops[TLS_BASE][TLS_SW ].poll = tls_sk_poll; 965*662fbcecSHannes Reinecke ops[TLS_BASE][TLS_SW ].read_sock = tls_sw_read_sock; 966f3911f73SJakub Kicinski 967f3911f73SJakub Kicinski ops[TLS_SW ][TLS_SW ] = ops[TLS_SW ][TLS_BASE]; 968f3911f73SJakub Kicinski ops[TLS_SW ][TLS_SW ].splice_read = tls_sw_splice_read; 969121dca78SJakub Kicinski ops[TLS_SW ][TLS_SW ].poll = tls_sk_poll; 970*662fbcecSHannes Reinecke ops[TLS_SW ][TLS_SW ].read_sock = tls_sw_read_sock; 971f3911f73SJakub Kicinski 972f3911f73SJakub Kicinski #ifdef CONFIG_TLS_DEVICE 973f3911f73SJakub Kicinski ops[TLS_HW ][TLS_BASE] = ops[TLS_BASE][TLS_BASE]; 974f3911f73SJakub Kicinski 975f3911f73SJakub Kicinski ops[TLS_HW ][TLS_SW ] = ops[TLS_BASE][TLS_SW ]; 976f3911f73SJakub Kicinski 977f3911f73SJakub Kicinski ops[TLS_BASE][TLS_HW ] = ops[TLS_BASE][TLS_SW ]; 978f3911f73SJakub Kicinski 979f3911f73SJakub Kicinski ops[TLS_SW ][TLS_HW ] = ops[TLS_SW ][TLS_SW ]; 980f3911f73SJakub Kicinski 981f3911f73SJakub Kicinski ops[TLS_HW ][TLS_HW ] = ops[TLS_HW ][TLS_SW ]; 982f3911f73SJakub Kicinski #endif 983f3911f73SJakub Kicinski #ifdef CONFIG_TLS_TOE 984f3911f73SJakub Kicinski ops[TLS_HW_RECORD][TLS_HW_RECORD] = *base; 985f3911f73SJakub Kicinski #endif 986f3911f73SJakub Kicinski } 987f3911f73SJakub Kicinski 98863a6b3feSAtul Gupta static void tls_build_proto(struct sock *sk) 98963a6b3feSAtul Gupta { 99063a6b3feSAtul Gupta int ip_ver = sk->sk_family == AF_INET6 ? TLSV6 : TLSV4; 9919a893949SWill Deacon struct proto *prot = READ_ONCE(sk->sk_prot); 99263a6b3feSAtul Gupta 99363a6b3feSAtul Gupta /* Build IPv6 TLS whenever the address of tcpv6 _prot changes */ 99463a6b3feSAtul Gupta if (ip_ver == TLSV6 && 9955bb4c45dSJakub Sitnicki unlikely(prot != smp_load_acquire(&saved_tcpv6_prot))) { 99663a6b3feSAtul Gupta mutex_lock(&tcpv6_prot_mutex); 9975bb4c45dSJakub Sitnicki if (likely(prot != saved_tcpv6_prot)) { 9985bb4c45dSJakub Sitnicki build_protos(tls_prots[TLSV6], prot); 999f3911f73SJakub Kicinski build_proto_ops(tls_proto_ops[TLSV6], 1000f3911f73SJakub Kicinski sk->sk_socket->ops); 10015bb4c45dSJakub Sitnicki smp_store_release(&saved_tcpv6_prot, prot); 100263a6b3feSAtul Gupta } 100363a6b3feSAtul Gupta mutex_unlock(&tcpv6_prot_mutex); 100463a6b3feSAtul Gupta } 100563a6b3feSAtul Gupta 100663a6b3feSAtul Gupta if (ip_ver == TLSV4 && 10075bb4c45dSJakub Sitnicki unlikely(prot != smp_load_acquire(&saved_tcpv4_prot))) { 100863a6b3feSAtul Gupta mutex_lock(&tcpv4_prot_mutex); 10095bb4c45dSJakub Sitnicki if (likely(prot != saved_tcpv4_prot)) { 10105bb4c45dSJakub Sitnicki build_protos(tls_prots[TLSV4], prot); 1011f3911f73SJakub Kicinski build_proto_ops(tls_proto_ops[TLSV4], 1012f3911f73SJakub Kicinski sk->sk_socket->ops); 10135bb4c45dSJakub Sitnicki smp_store_release(&saved_tcpv4_prot, prot); 101463a6b3feSAtul Gupta } 101563a6b3feSAtul Gupta mutex_unlock(&tcpv4_prot_mutex); 101663a6b3feSAtul Gupta } 101763a6b3feSAtul Gupta } 101863a6b3feSAtul Gupta 1019f66de3eeSBoris Pismenny static void build_protos(struct proto prot[TLS_NUM_CONFIG][TLS_NUM_CONFIG], 1020f13fe3e6SJakub Sitnicki const struct proto *base) 1021c113187dSBoris Pismenny { 1022f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_BASE] = *base; 1023f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_BASE].setsockopt = tls_setsockopt; 1024f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_BASE].getsockopt = tls_getsockopt; 1025f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_BASE].close = tls_sk_proto_close; 1026c113187dSBoris Pismenny 1027f66de3eeSBoris Pismenny prot[TLS_SW][TLS_BASE] = prot[TLS_BASE][TLS_BASE]; 1028f66de3eeSBoris Pismenny prot[TLS_SW][TLS_BASE].sendmsg = tls_sw_sendmsg; 1029df720d28SDavid Howells prot[TLS_SW][TLS_BASE].splice_eof = tls_sw_splice_eof; 1030c46234ebSDave Watson 1031f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_SW] = prot[TLS_BASE][TLS_BASE]; 1032f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_SW].recvmsg = tls_sw_recvmsg; 10337b50ecfcSCong Wang prot[TLS_BASE][TLS_SW].sock_is_readable = tls_sw_sock_is_readable; 1034f66de3eeSBoris Pismenny prot[TLS_BASE][TLS_SW].close = tls_sk_proto_close; 1035c46234ebSDave Watson 1036f66de3eeSBoris Pismenny prot[TLS_SW][TLS_SW] = prot[TLS_SW][TLS_BASE]; 1037f66de3eeSBoris Pismenny prot[TLS_SW][TLS_SW].recvmsg = tls_sw_recvmsg; 10387b50ecfcSCong Wang prot[TLS_SW][TLS_SW].sock_is_readable = tls_sw_sock_is_readable; 1039f66de3eeSBoris Pismenny prot[TLS_SW][TLS_SW].close = tls_sk_proto_close; 1040dd0bed16SAtul Gupta 1041e8f69799SIlya Lesokhin #ifdef CONFIG_TLS_DEVICE 1042e8f69799SIlya Lesokhin prot[TLS_HW][TLS_BASE] = prot[TLS_BASE][TLS_BASE]; 1043e8f69799SIlya Lesokhin prot[TLS_HW][TLS_BASE].sendmsg = tls_device_sendmsg; 1044d4c1e80bSDavid Howells prot[TLS_HW][TLS_BASE].splice_eof = tls_device_splice_eof; 1045e8f69799SIlya Lesokhin 1046e8f69799SIlya Lesokhin prot[TLS_HW][TLS_SW] = prot[TLS_BASE][TLS_SW]; 1047e8f69799SIlya Lesokhin prot[TLS_HW][TLS_SW].sendmsg = tls_device_sendmsg; 1048d4c1e80bSDavid Howells prot[TLS_HW][TLS_SW].splice_eof = tls_device_splice_eof; 10494799ac81SBoris Pismenny 10504799ac81SBoris Pismenny prot[TLS_BASE][TLS_HW] = prot[TLS_BASE][TLS_SW]; 10514799ac81SBoris Pismenny 10524799ac81SBoris Pismenny prot[TLS_SW][TLS_HW] = prot[TLS_SW][TLS_SW]; 10534799ac81SBoris Pismenny 10544799ac81SBoris Pismenny prot[TLS_HW][TLS_HW] = prot[TLS_HW][TLS_SW]; 1055e8f69799SIlya Lesokhin #endif 105653b4414aSJakub Kicinski #ifdef CONFIG_TLS_TOE 1057f66de3eeSBoris Pismenny prot[TLS_HW_RECORD][TLS_HW_RECORD] = *base; 10580eb8745eSJakub Kicinski prot[TLS_HW_RECORD][TLS_HW_RECORD].hash = tls_toe_hash; 10590eb8745eSJakub Kicinski prot[TLS_HW_RECORD][TLS_HW_RECORD].unhash = tls_toe_unhash; 106053b4414aSJakub Kicinski #endif 1061c113187dSBoris Pismenny } 1062c113187dSBoris Pismenny 10633c4d7559SDave Watson static int tls_init(struct sock *sk) 10643c4d7559SDave Watson { 10653c4d7559SDave Watson struct tls_context *ctx; 10663c4d7559SDave Watson int rc = 0; 10673c4d7559SDave Watson 106816bed0e6SJakub Kicinski tls_build_proto(sk); 106916bed0e6SJakub Kicinski 107053b4414aSJakub Kicinski #ifdef CONFIG_TLS_TOE 10710eb8745eSJakub Kicinski if (tls_toe_bypass(sk)) 107295fa1454SJohn Fastabend return 0; 107353b4414aSJakub Kicinski #endif 1074dd0bed16SAtul Gupta 1075d91c3e17SIlya Lesokhin /* The TLS ulp is currently supported only for TCP sockets 1076d91c3e17SIlya Lesokhin * in ESTABLISHED state. 1077d91c3e17SIlya Lesokhin * Supporting sockets in LISTEN state will require us 1078d91c3e17SIlya Lesokhin * to modify the accept implementation to clone rather then 1079d91c3e17SIlya Lesokhin * share the ulp context. 1080d91c3e17SIlya Lesokhin */ 1081d91c3e17SIlya Lesokhin if (sk->sk_state != TCP_ESTABLISHED) 10824a5cdc60SValentin Vidic return -ENOTCONN; 1083d91c3e17SIlya Lesokhin 10843c4d7559SDave Watson /* allocate tls context */ 108595fa1454SJohn Fastabend write_lock_bh(&sk->sk_callback_lock); 108608700dabSJakub Kicinski ctx = tls_ctx_create(sk); 10873c4d7559SDave Watson if (!ctx) { 10883c4d7559SDave Watson rc = -ENOMEM; 10893c4d7559SDave Watson goto out; 10903c4d7559SDave Watson } 10916d88207fSIlya Lesokhin 1092f66de3eeSBoris Pismenny ctx->tx_conf = TLS_BASE; 1093f66de3eeSBoris Pismenny ctx->rx_conf = TLS_BASE; 10946d88207fSIlya Lesokhin update_sk_prot(sk, ctx); 10953c4d7559SDave Watson out: 109695fa1454SJohn Fastabend write_unlock_bh(&sk->sk_callback_lock); 10973c4d7559SDave Watson return rc; 10983c4d7559SDave Watson } 10993c4d7559SDave Watson 110033bfe20dSJohn Fastabend static void tls_update(struct sock *sk, struct proto *p, 110133bfe20dSJohn Fastabend void (*write_space)(struct sock *sk)) 110295fa1454SJohn Fastabend { 110395fa1454SJohn Fastabend struct tls_context *ctx; 110495fa1454SJohn Fastabend 1105e34a07c0SJakub Kicinski WARN_ON_ONCE(sk->sk_prot == p); 1106e34a07c0SJakub Kicinski 110795fa1454SJohn Fastabend ctx = tls_get_ctx(sk); 110833bfe20dSJohn Fastabend if (likely(ctx)) { 110933bfe20dSJohn Fastabend ctx->sk_write_space = write_space; 111095fa1454SJohn Fastabend ctx->sk_proto = p; 111133bfe20dSJohn Fastabend } else { 1112b8e202d1SJakub Sitnicki /* Pairs with lockless read in sk_clone_lock(). */ 1113b8e202d1SJakub Sitnicki WRITE_ONCE(sk->sk_prot, p); 111433bfe20dSJohn Fastabend sk->sk_write_space = write_space; 111533bfe20dSJohn Fastabend } 111695fa1454SJohn Fastabend } 111795fa1454SJohn Fastabend 111858790314SJakub Kicinski static u16 tls_user_config(struct tls_context *ctx, bool tx) 111958790314SJakub Kicinski { 112058790314SJakub Kicinski u16 config = tx ? ctx->tx_conf : ctx->rx_conf; 112158790314SJakub Kicinski 112258790314SJakub Kicinski switch (config) { 112358790314SJakub Kicinski case TLS_BASE: 112458790314SJakub Kicinski return TLS_CONF_BASE; 112558790314SJakub Kicinski case TLS_SW: 112658790314SJakub Kicinski return TLS_CONF_SW; 112758790314SJakub Kicinski case TLS_HW: 112858790314SJakub Kicinski return TLS_CONF_HW; 112958790314SJakub Kicinski case TLS_HW_RECORD: 113058790314SJakub Kicinski return TLS_CONF_HW_RECORD; 113158790314SJakub Kicinski } 113258790314SJakub Kicinski return 0; 113358790314SJakub Kicinski } 113458790314SJakub Kicinski 113526811cc9SDavide Caratti static int tls_get_info(const struct sock *sk, struct sk_buff *skb) 113626811cc9SDavide Caratti { 113726811cc9SDavide Caratti u16 version, cipher_type; 113826811cc9SDavide Caratti struct tls_context *ctx; 113926811cc9SDavide Caratti struct nlattr *start; 114026811cc9SDavide Caratti int err; 114126811cc9SDavide Caratti 114226811cc9SDavide Caratti start = nla_nest_start_noflag(skb, INET_ULP_INFO_TLS); 114326811cc9SDavide Caratti if (!start) 114426811cc9SDavide Caratti return -EMSGSIZE; 114526811cc9SDavide Caratti 114626811cc9SDavide Caratti rcu_read_lock(); 114726811cc9SDavide Caratti ctx = rcu_dereference(inet_csk(sk)->icsk_ulp_data); 114826811cc9SDavide Caratti if (!ctx) { 114926811cc9SDavide Caratti err = 0; 115026811cc9SDavide Caratti goto nla_failure; 115126811cc9SDavide Caratti } 115226811cc9SDavide Caratti version = ctx->prot_info.version; 115326811cc9SDavide Caratti if (version) { 115426811cc9SDavide Caratti err = nla_put_u16(skb, TLS_INFO_VERSION, version); 115526811cc9SDavide Caratti if (err) 115626811cc9SDavide Caratti goto nla_failure; 115726811cc9SDavide Caratti } 115826811cc9SDavide Caratti cipher_type = ctx->prot_info.cipher_type; 115926811cc9SDavide Caratti if (cipher_type) { 116026811cc9SDavide Caratti err = nla_put_u16(skb, TLS_INFO_CIPHER, cipher_type); 116126811cc9SDavide Caratti if (err) 116226811cc9SDavide Caratti goto nla_failure; 116326811cc9SDavide Caratti } 116426811cc9SDavide Caratti err = nla_put_u16(skb, TLS_INFO_TXCONF, tls_user_config(ctx, true)); 116526811cc9SDavide Caratti if (err) 116626811cc9SDavide Caratti goto nla_failure; 116726811cc9SDavide Caratti 116826811cc9SDavide Caratti err = nla_put_u16(skb, TLS_INFO_RXCONF, tls_user_config(ctx, false)); 116926811cc9SDavide Caratti if (err) 117026811cc9SDavide Caratti goto nla_failure; 117126811cc9SDavide Caratti 1172c1318b39SBoris Pismenny if (ctx->tx_conf == TLS_HW && ctx->zerocopy_sendfile) { 1173b489a6e5SMaxim Mikityanskiy err = nla_put_flag(skb, TLS_INFO_ZC_RO_TX); 1174c1318b39SBoris Pismenny if (err) 1175c1318b39SBoris Pismenny goto nla_failure; 1176c1318b39SBoris Pismenny } 117788527790SJakub Kicinski if (ctx->rx_no_pad) { 117888527790SJakub Kicinski err = nla_put_flag(skb, TLS_INFO_RX_NO_PAD); 117988527790SJakub Kicinski if (err) 118088527790SJakub Kicinski goto nla_failure; 118188527790SJakub Kicinski } 1182c1318b39SBoris Pismenny 118326811cc9SDavide Caratti rcu_read_unlock(); 118426811cc9SDavide Caratti nla_nest_end(skb, start); 118526811cc9SDavide Caratti return 0; 118626811cc9SDavide Caratti 118726811cc9SDavide Caratti nla_failure: 118826811cc9SDavide Caratti rcu_read_unlock(); 118926811cc9SDavide Caratti nla_nest_cancel(skb, start); 119026811cc9SDavide Caratti return err; 119126811cc9SDavide Caratti } 119226811cc9SDavide Caratti 119326811cc9SDavide Caratti static size_t tls_get_info_size(const struct sock *sk) 119426811cc9SDavide Caratti { 119526811cc9SDavide Caratti size_t size = 0; 119626811cc9SDavide Caratti 119726811cc9SDavide Caratti size += nla_total_size(0) + /* INET_ULP_INFO_TLS */ 119826811cc9SDavide Caratti nla_total_size(sizeof(u16)) + /* TLS_INFO_VERSION */ 119926811cc9SDavide Caratti nla_total_size(sizeof(u16)) + /* TLS_INFO_CIPHER */ 120026811cc9SDavide Caratti nla_total_size(sizeof(u16)) + /* TLS_INFO_RXCONF */ 120126811cc9SDavide Caratti nla_total_size(sizeof(u16)) + /* TLS_INFO_TXCONF */ 1202b489a6e5SMaxim Mikityanskiy nla_total_size(0) + /* TLS_INFO_ZC_RO_TX */ 120388527790SJakub Kicinski nla_total_size(0) + /* TLS_INFO_RX_NO_PAD */ 120426811cc9SDavide Caratti 0; 120526811cc9SDavide Caratti 120626811cc9SDavide Caratti return size; 120726811cc9SDavide Caratti } 120826811cc9SDavide Caratti 1209d26b698dSJakub Kicinski static int __net_init tls_init_net(struct net *net) 1210d26b698dSJakub Kicinski { 1211d26b698dSJakub Kicinski int err; 1212d26b698dSJakub Kicinski 1213d26b698dSJakub Kicinski net->mib.tls_statistics = alloc_percpu(struct linux_tls_mib); 1214d26b698dSJakub Kicinski if (!net->mib.tls_statistics) 1215d26b698dSJakub Kicinski return -ENOMEM; 1216d26b698dSJakub Kicinski 1217d26b698dSJakub Kicinski err = tls_proc_init(net); 1218d26b698dSJakub Kicinski if (err) 1219d26b698dSJakub Kicinski goto err_free_stats; 1220d26b698dSJakub Kicinski 1221d26b698dSJakub Kicinski return 0; 1222d26b698dSJakub Kicinski err_free_stats: 1223d26b698dSJakub Kicinski free_percpu(net->mib.tls_statistics); 1224d26b698dSJakub Kicinski return err; 1225d26b698dSJakub Kicinski } 1226d26b698dSJakub Kicinski 1227d26b698dSJakub Kicinski static void __net_exit tls_exit_net(struct net *net) 1228d26b698dSJakub Kicinski { 1229d26b698dSJakub Kicinski tls_proc_fini(net); 1230d26b698dSJakub Kicinski free_percpu(net->mib.tls_statistics); 1231d26b698dSJakub Kicinski } 1232d26b698dSJakub Kicinski 1233d26b698dSJakub Kicinski static struct pernet_operations tls_proc_ops = { 1234d26b698dSJakub Kicinski .init = tls_init_net, 1235d26b698dSJakub Kicinski .exit = tls_exit_net, 1236d26b698dSJakub Kicinski }; 1237d26b698dSJakub Kicinski 12383c4d7559SDave Watson static struct tcp_ulp_ops tcp_tls_ulp_ops __read_mostly = { 12393c4d7559SDave Watson .name = "tls", 12403c4d7559SDave Watson .owner = THIS_MODULE, 12413c4d7559SDave Watson .init = tls_init, 124295fa1454SJohn Fastabend .update = tls_update, 124326811cc9SDavide Caratti .get_info = tls_get_info, 124426811cc9SDavide Caratti .get_info_size = tls_get_info_size, 12453c4d7559SDave Watson }; 12463c4d7559SDave Watson 12473c4d7559SDave Watson static int __init tls_register(void) 12483c4d7559SDave Watson { 1249d26b698dSJakub Kicinski int err; 1250d26b698dSJakub Kicinski 1251d26b698dSJakub Kicinski err = register_pernet_subsys(&tls_proc_ops); 1252d26b698dSJakub Kicinski if (err) 1253d26b698dSJakub Kicinski return err; 1254d26b698dSJakub Kicinski 125584c61fe1SJakub Kicinski err = tls_strp_dev_init(); 125684c61fe1SJakub Kicinski if (err) 125784c61fe1SJakub Kicinski goto err_pernet; 125884c61fe1SJakub Kicinski 12593d8c51b2STariq Toukan err = tls_device_init(); 126084c61fe1SJakub Kicinski if (err) 126184c61fe1SJakub Kicinski goto err_strp; 12623d8c51b2STariq Toukan 12633c4d7559SDave Watson tcp_register_ulp(&tcp_tls_ulp_ops); 12643c4d7559SDave Watson 12653c4d7559SDave Watson return 0; 126684c61fe1SJakub Kicinski err_strp: 126784c61fe1SJakub Kicinski tls_strp_dev_exit(); 126884c61fe1SJakub Kicinski err_pernet: 126984c61fe1SJakub Kicinski unregister_pernet_subsys(&tls_proc_ops); 127084c61fe1SJakub Kicinski return err; 12713c4d7559SDave Watson } 12723c4d7559SDave Watson 12733c4d7559SDave Watson static void __exit tls_unregister(void) 12743c4d7559SDave Watson { 12753c4d7559SDave Watson tcp_unregister_ulp(&tcp_tls_ulp_ops); 127684c61fe1SJakub Kicinski tls_strp_dev_exit(); 1277e8f69799SIlya Lesokhin tls_device_cleanup(); 1278d26b698dSJakub Kicinski unregister_pernet_subsys(&tls_proc_ops); 12793c4d7559SDave Watson } 12803c4d7559SDave Watson 12813c4d7559SDave Watson module_init(tls_register); 12823c4d7559SDave Watson module_exit(tls_unregister); 1283