Fix Dial function crash instance when there is no instance context in…

… the ctx
v2fly · Apr 2, 2021 · fc73774 · fc73774
1 parent e31bfe0
commit fc73774
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/functions.go b/functions.go
@@ -51,6 +51,11 @@ func Dial(ctx context.Context, v *Instance, dest net.Destination) (net.Conn, err
 	if dispatcher == nil {
 		return nil, newError("routing.Dispatcher is not registered in V2Ray core")
 	}
+
+	if ctx.Value(v2rayKey) == nil {
+		ctx = context.WithValue(ctx, v2rayKey, v)
+	}
+
 	r, err := dispatcher.(routing.Dispatcher).Dispatch(ctx, dest)
 	if err != nil {
 		return nil, err