yggdrasil-go/misc/tests/tcptest.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)
		count := 0
		for {
			time.Sleep(100 * time.Millisecond)
			for len(bss) < count {
				bss = append(bss, msg)
			}
			bss.WriteTo(send)
			count++
			//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 {
		msg := make([]byte, 1280*buffSize)
		n, err := sock.Read(msg)
		if err != nil {
			panic(err)
		}
		msg = msg[:n]
		fmt.Println("Read:", n)
		for len(msg) > 1280 {
			// handle message
			msg = msg[1280:]
		}
		// 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()

}