Adjust Network's SocketFactory to restrict host name resolution to that Network

Change-Id: Iab9a5a2e060fe261f4be9ba974c1a55fb6b9c98b
This commit is contained in:
Paul Jensen
2014-05-27 10:19:39 -04:00
parent f1f6fc8a09
commit 486d6ac868

View File

@@ -94,11 +94,26 @@ public class Network implements Parcelable {
mNetId = netId;
}
private void connectToHost(Socket socket, String host, int port) throws IOException {
// Lookup addresses only on this Network.
InetAddress[] hostAddresses = getAllByName(host);
// Try all but last address ignoring exceptions.
for (int i = 0; i < hostAddresses.length - 1; i++) {
try {
socket.connect(new InetSocketAddress(hostAddresses[i], port));
return;
} catch (IOException e) {
}
}
// Try last address. Do throw exceptions.
socket.connect(new InetSocketAddress(hostAddresses[hostAddresses.length - 1], port));
}
@Override
public Socket createSocket(String host, int port, InetAddress localHost, int localPort) throws IOException {
Socket socket = createSocket();
socket.bind(new InetSocketAddress(localHost, localPort));
socket.connect(new InetSocketAddress(host, port));
connectToHost(socket, host, port);
return socket;
}
@@ -121,7 +136,7 @@ public class Network implements Parcelable {
@Override
public Socket createSocket(String host, int port) throws IOException {
Socket socket = createSocket();
socket.connect(new InetSocketAddress(host, port));
connectToHost(socket, host, port);
return socket;
}