starainrt
09d972c7b7
- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
106 lines
3.9 KiB
Go
106 lines
3.9 KiB
Go
package notify
|
|
|
|
import (
|
|
"b612.me/stario"
|
|
"context"
|
|
"net"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
type ClientCommon struct {
|
|
alive atomic.Value
|
|
status Status
|
|
byeFromServer bool
|
|
conn net.Conn
|
|
mu sync.Mutex
|
|
msgID uint64
|
|
peerIdentity string
|
|
sessionEpoch uint64
|
|
sessionOwnerState atomic.Int32
|
|
sessionRuntime atomic.Pointer[clientSessionRuntime]
|
|
connectSource atomic.Pointer[clientConnectSource]
|
|
queue *stario.StarQueue
|
|
stopFn context.CancelFunc
|
|
stopCtx context.Context
|
|
parallelNum int
|
|
maxReadTimeout time.Duration
|
|
maxWriteTimeout time.Duration
|
|
keyExchangeFn func(c Client) error
|
|
linkFns map[string]func(message *Message)
|
|
defaultFns func(message *Message)
|
|
msgEn func([]byte, []byte) []byte
|
|
msgDe func([]byte, []byte) []byte
|
|
fastStreamEncode transportFastStreamEncoder
|
|
fastBulkEncode transportFastBulkEncoder
|
|
fastPlainEncode transportFastPlainEncoder
|
|
handshakeRsaPubKey []byte
|
|
SecretKey []byte
|
|
noFinSyncMsgMaxKeepSeconds int
|
|
lastHeartbeat int64
|
|
heartbeatPeriod time.Duration
|
|
wg stario.WaitGroup
|
|
netType NetType
|
|
showError bool
|
|
skipKeyExchange bool
|
|
useHeartBeat bool
|
|
sequenceDe func([]byte) (interface{}, error)
|
|
sequenceEn func(interface{}) ([]byte, error)
|
|
logicalSession *logicalSessionState
|
|
onFileEvent func(FileEvent)
|
|
fileEventObserver func(FileEvent)
|
|
fileTransferCfg fileTransferConfig
|
|
signalReliableCfg signalReliabilityConfig
|
|
streamRuntime *streamRuntime
|
|
recordRuntime *recordRuntime
|
|
bulkRuntime *bulkRuntime
|
|
connectionRetryState *connectionRetryState
|
|
securityReadyCheck bool
|
|
debugMode bool
|
|
}
|
|
|
|
func NewClient() Client {
|
|
transport := defaultModernPSKTransportBundle()
|
|
var client = ClientCommon{
|
|
maxReadTimeout: 0,
|
|
maxWriteTimeout: 0,
|
|
peerIdentity: newClientPeerIdentity(),
|
|
sequenceEn: encode,
|
|
sequenceDe: Decode,
|
|
keyExchangeFn: aesRsaHello,
|
|
SecretKey: nil,
|
|
handshakeRsaPubKey: defaultRsaPubKey,
|
|
msgEn: transport.msgEn,
|
|
msgDe: transport.msgDe,
|
|
fastStreamEncode: transport.fastStreamEncode,
|
|
fastBulkEncode: transport.fastBulkEncode,
|
|
fastPlainEncode: transport.fastPlainEncode,
|
|
skipKeyExchange: true,
|
|
securityReadyCheck: true,
|
|
}
|
|
client.alive.Store(false)
|
|
client.useHeartBeat = true
|
|
client.heartbeatPeriod = time.Second * 20
|
|
client.linkFns = make(map[string]func(*Message))
|
|
client.defaultFns = func(message *Message) {
|
|
return
|
|
}
|
|
client.wg = stario.NewWaitGroup(0)
|
|
client.fileTransferCfg = defaultFileTransferConfig()
|
|
client.signalReliableCfg = defaultSignalReliabilityConfig()
|
|
client.logicalSession = newLogicalSessionState(client.fileTransferCfg, client.signalReliableCfg)
|
|
client.streamRuntime = newStreamRuntime("cstrm")
|
|
client.recordRuntime = newRecordRuntime()
|
|
client.bulkRuntime = newBulkRuntime("cblk")
|
|
client.connectionRetryState = newConnectionRetryState()
|
|
client.onFileEvent = normalizeFileEventCallback(nil)
|
|
client.fileEventObserver = normalizeFileEventCallback(nil)
|
|
client.stopCtx, client.stopFn = context.WithCancel(context.Background())
|
|
client.sessionRuntime.Store(newClientSessionRuntimeBase(client.stopCtx, client.stopFn))
|
|
bindClientStreamControl(&client)
|
|
bindClientBulkControl(&client)
|
|
client.getTransferState().setBuiltinHandler(client.builtinFileTransferHandler)
|
|
return &client
|
|
}
|