码迷,mamicode.com
首页 > 其他好文 > 详细

shiro登录模块源码分析

时间:2017-07-27 21:23:01      阅读:331      评论:0      收藏:0      [点我收藏+]

标签:failed   tee   get   repo   ref   sse   turned   osi   登录   

接触和使用shiro还是有好大一段时间,可惜并没有搞明白其实的原理和机制。因为工作中使用的框架都封装好了,so......并没有去研究。

原来一直猜想的是shiro可以直接连接数据库,验证用户名和密码。但是又没在实体类中找到有什么特殊的标记注解之类的。

就有点好奇了,于是趁这两天工作不是那么紧张了,就跟了下源码。

如有不对,请留言指正。蟹蟹!

 

红色部分是主要代码;
 
UsernamePasswordToken token = new UsernamePasswordToken(user.getUserName(), password, rememberMe.equals("0") ? false : true);
Subject currentUser = AppUtils.getSubject();
try {
    currentUser.login(token);
} catch (AuthenticationException e) {
    token.clear();
    model.addAttribute("user_err", "用户名或密码错误");
    return loginUrl;
}
 
将登录委托给subjectManager
public void login(AuthenticationToken token) throws AuthenticationException {
    clearRunAsIdentitiesInternal();
    Subject subject = securityManager.login(this, token);
 
    PrincipalCollection principals;
 
    String host = null;
 
    if (subject instanceof DelegatingSubject) {
        DelegatingSubject delegating = (DelegatingSubject) subject;
        //we have to do this in case there are assumed identities - we don‘t want to lose the ‘real‘ principals:
        principals = delegating.principals;
        host = delegating.host;
    } else {
        principals = subject.getPrincipals();
    }
 
    if (principals == null || principals.isEmpty()) {
        String msg = "Principals returned from securityManager.login( token ) returned a null or " +
                "empty value.  This value must be non null and populated with one or more elements.";
        throw new IllegalStateException(msg);
    }
    this.principals = principals;
    this.authenticated = true;
    if (token instanceof HostAuthenticationToken) {
        host = ((HostAuthenticationToken) token).getHost();
    }
    if (host != null) {
        this.host = host;
    }
    Session session = subject.getSession(false);
    if (session != null) {
        this.session = decorate(session);
    } else {
        this.session = null;
    }
}
 
subjectManager调authenticate模块进行身份认证
 
public Subject login(Subject subject, AuthenticationToken token) throws AuthenticationException {
    AuthenticationInfo info;
    try {
        info = authenticate(token);
    } catch (AuthenticationException ae) {
        try {
            onFailedLogin(token, ae, subject);
        } catch (Exception e) {
            if (log.isInfoEnabled()) {
                log.info("onFailedLogin method threw an " +
                        "exception.  Logging and propagating original AuthenticationException.", e);
            }
        }
        throw ae; //propagate
    }
 
    Subject loggedIn = createSubject(token, info, subject);
 
    onSuccessfulLogin(token, info, loggedIn);
 
    return loggedIn;
}
 
 
public AuthenticationInfo authenticate(AuthenticationToken token) throws AuthenticationException {
    return this.authenticator.authenticate(token);
}
 
 
public final AuthenticationInfo authenticate(AuthenticationToken token) throws AuthenticationException {
 
    if (token == null) {
        throw new IllegalArgumentException("Method argumet (authentication token) cannot be null.");
    }
 
    log.trace("Authentication attempt received for token [{}]", token);
 
    AuthenticationInfo info;
    try {
        info = doAuthenticate(token);
        if (info == null) {
            String msg = "No account information found for authentication token [" + token + "] by this " +
                    "Authenticator instance.  Please check that it is configured correctly.";
            throw new AuthenticationException(msg);
        }
    } catch (Throwable t) {......
 
 
判断Realm是否有实现类
protected AuthenticationInfo doAuthenticate(AuthenticationToken authenticationToken) throws AuthenticationException {
    assertRealmsConfigured();
    //配置文件中注入了bean
    Collection<Realm> realms = getRealms();
    if (realms.size() == 1) {
        return doSingleRealmAuthentication(realms.iterator().next(), authenticationToken);
    } else {
//这一步还不清楚
        return doMultiRealmAuthentication(realms, authenticationToken);
    }
}
 
调用重写的登录方法getAuthenticationInfo
protected AuthenticationInfo doSingleRealmAuthentication(Realm realm, AuthenticationToken token) {
    if (!realm.supports(token)) {
        String msg = "Realm [" + realm + "] does not support authentication token [" +
                token + "].  Please ensure that the appropriate Realm implementation is " +
                "configured correctly or that the realm accepts AuthenticationTokens of this type.";
        throw new UnsupportedTokenException(msg);
    }
    AuthenticationInfo info = realm.getAuthenticationInfo(token);
    if (info == null) {
        String msg = "Realm [" + realm + "] was unable to find account data for the " + 
                "submitted AuthenticationToken [" + token + "].";
        throw new UnknownAccountException(msg);
    }
    return info;
}

 

最后调用自己实现的用户名密码验证

protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken arg0) {
    UsernamePasswordToken upToken = (UsernamePasswordToken) arg0;
    String password = String.valueOf(upToken.getPassword());
    SysUserMember us = sysUserRepository.doGetAuthenticationInfoAvailable(upToken.getUsername());
    if (us == null) {
        throw new AuthenticationException("查询无此用户.");
    } else if (StringUtils.isBlank(us.getUserName()) || StringUtils.isBlank(us.getUserPassword())) {
        throw new AuthenticationException("查询无此用户.");
    } else if (!us.getUserPassword().equals(password)) {
        throw new AuthenticationException("用户名或密码错误.");
    } else {
        SecurityUtils.getSubject().getSession().setAttribute(AppConfig.SESSION_KEY, us);
        SimpleAuthenticationInfo info = new SimpleAuthenticationInfo(us.getUserName(), us.getUserPassword(), getName());
        return info;
    }
}

shiro登录模块源码分析

标签:failed   tee   get   repo   ref   sse   turned   osi   登录   

原文地址:http://www.cnblogs.com/chenyongtao/p/7246938.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!