blob: 3c6ad829ddea4b5e9d6c30c07ab7f65843750443 [file] [log] [blame]
package runner
import (
"crypto/cipher"
"crypto/subtle"
"encoding/binary"
"errors"
"./poly1305"
)
// See RFC 7539.
func leftRotate(a uint32, n uint) uint32 {
return (a << n) | (a >> (32 - n))
}
func chaChaQuarterRound(state *[16]uint32, a, b, c, d int) {
state[a] += state[b]
state[d] = leftRotate(state[d]^state[a], 16)
state[c] += state[d]
state[b] = leftRotate(state[b]^state[c], 12)
state[a] += state[b]
state[d] = leftRotate(state[d]^state[a], 8)
state[c] += state[d]
state[b] = leftRotate(state[b]^state[c], 7)
}
func chaCha20Block(state *[16]uint32, out []byte) {
var workingState [16]uint32
copy(workingState[:], state[:])
for i := 0; i < 10; i++ {
chaChaQuarterRound(&workingState, 0, 4, 8, 12)
chaChaQuarterRound(&workingState, 1, 5, 9, 13)
chaChaQuarterRound(&workingState, 2, 6, 10, 14)
chaChaQuarterRound(&workingState, 3, 7, 11, 15)
chaChaQuarterRound(&workingState, 0, 5, 10, 15)
chaChaQuarterRound(&workingState, 1, 6, 11, 12)
chaChaQuarterRound(&workingState, 2, 7, 8, 13)
chaChaQuarterRound(&workingState, 3, 4, 9, 14)
}
for i := 0; i < 16; i++ {
binary.LittleEndian.PutUint32(out[i*4:i*4+4], workingState[i]+state[i])
}
}
// sliceForAppend takes a slice and a requested number of bytes. It returns a
// slice with the contents of the given slice followed by that many bytes and a
// second slice that aliases into it and contains only the extra bytes. If the
// original slice has sufficient capacity then no allocation is performed.
func sliceForAppend(in []byte, n int) (head, tail []byte) {
if total := len(in) + n; cap(in) >= total {
head = in[:total]
} else {
head = make([]byte, total)
copy(head, in)
}
tail = head[len(in):]
return
}
func chaCha20(out, in, key, nonce []byte, counter uint64) {
var state [16]uint32
state[0] = 0x61707865
state[1] = 0x3320646e
state[2] = 0x79622d32
state[3] = 0x6b206574
for i := 0; i < 8; i++ {
state[4+i] = binary.LittleEndian.Uint32(key[i*4 : i*4+4])
}
switch len(nonce) {
case 8:
state[14] = binary.LittleEndian.Uint32(nonce[0:4])
state[15] = binary.LittleEndian.Uint32(nonce[4:8])
case 12:
state[13] = binary.LittleEndian.Uint32(nonce[0:4])
state[14] = binary.LittleEndian.Uint32(nonce[4:8])
state[15] = binary.LittleEndian.Uint32(nonce[8:12])
default:
panic("bad nonce length")
}
for i := 0; i < len(in); i += 64 {
state[12] = uint32(counter)
var tmp [64]byte
chaCha20Block(&state, tmp[:])
count := 64
if len(in)-i < count {
count = len(in) - i
}
for j := 0; j < count; j++ {
out[i+j] = in[i+j] ^ tmp[j]
}
counter++
}
}
// chaCha20Poly1305 implements the AEAD from
// RFC 7539 and draft-agl-tls-chacha20poly1305-04.
type chaCha20Poly1305 struct {
key [32]byte
// oldMode, if true, indicates that the draft spec should be
// implemented rather than the final, RFC version.
oldMode bool
}
func newChaCha20Poly1305(key []byte) (cipher.AEAD, error) {
if len(key) != 32 {
return nil, errors.New("bad key length")
}
aead := new(chaCha20Poly1305)
copy(aead.key[:], key)
return aead, nil
}
func newChaCha20Poly1305Old(key []byte) (cipher.AEAD, error) {
if len(key) != 32 {
return nil, errors.New("bad key length")
}
aead := &chaCha20Poly1305{
oldMode: true,
}
copy(aead.key[:], key)
return aead, nil
}
func (c *chaCha20Poly1305) NonceSize() int {
if c.oldMode {
return 8
} else {
return 12
}
}
func (c *chaCha20Poly1305) Overhead() int { return 16 }
func (c *chaCha20Poly1305) poly1305(tag *[16]byte, nonce, ciphertext, additionalData []byte) {
input := make([]byte, 0, len(additionalData)+15+len(ciphertext)+15+8+8)
input = append(input, additionalData...)
var zeros [15]byte
if pad := len(input) % 16; pad != 0 {
input = append(input, zeros[:16-pad]...)
}
input = append(input, ciphertext...)
if pad := len(input) % 16; pad != 0 {
input = append(input, zeros[:16-pad]...)
}
input, out := sliceForAppend(input, 8)
binary.LittleEndian.PutUint64(out, uint64(len(additionalData)))
input, out = sliceForAppend(input, 8)
binary.LittleEndian.PutUint64(out, uint64(len(ciphertext)))
var poly1305Key [32]byte
chaCha20(poly1305Key[:], poly1305Key[:], c.key[:], nonce, 0)
poly1305.Sum(tag, input, &poly1305Key)
}
func (c *chaCha20Poly1305) poly1305Old(tag *[16]byte, nonce, ciphertext, additionalData []byte) {
input := make([]byte, 0, len(additionalData)+8+len(ciphertext)+8)
input = append(input, additionalData...)
input, out := sliceForAppend(input, 8)
binary.LittleEndian.PutUint64(out, uint64(len(additionalData)))
input = append(input, ciphertext...)
input, out = sliceForAppend(input, 8)
binary.LittleEndian.PutUint64(out, uint64(len(ciphertext)))
var poly1305Key [32]byte
chaCha20(poly1305Key[:], poly1305Key[:], c.key[:], nonce, 0)
poly1305.Sum(tag, input, &poly1305Key)
}
func (c *chaCha20Poly1305) Seal(dst, nonce, plaintext, additionalData []byte) []byte {
if len(nonce) != c.NonceSize() {
panic("Bad nonce length")
}
ret, out := sliceForAppend(dst, len(plaintext)+16)
chaCha20(out[:len(plaintext)], plaintext, c.key[:], nonce, 1)
var tag [16]byte
if c.oldMode {
c.poly1305Old(&tag, nonce, out[:len(plaintext)], additionalData)
} else {
c.poly1305(&tag, nonce, out[:len(plaintext)], additionalData)
}
copy(out[len(plaintext):], tag[:])
return ret
}
func (c *chaCha20Poly1305) Open(dst, nonce, ciphertext, additionalData []byte) ([]byte, error) {
if len(nonce) != c.NonceSize() {
panic("Bad nonce length")
}
if len(ciphertext) < 16 {
return nil, errors.New("chacha20: message authentication failed")
}
plaintextLen := len(ciphertext) - 16
var tag [16]byte
if c.oldMode {
c.poly1305Old(&tag, nonce, ciphertext[:plaintextLen], additionalData)
} else {
c.poly1305(&tag, nonce, ciphertext[:plaintextLen], additionalData)
}
if subtle.ConstantTimeCompare(tag[:], ciphertext[plaintextLen:]) != 1 {
return nil, errors.New("chacha20: message authentication failed")
}
ret, out := sliceForAppend(dst, plaintextLen)
chaCha20(out, ciphertext[:plaintextLen], c.key[:], nonce, 1)
return ret, nil
}