app/vppn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 8 Wiki Activity

Cleanup

Browse Source
This commit is contained in:
jdl 2025-01-04 13:52:50 +01:00
parent 36172bf310
commit 55f63043ee
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
node/supervisor.go
View File

@ -107,14 +107,10 @@ type peerStateData struct {
// ---------------------------------------------------------------------------- // ----------------------------------------------------------------------------
func (s *peerStateData) sendControlPacket(pkt interface{ Marshal([]byte) []byte }) { func (s *peerStateData) sendControlPacket(pkt interface{ Marshal([]byte) []byte }) {
s.limiter.Limit() s._sendControlPacket(pkt, s.staged)
_sendControlPacket(pkt, s.staged, s.buf1, s.buf2)
} }
func (s *peerStateData) sendControlPacketTo( func (s *peerStateData) sendControlPacketTo(pkt interface{ Marshal([]byte) []byte }, addr netip.AddrPort) {
pkt interface{ Marshal([]byte) []byte },
addr netip.AddrPort,
) {
if !addr.IsValid() { if !addr.IsValid() {
s.logf("ERROR: Attepted to send packet to invalid address: %v", addr) s.logf("ERROR: Attepted to send packet to invalid address: %v", addr)
return return
@ -122,7 +118,14 @@ func (s *peerStateData) sendControlPacketTo(
route := s.staged route := s.staged
route.Direct = true route.Direct = true
route.RemoteAddr = addr route.RemoteAddr = addr
s.limiter.Limit() s._sendControlPacket(pkt, route)
}
func (s *peerStateData) _sendControlPacket(pkt interface{ Marshal([]byte) []byte }, route peerRoute) {
if err := s.limiter.Limit(); err != nil {
s.logf("Not sending control packet: rate limited.") // Shouldn't happen.
return
}
_sendControlPacket(pkt, route, s.buf1, s.buf2) _sendControlPacket(pkt, route, s.buf1, s.buf2)
} }