1. 把tunnelid函数放到其他地方

2.在创建tunnel的时候显示每个tunnel属于哪个model
3.增加了queuecache的测试

protocol/tcp.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"github.com/TransX/log"
 	"github.com/TransX/stat"
+	"github.com/TransX/utils"
 	"net"
 	"os"
 )
@@ -75,22 +76,28 @@ func (this *TransTCP) Start(listenPort, destIP, destPort string, clientOrServer
 				log.Info("Dial %s", destConn.RemoteAddr().String())
 				//tunnel model : [ -->>server ---- client -->> ](this is a tunnel)
 				if clientOrServer == "client" {
-					ntSend := NewTunnel(listenerConn, destConn, SEND)
+					sendID := utils.TunnelID()
+					ntSend := NewTunnel(sendID, listenerConn, destConn, SEND)
 					ntSend.SetRegChan(tunMng.GetRegChan())
 					ntSend.SetUnRegChan(tunMng.GetUnregChan())
-					ntRev := NewTunnel(destConn, listenerConn, RECEIVE)
+					receiveID := utils.TunnelID()
+					ntRev := NewTunnel(receiveID, destConn, listenerConn, RECEIVE)
 					ntRev.SetRegChan(tunMng.GetRegChan())
 					ntRev.SetUnRegChan(tunMng.GetUnregChan())
+					printModelDetail(sendID, receiveID)
 					go ntSend.Run()
 					go ntRev.Run()
 				}
 				if clientOrServer == "server" {
-					ntRev := NewTunnel(listenerConn, destConn, RECEIVE)
+					receiveID := utils.TunnelID()
+					ntRev := NewTunnel(receiveID, listenerConn, destConn, RECEIVE)
 					ntRev.SetRegChan(tunMng.GetRegChan())
 					ntRev.SetUnRegChan(tunMng.GetUnregChan())
-					ntSend := NewTunnel(destConn, listenerConn, SEND)
+					sendID := utils.TunnelID()
+					ntSend := NewTunnel(sendID, destConn, listenerConn, SEND)
 					ntSend.SetRegChan(tunMng.GetRegChan())
 					ntSend.SetUnRegChan(tunMng.GetUnregChan())
+					printModelDetail(sendID, receiveID)
 					go ntRev.Run()
 					go ntSend.Run()
 				}
@@ -101,3 +108,7 @@ func (this *TransTCP) Start(listenPort, destIP, destPort string, clientOrServer
 		}
 	}
 }
+
+func printModelDetail(tunnelIDA, tunnelIDB string) {
+	log.Info("id %s and id %s belong to a model", tunnelIDA, tunnelIDB)
+}

protocol/tunnel.go

@@ -1,25 +1,15 @@
 package protocol
 
 import (
-	"bytes"
-	"crypto/md5"
-	"encoding/hex"
+
 	// "fmt"
 	"github.com/TransX/cache"
 	"github.com/TransX/log"
 	"github.com/TransX/tscipher"
 	"net"
-	"strconv"
-	"sync/atomic"
 	"time"
 )
 
-var seed int32
-
-func init() {
-	seed = 0
-}
-
 type Tunnel struct {
 	id              string
 	src             net.Conn
@@ -29,9 +19,9 @@ type Tunnel struct {
 	unregChan       chan interface{}
 }
 
-func NewTunnel(src, dest net.Conn, cipherDirection Direction) *Tunnel {
+func NewTunnel(id string, src, dest net.Conn, cipherDirection Direction) *Tunnel {
 	return &Tunnel{
-		id:              tunnelID(),
+		id:              id,
 		src:             src,
 		dest:            dest,
 		cipherDirection: cipherDirection,
@@ -65,17 +55,14 @@ func (this *Tunnel) Run() { //单向的，从src发送到dest
 	// cache := make([]byte, 1024*4) //4kB
 	//构建Carrier
 	queCache := cache.NewUnblockingQueueCache(1)
+	for i := 0; i < 1; i++ {
+		queCache.Put(make([]byte, 1024*4), 0)
+	}
 	msg := cache.NewBlockingQueueCache(1)
 	revCarrier := tscipher.NewCarrier(src, tscipher.NewCipher("XOR"), queCache, msg, id)
 	sendCarrier := tscipher.NewCarrier(dest, tscipher.NewCipher("XOR"), queCache, msg, id)
-	//timer
-
-	// for {
-	// nCh := make(chan int)
 	go this.receive(revCarrier)
 	go this.send(sendCarrier)
-	// log.Info("id %s send %d /receive %d duration %d ms", n, nByte, id, time.Since(srTimer)/1000)
-	// }
 
 }
 
@@ -97,7 +84,6 @@ func (this *Tunnel) receive(revCarrier *tscipher.Carrier) {
 			}
 		}
 	}()
-	// srTimer := time.Now() //send receive timer
 	var n int
 	var err error
 	for {
@@ -108,13 +94,11 @@ func (this *Tunnel) receive(revCarrier *tscipher.Carrier) {
 		} else {
 			n, err = tscipher.ReceiveData(revCarrier)
 		}
-		// fmt.Println("receive")
 		log.Info("id %s time to receive %d", id, time.Since(rTimer)/1000)
 		if err != nil {
 			log.Panic("Read panic. Tunnel id: %s. Remote Add: %s Local: %s. Err:%s", id, src.RemoteAddr().String(), src.LocalAddr().String(), err.Error())
 		}
-		log.Debug("Reived %d bytes (local add %s) from %s. Tunnel: id %s", n, src.LocalAddr().String(),src.RemoteAddr().String(), id)
-		// nCh <- n
+		log.Debug("Reived %d bytes (local add %s) from %s. Tunnel: id %s", n, src.LocalAddr().String(), src.RemoteAddr().String(), id)
 	}
 }
 
@@ -124,7 +108,6 @@ func (this *Tunnel) send(sendCarrier *tscipher.Carrier) {
 	cipherDirection := this.cipherDirection
 	id := this.id
 	defer func() {
-		// log.Debug("tunnel id %s ends", id)
 		//注销
 		this.unregChan <- this
 		if r := recover(); r != nil {
@@ -140,22 +123,12 @@ func (this *Tunnel) send(sendCarrier *tscipher.Carrier) {
 		sendCarrier.Cipher = nil
 	}
 	for {
-		// nByte := <-nCh
-		// fmt.Println("in send loop\n")
 		sTimer := time.Now() //send timer
-		_, err := tscipher.SendData(sendCarrier)
-		// fmt.Println("send")
+		n, err := tscipher.SendData(sendCarrier)
 		log.Info("id %s time to send %d", id, time.Since(sTimer)/1000)
 		if err != nil {
 			log.Panic("Write panic. ID: %s, Err: %s, Remote Add: %s", id, err, dest.RemoteAddr().String())
 		}
-		// log.Debug("Write %d bytes from %s to %s. Tunnel: %s . 18 bytes %x", n, dest.LocalAddr(), dest.RemoteAddr().String(), id, sendCarrier.Cache[:18])
+		log.Info("id %s Send %d bytes (local add %s),to %s", id, n, dest.LocalAddr().String(), dest.RemoteAddr().String())
 	}
 }
-
-func tunnelID() string {
-	nowString := time.Now().String() + strconv.Itoa(int(seed))
-	atomic.AddInt32(&seed, 1) //避免多线程情况下获得的种子相同
-	md5Byte := md5.Sum(bytes.NewBufferString(nowString).Bytes())
-	return hex.EncodeToString(md5Byte[:])
-}