Fix timeout

Jigsaw-Code · Dec 5, 2023 · 63f2e33 · 63f2e33
1 parent fec16af
commit 63f2e33
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/x/examples/find-strategy/main.go b/x/examples/find-strategy/main.go
@@ -382,14 +382,16 @@ func (f *StrategyFinder) testDNSClient(clientId string, client *dnsClient, testD
 	}
 	f.log("; status=ok (%v) ✅\n", ips)
 	for _, ip := range ips {
-		f.FindTLSStrategy(ctx, testDomain, ip, &transport.TCPStreamDialer{}, client)
+		f.FindTLSStrategy(testDomain, ip, &transport.TCPStreamDialer{}, client)
 	}
 	return nil
 }
 
-func (f *StrategyFinder) FindTLSStrategy(ctx context.Context, domain string, ip net.IP, dialer transport.StreamDialer, dnsClient DNSClient) error {
+func (f *StrategyFinder) FindTLSStrategy(domain string, ip net.IP, dialer transport.StreamDialer, dnsClient DNSClient) error {
 	address := net.JoinHostPort(domain, "443")
 	for _, strategy := range []string{"direct", "split:2", "omitsni", "tlsfrag"} {
+		ctx, cancel := context.WithTimeout(context.Background(), f.TestTimeout)
+		defer cancel()
 		f.log("  tls=%v", strategy)
 		var tlsConn transport.StreamConn
 		var tlsErr error