mirror of
https://github.com/cwinfo/yggdrasil-go.git
synced 2024-11-10 07:30:27 +00:00
104 lines
2.1 KiB
Go
104 lines
2.1 KiB
Go
package main
|
|
|
|
import "fmt"
|
|
import "net"
|
|
import "time"
|
|
import "flag"
|
|
import "os"
|
|
import "runtime/pprof"
|
|
|
|
// TODO look into netmap + libpcap to bypass the kernel as much as possible?
|
|
|
|
const buffSize = 32
|
|
|
|
func basic_test() {
|
|
|
|
// TODO need a way to look up who our link-local neighbors are for each iface!
|
|
|
|
addr, err := net.ResolveTCPAddr("tcp", "[::1]:9001")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
listener, err := net.ListenTCP("tcp", addr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer listener.Close()
|
|
|
|
go func() {
|
|
send, err := net.DialTCP("tcp", nil, addr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer send.Close()
|
|
msg := make([]byte, 1280)
|
|
bss := make(net.Buffers, 0, 1024)
|
|
for {
|
|
for len(bss) < 1 { //buffSize {
|
|
bss = append(bss, msg)
|
|
}
|
|
bss := net.Buffers{[]byte{0, 1, 2, 3}, []byte{0, 1}, msg}
|
|
bss.WriteTo(send)
|
|
//send.Write(msg)
|
|
}
|
|
}()
|
|
|
|
numPackets := 1000000
|
|
start := time.Now()
|
|
//msg := make([]byte, 1280)
|
|
sock, err := listener.AcceptTCP()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer sock.Close()
|
|
for i := 0; i < numPackets; i++ {
|
|
msg := make([]byte, 1280*buffSize)
|
|
n, err := sock.Read(msg)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
msg = msg[:n]
|
|
for len(msg) > 1286 {
|
|
// handle message
|
|
i++
|
|
msg = msg[1286:]
|
|
}
|
|
// handle remaining fragment of message
|
|
//fmt.Println(n)
|
|
}
|
|
timed := time.Since(start)
|
|
|
|
fmt.Printf("%f packets per second\n", float64(numPackets)/timed.Seconds())
|
|
|
|
_ = func(in chan<- int) {
|
|
close(in)
|
|
}
|
|
|
|
}
|
|
|
|
var cpuprofile = flag.String("cpuprofile", "", "write cpu profile `file`")
|
|
var memprofile = flag.String("memprofile", "", "write memory profile to this file")
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
if *cpuprofile != "" {
|
|
f, err := os.Create(*cpuprofile)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("could not create CPU profile: ", err))
|
|
}
|
|
if err := pprof.StartCPUProfile(f); err != nil {
|
|
panic(fmt.Sprintf("could not start CPU profile: ", err))
|
|
}
|
|
defer pprof.StopCPUProfile()
|
|
}
|
|
if *memprofile != "" {
|
|
f, err := os.Create(*memprofile)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("could not create memory profile: ", err))
|
|
}
|
|
defer func() { pprof.WriteHeapProfile(f); f.Close() }()
|
|
}
|
|
basic_test()
|
|
|
|
}
|