重构，用多态模式写数据发送和接收的代码。速度快了，可能和减少了判断有关。
This commit is contained in:
dmy@lab
@@ -3,14 +3,13 @@ package tscipher
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"github.com/TransX/cache"
|
||||
"github.com/TransX/log"
|
||||
"github.com/TransX/utils"
|
||||
"github.com/spf13/viper"
|
||||
"io"
|
||||
"net"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/TransX/cache"
|
||||
"github.com/TransX/utils"
|
||||
"github.com/spf13/viper"
|
||||
)
|
||||
|
||||
// var StartMark = []byte("#2v!") //should be constant
|
||||
@@ -141,82 +140,82 @@ func UnwrapPackage(pacakge []byte) (data []byte, rest []byte, err error) {
|
||||
|
||||
}
|
||||
|
||||
func SendData(carrier *Carrier) (n int, err error) {
|
||||
msg, nByte := carrier.Msg.Get()
|
||||
if len(msg) < nByte {
|
||||
log.Panic("Cache of send is too small")
|
||||
}
|
||||
if carrier.Cipher == nil {
|
||||
n, err = carrier.Conn.Write(msg[:nByte])
|
||||
carrier.Cache.Put(make([]byte, 1024*4), 1024*4)
|
||||
return
|
||||
}
|
||||
encrypedByte, err := carrier.Cipher.Encrypt(msg[:nByte])
|
||||
if err != nil {
|
||||
n = 0
|
||||
return
|
||||
}
|
||||
//打包
|
||||
wraped := WrapPackage(encrypedByte[:nByte])
|
||||
n, err = carrier.Conn.Write(wraped)
|
||||
carrier.Cache.Put(make([]byte, 1024*4), 1024*4)
|
||||
return
|
||||
}
|
||||
|
||||
func RowReceiveData(carrier *Carrier) (n int, err error) {
|
||||
cache, _ := carrier.Cache.Get()
|
||||
n, err = carrier.Conn.Read(cache)
|
||||
if err != nil {
|
||||
n = 0
|
||||
}
|
||||
carrier.Msg.Put(cache, n)
|
||||
return
|
||||
}
|
||||
|
||||
func ReceiveData(carrier *Carrier) (n int, err error) {
|
||||
wrapedPackage := carrier.GetReceiveBuff() //make([]byte, 0, cap(carrier.Cache))
|
||||
var packageData []byte
|
||||
var _rest []byte
|
||||
cache, _ := carrier.Cache.Get()
|
||||
for {
|
||||
//首先检查这个是不是完整的包，是就返回好了，免得被阻塞
|
||||
data, rest, err := UnwrapPackage(wrapedPackage)
|
||||
packageData = data
|
||||
_rest = rest
|
||||
if err, ok := err.(*NotPackageError); len(wrapedPackage) >= 18 && ok {
|
||||
log.Debug("return NotPackageError %s", carrier.AttachedTunnelID)
|
||||
return 0, err
|
||||
}
|
||||
if err == nil {
|
||||
//够一个完整的包
|
||||
capBuff := cap(carrier.GetReceiveBuff())
|
||||
_buff := make([]byte, 0, capBuff) //释放
|
||||
_buff = append(_buff, _rest...)
|
||||
carrier.SetReceiveBuff(_buff)
|
||||
break
|
||||
}
|
||||
//如果读到的数据不够一个完整的包
|
||||
if len(wrapedPackage) > 0 {
|
||||
n, err = carrier.Conn.Read(cache)
|
||||
if err != nil {
|
||||
log.Error("ERROR %s", err)
|
||||
}
|
||||
} else {
|
||||
n, err = io.ReadAtLeast(carrier.Conn, cache, 18)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
n = 0
|
||||
return n, err
|
||||
}
|
||||
wrapedPackage = append(wrapedPackage, cache[:n]...)
|
||||
}
|
||||
decrypted, err := carrier.Cipher.Decrypt(packageData)
|
||||
if err != nil {
|
||||
n = 0
|
||||
return
|
||||
}
|
||||
n = len(decrypted)
|
||||
carrier.Msg.Put(decrypted, n)
|
||||
return
|
||||
}
|
||||
// func SendData(carrier *Carrier) (n int, err error) {
|
||||
// msg, nByte := carrier.Msg.Get()
|
||||
// if len(msg) < nByte {
|
||||
// log.Panic("Cache of send is too small")
|
||||
// }
|
||||
// if carrier.Cipher == nil {
|
||||
// n, err = carrier.Conn.Write(msg[:nByte])
|
||||
// carrier.Cache.Put(make([]byte, 1024*4), 1024*4)
|
||||
// return
|
||||
// }
|
||||
// encrypedByte, err := carrier.Cipher.Encrypt(msg[:nByte])
|
||||
// if err != nil {
|
||||
// n = 0
|
||||
// return
|
||||
// }
|
||||
// //打包
|
||||
// wraped := WrapPackage(encrypedByte[:nByte])
|
||||
// n, err = carrier.Conn.Write(wraped)
|
||||
// carrier.Cache.Put(make([]byte, 1024*4), 1024*4)
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// func RowReceiveData(carrier *Carrier) (n int, err error) {
|
||||
// cache, _ := carrier.Cache.Get()
|
||||
// n, err = carrier.Conn.Read(cache)
|
||||
// if err != nil {
|
||||
// n = 0
|
||||
// }
|
||||
// carrier.Msg.Put(cache, n)
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// func ReceiveData(carrier *Carrier) (n int, err error) {
|
||||
// wrapedPackage := carrier.GetReceiveBuff() //make([]byte, 0, cap(carrier.Cache))
|
||||
// var packageData []byte
|
||||
// var _rest []byte
|
||||
// cache, _ := carrier.Cache.Get()
|
||||
// for {
|
||||
// //首先检查这个是不是完整的包，是就返回好了，免得被阻塞
|
||||
// data, rest, err := UnwrapPackage(wrapedPackage)
|
||||
// packageData = data
|
||||
// _rest = rest
|
||||
// if err, ok := err.(*NotPackageError); len(wrapedPackage) >= 18 && ok {
|
||||
// log.Debug("return NotPackageError %s", carrier.AttachedTunnelID)
|
||||
// return 0, err
|
||||
// }
|
||||
// if err == nil {
|
||||
// //够一个完整的包
|
||||
// capBuff := cap(carrier.GetReceiveBuff())
|
||||
// _buff := make([]byte, 0, capBuff) //释放
|
||||
// _buff = append(_buff, _rest...)
|
||||
// carrier.SetReceiveBuff(_buff)
|
||||
// break
|
||||
// }
|
||||
// //如果读到的数据不够一个完整的包
|
||||
// if len(wrapedPackage) > 0 {
|
||||
// n, err = carrier.Conn.Read(cache)
|
||||
// if err != nil {
|
||||
// log.Error("ERROR %s", err)
|
||||
// }
|
||||
// } else {
|
||||
// n, err = io.ReadAtLeast(carrier.Conn, cache, 18)
|
||||
// }
|
||||
//
|
||||
// if err != nil {
|
||||
// n = 0
|
||||
// return n, err
|
||||
// }
|
||||
// wrapedPackage = append(wrapedPackage, cache[:n]...)
|
||||
// }
|
||||
// decrypted, err := carrier.Cipher.Decrypt(packageData)
|
||||
// if err != nil {
|
||||
// n = 0
|
||||
// return
|
||||
// }
|
||||
// n = len(decrypted)
|
||||
// carrier.Msg.Put(decrypted, n)
|
||||
// return
|
||||
// }
|
||||
|
||||
Reference in New Issue
Block a user