1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084 |
- // Copyright © 2015 Daniel Fu <daniel820313@gmail.com>.
- // Copyright © 2019 Loki 'l0k18' Verloren <stalker.loki@protonmail.ch>.
- // Copyright © 2021 Gridfinity, LLC. <admin@gridfinity.com>.
- //
- // All rights reserved.
- //
- // All use of this code is governed by the MIT license.
- // The complete license is available in the LICENSE file.
- package gfcp_test
- import (
- "fmt"
- "io"
- "log"
- "net"
- "net/http"
- _ "net/http/pprof"
- "runtime"
- "sync"
- "testing"
- "time"
- "go.gridfinity.dev/gfcp"
- u "go.gridfinity.dev/leaktestfe"
- )
- const (
- portEcho = "127.0.0.1:9079"
- portSink = "127.0.0.1:19609"
- portTinyBufferEcho = "127.0.0.1:29609"
- portListerner = "127.0.0.1:9078"
- )
- func init() {
- go func() {
- log.Println(
- http.ListenAndServe(
- "127.0.0.1:8881",
- nil,
- ),
- )
- }()
- go echoServer()
- go sinkServer()
- go tinyBufferEchoServer()
- }
- func dialEcho() (
- *gfcp.UDPSession,
- error,
- ) {
- sess, err := gfcp.DialWithOptions(
- portEcho,
- 10,
- 3,
- )
- if err != nil {
- panic(
- err,
- )
- }
- sess.SetStreamMode(
- true,
- )
- sess.SetStreamMode(
- false,
- )
- sess.SetStreamMode(
- true,
- )
- sess.SetWindowSize(
- 1380,
- 1380,
- )
- sess.SetReadBuffer(
- 64 * 1024 * 1024,
- )
- sess.SetWriteBuffer(
- 64 * 1024 * 1024,
- )
- sess.SetStreamMode(
- true,
- )
- sess.SetNoDelay(
- 1,
- 10,
- 2,
- 1,
- )
- sess.SetMtu(
- 1400,
- )
- sess.SetMtu(
- 9000,
- )
- sess.SetMtu(
- 1400,
- )
- sess.SetACKNoDelay(
- true,
- )
- sess.SetACKNoDelay(
- false,
- )
- sess.SetDeadline(
- time.Now().Add(
- time.Minute,
- ),
- )
- return sess, err
- }
- func dialSink() (
- *gfcp.UDPSession,
- error,
- ) {
- sess, err := gfcp.DialWithOptions(
- portSink,
- 0,
- 0,
- )
- if err != nil {
- panic(
- err,
- )
- }
- sess.SetStreamMode(
- true,
- )
- sess.SetWindowSize(
- 1380,
- 1380,
- )
- sess.SetReadBuffer(
- 64 * 1024 * 1024,
- )
- sess.SetWriteBuffer(
- 64 * 1024 * 1024,
- )
- sess.SetStreamMode(
- true,
- )
- sess.SetNoDelay(
- 1,
- 10,
- 2,
- 1,
- )
- sess.SetMtu(
- 1400,
- )
- sess.SetACKNoDelay(
- false,
- )
- sess.SetDeadline(
- time.Now().Add(
- time.Minute,
- ),
- )
- return sess, err
- }
- func dialTinyBufferEcho() (
- *gfcp.UDPSession,
- error,
- ) {
- sess, err := gfcp.DialWithOptions(
- portTinyBufferEcho,
- 10,
- 3,
- )
- if err != nil {
- panic(
- err,
- )
- }
- return sess, err
- }
- func listenEcho() (
- net.Listener,
- error,
- ) {
- return gfcp.ListenWithOptions(
- portEcho,
- 10,
- 3,
- )
- }
- func listenTinyBufferEcho() (
- net.Listener,
- error,
- ) {
- return gfcp.ListenWithOptions(
- portTinyBufferEcho,
- 10,
- 3,
- )
- }
- func listenSink() (
- net.Listener,
- error,
- ) {
- return gfcp.ListenWithOptions(
- portSink,
- 0,
- 0,
- )
- }
- func echoServer() {
- l, err := listenEcho()
- if err != nil {
- panic(
- err,
- )
- }
- go func() {
- GFcplistener := l.(*gfcp.Listener)
- GFcplistener.SetReadBuffer(
- 64 * 1024 * 1024,
- )
- GFcplistener.SetWriteBuffer(
- 64 * 1024 * 1024,
- )
- GFcplistener.SetDSCP(
- 46,
- )
- for {
- s, err := l.Accept()
- if err != nil {
- return
- }
- s.(*gfcp.UDPSession).SetReadBuffer(
- 512 * 1024 * 1024,
- )
- s.(*gfcp.UDPSession).SetWriteBuffer(
- 512 * 1024 * 1024,
- )
- go handleEcho(s.(*gfcp.UDPSession))
- }
- }()
- }
- func sinkServer() {
- l, err := listenSink()
- if err != nil {
- panic(
- err,
- )
- }
- go func() {
- GFcplistener := l.(*gfcp.Listener)
- GFcplistener.SetReadBuffer(
- 64 * 1024 * 1024,
- )
- GFcplistener.SetWriteBuffer(
- 64 * 1024 * 1024,
- )
- GFcplistener.SetDSCP(
- 46,
- )
- for {
- s, err := l.Accept()
- if err != nil {
- return
- }
- go handleSink(s.(*gfcp.UDPSession))
- }
- }()
- }
- func tinyBufferEchoServer() {
- l, err := listenTinyBufferEcho()
- if err != nil {
- panic(
- err,
- )
- }
- go func() {
- for {
- s, err := l.Accept()
- if err != nil {
- return
- }
- go handleTinyBufferEcho(s.(*gfcp.UDPSession))
- }
- }()
- }
- func handleEcho(
- conn *gfcp.UDPSession,
- ) {
- conn.SetStreamMode(
- true,
- )
- conn.SetWindowSize(
- 8192,
- 8192,
- )
- conn.SetNoDelay(
- 1,
- 10,
- 2,
- 1,
- )
- conn.SetDSCP(
- 46,
- )
- conn.SetMtu(
- 1480,
- )
- conn.SetACKNoDelay(
- false,
- )
- conn.SetReadDeadline(
- time.Now().Add(
- time.Minute,
- ),
- )
- conn.SetWriteDeadline(
- time.Now().Add(
- time.Minute,
- ),
- )
- buf := make(
- []byte,
- 65536,
- )
- for {
- n, err := conn.Read(
- buf,
- )
- if err != nil {
- panic(
- err,
- )
- }
- conn.Write(
- buf[:n],
- )
- }
- }
- func handleSink(
- conn *gfcp.UDPSession,
- ) {
- conn.SetStreamMode(
- true,
- )
- conn.SetWindowSize(
- 8192,
- 8192,
- )
- conn.SetNoDelay(
- 1,
- 10,
- 2,
- 1,
- )
- conn.SetDSCP(
- 46,
- )
- conn.SetMtu(
- 1400,
- )
- conn.SetACKNoDelay(
- false,
- )
- conn.SetReadDeadline(
- time.Now().Add(
- time.Minute,
- ),
- )
- conn.SetWriteDeadline(
- time.Now().Add(
- time.Minute,
- ),
- )
- buf := make(
- []byte,
- 65536,
- )
- for {
- _, err := conn.Read(
- buf,
- )
- if err != nil {
- panic(
- err,
- )
- }
- }
- }
- func handleTinyBufferEcho(
- conn *gfcp.UDPSession,
- ) {
- conn.SetStreamMode(
- true,
- )
- buf := make(
- []byte,
- 2,
- )
- for {
- n, err := conn.Read(
- buf,
- )
- if err != nil {
- panic(
- err,
- )
- }
- conn.Write(
- buf[:n],
- )
- }
- }
- func TestTimeout(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- cli, err := dialEcho()
- if err != nil {
- panic(
- err,
- )
- }
- buf := make(
- []byte,
- 10,
- )
- cli.SetDeadline(
- time.Now().Add(
- time.Second,
- ),
- )
- <-time.After(
- 2 * time.Second,
- )
- n, err := cli.Read(
- buf,
- )
- if n != 0 || err == nil {
- t.Fail()
- }
- cli.Close()
- }
- func TestSendRecv(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- cli, err := dialEcho()
- if err != nil {
- panic(
- err,
- )
- }
- cli.SetWriteDelay(
- true,
- )
- cli.SetDUP(
- 1,
- )
- const (
- N = 100
- )
- buf := make(
- []byte,
- 10,
- )
- for i := 0; i < N; i++ {
- msg := fmt.Sprintf(
- "hello%v",
- i,
- )
- cli.Write(
- []byte(
- msg,
- ),
- )
- if n, err := cli.Read(
- buf,
- ); err == nil {
- if string(
- buf[:n],
- ) != msg {
- t.Fail()
- }
- } else {
- panic(
- err,
- )
- }
- }
- cli.Close()
- }
- func TestSendVector(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- cli, err := dialEcho()
- if err != nil {
- panic(
- err,
- )
- }
- cli.SetWriteDelay(
- false,
- )
- const N = 100
- buf := make(
- []byte,
- 20,
- )
- v := make(
- [][]byte,
- 2,
- )
- for i := 0; i < N; i++ {
- v[0] = []byte(
- fmt.Sprintf(
- "holas%v",
- i,
- ))
- v[1] = []byte(
- fmt.Sprintf(
- "amigo%v",
- i,
- ))
- msg :=
- fmt.Sprintf(
- "holas%vamigo%v",
- i,
- i,
- )
- cli.WriteBuffers(
- v,
- )
- if n, err := cli.Read(
- buf,
- ); err == nil {
- if string(
- buf[:n],
- ) != msg {
- t.Error(
- string(
- buf[:n],
- ),
- msg,
- )
- }
- } else {
- panic(
- err,
- )
- }
- }
- cli.Close()
- }
- func TestTinyBufferReceiver(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- cli, err := dialTinyBufferEcho()
- if err != nil {
- panic(
- err,
- )
- }
- const (
- N = 100
- )
- snd := byte(
- 0,
- )
- fillBuffer := func(
- buf []byte,
- ) {
- for i := 0; i < len(
- buf,
- ); i++ {
- buf[i] = snd
- snd++
- }
- }
- rcv := byte(
- 0,
- )
- check := func(
- buf []byte,
- ) bool {
- for i := 0; i < len(
- buf,
- ); i++ {
- if buf[i] != rcv {
- return false
- }
- rcv++
- }
- return true
- }
- sndbuf := make(
- []byte,
- 7,
- )
- rcvbuf := make(
- []byte,
- 7,
- )
- for i := 0; i < N; i++ {
- fillBuffer(
- sndbuf,
- )
- cli.Write(
- sndbuf,
- )
- if n, err := io.ReadFull(
- cli,
- rcvbuf,
- ); err == nil {
- if !check(
- rcvbuf[:n],
- ) {
- t.Fail()
- }
- } else {
- panic(
- err,
- )
- }
- }
- cli.Close()
- }
- func TestClose(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- cli, err := dialEcho()
- if err != nil {
- panic(
- err,
- )
- }
- buf := make(
- []byte,
- 10,
- )
- cli.Close()
- if cli.Close() == nil {
- t.Fail()
- }
- n, err := cli.Write(
- buf,
- )
- if n != 0 || err == nil {
- t.Fail()
- }
- n, err = cli.Read(
- buf,
- )
- if n != 0 || err == nil {
- t.Fail()
- }
- cli.Close()
- }
- func TestParallel(
- t *testing.T,
- ) {
- concurrent := 128
- if runtime.GOOS == "darwin" {
- t.Log(
- "\n--- WARN: Running on macOS: Retargetting concurrency:\t32",
- )
- concurrent = 32
- }
- if runtime.GOOS == "windows" {
- t.Log(
- "\n--- WARN: Running on Windows: Retargetting concurrency:\t8",
- )
- concurrent = 8
- }
- t.Log(
- fmt.Sprintf(
- "\n--- INFO: Target concurrency:\t%v",
- concurrent,
- ),
- )
- t.Parallel()
- t.Log(
- fmt.Sprintf(
- "\tStage 1/2:\tGoroutines:\t%v",
- runtime.NumGoroutine(),
- ),
- )
- defer u.Leakplug(
- t,
- )
- var wg sync.WaitGroup
- wg.Add(
- concurrent,
- )
- for i := 0; i < concurrent; i++ {
- go parallelClient(
- &wg,
- )
- }
- t.Log(
- fmt.Sprintf(
- "\tStage 2/2:\tGoroutines:\t%v",
- runtime.NumGoroutine(),
- ),
- )
- wg.Wait()
- t.Log(
- fmt.Sprintf(
- "\tStage 2/3:\tGoroutines:\t%v",
- runtime.NumGoroutine(),
- ),
- )
- }
- func parallelClient(
- wg *sync.WaitGroup,
- ) (
- err error,
- ) {
- cli, err := dialEcho()
- if err != nil {
- panic(
- err,
- )
- }
- err = echoTester(
- cli,
- 64,
- 64,
- )
- wg.Done()
- return
- }
- func BenchmarkEchoSpeed1K(
- b *testing.B,
- ) {
- speedclient(
- b,
- 1*1000,
- )
- }
- func BenchmarkEchoSpeed4K(
- b *testing.B,
- ) {
- speedclient(
- b,
- 4*1000,
- )
- }
- func BenchmarkEchoSpeed64K(
- b *testing.B,
- ) {
- speedclient(
- b,
- 64*1000,
- )
- }
- func BenchmarkEchoSpeed256K(
- b *testing.B,
- ) {
- speedclient(
- b,
- 256*1000,
- )
- }
- func BenchmarkEchoSpeed512K(
- b *testing.B,
- ) {
- speedclient(
- b,
- 512*1000,
- )
- }
- func BenchmarkEchoSpeed1M(
- b *testing.B,
- ) {
- speedclient(
- b,
- 1*1000*1000,
- )
- }
- func BenchmarkEchoSpeed4M(
- b *testing.B,
- ) {
- speedclient(
- b,
- 4*1000*1000,
- )
- }
- func BenchmarkEchoSpeed8M(
- b *testing.B,
- ) {
- speedclient(
- b,
- 8*1000*1000,
- )
- }
- func speedclient(
- b *testing.B,
- nbytes int,
- ) {
- b.ReportAllocs()
- cli, err := dialEcho()
- if err != nil {
- panic(
- err,
- )
- }
- if err := echoTester(
- cli,
- nbytes,
- b.N,
- ); err != nil {
- b.Fail()
- }
- b.SetBytes(
- int64(
- nbytes,
- ),
- )
- }
- func BenchmarkSinkSpeed1K(
- b *testing.B,
- ) {
- sinkclient(
- b,
- 1*1000,
- )
- }
- func BenchmarkSinkSpeed4K(
- b *testing.B,
- ) {
- sinkclient(
- b,
- 4*1000,
- )
- }
- func BenchmarkSinkSpeed64K(
- b *testing.B,
- ) {
- sinkclient(
- b,
- 64*1000,
- )
- }
- func BenchmarkSinkSpeed256K(
- b *testing.B,
- ) {
- sinkclient(
- b,
- 256*1000,
- )
- }
- func BenchmarkSinkSpeed512K(
- b *testing.B,
- ) {
- sinkclient(
- b,
- 512*1000,
- )
- }
- func BenchmarkSinkSpeed1M(
- b *testing.B,
- ) {
- sinkclient(
- b,
- 1*1000*1000,
- )
- }
- func BenchmarkSinkSpeed4M(
- b *testing.B,
- ) {
- sinkclient(
- b,
- 4*1000*1000,
- )
- }
- func BenchmarkSinkSpeed8M(
- b *testing.B,
- ) {
- sinkclient(
- b,
- 8*1000*1000,
- )
- }
- func sinkclient(
- b *testing.B,
- nbytes int,
- ) {
- b.ReportAllocs()
- cli, err := dialSink()
- if err != nil {
- panic(
- err,
- )
- }
- sinkTester(
- cli,
- nbytes,
- b.N,
- )
- b.SetBytes(
- int64(
- nbytes,
- ),
- )
- }
- func echoTester(
- cli net.Conn,
- msglen,
- msgcount int,
- ) error {
- buf := make(
- []byte,
- msglen,
- )
- for i := 0; i < msgcount; i++ {
- if _, err := cli.Write(
- buf,
- ); err != nil {
- return err
- }
- nrecv := 0
- for {
- n, err := cli.Read(
- buf,
- )
- if err != nil {
- return err
- }
- nrecv += n
- if nrecv == msglen {
- break
- }
- }
- }
- return nil
- }
- func sinkTester(
- cli *gfcp.UDPSession,
- msglen,
- msgcount int,
- ) error {
- buf := make(
- []byte,
- msglen,
- )
- for i := 0; i < msgcount; i++ {
- if _, err := cli.Write(
- buf,
- ); err != nil {
- return err
- }
- }
- return nil
- }
- func TestSnsi(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- t.Log(
- *gfcp.DefaultSnsi.Copy(),
- )
- t.Log(
- gfcp.DefaultSnsi.Header(),
- )
- t.Log(
- gfcp.DefaultSnsi.ToSlice(),
- )
- t.Log(
- "Resetting Snsi counters",
- )
- gfcp.DefaultSnsi.Reset()
- t.Log(
- gfcp.DefaultSnsi.ToSlice(),
- )
- }
- func TestListenerClose(
- t *testing.T,
- ) {
- defer u.Leakplug(
- t,
- )
- l, err := gfcp.ListenWithOptions(
- portListerner,
- 10,
- 3,
- )
- if err != nil {
- t.Fail()
- }
- l.SetReadDeadline(
- time.Now().Add(
- 3 * time.Second,
- ),
- )
- l.SetWriteDeadline(
- time.Now().Add(
- 3 * time.Second,
- ),
- )
- l.SetDeadline(
- time.Now().Add(
- 3 * time.Second,
- ),
- )
- time.Sleep(
- 1 * time.Millisecond,
- )
- if _, err := l.Accept(); err == nil {
- t.Fail()
- }
- l.Close()
- fakeaddr, _ := net.ResolveUDPAddr(
- "udp6",
- "127.0.0.1:7162",
- )
- if l.CloseSession(
- fakeaddr,
- ) {
- t.Fail()
- }
- }
|