001 /*
002 GRANITE DATA SERVICES
003 Copyright (C) 2011 GRANITE DATA SERVICES S.A.S.
004
005 This file is part of Granite Data Services.
006
007 Granite Data Services is free software; you can redistribute it and/or modify
008 it under the terms of the GNU Library General Public License as published by
009 the Free Software Foundation; either version 2 of the License, or (at your
010 option) any later version.
011
012 Granite Data Services is distributed in the hope that it will be useful, but
013 WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
014 FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public License
015 for more details.
016
017 You should have received a copy of the GNU Library General Public License
018 along with this library; if not, see <http://www.gnu.org/licenses/>.
019 */
020
021 package org.granite.messaging.amf.process;
022
023 import java.util.Date;
024
025 import org.granite.config.GraniteConfig;
026 import org.granite.context.GraniteContext;
027 import org.granite.logging.Logger;
028 import org.granite.messaging.service.ServiceException;
029 import org.granite.messaging.service.ServiceFactory;
030 import org.granite.messaging.service.ServiceInvoker;
031 import org.granite.messaging.service.security.SecurityService;
032 import org.granite.messaging.service.security.SecurityServiceException;
033 import org.granite.messaging.webapp.ServletGraniteContext;
034 import org.granite.util.UUIDUtil;
035
036 import flex.messaging.messages.AcknowledgeMessage;
037 import flex.messaging.messages.CommandMessage;
038 import flex.messaging.messages.ErrorMessage;
039 import flex.messaging.messages.Message;
040 import flex.messaging.messages.RemotingMessage;
041
042 /**
043 * @author Franck WOLFF
044 */
045 public abstract class AMF3MessageProcessor {
046
047 private static final Logger log = Logger.getLogger(AMF3MessageProcessor.class);
048
049 public static Message process(Message request) {
050 GraniteContext context = GraniteContext.getCurrentInstance();
051 AMF3MessageInterceptor interceptor = context.getGraniteConfig().getAmf3MessageInterceptor();
052
053 Message response = null;
054 try {
055 if (interceptor != null)
056 interceptor.before(request);
057
058 if (request instanceof RemotingMessage)
059 response = processRemotingMessage((RemotingMessage)request);
060 else if (request instanceof CommandMessage)
061 response = processCommandMessage((CommandMessage)request);
062 else
063 throw new IllegalArgumentException("Unknown request message type: " + request);
064 }
065 finally {
066 if (interceptor != null)
067 interceptor.after(request, response);
068 }
069
070 if (context.getSessionId() != null) {
071 response.setHeader("org.granite.sessionId", context.getSessionId());
072 if (((ServletGraniteContext)context).getSession(false) != null) {
073 long serverTime = new Date().getTime();
074 ((ServletGraniteContext)context).getSession().setAttribute(GraniteContext.SESSION_LAST_ACCESSED_TIME_KEY, serverTime);
075 response.setHeader("org.granite.time", serverTime);
076 response.setHeader("org.granite.sessionExp", ((ServletGraniteContext)context).getSession().getMaxInactiveInterval());
077 }
078 }
079
080 return response;
081 }
082
083 public static Message processCommandMessage(CommandMessage request) {
084
085 log.debug(">> Processing AMF3 request:\n%s", request);
086
087 Message response = null;
088 if (request.isSecurityOperation()) {
089 GraniteContext context = GraniteContext.getCurrentInstance();
090 GraniteConfig config = context.getGraniteConfig();
091
092 if (!config.hasSecurityService())
093 log.warn("Ignored security operation (no security settings in granite-config.xml): %s", request);
094 else {
095 SecurityService securityService = config.getSecurityService();
096 try {
097 if (request.isLoginOperation())
098 securityService.login(request.getBody(), (String)request.getHeader(Message.CREDENTIALS_CHARSET_HEADER));
099 else if (request.isLogoutOperation())
100 securityService.logout();
101 else
102 log.warn("Unknown security operation: %s", request);
103 } catch (Exception e) {
104 if (e instanceof SecurityServiceException) {
105 securityService.handleSecurityException((SecurityServiceException)e);
106 log.debug(e, "Could not process security operation: %s", request);
107 }
108 else
109 log.error(e, "Could not process security operation: %s", request);
110 response = new ErrorMessage(request, e);
111 }
112 }
113 }
114
115 if (response == null) {
116 response = new AcknowledgeMessage(request);
117 // For SDK 2.0.1_Hotfix2.
118 if (request.isSecurityOperation())
119 response.setBody("success");
120 }
121
122 // For SDK 2.0.1_Hotfix2.
123 if ("nil".equals(request.getHeader(Message.DS_ID_HEADER)))
124 response.getHeaders().put(Message.DS_ID_HEADER, UUIDUtil.randomUUID());
125
126 log.debug("<< Returning AMF3 response:\n%s", response);
127
128 return response;
129 }
130
131 public static Message processRemotingMessage(RemotingMessage request) {
132
133 log.debug(">> Processing AMF3 request:\n%s", request);
134
135 Message response = null;
136 try {
137 // Execute method on service.
138 ServiceFactory factory = ServiceFactory.getFactoryInstance(request);
139 ServiceInvoker<?> service = factory.getServiceInstance(request);
140 Object result = service.invoke(request);
141
142 response = new AcknowledgeMessage(request);
143 response.setBody(result);
144 } catch (ServiceException e) {
145 log.debug(e, "Could not process remoting message: %s", request);
146 response = new ErrorMessage(request, e);
147 }
148
149 log.debug("<< Returning AMF3 response:\n%s", response);
150
151 return response;
152 }
153 }