nim-libp2p/libp2p/crypto/chacha20poly1305.nim

92 lines
2.6 KiB
Nim

## Nim-Libp2p
## Copyright (c) 2020 Status Research & Development GmbH
## Licensed under either of
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
## at your option.
## This file may not be copied, modified, or distributed except according to
## those terms.
## This module integrates BearSSL ChaCha20+Poly1305
##
## This module uses unmodified parts of code from
## BearSSL library <https://bearssl.org/>
## Copyright(C) 2018 Thomas Pornin <pornin@bolet.org>.
# RFC @ https://tools.ietf.org/html/rfc7539
import bearssl
const
ChaChaPolyKeySize = 32
ChaChaPolyNonceSize = 12
ChaChaPolyTagSize = 16
type
ChaChaPoly* = object
ChaChaPolyKey* = array[ChaChaPolyKeySize, byte]
ChaChaPolyNonce* = array[ChaChaPolyNonceSize, byte]
ChaChaPolyTag* = array[ChaChaPolyTagSize, byte]
converter toChaChaPolyKey*(s: seq[byte]): ChaChaPolyKey =
assert s.len == ChaChaPolyKeySize
copyMem(addr result[0], unsafeaddr s[0], ChaChaPolyKeySize)
converter toChaChaPolyNonce*(s: seq[byte]): ChaChaPolyNonce =
assert s.len == ChaChaPolyNonceSize
copyMem(addr result[0], unsafeaddr s[0], ChaChaPolyNonceSize)
converter toChaChaPolyTag*(s: seq[byte]): ChaChaPolyTag =
assert s.len == ChaChaPolyTagSize
copyMem(addr result[0], unsafeaddr s[0], ChaChaPolyTagSize)
# bearssl allows us to use optimized versions
# this is reconciled at runtime
# we do this in the global scope / module init
# try for the best first
var
chachapoly_native_impl: Poly1305Run = poly1305CtmulqGet()
chacha_native_impl: Chacha20Run = chacha20Sse2Get()
# fall back if not available
if chachapoly_native_impl == nil:
chachapoly_native_impl = poly1305CtmulRun
if chacha_native_impl == nil:
chacha_native_impl = chacha20CtRun
proc encrypt*(_: type[ChaChaPoly],
key: var ChaChaPolyKey,
nonce: var ChaChaPolyNonce,
tag: var ChaChaPolyTag,
data: var openarray[byte],
aad: var openarray[byte]) =
chachapoly_native_impl(
addr key[0],
addr nonce[0],
addr data[0],
data.len,
addr aad[0],
aad.len,
addr tag[0],
chacha_native_impl,
#[encrypt]# 1.cint)
proc decrypt*(_: type[ChaChaPoly],
key: var ChaChaPolyKey,
nonce: var ChaChaPolyNonce,
tag: var ChaChaPolyTag,
data: var openarray[byte],
aad: var openarray[byte]) =
chachapoly_native_impl(
addr key[0],
addr nonce[0],
addr data[0],
data.len,
addr aad[0],
aad.len,
addr tag[0],
chacha_native_impl,
#[decrypt]# 0.cint)