229 lines
9.0 KiB
Java
229 lines
9.0 KiB
Java
/*
|
|
* 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.tomcat.websocket;
|
|
|
|
import java.net.URI;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Queue;
|
|
import java.util.concurrent.CountDownLatch;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
import javax.websocket.ClientEndpointConfig;
|
|
import javax.websocket.ClientEndpointConfig.Configurator;
|
|
import javax.websocket.ContainerProvider;
|
|
import javax.websocket.Session;
|
|
import javax.websocket.WebSocketContainer;
|
|
|
|
import org.junit.Assert;
|
|
import org.junit.Test;
|
|
|
|
import org.apache.catalina.Context;
|
|
import org.apache.catalina.authenticator.AuthenticatorBase;
|
|
import org.apache.catalina.servlets.DefaultServlet;
|
|
import org.apache.catalina.startup.Tomcat;
|
|
import org.apache.tomcat.util.descriptor.web.LoginConfig;
|
|
import org.apache.tomcat.util.descriptor.web.SecurityCollection;
|
|
import org.apache.tomcat.util.descriptor.web.SecurityConstraint;
|
|
import org.apache.tomcat.websocket.TesterMessageCountClient.BasicText;
|
|
import org.apache.tomcat.websocket.TesterMessageCountClient.TesterProgrammaticEndpoint;
|
|
|
|
public class TestWebSocketFrameClient extends WebSocketBaseTest {
|
|
|
|
private static final String USER = "Aladdin";
|
|
private static final String PWD = "open sesame";
|
|
private static final String ROLE = "role";
|
|
private static final String URI_PROTECTED = "/foo";
|
|
|
|
@Test
|
|
public void testConnectToServerEndpoint() throws Exception {
|
|
Tomcat tomcat = getTomcatInstance();
|
|
// No file system docBase required
|
|
Context ctx = tomcat.addContext("", null);
|
|
ctx.addApplicationListener(TesterFirehoseServer.Config.class.getName());
|
|
Tomcat.addServlet(ctx, "default", new DefaultServlet());
|
|
ctx.addServletMappingDecoded("/", "default");
|
|
|
|
tomcat.start();
|
|
|
|
WebSocketContainer wsContainer = ContainerProvider.getWebSocketContainer();
|
|
|
|
// BZ 62596
|
|
final StringBuilder dummyValue = new StringBuilder(4000);
|
|
for (int i = 0; i < 4000; i++) {
|
|
dummyValue.append('A');
|
|
}
|
|
ClientEndpointConfig clientEndpointConfig =
|
|
ClientEndpointConfig.Builder.create().configurator(new Configurator() {
|
|
@Override
|
|
public void beforeRequest(Map<String, List<String>> headers) {
|
|
headers.put("Dummy", Collections.singletonList(dummyValue.toString()));
|
|
super.beforeRequest(headers);
|
|
}
|
|
}).build();
|
|
|
|
Session wsSession = wsContainer.connectToServer(
|
|
TesterProgrammaticEndpoint.class,
|
|
clientEndpointConfig,
|
|
new URI("ws://localhost:" + getPort() +
|
|
TesterFirehoseServer.Config.PATH));
|
|
CountDownLatch latch =
|
|
new CountDownLatch(TesterFirehoseServer.MESSAGE_COUNT);
|
|
BasicText handler = new BasicText(latch);
|
|
wsSession.addMessageHandler(handler);
|
|
wsSession.getBasicRemote().sendText("Hello");
|
|
|
|
System.out.println("Sent Hello message, waiting for data");
|
|
|
|
// Ignore the latch result as the message count test below will tell us
|
|
// if the right number of messages arrived
|
|
handler.getLatch().await(TesterFirehoseServer.WAIT_TIME_MILLIS,
|
|
TimeUnit.MILLISECONDS);
|
|
|
|
Queue<String> messages = handler.getMessages();
|
|
Assert.assertEquals(
|
|
TesterFirehoseServer.MESSAGE_COUNT, messages.size());
|
|
for (String message : messages) {
|
|
Assert.assertEquals(TesterFirehoseServer.MESSAGE, message);
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public void testConnectToRootEndpoint() throws Exception {
|
|
Tomcat tomcat = getTomcatInstance();
|
|
// No file system docBase required
|
|
Context ctx = tomcat.addContext("", null);
|
|
ctx.addApplicationListener(TesterEchoServer.Config.class.getName());
|
|
Tomcat.addServlet(ctx, "default", new DefaultServlet());
|
|
ctx.addServletMappingDecoded("/", "default");
|
|
Context ctx2 = tomcat.addContext("/foo", null);
|
|
ctx2.addApplicationListener(TesterEchoServer.Config.class.getName());
|
|
Tomcat.addServlet(ctx2, "default", new DefaultServlet());
|
|
ctx2.addServletMappingDecoded("/", "default");
|
|
|
|
tomcat.start();
|
|
|
|
echoTester("",null);
|
|
echoTester("/",null);
|
|
echoTester("/foo",null);
|
|
echoTester("/foo/",null);
|
|
}
|
|
|
|
public void echoTester(String path, ClientEndpointConfig clientEndpointConfig)
|
|
throws Exception {
|
|
WebSocketContainer wsContainer = ContainerProvider.getWebSocketContainer();
|
|
|
|
if (clientEndpointConfig == null) {
|
|
clientEndpointConfig = ClientEndpointConfig.Builder.create().build();
|
|
}
|
|
Session wsSession = wsContainer.connectToServer(TesterProgrammaticEndpoint.class,
|
|
clientEndpointConfig, new URI("ws://localhost:" + getPort() + path));
|
|
CountDownLatch latch = new CountDownLatch(1);
|
|
BasicText handler = new BasicText(latch);
|
|
wsSession.addMessageHandler(handler);
|
|
wsSession.getBasicRemote().sendText("Hello");
|
|
|
|
boolean latchResult = handler.getLatch().await(10, TimeUnit.SECONDS);
|
|
Assert.assertTrue(latchResult);
|
|
|
|
Queue<String> messages = handler.getMessages();
|
|
Assert.assertEquals(1, messages.size());
|
|
for (String message : messages) {
|
|
Assert.assertEquals("Hello", message);
|
|
}
|
|
wsSession.close();
|
|
}
|
|
|
|
@Test
|
|
public void testConnectToBasicEndpoint() throws Exception {
|
|
|
|
Tomcat tomcat = getTomcatInstance();
|
|
Context ctx = tomcat.addContext(URI_PROTECTED, null);
|
|
ctx.addApplicationListener(TesterEchoServer.Config.class.getName());
|
|
Tomcat.addServlet(ctx, "default", new DefaultServlet());
|
|
ctx.addServletMappingDecoded("/", "default");
|
|
|
|
SecurityCollection collection = new SecurityCollection();
|
|
collection.addPatternDecoded("/");
|
|
String utf8User = "test";
|
|
String utf8Pass = "123\u00A3"; // pound sign
|
|
|
|
tomcat.addUser(utf8User, utf8Pass);
|
|
tomcat.addRole(utf8User, ROLE);
|
|
|
|
SecurityConstraint sc = new SecurityConstraint();
|
|
sc.addAuthRole(ROLE);
|
|
sc.addCollection(collection);
|
|
ctx.addConstraint(sc);
|
|
|
|
LoginConfig lc = new LoginConfig();
|
|
lc.setAuthMethod("BASIC");
|
|
ctx.setLoginConfig(lc);
|
|
|
|
AuthenticatorBase basicAuthenticator = new org.apache.catalina.authenticator.BasicAuthenticator();
|
|
ctx.getPipeline().addValve(basicAuthenticator);
|
|
|
|
tomcat.start();
|
|
|
|
ClientEndpointConfig clientEndpointConfig = ClientEndpointConfig.Builder.create().build();
|
|
clientEndpointConfig.getUserProperties().put(Constants.WS_AUTHENTICATION_USER_NAME, utf8User);
|
|
clientEndpointConfig.getUserProperties().put(Constants.WS_AUTHENTICATION_PASSWORD, utf8Pass);
|
|
|
|
echoTester(URI_PROTECTED, clientEndpointConfig);
|
|
|
|
}
|
|
|
|
@Test
|
|
public void testConnectToDigestEndpoint() throws Exception {
|
|
|
|
Tomcat tomcat = getTomcatInstance();
|
|
Context ctx = tomcat.addContext(URI_PROTECTED, null);
|
|
ctx.addApplicationListener(TesterEchoServer.Config.class.getName());
|
|
Tomcat.addServlet(ctx, "default", new DefaultServlet());
|
|
ctx.addServletMappingDecoded("/", "default");
|
|
|
|
SecurityCollection collection = new SecurityCollection();
|
|
collection.addPatternDecoded("/*");
|
|
|
|
tomcat.addUser(USER, PWD);
|
|
tomcat.addRole(USER, ROLE);
|
|
|
|
SecurityConstraint sc = new SecurityConstraint();
|
|
sc.addAuthRole(ROLE);
|
|
sc.addCollection(collection);
|
|
ctx.addConstraint(sc);
|
|
|
|
LoginConfig lc = new LoginConfig();
|
|
lc.setAuthMethod("DIGEST");
|
|
ctx.setLoginConfig(lc);
|
|
|
|
AuthenticatorBase digestAuthenticator = new org.apache.catalina.authenticator.DigestAuthenticator();
|
|
ctx.getPipeline().addValve(digestAuthenticator);
|
|
|
|
tomcat.start();
|
|
|
|
ClientEndpointConfig clientEndpointConfig = ClientEndpointConfig.Builder.create().build();
|
|
clientEndpointConfig.getUserProperties().put(Constants.WS_AUTHENTICATION_USER_NAME, USER);
|
|
clientEndpointConfig.getUserProperties().put(Constants.WS_AUTHENTICATION_PASSWORD,PWD);
|
|
|
|
echoTester(URI_PROTECTED, clientEndpointConfig);
|
|
|
|
}
|
|
|
|
}
|