com.mobius.software.telco.protocols.diameter.impl.app.ClientCCSessionImpl Maven / Gradle / Ivy
package com.mobius.software.telco.protocols.diameter.impl.app;
/*
* Mobius Software LTD
* Copyright 2023, Mobius Software LTD and individual contributors
* by the @authors tag.
*
* This program is free software: you can redistribute it and/or modify
* under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation; either version 3 of
* the License, or (at your option) any later version.
*
* This program 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see extends DiameterSessionImpl implements ClientCCSession
{
private DiameterProvider, ?, ?, ?, ?> provider;
private RetransmissionCallback retransmissionCallback=new RetransmissionCallback(this);
//for serialization
public ClientCCSessionImpl(Long applicationID)
{
super(applicationID);
}
public ClientCCSessionImpl(String sessionID, Long applicationID, String remoteHost, String remoteRealm, DiameterProvider, ?, ?, ?, ?> provider)
{
super(sessionID, applicationID, remoteHost, remoteRealm, provider);
this.provider = provider;
}
@SuppressWarnings("unchecked")
@Override
public void setProvider(DiameterProvider provider)
{
this.provider = (DiameterProvider, ?, ?, ?, ?>)provider;
super.setProvider(provider);
}
@Override
public void sendInitialRequest(R1 request, AsyncCallback callback)
{
if(getSessionState()!=null && getSessionState()!=SessionStateEnum.IDLE && getSessionState()!=SessionStateEnum.OPEN)
{
callback.onError(new DiameterException("session state is invalid, can not send message", null, ResultCodes.DIAMETER_UNABLE_TO_COMPLY, null));
return;
}
try
{
request.setSessionId(getID());
}
catch(MissingAvpException | AvpNotSupportedException ex)
{
//will not happen
}
final Long startTime = System.currentTimeMillis();
provider.getStack().getQueue().offerLast(new Task()
{
@Override
public long getStartTime()
{
return startTime;
}
@Override
public void execute()
{
setSessionState(SessionStateEnum.PENDING);
if(!isRetry())
{
if(request.getDestinationRealm()==null && getRemoteRealm()!=null)
{
try
{
request.setDestinationRealm(getRemoteRealm());
}
catch(MissingAvpException ex)
{
}
}
DiameterRequest oldRequest = getLastSendRequest();
setLastSentRequest(request);
requestSent(oldRequest==null, request, callback);
}
provider.getStack().sendRequest(request, callback);
}
});
}
@Override
public void sendReauthAnswer(A2 answer, AsyncCallback callback)
{
if(getSessionState()==null || (getSessionState()!=SessionStateEnum.IDLE && getSessionState()!=SessionStateEnum.OPEN))
{
callback.onError(new DiameterException("session state is invalid, can not send message", null, ResultCodes.DIAMETER_UNABLE_TO_COMPLY, null));
return;
}
try
{
answer.setSessionId(getID());
}
catch(MissingAvpException | AvpNotSupportedException ex)
{
//will not happen
}
final Long startTime = System.currentTimeMillis();
provider.getStack().getQueue().offerLast(new Task()
{
@Override
public long getStartTime()
{
return startTime;
}
@Override
public void execute()
{
answerSent(answer, null, callback);
provider.getStack().sendAnswer(answer, getRemoteHost(), getRemoteRealm(), callback);
}
});
}
@Override
public void sendSessionTerminationRequest(R4 request, AsyncCallback callback)
{
if(getSessionState()==null || (getSessionState()!=SessionStateEnum.IDLE && getSessionState()!=SessionStateEnum.OPEN))
{
callback.onError(new DiameterException("session state is invalid, can not send message", null, ResultCodes.DIAMETER_UNABLE_TO_COMPLY, null));
return;
}
try
{
request.setSessionId(getID());
}
catch(MissingAvpException | AvpNotSupportedException ex)
{
//will not happen
}
final Long startTime = System.currentTimeMillis();
provider.getStack().getQueue().offerLast(new Task()
{
@Override
public long getStartTime()
{
return startTime;
}
@Override
public void execute()
{
setSessionState(SessionStateEnum.DISCONNECTED);
if(request.getDestinationRealm()==null && getRemoteRealm()!=null)
{
try
{
request.setDestinationRealm(getRemoteRealm());
}
catch(MissingAvpException ex)
{
}
}
setLastSentRequest(request);
requestSent(false, request, callback);
provider.getStack().sendRequest(request, callback);
}
});
}
@Override
public void sendAbortSessionAnswer(A3 answer, AsyncCallback callback)
{
if(getSessionState()==null || (getSessionState()!=SessionStateEnum.IDLE && getSessionState()!=SessionStateEnum.OPEN))
{
callback.onError(new DiameterException("session state is invalid, can not send message", null, ResultCodes.DIAMETER_UNABLE_TO_COMPLY, null));
return;
}
try
{
answer.setSessionId(getID());
}
catch(MissingAvpException | AvpNotSupportedException ex)
{
//will not happen
}
final Long startTime = System.currentTimeMillis();
provider.getStack().getQueue().offerLast(new Task()
{
@Override
public long getStartTime()
{
return startTime;
}
@Override
public void execute()
{
if(answer.getIsError()==null || !answer.getIsError())
setSessionState(SessionStateEnum.DISCONNECTED);
answerSent(answer, null, callback);
provider.getStack().sendAnswer(answer, getRemoteHost(), getRemoteRealm(), callback);
}
});
}
@Override
public void requestReceived(DiameterRequest request,String linkID, AsyncCallback callback)
{
@SuppressWarnings("unchecked")
Collection> listeners = (Collection>) provider.getClientListeners().values();
if(request instanceof ReAuthRequest)
{
try
{
super.requestReceived(request, linkID, callback);
@SuppressWarnings("unchecked")
R2 castedRequest = (R2)request;
for(ClientCCListener listener:listeners)
listener.onReauthRequest(castedRequest, this, linkID, callback);
}
catch(Exception ex)
{
logger.warn("An error occured while processing incoming request," + ex.getMessage(),ex);
callback.onError(new DiameterException("Received unexpected request", null, ResultCodes.DIAMETER_COMMAND_UNSUPPORTED, null));
return;
}
}
else if(request instanceof AbortSessionRequest)
{
try
{
super.requestReceived(request, linkID, callback);
@SuppressWarnings("unchecked")
R3 castedRequest = (R3)request;
for(ClientCCListener listener:listeners)
listener.onAbortSessionRequest(castedRequest, this, linkID, callback);
}
catch(Exception ex)
{
logger.warn("An error occured while processing incoming request," + ex.getMessage(),ex);
callback.onError(new DiameterException("Received unexpected request", null, ResultCodes.DIAMETER_COMMAND_UNSUPPORTED, null));
return;
}
}
else
{
callback.onError(new DiameterException("Received unexpected request", null, ResultCodes.DIAMETER_COMMAND_UNSUPPORTED, null));
return;
}
}
@SuppressWarnings("unchecked")
@Override
public void answerReceived(DiameterAnswer answer, Long idleTime,Boolean stopSendTimer,String linkID, AsyncCallback callback)
{
try
{
validateAnswer(answer);
}
catch(DiameterException ex)
{
callback.onError(new DiameterException("Received unexpected answer", null, ResultCodes.DIAMETER_COMMAND_UNSUPPORTED, null));
return;
}
DiameterRequest request = getLastSendRequest();
Long newTime = null;
if(request!=null)
{
Collection> listeners = null;
if(provider.getClientListeners()!=null)
listeners = (Collection>) provider.getClientListeners().values();
if(request instanceof SessionTerminationRequest)
{
if(answer instanceof SessionTerminationAnswer)
{
try
{
A4 castedAnswer = (A4)answer;
setSessionState(SessionStateEnum.IDLE);
terminate(castedAnswer.getResultCode());
if(listeners!=null)
{
for(ClientCCListener listener:listeners)
listener.onSessionTerminationAnswer(castedAnswer, this, linkID, callback);
}
}
catch(Exception ex)
{
callback.onError(new DiameterException("Received unexpected answer", null, ResultCodes.DIAMETER_COMMAND_UNSUPPORTED, null));
return;
}
}
}
else
{
try
{
A1 castedAnswer = (A1)answer;
if(getSessionState()==SessionStateEnum.PENDING)
{
try
{
if(castedAnswer.getValidityTime()!=null)
newTime = castedAnswer.getValidityTime()*1000L;
}
catch(DiameterException ex)
{
//validity time may not be supported
}
if(((CreditControlRequest)request).getCcRequestType()!=null && ((CreditControlRequest)request).getCcRequestType()==CcRequestTypeEnum.TERMINATION_REQUEST)
{
setSessionState(SessionStateEnum.IDLE);
terminate(castedAnswer.getResultCode());
if(listeners!=null)
{
for(ClientCCListener listener:listeners)
listener.onInitialAnswer(castedAnswer, this, linkID, callback);
}
}
else if(castedAnswer.getResultCode()!=null && !castedAnswer.getIsError())
{
if(((CreditControlRequest)request).getCcRequestType()!=null && ((CreditControlRequest)request).getCcRequestType()==CcRequestTypeEnum.EVENT_REQUEST)
{
setSessionState(SessionStateEnum.IDLE);
terminate(castedAnswer.getResultCode());
if(listeners!=null)
{
for(ClientCCListener listener:listeners)
listener.onInitialAnswer(castedAnswer, this, linkID, callback);
}
}
else
{
setSessionState(SessionStateEnum.OPEN);
super.answerReceived(answer, newTime,!isRetry(), linkID, callback);
if(listeners!=null)
{
for(ClientCCListener listener:listeners)
listener.onInitialAnswer(castedAnswer, this, linkID, callback);
}
}
}
else
{
Boolean shouldRetry = false;
if(((CreditControlRequest)request).getCcRequestType()!=null && ((CreditControlRequest)request).getCcRequestType()==CcRequestTypeEnum.EVENT_REQUEST)
{
if(((CreditControlRequest)request).getRequestedAction()!=null)
{
switch(((CreditControlRequest)request).getRequestedAction())
{
case DIRECT_DEBITING:
if(castedAnswer.getResultCode()>=4000L && castedAnswer.getResultCode()<5000L)
if(castedAnswer.getDirectDebitingFailureHandling()!=null && castedAnswer.getDirectDebitingFailureHandling()==DirectDebitingFailureHandlingEnum.CONTINUE)
{
request.setIsRetransmit(true);
shouldRetry = true;
}
break;
case REFUND_ACCOUNT:
if(castedAnswer.getResultCode()>=4000L && castedAnswer.getResultCode()<5000L)
{
request.setIsRetransmit(true);
shouldRetry = true;
}
break;
case CHECK_BALANCE:
case PRICE_ENQUIRY:
default:
break;
}
}
}
if(!shouldRetry)
{
setSessionState(SessionStateEnum.IDLE);
terminate(castedAnswer.getResultCode());
if(listeners!=null)
{
for(ClientCCListener listener:listeners)
listener.onInitialAnswer(castedAnswer, this, linkID, callback);
}
}
else
{
super.answerReceived(answer, newTime,!isRetry(), linkID, callback);
if(listeners!=null)
{
for(ClientCCListener listener:listeners)
listener.onInitialAnswer(castedAnswer, this, linkID,callback);
}
R1 realRequest=(R1)request;
setIsRetry(true);
sendInitialRequest(realRequest, retransmissionCallback);
}
}
}
}
catch(Exception ex)
{
callback.onError(new DiameterException("Received unexpected answer", null, ResultCodes.DIAMETER_COMMAND_UNSUPPORTED, null));
return;
}
}
}
else
callback.onError(new DiameterException("Received unexpected answer", null, ResultCodes.DIAMETER_COMMAND_UNSUPPORTED, null));
}
@Override
public void onTimeout()
{
DiameterRequest request = getLastSendRequest();
if(request!=null)
{
if(!(request instanceof SessionTerminationRequest))
{
if(getSessionState()==SessionStateEnum.PENDING)
{
Boolean shouldRetry = false;
if(((CreditControlRequest)request).getCcRequestType()!=null && ((CreditControlRequest)request).getCcRequestType()==CcRequestTypeEnum.EVENT_REQUEST)
{
if(((CreditControlRequest)request).getRequestedAction()!=null)
{
switch(((CreditControlRequest)request).getRequestedAction())
{
case DIRECT_DEBITING:
shouldRetry = true;
break;
case REFUND_ACCOUNT:
request.setIsRetransmit(true);
shouldRetry = true;
break;
case CHECK_BALANCE:
case PRICE_ENQUIRY:
default:
break;
}
}
}
if(shouldRetry)
{
if(provider.getClientListeners()!=null)
{
@SuppressWarnings("unchecked")
Collection> listeners = (Collection>) provider.getClientListeners().values();
for(ClientCCListener listener:listeners)
listener.onTimeout(getLastSendRequest(),this);
}
@SuppressWarnings("unchecked")
R1 realRequest=(R1)request;
setIsRetry(true);
sendInitialRequest(realRequest, retransmissionCallback);
return;
}
}
}
}
super.onTimeout();
}
@Override
public boolean isServer()
{
return false;
}
private class RetransmissionCallback implements AsyncCallback
{
private DiameterSession session;
public RetransmissionCallback(DiameterSession session)
{
this.session = session;
}
@Override
public void onSuccess()
{
}
@Override
public void onError(DiameterException ex)
{
setSessionState(SessionStateEnum.IDLE);
terminate(ex.getErrorCode());
if(provider.getClientListeners()!=null)
{
@SuppressWarnings("unchecked")
Collection> listeners = (Collection>) provider.getClientListeners().values();
for(ClientCCListener listener:listeners)
listener.onTimeout(getLastSendRequest(),session);
}
}
}
} © 2015 - 2024 Weber Informatics LLC | Privacy Policy