/*
* JBoss, Home of Professional Open Source
* Copyright 2011, Red Hat, Inc. and/or its affiliates, and individual
* contributors as indicated by the @authors tag. All rights reserved.
* See the copyright.txt in the distribution for a full listing
* of individual contributors.
*
* This copyrighted material is made available to anyone wishing to use,
* modify, copy, or redistribute it subject to the terms and conditions
* of the GNU General Public License, v. 2.0.
*
* 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
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License,
* v. 2.0 along with this distribution; if not, write to the Free
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
* MA 02110-1301, USA.
*/
package org.mobicents.diameter.stack.functional.sh.base;
import org.jdiameter.api.Answer;
import org.jdiameter.api.Avp;
import org.jdiameter.api.AvpSet;
import org.jdiameter.api.IllegalDiameterStateException;
import org.jdiameter.api.InternalException;
import org.jdiameter.api.NetworkReqListener;
import org.jdiameter.api.OverloadException;
import org.jdiameter.api.Request;
import org.jdiameter.api.RouteException;
import org.jdiameter.api.app.AppAnswerEvent;
import org.jdiameter.api.app.AppRequestEvent;
import org.jdiameter.api.app.AppSession;
import org.jdiameter.api.sh.ServerShSession;
import org.jdiameter.api.sh.events.ProfileUpdateAnswer;
import org.jdiameter.api.sh.events.ProfileUpdateRequest;
import org.jdiameter.api.sh.events.PushNotificationAnswer;
import org.jdiameter.api.sh.events.PushNotificationRequest;
import org.jdiameter.api.sh.events.SubscribeNotificationsAnswer;
import org.jdiameter.api.sh.events.SubscribeNotificationsRequest;
import org.jdiameter.api.sh.events.UserDataAnswer;
import org.jdiameter.api.sh.events.UserDataRequest;
import org.jdiameter.common.impl.app.sh.ProfileUpdateAnswerImpl;
import org.jdiameter.common.impl.app.sh.PushNotificationRequestImpl;
import org.jdiameter.common.impl.app.sh.SubscribeNotificationsAnswerImpl;
import org.jdiameter.common.impl.app.sh.UserDataAnswerImpl;
import org.mobicents.diameter.stack.functional.Utils;
/**
* Base implementation of Server
*
* @author <a href="mailto:brainslog@gmail.com"> Alexandre Mendonca </a>
* @author <a href="mailto:baranowb@gmail.com"> Bartosz Baranowski </a>
*/
public class ServerPNR extends org.mobicents.diameter.stack.functional.sh.AbstractServer {
protected boolean sentSubscribeNotifications;
protected boolean sentProfileUpdate;
protected boolean sentUserData;
protected boolean sentPushNotification;
protected boolean receiveSubscribeNotifications;
protected boolean receiveProfileUpdate;
protected boolean receiveUserData;
protected boolean receivePushNotification;
protected SubscribeNotificationsRequest subscribeNotificationsRequest;
protected ProfileUpdateRequest profileUpdateRequest;
protected UserDataRequest userDataRequest;
// ------- send methods to trigger answer
public void sendSubscribeNotifications() throws Exception {
if (!this.receiveSubscribeNotifications || this.subscribeNotificationsRequest == null) {
fail("Did not receive SUBSCRIBE or answer already sent.", null);
throw new Exception("Did not receive SUBSCRIBE or answer already sent. Request: " + this.subscribeNotificationsRequest);
}
SubscribeNotificationsAnswer answer = new SubscribeNotificationsAnswerImpl((Request) this.subscribeNotificationsRequest.getMessage(), 2001);
AvpSet reqSet = subscribeNotificationsRequest.getMessage().getAvps();
AvpSet set = answer.getMessage().getAvps();
set.removeAvp(Avp.DESTINATION_HOST);
set.removeAvp(Avp.DESTINATION_REALM);
set.addAvp(reqSet.getAvp(Avp.CC_REQUEST_TYPE), reqSet.getAvp(Avp.CC_REQUEST_NUMBER), reqSet.getAvp(Avp.AUTH_APPLICATION_ID));
this.serverShSession.sendSubscribeNotificationsAnswer(answer);
Utils.printMessage(log, super.stack.getDictionary(), answer.getMessage(), true);
this.subscribeNotificationsRequest = null;
}
public void sendProfileUpdate() throws Exception {
if (!this.receiveProfileUpdate || this.profileUpdateRequest == null) {
fail("Did not receive UPDATE or answer already sent.", null);
throw new Exception("Did not receive UPDATE or answer already sent. Request: " + this.profileUpdateRequest);
}
ProfileUpdateAnswer answer = new ProfileUpdateAnswerImpl((Request) this.profileUpdateRequest.getMessage(), 2001);
AvpSet reqSet = profileUpdateRequest.getMessage().getAvps();
AvpSet set = answer.getMessage().getAvps();
set.removeAvp(Avp.DESTINATION_HOST);
set.removeAvp(Avp.DESTINATION_REALM);
set.addAvp(reqSet.getAvp(Avp.CC_REQUEST_TYPE), reqSet.getAvp(Avp.CC_REQUEST_NUMBER), reqSet.getAvp(Avp.AUTH_APPLICATION_ID));
this.serverShSession.sendProfileUpdateAnswer(answer);
Utils.printMessage(log, super.stack.getDictionary(), answer.getMessage(), true);
this.profileUpdateRequest = null;
}
public void sendUserData() throws Exception {
if (!this.receiveUserData || this.userDataRequest == null) {
fail("Did not receive USER DATA or answer already sent.", null);
throw new Exception("Did not receive USER DATA or answer already sent. Request: " + this.userDataRequest);
}
UserDataAnswer answer = new UserDataAnswerImpl((Request) this.userDataRequest.getMessage(), 2001);
AvpSet reqSet = userDataRequest.getMessage().getAvps();
AvpSet set = answer.getMessage().getAvps();
set.removeAvp(Avp.DESTINATION_HOST);
set.removeAvp(Avp.DESTINATION_REALM);
set.addAvp(reqSet.getAvp(Avp.CC_REQUEST_TYPE), reqSet.getAvp(Avp.CC_REQUEST_NUMBER), reqSet.getAvp(Avp.AUTH_APPLICATION_ID));
this.serverShSession.sendUserDataAnswer(answer);
Utils.printMessage(log, super.stack.getDictionary(), answer.getMessage(), true);
this.userDataRequest = null;
}
public void sendPushNotification() throws Exception {
if (super.serverShSession == null) {
super.serverShSession =
this.sessionFactory.getNewAppSession(this.sessionFactory.getSessionId("xxTESTxx"), getApplicationId(), ServerShSession.class, (Object) null);
}
PushNotificationRequest request = new PushNotificationRequestImpl(super.serverShSession.getSessions().get(0)
.createRequest(PushNotificationRequest.code, getApplicationId(), getClientRealmName()));
AvpSet avpSet = request.getMessage().getAvps();
// < Push-Notification-Request > ::= < Diameter Header: 309, REQ, PXY, 16777217 >
// < Session-Id >
// { Vendor-Specific-Application-Id }
AvpSet vendorSpecificApplicationId = avpSet.addGroupedAvp(Avp.VENDOR_SPECIFIC_APPLICATION_ID, 0, false, false);
// 1* [ Vendor-Id ]
vendorSpecificApplicationId.addAvp(Avp.VENDOR_ID, getApplicationId().getVendorId(), true);
// 0*1{ Auth-Application-Id }
vendorSpecificApplicationId.addAvp(Avp.AUTH_APPLICATION_ID, getApplicationId().getAuthAppId(), true);
// avpSet.addAvp(Avp.AUTH_SESSION_STATE, 1);
// { Origin-Host }
avpSet.removeAvp(Avp.ORIGIN_HOST);
avpSet.addAvp(Avp.ORIGIN_HOST, getServerURI(), true);
// { Origin-Realm }
// { Destination-Host }
avpSet.addAvp(Avp.ORIGIN_HOST, getClientURI(), true);
// { Destination-Realm }
// *[ Supported-Features ]
// { User-Identity }
AvpSet userIdentity = avpSet.addGroupedAvp(Avp.USER_IDENTITY, getApplicationId().getVendorId(), true, false);
// User-Identity ::= <AVP header: 700 10415>
// [Public-Identity]
userIdentity.addAvp(Avp.PUBLIC_IDENTITY, "tralalalal user", getApplicationId().getVendorId(), true, false, false);
// [MSISDN]
// *[AVP]
// [ Wildcarded-PSI ]
// [ Wildcarded-IMPU ]
// { User-Data }
avpSet.addAvp(Avp.USER_DATA_SH, "<xml><morexml></morexml></xml>", true);
// *[ AVP ]
// *[ Proxy-Info ]
// *[ Route-Record ]
this.serverShSession.sendPushNotificationRequest(request);
this.sentPushNotification = true;
Utils.printMessage(log, super.stack.getDictionary(), request.getMessage(), true);
}
// ------- initial, this will be triggered for first msg.
/*
* (non-Javadoc)
*
* @see org.jdiameter.api.NetworkReqListener#processRequest(org.jdiameter.api.Request)
*/
@Override
public Answer processRequest(Request request) {
int code = request.getCommandCode();
if (code != ProfileUpdateRequest.code && code != UserDataRequest.code && code != SubscribeNotificationsRequest.code) {
fail("Received Request with code not used by Sh!. Code[" + request.getCommandCode() + "]", null);
return null;
}
if (super.serverShSession == null) {
try {
super.serverShSession = this.sessionFactory.getNewAppSession(request.getSessionId(), getApplicationId(), ServerShSession.class, (Object) null);
((NetworkReqListener) this.serverShSession).processRequest(request);
}
catch (Exception e) {
fail(null, e);
}
}
else {
// do fail?
fail("Received Request in base listener, not in app specific!" + code, null);
}
return null;
}
// ------------- specific, app session listener.
/*
* (non-Javadoc)
*
* @see org.jdiameter.api.cca.ServerRoSessionListener#doOtherEvent(org.jdiameter.api.app.AppSession,
* org.jdiameter.api.app.AppRequestEvent, org.jdiameter.api.app.AppAnswerEvent)
*/
@Override
public void doOtherEvent(AppSession session, AppRequestEvent request, AppAnswerEvent answer)
throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {
fail("Received \"Other\" event, request[" + request + "], answer[" + answer + "], on session[" + session + "]", null);
}
@Override
public void doSubscribeNotificationsRequestEvent(ServerShSession session, SubscribeNotificationsRequest request)
throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {
if (this.receiveSubscribeNotifications) {
fail("Received SUBSCRIBE more than once!", null);
}
this.receiveSubscribeNotifications = true;
this.subscribeNotificationsRequest = request;
}
@Override
public void doProfileUpdateRequestEvent(ServerShSession session, ProfileUpdateRequest request)
throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {
if (this.receiveProfileUpdate) {
fail("Received UPDATE more than once!", null);
}
this.receiveProfileUpdate = true;
this.profileUpdateRequest = request;
}
@Override
public void doPushNotificationAnswerEvent(ServerShSession session, PushNotificationRequest request, PushNotificationAnswer answer) throws InternalException,
IllegalDiameterStateException, RouteException, OverloadException {
if (this.receivePushNotification) {
fail("Received NOTIFICATION more than once!", null);
}
this.receivePushNotification = true;
}
@Override
public void doUserDataRequestEvent(ServerShSession session, UserDataRequest request)
throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {
if (this.receiveUserData) {
fail("Received USER DATA more than once!", null);
}
this.receiveUserData = true;
this.userDataRequest = request;
}
@Override
public void receivedSuccessMessage(Request request, Answer answer) {
fail("Received \"SuccessMessage\" event, request[" + request + "], answer[" + answer + "]", null);
}
@Override
public void timeoutExpired(Request request) {
fail("Received \"Timoeout\" event, request[" + request + "]", null);
}
public boolean isSentSubscribeNotifications() {
return sentSubscribeNotifications;
}
public boolean isSentProfileUpdate() {
return sentProfileUpdate;
}
public boolean isSentUserData() {
return sentUserData;
}
public boolean isSentPushNotification() {
return sentPushNotification;
}
public boolean isReceiveSubscribeNotifications() {
return receiveSubscribeNotifications;
}
public boolean isReceiveProfileUpdate() {
return receiveProfileUpdate;
}
public boolean isReceiveUserData() {
return receiveUserData;
}
public boolean isReceivePushNotification() {
return receivePushNotification;
}
}