POACurrent.java

/*
 * Copyright (c) 1997, 2003, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * This code 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
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

package org.jboss.com.sun.corba.se.impl.oa.poa;

import java.util.EmptyStackException;

import org.jboss.com.sun.corba.se.impl.logging.POASystemException;
import org.jboss.com.sun.corba.se.spi.logging.CORBALogDomains;
import org.jboss.com.sun.corba.se.spi.oa.OAInvocationInfo;
import org.jboss.com.sun.corba.se.spi.oa.ObjectAdapter;
import org.jboss.com.sun.corba.se.spi.orb.ORB;
import org.omg.CORBA.CompletionStatus;
import org.omg.PortableServer.POA;
import org.omg.PortableServer.Servant;
import org.omg.PortableServer.CurrentPackage.NoContext;
import org.omg.PortableServer.ServantLocatorPackage.CookieHolder;

// XXX Needs to be turned into LocalObjectImpl.

public class POACurrent extends org.omg.CORBA.portable.ObjectImpl implements org.omg.PortableServer.Current
{
    private static final long serialVersionUID = 6652432719945390553L;

    private ORB orb;

    private POASystemException wrapper;

    public POACurrent(ORB orb)
    {
        this.orb = orb;
        wrapper = POASystemException.get(orb, CORBALogDomains.OA_INVOCATION);
    }

    public String[] _ids()
    {
        String[] ids = new String[1];
        ids[0] = "IDL:omg.org/PortableServer/Current:1.0";
        return ids;
    }

    // Standard OMG operations.

    public POA get_POA() throws NoContext
    {
        POA poa = (POA) (peekThrowNoContext().oa());
        throwNoContextIfNull(poa);
        return poa;
    }

    public byte[] get_object_id() throws NoContext
    {
        byte[] objectid = peekThrowNoContext().id();
        throwNoContextIfNull(objectid);
        return objectid;
    }

    // Implementation operations used by POA package.

    public ObjectAdapter getOA()
    {
        ObjectAdapter oa = peekThrowInternal().oa();
        throwInternalIfNull(oa);
        return oa;
    }

    public byte[] getObjectId()
    {
        byte[] objectid = peekThrowInternal().id();
        throwInternalIfNull(objectid);
        return objectid;
    }

    Servant getServant()
    {
        Servant servant = (Servant) (peekThrowInternal().getServantContainer());
        // If is OK for the servant to be null. This could happen if POAImpl.getServant is called but
        // POAImpl.internalGetServant throws an exception.
        return servant;
    }

    CookieHolder getCookieHolder()
    {
        CookieHolder cookieHolder = peekThrowInternal().getCookieHolder();
        throwInternalIfNull(cookieHolder);
        return cookieHolder;
    }

    // This is public so we can test the stack balance. It is not a security hole since this same info can be obtained
    // from PortableInterceptors.
    public String getOperation()
    {
        String operation = peekThrowInternal().getOperation();
        throwInternalIfNull(operation);
        return operation;
    }

    void setServant(Servant servant)
    {
        peekThrowInternal().setServant(servant);
    }

    //
    // Class utilities.
    //

    private OAInvocationInfo peekThrowNoContext() throws NoContext
    {
        OAInvocationInfo invocationInfo = null;
        try
        {
            invocationInfo = orb.peekInvocationInfo();
        }
        catch (EmptyStackException e)
        {
            throw new NoContext();
        }
        return invocationInfo;
    }

    private OAInvocationInfo peekThrowInternal()
    {
        OAInvocationInfo invocationInfo = null;
        try
        {
            invocationInfo = orb.peekInvocationInfo();
        }
        catch (EmptyStackException e)
        {
            // The completion status is maybe because this could happen after the servant has been invoked.
            throw wrapper.poacurrentUnbalancedStack(e);
        }
        return invocationInfo;
    }

    private void throwNoContextIfNull(Object o) throws NoContext
    {
        if (o == null)
        {
            throw new NoContext();
        }
    }

    private void throwInternalIfNull(Object o)
    {
        if (o == null)
        {
            throw wrapper.poacurrentNullField(CompletionStatus.COMPLETED_MAYBE);
        }
    }

    public org.omg.CORBA.Object get_reference() throws NoContext
    {
        Servant servant = get_servant();
        if (servant != null)
            return servant._this_object(orb);
        return null;
    }

    public Servant get_servant() throws NoContext
    {
        return getServant();
    }
}