mirror of
https://github.com/impact-eintr/netstack.git
synced 2025-10-06 13:26:49 +08:00
80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
package main
|
||
|
||
import (
|
||
"flag"
|
||
"log"
|
||
"net"
|
||
"netstack/tcpip"
|
||
"netstack/tcpip/stack"
|
||
"netstack/tcpip/transport/udp"
|
||
"netstack/waiter"
|
||
"os"
|
||
)
|
||
|
||
func main() {
|
||
flag.Parse()
|
||
if len(flag.Args()) != 2 {
|
||
log.Fatal("Usage: ", os.Args[0], "<listen-address> port")
|
||
}
|
||
|
||
log.SetFlags(log.Lshortfile | log.LstdFlags)
|
||
listeAddr := flag.Arg(0)
|
||
portName := flag.Arg(1)
|
||
|
||
Socket(listeAddr + ":" + portName)
|
||
}
|
||
|
||
func Socket(addr string) {
|
||
conn, err := net.Dial("tcp", addr)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
conn.Write([]byte("udp\xc0\xa8\x01\x01\x27\x0f")) // bind udp 192.168.1.1 9999
|
||
conn.Close()
|
||
}
|
||
|
||
//func tcpListen(s *stack.Stack, proto tcpip.NetworkProtocolNumber, localPort int) tcpip.Endpoint {
|
||
// var wq waiter.Queue
|
||
// // 新建一个tcp端
|
||
// ep, err := s.NewEndpoint(tcp.ProtocolNumber, proto, &wq)
|
||
// if err != nil {
|
||
// log.Fatal(err)
|
||
// }
|
||
//
|
||
// // 绑定IP和端口，这里的IP地址为空，表示绑定任何IP
|
||
// // 此时就会调用端口管理器
|
||
// if err := ep.Bind(tcpip.FullAddress{0, "", uint16(localPort)}, nil); err != nil {
|
||
// log.Fatal("Bind failed: ", err)
|
||
// }
|
||
//
|
||
// // 开始监听
|
||
// if err := ep.Listen(10); err != nil {
|
||
// log.Fatal("Listen failed: ", err)
|
||
// }
|
||
//
|
||
// return ep
|
||
//}
|
||
|
||
func udpListen(s *stack.Stack, proto tcpip.NetworkProtocolNumber, localPort int) tcpip.Endpoint {
|
||
var wq waiter.Queue
|
||
// 新建一个udp端
|
||
ep, err := s.NewEndpoint(udp.ProtocolNumber, proto, &wq)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
|
||
// 绑定IP和端口，这里的IP地址为空，表示绑定任何IP
|
||
// 0.0.0.0:9999 这台机器上的所有ip的9999段端口数据都会使用该传输层实现
|
||
// 此时就会调用端口管理器
|
||
if err := ep.Bind(tcpip.FullAddress{NIC: 0, Addr: "", Port: uint16(localPort)}, nil); err != nil {
|
||
log.Fatal("Bind failed: ", err)
|
||
}
|
||
|
||
if err := ep.Connect(tcpip.FullAddress{NIC: 0, Addr: "", Port: uint16(localPort)}); err != nil {
|
||
log.Fatal("Conn failed: ", err)
|
||
}
|
||
|
||
// 注意UDP是无连接的，它不需要Listen
|
||
return ep
|
||
}
|