11ca1b917SEric Biggers /* 2aa762409SEric Biggers * ChaCha and XChaCha stream ciphers, including ChaCha20 (RFC7539) 31ca1b917SEric Biggers * 41ca1b917SEric Biggers * Copyright (C) 2015 Martin Willi 51ca1b917SEric Biggers * Copyright (C) 2018 Google LLC 61ca1b917SEric Biggers * 71ca1b917SEric Biggers * This program is free software; you can redistribute it and/or modify 81ca1b917SEric Biggers * it under the terms of the GNU General Public License as published by 91ca1b917SEric Biggers * the Free Software Foundation; either version 2 of the License, or 101ca1b917SEric Biggers * (at your option) any later version. 111ca1b917SEric Biggers */ 121ca1b917SEric Biggers 131ca1b917SEric Biggers #include <asm/unaligned.h> 141ca1b917SEric Biggers #include <crypto/algapi.h> 151ca1b917SEric Biggers #include <crypto/chacha.h> 161ca1b917SEric Biggers #include <crypto/internal/skcipher.h> 171ca1b917SEric Biggers #include <linux/module.h> 181ca1b917SEric Biggers 191ca1b917SEric Biggers static void chacha_docrypt(u32 *state, u8 *dst, const u8 *src, 201ca1b917SEric Biggers unsigned int bytes, int nrounds) 211ca1b917SEric Biggers { 221ca1b917SEric Biggers /* aligned to potentially speed up crypto_xor() */ 231ca1b917SEric Biggers u8 stream[CHACHA_BLOCK_SIZE] __aligned(sizeof(long)); 241ca1b917SEric Biggers 251ca1b917SEric Biggers while (bytes >= CHACHA_BLOCK_SIZE) { 261ca1b917SEric Biggers chacha_block(state, stream, nrounds); 2729d97decSEric Biggers crypto_xor_cpy(dst, src, stream, CHACHA_BLOCK_SIZE); 281ca1b917SEric Biggers bytes -= CHACHA_BLOCK_SIZE; 291ca1b917SEric Biggers dst += CHACHA_BLOCK_SIZE; 3029d97decSEric Biggers src += CHACHA_BLOCK_SIZE; 311ca1b917SEric Biggers } 321ca1b917SEric Biggers if (bytes) { 331ca1b917SEric Biggers chacha_block(state, stream, nrounds); 3429d97decSEric Biggers crypto_xor_cpy(dst, src, stream, bytes); 351ca1b917SEric Biggers } 361ca1b917SEric Biggers } 371ca1b917SEric Biggers 381ca1b917SEric Biggers static int chacha_stream_xor(struct skcipher_request *req, 391ca1b917SEric Biggers struct chacha_ctx *ctx, u8 *iv) 401ca1b917SEric Biggers { 411ca1b917SEric Biggers struct skcipher_walk walk; 421ca1b917SEric Biggers u32 state[16]; 431ca1b917SEric Biggers int err; 441ca1b917SEric Biggers 451ca1b917SEric Biggers err = skcipher_walk_virt(&walk, req, false); 461ca1b917SEric Biggers 471ca1b917SEric Biggers crypto_chacha_init(state, ctx, iv); 481ca1b917SEric Biggers 491ca1b917SEric Biggers while (walk.nbytes > 0) { 501ca1b917SEric Biggers unsigned int nbytes = walk.nbytes; 511ca1b917SEric Biggers 521ca1b917SEric Biggers if (nbytes < walk.total) 537aceaaefSEric Biggers nbytes = round_down(nbytes, CHACHA_BLOCK_SIZE); 541ca1b917SEric Biggers 551ca1b917SEric Biggers chacha_docrypt(state, walk.dst.virt.addr, walk.src.virt.addr, 561ca1b917SEric Biggers nbytes, ctx->nrounds); 571ca1b917SEric Biggers err = skcipher_walk_done(&walk, walk.nbytes - nbytes); 581ca1b917SEric Biggers } 591ca1b917SEric Biggers 601ca1b917SEric Biggers return err; 611ca1b917SEric Biggers } 621ca1b917SEric Biggers 631ca1b917SEric Biggers void crypto_chacha_init(u32 *state, struct chacha_ctx *ctx, u8 *iv) 641ca1b917SEric Biggers { 651ca1b917SEric Biggers state[0] = 0x61707865; /* "expa" */ 661ca1b917SEric Biggers state[1] = 0x3320646e; /* "nd 3" */ 671ca1b917SEric Biggers state[2] = 0x79622d32; /* "2-by" */ 681ca1b917SEric Biggers state[3] = 0x6b206574; /* "te k" */ 691ca1b917SEric Biggers state[4] = ctx->key[0]; 701ca1b917SEric Biggers state[5] = ctx->key[1]; 711ca1b917SEric Biggers state[6] = ctx->key[2]; 721ca1b917SEric Biggers state[7] = ctx->key[3]; 731ca1b917SEric Biggers state[8] = ctx->key[4]; 741ca1b917SEric Biggers state[9] = ctx->key[5]; 751ca1b917SEric Biggers state[10] = ctx->key[6]; 761ca1b917SEric Biggers state[11] = ctx->key[7]; 771ca1b917SEric Biggers state[12] = get_unaligned_le32(iv + 0); 781ca1b917SEric Biggers state[13] = get_unaligned_le32(iv + 4); 791ca1b917SEric Biggers state[14] = get_unaligned_le32(iv + 8); 801ca1b917SEric Biggers state[15] = get_unaligned_le32(iv + 12); 811ca1b917SEric Biggers } 821ca1b917SEric Biggers EXPORT_SYMBOL_GPL(crypto_chacha_init); 831ca1b917SEric Biggers 841ca1b917SEric Biggers static int chacha_setkey(struct crypto_skcipher *tfm, const u8 *key, 851ca1b917SEric Biggers unsigned int keysize, int nrounds) 861ca1b917SEric Biggers { 871ca1b917SEric Biggers struct chacha_ctx *ctx = crypto_skcipher_ctx(tfm); 881ca1b917SEric Biggers int i; 891ca1b917SEric Biggers 901ca1b917SEric Biggers if (keysize != CHACHA_KEY_SIZE) 911ca1b917SEric Biggers return -EINVAL; 921ca1b917SEric Biggers 931ca1b917SEric Biggers for (i = 0; i < ARRAY_SIZE(ctx->key); i++) 941ca1b917SEric Biggers ctx->key[i] = get_unaligned_le32(key + i * sizeof(u32)); 951ca1b917SEric Biggers 961ca1b917SEric Biggers ctx->nrounds = nrounds; 971ca1b917SEric Biggers return 0; 981ca1b917SEric Biggers } 991ca1b917SEric Biggers 1001ca1b917SEric Biggers int crypto_chacha20_setkey(struct crypto_skcipher *tfm, const u8 *key, 1011ca1b917SEric Biggers unsigned int keysize) 1021ca1b917SEric Biggers { 1031ca1b917SEric Biggers return chacha_setkey(tfm, key, keysize, 20); 1041ca1b917SEric Biggers } 1051ca1b917SEric Biggers EXPORT_SYMBOL_GPL(crypto_chacha20_setkey); 1061ca1b917SEric Biggers 107aa762409SEric Biggers int crypto_chacha12_setkey(struct crypto_skcipher *tfm, const u8 *key, 108aa762409SEric Biggers unsigned int keysize) 109aa762409SEric Biggers { 110aa762409SEric Biggers return chacha_setkey(tfm, key, keysize, 12); 111aa762409SEric Biggers } 112aa762409SEric Biggers EXPORT_SYMBOL_GPL(crypto_chacha12_setkey); 113aa762409SEric Biggers 1141ca1b917SEric Biggers int crypto_chacha_crypt(struct skcipher_request *req) 1151ca1b917SEric Biggers { 1161ca1b917SEric Biggers struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req); 1171ca1b917SEric Biggers struct chacha_ctx *ctx = crypto_skcipher_ctx(tfm); 1181ca1b917SEric Biggers 1191ca1b917SEric Biggers return chacha_stream_xor(req, ctx, req->iv); 1201ca1b917SEric Biggers } 1211ca1b917SEric Biggers EXPORT_SYMBOL_GPL(crypto_chacha_crypt); 1221ca1b917SEric Biggers 1231ca1b917SEric Biggers int crypto_xchacha_crypt(struct skcipher_request *req) 1241ca1b917SEric Biggers { 1251ca1b917SEric Biggers struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req); 1261ca1b917SEric Biggers struct chacha_ctx *ctx = crypto_skcipher_ctx(tfm); 1271ca1b917SEric Biggers struct chacha_ctx subctx; 1281ca1b917SEric Biggers u32 state[16]; 1291ca1b917SEric Biggers u8 real_iv[16]; 1301ca1b917SEric Biggers 1311ca1b917SEric Biggers /* Compute the subkey given the original key and first 128 nonce bits */ 1321ca1b917SEric Biggers crypto_chacha_init(state, ctx, req->iv); 1331ca1b917SEric Biggers hchacha_block(state, subctx.key, ctx->nrounds); 1341ca1b917SEric Biggers subctx.nrounds = ctx->nrounds; 1351ca1b917SEric Biggers 1361ca1b917SEric Biggers /* Build the real IV */ 1371ca1b917SEric Biggers memcpy(&real_iv[0], req->iv + 24, 8); /* stream position */ 1381ca1b917SEric Biggers memcpy(&real_iv[8], req->iv + 16, 8); /* remaining 64 nonce bits */ 1391ca1b917SEric Biggers 1401ca1b917SEric Biggers /* Generate the stream and XOR it with the data */ 1411ca1b917SEric Biggers return chacha_stream_xor(req, &subctx, real_iv); 1421ca1b917SEric Biggers } 1431ca1b917SEric Biggers EXPORT_SYMBOL_GPL(crypto_xchacha_crypt); 1441ca1b917SEric Biggers 1451ca1b917SEric Biggers static struct skcipher_alg algs[] = { 1461ca1b917SEric Biggers { 1471ca1b917SEric Biggers .base.cra_name = "chacha20", 1481ca1b917SEric Biggers .base.cra_driver_name = "chacha20-generic", 1491ca1b917SEric Biggers .base.cra_priority = 100, 1501ca1b917SEric Biggers .base.cra_blocksize = 1, 1511ca1b917SEric Biggers .base.cra_ctxsize = sizeof(struct chacha_ctx), 1521ca1b917SEric Biggers .base.cra_module = THIS_MODULE, 1531ca1b917SEric Biggers 1541ca1b917SEric Biggers .min_keysize = CHACHA_KEY_SIZE, 1551ca1b917SEric Biggers .max_keysize = CHACHA_KEY_SIZE, 1561ca1b917SEric Biggers .ivsize = CHACHA_IV_SIZE, 1571ca1b917SEric Biggers .chunksize = CHACHA_BLOCK_SIZE, 1581ca1b917SEric Biggers .setkey = crypto_chacha20_setkey, 1591ca1b917SEric Biggers .encrypt = crypto_chacha_crypt, 1601ca1b917SEric Biggers .decrypt = crypto_chacha_crypt, 1611ca1b917SEric Biggers }, { 1621ca1b917SEric Biggers .base.cra_name = "xchacha20", 1631ca1b917SEric Biggers .base.cra_driver_name = "xchacha20-generic", 1641ca1b917SEric Biggers .base.cra_priority = 100, 1651ca1b917SEric Biggers .base.cra_blocksize = 1, 1661ca1b917SEric Biggers .base.cra_ctxsize = sizeof(struct chacha_ctx), 1671ca1b917SEric Biggers .base.cra_module = THIS_MODULE, 1681ca1b917SEric Biggers 1691ca1b917SEric Biggers .min_keysize = CHACHA_KEY_SIZE, 1701ca1b917SEric Biggers .max_keysize = CHACHA_KEY_SIZE, 1711ca1b917SEric Biggers .ivsize = XCHACHA_IV_SIZE, 1721ca1b917SEric Biggers .chunksize = CHACHA_BLOCK_SIZE, 1731ca1b917SEric Biggers .setkey = crypto_chacha20_setkey, 1741ca1b917SEric Biggers .encrypt = crypto_xchacha_crypt, 1751ca1b917SEric Biggers .decrypt = crypto_xchacha_crypt, 176aa762409SEric Biggers }, { 177aa762409SEric Biggers .base.cra_name = "xchacha12", 178aa762409SEric Biggers .base.cra_driver_name = "xchacha12-generic", 179aa762409SEric Biggers .base.cra_priority = 100, 180aa762409SEric Biggers .base.cra_blocksize = 1, 181aa762409SEric Biggers .base.cra_ctxsize = sizeof(struct chacha_ctx), 182aa762409SEric Biggers .base.cra_module = THIS_MODULE, 183aa762409SEric Biggers 184aa762409SEric Biggers .min_keysize = CHACHA_KEY_SIZE, 185aa762409SEric Biggers .max_keysize = CHACHA_KEY_SIZE, 186aa762409SEric Biggers .ivsize = XCHACHA_IV_SIZE, 187aa762409SEric Biggers .chunksize = CHACHA_BLOCK_SIZE, 188aa762409SEric Biggers .setkey = crypto_chacha12_setkey, 189aa762409SEric Biggers .encrypt = crypto_xchacha_crypt, 190aa762409SEric Biggers .decrypt = crypto_xchacha_crypt, 1911ca1b917SEric Biggers } 1921ca1b917SEric Biggers }; 1931ca1b917SEric Biggers 1941ca1b917SEric Biggers static int __init chacha_generic_mod_init(void) 1951ca1b917SEric Biggers { 1961ca1b917SEric Biggers return crypto_register_skciphers(algs, ARRAY_SIZE(algs)); 1971ca1b917SEric Biggers } 1981ca1b917SEric Biggers 1991ca1b917SEric Biggers static void __exit chacha_generic_mod_fini(void) 2001ca1b917SEric Biggers { 2011ca1b917SEric Biggers crypto_unregister_skciphers(algs, ARRAY_SIZE(algs)); 2021ca1b917SEric Biggers } 2031ca1b917SEric Biggers 204c4741b23SEric Biggers subsys_initcall(chacha_generic_mod_init); 2051ca1b917SEric Biggers module_exit(chacha_generic_mod_fini); 2061ca1b917SEric Biggers 2071ca1b917SEric Biggers MODULE_LICENSE("GPL"); 2081ca1b917SEric Biggers MODULE_AUTHOR("Martin Willi <martin@strongswan.org>"); 2091ca1b917SEric Biggers MODULE_DESCRIPTION("ChaCha and XChaCha stream ciphers (generic)"); 2101ca1b917SEric Biggers MODULE_ALIAS_CRYPTO("chacha20"); 2111ca1b917SEric Biggers MODULE_ALIAS_CRYPTO("chacha20-generic"); 2121ca1b917SEric Biggers MODULE_ALIAS_CRYPTO("xchacha20"); 2131ca1b917SEric Biggers MODULE_ALIAS_CRYPTO("xchacha20-generic"); 214aa762409SEric Biggers MODULE_ALIAS_CRYPTO("xchacha12"); 215aa762409SEric Biggers MODULE_ALIAS_CRYPTO("xchacha12-generic"); 216