123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095 |
- // Copyright © 2015 Daniel Fu <daniel820313@gmail.com>.
- // Copyright © 2019 Loki 'l0k18' Verloren <stalker.loki@protonmail.ch>.
- // Copyright © 2020 Gridfinity, LLC. <admin@gridfinity.com>.
- // Copyright © 2020 Jeffrey H. Johnson <jeff@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"
- // "hash/fnv"
- "io"
- "log"
- "net"
- "net/http"
- _ "net/http/pprof"
- "runtime"
- "sync"
- "testing"
- "time"
- "go.gridfinity.dev/gfcp"
- u "go.gridfinity.dev/leaktestfe"
- // "golang.org/x/crypto/pbkdf2"
- )
- 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"
- )
- /*var (
- key = []byte(
- "testkey",
- )
- pass = pbkdf2.Key(
- key,
- []byte(portSink),
- 4096,
- 32,
- fnv.New128a,
- )
- )*/
- 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 := 1024
- if runtime.GOOS == "darwin" {
- t.Log(
- "\n--- WARN: Running on macOS: Retargetting concurrency:\t128",
- )
- concurrent = 128
- }
- 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 parallel_client(
- &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 parallel_client(
- wg *sync.WaitGroup,
- ) (
- err error,
- ) {
- cli, err := dialEcho()
- if err != nil {
- panic(
- err,
- )
- }
- err = echo_tester(
- 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 := echo_tester(
- 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,
- )
- }
- sink_tester(
- cli,
- nbytes,
- b.N,
- )
- b.SetBytes(
- int64(
- nbytes,
- ),
- )
- }
- func echo_tester(
- 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
- } else {
- nrecv += n
- if nrecv == msglen {
- break
- }
- }
- }
- }
- return nil
- }
- func sink_tester(
- 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()
- }
- }
|