fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-29 23:30:37 +08:00
Code Activity

add mtu ctrl

Browse Source
This commit is contained in:
fumiama
2022-01-01 17:46:36 +08:00
parent 55f03d21f3
commit 8a43e379a5
1 changed files with 36 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
gold/link/me.go
View File

@@ -2,7 +2,6 @@ package link
import ( import (
"encoding/binary" "encoding/binary"
"io"
"net" "net"
"strconv" "strconv"
"sync" "sync"
@@ -125,9 +124,9 @@ func (m *Me) ListenFromNIC() {
continue continue
} }
packet = packet[:n] packet = packet[:n]
n, rem := m.send(m.nic, packet) n, rem := m.sendAllSameDst(packet)
for len(rem) > 20 && n > 0 { for len(rem) > 20 && n > 0 {
n, rem = m.send(m.nic, rem) n, rem = m.sendAllSameDst(rem)
} }
if len(rem) > 0 { if len(rem) > 0 {
logrus.Debugln("[me] remain", len(rem), "bytes to send") logrus.Debugln("[me] remain", len(rem), "bytes to send")
@@ -143,36 +142,45 @@ func (m *Me) ListenFromNIC() {
} }
} }
func (m *Me) send(nc io.Reader, packet []byte) (n int, rem []byte) { type PacketID [2]byte
if !waterutil.IsIPv4(packet) {
if waterutil.IsIPv6(packet) { func newpacketid(packet []byte) PacketID {
n = int(binary.BigEndian.Uint16(packet[4:6])) + 40 return waterutil.IPv4Identification(packet)
if n > len(packet) {
rem = packet
logrus.Warnln("[me] skip to send", len(packet), "bytes ipv6 packet head")
} else {
rem = packet[n:]
logrus.Warnln("[me] skip to send", n, "bytes ipv6 packet")
} }
func (p PacketID) issame(packet []byte) bool {
return p == waterutil.IPv4Identification(packet)
}
func (m *Me) sendAllSameDst(packet []byte) (n int, rem []byte) {
rem = packet
if !waterutil.IsIPv4(packet) {
for waterutil.IsIPv6(rem) {
pktl := int(binary.BigEndian.Uint16(packet[4:6])) + 40
if pktl > len(rem) {
return return
} }
logrus.Warnln("[me] skip to send", len(packet), "bytes non-ipv4/v6 packet") n += pktl
rem = packet[n:]
}
if !waterutil.IsIPv4(rem) {
logrus.Warnln("[me] skip to send", len(rem), "bytes non-ipv4/v6 packet")
return len(packet), nil return len(packet), nil
} }
totl := waterutil.IPv4TotalLength(packet) }
if int(totl) > len(packet) { p := newpacketid(rem)
buf := make([]byte, int(totl)) for len(rem) > 20 && p.issame(rem) {
copy(buf, packet) totl := waterutil.IPv4TotalLength(rem)
cnt, err := m.nic.Read(buf[len(packet):]) if int(totl) > len(rem) {
if err != nil { break
rem = packet }
n += int(totl)
rem = packet[n:]
}
if n == 0 {
return return
} }
packet = buf[:cnt+len(packet)] packet = packet[:n]
}
rem = packet[totl:]
packet = packet[:totl]
n = int(totl)
dst := waterutil.IPv4Destination(packet) dst := waterutil.IPv4Destination(packet)
logrus.Debugln("[me] sending", len(packet), "bytes packet from :"+strconv.Itoa(int(m.SrcPort())), "to", dst.String()+":"+strconv.Itoa(int(m.DstPort()))) logrus.Debugln("[me] sending", len(packet), "bytes packet from :"+strconv.Itoa(int(m.SrcPort())), "to", dst.String()+":"+strconv.Itoa(int(m.DstPort())))
lnk := m.router.NextHop(dst.String()) lnk := m.router.NextHop(dst.String())