Eclipse
/
jetty.project
mirror of https://github.com/jetty/jetty.project.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Issue #3734 - throw ISE for WebSocket suspend after close (#4095) 

Signed-off-by: Lachlan Roberts <lachlan@webtide.com>
This commit is contained in:
Lachlan 2019-10-08 09:47:48 +11:00 committed by GitHub
parent 3f6119eb13
commit d47264960e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
jetty-websocket/jetty-websocket-common/src/main/java/org/eclipse/jetty/websocket/common/JettyWebSocketFrameHandler.java
View File

@ -48,7 +48,8 @@ public class JettyWebSocketFrameHandler implements FrameHandler
{
DEMANDING,
SUSPENDING,
SUSPENDED
SUSPENDED,
CLOSED
}
private final Logger log;
@ -191,7 +192,6 @@ public class JettyWebSocketFrameHandler implements FrameHandler
state = SuspendState.SUSPENDED;
return;
case SUSPENDED:
default:
throw new IllegalStateException();
}
@ -280,6 +280,12 @@ public class JettyWebSocketFrameHandler implements FrameHandler
@Override
public void onClosed(CloseStatus closeStatus, Callback callback)
{
synchronized (this)
{
// We are now closed and cannot suspend or resume
state = SuspendState.CLOSED;
}
try
{
if (closeHandle != null)
@ -405,18 +411,15 @@ public class JettyWebSocketFrameHandler implements FrameHandler
state = SuspendState.SUSPENDING;
break;
case SUSPENDED:
case SUSPENDING:
throw new IllegalStateException("Already Suspended");
default:
throw new IllegalStateException();
throw new IllegalStateException(state.name());
}
}
}
public void resume()
{
boolean needDemand = false;
Runnable delayedFrame = null;
synchronized (this)
{
@ -426,6 +429,7 @@ public class JettyWebSocketFrameHandler implements FrameHandler
throw new IllegalStateException("Already Resumed");
case SUSPENDED:
needDemand = true;
delayedFrame = delayedOnFrame;
delayedOnFrame = null;
state = SuspendState.DEMANDING;
@ -438,14 +442,17 @@ public class JettyWebSocketFrameHandler implements FrameHandler
break;
default:
throw new IllegalStateException();
throw new IllegalStateException(state.name());
}
}
if (delayedFrame != null)
delayedFrame.run();
else
session.getCoreSession().demand(1);
if (needDemand)
{
if (delayedFrame != null)
delayedFrame.run();
else
session.getCoreSession().demand(1);
}
}
private void demand()
@ -459,15 +466,12 @@ public class JettyWebSocketFrameHandler implements FrameHandler
demand = true;
break;
case SUSPENDED:
throw new IllegalStateException("Suspended");
case SUSPENDING:
state = SuspendState.SUSPENDED;
break;
default:
throw new IllegalStateException();
throw new IllegalStateException(state.name());
}
}

2
jetty-websocket/jetty-websocket-tests/src/test/java/org/eclipse/jetty/websocket/tests/SuspendResumeTest.java
View File

@ -36,7 +36,6 @@ import org.eclipse.jetty.websocket.server.JettyWebSocketServletFactory;
import org.eclipse.jetty.websocket.server.config.JettyWebSocketServletContainerInitializer;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
import static org.hamcrest.MatcherAssert.assertThat;
@ -170,7 +169,6 @@ public class SuspendResumeTest
assertNull(serverSocket.error);
}
@Disabled
@Test
public void testSuspendAfterClose() throws Exception
{

2
jetty-websocket/websocket-core/src/main/java/org/eclipse/jetty/websocket/core/internal/WebSocketCoreSession.java
View File

@ -470,7 +470,7 @@ public class WebSocketCoreSession implements IncomingFrames, FrameHandler.CoreSe
if (!demanding)
throw new IllegalStateException("FrameHandler is not demanding: " + this);
if (!sessionState.isInputOpen())
throw new IllegalStateException("FrameHandler input not open: " + this); // TODO Perhaps this is a NOOP?
throw new IllegalStateException("FrameHandler input not open: " + this);
connection.demand(n);
}