mirror of https://github.com/XTLS/Xray-core
49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
package quic
|
|
|
|
import (
|
|
"crypto/aes"
|
|
"crypto/cipher"
|
|
"crypto/sha256"
|
|
"errors"
|
|
|
|
"github.com/xtls/xray-core/common"
|
|
"github.com/xtls/xray-core/common/protocol"
|
|
"github.com/xtls/xray-core/transport/internet"
|
|
"golang.org/x/crypto/chacha20poly1305"
|
|
)
|
|
|
|
func getAuth(config *Config) (cipher.AEAD, error) {
|
|
security := config.Security.GetSecurityType()
|
|
if security == protocol.SecurityType_NONE {
|
|
return nil, nil
|
|
}
|
|
|
|
salted := []byte(config.Key + "xray-quic-salt")
|
|
key := sha256.Sum256(salted)
|
|
|
|
if security == protocol.SecurityType_AES128_GCM {
|
|
block, err := aes.NewCipher(key[:16])
|
|
common.Must(err)
|
|
return cipher.NewGCM(block)
|
|
}
|
|
|
|
if security == protocol.SecurityType_CHACHA20_POLY1305 {
|
|
return chacha20poly1305.New(key[:])
|
|
}
|
|
|
|
return nil, errors.New("unsupported security type")
|
|
}
|
|
|
|
func getHeader(config *Config) (internet.PacketHeader, error) {
|
|
if config.Header == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
msg, err := config.Header.GetInstance()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return internet.CreatePacketHeader(msg)
|
|
}
|