src/jlibrtp/RTPReceiverThread.java
author nikita@nikita-rack
Thu, 09 Apr 2009 20:26:58 +0200
changeset 99 8de21ac527ce
parent 13 e684f11070d5
permissions -rw-r--r--
revert pour refaire un push propre
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
13
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
     1
/**
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
     2
 * Java RTP Library (jlibrtp)
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
     3
 * Copyright (C) 2006 Arne Kepp
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
     4
 * 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
     5
 * This library is free software; you can redistribute it and/or
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
     6
 * modify it under the terms of the GNU Lesser General Public
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
     7
 * License as published by the Free Software Foundation; either
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
     8
 * version 2.1 of the License, or (at your option) any later version.
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
     9
 *
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    10
 * This library is distributed in the hope that it will be useful,
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    12
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    13
 * Lesser General Public License for more details.
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    14
 * 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    15
 * You should have received a copy of the GNU Lesser General Public
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    16
 * License along with this library; if not, write to the Free Software
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    17
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    18
 */
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    19
package jlibrtp;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    20
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    21
import java.io.IOException;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    22
import java.net.DatagramPacket;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    23
import java.net.InetSocketAddress;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    24
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    25
/**
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    26
 * The RTP receiver thread waits on the designated UDP socket for new packets.
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    27
 * 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    28
 * Once one arrives, it is parsed and tested. We also check the ip-address of the sender. 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    29
 * If accepted, the packet is added onto the packet buffer of the participant.
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    30
 * 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    31
 * A separate thread moves the packet from the packet buffer to the application.
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    32
 * 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    33
 * @author Arne Kepp
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    34
 */
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    35
public class RTPReceiverThread extends Thread {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    36
	/** Parent RTP Session */
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    37
	RTPSession rtpSession = null;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    38
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    39
	RTPReceiverThread(RTPSession session) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    40
		rtpSession = session;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    41
		if(RTPSession.rtpDebugLevel > 1) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    42
			System.out.println("<-> RTPReceiverThread created");
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    43
		} 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    44
	}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    45
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    46
	public void run() {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    47
		if(RTPSession.rtpDebugLevel > 1) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    48
			if(rtpSession.mcSession) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    49
				System.out.println("-> RTPReceiverThread.run() starting on MC " + rtpSession.rtpMCSock.getLocalPort() );
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    50
			} else {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    51
				System.out.println("-> RTPReceiverThread.run() starting on " + rtpSession.rtpSock.getLocalPort() );
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    52
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    53
		}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    54
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    55
		while(!rtpSession.endSession) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    56
			if(RTPSession.rtpDebugLevel > 6) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    57
				if(rtpSession.mcSession) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    58
					System.out.println("-> RTPReceiverThread.run() waiting for MC packet on " + rtpSession.rtpMCSock.getLocalPort() );
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    59
				} else {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    60
					System.out.println("-> RTPReceiverThread.run() waiting for packet on " + rtpSession.rtpSock.getLocalPort() );
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    61
				}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    62
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    63
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    64
			// Prepare a packet
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    65
			byte[] rawPkt = new byte[1500];
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    66
			DatagramPacket packet = new DatagramPacket(rawPkt, rawPkt.length);
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    67
			// Wait for it to arrive
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    68
			if(! rtpSession.mcSession) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    69
				//Unicast
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    70
				try {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    71
					rtpSession.rtpSock.receive(packet);
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    72
				} catch (IOException e) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    73
					if(!rtpSession.endSession) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    74
						e.printStackTrace();
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    75
					} else {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    76
						continue;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    77
					}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    78
				}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    79
			} else {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    80
				//Multicast 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    81
				try {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    82
					rtpSession.rtpMCSock.receive(packet);
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    83
				} catch (IOException e) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    84
					if(!rtpSession.endSession) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    85
						e.printStackTrace();
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    86
					} else {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    87
						continue;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    88
					}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    89
				}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    90
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    91
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    92
			// Parse the received RTP (?) packet
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    93
			RtpPkt pkt = new RtpPkt(rawPkt, packet.getLength());
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    94
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    95
			// Check whether it was valid.
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    96
			if(pkt == null) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    97
				System.out.println("Received invalid RTP packet. Ignoring");
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    98
				continue;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
    99
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   100
			
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   101
			long pktSsrc = pkt.getSsrc();
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   102
			
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   103
			// Check for loops and SSRC collisions
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   104
			if( rtpSession.ssrc == pktSsrc )
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   105
				rtpSession.resolveSsrcConflict();
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   106
			
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   107
			long[] csrcArray = pkt.getCsrcArray();
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   108
			if( csrcArray != null) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   109
				for(int i=0; i< csrcArray.length; i++) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   110
					if(csrcArray[i] == rtpSession.ssrc);
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   111
						rtpSession.resolveSsrcConflict();
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   112
				}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   113
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   114
			
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   115
			if(RTPSession.rtpDebugLevel > 17) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   116
				System.out.println("-> RTPReceiverThread.run() rcvd packet, seqNum " + pktSsrc );
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   117
				if(RTPSession.rtpDebugLevel > 10) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   118
					String str = new String(pkt.getPayload());
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   119
					System.out.println("-> RTPReceiverThread.run() payload is " + str );
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   120
				}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   121
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   122
			
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   123
			//Find the participant in the database based on SSRC
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   124
			Participant part = rtpSession.partDb.getParticipant(pktSsrc);
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   125
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   126
			if(part == null) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   127
				InetSocketAddress nullSocket = null;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   128
				part = new Participant((InetSocketAddress) packet.getSocketAddress(), nullSocket, pkt.getSsrc());
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   129
				part.unexpected = true;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   130
				rtpSession.partDb.addParticipant(1,part);
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   131
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   132
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   133
			// Do checks on whether the datagram came from the expected source for that SSRC.
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   134
			if(part.rtpAddress == null || packet.getAddress().equals(part.rtpAddress.getAddress())) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   135
				PktBuffer pktBuffer = part.pktBuffer;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   136
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   137
				if(pktBuffer != null) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   138
					//A buffer already exists, append to it
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   139
					pktBuffer.addPkt(pkt);
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   140
				} else {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   141
					// Create a new packet/frame buffer
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   142
					pktBuffer = new PktBuffer(this.rtpSession, part,pkt);
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   143
					part.pktBuffer = pktBuffer;
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   144
				}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   145
			} else {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   146
				System.out.println("RTPReceiverThread: Got an unexpected packet from " + pkt.getSsrc() 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   147
						+ " the sending ip-address was " + packet.getAddress().toString() 
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   148
						+ ", we expected from " + part.rtpAddress.toString());
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   149
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   150
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   151
			// Statistics for receiver report.
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   152
			part.updateRRStats(packet.getLength(), pkt);
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   153
			// Upate liveness
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   154
			part.lastRtpPkt = System.currentTimeMillis();
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   155
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   156
			if(RTPSession.rtpDebugLevel > 5) {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   157
				System.out.println("<-> RTPReceiverThread signalling pktBufDataReady");
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   158
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   159
			
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   160
			// Signal the thread that pushes data to application
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   161
			rtpSession.pktBufLock.lock();
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   162
			try { rtpSession.pktBufDataReady.signalAll(); } finally {
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   163
				rtpSession.pktBufLock.unlock();
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   164
			}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   165
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   166
		}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   167
	}
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   168
e684f11070d5 ajout de jlibrtp
nikita@nikita-rack
parents:
diff changeset
   169
}