1.加了分包发的功能

2.testing没什么问题，但是使用时还是感觉掉数据

Signed-off-by: dmy@lab <dmy@lab.com>

tscipher/cipher.go

@@ -1,9 +1,16 @@
 package tscipher
 
 import (
+	"bytes"
+	"errors"
+	"github.com/TransX/log"
+	"github.com/TransX/utils"
 	"net"
 )
 
+var StartMark = []byte("#2v!") //should be constant
+var EndMark = []byte("_=1z")   //should be constant
+
 type Cipher interface {
 	Decrypt(data []byte) (decrypted []byte, err error)
 	Encrypt(data []byte) (encryped []byte, err error)
@@ -28,6 +35,43 @@ func NewCipher(cipherName string) (cipher Cipher) {
 	return nil //TODO:临时这样处理
 }
 
+func WrapPackage(data []byte) []byte { //把要加密传输的数据打包成一定的格式，避免发送了100自己，只收到90字节的问题。
+	sizeOfData := len(data)
+	binSize := utils.Int2binary(sizeOfData, 10)
+	header := append(append(StartMark, binSize...), EndMark...)
+	//加密
+	key := []byte("hahahehe~-1!")
+	cipheredHeader := make([]byte, len(header))
+	for i, v := range header {
+		cipheredHeader[i] = v ^ key[i%len(key)]
+	}
+	return append(cipheredHeader, data...)
+}
+
+func UnwrapPackage(pacakge []byte) (packageSize int, data []byte, err error) {
+	//前14个字节是header
+	cipheredHeader := pacakge[:18]
+	header := make([]byte, len(cipheredHeader))
+	key := []byte("hahahehe~-1!")
+	for i, v := range cipheredHeader {
+		header[i] = v ^ key[i%len(key)]
+	}
+	start := header[:4]
+	end := header[14:]
+	binSize := header[4:14]
+	if bytes.Compare(start, StartMark) == 0 && bytes.Compare(end, EndMark) == 0 {
+		packageSize = utils.Binary2Int(binSize)
+		data = pacakge[18:]
+		err = nil
+	} else {
+		packageSize = 0
+		data = pacakge
+		err = errors.New("not a package")
+	}
+	return
+
+}
+
 func SendData(carrier *Carrier, nByte int) (n int, err error) {
 	if carrier.Cipher == nil {
 		n, err = carrier.Conn.Write(carrier.Cache[:nByte])
@@ -38,7 +82,9 @@ func SendData(carrier *Carrier, nByte int) (n int, err error) {
 		n = 0
 		return
 	}
-	n, err = carrier.Conn.Write(encrypedByte[:nByte])
+	//打包
+	wraped := WrapPackage(encrypedByte[:nByte])
+	n, err = carrier.Conn.Write(wraped)
 	copy(carrier.Cache, encrypedByte[:nByte]) // in case of debugging
 	return
 }
@@ -60,12 +106,46 @@ func ReceiveData(carrier *Carrier) (n int, err error) {
 	if carrier.Cipher == nil {
 		return
 	}
-	decrypted, err := carrier.Cipher.Decrypt(carrier.Cache[:n])
-	copy(carrier.Cache, decrypted[:n])
+	//解包
+	wrapedPackage := carrier.Cache[:n]
+	packageSize, data, err := UnwrapPackage(wrapedPackage)
+	realData := make([]byte, 0, packageSize)
+	// log.Info("packageSize %d data size %d", packageSize, len(data))
+	if err == nil && packageSize == len(data) { //读到的是一个完整的包
+		realData = data
+		log.Info("read a complete package")
+	} else {
+		gotSize := len(data)
+		for {
+			n, err = carrier.Conn.Read(carrier.Cache)
+			if err != nil {
+				n = 0
+				return
+			}
+			wrapedPackage = carrier.Cache[:n]
+			log.Info("got partial package size %d", n)
+			_, data, err = UnwrapPackage(wrapedPackage)
+			if err == nil {
+				n = 0
+				err = errors.New("partial package lost")
+				return
+			}
+			gotSize += len(data)
+			realData = append(realData, data...)
+			if gotSize == packageSize {
+				log.Info("got enough:. packageSize %d, real size %d", packageSize, gotSize)
+				break
+			}
+		}
+
+	}
+	decrypted, err := carrier.Cipher.Decrypt(realData)
 	if err != nil {
 		n = 0
 		return
 	}
+	n = len(decrypted)
+	copy(carrier.Cache, decrypted)
 	return
 }