/* * Copyright (c) 2011, 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. */ /* * @test * @bug 6964547 * @run main/othervm SocksProxyVersion * @summary test socksProxyVersion system property */ import java.net.InetSocketAddress; import java.net.ServerSocket; import java.net.Socket; import java.net.SocketException; import java.io.DataInputStream; import java.io.IOException; public class SocksProxyVersion implements Runnable { final ServerSocket ss; volatile boolean failed; public static void main(String[] args) throws Exception { new SocksProxyVersion(); } @SuppressWarnings("try") public SocksProxyVersion() throws Exception { ss = new ServerSocket(0); try (ServerSocket socket = ss) { runTest(); } } void runTest() throws Exception { int port = ss.getLocalPort(); Thread serverThread = new Thread(this); serverThread.start(); System.setProperty("socksProxyHost", "localhost"); System.setProperty("socksProxyPort", Integer.toString(port)); // SOCKS V4 System.setProperty("socksProxyVersion", Integer.toString(4)); try (Socket s = new Socket()) { s.connect(new InetSocketAddress("localhost", port)); } catch (SocketException e) { // java.net.SocketException: Malformed reply from SOCKS server // This exception is OK, since the "server" does not implement // the socks protocol. It just verifies the version and closes. } // SOCKS V5 System.setProperty("socksProxyVersion", Integer.toString(5)); try (Socket s = new Socket()) { s.connect(new InetSocketAddress("localhost", port)); } catch (SocketException e) { /* OK */ } serverThread.join(); if (failed) { throw new RuntimeException("socksProxyVersion not being set correctly"); } } public void run() { try { try (Socket s = ss.accept()) { int version = (s.getInputStream()).read(); if (version != 4) { System.out.println("Got " + version + ", expected 4"); failed = true; } } try (Socket s = ss.accept()) { int version = (s.getInputStream()).read(); if (version != 5) { System.out.println("Got " + version + ", expected 5"); failed = true; } } } catch (IOException e) { e.printStackTrace(); } } }