mirror of
https://github.com/impact-eintr/netstack.git
synced 2025-10-06 05:16:50 +08:00
wsl code edit
This commit is contained in:
impact-eintr
@@ -1,160 +1,160 @@
|
||||
package stack_test
|
||||
|
||||
import (
|
||||
"log"
|
||||
"netstack/tcpip"
|
||||
"netstack/tcpip/buffer"
|
||||
"netstack/tcpip/link/channel"
|
||||
"netstack/tcpip/stack"
|
||||
"testing"
|
||||
)
|
||||
|
||||
const (
|
||||
fakeNetHeaderLen = 12
|
||||
defaultMTU = 65536
|
||||
)
|
||||
|
||||
type fakeNetworkEndpoint struct {
|
||||
nicid tcpip.NICID
|
||||
id stack.NetworkEndpointID
|
||||
proto *fakeNetworkProtocol
|
||||
dispatcher stack.TransportDispatcher
|
||||
linkEP stack.LinkEndpoint
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) DefaultTTL() uint8 {
|
||||
return 123
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) MTU() uint32 {
|
||||
return f.linkEP.MTU() - uint32(f.MaxHeaderLength())
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) Capabilities() stack.LinkEndpointCapabilities {
|
||||
return f.linkEP.Capabilities()
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) MaxHeaderLength() uint16 {
|
||||
return f.linkEP.MaxHeaderLength() + fakeNetHeaderLen
|
||||
}
|
||||
func (f *fakeNetworkEndpoint) WritePacket(r *stack.Route, hdr buffer.Prependable, payload buffer.VectorisedView,
|
||||
protocol tcpip.TransportProtocolNumber, ttl uint8) *tcpip.Error {
|
||||
b := hdr.Prepend(fakeNetHeaderLen)
|
||||
copy(b[:4], []byte(r.RemoteAddress))
|
||||
copy(b[4:8], []byte(f.id.LocalAddress))
|
||||
b[8] = byte(protocol)
|
||||
log.Println("写入网络层数据 下一层去往链路层", b, payload)
|
||||
|
||||
return f.linkEP.WritePacket(r, hdr, payload, 114514)
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) ID() *stack.NetworkEndpointID {
|
||||
return &f.id
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) NICID() tcpip.NICID {
|
||||
return f.nicid
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) HandlePacket(r *stack.Route, vv buffer.VectorisedView) {
|
||||
log.Println("执行这个函数 接下来它会去向传输层分发数据")
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) Close() {}
|
||||
|
||||
// dst|src|payload
|
||||
type fakeNetworkProtocol struct{}
|
||||
|
||||
func (f *fakeNetworkProtocol) Number() tcpip.NetworkProtocolNumber {
|
||||
return 114514
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) NewEndpoint(nicid tcpip.NICID, addr tcpip.Address, linkAddrCache stack.LinkAddressCache,
|
||||
dispatcher stack.TransportDispatcher, linkEP stack.LinkEndpoint) (stack.NetworkEndpoint, *tcpip.Error) {
|
||||
return &fakeNetworkEndpoint{
|
||||
nicid: nicid,
|
||||
id: stack.NetworkEndpointID{addr},
|
||||
proto: f,
|
||||
dispatcher: dispatcher,
|
||||
linkEP: linkEP,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) MinimumPacketSize() int {
|
||||
return fakeNetHeaderLen
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) ParseAddresses(v buffer.View) (src, dst tcpip.Address) {
|
||||
return tcpip.Address(v[4:8]), tcpip.Address(v[0:4])
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) SetOption(option interface{}) *tcpip.Error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) Option(option interface{}) *tcpip.Error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
stack.RegisterNetworkProtocolFactory("fakeNet", func() stack.NetworkProtocol {
|
||||
return &fakeNetworkProtocol{}
|
||||
})
|
||||
}
|
||||
|
||||
func TestStackBase(t *testing.T) {
|
||||
|
||||
myStack := stack.New([]string{"fakeNet"}, nil, stack.Options{})
|
||||
id1, ep1 := channel.New(10, defaultMTU, "00:15:5d:26:d7:a1") // 这是一个物理设备
|
||||
|
||||
if err := myStack.CreateNIC(1, id1); err != nil { // 将上面的物理设备抽象成我们的网卡对象
|
||||
panic(err)
|
||||
}
|
||||
myStack.AddAddress(1, 114514, "\x0a\xff\x01\x01") // 给网卡对象绑定一个IP地址 可以绑定多个
|
||||
|
||||
id2, _ := channel.New(10, defaultMTU, "50:5B:C2:D0:96:57") // 这是一个物理设备
|
||||
if err := myStack.CreateNIC(2, id2); err != nil { // 将上面的物理设备抽象成我们的网卡对象
|
||||
panic(err)
|
||||
}
|
||||
myStack.AddAddress(2, 114514, "\x0a\xff\x01\x02") // 给网卡对象绑定一个IP地址 可以绑定多个
|
||||
|
||||
buf := buffer.NewView(30)
|
||||
for i := range buf {
|
||||
buf[i] = 0
|
||||
}
|
||||
// dst 10.255.1.2
|
||||
buf[0] = '\x0a'
|
||||
buf[1] = '\xff'
|
||||
buf[2] = '\x01'
|
||||
buf[3] = '\x02'
|
||||
// src 10.255.1.1
|
||||
buf[4] = '\x0a'
|
||||
buf[5] = '\xff'
|
||||
buf[6] = '\x01'
|
||||
buf[7] = '\x01'
|
||||
|
||||
myStack.SetRouteTable([]tcpip.Route{
|
||||
{"\x01", "\x01", "\x00", 1},
|
||||
{"\x00", "\x01", "\x00", 2},
|
||||
})
|
||||
|
||||
sendTo(t, myStack, tcpip.Address("\x0a\xff\x01\x02"))
|
||||
|
||||
//log.Println(ep1.Drain())
|
||||
p := <-ep1.C
|
||||
log.Println(p)
|
||||
}
|
||||
|
||||
func sendTo(t *testing.T, s *stack.Stack, addr tcpip.Address) {
|
||||
r, err := s.FindRoute(0, "", addr, 114514)
|
||||
if err != nil {
|
||||
t.Fatalf("FindRoute failed: %v", err)
|
||||
}
|
||||
defer r.Release()
|
||||
|
||||
hdr := buffer.NewPrependable(int(r.MaxHeaderLength()))
|
||||
if err := r.WritePacket(hdr, buffer.VectorisedView{}, 10086, 123); err != nil {
|
||||
t.Errorf("WritePacket failed: %v", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
package stack_test
|
||||
|
||||
import (
|
||||
"log"
|
||||
"netstack/tcpip"
|
||||
"netstack/tcpip/buffer"
|
||||
"netstack/tcpip/link/channel"
|
||||
"netstack/tcpip/stack"
|
||||
"testing"
|
||||
)
|
||||
|
||||
const (
|
||||
fakeNetHeaderLen = 12
|
||||
defaultMTU = 65536
|
||||
)
|
||||
|
||||
type fakeNetworkEndpoint struct {
|
||||
nicid tcpip.NICID
|
||||
id stack.NetworkEndpointID
|
||||
proto *fakeNetworkProtocol
|
||||
dispatcher stack.TransportDispatcher
|
||||
linkEP stack.LinkEndpoint
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) DefaultTTL() uint8 {
|
||||
return 123
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) MTU() uint32 {
|
||||
return f.linkEP.MTU() - uint32(f.MaxHeaderLength())
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) Capabilities() stack.LinkEndpointCapabilities {
|
||||
return f.linkEP.Capabilities()
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) MaxHeaderLength() uint16 {
|
||||
return f.linkEP.MaxHeaderLength() + fakeNetHeaderLen
|
||||
}
|
||||
func (f *fakeNetworkEndpoint) WritePacket(r *stack.Route, hdr buffer.Prependable, payload buffer.VectorisedView,
|
||||
protocol tcpip.TransportProtocolNumber, ttl uint8) *tcpip.Error {
|
||||
b := hdr.Prepend(fakeNetHeaderLen)
|
||||
copy(b[:4], []byte(r.RemoteAddress))
|
||||
copy(b[4:8], []byte(f.id.LocalAddress))
|
||||
b[8] = byte(protocol)
|
||||
log.Println("写入网络层数据 下一层去往链路层", b, payload)
|
||||
|
||||
return f.linkEP.WritePacket(r, hdr, payload, 114514)
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) ID() *stack.NetworkEndpointID {
|
||||
return &f.id
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) NICID() tcpip.NICID {
|
||||
return f.nicid
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) HandlePacket(r *stack.Route, vv buffer.VectorisedView) {
|
||||
log.Println("执行这个函数 接下来它会去向传输层分发数据")
|
||||
}
|
||||
|
||||
func (f *fakeNetworkEndpoint) Close() {}
|
||||
|
||||
// dst|src|payload
|
||||
type fakeNetworkProtocol struct{}
|
||||
|
||||
func (f *fakeNetworkProtocol) Number() tcpip.NetworkProtocolNumber {
|
||||
return 114514
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) NewEndpoint(nicid tcpip.NICID, addr tcpip.Address, linkAddrCache stack.LinkAddressCache,
|
||||
dispatcher stack.TransportDispatcher, linkEP stack.LinkEndpoint) (stack.NetworkEndpoint, *tcpip.Error) {
|
||||
return &fakeNetworkEndpoint{
|
||||
nicid: nicid,
|
||||
id: stack.NetworkEndpointID{addr},
|
||||
proto: f,
|
||||
dispatcher: dispatcher,
|
||||
linkEP: linkEP,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) MinimumPacketSize() int {
|
||||
return fakeNetHeaderLen
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) ParseAddresses(v buffer.View) (src, dst tcpip.Address) {
|
||||
return tcpip.Address(v[4:8]), tcpip.Address(v[0:4])
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) SetOption(option interface{}) *tcpip.Error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *fakeNetworkProtocol) Option(option interface{}) *tcpip.Error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
stack.RegisterNetworkProtocolFactory("fakeNet", func() stack.NetworkProtocol {
|
||||
return &fakeNetworkProtocol{}
|
||||
})
|
||||
}
|
||||
|
||||
func TestStackBase(t *testing.T) {
|
||||
|
||||
myStack := stack.New([]string{"fakeNet"}, nil, stack.Options{})
|
||||
id1, ep1 := channel.New(10, defaultMTU, "00:15:5d:26:d7:a1") // 这是一个物理设备
|
||||
|
||||
if err := myStack.CreateNIC(1, id1); err != nil { // 将上面的物理设备抽象成我们的网卡对象
|
||||
panic(err)
|
||||
}
|
||||
myStack.AddAddress(1, 114514, "\x0a\xff\x01\x01") // 给网卡对象绑定一个IP地址 可以绑定多个
|
||||
|
||||
id2, _ := channel.New(10, defaultMTU, "50:5B:C2:D0:96:57") // 这是一个物理设备
|
||||
if err := myStack.CreateNIC(2, id2); err != nil { // 将上面的物理设备抽象成我们的网卡对象
|
||||
panic(err)
|
||||
}
|
||||
myStack.AddAddress(2, 114514, "\x0a\xff\x01\x02") // 给网卡对象绑定一个IP地址 可以绑定多个
|
||||
|
||||
buf := buffer.NewView(30)
|
||||
for i := range buf {
|
||||
buf[i] = 0
|
||||
}
|
||||
// dst 10.255.1.2
|
||||
buf[0] = '\x0a'
|
||||
buf[1] = '\xff'
|
||||
buf[2] = '\x01'
|
||||
buf[3] = '\x02'
|
||||
// src 10.255.1.1
|
||||
buf[4] = '\x0a'
|
||||
buf[5] = '\xff'
|
||||
buf[6] = '\x01'
|
||||
buf[7] = '\x01'
|
||||
|
||||
myStack.SetRouteTable([]tcpip.Route{
|
||||
{"\x01", "\x01", "\x00", 1},
|
||||
{"\x00", "\x01", "\x00", 2},
|
||||
})
|
||||
|
||||
sendTo(t, myStack, tcpip.Address("\x0a\xff\x01\x02"))
|
||||
|
||||
//log.Println(ep1.Drain())
|
||||
p := <-ep1.C
|
||||
log.Println(p)
|
||||
}
|
||||
|
||||
func sendTo(t *testing.T, s *stack.Stack, addr tcpip.Address) {
|
||||
r, err := s.FindRoute(0, "", addr, 114514)
|
||||
if err != nil {
|
||||
t.Fatalf("FindRoute failed: %v", err)
|
||||
}
|
||||
defer r.Release()
|
||||
|
||||
hdr := buffer.NewPrependable(int(r.MaxHeaderLength()))
|
||||
if err := r.WritePacket(hdr, buffer.VectorisedView{}, 10086, 123); err != nil {
|
||||
t.Errorf("WritePacket failed: %v", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user