mirror of
https://github.com/vanitasvitae/Smack.git
synced 2024-06-26 13:24:49 +02:00
This also moves the logic to send error IQ replies from "when there is no IQ provider registerd" to "when there is no IQ request handler registered". Which has for example the advantage that IQ parsing no longer asks for a connection instance.
311 lines
12 KiB
Java
311 lines
12 KiB
Java
/**
|
|
*
|
|
* Copyright the original author or authors
|
|
*
|
|
* Licensed 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.jivesoftware.smackx.bytestreams.socks5;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.mockito.Mockito.mock;
|
|
import static org.mockito.Mockito.never;
|
|
import static org.mockito.Mockito.verify;
|
|
|
|
import org.jivesoftware.smack.XMPPConnection;
|
|
import org.jivesoftware.smack.packet.IQ;
|
|
import org.jivesoftware.smack.packet.XMPPError;
|
|
import org.jivesoftware.smackx.bytestreams.BytestreamRequest;
|
|
import org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream;
|
|
import org.jivesoftware.smackx.disco.ServiceDiscoveryManager;
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
import org.mockito.ArgumentCaptor;
|
|
import org.powermock.reflect.Whitebox;
|
|
|
|
/**
|
|
* Test for the InitiationListener class.
|
|
*
|
|
* @author Henning Staib
|
|
*/
|
|
public class InitiationListenerTest {
|
|
|
|
String initiatorJID = "initiator@xmpp-server/Smack";
|
|
String targetJID = "target@xmpp-server/Smack";
|
|
String xmppServer = "xmpp-server";
|
|
String proxyJID = "proxy.xmpp-server";
|
|
String proxyAddress = "127.0.0.1";
|
|
String sessionID = "session_id";
|
|
|
|
XMPPConnection connection;
|
|
Socks5BytestreamManager byteStreamManager;
|
|
InitiationListener initiationListener;
|
|
Bytestream initBytestream;
|
|
|
|
/**
|
|
* Initialize fields used in the tests.
|
|
*/
|
|
@Before
|
|
public void setup() {
|
|
|
|
// mock connection
|
|
connection = mock(XMPPConnection.class);
|
|
|
|
// create service discovery manager for mocked connection
|
|
ServiceDiscoveryManager.getInstanceFor(connection);
|
|
|
|
// initialize Socks5ByteStreamManager to get the InitiationListener
|
|
byteStreamManager = Socks5BytestreamManager.getBytestreamManager(connection);
|
|
|
|
// get the InitiationListener from Socks5ByteStreamManager
|
|
initiationListener = Whitebox.getInternalState(byteStreamManager, InitiationListener.class);
|
|
|
|
// create a SOCKS5 Bytestream initiation packet
|
|
initBytestream = Socks5PacketUtils.createBytestreamInitiation(initiatorJID, targetJID,
|
|
sessionID);
|
|
initBytestream.addStreamHost(proxyJID, proxyAddress, 7777);
|
|
|
|
}
|
|
|
|
/**
|
|
* If no listeners are registered for incoming SOCKS5 Bytestream requests, all request should be
|
|
* rejected with an error.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldRespondWithError() throws Exception {
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// capture reply to the SOCKS5 Bytestream initiation
|
|
ArgumentCaptor<IQ> argument = ArgumentCaptor.forClass(IQ.class);
|
|
verify(connection).sendPacket(argument.capture());
|
|
|
|
// assert that reply is the correct error packet
|
|
assertEquals(initiatorJID, argument.getValue().getTo());
|
|
assertEquals(IQ.Type.error, argument.getValue().getType());
|
|
assertEquals(XMPPError.Condition.not_acceptable,
|
|
argument.getValue().getError().getCondition());
|
|
|
|
}
|
|
|
|
/**
|
|
* If a listener for all requests is registered it should be notified on incoming requests.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldInvokeListenerForAllRequests() throws Exception {
|
|
|
|
// add listener
|
|
Socks5BytestreamListener listener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(listener);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert listener is called once
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(listener).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert that listener is called for the correct request
|
|
assertEquals(initiatorJID, byteStreamRequest.getValue().getFrom());
|
|
|
|
}
|
|
|
|
/**
|
|
* If a listener for a specific user in registered it should be notified on incoming requests
|
|
* for that user.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldInvokeListenerForUser() throws Exception {
|
|
|
|
// add listener
|
|
Socks5BytestreamListener listener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(listener, initiatorJID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert listener is called once
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(listener).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert that reply is the correct error packet
|
|
assertEquals(initiatorJID, byteStreamRequest.getValue().getFrom());
|
|
|
|
}
|
|
|
|
/**
|
|
* If listener for a specific user is registered it should not be notified on incoming requests
|
|
* from other users.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldNotInvokeListenerForUser() throws Exception {
|
|
|
|
// add listener for request of user "other_initiator"
|
|
Socks5BytestreamListener listener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(listener, "other_" + initiatorJID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert listener is not called
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(listener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// capture reply to the SOCKS5 Bytestream initiation
|
|
ArgumentCaptor<IQ> argument = ArgumentCaptor.forClass(IQ.class);
|
|
verify(connection).sendPacket(argument.capture());
|
|
|
|
// assert that reply is the correct error packet
|
|
assertEquals(initiatorJID, argument.getValue().getTo());
|
|
assertEquals(IQ.Type.error, argument.getValue().getType());
|
|
assertEquals(XMPPError.Condition.not_acceptable,
|
|
argument.getValue().getError().getCondition());
|
|
}
|
|
|
|
/**
|
|
* If a user specific listener and an all requests listener is registered only the user specific
|
|
* listener should be notified.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldNotInvokeAllRequestsListenerIfUserListenerExists() throws Exception {
|
|
|
|
// add listener for all request
|
|
Socks5BytestreamListener allRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(allRequestsListener);
|
|
|
|
// add listener for request of user "initiator"
|
|
Socks5BytestreamListener userRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(userRequestsListener, initiatorJID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert user request listener is called once
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(userRequestsListener).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert all requests listener is not called
|
|
byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(allRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
}
|
|
|
|
/**
|
|
* If a user specific listener and an all requests listener is registered only the all requests
|
|
* listener should be notified on an incoming request for another user.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldInvokeAllRequestsListenerIfUserListenerExists() throws Exception {
|
|
|
|
// add listener for all request
|
|
Socks5BytestreamListener allRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(allRequestsListener);
|
|
|
|
// add listener for request of user "other_initiator"
|
|
Socks5BytestreamListener userRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(userRequestsListener, "other_"
|
|
+ initiatorJID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert user request listener is not called
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(userRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert all requests listener is called
|
|
byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(allRequestsListener).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
}
|
|
|
|
/**
|
|
* If a request with a specific session ID should be ignored no listeners should be notified.
|
|
*
|
|
* @throws Exception should not happen
|
|
*/
|
|
@Test
|
|
public void shouldIgnoreSocks5BytestreamRequestOnce() throws Exception {
|
|
|
|
// add listener for all request
|
|
Socks5BytestreamListener allRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(allRequestsListener);
|
|
|
|
// add listener for request of user "initiator"
|
|
Socks5BytestreamListener userRequestsListener = mock(Socks5BytestreamListener.class);
|
|
byteStreamManager.addIncomingBytestreamListener(userRequestsListener, initiatorJID);
|
|
|
|
// ignore session ID
|
|
byteStreamManager.ignoreBytestreamRequestOnce(sessionID);
|
|
|
|
// run the listener with the initiation packet
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert user request listener is not called
|
|
ArgumentCaptor<BytestreamRequest> byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(userRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert all requests listener is not called
|
|
byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(allRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// run the listener with the initiation packet again
|
|
initiationListener.handleIQRequest(initBytestream);
|
|
|
|
// wait because packet is processed in an extra thread
|
|
Thread.sleep(200);
|
|
|
|
// assert user request listener is called on the second request with the same session ID
|
|
verify(userRequestsListener).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
// assert all requests listener is not called
|
|
byteStreamRequest = ArgumentCaptor.forClass(BytestreamRequest.class);
|
|
verify(allRequestsListener, never()).incomingBytestreamRequest(byteStreamRequest.capture());
|
|
|
|
}
|
|
|
|
}
|