Ensure local addresses aren't null

libs/nio/src/main/java/org/elasticsearch/nio/NioChannel.java

@@ -19,7 +19,6 @@
 
 package org.elasticsearch.nio;
 
-import java.io.IOException;
 import java.net.InetSocketAddress;
 import java.nio.channels.NetworkChannel;
 import java.util.function.BiConsumer;
@@ -32,20 +31,10 @@
  */
 public abstract class NioChannel {
 
-    private final InetSocketAddress localAddress;
-
-    NioChannel(NetworkChannel socketChannel) throws IOException {
-        this.localAddress = (InetSocketAddress) socketChannel.getLocalAddress();
-    }
-
     public boolean isOpen() {
         return getContext().isOpen();
     }
 
-    public InetSocketAddress getLocalAddress() {
-        return localAddress;
-    }
-
     /**
      * Adds a close listener to the channel. Multiple close listeners can be added. There is no guarantee
      * about the order in which close listeners will be executed. If the channel is already closed, the
@@ -64,6 +53,8 @@ public void close() {
         getContext().closeChannel();
     }
 
+    public abstract InetSocketAddress getLocalAddress();
+
     public abstract NetworkChannel getRawChannel();
 
     public abstract ChannelContext<?> getContext();

libs/nio/src/main/java/org/elasticsearch/nio/NioServerSocketChannel.java

@@ -19,19 +19,19 @@
 
 package org.elasticsearch.nio;
 
-import java.io.IOException;
+import java.net.InetSocketAddress;
 import java.nio.channels.ServerSocketChannel;
 import java.util.concurrent.atomic.AtomicBoolean;
 
 public class NioServerSocketChannel extends NioChannel {
 
-    private final ServerSocketChannel socketChannel;
+    private final ServerSocketChannel serverSocketChannel;
     private final AtomicBoolean contextSet = new AtomicBoolean(false);
+    private volatile InetSocketAddress localAddress;
     private ServerChannelContext context;
 
-    public NioServerSocketChannel(ServerSocketChannel socketChannel) throws IOException {
-        super(socketChannel);
-        this.socketChannel = socketChannel;
+    public NioServerSocketChannel(ServerSocketChannel serverSocketChannel) {
+        this.serverSocketChannel = serverSocketChannel;
     }
 
     /**
@@ -48,9 +48,17 @@ public void setContext(ServerChannelContext context) {
         }
     }
 
+    @Override
+    public InetSocketAddress getLocalAddress() {
+        if (localAddress == null) {
+            localAddress = (InetSocketAddress) serverSocketChannel.socket().getLocalSocketAddress();
+        }
+        return localAddress;
+    }
+
     @Override
     public ServerSocketChannel getRawChannel() {
-        return socketChannel;
+        return serverSocketChannel;
     }
 
     @Override

libs/nio/src/main/java/org/elasticsearch/nio/NioSocketChannel.java

@@ -27,15 +27,20 @@
 
 public class NioSocketChannel extends NioChannel {
 
-    private final InetSocketAddress remoteAddress;
     private final AtomicBoolean contextSet = new AtomicBoolean(false);
     private final SocketChannel socketChannel;
+    private volatile InetSocketAddress remoteAddress;
+    private volatile InetSocketAddress localAddress;
     private SocketChannelContext context;
 
-    public NioSocketChannel(SocketChannel socketChannel) throws IOException {
-        super(socketChannel);
+    public NioSocketChannel(SocketChannel socketChannel) {
         this.socketChannel = socketChannel;
-        this.remoteAddress = (InetSocketAddress) socketChannel.getRemoteAddress();
+        try {
+            remoteAddress = (InetSocketAddress) socketChannel.getRemoteAddress();
+        } catch (IOException e) {
+            // This exception will be thrown if the channel is closed. But we do not really care when
+            // we are just attempting to read the remote address.
+        }
     }
 
     public void setContext(SocketChannelContext context) {
@@ -46,6 +51,14 @@ public void setContext(SocketChannelContext context) {
         }
     }
 
+    @Override
+    public InetSocketAddress getLocalAddress() {
+        if (localAddress == null) {
+            localAddress = (InetSocketAddress) socketChannel.socket().getLocalSocketAddress();
+        }
+        return localAddress;
+    }
+
     @Override
     public SocketChannel getRawChannel() {
         return socketChannel;

plugins/transport-nio/src/main/java/org/elasticsearch/http/nio/NioHttpChannel.java

@@ -24,12 +24,11 @@
 import org.elasticsearch.http.HttpResponse;
 import org.elasticsearch.nio.NioSocketChannel;
 
-import java.io.IOException;
 import java.nio.channels.SocketChannel;
 
 public class NioHttpChannel extends NioSocketChannel implements HttpChannel {
 
-    NioHttpChannel(SocketChannel socketChannel) throws IOException {
+    NioHttpChannel(SocketChannel socketChannel) {
         super(socketChannel);
     }
 

plugins/transport-nio/src/main/java/org/elasticsearch/transport/nio/NioTcpChannel.java

@@ -32,7 +32,7 @@ public class NioTcpChannel extends NioSocketChannel implements TcpChannel {
 
     private final String profile;
 
-    public NioTcpChannel(String profile, SocketChannel socketChannel) throws IOException {
+    public NioTcpChannel(String profile, SocketChannel socketChannel) {
         super(socketChannel);
         this.profile = profile;
     }

plugins/transport-nio/src/main/java/org/elasticsearch/transport/nio/NioTcpServerChannel.java

@@ -24,7 +24,6 @@
 import org.elasticsearch.nio.NioServerSocketChannel;
 import org.elasticsearch.transport.TcpChannel;
 
-import java.io.IOException;
 import java.net.InetSocketAddress;
 import java.nio.channels.ServerSocketChannel;
 
@@ -36,7 +35,7 @@ public class NioTcpServerChannel extends NioServerSocketChannel implements TcpCh
 
     private final String profile;
 
-    public NioTcpServerChannel(String profile, ServerSocketChannel socketChannel) throws IOException {
+    public NioTcpServerChannel(String profile, ServerSocketChannel socketChannel) {
         super(socketChannel);
         this.profile = profile;
     }
@@ -47,7 +46,7 @@ public void sendMessage(BytesReference reference, ActionListener<Void> listener)
     }
 
     @Override
-    public void setSoLinger(int value) throws IOException {
+    public void setSoLinger(int value) {
         throw new UnsupportedOperationException("Cannot set SO_LINGER on a server channel.");
     }
 

test/framework/src/main/java/org/elasticsearch/transport/nio/MockNioTransport.java

@@ -164,7 +164,7 @@ public MockSocketChannel createChannel(NioSelector selector, SocketChannel chann
 
         @Override
         public MockServerChannel createServerChannel(NioSelector selector, ServerSocketChannel channel) throws IOException {
-            MockServerChannel nioServerChannel = new MockServerChannel(profileName, channel, this, selector);
+            MockServerChannel nioServerChannel = new MockServerChannel(profileName, channel);
             Consumer<Exception> exceptionHandler = (e) -> logger.error(() ->
                 new ParameterizedMessage("exception from server channel caught on transport layer [{}]", channel), e);
             ServerChannelContext context = new ServerChannelContext(nioServerChannel, this, selector, MockNioTransport.this::acceptChannel,
@@ -195,8 +195,7 @@ private static class MockServerChannel extends NioServerSocketChannel implements
 
         private final String profile;
 
-        MockServerChannel(String profile, ServerSocketChannel channel, ChannelFactory<?, ?> channelFactory, NioSelector selector)
-            throws IOException {
+        MockServerChannel(String profile, ServerSocketChannel channel) {
             super(channel);
             this.profile = profile;
         }
@@ -236,8 +235,7 @@ private static class MockSocketChannel extends NioSocketChannel implements TcpCh
 
         private final String profile;
 
-        private MockSocketChannel(String profile, java.nio.channels.SocketChannel socketChannel, NioSelector selector)
-            throws IOException {
+        private MockSocketChannel(String profile, java.nio.channels.SocketChannel socketChannel, NioSelector selector) {
             super(socketChannel);
             this.profile = profile;
         }