google永远的神

tcpip/stack/linkaddresscache.go

@@ -0,0 +1,45 @@
+package stack
+
+import (
+	"sync"
+	"time"
+
+	"github.com/impact-eintr/netstack/tcpip"
+)
+
+const linkAddrCacheSize = 512 // 最大缓存条目
+
+// 是一个固定大小的缓存，将 IP 地址映射到链接地址
+// 条目存储在环形缓冲区中，最旧的条目首先被替换。
+// 这个结构体对于并发使用是安全的
+type linkAddrCache struct {
+	// 缓存条目的有效期
+	ageLimit time.Duration
+	// 等待链接请求解析地址的时间
+	resolutionTimeout time.Duration
+	// 地址在失败前尝试解析的次数
+	resolutionAttempts int
+
+	mu      sync.Mutex
+	cache   map[tcpip.FullAddress]*linkAddrEntry
+	next    int // 下一个可用条目的数组索引
+	entries [linkAddrCacheSize]linkAddrEntry
+}
+
+// linkAddrCache 中的一个条目
+type linkAddrEntry struct {
+	addr       tcpip.FullAddress
+	linkAddr   tcpip.LinkAddress
+	expiration time.Time
+	s          entryState
+}
+
+// entryState 控制缓存中单个条目的状态
+type entryState int
+
+const (
+	incomplete entryState = iota
+	ready
+	failed
+	expired // 失效的
+)

tcpip/stack/nic.go

@@ -1,5 +1,11 @@
 package stack
 
+import (
+	"sync"
+
+	"github.com/impact-eintr/netstack/tcpip"
+)
+
 type referencedNetworkEndpoint struct {
 	//ilist.Entry
 	//refs     int32
@@ -9,3 +15,25 @@ type referencedNetworkEndpoint struct {
 	//linkCache LinkAddressCache
 	//holdsInserRef bool
 }
+
+// 代表一个网卡对象
+type NIC struct {
+	stack *Stack
+	// 每个网卡唯一的标识号
+	id tcpip.NICID
+	// 网卡名 可有可无
+	name string
+	// 链路层端
+	linkEP LinkEndpoint
+	// 传输层的解复用
+	demux *transportDemuxer
+
+	mu          sync.RWMutex
+	spoofing    bool
+	promiscuous bool
+	primary     map[tcpip.NetworkProtocolNumber]*ilist.List
+	// 网络层端的记录
+	endpoints map[tcpip.NetworkEndpointID]*referencedNetworkEndpoint
+	// 子网的记录
+	subnets []tcpip.Subnet
+}

tcpip/stack/registration.go

@@ -46,13 +46,6 @@ const (
 	CapabilityLoopback
 )
 
-// 包含网络协议栈用于在 数据链路层 处理数据包后将数据包传送到适当网络端点的方法。
-type NetworkDispatcher interface {
-	// deliver 递送
-	DeliverNetworkPacket(linkEP LinkEndpoint, dstLinkAddr, srcLinkAddr tcpip.LinkAddress,
-		protocol tcpip.NetworkProtocolNumber, vv buffer.VectorisedView)
-}
-
 var (
 	// 传输层协议的注册存储结构
 	//transportProtocols = make(map[string]TransportProtocolFactory)
@@ -77,3 +70,29 @@ func RegisterLinkEndpoint(linkEP LinkEndpoint) tcpip.LinkEndpointID {
 	return v
 
 }
+
+type TransportProtocol interface {
+}
+
+type TransportEndpointID struct {
+	LocalPort     uint16
+	LocalAddress  tcpip.Address
+	RemotePort    uint16
+	RemoteAddress tcpip.Address
+}
+
+type NetworkProtocol interface {
+	//Number() tcpip.NetworkProtocolNumber
+	//MinimumPacketSize() int
+	//ParseAddresses(v buffer.View) (src, dst tcpip.Address)
+}
+
+// 包含网络协议栈用于在 数据链路层 处理数据包后将数据包传送到适当网络端点的方法。
+type NetworkDispatcher interface {
+	// deliver 递送
+	DeliverNetworkPacket(linkEP LinkEndpoint, dstLinkAddr, srcLinkAddr tcpip.LinkAddress,
+		protocol tcpip.NetworkProtocolNumber, vv buffer.VectorisedView)
+}
+
+type LinkAddressResolver interface {
+}

tcpip/stack/stack.go

@@ -0,0 +1,48 @@
+package stack
+
+import (
+	"sync"
+
+	"github.com/impact-eintr/netstack/tcpip"
+	"github.com/impact-eintr/netstack/tcpip/buffer"
+	"github.com/impact-eintr/netstack/tcpip/ports"
+)
+
+type transportProtocolState struct {
+	proto          TransportProtocol
+	defaultHandler func(*Route, TransportEndpointID, buffer.VectorisedView) bool
+}
+
+type Stack struct {
+	transportProtocols map[tcpip.TransportProtocolNumber]*transportProtocolState
+	networkProtocols   map[tcpip.NetworkProtocolNumber]NetworkProtocol
+	linkAddrResolvers  map[tcpip.NetworkProtocolNumber]LinkAddressResolver
+
+	demux *transportDemuxer
+
+	stats tcpip.Stats
+
+	linkAddrCache *linkAddrCache
+
+	mu         sync.RWMutex
+	nics       map[tcpip.NICID]*NIC
+	forwarding bool
+
+	routeTable []tcpip.Route
+
+	*ports.PortManager
+	tcpProbeFunc TCPProbeFunc
+	clock        tcpip.Clock
+}
+
+type Options struct {
+	Clock tcpip.Clock
+	Stats tcpip.Stats
+}
+
+// TCPProbeFunc 是要传递给 stack.AddTCPProbe 的 TCP 探测函数的预期函数类型
+type TCPProbeFunc func(s TCPEndpointState)
+
+// TCPEndpointState 是 TCP 端点内部状态的副本
+type TCPEndpointState struct {
+}

tcpip/stack/transport_demuxer.go

@@ -0,0 +1,25 @@
+package stack
+
+import (
+	"sync"
+
+	"github.com/impact-eintr/netstack/tcpip"
+)
+
+// 解复用针对传输端点的数据包 在他们被网络层解析之后
+// 它执行两级解复用 首先基于网络协议和传输协议 然后基于端点ID
+type transportDemuxer struct {
+	protocol map[protocolIDs]*transportEndpoints
+}
+
+// 管理给定协议的所有端点
+type transportEndpoints struct {
+	mu        sync.RWMutex
+	endpoints map[TransportEndpointID]*transportEndpoints
+}
+
+// 网络层协议号和传输层协议号的组合 当作分流器的key值
+type protocolIDs struct {
+	network   tcpip.NetworkProtocolNumber
+	transport tcpip.TransportProtocolNumber
+}