noot

lib/fed/codecs/gnetcodec/codec.go

-package netconncodec
-
-import (
-	"crypto/tls"
-	"errors"
-	"fmt"
-	"net"
-	"sync"
-
-	"gfx.cafe/gfx/pggat/lib/fed"
-	"gfx.cafe/gfx/pggat/lib/util/decorator"
-)
-
-type Codec struct {
-	noCopy decorator.NoCopy
-
-	conn net.Conn
-	ssl  bool
-
-	encoder fed.Encoder
-	decoder fed.Decoder
-
-	mu sync.RWMutex
-}
-
-func NewCodec(rw net.Conn) fed.PacketCodec {
-	c := &Codec{
-		conn: rw,
-	}
-	c.encoder.Reset(rw)
-	c.decoder.Reset(rw)
-	return c
-}
-
-func (c *Codec) ReadPacket(typed bool) (fed.Packet, error) {
-	if err := c.decoder.Next(typed); err != nil {
-		return nil, err
-	}
-	return fed.PendingPacket{
-		Decoder: &c.decoder,
-	}, nil
-}
-
-func (c *Codec) WritePacket(packet fed.Packet) error {
-	err := c.encoder.Next(packet.Type(), packet.Length())
-	if err != nil {
-		return err
-	}
-
-	return packet.WriteTo(&c.encoder)
-}
-func (c *Codec) WriteByte(b byte) error {
-	return c.encoder.WriteByte(b)
-}
-
-func (c *Codec) ReadByte() (byte, error) {
-	if err := c.Flush(); err != nil {
-		return 0, err
-	}
-
-	return c.decoder.ReadByte()
-}
-
-func (c *Codec) Flush() error {
-	return c.encoder.Flush()
-}
-
-func (c *Codec) Close() error {
-	if err := c.encoder.Flush(); err != nil {
-		return err
-	}
-	return c.conn.Close()
-}
-
-func (c *Codec) LocalAddr() net.Addr {
-	return c.conn.LocalAddr()
-}
-
-func (c *Codec) SSL() bool {
-	return c.ssl
-}
-
-func (c *Codec) EnableSSL(config *tls.Config, isClient bool) error {
-	c.mu.Lock()
-	defer c.mu.Unlock()
-	if c.ssl {
-		return errors.New("SSL is already enabled")
-	}
-	c.ssl = true
-
-	// Flush buffers
-	if err := c.Flush(); err != nil {
-		return err
-	}
-	if c.decoder.Buffered() > 0 {
-		return errors.New("expected empty read buffer")
-	}
-
-	var sslConn *tls.Conn
-	if isClient {
-		sslConn = tls.Client(c.conn, config)
-	} else {
-		sslConn = tls.Server(c.conn, config)
-	}
-	c.encoder.Reset(sslConn)
-	c.decoder.Reset(sslConn)
-	c.conn = sslConn
-	err := sslConn.Handshake()
-	if err != nil {
-		return fmt.Errorf("ssl handshake fail client(%v): %w", isClient, err)
-	}
-	return nil
-}