1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
/*
* Copyright (c) 2019, 2022, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
import org.testng.Assert;
import org.testng.annotations.AfterTest;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.Test;
import sun.net.spi.DefaultProxySelector;
import java.io.IOException;
import java.net.InetAddress;
import java.net.Proxy;
import java.net.ProxySelector;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.List;
/**
* @test
* @bug 8230310
* @summary Tests java.net.SocksSocketImpl
* @run testng SocksSocketImplTest
* @modules java.base/sun.net.spi:+open
*/
public class SocksSocketImplTest {
private ProxySelector previousDefault;
@BeforeTest
public void beforeTest() {
previousDefault = ProxySelector.getDefault();
ProxySelector.setDefault(new SchemeStrippedProxySelector());
}
@AfterTest
public void afterTest() {
ProxySelector.setDefault(previousDefault);
}
/**
* Creates a socket connection, which internally triggers proxy selection for the target
* address. The test has been configured to use a {@link SchemeStrippedProxySelector ProxySelector}
* which throws a {@link IllegalArgumentException}. This test then verifies that this IAE gets wrapped
* by {@code java.net.SocksSocketImpl} into an {@link IOException} before being thrown
*
* @throws Exception
*/
@Test
public void testIOEOnProxySelection() throws Exception {
final int backlog = -1;
final int port = 0;
try (ServerSocket ss = new ServerSocket(port, backlog, InetAddress.getLoopbackAddress());
Socket s1 = new Socket(ss.getInetAddress(), ss.getLocalPort());
Socket s2 = ss.accept()) {
Assert.fail("IOException was expected to be thrown, but wasn't");
} catch (IOException ioe) {
// expected
// now verify the IOE was thrown for the correct expected reason
if (!(ioe.getCause() instanceof IllegalArgumentException)) {
// rethrow this so that the test output failure will capture the entire/real
// cause in its stacktrace
throw ioe;
}
}
}
/**
* A {@link ProxySelector} which strips the "scheme" part of the {@link URI}
* before delegating the selection to the {@link DefaultProxySelector}.
* This is to ensure that the {@code DefaultProxySelector} throws an {@link IllegalArgumentException}
* during selection of the proxy
*/
private static final class SchemeStrippedProxySelector extends DefaultProxySelector {
@Override
public List<Proxy> select(final URI uri) {
System.out.println("Proxy selection for " + uri);
final URI schemeStrippedURI;
try {
// strip the scheme and pass the rest
schemeStrippedURI = new URI(null, uri.getHost(), uri.getPath(), null);
} catch (URISyntaxException e) {
throw new RuntimeException(e);
}
System.out.println("Scheme stripped URI " + schemeStrippedURI + " is being used to select a proxy");
return super.select(schemeStrippedURI);
}
}
}
|
