首页 > 代码库 > JAAS
JAAS
package apache.shrio;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import javax.security.auth.callback.Callback;
import javax.security.auth.callback.CallbackHandler;
import javax.security.auth.callback.NameCallback;
import javax.security.auth.callback.PasswordCallback;
import javax.security.auth.callback.UnsupportedCallbackException;
public class SimpleCallbackHandler implements CallbackHandler {
@Override
public void handle(Callback[] callbacks) throws IOException,
UnsupportedCallbackException {
// TODO Auto-generated method stub
for(Callback cb:callbacks) {
if(cb instanceof NameCallback) {
NameCallback nc = (NameCallback)cb;
System.out.println(nc.getPrompt());
System.out.flush();
nc.setName(new BufferedReader(new InputStreamReader(System.in)).readLine());
} else if(cb instanceof PasswordCallback) {
PasswordCallback pc = (PasswordCallback)cb;
System.out.println(pc.getPrompt());
System.out.flush();
pc.setPassword(new BufferedReader(new InputStreamReader(System.in)).readLine().toCharArray());
}
}
}
}
package apache.shrio;
import java.io.IOException;
import java.security.Principal;
import java.util.Map;
import java.util.Set;
import javax.security.auth.Subject;
import javax.security.auth.callback.Callback;
import javax.security.auth.callback.CallbackHandler;
import javax.security.auth.callback.NameCallback;
import javax.security.auth.callback.PasswordCallback;
import javax.security.auth.callback.UnsupportedCallbackException;
import javax.security.auth.login.LoginException;
import javax.security.auth.spi.LoginModule;
public class SimpleLoginModule implements LoginModule {
private String userName;
private char[] password;
private Subject subject;
private DemoPrincipal principal;
private CallbackHandler callbackHandler;
private Map sharedState;
private Map options;
private String debug;
@Override
public void initialize(Subject subject, CallbackHandler callbackHandler,
Map<String, ?> sharedState, Map<String, ?> options) {
// TODO Auto-generated method stub
this.subject = subject;
this.callbackHandler = callbackHandler;
this.sharedState = sharedState;
this.options = options;
this.debug = (String)options.get("debug");
}
@Override
public boolean login() throws LoginException {
// TODO Auto-generated method stub
Callback[] cbs = new Callback[2];
cbs[0] = new NameCallback("用户名:");
cbs[1] = new PasswordCallback("密码:",false);
try {
callbackHandler.handle(cbs);
userName = ((NameCallback)cbs[0]).getName();
password = ((PasswordCallback)cbs[1]).getPassword();
if(debug.equals("true")) {
System.out.println("你输入的用户名为:"+userName);
System.out.println("你输入的密码为:"+new String(password));
}
if(userName.equals("test") && new String(password).equals("123123")) {
System.out.println("验证成功");
principal = new DemoPrincipal(userName);
return true;
} else {
System.out.println("验证失败");
userName = null;
password = null;
return false;
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (UnsupportedCallbackException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return false;
}
@Override
public boolean commit() throws LoginException {
// TODO Auto-generated method stub
System.out.println("commit()");
Set<Principal> principals = subject.getPrincipals();
if(!principals.contains(principal)) {
subject.getPrincipals().add(principal);
}
return true;
}
@Override
public boolean abort() throws LoginException {
// TODO Auto-generated method stub
System.out.println("abort()");
return false;
}
@Override
public boolean logout() throws LoginException {
// TODO Auto-generated method stub
System.out.println("logout()");
return false;
}
/*public String getUserName() {
return userName;
}
public void setUserName(String userName) {
this.userName = userName;
}
public char[] getPassword() {
return password;
}
public void setPassword(char[] password) {
this.password = password;
}
public Subject getSubject() {
return subject;
}
public void setSubject(Subject subject) {
this.subject = subject;
}
public CallbackHandler getCallbackHandler() {
return callbackHandler;
}
public void setCallbackHandler(CallbackHandler callbackHandler) {
this.callbackHandler = callbackHandler;
}
public Map getSharedState() {
return sharedState;
}
public void setSharedState(Map sharedState) {
this.sharedState = sharedState;
}
public Map getOptions() {
return options;
}
public void setOptions(Map options) {
this.options = options;
}
public String getDebug() {
return debug;
}
public void setDebug(String debug) {
this.debug = debug;
}
*/
}
package apache.shrio;
import java.security.Principal;
public class DemoPrincipal implements Principal {
private String name;
public DemoPrincipal(String name) {
this.name = name;
}
@Override
public String getName() {
// TODO Auto-generated method stub
return name;
}
}
package apache.shrio;
import java.security.Principal;
import javax.security.auth.Subject;
import javax.security.auth.login.LoginContext;
import javax.security.auth.login.LoginException;
public class TestJAAS {
/**
* @param args
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
System.setProperty("java.security.auth.login.config", "F:\\workspace\\shrio\\src\\main\\java\\jaas.config");
try {
LoginContext lc = new LoginContext("test",new SimpleCallbackHandler());
lc.login();
Subject subject = lc.getSubject();
for(Principal p:subject.getPrincipals()) {
System.out.println(p.getName());
}
System.out.println("登录成功");
} catch (LoginException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
test {
apache.shrio.SimpleLoginModule required debug=true;
};
JAAS