package com.genersoft.iot.vmp.conf.security;
|
|
import com.genersoft.iot.vmp.conf.security.dto.JwtUser;
|
import com.genersoft.iot.vmp.service.IUserService;
|
import com.genersoft.iot.vmp.storager.dao.dto.User;
|
import org.jose4j.jwk.RsaJsonWebKey;
|
import org.jose4j.jwk.RsaJwkGenerator;
|
import org.jose4j.jws.AlgorithmIdentifiers;
|
import org.jose4j.jws.JsonWebSignature;
|
import org.jose4j.jwt.JwtClaims;
|
import org.jose4j.jwt.NumericDate;
|
import org.jose4j.jwt.consumer.ErrorCodes;
|
import org.jose4j.jwt.consumer.InvalidJwtException;
|
import org.jose4j.jwt.consumer.JwtConsumer;
|
import org.jose4j.jwt.consumer.JwtConsumerBuilder;
|
import org.jose4j.lang.JoseException;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.InitializingBean;
|
import org.springframework.stereotype.Component;
|
|
import javax.annotation.Resource;
|
import java.time.LocalDateTime;
|
import java.time.ZoneOffset;
|
|
@Component
|
public class JwtUtils implements InitializingBean {
|
|
private static final Logger logger = LoggerFactory.getLogger(JwtUtils.class);
|
|
private static final String HEADER = "access-token";
|
|
private static final String AUDIENCE = "Audience";
|
|
private static final String keyId = "3e79646c4dbc408383a9eed09f2b85ae";
|
|
/**
|
* token过期时间(分钟)
|
*/
|
public static final long expirationTime = 30 * 24 * 60;
|
|
private static RsaJsonWebKey rsaJsonWebKey;
|
|
private static IUserService userService;
|
|
@Resource
|
public void setUserService(IUserService userService) {
|
JwtUtils.userService = userService;
|
}
|
|
@Override
|
public void afterPropertiesSet() {
|
try {
|
rsaJsonWebKey = generateRsaJsonWebKey();
|
} catch (JoseException e) {
|
logger.error("生成RsaJsonWebKey报错。", e);
|
}
|
}
|
|
/**
|
* 创建密钥对
|
* @throws JoseException JoseException
|
*/
|
private RsaJsonWebKey generateRsaJsonWebKey() throws JoseException {
|
// 生成一个RSA密钥对，该密钥对将用于JWT的签名和验证，包装在JWK中
|
RsaJsonWebKey rsaJsonWebKey = RsaJwkGenerator.generateJwk(2048);
|
// 给JWK一个密钥ID
|
rsaJsonWebKey.setKeyId(keyId);
|
return rsaJsonWebKey;
|
}
|
|
public static String createToken(String username) {
|
try {
|
/*
|
* “iss” (issuer) 发行人
|
* “sub” (subject) 主题
|
* “aud” (audience) 接收方 用户
|
* “exp” (expiration time) 到期时间
|
* “nbf” (not before) 在此之前不可用
|
* “iat” (issued at) jwt的签发时间
|
*/
|
JwtClaims claims = new JwtClaims();
|
claims.setGeneratedJwtId();
|
claims.setIssuedAtToNow();
|
// 令牌将过期的时间 分钟
|
claims.setExpirationTimeMinutesInTheFuture(expirationTime);
|
claims.setNotBeforeMinutesInThePast(0);
|
claims.setSubject("login");
|
claims.setAudience(AUDIENCE);
|
//添加自定义参数,必须是字符串类型
|
claims.setClaim("userName", username);
|
|
//jws
|
JsonWebSignature jws = new JsonWebSignature();
|
//签名算法RS256
|
jws.setAlgorithmHeaderValue(AlgorithmIdentifiers.RSA_USING_SHA256);
|
jws.setKeyIdHeaderValue(keyId);
|
jws.setPayload(claims.toJson());
|
|
jws.setKey(rsaJsonWebKey.getPrivateKey());
|
|
//get token
|
return jws.getCompactSerialization();
|
} catch (JoseException e) {
|
logger.error("[Token生成失败]： {}", e.getMessage());
|
}
|
|
return null;
|
}
|
|
public static String getHeader() {
|
return HEADER;
|
}
|
|
public static JwtUser verifyToken(String token) {
|
|
JwtUser jwtUser = new JwtUser();
|
|
try {
|
JwtConsumer consumer = new JwtConsumerBuilder()
|
.setRequireExpirationTime()
|
.setMaxFutureValidityInMinutes(5256000)
|
.setAllowedClockSkewInSeconds(30)
|
.setRequireSubject()
|
//.setExpectedIssuer("")
|
.setExpectedAudience(AUDIENCE)
|
.setVerificationKey(rsaJsonWebKey.getPublicKey())
|
.build();
|
|
JwtClaims claims = consumer.processToClaims(token);
|
NumericDate expirationTime = claims.getExpirationTime();
|
// 判断是否即将过期, 默认剩余时间小于5分钟未即将过期
|
// 剩余时间 （秒）
|
long timeRemaining = LocalDateTime.now().toEpochSecond(ZoneOffset.ofHours(8)) - expirationTime.getValue();
|
if (timeRemaining < 5 * 60) {
|
jwtUser.setStatus(JwtUser.TokenStatus.EXPIRING_SOON);
|
} else {
|
jwtUser.setStatus(JwtUser.TokenStatus.NORMAL);
|
}
|
|
String username = (String) claims.getClaimValue("userName");
|
User user = userService.getUserByUsername(username);
|
|
jwtUser.setUserName(username);
|
jwtUser.setPassword(user.getPassword());
|
jwtUser.setRoleId(user.getRole().getId());
|
|
return jwtUser;
|
} catch (InvalidJwtException e) {
|
if (e.hasErrorCode(ErrorCodes.EXPIRED)) {
|
jwtUser.setStatus(JwtUser.TokenStatus.EXPIRED);
|
} else {
|
jwtUser.setStatus(JwtUser.TokenStatus.EXCEPTION);
|
}
|
return jwtUser;
|
} catch (Exception e) {
|
logger.error("[Token解析失败]： {}", e.getMessage());
|
jwtUser.setStatus(JwtUser.TokenStatus.EXPIRED);
|
return jwtUser;
|
}
|
}
|
}
|
