| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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.apache.ignite.jvmtest; |
| |
| import java.io.IOException; |
| import java.net.InetAddress; |
| import java.net.InetSocketAddress; |
| import java.net.ServerSocket; |
| import java.net.Socket; |
| import java.util.concurrent.Callable; |
| import java.util.concurrent.atomic.AtomicBoolean; |
| import java.util.concurrent.atomic.AtomicReference; |
| import javax.swing.JOptionPane; |
| import org.apache.ignite.internal.IgniteInternalFuture; |
| import org.apache.ignite.internal.util.typedef.X; |
| import org.apache.ignite.internal.util.typedef.internal.U; |
| import org.apache.ignite.testframework.GridTestUtils; |
| import org.jetbrains.annotations.Nullable; |
| import org.junit.Test; |
| |
| /** |
| * |
| */ |
| public class NetworkFailureTest { |
| /** |
| * @throws Exception If failed. |
| */ |
| @Test |
| public void testNetworkFailure() throws Exception { |
| final AtomicBoolean done = new AtomicBoolean(); |
| |
| final InetAddress addr = InetAddress.getByName("192.168.0.100"); |
| |
| IgniteInternalFuture<?> fut1 = GridTestUtils.runMultiThreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| ServerSocket srvSock = null; |
| Socket sock = null; |
| |
| try { |
| srvSock = new ServerSocket(60000, 0, addr); |
| |
| sock = srvSock.accept(); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", sndBuf=" + sock.getSendBufferSize() + ']'); |
| |
| sock.setKeepAlive(true); |
| |
| sock.setSoTimeout(2000); |
| |
| sock.setSendBufferSize(256 * 1024); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", rcvBuf=" + sock.getReceiveBufferSize() + ']'); |
| |
| while (!done.get()) |
| X.println("Read from socket: " + sock.getInputStream().read()); |
| |
| return null; |
| } |
| finally { |
| U.closeQuiet(srvSock); |
| U.closeQuiet(sock); |
| } |
| } |
| }, |
| 1, |
| "server" |
| ); |
| |
| IgniteInternalFuture<?> fut2 = GridTestUtils.runMultiThreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| Socket sock = null; |
| |
| try { |
| sock = new Socket(addr, 60000); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", sndBuf=" + sock.getSendBufferSize() + ']'); |
| |
| sock.setKeepAlive(true); |
| |
| sock.setSoTimeout(2000); |
| |
| sock.setSendBufferSize(256 * 1024); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", sndBuf=" + sock.getSendBufferSize() + ']'); |
| |
| int i = 0; |
| |
| while (!done.get()) { |
| sock.getOutputStream().write(++i); |
| |
| sock.getOutputStream().flush(); |
| |
| X.println("Wrote to socket: " + i); |
| |
| X.println("Socket connected: " + sock.isConnected()); |
| X.println("Socket keep alive: " + sock.getKeepAlive()); |
| |
| U.sleep(1000); |
| } |
| |
| return null; |
| } |
| finally { |
| U.closeQuiet(sock); |
| } |
| } |
| }, |
| 1, |
| "client" |
| ); |
| |
| JOptionPane.showMessageDialog(null, "Unplug network cable." + U.nl() + |
| "Press OK to finish."); |
| |
| done.set(true); |
| |
| fut1.get(); |
| fut2.get(); |
| } |
| |
| /** |
| * @throws Exception If failed. |
| */ |
| @Test |
| public void testReadTimeout() throws Exception { |
| final InetAddress addr = InetAddress.getByName("192.168.3.10"); |
| |
| IgniteInternalFuture<?> fut1 = GridTestUtils.runMultiThreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| ServerSocket srvSock = null; |
| Socket sock = null; |
| |
| try { |
| srvSock = new ServerSocket(60000, 0, addr); |
| |
| sock = srvSock.accept(); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", sndBuf=" + sock.getSendBufferSize() + |
| ", NODELAY=" + sock.getTcpNoDelay() + ']'); |
| |
| sock.setSoTimeout(2000); |
| sock.setTcpNoDelay(true); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", sndBuf=" + sock.getSendBufferSize() + |
| ", NODELAY=" + sock.getTcpNoDelay() + ']'); |
| |
| sock.getInputStream().read(); |
| } |
| catch (IOException e) { |
| X.println("Caught expected exception: " + e); |
| |
| e.printStackTrace(); |
| } |
| finally { |
| U.closeQuiet(srvSock); |
| U.closeQuiet(sock); |
| } |
| |
| return null; |
| } |
| }, |
| 1, |
| "server" |
| ); |
| |
| IgniteInternalFuture<?> fut2 = GridTestUtils.runMultiThreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| Socket sock = null; |
| |
| try { |
| sock = new Socket(addr, 60000); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", sndBuf=" + sock.getSendBufferSize() + |
| ", NODELAY=" + sock.getTcpNoDelay() + ']'); |
| |
| sock.setTcpNoDelay(true); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", sndBuf=" + sock.getSendBufferSize() + |
| ", NODELAY=" + sock.getTcpNoDelay() + ']'); |
| |
| Thread.sleep(10000); |
| |
| return null; |
| } |
| finally { |
| U.closeQuiet(sock); |
| } |
| } |
| }, |
| 1, |
| "client" |
| ); |
| |
| fut1.get(); |
| fut2.get(); |
| } |
| |
| /** |
| * @throws Exception If failed. |
| */ |
| @Test |
| public void testSocketCloseOnTimeout() throws Exception { |
| final AtomicBoolean done = new AtomicBoolean(); |
| |
| final InetAddress addr = InetAddress.getByName("192.168.0.100"); |
| |
| IgniteInternalFuture<?> fut1 = GridTestUtils.runMultiThreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| ServerSocket srvSock = null; |
| Socket sock = null; |
| |
| try { |
| srvSock = new ServerSocket(60000, 0, addr); |
| |
| sock = srvSock.accept(); |
| |
| while (!done.get()) |
| U.sleep(1000); |
| |
| return null; |
| } |
| finally { |
| U.closeQuiet(srvSock); |
| U.closeQuiet(sock); |
| } |
| } |
| }, |
| 1, |
| "server" |
| ); |
| |
| final AtomicReference<Socket> sockRef = new AtomicReference<>(); |
| |
| IgniteInternalFuture<?> fut2 = GridTestUtils.runMultiThreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| Socket sock = null; |
| |
| try { |
| sock = new Socket(addr, 60000); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", sndBuf=" + sock.getSendBufferSize() + ']'); |
| |
| sockRef.set(sock); |
| |
| sock.getOutputStream().write( |
| new byte[(sock.getSendBufferSize() + sock.getReceiveBufferSize()) * 2]); |
| |
| assert false : "Message has been written."; |
| } |
| catch (IOException e) { |
| X.println("Caught expected exception: " + e); |
| |
| e.printStackTrace(); |
| } |
| finally { |
| U.closeQuiet(sock); |
| } |
| |
| return null; |
| } |
| }, |
| 1, |
| "client" |
| ); |
| |
| IgniteInternalFuture<?> fut3 = GridTestUtils.runMultiThreadedAsync( |
| new Callable<Object>() { |
| @Nullable @Override public Object call() throws Exception { |
| while (true) { |
| U.sleep(1000); |
| |
| Socket sock = sockRef.get(); |
| |
| if (sock != null) { |
| U.sleep(1000); |
| |
| U.closeQuiet(sock); |
| |
| return null; |
| } |
| } |
| } |
| }, |
| 1, |
| "client" |
| ); |
| |
| fut2.get(); |
| fut3.get(); |
| |
| done.set(true); |
| |
| fut1.get(); |
| } |
| |
| /** |
| * @throws Exception If failed. |
| */ |
| @Test |
| public void testConnectionTime() throws Exception { |
| X.println("Unexistent host."); |
| checkConnection(InetAddress.getByName("192.168.0.222")); |
| |
| X.println("Unlistened port."); |
| checkConnection(InetAddress.getByName("192.168.0.1")); |
| } |
| |
| /** |
| * @param addr Address to check connection to. |
| */ |
| private void checkConnection(InetAddress addr) { |
| long start = System.currentTimeMillis(); |
| |
| Socket sock = null; |
| |
| try { |
| sock = openSocket(addr, 80); |
| } |
| catch (Exception e) { |
| X.println("Caught exception: " + e.getClass().getSimpleName() + " - " + e.getMessage()); |
| } |
| finally { |
| X.println("Time taken: " + (System.currentTimeMillis() - start)); |
| |
| U.closeQuiet(sock); |
| } |
| } |
| |
| /** |
| * @param addr Remote address. |
| * @param port Remote port. |
| * @return Opened socket. |
| * @throws IOException If failed. |
| */ |
| private Socket openSocket(InetAddress addr, int port) throws IOException { |
| Socket sock = new Socket(); |
| |
| sock.bind(new InetSocketAddress(InetAddress.getByName("192.168.0.100"), 0)); |
| |
| sock.connect(new InetSocketAddress(addr, port), 1); |
| |
| X.println("Socket [timeout=" + sock.getSoTimeout() + ", linger=" + sock.getSoLinger() + |
| ", sndBuf=" + sock.getSendBufferSize() + ", sndBuf=" + sock.getSendBufferSize() + ']'); |
| |
| return sock; |
| } |
| } |