server.go 1004 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. package tls
  2. import (
  3. "context"
  4. "net"
  5. "os"
  6. "github.com/sagernet/sing-box/common/badtls"
  7. C "github.com/sagernet/sing-box/constant"
  8. "github.com/sagernet/sing-box/log"
  9. "github.com/sagernet/sing-box/option"
  10. aTLS "github.com/sagernet/sing/common/tls"
  11. )
  12. func NewServer(ctx context.Context, logger log.Logger, options option.InboundTLSOptions) (ServerConfig, error) {
  13. if !options.Enabled {
  14. return nil, nil
  15. }
  16. if options.Reality != nil && options.Reality.Enabled {
  17. return NewRealityServer(ctx, logger, options)
  18. }
  19. return NewSTDServer(ctx, logger, options)
  20. }
  21. func ServerHandshake(ctx context.Context, conn net.Conn, config ServerConfig) (Conn, error) {
  22. ctx, cancel := context.WithTimeout(ctx, C.TCPTimeout)
  23. defer cancel()
  24. tlsConn, err := aTLS.ServerHandshake(ctx, conn, config)
  25. if err != nil {
  26. return nil, err
  27. }
  28. readWaitConn, err := badtls.NewReadWaitConn(tlsConn)
  29. if err == nil {
  30. return readWaitConn, nil
  31. } else if err != os.ErrInvalid {
  32. return nil, err
  33. }
  34. return tlsConn, nil
  35. }