io.fabric8.mq.protocol.stomp.StompVertxCodec Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fabric8mq Show documentation
Fabric8 MQ Broker
There is a newer version: 2.2.96
/**
 * 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 io.fabric8.mq.protocol.stomp;

import io.fabric8.mq.protocol.ProtocolException;
import org.apache.activemq.transport.stomp.Stomp;
import org.apache.activemq.transport.stomp.StompFrame;
import org.apache.activemq.util.ByteArrayOutputStream;
import org.fusesource.hawtbuf.DataByteArrayInputStream;

import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;

public class StompVertxCodec {

    private final static byte[] crlfcrlf = new byte[]{'\r', '\n', '\r', '\n'};
    private final StompWriteStream writeStream;
    private final StompWireFormat wireFormat;
    private ByteArrayOutputStream currentCommand = new ByteArrayOutputStream();
    private boolean processedHeaders = false;
    private String action;
    private HashMap headers;
    private int contentLength = -1;
    private int readLength = 0;
    private int previousByte = -1;
    private boolean awaitingCommandStart = true;
    private String version = Stomp.DEFAULT_VERSION;

    public StompVertxCodec(StompWriteStream writeStream, StompWireFormat wireFormat) {
        this.writeStream = writeStream;
        this.wireFormat = wireFormat;
    }

    public static String detectVersion(Map headers) throws ProtocolException {
        String accepts = headers.get(Stomp.Headers.Connect.ACCEPT_VERSION);

        if (accepts == null) {
            accepts = Stomp.DEFAULT_VERSION;
        }
        HashSet acceptsVersions = new HashSet<>(Arrays.asList(accepts.trim().split(Stomp.COMMA)));
        acceptsVersions.retainAll(Arrays.asList(Stomp.SUPPORTED_PROTOCOL_VERSIONS));
        if (acceptsVersions.isEmpty()) {
            throw new ProtocolException("Invalid Protocol version[" + accepts + "], supported versions are: " +
                                            Arrays.toString(Stomp.SUPPORTED_PROTOCOL_VERSIONS), true);
        } else {
            return Collections.max(acceptsVersions);
        }
    }

    public void parse(DataByteArrayInputStream input, int readSize) throws Exception {
        int i = 0;
        int b;
        while (i++ < readSize) {
            b = input.read();
            // skip repeating nulls
            if (!processedHeaders && previousByte == 0 && b == 0) {
                continue;
            }

            if (!processedHeaders) {

                // skip heart beat commands.
                if (awaitingCommandStart && b == '\n') {
                    continue;
                } else {
                    awaitingCommandStart = false;   // non-newline indicates next frame.
                }

                currentCommand.write(b);
                // end of headers section, parse action and header
                if (b == '\n' && (previousByte == '\n' || currentCommand.endsWith(crlfcrlf))) {
                    DataByteArrayInputStream data = new DataByteArrayInputStream(currentCommand.toByteArray());
                    action = wireFormat.parseAction(data);
                    headers = wireFormat.parseHeaders(data);

                    String contentLengthHeader = headers.get(Stomp.Headers.CONTENT_LENGTH);
                    if ((action.equals(Stomp.Commands.SEND) || action.equals(Stomp.Responses.MESSAGE)) && contentLengthHeader != null) {
                        contentLength = wireFormat.parseContentLength(contentLengthHeader);
                    } else {
                        contentLength = -1;
                    }

                    processedHeaders = true;
                    currentCommand.reset();
                }

            } else {

                if (contentLength == -1) {
                    // end of command reached, unmarshal
                    if (b == 0) {
                        processCommand();
                    } else {
                        currentCommand.write(b);
                    }
                } else {
                    // read desired content length
                    if (readLength++ == contentLength) {
                        processCommand();
                        readLength = 0;
                    } else {
                        currentCommand.write(b);
                    }
                }
            }

            previousByte = b;
        }
    }

    protected void processCommand() throws Exception {
        StompFrame frame = new StompFrame(action, headers, currentCommand.toByteArray());
        writeStream.consume(frame);
        processedHeaders = false;
        awaitingCommandStart = true;
        currentCommand.reset();
        contentLength = -1;
    }
}