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

test.unit.gov.nist.javax.sip.stack.AckReTransmissionTest Maven / Gradle / Ivy

/*
* Conditions Of Use 
* 
* This software was developed by employees of the National Institute of
* Standards and Technology (NIST), and others. 
* This software is has been contributed to the public domain. 
* As a result, a formal license is not needed to use the software.
* 
* This software is provided "AS IS."  
* NIST MAKES NO WARRANTY OF ANY KIND, EXPRESS, IMPLIED
* OR STATUTORY, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTY OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, NON-INFRINGEMENT
* AND DATA ACCURACY.  NIST does not warrant or make any representations
* regarding the use of the software or the results thereof, including but
* not limited to the correctness, accuracy, reliability or usefulness of
* the software.
* 
* 
*/
/**
 * 
 */
package test.unit.gov.nist.javax.sip.stack;

import java.util.ArrayList;
import java.util.EventObject;

import javax.sip.ClientTransaction;
import javax.sip.Dialog;
import javax.sip.DialogTerminatedEvent;
import javax.sip.IOExceptionEvent;
import javax.sip.ListeningPoint;
import javax.sip.RequestEvent;
import javax.sip.ResponseEvent;
import javax.sip.ServerTransaction;
import javax.sip.SipListener;
import javax.sip.SipProvider;
import javax.sip.TimeoutEvent;
import javax.sip.Transaction;
import javax.sip.TransactionTerminatedEvent;
import javax.sip.address.Address;
import javax.sip.address.SipURI;
import javax.sip.header.CSeqHeader;
import javax.sip.header.CallIdHeader;
import javax.sip.header.ContactHeader;
import javax.sip.header.ContentTypeHeader;
import javax.sip.header.FromHeader;
import javax.sip.header.Header;
import javax.sip.header.MaxForwardsHeader;
import javax.sip.header.RouteHeader;
import javax.sip.header.ToHeader;
import javax.sip.header.ViaHeader;
import javax.sip.message.Request;
import javax.sip.message.Response;

import org.apache.log4j.ConsoleAppender;
import org.apache.log4j.Logger;
import org.apache.log4j.SimpleLayout;
import org.apache.log4j.helpers.NullEnumeration;

import test.tck.msgflow.callflows.ProtocolObjects;
import test.tck.msgflow.callflows.ScenarioHarness;

/**
 * 
 */
public class AckReTransmissionTest extends ScenarioHarness implements SipListener {

	
	protected Shootist shootist;

	private Shootme shootme;

	private static Logger logger = Logger.getLogger("test.tck");

	static {
		if (!logger.isAttached(console))
			logger.addAppender(console);
	}
	
	 class Shootme  implements SipListener {


			private ProtocolObjects  protocolObjects;
			
			
			// To run on two machines change these to suit.
			public static final String myAddress = "127.0.0.1";

			public static final int myPort = 5070;

			private ServerTransaction inviteTid;

			
			private Dialog dialog;

			private ServerTransaction reSendSt = null;
			private Response reSendResponse = null;
			private int dropAckCount = 0;

			public Shootme(ProtocolObjects protocolObjects) {
				this.protocolObjects = protocolObjects;
			}
			

			public void processRequest(RequestEvent requestEvent) {
				Request request = requestEvent.getRequest();
				ServerTransaction serverTransactionId = requestEvent
						.getServerTransaction();

				logger.info("\n\nRequest " + request.getMethod()
						+ " received at " + protocolObjects.sipStack.getStackName()
						+ " with server transaction id " + serverTransactionId);

				if (request.getMethod().equals(Request.INVITE)) {
					processInvite(requestEvent, serverTransactionId);
				} else if (request.getMethod().equals(Request.ACK)) {
					processAck(requestEvent, serverTransactionId);
				}

			}

			/**
			 * Process the ACK request. Send the bye and complete the call flow.
			 */
			public void processAck(RequestEvent requestEvent,
					ServerTransaction serverTransaction) {
				logger.info("shootme: got an ACK " + requestEvent.getRequest());
				if (dropAckCount <= 3){
					boolean skip = false;
					//drop the ACK, resend 200 OK
					try {
						reSendSt.sendResponse(reSendResponse);
						logger.info("shootme: resending the previous 200");
				    }
					catch (Exception ex) {
						String s = "Unexpected error";
						logger.error(s, ex);
						AckReTransmissionTest.fail(s);
						skip = true;
					}
					if(!skip) {
						dropAckCount++;
						return;
					}
				}
				SipProvider sipProvider = (SipProvider) requestEvent.getSource();
				try {
					AckReTransmissionTest.assertTrue("ACK was not successfully retransmitted 4 times", 4 == dropAckCount);
					//Create BYE request
					Request byeRequest = dialog.createRequest(Request.BYE);
					ClientTransaction ct = sipProvider.getNewClientTransaction(byeRequest);
					dialog.sendRequest(ct);
					reSendSt = null;
					reSendResponse = null;
					logger.info("shootme: Sending a BYE");
				} catch (Exception ex) {
					String s = "Unexpected error";
					logger.error(s,ex);
					AckReTransmissionTest.fail(s);
				}
			}

			/**
			 * Process the invite request.
			 */
			public void processInvite(RequestEvent requestEvent,
					ServerTransaction serverTransaction) {
				SipProvider sipProvider = (SipProvider) requestEvent.getSource();
				Request request = requestEvent.getRequest();
				logger.info("Got an INVITE  " + request);
				try {
					Response response = protocolObjects.messageFactory.createResponse(180, request);
					ToHeader toHeader = (ToHeader) response.getHeader(ToHeader.NAME);
					toHeader.setTag("4321");
					Address address = protocolObjects.addressFactory.createAddress("Shootme ");
					ContactHeader contactHeader = protocolObjects.headerFactory
							.createContactHeader(address);
					response.addHeader(contactHeader);
					ServerTransaction st = requestEvent.getServerTransaction();

					if (st == null) {
						st = sipProvider.getNewServerTransaction(request);
						logger.info("Server transaction created!" + request);

						logger.info("Dialog = " + st.getDialog());
					}

					byte[] content = request.getRawContent();
					if (content != null) {
						logger.info(" content = " + new String(content));
						ContentTypeHeader contentTypeHeader = protocolObjects.headerFactory
								.createContentTypeHeader("application", "sdp");
						logger.info("response = " + response);
						response.setContent(content, contentTypeHeader);
					}
					dialog = st.getDialog();
					if (dialog != null) {
						logger.info("Dialog " + dialog);
						logger.info("Dialog state " + dialog.getState());
					}
					st.sendResponse(response);
					response = protocolObjects.messageFactory.createResponse(200, request);
					toHeader = (ToHeader) response.getHeader(ToHeader.NAME);
					toHeader.setTag("4321");
					// Application is supposed to set.
					response.addHeader(contactHeader);
					st.sendResponse(response);
					reSendSt = st;
					reSendResponse = response;
					logger.info("TxState after sendResponse = " + st.getState());
					this.inviteTid = st;
				} catch (Exception ex) {
					String s = "unexpected exception";
					
					logger.error(s,ex);
					AckReTransmissionTest.fail(s);
				}
			}

			
			public void processResponse(ResponseEvent responseReceivedEvent) {
				logger.info("Got a response");
				Response response = (Response) responseReceivedEvent.getResponse();
				Transaction tid = responseReceivedEvent.getClientTransaction();

				logger.info("Response received with client transaction id "
						+ tid + ":\n" + response);
				try {
					if (response.getStatusCode() == Response.OK
							&& ((CSeqHeader) response.getHeader(CSeqHeader.NAME))
									.getMethod().equals(Request.INVITE)) {
						Dialog dialog = tid.getDialog();
						CSeqHeader cseq = (CSeqHeader) response.getHeader(CSeqHeader.NAME);
						Request ackRequest = dialog.createAck(cseq.getSeqNumber());
						dialog.sendAck(ackRequest);
					}
					if ( tid != null ) {
					    Dialog dialog = tid.getDialog();
					    logger.info("Dialog State = " + dialog.getState());			}
				} catch (Exception ex) {			
					String s = "Unexpected exception";			
					logger.error(s,ex);
					AckReTransmissionTest.fail(s);
				}

			}

			public void processTimeout(javax.sip.TimeoutEvent timeoutEvent) {
				Transaction transaction;
				if (timeoutEvent.isServerTransaction()) {
					transaction = timeoutEvent.getServerTransaction();
				} else {
					transaction = timeoutEvent.getClientTransaction();
				}
				logger.info("state = " + transaction.getState());
				logger.info("dialog = " + transaction.getDialog());
				logger.info("dialogState = "
						+ transaction.getDialog().getState());
				logger.info("Transaction Time out");
			}

			

			
			public SipProvider createSipProvider() throws Exception {
				ListeningPoint lp = protocolObjects.sipStack.createListeningPoint(myAddress,
						myPort, protocolObjects.transport);

				
				SipProvider sipProvider = protocolObjects.sipStack.createSipProvider(lp);
				return sipProvider;
			}
			
			
	

			public void checkState() {
				AckReTransmissionTest.assertTrue("ACK was not successfully retransmitted 4 times", 4==dropAckCount);
			}
			/*
			 * (non-Javadoc)
			 * 
			 * @see javax.sip.SipListener#processIOException(javax.sip.IOExceptionEvent)
			 */
			public void processIOException(IOExceptionEvent exceptionEvent) {
				logger.error("An IO Exception was detected : "
						+ exceptionEvent.getHost());

			}

			/*
			 * (non-Javadoc)
			 * 
			 * @see javax.sip.SipListener#processTransactionTerminated(javax.sip.TransactionTerminatedEvent)
			 */
			public void processTransactionTerminated(
					TransactionTerminatedEvent transactionTerminatedEvent) {
				logger.info("Tx terminated event ");

			}

			/*
			 * (non-Javadoc)
			 * 
			 * @see javax.sip.SipListener#processDialogTerminated(javax.sip.DialogTerminatedEvent)
			 */
			public void processDialogTerminated(
					DialogTerminatedEvent dialogTerminatedEvent) {
				logger.info("Dialog terminated event detected ");

			}

		}

	class Shootist  implements SipListener {

		private SipProvider provider;

		private int reInviteCount;

		private ContactHeader contactHeader;

		private ListeningPoint listeningPoint;

		private int counter;

		private  String PEER_ADDRESS = Shootme.myAddress;

		private  int PEER_PORT = Shootme.myPort;

		private  String peerHostPort = PEER_ADDRESS + ":" + PEER_PORT;

		// To run on two machines change these to suit.
		public static final String myAddress = "127.0.0.1";

		private static final int myPort = 5060;

		protected ClientTransaction inviteTid;

		
		
		
		private ProtocolObjects protocolObjects;

		private Dialog dialog;

		
		
		

		public Shootist(ProtocolObjects protocolObjects) {
			super();
			this.protocolObjects = protocolObjects;
		
		}

		

		public void processRequest(RequestEvent requestReceivedEvent) {
			Request request = requestReceivedEvent.getRequest();
			ServerTransaction serverTransactionId = requestReceivedEvent
					.getServerTransaction();

			logger.info("\n\nRequest " + request.getMethod() + " received at "
					+ protocolObjects.sipStack.getStackName()
					+ " with server transaction id " + serverTransactionId);

			if (request.getMethod().equals(Request.BYE))
				processBye(request, serverTransactionId);
		}

		public void processBye(Request request,
				ServerTransaction serverTransactionId) {
			try {
				logger.info("shootist:  got a bye .");
				if (serverTransactionId == null) {
					logger.info("shootist:  null TID.");
					return;
				}
				Response response = protocolObjects.messageFactory.createResponse(
						200, request);
				serverTransactionId.sendResponse(response);
			} catch (Exception ex) {
				logger.error("unexpected exception",ex);
				AckReTransmissionTest.fail("unexpected exception");
				
			}
		}

		public void processResponse(ResponseEvent responseReceivedEvent) {
			logger.info("Got a response");

			Response response = (Response) responseReceivedEvent.getResponse();
			Transaction tid = responseReceivedEvent.getClientTransaction();

			logger.info("Response received with client transaction id " + tid
					+ ":\n" + response.getStatusCode());
			if (tid == null) {
				logger.info("Stray response -- dropping ");
				return;
			}
			logger.info("transaction state is " + tid.getState());
			logger.info("Dialog = " + tid.getDialog());
			logger.info("Dialog State is " + tid.getDialog().getState());
			SipProvider provider = (SipProvider) responseReceivedEvent.getSource();

			try {
				if (response.getStatusCode() == Response.OK
						&& ((CSeqHeader) response.getHeader(CSeqHeader.NAME))
								.getMethod().equals(Request.INVITE)) {
					
					Dialog dialog = tid.getDialog();
					CSeqHeader cseq = (CSeqHeader) response.getHeader(CSeqHeader.NAME);
					Request ackRequest = dialog.createAck(cseq.getSeqNumber());
					logger.info("Ack request to send = " + ackRequest);
					logger.info("Sending ACK");
					dialog.sendAck(ackRequest);
				}
			} catch (Exception ex) {
				logger.error(ex);
				AckReTransmissionTest.fail("unexpected exception");
			}

		}

		public void processTimeout(javax.sip.TimeoutEvent timeoutEvent) {

			logger.info("Transaction Time out");
			logger.info("TimeoutEvent " + timeoutEvent.getTimeout());
		}

		public SipProvider createSipProvider() {
			try {
				listeningPoint = protocolObjects.sipStack.createListeningPoint(
						myAddress, myPort, protocolObjects.transport);

				provider = protocolObjects.sipStack
						.createSipProvider(listeningPoint);
				return provider;
			} catch (Exception ex) {
				logger.error(ex);
				AckReTransmissionTest.fail("unable to create provider");
				return null;
			}
		}

		public void sendInvite() {

			try {

				// Note that a provider has multiple listening points.
				// all the listening points must have the same IP address
				// and port but differ in their transport parameters.

				String fromName = "BigGuy";
				String fromSipAddress = "here.com";
				String fromDisplayName = "The Master Blaster";

				String toSipAddress = "there.com";
				String toUser = "LittleGuy";
				String toDisplayName = "The Little Blister";

				// create >From Header
				SipURI fromAddress = protocolObjects.addressFactory.createSipURI(
						fromName, fromSipAddress);

				Address fromNameAddress = protocolObjects.addressFactory
						.createAddress(fromAddress);
				fromNameAddress.setDisplayName(fromDisplayName);
				FromHeader fromHeader = protocolObjects.headerFactory
						.createFromHeader(fromNameAddress, new Integer((int) (Math
								.random() * Integer.MAX_VALUE)).toString());

				// create To Header
				SipURI toAddress = protocolObjects.addressFactory.createSipURI(
						toUser, toSipAddress);
				Address toNameAddress = protocolObjects.addressFactory
						.createAddress(toAddress);
				toNameAddress.setDisplayName(toDisplayName);
				ToHeader toHeader = protocolObjects.headerFactory.createToHeader(
						toNameAddress, null);

				// create Request URI
				SipURI requestURI = protocolObjects.addressFactory.createSipURI(
						toUser, peerHostPort);

				// Create ViaHeaders

				ArrayList viaHeaders = new ArrayList();
				int port = provider.getListeningPoint(protocolObjects.transport)
						.getPort();

				ViaHeader viaHeader = protocolObjects.headerFactory
						.createViaHeader(myAddress, port,
								protocolObjects.transport, null);

				// add via headers
				viaHeaders.add(viaHeader);

				// Create ContentTypeHeader
				ContentTypeHeader contentTypeHeader = protocolObjects.headerFactory
						.createContentTypeHeader("application", "sdp");

				// Create a new CallId header
				CallIdHeader callIdHeader = provider.getNewCallId();
				// JvB: Make sure that the implementation matches the messagefactory
				callIdHeader = protocolObjects.headerFactory.createCallIdHeader( callIdHeader.getCallId() );


				// Create a new Cseq header
				CSeqHeader cSeqHeader = protocolObjects.headerFactory
						.createCSeqHeader(1L, Request.INVITE);

				// Create a new MaxForwardsHeader
				MaxForwardsHeader maxForwards = protocolObjects.headerFactory
						.createMaxForwardsHeader(70);

				// Create the request.
				Request request = protocolObjects.messageFactory.createRequest(
						requestURI, Request.INVITE, callIdHeader, cSeqHeader,
						fromHeader, toHeader, viaHeaders, maxForwards);
				// Create contact headers

				// Create the contact name address.
				SipURI contactURI = protocolObjects.addressFactory.createSipURI(
						fromName, myAddress);
				contactURI.setPort(provider.getListeningPoint(
						protocolObjects.transport).getPort());

				Address contactAddress = protocolObjects.addressFactory
						.createAddress(contactURI);

				// Add the contact address.
				contactAddress.setDisplayName(fromName);

				contactHeader = protocolObjects.headerFactory
						.createContactHeader(contactAddress);
				request.addHeader(contactHeader);

				// Add the extension header.
				Header extensionHeader = protocolObjects.headerFactory
						.createHeader("My-Header", "my header value");
				request.addHeader(extensionHeader);

				String sdpData = "v=0\r\n"
						+ "o=4855 13760799956958020 13760799956958020"
						+ " IN IP4  129.6.55.78\r\n" + "s=mysession session\r\n"
						+ "p=+46 8 52018010\r\n" + "c=IN IP4  129.6.55.78\r\n"
						+ "t=0 0\r\n" + "m=audio 6022 RTP/AVP 0 4 18\r\n"
						+ "a=rtpmap:0 PCMU/8000\r\n" + "a=rtpmap:4 G723/8000\r\n"
						+ "a=rtpmap:18 G729A/8000\r\n" + "a=ptime:20\r\n";

				request.setContent(sdpData, contentTypeHeader);

				// The following is the preferred method to route requests
				// to the peer. Create a route header and set the "lr"
				// parameter for the router header.

				Address address = protocolObjects.addressFactory
						.createAddress("");
				// SipUri sipUri = (SipUri) address.getURI();
				// sipUri.setPort(PEER_PORT);

				RouteHeader routeHeader = protocolObjects.headerFactory
						.createRouteHeader(address);
				((SipURI)address.getURI()).setLrParam();
				request.addHeader(routeHeader);
				extensionHeader = protocolObjects.headerFactory.createHeader(
						"My-Other-Header", "my new header value ");
				request.addHeader(extensionHeader);

				Header callInfoHeader = protocolObjects.headerFactory.createHeader(
						"Call-Info", "");
				request.addHeader(callInfoHeader);

				// Create the client transaction.
				this.inviteTid = provider.getNewClientTransaction(request);
				this.dialog = this.inviteTid.getDialog();
				// Note that the response may have arrived right away so 
				// we cannot check after the message is sent.
				AckReTransmissionTest.assertTrue(this.dialog.getState() == null);
			
				// send the request out.
				this.inviteTid.sendRequest();

					
			} catch (Exception ex) {
				logger.error("Unexpected exception", ex);
				AckReTransmissionTest.fail("unexpected exception");
			}
		}

		
		
		/*
		 * (non-Javadoc)
		 * 
		 * @see javax.sip.SipListener#processIOException(javax.sip.IOExceptionEvent)
		 */
		public void processIOException(IOExceptionEvent exceptionEvent) {
			logger.error("IO Exception!");
			AckReTransmissionTest.fail("Unexpected exception");

		}

		/*
		 * (non-Javadoc)
		 * 
		 * @see javax.sip.SipListener#processTransactionTerminated(javax.sip.TransactionTerminatedEvent)
		 */
		public void processTransactionTerminated(
				TransactionTerminatedEvent transactionTerminatedEvent) {

			logger.info("Transaction Terminated Event!");
		}

		/*
		 * (non-Javadoc)
		 * 
		 * @see javax.sip.SipListener#processDialogTerminated(javax.sip.DialogTerminatedEvent)
		 */
		public void processDialogTerminated(
				DialogTerminatedEvent dialogTerminatedEvent) {
			logger.info("Dialog Terminated Event!");

		}
	}

	private SipListener getSipListener(EventObject sipEvent) {
		SipProvider source = (SipProvider) sipEvent.getSource();
		SipListener listener = (SipListener) providerTable.get(source);
		assertTrue(listener != null);
		return listener;
	}

	public AckReTransmissionTest()
	{
		super("AckReTransmissionTest", true);
	}

	public void setUp() {

		try {
			this.transport = "udp";
			
			super.setUp();
			shootist = new Shootist(riProtocolObjects);
			SipProvider shootistProvider = shootist.createSipProvider();
			providerTable.put(shootistProvider, shootist);

			shootme = new Shootme(tiProtocolObjects);
			SipProvider shootmeProvider = shootme.createSipProvider();
			providerTable.put(shootmeProvider, shootme);
			shootistProvider.addSipListener(this);
			shootmeProvider.addSipListener(this);

			riProtocolObjects.start();
			if (tiProtocolObjects != riProtocolObjects)
				tiProtocolObjects.start();
		} catch (Exception ex) {
			ex.printStackTrace();
			fail("unexpected exception ");
		}
	}

	public void testAckReTransmission() {
		this.shootist.sendInvite();
	}

	public void tearDown() {
		try {
			Thread.sleep(2000);
			this.shootme.checkState();
			tiProtocolObjects.destroy();
			if (tiProtocolObjects != riProtocolObjects)
				riProtocolObjects.destroy();
			Thread.sleep(1000);
			this.providerTable.clear();
			logTestCompleted();
		} catch (Exception ex) {
			ex.printStackTrace();
		}

	}

	public void processRequest(RequestEvent requestEvent) {
		getSipListener(requestEvent).processRequest(requestEvent);

	}

	public void processResponse(ResponseEvent responseEvent) {
		getSipListener(responseEvent).processResponse(responseEvent);

	}

	public void processTimeout(TimeoutEvent timeoutEvent) {
		getSipListener(timeoutEvent).processTimeout(timeoutEvent);
	}

	public void processIOException(IOExceptionEvent exceptionEvent) {
		fail("unexpected exception");

	}

	public void processTransactionTerminated(
			TransactionTerminatedEvent transactionTerminatedEvent) {
		getSipListener(transactionTerminatedEvent)
				.processTransactionTerminated(transactionTerminatedEvent);

	}

	public void processDialogTerminated(
			DialogTerminatedEvent dialogTerminatedEvent) {
		getSipListener(dialogTerminatedEvent).processDialogTerminated(
				dialogTerminatedEvent);

	}

}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy