All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.red5.server.net.rtmp.message.Packet Maven / Gradle / Ivy

There is a newer version: 2.3.3.1
Show newest version
/*
 * RED5 Open Source Media Server - https://github.com/Red5/
 * 
 * Copyright 2006-2016 by respective authors (see below). All rights reserved.
 * 
 * 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.red5.server.net.rtmp.message;

import java.io.Externalizable;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.util.concurrent.atomic.AtomicBoolean;

import org.apache.mina.core.buffer.IoBuffer;
import org.red5.server.net.rtmp.event.IRTMPEvent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * RTMP packet. Consists of packet header, data and event context.
 */
public class Packet implements Externalizable {

    private static final long serialVersionUID = -6415050845346626950L;

    private static Logger log = LoggerFactory.getLogger(Packet.class);

    private static final boolean noCopy = System.getProperty("packet.noCopy") == null ? false : Boolean.valueOf(System.getProperty("packet.noCopy"));

    /**
     * Header
     */
    private Header header;

    /**
     * RTMP event
     */
    private IRTMPEvent message;

    /**
     * Packet data
     */
    private IoBuffer data;

    /**
     * Expiration time
     */
    private transient long expirationTime = 0L;

    /**
     * Flag representing processed status
     */
    private transient final AtomicBoolean processed = new AtomicBoolean(false);

    public Packet() {
        log.trace("ctor");
    }

    /**
     * Create packet with given header.
     * 
     * @param header
     *            Packet header
     */
    public Packet(Header header) {
        if (log.isTraceEnabled()) {
            log.trace("Header: {}", header);
        }
        this.header = header;
        data = IoBuffer.allocate(header.getSize()).setAutoExpand(true);
    }

    /**
     * Create packet with given header and event context.
     * 
     * @param header
     *            RTMP header
     * @param event
     *            RTMP message
     */
    public Packet(Header header, IRTMPEvent event) {
        if (log.isTraceEnabled()) {
            log.trace("Header: {} event: {}", header, event);
        }
        this.header = header;
        this.message = event;
    }

    /**
     * Getter for header
     *
     * @return Packet header
     */
    public Header getHeader() {
        return header;
    }

    /**
     * Setter for event context
     *
     * @param message
     *            RTMP event context
     */
    public void setMessage(IRTMPEvent message) {
        this.message = message;
    }

    /**
     * Getter for event context
     *
     * @return RTMP event context
     */
    public IRTMPEvent getMessage() {
        return message;
    }

    /**
     * Setter for data
     *
     * @param buffer
     *            Packet data
     */
    public void setData(IoBuffer buffer) {
        if (noCopy) {
            log.trace("Using buffer reference");
            this.data = buffer;
        } else {
            // try the backing array first if it exists
            if (buffer.hasArray()) {
                log.trace("Buffer has backing array, making a copy");
                byte[] copy = new byte[buffer.limit()];
                buffer.mark();
                buffer.get(copy);
                buffer.reset();
                data = IoBuffer.wrap(copy);
            } else {
                log.trace("Buffer has no backing array, using ByteBuffer");
                // fallback to ByteBuffer
                data.put(buffer.buf()).flip();
            }
        }
    }

    /**
     * Getter for data
     *
     * @return Packet data
     */
    public IoBuffer getData() {
        return data;
    }

    /**
     * Returns whether or not the packet has a data buffer.
     * 
     * @return true if data buffer exists and false otherwise
     */
    public boolean hasData() {
        return data != null;
    }

    /**
     * Clears the data buffer.
     */
    public void clearData() {
        if (data != null) {
            data.clear();
            data.free();
            data = null;
        }
    }

    /**
     * Return the expiration time.
     * 
     * @return expirationTime
     */
    public long getExpirationTime() {
        return expirationTime;
    }

    /**
     * Set the expiration time.
     * 
     * @param expirationTime new expiration time to set
     */
    public void setExpirationTime(long expirationTime) {
        this.expirationTime = expirationTime;
    }

    /**
     * Returns true if expiration time has been reached and false otherwise.
     * 
     * @return expired or not
     */
    public boolean isExpired() {
        // if expirationTime is zero, the expiration is not used
        return expirationTime > 0L ? System.currentTimeMillis() > expirationTime : false;
    }

    /**
     * Sets the processed flag.
     * 
     * @param isProcessed
     *            true if processed and false otherwise
     */
    public void setProcessed(boolean isProcessed) {
        this.processed.set(isProcessed);
    }

    /**
     * Gets the processed flag.
     * 
     * @return true if processed and false otherwise
     */
    public boolean isProcessed() {
        return this.processed.get();
    }

    public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
        header = (Header) in.readObject();
        message = (IRTMPEvent) in.readObject();
        message.setHeader(header);
        message.setTimestamp(header.getTimer());
    }

    public void writeExternal(ObjectOutput out) throws IOException {
        out.writeObject(header);
        out.writeObject(message);
    }

    /* (non-Javadoc)
     * @see java.lang.Object#toString()
     */
    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder("Packet [");
        if (header != null) {
            sb.append("[header data type=" + header.getDataType() + ", channel=" + header.getChannelId() + ", timer=" + header.getTimer() + "]");
        } else {
            sb.append("[header=null]");
        }
        if (message != null) {
            sb.append(", [message timestamp=" + message.getTimestamp() + "]");
        } else {
            sb.append(", [message=null]");
        }
        sb.append("]");
        return sb.toString();
    }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy