moonlight-android/moonlight-common/src/com/limelight/nvstream/rtsp/RtspConnection.java

package com.limelight.nvstream.rtsp;

import java.io.IOException;
import java.net.InetAddress;
import java.net.Inet6Address;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.util.HashMap;

import com.limelight.nvstream.StreamConfiguration;
import com.tinyrtsp.rtsp.message.RtspMessage;
import com.tinyrtsp.rtsp.message.RtspRequest;
import com.tinyrtsp.rtsp.message.RtspResponse;
import com.tinyrtsp.rtsp.parser.RtspStream;

public class RtspConnection {
	public static final int PORT = 48010;
	public static final int RTSP_TIMEOUT = 5000;
	
	// GFE 2.2.2+
	public static final int CLIENT_VERSION = 11;
	
	private int sequenceNumber = 1;
	private int sessionId = 0;
	
	private InetAddress host;
	private String hostStr;
	
	public RtspConnection(InetAddress host) {
		this.host = host;
		if (host instanceof Inet6Address) {
			// RFC2732-formatted IPv6 address for use in URL
			this.hostStr = "["+host.getHostAddress()+"]";
		}
		else {
			this.hostStr = host.getHostAddress();
		}
	}

	private RtspRequest createRtspRequest(String command, String target) {
		RtspRequest m = new RtspRequest(command, target, "RTSP/1.0",
				sequenceNumber++, new HashMap<String, String>(), null);
		m.setOption("X-GS-ClientVersion", ""+CLIENT_VERSION);
		return m;
	}
	
	private RtspResponse transactRtspMessage(RtspMessage m) throws IOException {
		Socket s = new Socket();
		try {
			s.setTcpNoDelay(true);
			s.connect(new InetSocketAddress(host, PORT), RTSP_TIMEOUT);
			
			RtspStream rtspStream = new RtspStream(s.getInputStream(), s.getOutputStream());
			try {
				rtspStream.write(m);
				return (RtspResponse) rtspStream.read();
			} finally {
				rtspStream.close();
			}
		} finally {
			s.close();
		}
	}
	
	private RtspResponse requestOptions() throws IOException {
		RtspRequest m = createRtspRequest("OPTIONS", "rtsp://"+hostStr);
		return transactRtspMessage(m);
	}
	
	private RtspResponse requestDescribe() throws IOException {
		RtspRequest m = createRtspRequest("DESCRIBE", "rtsp://"+hostStr);
		m.setOption("Accept", "application/sdp");
		m.setOption("If-Modified-Since", "Thu, 01 Jan 1970 00:00:00 GMT");
		return transactRtspMessage(m);
	}
	
	private RtspResponse setupStream(String streamName) throws IOException {
		RtspRequest m = createRtspRequest("SETUP", "streamid="+streamName);
		if (sessionId != 0) {
			m.setOption("Session", ""+sessionId);
		}
		m.setOption("Transport", " ");
		m.setOption("If-Modified-Since", "Thu, 01 Jan 1970 00:00:00 GMT");
		return transactRtspMessage(m);
	}
	
	private RtspResponse playStream(String streamName) throws IOException {
		RtspRequest m = createRtspRequest("PLAY", "streamid="+streamName);
		m.setOption("Session", ""+sessionId);
		return transactRtspMessage(m);
	}
	
	private RtspResponse sendVideoAnnounce(StreamConfiguration sc) throws IOException {
		RtspRequest m = createRtspRequest("ANNOUNCE", "streamid=video");
		m.setOption("Session", ""+sessionId);
		m.setOption("Content-type", "application/sdp");
		m.setPayload(SdpGenerator.generateSdpFromConfig(host, sc));
		m.setOption("Content-length", ""+m.getPayload().length());
		return transactRtspMessage(m);
	}
	
	public void doRtspHandshake(StreamConfiguration sc) throws IOException {
		RtspResponse r;
		
		r = requestOptions();
		if (r.getStatusCode() != 200) {
			throw new IOException("RTSP OPTIONS request failed: "+r.getStatusCode());
		}
		
		r = requestDescribe();
		if (r.getStatusCode() != 200) {
			throw new IOException("RTSP DESCRIBE request failed: "+r.getStatusCode());
		}
		
		r = setupStream("audio");
		if (r.getStatusCode() != 200) {
			throw new IOException("RTSP SETUP request failed: "+r.getStatusCode());
		}
		
		try {
			sessionId = Integer.parseInt(r.getOption("Session"));
		} catch (NumberFormatException e) {
			throw new IOException("RTSP SETUP response was malformed");
		}
		
		r = setupStream("video");
		if (r.getStatusCode() != 200) {
			throw new IOException("RTSP SETUP request failed: "+r.getStatusCode());
		}
		
		r = sendVideoAnnounce(sc);
		if (r.getStatusCode() != 200) {
			throw new IOException("RTSP ANNOUNCE request failed: "+r.getStatusCode());
		}
		
		r = playStream("video");
		if (r.getStatusCode() != 200) {
			throw new IOException("RTSP PLAY request failed: "+r.getStatusCode());
		}
		r = playStream("audio");
		if (r.getStatusCode() != 200) {
			throw new IOException("RTSP PLAY request failed: "+r.getStatusCode());
		}
	}
}