mirror of
https://github.com/fumiama/WireGold.git
synced 2026-06-05 07:50:24 +08:00
66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
package link
|
||
|
||
import (
|
||
"net"
|
||
"unsafe"
|
||
|
||
tea "github.com/fumiama/gofastTEA"
|
||
)
|
||
|
||
var (
|
||
// 本机私钥
|
||
// 利用 Curve25519 生成
|
||
// https://pkg.go.dev/golang.org/x/crypto/curve25519
|
||
// https://www.zhihu.com/question/266758647
|
||
privKey [32]byte
|
||
// 本机虚拟 ip
|
||
me net.IP
|
||
// 本机 endpoint
|
||
myend *net.UDPAddr
|
||
)
|
||
|
||
// SetMyself 设置本机参数
|
||
func SetMyself(privateKey [32]byte, myIP string, myEndpoint string) {
|
||
privKey = privateKey
|
||
var err error
|
||
myend, err = net.ResolveUDPAddr("udp", myEndpoint)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
me = net.ParseIP(myIP)
|
||
myconn, err = listen()
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
}
|
||
|
||
// Encode 使用 TEA 加密
|
||
func (l *Link) Encode(b []byte) (eb []byte, err error) {
|
||
if b == nil {
|
||
return
|
||
}
|
||
if l.key == nil {
|
||
eb = b
|
||
} else {
|
||
// 在此处填写加密逻辑，密钥是l.key，输入是b，输出是eb
|
||
// 不用写return，直接赋值给eb即可
|
||
eb = (*tea.TEA)(unsafe.Pointer(&privKey)).Encrypt(b)
|
||
}
|
||
return
|
||
}
|
||
|
||
// Decode 使用 TEA 解密
|
||
func (l *Link) Decode(b []byte) (db []byte, err error) {
|
||
if b == nil {
|
||
return
|
||
}
|
||
if l.key == nil {
|
||
db = b
|
||
} else {
|
||
// 在此处填写解密逻辑，密钥是l.key，输入是b，输出是db
|
||
// 不用写return，直接赋值给db即可
|
||
db = (*tea.TEA)(unsafe.Pointer(&privKey)).Decrypt(b)
|
||
}
|
||
return
|
||
}
|