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> 413c4d7559SDave Watson 423c4d7559SDave Watson #include <net/tls.h> 433c4d7559SDave Watson 443c4d7559SDave Watson MODULE_AUTHOR("Mellanox Technologies"); 453c4d7559SDave Watson MODULE_DESCRIPTION("Transport Layer Security Support"); 463c4d7559SDave Watson MODULE_LICENSE("Dual BSD/GPL"); 473c4d7559SDave Watson 486d88207fSIlya Lesokhin enum { 496d88207fSIlya Lesokhin TLS_BASE_TX, 506d88207fSIlya Lesokhin TLS_SW_TX, 516d88207fSIlya Lesokhin TLS_NUM_CONFIG, 526d88207fSIlya Lesokhin }; 536d88207fSIlya Lesokhin 546d88207fSIlya Lesokhin static struct proto tls_prots[TLS_NUM_CONFIG]; 556d88207fSIlya Lesokhin 566d88207fSIlya Lesokhin static inline void update_sk_prot(struct sock *sk, struct tls_context *ctx) 576d88207fSIlya Lesokhin { 586d88207fSIlya Lesokhin sk->sk_prot = &tls_prots[ctx->tx_conf]; 596d88207fSIlya Lesokhin } 603c4d7559SDave Watson 613c4d7559SDave Watson int wait_on_pending_writer(struct sock *sk, long *timeo) 623c4d7559SDave Watson { 633c4d7559SDave Watson int rc = 0; 643c4d7559SDave Watson DEFINE_WAIT_FUNC(wait, woken_wake_function); 653c4d7559SDave Watson 663c4d7559SDave Watson add_wait_queue(sk_sleep(sk), &wait); 673c4d7559SDave Watson while (1) { 683c4d7559SDave Watson if (!*timeo) { 693c4d7559SDave Watson rc = -EAGAIN; 703c4d7559SDave Watson break; 713c4d7559SDave Watson } 723c4d7559SDave Watson 733c4d7559SDave Watson if (signal_pending(current)) { 743c4d7559SDave Watson rc = sock_intr_errno(*timeo); 753c4d7559SDave Watson break; 763c4d7559SDave Watson } 773c4d7559SDave Watson 783c4d7559SDave Watson if (sk_wait_event(sk, timeo, !sk->sk_write_pending, &wait)) 793c4d7559SDave Watson break; 803c4d7559SDave Watson } 813c4d7559SDave Watson remove_wait_queue(sk_sleep(sk), &wait); 823c4d7559SDave Watson return rc; 833c4d7559SDave Watson } 843c4d7559SDave Watson 853c4d7559SDave Watson int tls_push_sg(struct sock *sk, 863c4d7559SDave Watson struct tls_context *ctx, 873c4d7559SDave Watson struct scatterlist *sg, 883c4d7559SDave Watson u16 first_offset, 893c4d7559SDave Watson int flags) 903c4d7559SDave Watson { 913c4d7559SDave Watson int sendpage_flags = flags | MSG_SENDPAGE_NOTLAST; 923c4d7559SDave Watson int ret = 0; 933c4d7559SDave Watson struct page *p; 943c4d7559SDave Watson size_t size; 953c4d7559SDave Watson int offset = first_offset; 963c4d7559SDave Watson 973c4d7559SDave Watson size = sg->length - offset; 983c4d7559SDave Watson offset += sg->offset; 993c4d7559SDave Watson 1003c4d7559SDave Watson while (1) { 1013c4d7559SDave Watson if (sg_is_last(sg)) 1023c4d7559SDave Watson sendpage_flags = flags; 1033c4d7559SDave Watson 1043c4d7559SDave Watson /* is sending application-limited? */ 1053c4d7559SDave Watson tcp_rate_check_app_limited(sk); 1063c4d7559SDave Watson p = sg_page(sg); 1073c4d7559SDave Watson retry: 1083c4d7559SDave Watson ret = do_tcp_sendpages(sk, p, offset, size, sendpage_flags); 1093c4d7559SDave Watson 1103c4d7559SDave Watson if (ret != size) { 1113c4d7559SDave Watson if (ret > 0) { 1123c4d7559SDave Watson offset += ret; 1133c4d7559SDave Watson size -= ret; 1143c4d7559SDave Watson goto retry; 1153c4d7559SDave Watson } 1163c4d7559SDave Watson 1173c4d7559SDave Watson offset -= sg->offset; 1183c4d7559SDave Watson ctx->partially_sent_offset = offset; 1193c4d7559SDave Watson ctx->partially_sent_record = (void *)sg; 1203c4d7559SDave Watson return ret; 1213c4d7559SDave Watson } 1223c4d7559SDave Watson 1233c4d7559SDave Watson put_page(p); 1243c4d7559SDave Watson sk_mem_uncharge(sk, sg->length); 1253c4d7559SDave Watson sg = sg_next(sg); 1263c4d7559SDave Watson if (!sg) 1273c4d7559SDave Watson break; 1283c4d7559SDave Watson 1293c4d7559SDave Watson offset = sg->offset; 1303c4d7559SDave Watson size = sg->length; 1313c4d7559SDave Watson } 1323c4d7559SDave Watson 1333c4d7559SDave Watson clear_bit(TLS_PENDING_CLOSED_RECORD, &ctx->flags); 1343c4d7559SDave Watson 1353c4d7559SDave Watson return 0; 1363c4d7559SDave Watson } 1373c4d7559SDave Watson 1383c4d7559SDave Watson static int tls_handle_open_record(struct sock *sk, int flags) 1393c4d7559SDave Watson { 1403c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 1413c4d7559SDave Watson 1423c4d7559SDave Watson if (tls_is_pending_open_record(ctx)) 1433c4d7559SDave Watson return ctx->push_pending_record(sk, flags); 1443c4d7559SDave Watson 1453c4d7559SDave Watson return 0; 1463c4d7559SDave Watson } 1473c4d7559SDave Watson 1483c4d7559SDave Watson int tls_proccess_cmsg(struct sock *sk, struct msghdr *msg, 1493c4d7559SDave Watson unsigned char *record_type) 1503c4d7559SDave Watson { 1513c4d7559SDave Watson struct cmsghdr *cmsg; 1523c4d7559SDave Watson int rc = -EINVAL; 1533c4d7559SDave Watson 1543c4d7559SDave Watson for_each_cmsghdr(cmsg, msg) { 1553c4d7559SDave Watson if (!CMSG_OK(msg, cmsg)) 1563c4d7559SDave Watson return -EINVAL; 1573c4d7559SDave Watson if (cmsg->cmsg_level != SOL_TLS) 1583c4d7559SDave Watson continue; 1593c4d7559SDave Watson 1603c4d7559SDave Watson switch (cmsg->cmsg_type) { 1613c4d7559SDave Watson case TLS_SET_RECORD_TYPE: 1623c4d7559SDave Watson if (cmsg->cmsg_len < CMSG_LEN(sizeof(*record_type))) 1633c4d7559SDave Watson return -EINVAL; 1643c4d7559SDave Watson 1653c4d7559SDave Watson if (msg->msg_flags & MSG_MORE) 1663c4d7559SDave Watson return -EINVAL; 1673c4d7559SDave Watson 1683c4d7559SDave Watson rc = tls_handle_open_record(sk, msg->msg_flags); 1693c4d7559SDave Watson if (rc) 1703c4d7559SDave Watson return rc; 1713c4d7559SDave Watson 1723c4d7559SDave Watson *record_type = *(unsigned char *)CMSG_DATA(cmsg); 1733c4d7559SDave Watson rc = 0; 1743c4d7559SDave Watson break; 1753c4d7559SDave Watson default: 1763c4d7559SDave Watson return -EINVAL; 1773c4d7559SDave Watson } 1783c4d7559SDave Watson } 1793c4d7559SDave Watson 1803c4d7559SDave Watson return rc; 1813c4d7559SDave Watson } 1823c4d7559SDave Watson 1833c4d7559SDave Watson int tls_push_pending_closed_record(struct sock *sk, struct tls_context *ctx, 1843c4d7559SDave Watson int flags, long *timeo) 1853c4d7559SDave Watson { 1863c4d7559SDave Watson struct scatterlist *sg; 1873c4d7559SDave Watson u16 offset; 1883c4d7559SDave Watson 1893c4d7559SDave Watson if (!tls_is_partially_sent_record(ctx)) 1903c4d7559SDave Watson return ctx->push_pending_record(sk, flags); 1913c4d7559SDave Watson 1923c4d7559SDave Watson sg = ctx->partially_sent_record; 1933c4d7559SDave Watson offset = ctx->partially_sent_offset; 1943c4d7559SDave Watson 1953c4d7559SDave Watson ctx->partially_sent_record = NULL; 1963c4d7559SDave Watson return tls_push_sg(sk, ctx, sg, offset, flags); 1973c4d7559SDave Watson } 1983c4d7559SDave Watson 1993c4d7559SDave Watson static void tls_write_space(struct sock *sk) 2003c4d7559SDave Watson { 2013c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 2023c4d7559SDave Watson 2033c4d7559SDave Watson if (!sk->sk_write_pending && tls_is_pending_closed_record(ctx)) { 2043c4d7559SDave Watson gfp_t sk_allocation = sk->sk_allocation; 2053c4d7559SDave Watson int rc; 2063c4d7559SDave Watson long timeo = 0; 2073c4d7559SDave Watson 2083c4d7559SDave Watson sk->sk_allocation = GFP_ATOMIC; 2093c4d7559SDave Watson rc = tls_push_pending_closed_record(sk, ctx, 2103c4d7559SDave Watson MSG_DONTWAIT | 2113c4d7559SDave Watson MSG_NOSIGNAL, 2123c4d7559SDave Watson &timeo); 2133c4d7559SDave Watson sk->sk_allocation = sk_allocation; 2143c4d7559SDave Watson 2153c4d7559SDave Watson if (rc < 0) 2163c4d7559SDave Watson return; 2173c4d7559SDave Watson } 2183c4d7559SDave Watson 2193c4d7559SDave Watson ctx->sk_write_space(sk); 2203c4d7559SDave Watson } 2213c4d7559SDave Watson 2223c4d7559SDave Watson static void tls_sk_proto_close(struct sock *sk, long timeout) 2233c4d7559SDave Watson { 2243c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 2253c4d7559SDave Watson long timeo = sock_sndtimeo(sk, 0); 2263c4d7559SDave Watson void (*sk_proto_close)(struct sock *sk, long timeout); 2273c4d7559SDave Watson 2283c4d7559SDave Watson lock_sock(sk); 2293c4d7559SDave Watson 2303c4d7559SDave Watson if (!tls_complete_pending_work(sk, ctx, 0, &timeo)) 2313c4d7559SDave Watson tls_handle_open_record(sk, 0); 2323c4d7559SDave Watson 2333c4d7559SDave Watson if (ctx->partially_sent_record) { 2343c4d7559SDave Watson struct scatterlist *sg = ctx->partially_sent_record; 2353c4d7559SDave Watson 2363c4d7559SDave Watson while (1) { 2373c4d7559SDave Watson put_page(sg_page(sg)); 2383c4d7559SDave Watson sk_mem_uncharge(sk, sg->length); 2393c4d7559SDave Watson 2403c4d7559SDave Watson if (sg_is_last(sg)) 2413c4d7559SDave Watson break; 2423c4d7559SDave Watson sg++; 2433c4d7559SDave Watson } 2443c4d7559SDave Watson } 2453c4d7559SDave Watson ctx->free_resources(sk); 2463c4d7559SDave Watson kfree(ctx->rec_seq); 2473c4d7559SDave Watson kfree(ctx->iv); 2483c4d7559SDave Watson 2493c4d7559SDave Watson sk_proto_close = ctx->sk_proto_close; 2503c4d7559SDave Watson kfree(ctx); 2513c4d7559SDave Watson 2523c4d7559SDave Watson release_sock(sk); 2533c4d7559SDave Watson sk_proto_close(sk, timeout); 2543c4d7559SDave Watson } 2553c4d7559SDave Watson 2563c4d7559SDave Watson static int do_tls_getsockopt_tx(struct sock *sk, char __user *optval, 2573c4d7559SDave Watson int __user *optlen) 2583c4d7559SDave Watson { 2593c4d7559SDave Watson int rc = 0; 2603c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 2613c4d7559SDave Watson struct tls_crypto_info *crypto_info; 2623c4d7559SDave Watson int len; 2633c4d7559SDave Watson 2643c4d7559SDave Watson if (get_user(len, optlen)) 2653c4d7559SDave Watson return -EFAULT; 2663c4d7559SDave Watson 2673c4d7559SDave Watson if (!optval || (len < sizeof(*crypto_info))) { 2683c4d7559SDave Watson rc = -EINVAL; 2693c4d7559SDave Watson goto out; 2703c4d7559SDave Watson } 2713c4d7559SDave Watson 2723c4d7559SDave Watson if (!ctx) { 2733c4d7559SDave Watson rc = -EBUSY; 2743c4d7559SDave Watson goto out; 2753c4d7559SDave Watson } 2763c4d7559SDave Watson 2773c4d7559SDave Watson /* get user crypto info */ 2783c4d7559SDave Watson crypto_info = &ctx->crypto_send; 2793c4d7559SDave Watson 2803c4d7559SDave Watson if (!TLS_CRYPTO_INFO_READY(crypto_info)) { 2813c4d7559SDave Watson rc = -EBUSY; 2823c4d7559SDave Watson goto out; 2833c4d7559SDave Watson } 2843c4d7559SDave Watson 2855a3b886cSMatthias Rosenfelder if (len == sizeof(*crypto_info)) { 286ac55cd61SDan Carpenter if (copy_to_user(optval, crypto_info, sizeof(*crypto_info))) 287ac55cd61SDan Carpenter rc = -EFAULT; 2883c4d7559SDave Watson goto out; 2893c4d7559SDave Watson } 2903c4d7559SDave Watson 2913c4d7559SDave Watson switch (crypto_info->cipher_type) { 2923c4d7559SDave Watson case TLS_CIPHER_AES_GCM_128: { 2933c4d7559SDave Watson struct tls12_crypto_info_aes_gcm_128 * 2943c4d7559SDave Watson crypto_info_aes_gcm_128 = 2953c4d7559SDave Watson container_of(crypto_info, 2963c4d7559SDave Watson struct tls12_crypto_info_aes_gcm_128, 2973c4d7559SDave Watson info); 2983c4d7559SDave Watson 2993c4d7559SDave Watson if (len != sizeof(*crypto_info_aes_gcm_128)) { 3003c4d7559SDave Watson rc = -EINVAL; 3013c4d7559SDave Watson goto out; 3023c4d7559SDave Watson } 3033c4d7559SDave Watson lock_sock(sk); 3043c4d7559SDave Watson memcpy(crypto_info_aes_gcm_128->iv, ctx->iv, 3053c4d7559SDave Watson TLS_CIPHER_AES_GCM_128_IV_SIZE); 3063c4d7559SDave Watson release_sock(sk); 307ac55cd61SDan Carpenter if (copy_to_user(optval, 3083c4d7559SDave Watson crypto_info_aes_gcm_128, 309ac55cd61SDan Carpenter sizeof(*crypto_info_aes_gcm_128))) 310ac55cd61SDan Carpenter rc = -EFAULT; 3113c4d7559SDave Watson break; 3123c4d7559SDave Watson } 3133c4d7559SDave Watson default: 3143c4d7559SDave Watson rc = -EINVAL; 3153c4d7559SDave Watson } 3163c4d7559SDave Watson 3173c4d7559SDave Watson out: 3183c4d7559SDave Watson return rc; 3193c4d7559SDave Watson } 3203c4d7559SDave Watson 3213c4d7559SDave Watson static int do_tls_getsockopt(struct sock *sk, int optname, 3223c4d7559SDave Watson char __user *optval, int __user *optlen) 3233c4d7559SDave Watson { 3243c4d7559SDave Watson int rc = 0; 3253c4d7559SDave Watson 3263c4d7559SDave Watson switch (optname) { 3273c4d7559SDave Watson case TLS_TX: 3283c4d7559SDave Watson rc = do_tls_getsockopt_tx(sk, optval, optlen); 3293c4d7559SDave Watson break; 3303c4d7559SDave Watson default: 3313c4d7559SDave Watson rc = -ENOPROTOOPT; 3323c4d7559SDave Watson break; 3333c4d7559SDave Watson } 3343c4d7559SDave Watson return rc; 3353c4d7559SDave Watson } 3363c4d7559SDave Watson 3373c4d7559SDave Watson static int tls_getsockopt(struct sock *sk, int level, int optname, 3383c4d7559SDave Watson char __user *optval, int __user *optlen) 3393c4d7559SDave Watson { 3403c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 3413c4d7559SDave Watson 3423c4d7559SDave Watson if (level != SOL_TLS) 3433c4d7559SDave Watson return ctx->getsockopt(sk, level, optname, optval, optlen); 3443c4d7559SDave Watson 3453c4d7559SDave Watson return do_tls_getsockopt(sk, optname, optval, optlen); 3463c4d7559SDave Watson } 3473c4d7559SDave Watson 3483c4d7559SDave Watson static int do_tls_setsockopt_tx(struct sock *sk, char __user *optval, 3493c4d7559SDave Watson unsigned int optlen) 3503c4d7559SDave Watson { 3513c4d7559SDave Watson struct tls_crypto_info *crypto_info, tmp_crypto_info; 3523c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 3533c4d7559SDave Watson int rc = 0; 3546d88207fSIlya Lesokhin int tx_conf; 3553c4d7559SDave Watson 3563c4d7559SDave Watson if (!optval || (optlen < sizeof(*crypto_info))) { 3573c4d7559SDave Watson rc = -EINVAL; 3583c4d7559SDave Watson goto out; 3593c4d7559SDave Watson } 3603c4d7559SDave Watson 3613c4d7559SDave Watson rc = copy_from_user(&tmp_crypto_info, optval, sizeof(*crypto_info)); 3623c4d7559SDave Watson if (rc) { 3633c4d7559SDave Watson rc = -EFAULT; 3643c4d7559SDave Watson goto out; 3653c4d7559SDave Watson } 3663c4d7559SDave Watson 3673c4d7559SDave Watson /* check version */ 3683c4d7559SDave Watson if (tmp_crypto_info.version != TLS_1_2_VERSION) { 3693c4d7559SDave Watson rc = -ENOTSUPP; 3703c4d7559SDave Watson goto out; 3713c4d7559SDave Watson } 3723c4d7559SDave Watson 3733c4d7559SDave Watson /* get user crypto info */ 3743c4d7559SDave Watson crypto_info = &ctx->crypto_send; 3753c4d7559SDave Watson 3763c4d7559SDave Watson /* Currently we don't support set crypto info more than one time */ 3773c4d7559SDave Watson if (TLS_CRYPTO_INFO_READY(crypto_info)) 3783c4d7559SDave Watson goto out; 3793c4d7559SDave Watson 3803c4d7559SDave Watson switch (tmp_crypto_info.cipher_type) { 3813c4d7559SDave Watson case TLS_CIPHER_AES_GCM_128: { 3823c4d7559SDave Watson if (optlen != sizeof(struct tls12_crypto_info_aes_gcm_128)) { 3833c4d7559SDave Watson rc = -EINVAL; 3843c4d7559SDave Watson goto out; 3853c4d7559SDave Watson } 3863c4d7559SDave Watson rc = copy_from_user( 3873c4d7559SDave Watson crypto_info, 3883c4d7559SDave Watson optval, 3893c4d7559SDave Watson sizeof(struct tls12_crypto_info_aes_gcm_128)); 3903c4d7559SDave Watson 3913c4d7559SDave Watson if (rc) { 3923c4d7559SDave Watson rc = -EFAULT; 3933c4d7559SDave Watson goto err_crypto_info; 3943c4d7559SDave Watson } 3953c4d7559SDave Watson break; 3963c4d7559SDave Watson } 3973c4d7559SDave Watson default: 3983c4d7559SDave Watson rc = -EINVAL; 3993c4d7559SDave Watson goto out; 4003c4d7559SDave Watson } 4013c4d7559SDave Watson 4023c4d7559SDave Watson ctx->sk_write_space = sk->sk_write_space; 4033c4d7559SDave Watson sk->sk_write_space = tls_write_space; 4043c4d7559SDave Watson 4053c4d7559SDave Watson ctx->sk_proto_close = sk->sk_prot->close; 4063c4d7559SDave Watson 4073c4d7559SDave Watson /* currently SW is default, we will have ethtool in future */ 4083c4d7559SDave Watson rc = tls_set_sw_offload(sk, ctx); 4096d88207fSIlya Lesokhin tx_conf = TLS_SW_TX; 4103c4d7559SDave Watson if (rc) 4113c4d7559SDave Watson goto err_crypto_info; 4123c4d7559SDave Watson 4136d88207fSIlya Lesokhin ctx->tx_conf = tx_conf; 4146d88207fSIlya Lesokhin update_sk_prot(sk, ctx); 4153c4d7559SDave Watson goto out; 4163c4d7559SDave Watson 4173c4d7559SDave Watson err_crypto_info: 4183c4d7559SDave Watson memset(crypto_info, 0, sizeof(*crypto_info)); 4193c4d7559SDave Watson out: 4203c4d7559SDave Watson return rc; 4213c4d7559SDave Watson } 4223c4d7559SDave Watson 4233c4d7559SDave Watson static int do_tls_setsockopt(struct sock *sk, int optname, 4243c4d7559SDave Watson char __user *optval, unsigned int optlen) 4253c4d7559SDave Watson { 4263c4d7559SDave Watson int rc = 0; 4273c4d7559SDave Watson 4283c4d7559SDave Watson switch (optname) { 4293c4d7559SDave Watson case TLS_TX: 4303c4d7559SDave Watson lock_sock(sk); 4313c4d7559SDave Watson rc = do_tls_setsockopt_tx(sk, optval, optlen); 4323c4d7559SDave Watson release_sock(sk); 4333c4d7559SDave Watson break; 4343c4d7559SDave Watson default: 4353c4d7559SDave Watson rc = -ENOPROTOOPT; 4363c4d7559SDave Watson break; 4373c4d7559SDave Watson } 4383c4d7559SDave Watson return rc; 4393c4d7559SDave Watson } 4403c4d7559SDave Watson 4413c4d7559SDave Watson static int tls_setsockopt(struct sock *sk, int level, int optname, 4423c4d7559SDave Watson char __user *optval, unsigned int optlen) 4433c4d7559SDave Watson { 4443c4d7559SDave Watson struct tls_context *ctx = tls_get_ctx(sk); 4453c4d7559SDave Watson 4463c4d7559SDave Watson if (level != SOL_TLS) 4473c4d7559SDave Watson return ctx->setsockopt(sk, level, optname, optval, optlen); 4483c4d7559SDave Watson 4493c4d7559SDave Watson return do_tls_setsockopt(sk, optname, optval, optlen); 4503c4d7559SDave Watson } 4513c4d7559SDave Watson 4523c4d7559SDave Watson static int tls_init(struct sock *sk) 4533c4d7559SDave Watson { 4543c4d7559SDave Watson struct inet_connection_sock *icsk = inet_csk(sk); 4553c4d7559SDave Watson struct tls_context *ctx; 4563c4d7559SDave Watson int rc = 0; 4573c4d7559SDave Watson 4583c4d7559SDave Watson /* allocate tls context */ 4593c4d7559SDave Watson ctx = kzalloc(sizeof(*ctx), GFP_KERNEL); 4603c4d7559SDave Watson if (!ctx) { 4613c4d7559SDave Watson rc = -ENOMEM; 4623c4d7559SDave Watson goto out; 4633c4d7559SDave Watson } 4643c4d7559SDave Watson icsk->icsk_ulp_data = ctx; 4653c4d7559SDave Watson ctx->setsockopt = sk->sk_prot->setsockopt; 4663c4d7559SDave Watson ctx->getsockopt = sk->sk_prot->getsockopt; 4676d88207fSIlya Lesokhin 4686d88207fSIlya Lesokhin ctx->tx_conf = TLS_BASE_TX; 4696d88207fSIlya Lesokhin update_sk_prot(sk, ctx); 4703c4d7559SDave Watson out: 4713c4d7559SDave Watson return rc; 4723c4d7559SDave Watson } 4733c4d7559SDave Watson 4743c4d7559SDave Watson static struct tcp_ulp_ops tcp_tls_ulp_ops __read_mostly = { 4753c4d7559SDave Watson .name = "tls", 4763c4d7559SDave Watson .owner = THIS_MODULE, 4773c4d7559SDave Watson .init = tls_init, 4783c4d7559SDave Watson }; 4793c4d7559SDave Watson 4806d88207fSIlya Lesokhin static void build_protos(struct proto *prot, struct proto *base) 4816d88207fSIlya Lesokhin { 4826d88207fSIlya Lesokhin prot[TLS_BASE_TX] = *base; 4836d88207fSIlya Lesokhin prot[TLS_BASE_TX].setsockopt = tls_setsockopt; 4846d88207fSIlya Lesokhin prot[TLS_BASE_TX].getsockopt = tls_getsockopt; 4856d88207fSIlya Lesokhin 4866d88207fSIlya Lesokhin prot[TLS_SW_TX] = prot[TLS_BASE_TX]; 4876d88207fSIlya Lesokhin prot[TLS_SW_TX].close = tls_sk_proto_close; 4886d88207fSIlya Lesokhin prot[TLS_SW_TX].sendmsg = tls_sw_sendmsg; 4896d88207fSIlya Lesokhin prot[TLS_SW_TX].sendpage = tls_sw_sendpage; 4906d88207fSIlya Lesokhin } 4916d88207fSIlya Lesokhin 4923c4d7559SDave Watson static int __init tls_register(void) 4933c4d7559SDave Watson { 4946d88207fSIlya Lesokhin build_protos(tls_prots, &tcp_prot); 4953c4d7559SDave Watson 4963c4d7559SDave Watson tcp_register_ulp(&tcp_tls_ulp_ops); 4973c4d7559SDave Watson 4983c4d7559SDave Watson return 0; 4993c4d7559SDave Watson } 5003c4d7559SDave Watson 5013c4d7559SDave Watson static void __exit tls_unregister(void) 5023c4d7559SDave Watson { 5033c4d7559SDave Watson tcp_unregister_ulp(&tcp_tls_ulp_ops); 5043c4d7559SDave Watson } 5053c4d7559SDave Watson 5063c4d7559SDave Watson module_init(tls_register); 5073c4d7559SDave Watson module_exit(tls_unregister); 508