49765
|
1 |
/*
|
|
2 |
* Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
|
|
3 |
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
|
4 |
*
|
|
5 |
* This code is free software; you can redistribute it and/or modify it
|
|
6 |
* under the terms of the GNU General Public License version 2 only, as
|
|
7 |
* published by the Free Software Foundation.
|
|
8 |
*
|
|
9 |
* This code is distributed in the hope that it will be useful, but WITHOUT
|
|
10 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
11 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
12 |
* version 2 for more details (a copy is included in the LICENSE file that
|
|
13 |
* accompanied this code).
|
|
14 |
*
|
|
15 |
* You should have received a copy of the GNU General Public License version
|
|
16 |
* 2 along with this work; if not, write to the Free Software Foundation,
|
|
17 |
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
18 |
*
|
|
19 |
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
|
20 |
* or visit www.oracle.com if you need additional information or have any
|
|
21 |
* questions.
|
|
22 |
*/
|
|
23 |
|
|
24 |
/*
|
|
25 |
* @test
|
|
26 |
* @build DummyWebSocketServer
|
|
27 |
* @run testng/othervm
|
|
28 |
* -Djdk.internal.httpclient.debug=true
|
|
29 |
* -Djdk.internal.httpclient.websocket.debug=true
|
|
30 |
* BlowupOutputQueue
|
|
31 |
*/
|
|
32 |
|
|
33 |
import org.testng.annotations.Test;
|
|
34 |
|
|
35 |
import java.io.IOException;
|
|
36 |
import java.net.http.WebSocket;
|
|
37 |
import java.nio.ByteBuffer;
|
|
38 |
import java.nio.CharBuffer;
|
|
39 |
import java.util.List;
|
|
40 |
import java.util.concurrent.CompletableFuture;
|
|
41 |
import java.util.concurrent.CompletionStage;
|
|
42 |
import java.util.concurrent.CountDownLatch;
|
|
43 |
import java.util.concurrent.TimeUnit;
|
|
44 |
import java.util.concurrent.TimeoutException;
|
|
45 |
|
|
46 |
import static java.net.http.HttpClient.newHttpClient;
|
|
47 |
import static org.testng.Assert.assertFalse;
|
|
48 |
|
|
49 |
public class BlowupOutputQueue extends PendingOperations {
|
|
50 |
|
|
51 |
CompletableFuture<WebSocket> cfText;
|
|
52 |
CompletableFuture<WebSocket> cfPing;
|
|
53 |
CompletableFuture<WebSocket> cfClose;
|
|
54 |
MockListener listener;
|
|
55 |
|
|
56 |
/*
|
|
57 |
* The idea is to arrange things such that the internal queue will be fully
|
|
58 |
* utilized and then make sure there won't be any errors caused by that.
|
|
59 |
*
|
|
60 |
* First, fill the queue with Text messages. Once done, send a Ping message.
|
|
61 |
* At this stage, there are at least 2 messages are in queue. Now, start
|
|
62 |
* receiving. Received Ping messages will cause automatic Pong replies. When
|
|
63 |
* all done, there will be at least 3 messages in the queue. (As at least
|
|
64 |
* the a single Ping has to be replied). Then send a Close message. Now
|
|
65 |
* there are at least 4 messages in the queue. Finally, receive the last
|
|
66 |
* message which is a Close message. This will cause an automatic reply with
|
|
67 |
* a Close message from the client side. All in all there should be at least
|
|
68 |
* 5 messages in the queue.
|
|
69 |
*/
|
|
70 |
@Test
|
|
71 |
public void full() throws Exception {
|
|
72 |
int N = 32;
|
|
73 |
int[] incoming = new int[2 * (N + 1)]; // 2 bytes per message
|
|
74 |
for (int i = 0; i < incoming.length - 2; i += 2) {
|
|
75 |
// <PING>
|
|
76 |
incoming[i + 0] = 0x89;
|
|
77 |
incoming[i + 1] = 0x00;
|
|
78 |
}
|
|
79 |
// <CLOSE>
|
|
80 |
incoming[incoming.length - 2] = 0x88;
|
|
81 |
incoming[incoming.length - 1] = 0x00;
|
|
82 |
|
|
83 |
repeatable(() -> {
|
|
84 |
CountDownLatch allButCloseReceived = new CountDownLatch(N);
|
|
85 |
server = Support.writingServer(incoming);
|
|
86 |
server.open();
|
|
87 |
listener = new MockListener() {
|
|
88 |
|
|
89 |
@Override
|
|
90 |
protected void replenish(WebSocket webSocket) {
|
|
91 |
/* do nothing */
|
|
92 |
}
|
|
93 |
|
|
94 |
@Override
|
|
95 |
protected CompletionStage<?> onPing0(WebSocket webSocket,
|
|
96 |
ByteBuffer message) {
|
|
97 |
allButCloseReceived.countDown();
|
|
98 |
return null;
|
|
99 |
}
|
|
100 |
};
|
|
101 |
webSocket = newHttpClient().newWebSocketBuilder()
|
|
102 |
.buildAsync(server.getURI(), listener)
|
|
103 |
.join();
|
|
104 |
CharBuffer data = CharBuffer.allocate(65536);
|
|
105 |
for (int i = 0; ; i++) { // fill up the send buffer
|
|
106 |
long start = System.currentTimeMillis();
|
|
107 |
System.out.printf("begin cycle #%s at %s%n", i, start);
|
|
108 |
cfText = webSocket.sendText(data, true);
|
|
109 |
try {
|
|
110 |
cfText.get(MAX_WAIT_SEC, TimeUnit.SECONDS);
|
|
111 |
data.clear();
|
|
112 |
} catch (TimeoutException e) {
|
|
113 |
break;
|
|
114 |
} finally {
|
|
115 |
long stop = System.currentTimeMillis();
|
|
116 |
System.out.printf("end cycle #%s at %s (%s ms)%n", i, stop, stop - start);
|
|
117 |
}
|
|
118 |
}
|
|
119 |
cfPing = webSocket.sendPing(ByteBuffer.allocate(125));
|
|
120 |
webSocket.request(N);
|
|
121 |
allButCloseReceived.await();
|
|
122 |
webSocket.request(1); // Receive the last message: Close
|
|
123 |
return null;
|
|
124 |
}, () -> cfText.isDone());
|
|
125 |
List<MockListener.Invocation> invocations = listener.invocations();
|
|
126 |
cfClose = webSocket.sendClose(WebSocket.NORMAL_CLOSURE, "ok");
|
|
127 |
|
|
128 |
assertFalse(invocations.contains(new MockListener.OnError(webSocket, IOException.class)));
|
|
129 |
assertFalse(cfText.isDone());
|
|
130 |
assertFalse(cfPing.isDone());
|
|
131 |
assertFalse(cfClose.isDone());
|
|
132 |
}
|
|
133 |
}
|