wireguard-go/noise-types.go

81 lines
1.7 KiB
Go
Raw Normal View History

/* SPDX-License-Identifier: GPL-2.0
*
* Copyright (C) 2017-2018 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
2018-05-18 22:34:56 +00:00
* Copyright (C) 2017-2018 Mathias N. Hall-Andersen <mathias@hall-andersen.dk>.
*/
package main
import (
2017-08-04 14:15:53 +00:00
"crypto/subtle"
"encoding/hex"
"errors"
"golang.org/x/crypto/chacha20poly1305"
)
const (
NoisePublicKeySize = 32
NoisePrivateKeySize = 32
)
type (
2017-06-23 11:45:32 +00:00
NoisePublicKey [NoisePublicKeySize]byte
NoisePrivateKey [NoisePrivateKeySize]byte
NoiseSymmetricKey [chacha20poly1305.KeySize]byte
2017-06-23 11:45:32 +00:00
NoiseNonce uint64 // padded to 12-bytes
)
2017-06-01 19:31:30 +00:00
func loadExactHex(dst []byte, src string) error {
slice, err := hex.DecodeString(src)
if err != nil {
return err
}
2017-06-01 19:31:30 +00:00
if len(slice) != len(dst) {
2018-05-13 17:33:41 +00:00
return errors.New("hex string does not fit the slice")
}
2017-06-01 19:31:30 +00:00
copy(dst, slice)
return nil
}
func (key NoisePrivateKey) IsZero() bool {
2017-08-04 14:15:53 +00:00
var zero NoisePrivateKey
return key.Equals(zero)
}
func (key NoisePrivateKey) Equals(tar NoisePrivateKey) bool {
return subtle.ConstantTimeCompare(key[:], tar[:]) == 1
}
2017-06-01 19:31:30 +00:00
func (key *NoisePrivateKey) FromHex(src string) error {
return loadExactHex(key[:], src)
}
func (key NoisePrivateKey) ToHex() string {
return hex.EncodeToString(key[:])
}
2017-06-01 19:31:30 +00:00
func (key *NoisePublicKey) FromHex(src string) error {
return loadExactHex(key[:], src)
}
func (key NoisePublicKey) ToHex() string {
return hex.EncodeToString(key[:])
}
2017-08-04 14:15:53 +00:00
func (key NoisePublicKey) IsZero() bool {
var zero NoisePublicKey
return key.Equals(zero)
}
func (key NoisePublicKey) Equals(tar NoisePublicKey) bool {
return subtle.ConstantTimeCompare(key[:], tar[:]) == 1
}
2017-06-01 19:31:30 +00:00
func (key *NoiseSymmetricKey) FromHex(src string) error {
return loadExactHex(key[:], src)
}
2017-06-01 19:31:30 +00:00
func (key NoiseSymmetricKey) ToHex() string {
return hex.EncodeToString(key[:])
}