mirror of https://github.com/apache/activemq.git
add maxFrameSize to the transport and enforce across the TCP, SSL, NIO and NIO+SSL transport connectors.
This commit is contained in:
parent
7c41ebc912
commit
3454a8b596
|
@ -26,6 +26,7 @@ import org.fusesource.mqtt.codec.MQTTFrame;
|
|||
public class MQTTCodec {
|
||||
|
||||
private final MQTTFrameSink frameSink;
|
||||
private final MQTTWireFormat wireFormat;
|
||||
|
||||
private byte header;
|
||||
private int contentLength = -1;
|
||||
|
@ -43,10 +44,20 @@ public class MQTTCodec {
|
|||
}
|
||||
|
||||
public MQTTCodec(MQTTFrameSink sink) {
|
||||
this(sink, null);
|
||||
}
|
||||
|
||||
public MQTTCodec(MQTTFrameSink sink, MQTTWireFormat wireFormat) {
|
||||
this.frameSink = sink;
|
||||
this.wireFormat = wireFormat;
|
||||
}
|
||||
|
||||
public MQTTCodec(final TcpTransport transport) {
|
||||
this(transport, null);
|
||||
}
|
||||
|
||||
public MQTTCodec(final TcpTransport transport, MQTTWireFormat wireFormat) {
|
||||
this.wireFormat = wireFormat;
|
||||
this.frameSink = new MQTTFrameSink() {
|
||||
|
||||
@Override
|
||||
|
@ -79,6 +90,10 @@ public class MQTTCodec {
|
|||
frameSink.onFrame(frame);
|
||||
}
|
||||
|
||||
private int getMaxFrameSize() {
|
||||
return wireFormat != null ? wireFormat.getMaxFrameSize() : MQTTWireFormat.MAX_MESSAGE_LENGTH;
|
||||
}
|
||||
|
||||
//----- Prepare the current frame parser for use -------------------------//
|
||||
|
||||
private FrameParser initializeHeaderParser() throws IOException {
|
||||
|
@ -151,6 +166,10 @@ public class MQTTCodec {
|
|||
processCommand();
|
||||
currentParser = initializeHeaderParser();
|
||||
} else {
|
||||
if (length > getMaxFrameSize()) {
|
||||
throw new IOException("The maximum message length was exceeded");
|
||||
}
|
||||
|
||||
currentParser = initializeContentParser();
|
||||
contentLength = length;
|
||||
}
|
||||
|
|
|
@ -42,7 +42,7 @@ public class MQTTNIOSSLTransport extends NIOSSLTransport {
|
|||
|
||||
@Override
|
||||
protected void initializeStreams() throws IOException {
|
||||
codec = new MQTTCodec(this);
|
||||
codec = new MQTTCodec(this, (MQTTWireFormat) getWireFormat());
|
||||
super.initializeStreams();
|
||||
if (inputBuffer.position() != 0 && inputBuffer.hasRemaining()) {
|
||||
serviceRead();
|
||||
|
|
|
@ -56,17 +56,20 @@ public class MQTTNIOTransport extends TcpTransport {
|
|||
super(wireFormat, socket);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void initializeStreams() throws IOException {
|
||||
channel = socket.getChannel();
|
||||
channel.configureBlocking(false);
|
||||
// listen for events telling us when the socket is readable.
|
||||
selection = SelectorManager.getInstance().register(channel, new SelectorManager.Listener() {
|
||||
@Override
|
||||
public void onSelect(SelectorSelection selection) {
|
||||
if (!isStopped()) {
|
||||
serviceRead();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onError(SelectorSelection selection, Throwable error) {
|
||||
if (error instanceof IOException) {
|
||||
onException((IOException) error);
|
||||
|
@ -78,9 +81,9 @@ public class MQTTNIOTransport extends TcpTransport {
|
|||
|
||||
inputBuffer = ByteBuffer.allocate(8 * 1024);
|
||||
NIOOutputStream outPutStream = new NIOOutputStream(channel, 8 * 1024);
|
||||
this.dataOut = new DataOutputStream(outPutStream);
|
||||
this.buffOut = outPutStream;
|
||||
codec = new MQTTCodec(this);
|
||||
dataOut = new DataOutputStream(outPutStream);
|
||||
buffOut = outPutStream;
|
||||
codec = new MQTTCodec(this, (MQTTWireFormat) getWireFormat());
|
||||
}
|
||||
|
||||
private void serviceRead() {
|
||||
|
@ -116,12 +119,14 @@ public class MQTTNIOTransport extends TcpTransport {
|
|||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void doStart() throws Exception {
|
||||
connect();
|
||||
selection.setInterestOps(SelectionKey.OP_READ);
|
||||
selection.enable();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void doStop(ServiceStopper stopper) throws Exception {
|
||||
try {
|
||||
if (selection != null) {
|
||||
|
|
|
@ -258,4 +258,22 @@ public class MQTTTransportFilter extends TransportFilter implements MQTTTranspor
|
|||
public void setActiveMQSubscriptionPrefetch(int activeMQSubscriptionPrefetch) {
|
||||
protocolConverter.setActiveMQSubscriptionPrefetch(activeMQSubscriptionPrefetch);
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the maximum number of bytes a single MQTT message frame is allowed to be.
|
||||
*/
|
||||
public int getMaxFrameSize() {
|
||||
return wireFormat.getMaxFrameSize();
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets the maximum frame size for an incoming MQTT frame. The protocl limit is
|
||||
* 256 megabytes and this value cannot be set higher.
|
||||
*
|
||||
* @param maxFrameSize
|
||||
* the maximum allowed frame size for a single MQTT frame.
|
||||
*/
|
||||
public void setMaxFrameSize(int maxFrameSize) {
|
||||
wireFormat.setMaxFrameSize(maxFrameSize);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -40,6 +40,8 @@ public class MQTTWireFormat implements WireFormat {
|
|||
|
||||
private int version = 1;
|
||||
|
||||
private int maxFrameSize = MAX_MESSAGE_LENGTH;
|
||||
|
||||
@Override
|
||||
public ByteSequence marshal(Object command) throws IOException {
|
||||
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
||||
|
@ -93,7 +95,7 @@ public class MQTTWireFormat implements WireFormat {
|
|||
while ((digit & 0x80) != 0);
|
||||
|
||||
if (length >= 0) {
|
||||
if (length > MAX_MESSAGE_LENGTH) {
|
||||
if (length > getMaxFrameSize()) {
|
||||
throw new IOException("The maximum message length was exceeded");
|
||||
}
|
||||
|
||||
|
@ -124,4 +126,22 @@ public class MQTTWireFormat implements WireFormat {
|
|||
public int getVersion() {
|
||||
return this.version;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the maximum number of bytes a single MQTT message frame is allowed to be.
|
||||
*/
|
||||
public int getMaxFrameSize() {
|
||||
return maxFrameSize;
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets the maximum frame size for an incoming MQTT frame. The protocl limit is
|
||||
* 256 megabytes and this value cannot be set higher.
|
||||
*
|
||||
* @param maxFrameSize
|
||||
* the maximum allowed frame size for a single MQTT frame.
|
||||
*/
|
||||
public void setMaxFrameSize(int maxFrameSize) {
|
||||
this.maxFrameSize = Math.min(MAX_MESSAGE_LENGTH, maxFrameSize);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -0,0 +1,122 @@
|
|||
/**
|
||||
* 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.activemq.transport.mqtt;
|
||||
|
||||
import static org.junit.Assert.fail;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.Collection;
|
||||
|
||||
import org.fusesource.mqtt.client.BlockingConnection;
|
||||
import org.fusesource.mqtt.client.MQTT;
|
||||
import org.fusesource.mqtt.client.QoS;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.junit.runners.Parameterized;
|
||||
import org.junit.runners.Parameterized.Parameters;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
/**
|
||||
* Test that the maxFrameSize configuration value is applied across the transports.
|
||||
*/
|
||||
@RunWith(Parameterized.class)
|
||||
public class MQTTMaxFrameSizeTest extends MQTTTestSupport {
|
||||
|
||||
private static final Logger LOG = LoggerFactory.getLogger(MQTTMaxFrameSizeTest.class);
|
||||
|
||||
private final int maxFrameSize;
|
||||
|
||||
@Parameters(name="{0}")
|
||||
public static Collection<Object[]> data() {
|
||||
return Arrays.asList(new Object[][] {
|
||||
{ "mqtt", false, 1024 },
|
||||
{ "mqtt+ssl", true, 1024 },
|
||||
{ "mqtt+nio", false, 1024 },
|
||||
{ "mqtt+nio+ssl", true, 1024 }
|
||||
});
|
||||
}
|
||||
|
||||
public MQTTMaxFrameSizeTest(String connectorScheme, boolean useSSL, int maxFrameSize) {
|
||||
super(connectorScheme, useSSL);
|
||||
|
||||
this.maxFrameSize = maxFrameSize;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getProtocolConfig() {
|
||||
return "?transport.maxFrameSize=" + maxFrameSize;
|
||||
}
|
||||
|
||||
@Test(timeout = 30000)
|
||||
public void testFrameSizeToLargeClosesConnection() throws Exception {
|
||||
|
||||
LOG.debug("Starting test on connector {} for frame size: {}", getProtocolScheme(), maxFrameSize);
|
||||
|
||||
MQTT mqtt = createMQTTConnection();
|
||||
mqtt.setClientId(getName());
|
||||
mqtt.setKeepAlive((short) 10);
|
||||
mqtt.setVersion("3.1.1");
|
||||
|
||||
BlockingConnection connection = mqtt.blockingConnection();
|
||||
connection.connect();
|
||||
|
||||
final int payloadSize = maxFrameSize + 100;
|
||||
|
||||
byte[] payload = new byte[payloadSize];
|
||||
for (int i = 0; i < payloadSize; ++i) {
|
||||
payload[i] = 42;
|
||||
}
|
||||
|
||||
try {
|
||||
connection.publish(getTopicName(), payload, QoS.AT_LEAST_ONCE, false);
|
||||
fail("should have thrown an exception");
|
||||
} catch (Exception ex) {
|
||||
} finally {
|
||||
connection.disconnect();
|
||||
}
|
||||
}
|
||||
|
||||
@Test(timeout = 30000)
|
||||
public void testFrameSizeNotExceededWorks() throws Exception {
|
||||
|
||||
LOG.debug("Starting test on connector {} for frame size: {}", getProtocolScheme(), maxFrameSize);
|
||||
|
||||
MQTT mqtt = createMQTTConnection();
|
||||
mqtt.setClientId(getName());
|
||||
mqtt.setKeepAlive((short) 10);
|
||||
mqtt.setVersion("3.1.1");
|
||||
|
||||
BlockingConnection connection = mqtt.blockingConnection();
|
||||
connection.connect();
|
||||
|
||||
final int payloadSize = maxFrameSize / 2;
|
||||
|
||||
byte[] payload = new byte[payloadSize];
|
||||
for (int i = 0; i < payloadSize; ++i) {
|
||||
payload[i] = 42;
|
||||
}
|
||||
|
||||
try {
|
||||
connection.publish(getTopicName(), payload, QoS.AT_LEAST_ONCE, false);
|
||||
} catch (Exception ex) {
|
||||
fail("should not have thrown an exception");
|
||||
} finally {
|
||||
connection.disconnect();
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue