feat(notify): 重构通信内核并补齐 stream/bulk/record/transfer 能力

- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
2026-04-15 15:24:36 +08:00
parent d14d13c393
commit 09d972c7b7
216 changed files with 51374 additions and 1715 deletions
@@ -0,0 +1,238 @@
+package notify
+
+import "errors"
+
+var (
+	errClientSessionQueueUnavailable = errors.New("client session queue is unavailable")
+	errServerSessionQueueUnavailable = errors.New("server session queue is unavailable")
+	errTransportPayloadEncryptFailed = errors.New("transport payload encrypt failed")
+	errTransportPayloadDecryptFailed = errors.New("transport payload decrypt failed")
+)
+
+func (c *ClientCommon) encodeTransferMsg(msg TransferMsg) ([]byte, error) {
+	data, err := c.sequenceEn(msg)
+	if err != nil {
+		return nil, err
+	}
+	data = c.msgEn(c.SecretKey, data)
+	queue := c.clientQueueSnapshot()
+	if queue == nil {
+		return nil, errClientSessionQueueUnavailable
+	}
+	return queue.BuildMessage(data), nil
+}
+
+func (c *ClientCommon) decodeTransferMsg(data []byte) (TransferMsg, error) {
+	msg, err := c.sequenceDe(c.msgDe(c.SecretKey, data))
+	if err != nil {
+		return TransferMsg{}, err
+	}
+	transfer, ok := msg.(TransferMsg)
+	if !ok {
+		return TransferMsg{}, errors.New("invalid transfer message")
+	}
+	return transfer, nil
+}
+
+func (s *ServerCommon) encodeTransferMsg(c *ClientConn, msg TransferMsg) ([]byte, error) {
+	data, err := s.sequenceEn(msg)
+	if err != nil {
+		return nil, err
+	}
+	msgEn := c.clientConnMsgEnSnapshot()
+	secretKey := c.clientConnSecretKeySnapshot()
+	data = msgEn(secretKey, data)
+	queue := s.serverQueueSnapshot()
+	if queue == nil {
+		return nil, errServerSessionQueueUnavailable
+	}
+	return queue.BuildMessage(data), nil
+}
+
+func (s *ServerCommon) decodeTransferMsg(c *ClientConn, data []byte) (TransferMsg, error) {
+	msgDe := c.clientConnMsgDeSnapshot()
+	secretKey := c.clientConnSecretKeySnapshot()
+	msg, err := s.sequenceDe(msgDe(secretKey, data))
+	if err != nil {
+		return TransferMsg{}, err
+	}
+	transfer, ok := msg.(TransferMsg)
+	if !ok {
+		return TransferMsg{}, errors.New("invalid transfer message")
+	}
+	return transfer, nil
+}
+
+func (c *ClientCommon) encodeEnvelopePayload(env Envelope) ([]byte, error) {
+	data, err := c.encodeEnvelopePlain(env)
+	if err != nil {
+		return nil, err
+	}
+	return c.encryptTransportPayload(data)
+}
+
+func (c *ClientCommon) encodeEnvelopePlain(env Envelope) ([]byte, error) {
+	data, err := c.sequenceEn(env)
+	if err != nil {
+		return nil, err
+	}
+	return data, nil
+}
+
+func (c *ClientCommon) encryptTransportPayload(data []byte) ([]byte, error) {
+	encoded := c.msgEn(c.SecretKey, data)
+	if encoded == nil && len(data) != 0 {
+		return nil, errTransportPayloadEncryptFailed
+	}
+	return encoded, nil
+}
+
+func (c *ClientCommon) encodeEnvelope(env Envelope) ([]byte, error) {
+	data, err := c.encodeEnvelopePayload(env)
+	if err != nil {
+		return nil, err
+	}
+	queue := c.clientQueueSnapshot()
+	if queue == nil {
+		return nil, errClientSessionQueueUnavailable
+	}
+	return queue.BuildMessage(data), nil
+}
+
+func (c *ClientCommon) decodeEnvelope(data []byte) (Envelope, error) {
+	plain, err := c.decryptTransportPayload(data)
+	if err != nil {
+		return Envelope{}, err
+	}
+	return c.decodeEnvelopePlain(plain)
+}
+
+func (c *ClientCommon) decryptTransportPayload(data []byte) ([]byte, error) {
+	plain := c.msgDe(c.SecretKey, data)
+	if plain == nil && len(data) != 0 {
+		return nil, errTransportPayloadDecryptFailed
+	}
+	return plain, nil
+}
+
+func (c *ClientCommon) decodeEnvelopePlain(data []byte) (Envelope, error) {
+	msg, err := c.sequenceDe(data)
+	if err != nil {
+		return Envelope{}, err
+	}
+	env, ok := msg.(Envelope)
+	if ok {
+		return env, nil
+	}
+	transfer, ok := msg.(TransferMsg)
+	if !ok {
+		return Envelope{}, errors.New("invalid envelope")
+	}
+	wrapped, err := wrapTransferMsgEnvelope(transfer, c.sequenceEn)
+	if err != nil {
+		return Envelope{}, err
+	}
+	return wrapped, nil
+}
+
+func (s *ServerCommon) encodeEnvelope(c *ClientConn, env Envelope) ([]byte, error) {
+	return s.encodeEnvelopeLogical(logicalConnFromClient(c), env)
+}
+
+func (s *ServerCommon) encodeEnvelopePayloadLogical(logical *LogicalConn, env Envelope) ([]byte, error) {
+	if logical == nil {
+		return nil, errTransportDetached
+	}
+	data, err := s.encodeEnvelopePlain(env)
+	if err != nil {
+		return nil, err
+	}
+	return s.encryptTransportPayloadLogical(logical, data)
+}
+
+func (s *ServerCommon) encodeEnvelopePlain(env Envelope) ([]byte, error) {
+	data, err := s.sequenceEn(env)
+	if err != nil {
+		return nil, err
+	}
+	return data, nil
+}
+
+func (s *ServerCommon) encryptTransportPayloadLogical(logical *LogicalConn, data []byte) ([]byte, error) {
+	if logical == nil {
+		return nil, errTransportDetached
+	}
+	msgEn := logical.msgEnSnapshot()
+	secretKey := logical.secretKeySnapshot()
+	if msgEn == nil {
+		return nil, errTransportDetached
+	}
+	encoded := msgEn(secretKey, data)
+	if encoded == nil && len(data) != 0 {
+		return nil, errTransportPayloadEncryptFailed
+	}
+	return encoded, nil
+}
+
+func (s *ServerCommon) encodeEnvelopeLogical(logical *LogicalConn, env Envelope) ([]byte, error) {
+	data, err := s.encodeEnvelopePayloadLogical(logical, env)
+	if err != nil {
+		return nil, err
+	}
+	queue := s.serverQueueSnapshot()
+	if queue == nil {
+		return nil, errServerSessionQueueUnavailable
+	}
+	return queue.BuildMessage(data), nil
+}
+
+func (s *ServerCommon) decodeEnvelope(c *ClientConn, data []byte) (Envelope, error) {
+	return s.decodeEnvelopeLogical(logicalConnFromClient(c), data)
+}
+
+func (s *ServerCommon) decodeEnvelopeLogical(logical *LogicalConn, data []byte) (Envelope, error) {
+	if logical == nil {
+		return Envelope{}, errTransportDetached
+	}
+	plain, err := s.decryptTransportPayloadLogical(logical, data)
+	if err != nil {
+		return Envelope{}, err
+	}
+	return s.decodeEnvelopePlain(plain)
+}
+
+func (s *ServerCommon) decryptTransportPayloadLogical(logical *LogicalConn, data []byte) ([]byte, error) {
+	if logical == nil {
+		return nil, errTransportDetached
+	}
+	msgDe := logical.msgDeSnapshot()
+	secretKey := logical.secretKeySnapshot()
+	if msgDe == nil {
+		return nil, errTransportDetached
+	}
+	plain := msgDe(secretKey, data)
+	if plain == nil && len(data) != 0 {
+		return nil, errTransportPayloadDecryptFailed
+	}
+	return plain, nil
+}
+
+func (s *ServerCommon) decodeEnvelopePlain(data []byte) (Envelope, error) {
+	msg, err := s.sequenceDe(data)
+	if err != nil {
+		return Envelope{}, err
+	}
+	env, ok := msg.(Envelope)
+	if ok {
+		return env, nil
+	}
+	transfer, ok := msg.(TransferMsg)
+	if !ok {
+		return Envelope{}, errors.New("invalid envelope")
+	}
+	wrapped, err := wrapTransferMsgEnvelope(transfer, s.sequenceEn)
+	if err != nil {
+		return Envelope{}, err
+	}
+	return wrapped, nil
+}