1 /* $NetBSD: kexsntrup761x25519.c,v 1.4 2024/09/24 21:32:18 christos Exp $ */ 2 /* $OpenBSD: kexsntrup761x25519.c,v 1.3 2024/09/15 02:20:51 djm Exp $ */ 3 4 /* 5 * Copyright (c) 2019 Markus Friedl. All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 */ 27 28 #include "includes.h" 29 __RCSID("$NetBSD: kexsntrup761x25519.c,v 1.4 2024/09/24 21:32:18 christos Exp $"); 30 31 #include <sys/types.h> 32 33 #include <stdio.h> 34 #include <string.h> 35 #include <signal.h> 36 37 #include "sshkey.h" 38 #include "kex.h" 39 #include "sshbuf.h" 40 #include "digest.h" 41 #include "ssherr.h" 42 43 volatile crypto_int16 crypto_int16_optblocker = 0; 44 volatile crypto_int32 crypto_int32_optblocker = 0; 45 volatile crypto_int64 crypto_int64_optblocker = 0; 46 47 int 48 kex_kem_sntrup761x25519_keypair(struct kex *kex) 49 { 50 struct sshbuf *buf = NULL; 51 u_char *cp = NULL; 52 size_t need; 53 int r; 54 55 if ((buf = sshbuf_new()) == NULL) 56 return SSH_ERR_ALLOC_FAIL; 57 need = crypto_kem_sntrup761_PUBLICKEYBYTES + CURVE25519_SIZE; 58 if ((r = sshbuf_reserve(buf, need, &cp)) != 0) 59 goto out; 60 crypto_kem_sntrup761_keypair(cp, kex->sntrup761_client_key); 61 #ifdef DEBUG_KEXECDH 62 dump_digest("client public key sntrup761:", cp, 63 crypto_kem_sntrup761_PUBLICKEYBYTES); 64 #endif 65 cp += crypto_kem_sntrup761_PUBLICKEYBYTES; 66 kexc25519_keygen(kex->c25519_client_key, cp); 67 #ifdef DEBUG_KEXECDH 68 dump_digest("client public key c25519:", cp, CURVE25519_SIZE); 69 #endif 70 kex->client_pub = buf; 71 buf = NULL; 72 out: 73 sshbuf_free(buf); 74 return r; 75 } 76 77 int 78 kex_kem_sntrup761x25519_enc(struct kex *kex, 79 const struct sshbuf *client_blob, struct sshbuf **server_blobp, 80 struct sshbuf **shared_secretp) 81 { 82 struct sshbuf *server_blob = NULL; 83 struct sshbuf *buf = NULL; 84 const u_char *client_pub; 85 u_char *kem_key, *ciphertext, *server_pub; 86 u_char server_key[CURVE25519_SIZE]; 87 u_char hash[SSH_DIGEST_MAX_LENGTH]; 88 size_t need; 89 int r; 90 91 *server_blobp = NULL; 92 *shared_secretp = NULL; 93 94 /* client_blob contains both KEM and ECDH client pubkeys */ 95 need = crypto_kem_sntrup761_PUBLICKEYBYTES + CURVE25519_SIZE; 96 if (sshbuf_len(client_blob) != need) { 97 r = SSH_ERR_SIGNATURE_INVALID; 98 goto out; 99 } 100 client_pub = sshbuf_ptr(client_blob); 101 #ifdef DEBUG_KEXECDH 102 dump_digest("client public key sntrup761:", client_pub, 103 crypto_kem_sntrup761_PUBLICKEYBYTES); 104 dump_digest("client public key 25519:", 105 client_pub + crypto_kem_sntrup761_PUBLICKEYBYTES, 106 CURVE25519_SIZE); 107 #endif 108 /* allocate buffer for concatenation of KEM key and ECDH shared key */ 109 /* the buffer will be hashed and the result is the shared secret */ 110 if ((buf = sshbuf_new()) == NULL) { 111 r = SSH_ERR_ALLOC_FAIL; 112 goto out; 113 } 114 if ((r = sshbuf_reserve(buf, crypto_kem_sntrup761_BYTES, 115 &kem_key)) != 0) 116 goto out; 117 /* allocate space for encrypted KEM key and ECDH pub key */ 118 if ((server_blob = sshbuf_new()) == NULL) { 119 r = SSH_ERR_ALLOC_FAIL; 120 goto out; 121 } 122 need = crypto_kem_sntrup761_CIPHERTEXTBYTES + CURVE25519_SIZE; 123 if ((r = sshbuf_reserve(server_blob, need, &ciphertext)) != 0) 124 goto out; 125 /* generate and encrypt KEM key with client key */ 126 crypto_kem_sntrup761_enc(ciphertext, kem_key, client_pub); 127 /* generate ECDH key pair, store server pubkey after ciphertext */ 128 server_pub = ciphertext + crypto_kem_sntrup761_CIPHERTEXTBYTES; 129 kexc25519_keygen(server_key, server_pub); 130 /* append ECDH shared key */ 131 client_pub += crypto_kem_sntrup761_PUBLICKEYBYTES; 132 if ((r = kexc25519_shared_key_ext(server_key, client_pub, buf, 1)) < 0) 133 goto out; 134 if ((r = ssh_digest_buffer(kex->hash_alg, buf, hash, sizeof(hash))) != 0) 135 goto out; 136 #ifdef DEBUG_KEXECDH 137 dump_digest("server public key 25519:", server_pub, CURVE25519_SIZE); 138 dump_digest("server cipher text:", ciphertext, 139 crypto_kem_sntrup761_CIPHERTEXTBYTES); 140 dump_digest("server kem key:", kem_key, crypto_kem_sntrup761_BYTES); 141 dump_digest("concatenation of KEM key and ECDH shared key:", 142 sshbuf_ptr(buf), sshbuf_len(buf)); 143 #endif 144 /* string-encoded hash is resulting shared secret */ 145 sshbuf_reset(buf); 146 if ((r = sshbuf_put_string(buf, hash, 147 ssh_digest_bytes(kex->hash_alg))) != 0) 148 goto out; 149 #ifdef DEBUG_KEXECDH 150 dump_digest("encoded shared secret:", sshbuf_ptr(buf), sshbuf_len(buf)); 151 #endif 152 *server_blobp = server_blob; 153 *shared_secretp = buf; 154 server_blob = NULL; 155 buf = NULL; 156 out: 157 explicit_bzero(hash, sizeof(hash)); 158 explicit_bzero(server_key, sizeof(server_key)); 159 sshbuf_free(server_blob); 160 sshbuf_free(buf); 161 return r; 162 } 163 164 int 165 kex_kem_sntrup761x25519_dec(struct kex *kex, 166 const struct sshbuf *server_blob, struct sshbuf **shared_secretp) 167 { 168 struct sshbuf *buf = NULL; 169 u_char *kem_key = NULL; 170 const u_char *ciphertext, *server_pub; 171 u_char hash[SSH_DIGEST_MAX_LENGTH]; 172 size_t need; 173 int r, decoded; 174 175 *shared_secretp = NULL; 176 177 need = crypto_kem_sntrup761_CIPHERTEXTBYTES + CURVE25519_SIZE; 178 if (sshbuf_len(server_blob) != need) { 179 r = SSH_ERR_SIGNATURE_INVALID; 180 goto out; 181 } 182 ciphertext = sshbuf_ptr(server_blob); 183 server_pub = ciphertext + crypto_kem_sntrup761_CIPHERTEXTBYTES; 184 #ifdef DEBUG_KEXECDH 185 dump_digest("server cipher text:", ciphertext, 186 crypto_kem_sntrup761_CIPHERTEXTBYTES); 187 dump_digest("server public key c25519:", server_pub, CURVE25519_SIZE); 188 #endif 189 /* hash concatenation of KEM key and ECDH shared key */ 190 if ((buf = sshbuf_new()) == NULL) { 191 r = SSH_ERR_ALLOC_FAIL; 192 goto out; 193 } 194 if ((r = sshbuf_reserve(buf, crypto_kem_sntrup761_BYTES, 195 &kem_key)) != 0) 196 goto out; 197 decoded = crypto_kem_sntrup761_dec(kem_key, ciphertext, 198 kex->sntrup761_client_key); 199 if ((r = kexc25519_shared_key_ext(kex->c25519_client_key, server_pub, 200 buf, 1)) < 0) 201 goto out; 202 if ((r = ssh_digest_buffer(kex->hash_alg, buf, hash, sizeof(hash))) != 0) 203 goto out; 204 #ifdef DEBUG_KEXECDH 205 dump_digest("client kem key:", kem_key, crypto_kem_sntrup761_BYTES); 206 dump_digest("concatenation of KEM key and ECDH shared key:", 207 sshbuf_ptr(buf), sshbuf_len(buf)); 208 #endif 209 sshbuf_reset(buf); 210 if ((r = sshbuf_put_string(buf, hash, 211 ssh_digest_bytes(kex->hash_alg))) != 0) 212 goto out; 213 #ifdef DEBUG_KEXECDH 214 dump_digest("encoded shared secret:", sshbuf_ptr(buf), sshbuf_len(buf)); 215 #endif 216 if (decoded != 0) { 217 r = SSH_ERR_SIGNATURE_INVALID; 218 goto out; 219 } 220 *shared_secretp = buf; 221 buf = NULL; 222 out: 223 explicit_bzero(hash, sizeof(hash)); 224 sshbuf_free(buf); 225 return r; 226 } 227