From 7afd1fdf46f1ba64eca613aa57e2aa74b5454863 Mon Sep 17 00:00:00 2001 From: Florian Schmaus Date: Mon, 18 Nov 2019 17:44:08 +0100 Subject: [PATCH 1/3] proxy: make it the caller's reponsibility to close the socket This makes the code shorter as there is now a single place where the socket should be closed. --- .../smack/proxy/ProxySocketConnection.java | 12 +- .../proxy/Socks4ProxySocketConnection.java | 94 ++++------ .../proxy/Socks5ProxySocketConnection.java | 167 ++++++++---------- .../smack/tcp/XMPPTCPConnection.java | 1 + 4 files changed, 123 insertions(+), 151 deletions(-) diff --git a/smack-core/src/main/java/org/jivesoftware/smack/proxy/ProxySocketConnection.java b/smack-core/src/main/java/org/jivesoftware/smack/proxy/ProxySocketConnection.java index aad3cb5bf..7f31f94dc 100644 --- a/smack-core/src/main/java/org/jivesoftware/smack/proxy/ProxySocketConnection.java +++ b/smack-core/src/main/java/org/jivesoftware/smack/proxy/ProxySocketConnection.java @@ -1,6 +1,6 @@ /** * - * Copyright 2015-2016 Florian Schmaus. + * Copyright 2015-2019 Florian Schmaus. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -21,6 +21,16 @@ import java.net.Socket; public interface ProxySocketConnection { + /** + * Initiate a connection to the given host on the given port. Note that the caller is responsible for closing the + * socket in case this method throws. + * + * @param socket the socket to use to initiate the connection to the proxy. + * @param host the host to connect to. + * @param port the port to connect to. + * @param timeout the timeout in milliseconds. + * @throws IOException in case an I/O error occurs. + */ void connect(Socket socket, String host, int port, int timeout) throws IOException; diff --git a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java index 2c2658c99..614957f87 100644 --- a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java +++ b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java @@ -39,20 +39,17 @@ public class Socks4ProxySocketConnection implements ProxySocketConnection { @Override public void connect(Socket socket, String host, int port, int timeout) throws IOException { - InputStream in = null; - OutputStream out = null; String proxy_host = proxy.getProxyAddress(); int proxy_port = proxy.getProxyPort(); String user = proxy.getProxyUsername(); - try { - socket.connect(new InetSocketAddress(proxy_host, proxy_port), timeout); - in = socket.getInputStream(); - out = socket.getOutputStream(); - socket.setTcpNoDelay(true); + socket.connect(new InetSocketAddress(proxy_host, proxy_port), timeout); + InputStream in = socket.getInputStream(); + OutputStream out = socket.getOutputStream(); + socket.setTcpNoDelay(true); - byte[] buf = new byte[1024]; - int index = 0; + byte[] buf = new byte[1024]; + int index = 0; /* 1) CONNECT @@ -72,25 +69,25 @@ public class Socks4ProxySocketConnection implements ProxySocketConnection { of all zero bits. */ - buf[index++] = 4; - buf[index++] = 1; + buf[index++] = 4; + buf[index++] = 1; - buf[index++] = (byte) (port >>> 8); - buf[index++] = (byte) (port & 0xff); + buf[index++] = (byte) (port >>> 8); + buf[index++] = (byte) (port & 0xff); - InetAddress inetAddress = InetAddress.getByName(proxy_host); - byte[] byteAddress = inetAddress.getAddress(); - for (int i = 0; i < byteAddress.length; i++) { - buf[index++] = byteAddress[i]; - } + InetAddress inetAddress = InetAddress.getByName(proxy_host); + byte[] byteAddress = inetAddress.getAddress(); + for (int i = 0; i < byteAddress.length; i++) { + buf[index++] = byteAddress[i]; + } - if (user != null) { - byte[] userBytes = user.getBytes(StandardCharsets.UTF_8); - System.arraycopy(userBytes, 0, buf, index, user.length()); - index += user.length(); - } - buf[index++] = 0; - out.write(buf, 0, index); + if (user != null) { + byte[] userBytes = user.getBytes(StandardCharsets.UTF_8); + System.arraycopy(userBytes, 0, buf, index, user.length()); + index += user.length(); + } + buf[index++] = 0; + out.write(buf, 0, index); /* The SOCKS server checks to see whether such a request should be granted @@ -119,43 +116,26 @@ public class Socks4ProxySocketConnection implements ProxySocketConnection { The remaining fields are ignored. */ - int len = 6; - int s = 0; - while (s < len) { - int i = in.read(buf, s, len - s); - if (i <= 0) { - throw new ProxyException(ProxyInfo.ProxyType.SOCKS4, - "stream is closed"); - } - s += i; - } - if (buf[0] != 0) { + int len = 6; + int s = 0; + while (s < len) { + int i = in.read(buf, s, len - s); + if (i <= 0) { throw new ProxyException(ProxyInfo.ProxyType.SOCKS4, - "server returns VN " + buf[0]); + "stream is closed"); } - if (buf[1] != 90) { - try { - socket.close(); - } - catch (Exception eee) { - } - String message = "ProxySOCKS4: server returns CD " + buf[1]; - throw new ProxyException(ProxyInfo.ProxyType.SOCKS4, message); - } - byte[] temp = new byte[2]; - in.read(temp, 0, 2); + s += i; } - catch (RuntimeException e) { - throw e; + if (buf[0] != 0) { + throw new ProxyException(ProxyInfo.ProxyType.SOCKS4, + "server returns VN " + buf[0]); } - catch (Exception e) { - try { - socket.close(); - } - catch (Exception eee) { - } - throw new ProxyException(ProxyInfo.ProxyType.SOCKS4, e.toString()); + if (buf[1] != 90) { + String message = "ProxySOCKS4: server returns CD " + buf[1]; + throw new ProxyException(ProxyInfo.ProxyType.SOCKS4, message); } + byte[] temp = new byte[2]; + in.read(temp, 0, 2); } } diff --git a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java index 3c8a080e6..c91402218 100644 --- a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java +++ b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java @@ -22,9 +22,6 @@ import java.io.OutputStream; import java.net.InetSocketAddress; import java.net.Socket; import java.nio.charset.StandardCharsets; -import java.util.logging.Logger; - -import org.jivesoftware.smack.util.CloseableUtil; /** * Socket factory for Socks5 proxy. @@ -32,7 +29,6 @@ import org.jivesoftware.smack.util.CloseableUtil; * @author Atul Aggarwal */ public class Socks5ProxySocketConnection implements ProxySocketConnection { - private static final Logger LOGGER = Logger.getLogger(Socks5ProxySocketConnection.class.getName()); private final ProxyInfo proxy; @@ -43,22 +39,19 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { @Override public void connect(Socket socket, String host, int port, int timeout) throws IOException { - InputStream in = null; - OutputStream out = null; String proxy_host = proxy.getProxyAddress(); int proxy_port = proxy.getProxyPort(); String user = proxy.getProxyUsername(); String passwd = proxy.getProxyPassword(); - try { - socket.connect(new InetSocketAddress(proxy_host, proxy_port), timeout); - in = socket.getInputStream(); - out = socket.getOutputStream(); + socket.connect(new InetSocketAddress(proxy_host, proxy_port), timeout); + InputStream in = socket.getInputStream(); + OutputStream out = socket.getOutputStream(); - socket.setTcpNoDelay(true); + socket.setTcpNoDelay(true); - byte[] buf = new byte[1024]; - int index = 0; + byte[] buf = new byte[1024]; + int index = 0; /* +----+----------+----------+ @@ -81,13 +74,13 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { o X'FF' NO ACCEPTABLE METHODS */ - buf[index++] = 5; + buf[index++] = 5; - buf[index++] = 2; - buf[index++] = 0; // NO AUTHENTICATION REQUIRED - buf[index++] = 2; // USERNAME/PASSWORD + buf[index++] = 2; + buf[index++] = 0; // NO AUTHENTICATION REQUIRED + buf[index++] = 2; // USERNAME/PASSWORD - out.write(buf, 0, index); + out.write(buf, 0, index); /* The server selects from one of the methods given in METHODS, and @@ -99,17 +92,17 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { | 1 | 1 | +----+--------+ */ - fill(in, buf, 2); + fill(in, buf, 2); - boolean check = false; - switch (buf[1] & 0xff) { - case 0: // NO AUTHENTICATION REQUIRED - check = true; + boolean check = false; + switch (buf[1] & 0xff) { + case 0: // NO AUTHENTICATION REQUIRED + check = true; + break; + case 2: // USERNAME/PASSWORD + if (user == null || passwd == null) { break; - case 2: // USERNAME/PASSWORD - if (user == null || passwd == null) { - break; - } + } /* Once the SOCKS V5 server has started, and the client has selected the @@ -130,20 +123,19 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { PASSWD field that follows. The PASSWD field contains the password association with the given UNAME. */ - index = 0; - buf[index++] = 1; - buf[index++] = (byte) user.length(); - byte[] userBytes = user.getBytes(StandardCharsets.UTF_8); - System.arraycopy(userBytes, 0, buf, index, - user.length()); - index += user.length(); - byte[] passwordBytes = passwd.getBytes(StandardCharsets.UTF_8); - buf[index++] = (byte) passwordBytes.length; - System.arraycopy(passwordBytes, 0, buf, index, - passwd.length()); - index += passwd.length(); - - out.write(buf, 0, index); + index = 0; + buf[index++] = 1; + buf[index++] = (byte) user.length(); + byte[] userBytes = user.getBytes(StandardCharsets.UTF_8); + System.arraycopy(userBytes, 0, buf, index, + user.length()); + index += user.length(); + byte[] passwordBytes = passwd.getBytes(StandardCharsets.UTF_8); + buf[index++] = (byte) passwordBytes.length; + System.arraycopy(passwordBytes, 0, buf, index, + passwd.length()); + index += passwd.length(); + out.write(buf, 0, index); /* The server verifies the supplied UNAME and PASSWD, and sends the @@ -159,19 +151,18 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { `failure' (STATUS value other than X'00') status, it MUST close the connection. */ - fill(in, buf, 2); - if (buf[1] == 0) { - check = true; - } - break; - default: - } + fill(in, buf, 2); + if (buf[1] == 0) { + check = true; + } + break; + default: + } - if (!check) { - CloseableUtil.maybeClose(socket, LOGGER); - throw new ProxyException(ProxyInfo.ProxyType.SOCKS5, - "fail in SOCKS5 proxy"); - } + if (!check) { + throw new ProxyException(ProxyInfo.ProxyType.SOCKS5, + "fail in SOCKS5 proxy"); + } /* The SOCKS request is formed as follows: @@ -199,21 +190,21 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { order */ - index = 0; - buf[index++] = 5; - buf[index++] = 1; // CONNECT - buf[index++] = 0; + index = 0; + buf[index++] = 5; + buf[index++] = 1; // CONNECT + buf[index++] = 0; - byte[] hostb = host.getBytes(StandardCharsets.UTF_8); - int len = hostb.length; - buf[index++] = 3; // DOMAINNAME - buf[index++] = (byte) len; - System.arraycopy(hostb, 0, buf, index, len); - index += len; - buf[index++] = (byte) (port >>> 8); - buf[index++] = (byte) (port & 0xff); + byte[] hostb = host.getBytes(StandardCharsets.UTF_8); + int len = hostb.length; + buf[index++] = 3; // DOMAINNAME + buf[index++] = (byte) len; + System.arraycopy(hostb, 0, buf, index, len); + index += len; + buf[index++] = (byte) (port >>> 8); + buf[index++] = (byte) (port & 0xff); - out.write(buf, 0, index); + out.write(buf, 0, index); /* The SOCKS request information is sent by the client as soon as it has @@ -250,35 +241,25 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { o BND.PORT server bound port in network octet order */ - fill(in, buf, 4); + fill(in, buf, 4); - if (buf[1] != 0) { - CloseableUtil.maybeClose(socket, LOGGER); - throw new ProxyException(ProxyInfo.ProxyType.SOCKS5, - "server returns " + buf[1]); - } + if (buf[1] != 0) { + throw new ProxyException(ProxyInfo.ProxyType.SOCKS5, + "server returns " + buf[1]); + } - switch (buf[3] & 0xff) { - case 1: - fill(in, buf, 6); - break; - case 3: - fill(in, buf, 1); - fill(in, buf, (buf[0] & 0xff) + 2); - break; - case 4: - fill(in, buf, 18); - break; - default: - } - } - catch (RuntimeException e) { - throw e; - } - catch (Exception e) { - CloseableUtil.maybeClose(socket, LOGGER); - // TODO convert to IOException(e) when minimum Android API level is 9 or higher - throw new IOException(e.getLocalizedMessage()); + switch (buf[3] & 0xff) { + case 1: + fill(in, buf, 6); + break; + case 3: + fill(in, buf, 1); + fill(in, buf, (buf[0] & 0xff) + 2); + break; + case 4: + fill(in, buf, 18); + break; + default: } } diff --git a/smack-tcp/src/main/java/org/jivesoftware/smack/tcp/XMPPTCPConnection.java b/smack-tcp/src/main/java/org/jivesoftware/smack/tcp/XMPPTCPConnection.java index df6a83156..c4ee49ca9 100644 --- a/smack-tcp/src/main/java/org/jivesoftware/smack/tcp/XMPPTCPConnection.java +++ b/smack-tcp/src/main/java/org/jivesoftware/smack/tcp/XMPPTCPConnection.java @@ -606,6 +606,7 @@ public class XMPPTCPConnection extends AbstractXMPPConnection { try { proxyInfo.getProxySocketConnection().connect(socket, host, port, timeout); } catch (IOException e) { + CloseableUtil.maybeClose(socket, LOGGER); hostAddress.setException(e); failedAddresses.add(hostAddress); continue; From 14cc0c1e418c9ea6a506065c5b43793a2ac918fe Mon Sep 17 00:00:00 2001 From: Florian Schmaus Date: Mon, 18 Nov 2019 19:37:10 +0100 Subject: [PATCH 2/3] proxy: do not call setTcpNoDelay() but instead flush the stream --- .../jivesoftware/smack/proxy/Socks4ProxySocketConnection.java | 2 +- .../jivesoftware/smack/proxy/Socks5ProxySocketConnection.java | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java index 614957f87..edab289f1 100644 --- a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java +++ b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java @@ -46,7 +46,6 @@ public class Socks4ProxySocketConnection implements ProxySocketConnection { socket.connect(new InetSocketAddress(proxy_host, proxy_port), timeout); InputStream in = socket.getInputStream(); OutputStream out = socket.getOutputStream(); - socket.setTcpNoDelay(true); byte[] buf = new byte[1024]; int index = 0; @@ -88,6 +87,7 @@ public class Socks4ProxySocketConnection implements ProxySocketConnection { } buf[index++] = 0; out.write(buf, 0, index); + out.flush(); /* The SOCKS server checks to see whether such a request should be granted diff --git a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java index c91402218..299cf1d89 100644 --- a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java +++ b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java @@ -48,8 +48,6 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { InputStream in = socket.getInputStream(); OutputStream out = socket.getOutputStream(); - socket.setTcpNoDelay(true); - byte[] buf = new byte[1024]; int index = 0; @@ -81,6 +79,7 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { buf[index++] = 2; // USERNAME/PASSWORD out.write(buf, 0, index); + out.flush(); /* The server selects from one of the methods given in METHODS, and @@ -205,6 +204,7 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { buf[index++] = (byte) (port & 0xff); out.write(buf, 0, index); + out.flush(); /* The SOCKS request information is sent by the client as soon as it has From a6e2523648b0fa15fdfc984203b8ffc698a84dd3 Mon Sep 17 00:00:00 2001 From: Florian Schmaus Date: Mon, 18 Nov 2019 21:08:03 +0100 Subject: [PATCH 3/3] proxy: modernize code by using (ByteArrayOutput|DataInput)Stream --- .../proxy/Socks4ProxySocketConnection.java | 54 ++++----- .../proxy/Socks5ProxySocketConnection.java | 108 +++++++++--------- .../smack/util/OutputStreamUtil.java | 39 +++++++ 3 files changed, 113 insertions(+), 88 deletions(-) create mode 100644 smack-core/src/main/java/org/jivesoftware/smack/util/OutputStreamUtil.java diff --git a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java index edab289f1..a30777a34 100644 --- a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java +++ b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks4ProxySocketConnection.java @@ -16,6 +16,8 @@ */ package org.jivesoftware.smack.proxy; +import java.io.ByteArrayOutputStream; +import java.io.DataInputStream; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; @@ -24,6 +26,8 @@ import java.net.InetSocketAddress; import java.net.Socket; import java.nio.charset.StandardCharsets; +import org.jivesoftware.smack.util.OutputStreamUtil; + /** * Socket factory for socks4 proxy. * @@ -45,10 +49,11 @@ public class Socks4ProxySocketConnection implements ProxySocketConnection { socket.connect(new InetSocketAddress(proxy_host, proxy_port), timeout); InputStream in = socket.getInputStream(); + DataInputStream dis = new DataInputStream(in); OutputStream out = socket.getOutputStream(); - byte[] buf = new byte[1024]; - int index = 0; + ByteArrayOutputStream outBuf = new ByteArrayOutputStream(); + byte[] inBuf; /* 1) CONNECT @@ -68,26 +73,22 @@ public class Socks4ProxySocketConnection implements ProxySocketConnection { of all zero bits. */ - buf[index++] = 4; - buf[index++] = 1; + outBuf.write(4); + outBuf.write(1); - buf[index++] = (byte) (port >>> 8); - buf[index++] = (byte) (port & 0xff); + outBuf.write(port >>> 8); + outBuf.write(port & 0xff); InetAddress inetAddress = InetAddress.getByName(proxy_host); byte[] byteAddress = inetAddress.getAddress(); - for (int i = 0; i < byteAddress.length; i++) { - buf[index++] = byteAddress[i]; - } + outBuf.write(byteAddress); if (user != null) { byte[] userBytes = user.getBytes(StandardCharsets.UTF_8); - System.arraycopy(userBytes, 0, buf, index, user.length()); - index += user.length(); + outBuf.write(userBytes); } - buf[index++] = 0; - out.write(buf, 0, index); - out.flush(); + outBuf.write(0); + OutputStreamUtil.writeResetAndFlush(outBuf, out); /* The SOCKS server checks to see whether such a request should be granted @@ -116,26 +117,17 @@ public class Socks4ProxySocketConnection implements ProxySocketConnection { The remaining fields are ignored. */ - int len = 6; - int s = 0; - while (s < len) { - int i = in.read(buf, s, len - s); - if (i <= 0) { - throw new ProxyException(ProxyInfo.ProxyType.SOCKS4, - "stream is closed"); - } - s += i; - } - if (buf[0] != 0) { + inBuf = new byte[6]; + dis.readFully(inBuf); + if (inBuf[0] != 0) { throw new ProxyException(ProxyInfo.ProxyType.SOCKS4, - "server returns VN " + buf[0]); + "server returns VN " + inBuf[0]); } - if (buf[1] != 90) { - String message = "ProxySOCKS4: server returns CD " + buf[1]; + if (inBuf[1] != 90) { + String message = "ProxySOCKS4: server returns CD " + inBuf[1]; throw new ProxyException(ProxyInfo.ProxyType.SOCKS4, message); } - byte[] temp = new byte[2]; - in.read(temp, 0, 2); + inBuf = new byte[2]; + dis.readFully(inBuf); } - } diff --git a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java index 299cf1d89..d81f4d65f 100644 --- a/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java +++ b/smack-core/src/main/java/org/jivesoftware/smack/proxy/Socks5ProxySocketConnection.java @@ -16,6 +16,8 @@ */ package org.jivesoftware.smack.proxy; +import java.io.ByteArrayOutputStream; +import java.io.DataInputStream; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; @@ -23,6 +25,8 @@ import java.net.InetSocketAddress; import java.net.Socket; import java.nio.charset.StandardCharsets; +import org.jivesoftware.smack.util.OutputStreamUtil; + /** * Socket factory for Socks5 proxy. * @@ -46,10 +50,11 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { socket.connect(new InetSocketAddress(proxy_host, proxy_port), timeout); InputStream in = socket.getInputStream(); + DataInputStream dis = new DataInputStream(in); OutputStream out = socket.getOutputStream(); - byte[] buf = new byte[1024]; - int index = 0; + ByteArrayOutputStream outBuf = new ByteArrayOutputStream(); + byte[] inBuf; /* +----+----------+----------+ @@ -72,14 +77,13 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { o X'FF' NO ACCEPTABLE METHODS */ - buf[index++] = 5; + outBuf.write(5); - buf[index++] = 2; - buf[index++] = 0; // NO AUTHENTICATION REQUIRED - buf[index++] = 2; // USERNAME/PASSWORD + outBuf.write(2); + outBuf.write(0); // NO AUTHENTICATION REQUIRED + outBuf.write(2); // USERNAME/PASSWORD - out.write(buf, 0, index); - out.flush(); + OutputStreamUtil.writeResetAndFlush(outBuf, out); /* The server selects from one of the methods given in METHODS, and @@ -91,10 +95,11 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { | 1 | 1 | +----+--------+ */ - fill(in, buf, 2); + inBuf = new byte[2]; + dis.readFully(inBuf); boolean check = false; - switch (buf[1] & 0xff) { + switch (inBuf[1] & 0xff) { case 0: // NO AUTHENTICATION REQUIRED check = true; break; @@ -122,19 +127,16 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { PASSWD field that follows. The PASSWD field contains the password association with the given UNAME. */ - index = 0; - buf[index++] = 1; - buf[index++] = (byte) user.length(); + outBuf.write(1); byte[] userBytes = user.getBytes(StandardCharsets.UTF_8); - System.arraycopy(userBytes, 0, buf, index, - user.length()); - index += user.length(); + OutputStreamUtil.writeByteSafe(outBuf, userBytes.length, "Username to long"); + outBuf.write(userBytes); + byte[] passwordBytes = passwd.getBytes(StandardCharsets.UTF_8); - buf[index++] = (byte) passwordBytes.length; - System.arraycopy(passwordBytes, 0, buf, index, - passwd.length()); - index += passwd.length(); - out.write(buf, 0, index); + OutputStreamUtil.writeByteSafe(outBuf, passwordBytes.length, "Password to long"); + outBuf.write(passwordBytes); + + OutputStreamUtil.writeResetAndFlush(outBuf, out); /* The server verifies the supplied UNAME and PASSWD, and sends the @@ -150,8 +152,9 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { `failure' (STATUS value other than X'00') status, it MUST close the connection. */ - fill(in, buf, 2); - if (buf[1] == 0) { + inBuf = new byte[2]; + dis.readFully(inBuf); + if (inBuf[1] == 0) { check = true; } break; @@ -189,22 +192,19 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { order */ - index = 0; - buf[index++] = 5; - buf[index++] = 1; // CONNECT - buf[index++] = 0; + outBuf.write(5); + outBuf.write(1); // CONNECT + outBuf.write(0); byte[] hostb = host.getBytes(StandardCharsets.UTF_8); int len = hostb.length; - buf[index++] = 3; // DOMAINNAME - buf[index++] = (byte) len; - System.arraycopy(hostb, 0, buf, index, len); - index += len; - buf[index++] = (byte) (port >>> 8); - buf[index++] = (byte) (port & 0xff); + outBuf.write(3); // DOMAINNAME + OutputStreamUtil.writeByteSafe(outBuf, len, "Hostname too long"); + outBuf.write(hostb); + outBuf.write(port >>> 8); + outBuf.write(port & 0xff); - out.write(buf, 0, index); - out.flush(); + OutputStreamUtil.writeResetAndFlush(outBuf, out); /* The SOCKS request information is sent by the client as soon as it has @@ -241,39 +241,33 @@ public class Socks5ProxySocketConnection implements ProxySocketConnection { o BND.PORT server bound port in network octet order */ - fill(in, buf, 4); + inBuf = new byte[4]; + dis.readFully(inBuf); - if (buf[1] != 0) { + if (inBuf[1] != 0) { throw new ProxyException(ProxyInfo.ProxyType.SOCKS5, - "server returns " + buf[1]); + "server returns " + inBuf[1]); } - switch (buf[3] & 0xff) { + final int addressBytes; + // TODO: Use Byte.toUnsignedInt() once Smack's minimum Android SDK level is 26 or higher. + final int atyp = inBuf[3] & 0xff; + switch (atyp) { case 1: - fill(in, buf, 6); + addressBytes = 4; break; case 3: - fill(in, buf, 1); - fill(in, buf, (buf[0] & 0xff) + 2); + byte domainnameLengthByte = dis.readByte(); + // TODO: Use Byte.toUnsignedInt() once Smack's minimum Android SDK level is 26 or higher. + addressBytes = domainnameLengthByte & 0xff; break; case 4: - fill(in, buf, 18); + addressBytes = 16; break; default: + throw new IOException("Unknown ATYP value: " + atyp); } + inBuf = new byte[addressBytes + 2]; + dis.readFully(inBuf); } - - private static void fill(InputStream in, byte[] buf, int len) - throws IOException { - int s = 0; - while (s < len) { - int i = in.read(buf, s, len - s); - if (i <= 0) { - throw new ProxyException(ProxyInfo.ProxyType.SOCKS5, "stream " + - "is closed"); - } - s += i; - } - } - } diff --git a/smack-core/src/main/java/org/jivesoftware/smack/util/OutputStreamUtil.java b/smack-core/src/main/java/org/jivesoftware/smack/util/OutputStreamUtil.java new file mode 100644 index 000000000..9f27a09a3 --- /dev/null +++ b/smack-core/src/main/java/org/jivesoftware/smack/util/OutputStreamUtil.java @@ -0,0 +1,39 @@ +/** + * + * Copyright 2019 Florian Schmaus + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.jivesoftware.smack.util; + +import java.io.ByteArrayOutputStream; +import java.io.IOException; +import java.io.OutputStream; + +public class OutputStreamUtil { + + public static void writeByteSafe(OutputStream outputStream, int i, String message) throws IOException { + if (i < 0 || i > 0xff) { + throw new IOException(message + ". The value " + i + " is not within the allowed range for bytes"); + } + outputStream.write(i); + } + + public static void writeResetAndFlush(ByteArrayOutputStream byteArrayOutputStream, OutputStream outputStream) + throws IOException { + byteArrayOutputStream.writeTo(outputStream); + byteArrayOutputStream.reset(); + outputStream.flush(); + } + +}