| /* |
| * Copyright 2004 The Apache Software Foundation |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| */ |
| package org.apache.directory.server.ldap.support; |
| |
| |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.Map; |
| |
| import org.apache.directory.server.ldap.ExtendedOperationHandler; |
| import org.apache.directory.server.ldap.SessionRegistry; |
| import org.apache.directory.shared.ldap.message.ExtendedRequest; |
| import org.apache.directory.shared.ldap.message.ExtendedResponse; |
| import org.apache.directory.shared.ldap.message.LdapResult; |
| import org.apache.directory.shared.ldap.message.ResultCodeEnum; |
| import org.apache.directory.shared.ldap.util.ExceptionUtils; |
| import org.apache.mina.common.IoSession; |
| import org.apache.mina.handler.demux.MessageHandler; |
| |
| |
| /** |
| * A single reply handler for {@link org.apache.directory.shared.ldap.message.ExtendedRequest}s. |
| * |
| * @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a> |
| * @version $Rev$ |
| */ |
| public class ExtendedHandler implements MessageHandler |
| { |
| private Map handlers = new HashMap(); |
| |
| |
| public ExtendedOperationHandler addHandler( ExtendedOperationHandler eoh ) |
| { |
| synchronized ( handlers ) |
| { |
| return ( ExtendedOperationHandler ) handlers.put( eoh.getOid(), eoh ); |
| } |
| } |
| |
| |
| public ExtendedOperationHandler removeHandler( String oid ) |
| { |
| synchronized ( handlers ) |
| { |
| return ( ExtendedOperationHandler ) handlers.remove( oid ); |
| } |
| } |
| |
| |
| public ExtendedOperationHandler getHandler( String oid ) |
| { |
| return ( ExtendedOperationHandler ) handlers.get( oid ); |
| } |
| |
| |
| public Map getHandlerMap() |
| { |
| return Collections.unmodifiableMap( handlers ); |
| } |
| |
| |
| public void messageReceived( IoSession session, Object request ) |
| { |
| ExtendedRequest req = ( ExtendedRequest ) request; |
| ExtendedOperationHandler handler = ( ExtendedOperationHandler ) handlers.get( req.getOid() ); |
| |
| if ( handler == null ) |
| { |
| // As long as no extended operations are implemented, send appropriate |
| // error back to the client. |
| String msg = "Unrecognized extended operation EXTENSION_OID: " + req.getOid(); |
| LdapResult result = req.getResultResponse().getLdapResult(); |
| result.setResultCode( ResultCodeEnum.PROTOCOLERROR ); |
| result.setErrorMessage( msg ); |
| session.write( req.getResultResponse() ); |
| } |
| else |
| { |
| try |
| { |
| handler.handleExtendedOperation( session, SessionRegistry.getSingleton(), req ); |
| } |
| catch ( Exception e ) |
| { |
| LdapResult result = req.getResultResponse().getLdapResult(); |
| result.setResultCode( ResultCodeEnum.UNAVAILABLE ); |
| result.setErrorMessage( "Extended operation handler for the specified EXTENSION_OID (" + req.getOid() |
| + ") has failed to process your request:\n" + ExceptionUtils.getStackTrace( e ) ); |
| ( ( ExtendedResponse ) req.getResultResponse() ).setResponse( new byte[0] ); |
| session.write( req.getResultResponse() ); |
| } |
| } |
| } |
| } |