mirror of https://github.com/apache/activemq.git
The MQTT transport will now throw an exception if a PINGREQ is sent to the broker if a CONNECT packet has not been received first as the spec says CONNECT must be the first packet sent.
This commit is contained in:
parent
c02bc64846
commit
6dacef1c95
|
@ -196,6 +196,7 @@ public class MQTTProtocolConverter {
|
||||||
switch (frame.messageType()) {
|
switch (frame.messageType()) {
|
||||||
case PINGREQ.TYPE:
|
case PINGREQ.TYPE:
|
||||||
LOG.debug("Received a ping from client: " + getClientId());
|
LOG.debug("Received a ping from client: " + getClientId());
|
||||||
|
checkConnected();
|
||||||
sendToMQTT(PING_RESP_FRAME);
|
sendToMQTT(PING_RESP_FRAME);
|
||||||
LOG.debug("Sent Ping Response to " + getClientId());
|
LOG.debug("Sent Ping Response to " + getClientId());
|
||||||
break;
|
break;
|
||||||
|
|
|
@ -25,6 +25,8 @@ import static org.junit.Assert.assertNull;
|
||||||
import static org.junit.Assert.assertTrue;
|
import static org.junit.Assert.assertTrue;
|
||||||
import static org.junit.Assert.fail;
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
|
import java.io.EOFException;
|
||||||
|
import java.lang.reflect.Method;
|
||||||
import java.net.ProtocolException;
|
import java.net.ProtocolException;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
|
@ -56,13 +58,21 @@ import org.apache.activemq.command.ActiveMQMessage;
|
||||||
import org.apache.activemq.command.ActiveMQTopic;
|
import org.apache.activemq.command.ActiveMQTopic;
|
||||||
import org.apache.activemq.util.ByteSequence;
|
import org.apache.activemq.util.ByteSequence;
|
||||||
import org.apache.activemq.util.Wait;
|
import org.apache.activemq.util.Wait;
|
||||||
|
import org.apache.activemq.util.Wait.Condition;
|
||||||
|
import org.fusesource.hawtdispatch.transport.Transport;
|
||||||
import org.fusesource.mqtt.client.BlockingConnection;
|
import org.fusesource.mqtt.client.BlockingConnection;
|
||||||
|
import org.fusesource.mqtt.client.Callback;
|
||||||
|
import org.fusesource.mqtt.client.CallbackConnection;
|
||||||
|
import org.fusesource.mqtt.client.FutureConnection;
|
||||||
import org.fusesource.mqtt.client.MQTT;
|
import org.fusesource.mqtt.client.MQTT;
|
||||||
import org.fusesource.mqtt.client.Message;
|
import org.fusesource.mqtt.client.Message;
|
||||||
import org.fusesource.mqtt.client.QoS;
|
import org.fusesource.mqtt.client.QoS;
|
||||||
import org.fusesource.mqtt.client.Topic;
|
import org.fusesource.mqtt.client.Topic;
|
||||||
import org.fusesource.mqtt.client.Tracer;
|
import org.fusesource.mqtt.client.Tracer;
|
||||||
import org.fusesource.mqtt.codec.MQTTFrame;
|
import org.fusesource.mqtt.codec.MQTTFrame;
|
||||||
|
import org.fusesource.mqtt.codec.MQTTProtocolCodec;
|
||||||
|
import org.fusesource.mqtt.codec.PINGREQ;
|
||||||
|
import org.fusesource.mqtt.codec.PINGRESP;
|
||||||
import org.fusesource.mqtt.codec.PUBLISH;
|
import org.fusesource.mqtt.codec.PUBLISH;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.slf4j.Logger;
|
import org.slf4j.Logger;
|
||||||
|
@ -1961,4 +1971,116 @@ public class MQTTTest extends MQTTTestSupport {
|
||||||
|
|
||||||
connection.disconnect();
|
connection.disconnect();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test(timeout = 15 * 1000, expected=EOFException.class)
|
||||||
|
public void testPingReqWithoutConnectFail31() throws Exception {
|
||||||
|
MQTT mqtt = createMQTTConnection();
|
||||||
|
mqtt.setClientId("clientId");
|
||||||
|
mqtt.setVersion("3.1");
|
||||||
|
testPingReqWithoutConnectFail(mqtt);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test(timeout = 15 * 1000, expected=EOFException.class)
|
||||||
|
public void testPingReqWithoutConnectFail311() throws Exception {
|
||||||
|
MQTT mqtt = createMQTTConnection();
|
||||||
|
mqtt.setClientId("clientId");
|
||||||
|
mqtt.setVersion("3.1.1");
|
||||||
|
testPingReqWithoutConnectFail(mqtt);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test(timeout = 15 * 1000)
|
||||||
|
public void testPingReqConnectSuccess31() throws Exception {
|
||||||
|
MQTT mqtt = createMQTTConnection();
|
||||||
|
mqtt.setClientId("clientId");
|
||||||
|
mqtt.setVersion("3.1");
|
||||||
|
testPingReqConnectSuccess(mqtt);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test(timeout = 15 * 1000)
|
||||||
|
public void testPingReqConnectSuccess311() throws Exception {
|
||||||
|
MQTT mqtt = createMQTTConnection();
|
||||||
|
mqtt.setClientId("clientId");
|
||||||
|
mqtt.setVersion("3.1.1");
|
||||||
|
testPingReqConnectSuccess(mqtt);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void testPingReqWithoutConnectFail(final MQTT mqtt) throws Exception {
|
||||||
|
final CountDownLatch latch = new CountDownLatch(1);
|
||||||
|
final AtomicReference<Transport> transport = new AtomicReference<>();
|
||||||
|
final AtomicReference<Throwable> error = new AtomicReference<>();
|
||||||
|
final Callback<Transport> con = new Callback<Transport>() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onSuccess(Transport value) {
|
||||||
|
transport.set(value);
|
||||||
|
latch.countDown();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onFailure(Throwable value) {
|
||||||
|
error.set(value);
|
||||||
|
latch.countDown();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
//Connect to the transport by using the createTransport method with a custom callback
|
||||||
|
//This will ensure that we connect without sending a CONNECT packet for testing
|
||||||
|
//and that we won't receive automatically
|
||||||
|
CallbackConnection connection = new CallbackConnection(mqtt);
|
||||||
|
Method createTransportMethod = connection.getClass().getDeclaredMethod("createTransport", Callback.class);
|
||||||
|
createTransportMethod.setAccessible(true);
|
||||||
|
createTransportMethod.invoke(connection, con);
|
||||||
|
latch.await();
|
||||||
|
|
||||||
|
//Make sure no error on connect
|
||||||
|
if (error.get() != null) {
|
||||||
|
LOG.error(error.get().getMessage(), error.get());
|
||||||
|
fail(error.get().getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
//Send a PINGREQ without a connect packet first
|
||||||
|
final MQTTProtocolCodec codec = new MQTTProtocolCodec();
|
||||||
|
codec.setTransport(transport.get());
|
||||||
|
transport.get().offer(new PINGREQ().encode());
|
||||||
|
|
||||||
|
//Protocol should throw an exception since we never sent a CONNECT
|
||||||
|
Wait.waitFor(new Condition() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isSatisified() throws Exception {
|
||||||
|
//Wait for exception to be thrown
|
||||||
|
codec.read();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}, 5000, 100);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private void testPingReqConnectSuccess(final MQTT mqtt) throws Exception {
|
||||||
|
final CountDownLatch pingRespReceived = new CountDownLatch(1);
|
||||||
|
//Tracer to assert we received the response by waiting for it
|
||||||
|
mqtt.setTracer(new Tracer() {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void onReceive(MQTTFrame frame) {
|
||||||
|
if (frame.messageType() == PINGRESP.TYPE) {
|
||||||
|
pingRespReceived.countDown();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
});
|
||||||
|
CallbackConnection callbackConnection = new CallbackConnection(mqtt);
|
||||||
|
BlockingConnection connection = new BlockingConnection(new FutureConnection(callbackConnection));
|
||||||
|
connection.connect();
|
||||||
|
Transport transport = callbackConnection.transport();
|
||||||
|
|
||||||
|
//SEND a PINGREQ and wait for the response
|
||||||
|
final MQTTProtocolCodec codec = new MQTTProtocolCodec();
|
||||||
|
codec.setTransport(transport);
|
||||||
|
transport.offer(new PINGREQ().encode());
|
||||||
|
|
||||||
|
//Wait for the response
|
||||||
|
assertTrue(pingRespReceived.await(5, TimeUnit.SECONDS));
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue