1 17 package org.apache.geronimo.security.jaas.client; 18 19 import java.security.Principal ; 20 import java.util.HashSet ; 21 import java.util.Iterator ; 22 import java.util.Map ; 23 import java.util.Set ; 24 import javax.security.auth.Subject ; 25 import javax.security.auth.callback.CallbackHandler ; 26 import javax.security.auth.login.LoginException ; 27 import javax.security.auth.spi.LoginModule ; 28 29 import org.apache.geronimo.security.DomainPrincipal; 30 import org.apache.geronimo.security.RealmPrincipal; 31 import org.apache.geronimo.security.jaas.LoginModuleControlFlag; 32 33 34 37 public class WrappingClientLoginModuleProxy extends ClientLoginModuleProxy { 38 private final String loginDomainName; 39 private final String realmName; 40 private final Subject localSubject = new Subject (); 41 42 public WrappingClientLoginModuleProxy(LoginModuleControlFlag controlFlag, Subject subject, LoginModule source, 43 String loginDomainName, String realmName) 44 { 45 super(controlFlag, subject, source); 46 this.loginDomainName = loginDomainName; 47 this.realmName = realmName; 48 } 49 50 public void initialize(Subject subject, CallbackHandler callbackHandler, Map sharedState, Map options) { 51 super.initialize(localSubject, callbackHandler, sharedState, options); 52 } 53 54 public boolean commit() throws LoginException { 55 boolean result = super.commit(); 56 57 Set wrapped = new HashSet (); 58 for (Iterator iter = subject.getPrincipals().iterator(); iter.hasNext();) { 59 Principal principal = (Principal ) iter.next(); 60 61 wrapped.add(new DomainPrincipal(loginDomainName, principal)); 62 wrapped.add(new RealmPrincipal(realmName, loginDomainName, principal)); 63 } 64 localSubject.getPrincipals().addAll(wrapped); 65 subject.getPrincipals().addAll(localSubject.getPrincipals()); 66 67 return result; 68 } 69 70 public boolean logout() throws LoginException { 71 boolean result = super.logout(); 72 73 subject.getPrincipals().removeAll(localSubject.getPrincipals()); 74 localSubject.getPrincipals().clear(); 75 76 return result; 77 } 78 } | Popular Tags |