8334719: (se) Deferred close of SelectableChannel may result in a Selector doing the final close before concurrent I/O on channel has completed

Reviewed-by: lucy
Backport-of: 9bb675f89dd1eeec423ca96cb3f96d29f5de477c

Author: GoeLin

src/java.base/share/classes/sun/nio/ch/DatagramChannelImpl.java

@@ -1927,6 +1927,11 @@ protected void implCloseSelectableChannel() throws IOException {
 
     @Override
     public void kill() {
+        // wait for any read/write operations to complete before trying to close
+        readLock.lock();
+        readLock.unlock();
+        writeLock.lock();
+        writeLock.unlock();
         synchronized (stateLock) {
             if (state == ST_CLOSING) {
                 tryFinishClose();

src/java.base/share/classes/sun/nio/ch/ServerSocketChannelImpl.java

@@ -653,6 +653,9 @@ protected void implCloseSelectableChannel() throws IOException {
 
     @Override
     public void kill() {
+        // wait for any accept operation to complete before trying to close
+        acceptLock.lock();
+        acceptLock.unlock();
         synchronized (stateLock) {
             if (state == ST_CLOSING) {
                 tryFinishClose();

src/java.base/share/classes/sun/nio/ch/SocketChannelImpl.java

@@ -1127,6 +1127,11 @@ protected void implCloseSelectableChannel() throws IOException {
 
     @Override
     public void kill() {
+        // wait for any read/write operations to complete before trying to close
+        readLock.lock();
+        readLock.unlock();
+        writeLock.lock();
+        writeLock.unlock();
         synchronized (stateLock) {
             if (state == ST_CLOSING) {
                 tryFinishClose();

src/java.base/unix/classes/sun/nio/ch/SinkChannelImpl.java

@@ -201,6 +201,9 @@ protected void implCloseSelectableChannel() throws IOException {
 
     @Override
     public void kill() {
+        // wait for any write operation to complete before trying to close
+        writeLock.lock();
+        writeLock.unlock();
         synchronized (stateLock) {
             if (state == ST_CLOSING) {
                 tryFinishClose();

src/java.base/unix/classes/sun/nio/ch/SourceChannelImpl.java

@@ -200,6 +200,9 @@ protected void implCloseSelectableChannel() throws IOException {
     }
     @Override
     public void kill() {
+        // wait for any read operation to complete before trying to close
+        readLock.lock();
+        readLock.unlock();
         synchronized (stateLock) {
             assert !isOpen();
             if (state == ST_CLOSING) {