Cloak/internal/multiplex/crypto.go

90 lines
1.8 KiB
Go

package multiplex
import (
"crypto/aes"
"crypto/cipher"
"crypto/rand"
"golang.org/x/crypto/chacha20poly1305"
)
type Crypto interface {
encrypt([]byte, []byte) ([]byte, error)
decrypt([]byte, []byte) ([]byte, error)
}
type Plain struct{}
func (p *Plain) encrypt(plaintext []byte, nonce []byte) ([]byte, error) {
salt := make([]byte, 16)
rand.Read(salt)
return append(plaintext, salt...), nil
}
func (p *Plain) decrypt(buf []byte, nonce []byte) ([]byte, error) {
return buf[:len(buf)-16], nil
}
type AESGCM struct {
cipher cipher.AEAD
}
func MakeAESGCMCipher(key []byte) (*AESGCM, error) {
c, err := aes.NewCipher(key)
if err != nil {
return nil, err
}
g, err := cipher.NewGCM(c)
if err != nil {
return nil, err
}
ret := AESGCM{
g,
}
return &ret, nil
}
func (a *AESGCM) encrypt(plaintext []byte, nonce []byte) ([]byte, error) {
ciphertext := a.cipher.Seal(nil, nonce, plaintext, nil)
ret := make([]byte, len(plaintext)+16)
copy(ret, ciphertext)
return ret, nil
}
func (a *AESGCM) decrypt(ciphertext []byte, nonce []byte) ([]byte, error) {
plain, err := a.cipher.Open(nil, nonce, ciphertext, nil)
if err != nil {
return nil, err
}
return plain, nil
}
type C20P1305 struct {
cipher cipher.AEAD
}
func MakeCPCipher(key []byte) (*C20P1305, error) {
c, err := chacha20poly1305.New(key)
if err != nil {
return nil, err
}
ret := C20P1305{
c,
}
return &ret, nil
}
func (c *C20P1305) encrypt(plaintext []byte, nonce []byte) ([]byte, error) {
ciphertext := c.cipher.Seal(nil, nonce, plaintext, nil)
ret := make([]byte, len(plaintext)+16)
copy(ret, ciphertext)
return ret, nil
}
func (c *C20P1305) decrypt(ciphertext []byte, nonce []byte) ([]byte, error) {
plain, err := c.cipher.Open(nil, nonce, ciphertext, nil)
if err != nil {
return nil, err
}
return plain, nil
}