fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-27 06:10:26 +08:00
Code Activity

feat: add ztsd for data compressing

Browse Source
This commit is contained in:
源文雨
2023-08-04 12:03:49 +08:00
parent 96e31e22e9
commit 47d8e35d19
8 changed files with 44 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
gold/link/send.go
View File

@@ -1,12 +1,16 @@
package link
import (
"bytes"
"errors"
"fmt"
"io"
"math/rand"
"sync/atomic"
"github.com/fumiama/WireGold/gold/head"
"github.com/fumiama/WireGold/helper"
"github.com/klauspost/compress/zstd"
"github.com/sirupsen/logrus"
)
@@ -21,21 +25,13 @@ func (l *Link) WriteAndPut(p *head.Packet, istransfer bool) (n int, err error) {
logrus.Debugln("[send] mtu:", mtu, ", count:", sndcnt, ", additional data:", uint16(sndcnt))
if len(p.Data) <= int(mtu) {
if !istransfer {
p.FillHash()
if l.aead != nil {
p.Data = l.EncodePreshared(uint16(sndcnt), p.Data)
}
p.Data = l.Encode(teatype, p.Data)
l.encrypt(p, uint16(sndcnt), teatype)
}
defer p.Put()
return l.write(p, teatype, uint16(sndcnt), mtu, uint32(len(p.Data)), 0, istransfer, false)
}
if !istransfer {
p.FillHash()
if l.aead != nil {
p.Data = l.EncodePreshared(uint16(sndcnt), p.Data)
}
p.Data = l.Encode(teatype, p.Data)
l.encrypt(p, uint16(sndcnt), teatype)
}
data := p.Data
ttl := p.TTL
@@ -62,6 +58,22 @@ func (l *Link) WriteAndPut(p *head.Packet, istransfer bool) (n int, err error) {
return n, err
}
func (l *Link) encrypt(p *head.Packet, sndcnt uint16, teatype uint8) {
p.FillHash()
if l.usezstd {
w := helper.SelectWriter()
defer helper.PutWriter(w)
enc, _ := zstd.NewWriter(w, zstd.WithEncoderLevel(zstd.SpeedFastest))
defer enc.Close()
_, _ = io.Copy(enc, bytes.NewReader(p.Data))
p.Data = w.Bytes()
}
if l.aead != nil {
p.Data = l.EncodePreshared(sndcnt, p.Data)
}
p.Data = l.Encode(teatype, p.Data)
}
// write 向 peer 发一个包
func (l *Link) write(p *head.Packet, teatype uint8, additional, mtu uint16, datasz uint32, offset uint16, istransfer, hasmore bool) (n int, err error) {
var d []byte