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

WIP: duplicate / counter checking. Working well

Browse Source
This commit is contained in:
jdl 2024-12-16 20:55:08 +01:00
parent 62bf956c10
commit a561dd3cce
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
node/conn.go
View File

@ -61,16 +61,20 @@ func (w *connWriter) WriteTo(remoteIP, packetType byte, data []byte) error {
type connReader struct {
*net.UDPConn
localIP byte
counters [256]uint64
dupChecks [256]*dupCheck
lookup func(byte) *peer
}
func newConnReader(conn *net.UDPConn, localIP byte, lookup func(byte) *peer) *connReader {
return &connReader{
r := &connReader{
UDPConn: conn,
localIP: localIP,
lookup: lookup,
}
for i := range r.dupChecks {
r.dupChecks[i] = newDupCheck(0)
}
return r
}
func (r *connReader) Read(buf []byte) (remoteAddr netip.AddrPort, h header, data []byte, err error) {
@ -94,8 +98,8 @@ func (r *connReader) Read(buf []byte) (remoteAddr netip.AddrPort, h header, data
continue // Packet is corrupt.
}
if h.Counter > r.counters[h.SourceIP] {
r.counters[h.SourceIP] = h.Counter
if r.dupChecks[h.SourceIP].IsDup(h.Counter) {
continue
}
return