SecurityContext.java

package org.objectweb.security.context;

import java.io.Serializable;
import java.security.Principal;
import java.util.ArrayList;
import java.util.Iterator;

import org.objectweb.jonas.security.jacc.JPolicyUserRoleMapping;

public class SecurityContext implements Serializable {

    private String principalName = null;

    private final String[] roles;

    private ArrayList runAsRoleStack = null;

    private ArrayList runAsPrincipalStack = null;

    private ArrayList runAsPrincipalRolesStack = null;

    public SecurityContext() {
        this("ANONYMOUS");
    }

    public SecurityContext(String principalName, String[] roles) {
        this.principalName = principalName;
        this.roles = roles;
    }

    public SecurityContext(String principalName, ArrayList arrayRoles) {
        this(principalName, arrayRoles, null, null, null);
    }

    protected SecurityContext(String principalName, ArrayList arrayRoles, ArrayList arrayRunas, ArrayList arrayRunasPrincipal, ArrayList arrayRunasPrincipalRoles) {
        this.principalName = principalName;
        String[] overridedRoles = JPolicyUserRoleMapping.getGlobalMappingForPrincipal(principalName);
        if (overridedRoles != null) {
            this.roles = overridedRoles;
        } else {
            if (arrayRoles != null) {
                //Convert list into array
                String[] r = new String[arrayRoles.size()];
                r = (String[]) arrayRoles.toArray(r);
                this.roles = r;
            } else {
                this.roles = null;
            }
        }

        this.runAsRoleStack = arrayRunas;
        this.runAsPrincipalStack = arrayRunasPrincipal;
        this.runAsPrincipalRolesStack = arrayRunasPrincipalRoles;
    }

    public SecurityContext(String principalName) {
        this.principalName = principalName;
        this.roles = new String[] {"JOnAS"};
    }

    public Principal getCallerPrincipal(boolean inRunAs) {
        return new InternalPrincipal(inRunAs);
    }

    public String[] getCallerPrincipalRoles(boolean inRunAs) {
        String[] runAsRoles = null;
        if (inRunAs) {
            runAsRoles = peekLastRunAsPrincipalRoles();
        } else {
            runAsRoles = peekRunAsPrincipalRoles();
        }
        if (runAsRoles != null) {
            return runAsRoles;
        } else {
            return roles;
        }
    }

    public void pushRunAsRole(String role) {
        getRunAsRoleStack().add(role);
    }

    public void pushRunAsPrincipal(String principalName, String[] roles) {
        getRunAsPrincipalStack().add(principalName);
        getRunAsPrincipalRolesStack().add(roles);
    }

    public void popRunAs() {
        if (!getRunAsRoleStack().isEmpty()) {
            getRunAsRoleStack().remove(getRunAsRoleStack().size() - 1);
        }
        if (!getRunAsPrincipalStack().isEmpty()) {
            getRunAsPrincipalStack().remove(getRunAsPrincipalStack().size() - 1);
        }
        if (!getRunAsPrincipalRolesStack().isEmpty()) {
            getRunAsPrincipalRolesStack().remove(getRunAsPrincipalRolesStack().size() - 1);
        }
    }

    public String peekRunAsRole() {
        if (getRunAsRoleStack().isEmpty()) {
            return null;
        } else {
            return (String) getRunAsRoleStack().get(getRunAsRoleStack().size() - 1);
        }
    }

    public String peekRunAsPrincipal() {
        if (getRunAsPrincipalStack().isEmpty()) {
            return null;
        } else {
            return (String) getRunAsPrincipalStack().get(getRunAsPrincipalStack().size() - 1);
        }
    }

    public String peekLastRunAsPrincipal() {
        if (getRunAsPrincipalStack().size() < 2) {
            return null;
        } else {
            return (String) getRunAsPrincipalStack().get(getRunAsPrincipalStack().size() - 2);
        }
    }

    public String[] peekRunAsPrincipalRoles() {
        if (getRunAsPrincipalRolesStack().isEmpty()) {
            return null;
        } else {
            return (String[]) getRunAsPrincipalRolesStack().get(getRunAsPrincipalRolesStack().size() - 1);
        }
    }

    public String[] peekLastRunAsPrincipalRoles() {
        if (getRunAsPrincipalRolesStack().size() < 2) {
            return null;
        } else {
            return (String[]) getRunAsPrincipalRolesStack().get(getRunAsPrincipalRolesStack().size() - 2);
        }
    }

    public ArrayList getRunAsRoleStack() {
        if (runAsRoleStack == null) {
            runAsRoleStack = new ArrayList();
        }
        return runAsRoleStack;
    }

    public ArrayList getRunAsPrincipalRolesStack() {
        if (runAsPrincipalRolesStack == null) {
            runAsPrincipalRolesStack = new ArrayList();
        }
        return runAsPrincipalRolesStack;
    }

    public ArrayList getRunAsPrincipalStack() {
        if (runAsPrincipalStack == null) {
            runAsPrincipalStack = new ArrayList();
        }
        return runAsPrincipalStack;
    }

    public String toString() {
        String txt = "principal : name = " + principalName + "\n";
        if (roles != null) {
            for (int i = 0; i < roles.length; i++) {
                txt += "role[" + i + "] = " + roles[i] + "\n";
            }
        }
        if (runAsRoleStack != null) {
            Iterator iRunas = runAsRoleStack.iterator();
            int i = 0;
            while (iRunas.hasNext()) {
                txt += "runas[" + i + "] = " + ((String) iRunas.next()) + "\n";
            }
        }
        return txt;
    }

    protected String getPrincipalName(boolean runningRunAs) {
        String principal = null;
        if (runningRunAs) {
            principal = peekLastRunAsPrincipal();
        } else {
            principal = peekRunAsPrincipal();
        }
        if (principal != null) {
            return principal;
        } else {
            return principalName;
        }
    }

    public String getPrincipalName() {
        return principalName;
    }

    protected String[] getRoles() {
        return roles;
    }

    class InternalPrincipal implements Principal {

        private boolean inRunAs = false;

        public InternalPrincipal(boolean inRunAs) {
            super();
            this.inRunAs = inRunAs;

        }

        public String getName() {
            return getPrincipalName(inRunAs);
        }

        public boolean equals(Object o) {
            if (o instanceof Principal) {
                return getPrincipalName(inRunAs).equals(((Principal) o).getName());
            }
            return false;
        }

        public int hashCode() {
            return getPrincipalName(inRunAs).hashCode();
        }

        public String toString() {
            return "name = " + getPrincipalName(inRunAs);
        }
    };

}

---