void/amneziawg-go
2
0
Fork 0
mirror of https://github.com/amnezia-vpn/amneziawg-go.git synced 2025-07-28 07:52:50 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

send junk in one batch with other

Browse source
This commit is contained in:
Mark Puha 2023-09-13 07:57:16 +02:00
parent c1670b484d
commit ba6bb0bb8f
1 changed files with 11 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
device/send.go
View file

@ -9,7 +9,6 @@ import (
"bytes"
"encoding/binary"
"errors"
"fmt"
"math/rand"
"net"
"os"
@ -125,16 +124,18 @@ func (peer *Peer) SendHandshakeInitiation(isRetry bool) error {
)
return err
}
var sendBuffer [][]byte
// so only packet processed for cookie generation
var junkedHeader []byte
if peer.device.isAdvancedSecurityOn() {
peer.device.aSecMux.RLock()
err = peer.sendJunkPackets()
junks, err := peer.createJunkPackets()
if err != nil {
peer.device.aSecMux.RUnlock()
peer.device.log.Errorf("%v - %v", peer, err)
return err
}
sendBuffer = append(sendBuffer, junks...)
if peer.device.aSecCfg.initPacketJunkSize != 0 {
buf := make([]byte, 0, peer.device.aSecCfg.initPacketJunkSize)
writer := bytes.NewBuffer(buf[:0])
@ -158,7 +159,9 @@ func (peer *Peer) SendHandshakeInitiation(isRetry bool) error {
peer.timersAnyAuthenticatedPacketTraversal()
peer.timersAnyAuthenticatedPacketSent()
err = peer.SendBuffers([][]byte{junkedHeader})
sendBuffer = append(sendBuffer, junkedHeader)
err = peer.SendBuffers(sendBuffer)
if err != nil {
peer.device.log.Errorf(
"%v - Failed to send handshake initiation: %v",
@ -481,9 +484,9 @@ top:
}
}
func (peer *Peer) sendJunkPackets() error {
func (peer *Peer) createJunkPackets() ([][]byte, error) {
if peer.device.aSecCfg.junkPacketCount == 0 {
return nil
return nil, nil
}
junks := make([][]byte, 0, peer.device.aSecCfg.junkPacketCount)
@ -499,15 +502,11 @@ func (peer *Peer) sendJunkPackets() error {
peer,
err,
)
return err
return nil, err
}
junks = append(junks, junk)
}
err := peer.SendBuffers(junks)
if err != nil {
return fmt.Errorf("failed to send junks: %v", err)
}
return nil
return junks, nil
}
func (peer *Peer) FlushStagedPackets() {