nps/lib/mux/mux_test.go

package mux

import (
	"net"
	"net/http"
	_ "net/http/pprof"
	"sync"
	"testing"
	"time"

	"github.com/astaxie/beego/logs"
	"github.com/cnlh/nps/lib/common"
)

var conn1 net.Conn
var conn2 net.Conn

func TestNewMux(t *testing.T) {
	go func() {
		http.ListenAndServe("0.0.0.0:8889", nil)
	}()
	logs.EnableFuncCallDepth(true)
	logs.SetLogFuncCallDepth(3)
	server()
	client()
	time.Sleep(time.Second * 3)
	go func() {
		m2 := NewMux(conn2, "tcp")
		for {
			logs.Warn("npc starting accept")
			c, err := m2.Accept()
			if err != nil {
				logs.Warn(err)
				continue
			}
			logs.Warn("npc accept success ")
			c2, err := net.Dial("tcp", "127.0.0.1:80")
			if err != nil {
				logs.Warn(err)
				continue
			}
			go func(c2 net.Conn, c net.Conn) {
				wg := sync.WaitGroup{}
				wg.Add(1)
				go func() {
					_, err = common.CopyBuffer(c2, c)
					if err != nil {
						c2.Close()
						c.Close()
						logs.Warn("close npc by copy from nps", err)
					}
					wg.Done()
				}()
				wg.Add(1)
				go func() {
					_, err = common.CopyBuffer(c, c2)
					if err != nil {
						c2.Close()
						c.Close()
						logs.Warn("close npc by copy from server", err)
					}
					wg.Done()
				}()
				logs.Warn("npc wait")
				wg.Wait()
			}(c2, c)
		}
	}()

	go func() {
		m1 := NewMux(conn1, "tcp")
		l, err := net.Listen("tcp", "127.0.0.1:7777")
		if err != nil {
			logs.Warn(err)
		}
		for {
			logs.Warn("nps starting accept")
			conn, err := l.Accept()
			if err != nil {
				logs.Warn(err)
				continue
			}
			logs.Warn("nps accept success starting new conn")
			tmpCpnn, err := m1.NewConn()
			if err != nil {
				logs.Warn("nps new conn err ", err)
				continue
			}
			logs.Warn("nps new conn success ", tmpCpnn.connId)
			go func(tmpCpnn net.Conn, conn net.Conn) {
				go func() {
					_, err := common.CopyBuffer(tmpCpnn, conn)
					if err != nil {
						conn.Close()
						tmpCpnn.Close()
						logs.Warn("close nps by copy from user")
					}
				}()
				//time.Sleep(time.Second)
				_, err = common.CopyBuffer(conn, tmpCpnn)
				if err != nil {
					conn.Close()
					tmpCpnn.Close()
					logs.Warn("close nps by copy from npc ")
				}
			}(tmpCpnn, conn)
		}
	}()

	for {
		time.Sleep(time.Second * 5)
	}
}

func server() {
	var err error
	l, err := net.Listen("tcp", "127.0.0.1:9999")
	if err != nil {
		logs.Warn(err)
	}
	go func() {
		conn1, err = l.Accept()
		if err != nil {
			logs.Warn(err)
		}
	}()
	return
}

func client() {
	var err error
	conn2, err = net.Dial("tcp", "127.0.0.1:9999")
	if err != nil {
		logs.Warn(err)
	}
}

func TestNewConn(t *testing.T) {
	buf := common.GetBufPoolCopy()
	logs.Warn(len(buf), cap(buf))
	//b := pool.GetBufPoolCopy()
	//b[0] = 1
	//b[1] = 2
	//b[2] = 3
	b := []byte{1, 2, 3}
	logs.Warn(copy(buf[:3], b), len(buf), cap(buf))
	logs.Warn(len(buf), buf[0])
}