test.tck.msgflow.callflows.tls.Shootme 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.tck.msgflow.callflows.tls;
import javax.sip.*;
import javax.sip.address.*;
import javax.sip.header.*;
import javax.sip.message.*;
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;
/**
* This class is a UAC template.
*
* @author M. Ranganathan
*/
public 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 static Logger logger = Logger.getLogger(Shootme.class);
static{
if (logger.getAllAppenders().equals(NullEnumeration.getInstance())) {
logger.addAppender(new ConsoleAppender(new SimpleLayout()));
}
}
private Dialog dialog;
private boolean okRecieved;
class ApplicationData {
protected int ackCount;
}
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);
} else if (request.getMethod().equals(Request.BYE)) {
processBye(requestEvent, serverTransactionId);
}
}
/**
* Process the ACK request. Send the bye and complete the call flow.
*/
public void processAck(RequestEvent requestEvent,
ServerTransaction serverTransaction) {
SipProvider sipProvider = (SipProvider) requestEvent.getSource();
try {
logger.info("shootme: got an ACK "
+ requestEvent.getRequest());
int ackCount = ((ApplicationData) dialog.getApplicationData()).ackCount;
if (ackCount == 1) {
dialog = inviteTid.getDialog();
this.sendReInvite(sipProvider);
} else
((ApplicationData) dialog.getApplicationData()).ackCount++;
} catch (Exception ex) {
String s = "Unexpected error";
logger.error(s,ex);
TlsTest.fail(s);
}
}
/**
* Process the invite request.
*/
public void processInvite(RequestEvent requestEvent,
ServerTransaction st) {
SipProvider sipProvider = (SipProvider) requestEvent.getSource();
Request request = requestEvent.getRequest();
logger.info("Got an INVITE " + request);
try {
logger.info("shootme: got an Invite sending OK");
// logger.info("shootme: " + request);
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);
if (st == null) {
st = sipProvider.getNewServerTransaction(request);
logger.info("Server transaction created!" + request);
logger.info("Dialog = " + st.getDialog());
if (st.getDialog().getApplicationData() == null) {
st.getDialog().setApplicationData(new ApplicationData());
}
} else {
// If Server transaction is not null, then
// this is a re-invite.
logger.info("This is a RE INVITE ");
TlsTest.assertSame( "Dialog mismatch ", st.getDialog(),this.dialog);
TlsTest.assertTrue( "Secure dialog", this.dialog.isSecure() );
}
// Thread.sleep(5000);
logger.info("got a server tranasaction " + st);
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);
logger.info("TxState after sendResponse = " + st.getState());
this.inviteTid = st;
} catch (Exception ex) {
String s = "unexpected exception";
logger.error(s,ex);
TlsTest.fail(s);
}
}
public void sendReInvite(SipProvider sipProvider) throws Exception {
Request inviteRequest = dialog.createRequest(Request.INVITE);
ClientTransaction ct = sipProvider
.getNewClientTransaction(inviteRequest);
dialog.sendRequest(ct);
}
/**
* Process the bye request.
*/
public void processBye(RequestEvent requestEvent,
ServerTransaction serverTransactionId) {
SipProvider sipProvider = (SipProvider) requestEvent.getSource();
Request request = requestEvent.getRequest();
try {
logger.info("shootme: got a bye sending OK.");
Response response = protocolObjects.messageFactory.createResponse(200, request);
if (serverTransactionId != null) {
serverTransactionId.sendResponse(response);
logger.info("Dialog State is "
+ serverTransactionId.getDialog().getState());
} else {
logger.info("null server tx.");
// sipProvider.sendResponse(response);
}
} catch (Exception ex) {
String s = "Unexpected exception";
logger.error(s,ex);
TlsTest.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)) {
this.okRecieved = true;
TlsTest.assertNotNull( "INVITE 200 response should match a transaction", tid );
Dialog dialog = tid.getDialog();
CSeqHeader cseq = (CSeqHeader) response.getHeader(CSeqHeader.NAME);
Request request = dialog.createAck(cseq.getSeqNumber());
dialog.sendAck(request);
}
if ( tid != null ) {
Dialog dialog = tid.getDialog();
logger.info("Dalog State = " + dialog.getState());
}
} catch (Exception ex) {
String s = "Unexpected exception";
logger.error(s,ex);
TlsTest.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 static void main(String args[]) throws Exception {
logger.addAppender( new ConsoleAppender(new SimpleLayout()));
ProtocolObjects protocolObjects = new ProtocolObjects("shootme", "gov.nist","udp",true,false, false);
Shootme shootme = new Shootme(protocolObjects);
shootme.createSipProvider().addSipListener(shootme);
}
public void checkState() {
ApplicationData data = (ApplicationData) dialog.getApplicationData();
TlsTest.assertTrue(data.ackCount == 1);
TlsTest.assertTrue(okRecieved);
}
/*
* (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 ");
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy