com.mobius.software.telco.protocols.diameter.impl.app.ClientAuthSessionImpl 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 ClientAuthSession
{
private DiameterProvider, ?, ?, ?, ?> provider;
//for serialization
public ClientAuthSessionImpl(Long applicationID)
{
super(applicationID);
}
public ClientAuthSessionImpl(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(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, new CallbackWrapper(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()
{
if(answer.getIsError()!=null && answer.getIsError())
{
setSessionState(SessionStateEnum.IDLE);
terminate(answer.getResultCode());
}
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);
}
});
}
@SuppressWarnings("unchecked")
@Override
public void requestReceived(DiameterRequest request, String linkID, AsyncCallback callback)
{
Collection> listeners = null;
if(provider.getClientListeners()!=null)
listeners = (Collection>) provider.getClientListeners().values();
if(request instanceof ReAuthRequest)
{
try
{
R2 castedRequest = (R2)request;
super.requestReceived(request, linkID, callback);
if(listeners!=null)
{
for(ClientAuthListener 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
{
R3 castedRequest = (R3)request;
super.requestReceived(request, linkID, callback);
for(ClientAuthListener 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();
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(ClientAuthListener 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)
{
if(castedAnswer.getResultCode()!=null && !castedAnswer.getIsError())
{
setSessionState(SessionStateEnum.OPEN);
super.answerReceived(answer, idleTime, stopSendTimer, linkID, callback);
if(listeners!=null)
{
for(ClientAuthListener listener:listeners)
listener.onInitialAnswer(castedAnswer, this, linkID, callback);
}
}
else
{
setSessionState(SessionStateEnum.IDLE);
terminate(castedAnswer.getResultCode());
if(listeners!=null)
{
for(ClientAuthListener listener:listeners)
listener.onInitialAnswer(castedAnswer, this, linkID, callback);
}
}
}
}
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 boolean isServer()
{
return false;
}
private class CallbackWrapper implements AsyncCallback
{
private AsyncCallback realCallback;
public CallbackWrapper(AsyncCallback realCallback)
{
this.realCallback = realCallback;
}
@Override
public void onSuccess()
{
realCallback.onSuccess();
}
@Override
public void onError(DiameterException ex)
{
setSessionState(SessionStateEnum.IDLE);
terminate(ex.getErrorCode());
realCallback.onError(ex);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy