mirror of https://github.com/bjdgyc/anylink.git
112 lines
1.7 KiB
Go
112 lines
1.7 KiB
Go
package handler
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/bjdgyc/anylink/base"
|
|
"github.com/bjdgyc/anylink/sessdata"
|
|
)
|
|
|
|
// 不允许直接修改
|
|
// [6] => PType
|
|
var plHeader = []byte{
|
|
'S', 'T', 'F', 1,
|
|
0x00, 0x00, /* Length */
|
|
0x00, /* Type */
|
|
0x00, /* Unknown */
|
|
}
|
|
|
|
var plPool = sync.Pool{
|
|
New: func() interface{} {
|
|
b := make([]byte, BufferSize)
|
|
pl := sessdata.Payload{
|
|
LType: sessdata.LTypeIPData,
|
|
PType: 0x00,
|
|
Data: b,
|
|
}
|
|
// fmt.Println("plPool-init", len(pl.Data), cap(pl.Data))
|
|
return &pl
|
|
},
|
|
}
|
|
|
|
func getPayload() *sessdata.Payload {
|
|
pl := plPool.Get().(*sessdata.Payload)
|
|
return pl
|
|
}
|
|
|
|
func putPayload(pl *sessdata.Payload) {
|
|
// 错误数据丢弃
|
|
if cap(pl.Data) != BufferSize {
|
|
base.Warn("payload cap is err", cap(pl.Data))
|
|
return
|
|
}
|
|
|
|
pl.LType = sessdata.LTypeIPData
|
|
pl.PType = 0x00
|
|
pl.Data = pl.Data[:BufferSize]
|
|
plPool.Put(pl)
|
|
}
|
|
|
|
var bytePool = sync.Pool{
|
|
New: func() interface{} {
|
|
b := make([]byte, BufferSize)
|
|
// fmt.Println("bytePool-init")
|
|
return &b
|
|
},
|
|
}
|
|
|
|
func getByteZero() *[]byte {
|
|
b := bytePool.Get().(*[]byte)
|
|
*b = (*b)[:0]
|
|
return b
|
|
}
|
|
|
|
func getByteFull() *[]byte {
|
|
b := bytePool.Get().(*[]byte)
|
|
return b
|
|
}
|
|
func putByte(b *[]byte) {
|
|
*b = (*b)[:BufferSize]
|
|
bytePool.Put(b)
|
|
}
|
|
|
|
// 长度 34 小对象
|
|
var byte34Pool = sync.Pool{
|
|
New: func() interface{} {
|
|
b := make([]byte, 34)
|
|
return &b
|
|
},
|
|
}
|
|
|
|
func getByte34() *[]byte {
|
|
b := byte34Pool.Get().(*[]byte)
|
|
return b
|
|
}
|
|
|
|
func putByte34(b *[]byte) {
|
|
*b = (*b)[:34]
|
|
byte34Pool.Put(b)
|
|
}
|
|
|
|
type BufferPool struct {
|
|
sync.Pool
|
|
}
|
|
|
|
// 长度 51 小对象
|
|
var byte51Pool = sync.Pool{
|
|
New: func() interface{} {
|
|
b := make([]byte, 51)
|
|
return &b
|
|
},
|
|
}
|
|
|
|
func getByte51() *[]byte {
|
|
b := byte51Pool.Get().(*[]byte)
|
|
return b
|
|
}
|
|
|
|
func putByte51(b *[]byte) {
|
|
*b = (*b)[:51]
|
|
byte51Pool.Put(b)
|
|
}
|