dial.go 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. package domainsocket
  2. import (
  3. "context"
  4. "v2ray.com/core/common"
  5. "v2ray.com/core/common/net"
  6. "v2ray.com/core/transport/internet"
  7. "v2ray.com/core/transport/internet/tls"
  8. )
  9. func getSettingsFromContext(ctx context.Context) *Config {
  10. rawSettings := internet.TransportSettingsFromContext(ctx)
  11. if rawSettings == nil {
  12. return nil
  13. }
  14. return rawSettings.(*Config)
  15. }
  16. func Dial(ctx context.Context, dest net.Destination) (internet.Connection, error) {
  17. settings := getSettingsFromContext(ctx)
  18. if settings == nil {
  19. return nil, newError("domain socket settings is not specified.").AtError()
  20. }
  21. addr, err := settings.GetUnixAddr()
  22. if err != nil {
  23. return nil, err
  24. }
  25. conn, err := net.DialUnix("unix", nil, addr)
  26. if err != nil {
  27. return nil, newError("failed to dial unix: ", settings.Path).Base(err).AtWarning()
  28. }
  29. if config := tls.ConfigFromContext(ctx); config != nil {
  30. return tls.Client(conn, config.GetTLSConfig(tls.WithDestination(dest))), nil
  31. }
  32. return conn, nil
  33. }
  34. func init() {
  35. common.Must(internet.RegisterTransportDialer(internet.TransportProtocol_DomainSocket, Dial))
  36. }