org.red5.server.net.rtmp.RTMPMinaConnection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of android-rmtp-client Show documentation
A standalone RTMP client library ported from the Red5 project
The newest version!
package org.red5.server.net.rtmp;

/*
 * RED5 Open Source Flash Server - http://code.google.com/p/red5/
 *
 * Copyright (c) 2006-2010 by respective authors (see below). All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under the
 * terms of the GNU Lesser General Public License as published by the Free Software
 * Foundation; either version 2.1 of the License, or (at your option) any later
 * version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License along
 * with this library; if not, write to the Free Software Foundation, Inc.,
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

//import java.beans.ConstructorProperties;
import java.net.InetSocketAddress;
import java.net.SocketAddress;
import java.util.ArrayList;
import java.util.Collections;

//import javax.management.ObjectName;

import org.apache.mina.core.buffer.IoBuffer;
import org.apache.mina.core.filterchain.IoFilterChain;
import org.apache.mina.core.session.IoSession;
//import org.red5.server.api.IContext;
//import org.red5.server.api.IScope;
//import org.red5.server.net.filter.TrafficShapingFilter;
import org.red5.server.net.ProtocolState;
import org.red5.server.net.rtmp.codec.RTMP;
//import org.red5.server.net.rtmp.event.ClientBW;
//import org.red5.server.net.rtmp.event.ServerBW;
import org.red5.server.net.rtmp.message.Packet;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Represents an RTMP connection using Mina.
 * 
 * @see "http://mina.apache.org/report/trunk/apidocs/org/apache/mina/core/session/IoSession.html"
 * 
 * @author Paul Gregoire
 */
public class RTMPMinaConnection extends RTMPConnection {
	
	protected static Logger log = LoggerFactory.getLogger(RTMPMinaConnection.class);

	/**
	 * MINA I/O session, connection between two end points
	 */
	private volatile IoSession ioSession;
	
	/**
	 * MBean object name used for de/registration purposes.
	 */
	//private volatile ObjectName oName;

	{
		log.debug("RTMPMinaConnection created");
	}
	
	/** Constructs a new RTMPMinaConnection. */
	//@ConstructorProperties(value={"persistent"})
	public RTMPMinaConnection() {
		super(PERSISTENT);
	}

	/** {@inheritDoc} */
	@Override
	public void close() {
		super.close();
		if (ioSession != null) {
			IoFilterChain filters =	ioSession.getFilterChain();
			//check if it exists and remove
			if (filters.contains("bandwidthFilter")) {
        		ioSession.getFilterChain().remove("bandwidthFilter");
			}
			// update our state
			if (ioSession.containsAttribute(ProtocolState.SESSION_KEY)) {
				RTMP rtmp = (RTMP) ioSession.getAttribute(ProtocolState.SESSION_KEY);
				log.debug("RTMP state: {}", rtmp);
				rtmp.setState(RTMP.STATE_DISCONNECTING);
			}
			// accept no further incoming data
			ioSession.suspendRead();
			// close now, no flushing, no waiting
			ioSession.close(true);
			// only close socket after all pending data has been sent, this does not
			// work as expected when using RTMPE
			//CloseFuture future = ioSession.close(false);
			// wait until the connection is closed
			//future.awaitUninterruptibly();
			// now connection should be closed.
			//if (!future.isClosed()) {
				// force the close
			//	ioSession.close(true);				
			//}
		}
		//de-register with JMX
//		try {
//		    JMXAgent.unregisterMBean(oName);
//		} catch (Exception e) {
//		    //sometimes the client is not registered in jmx
//		}
	}

	@SuppressWarnings("cast")
	@Override
	public boolean connect( Object[] params) {
//		log.debug("Connect scope: {}", newScope);
		boolean success = super.connect( params);
		if (!success) {
			return false;
		}
		@SuppressWarnings("unused")
		String hostStr = host;
		@SuppressWarnings("unused")
		int port = 1935;
		if (host != null && host.indexOf(":") > -1) {
			String[] arr = host.split(":");
			hostStr = arr[0];
			port = Integer.parseInt(arr[1]);
		}
		//if the client is null for some reason, skip the jmx registration
		if (client != null) {
    		try {
				String cName = this.getClass().getName();
				if (cName.indexOf('.') != -1) {
					cName = cName.substring(cName.lastIndexOf('.')).replaceFirst("[\\.]", "");
				}				
			    // Create a new mbean for this instance
//			    oName = JMXFactory.createObjectName("type", cName, "connectionType", type, "host", hostStr, "port", port + "", "clientId", client.getId());
//			    JMXAgent.registerMBean(this, this.getClass().getName(),	RTMPMinaConnectionMXBean.class, oName);		
    		} catch (Exception e) {
    			log.warn("Exception registering mbean", e);
    		}
		} else {
            log.warn("Client was null");			
		}
		//add bandwidth filter
		if (ioSession != null) {
			log.debug("Top level scope detected, configuration will be applied if it exists");
//			IContext ctx = scope.getContext();
//			if (ctx != null) {
//				log.debug("Context was found");	
//				IoFilterChain filters =	ioSession.getFilterChain();
//				//add it if it does not exist
//				if (!filters.contains("bandwidthFilter")) {
//					//look for the bean first
//					if (ctx.hasBean("bandwidthFilter")) {
//						TrafficShapingFilter filter = (TrafficShapingFilter) ctx.getBean("bandwidthFilter");
//                		//load the each after the last
//                		ioSession.getFilterChain().addAfter("protocolFilter", "bandwidthFilter", filter);                		
//                		//notify client about new bandwidth settings (in bytes per second)
//                		int downStream = filter.getMaxReadThroughput();
//                		int upStream = filter.getMaxWriteThroughput();
//                		if (downStream > 0) {
//                			ServerBW serverBW = new ServerBW((int) downStream / 8);
//                			getChannel(2).write(serverBW);
//                		}
//                		if (upStream > 0) {
//                			ClientBW clientBW = new ClientBW((int) upStream / 8, (byte) 0);
//                			getChannel(2).write(clientBW);
//            				// Update generation of BytesRead messages
//            				// TODO: what are the correct values here?
//            				bytesReadInterval = (int) upStream / 8;
//            				nextBytesRead = (int) getWrittenBytes();
//                		}
//               		
//					}
//				}
//			}
		} else {
			log.debug("Session was null");
		}
		return success;
	}

	/**
	 * Return MINA I/O session.
	 *
	 * @return MINA O/I session, connection between two end-points
	 */
	public IoSession getIoSession() {
		return ioSession;
	}

	/** {@inheritDoc} */
	@Override
	public long getPendingMessages() {
		if (ioSession != null) {
			return ioSession.getScheduledWriteMessages();
		}
		return 0;
	}

	/** {@inheritDoc} */
	@Override
	public long getReadBytes() {
		if (ioSession != null) {
			return ioSession.getReadBytes();
		}
		return 0;
	}

	/** {@inheritDoc} */
	@Override
	public long getWrittenBytes() {
		if (ioSession != null) {
			return ioSession.getWrittenBytes();
		}
		return 0;
	}

	public void invokeMethod(String method) {
		invoke(method);
	}

	/** {@inheritDoc} */
	@Override
	public boolean isConnected() {
		// XXX Paul: not sure isClosing is actually working as we expect here
		return true;
		//return super.isConnected() && (ioSession != null) && ioSession.isConnected(); // && !ioSession.isClosing();
	}

	/** {@inheritDoc} */
	@Override
	protected void onInactive() {
		this.close();
	}

	/** {@inheritDoc} */
	@Override
	public void rawWrite(IoBuffer out) {
		if (ioSession != null) {
			ioSession.write(out);
		}
	}

	/**
	 * Setter for MINA I/O session (connection).
	 *
	 * @param protocolSession  Protocol session
	 */
	public void setIoSession(IoSession protocolSession) {
		SocketAddress remote = protocolSession.getRemoteAddress();
		if (remote instanceof InetSocketAddress) {
			remoteAddress = ((InetSocketAddress) remote).getAddress().getHostAddress();
			remotePort = ((InetSocketAddress) remote).getPort();
		} else {
			remoteAddress = remote.toString();
			remotePort = -1;
		}
		remoteAddresses = new ArrayList(1);
		remoteAddresses.add(remoteAddress);
		remoteAddresses = Collections.unmodifiableList(remoteAddresses);
		this.ioSession = protocolSession;
	}

	/** {@inheritDoc} */
	@Override
	public void write(Packet out) {
		if (ioSession != null) {
			writingMessage(out);
			ioSession.write(out);
		}
	}

//	@Override
//	public IScope getScope() {
//		// TODO Auto-generated method stub
//		return null;
//	}
}