mirror of
https://github.com/XTLS/Xray-core.git
synced 2024-11-23 15:33:03 +00:00
22 lines
529 B
Go
22 lines
529 B
Go
|
package tls
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"crypto/tls"
|
||
|
|
|
||
|
|
"github.com/xtls/xray-core/common/net"
|
||
|
|
)
|
||
|
|
|
||
|
|
type customClientKey struct{}
|
||
|
|
|
||
|
|
type CustomClientFunc func(conn net.Conn, config *tls.Config) net.Conn
|
||
|
|
|
||
|
|
func CustomClientFromContext(ctx context.Context) (CustomClientFunc, bool) {
|
||
|
|
client, loaded := ctx.Value(customClientKey{}).(CustomClientFunc)
|
||
|
|
return client, loaded
|
||
|
|
}
|
||
|
|
|
||
|
|
func ContextWithCustomClient(ctx context.Context, customClient CustomClientFunc) context.Context {
|
||
|
|
return context.WithValue(ctx, customClientKey{}, customClient)
|
||
|
|
}
|