001: /*
002: * JBoss, Home of Professional Open Source.
003: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
004: * as indicated by the @author tags. See the copyright.txt file in the
005: * distribution for a full listing of individual contributors.
006: *
007: * This is free software; you can redistribute it and/or modify it
008: * under the terms of the GNU Lesser General Public License as
009: * published by the Free Software Foundation; either version 2.1 of
010: * the License, or (at your option) any later version.
011: *
012: * This software is distributed in the hope that it will be useful,
013: * but WITHOUT ANY WARRANTY; without even the implied warranty of
014: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
015: * Lesser General Public License for more details.
016: *
017: * You should have received a copy of the GNU Lesser General Public
018: * License along with this software; if not, write to the Free
019: * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
020: * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
021: */
022: package org.jboss.test.security.ejb;
023:
024: import java.lang.reflect.Method;
025: import java.lang.reflect.InvocationTargetException;
026: import java.rmi.RemoteException;
027: import java.security.Principal;
028: import javax.ejb.EJBException;
029: import javax.ejb.EntityBean;
030: import javax.ejb.EntityContext;
031: import javax.ejb.CreateException;
032: import javax.naming.InitialContext;
033: import javax.naming.NamingException;
034: import javax.security.auth.Subject;
035:
036: /** A BMP entity bean that creates beans on the fly with
037: a key equal to that passed to findByPrimaryKey. Obviously
038: not a real entity bean. It is used to test Principal propagation
039: using the echo method.
040:
041: @author Scott.Stark@jboss.org
042: @version $Revision: 57211 $
043: */
044: public class EntityBeanImpl implements EntityBean {
045: org.apache.log4j.Category log = org.apache.log4j.Category
046: .getInstance(getClass());
047:
048: private String key;
049: private EntityContext context;
050:
051: public void ejbActivate() {
052: log.debug("EntityBean.ejbActivate() called");
053: }
054:
055: public void ejbPassivate() {
056: log.debug("EntityBean.ejbPassivate() called");
057: }
058:
059: public void ejbRemove() {
060: log.debug("EntityBean.ejbRemove() called");
061: }
062:
063: public void ejbLoad() {
064: log.debug("EntityBean.ejbLoad() called");
065: key = (String) context.getPrimaryKey();
066: }
067:
068: public void ejbStore() {
069: log.debug("EntityBean.ejbStore() called");
070: }
071:
072: public void setEntityContext(EntityContext context) {
073: this .context = context;
074: }
075:
076: public void unsetEntityContext() {
077: this .context = null;
078: }
079:
080: public String ejbCreate(String key) throws CreateException {
081: this .key = key;
082: return key;
083: }
084:
085: public void ejbPostCreate(String key) {
086: }
087:
088: public String echo(String arg) {
089: log.debug("EntityBean.echo, arg=" + arg);
090: Principal p = context.getCallerPrincipal();
091: boolean isInternalRole = context.isCallerInRole("InternalRole");
092: log.debug("EntityBean.echo, callerPrincipal=" + p);
093: log.debug("EntityBean.echo, isCallerInRole('InternalRole')="
094: + isInternalRole);
095: // Check the java:comp/env/security/security-domain
096: try {
097: InitialContext ctx = new InitialContext();
098: Object securityMgr = ctx
099: .lookup("java:comp/env/security/security-domain");
100: log
101: .debug("Checking java:comp/env/security/security-domain");
102: if (securityMgr == null)
103: throw new EJBException(
104: "Failed to find security mgr under: java:comp/env/security/security-domain");
105: log.debug("Found SecurityManager: " + securityMgr);
106: Subject activeSubject = (Subject) ctx
107: .lookup("java:comp/env/security/subject");
108: log.debug("ActiveSubject: " + activeSubject);
109: } catch (NamingException e) {
110: log.debug("failed", e);
111: throw new EJBException("Naming exception: "
112: + e.toString(true));
113: }
114: return p.getName();
115: }
116:
117: public String ejbFindByPrimaryKey(String key) {
118: log.debug("EntityBean.ejbFindByPrimaryKey, key=" + key);
119: return key;
120: }
121: }
|