HADOOP-18851. Performance improvement for DelegationTokenSecretManager (#6803)
This commit is contained in:
parent
a97e3022de
commit
f8dce6c501
@ -33,6 +33,7 @@
|
|||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
import java.util.concurrent.locks.ReentrantReadWriteLock;
|
||||||
|
|
||||||
import javax.crypto.SecretKey;
|
import javax.crypto.SecretKey;
|
||||||
|
|
||||||
@ -122,10 +123,10 @@ private String formatTokenId(TokenIdent id) {
|
|||||||
*/
|
*/
|
||||||
private DelegationKey currentKey;
|
private DelegationKey currentKey;
|
||||||
|
|
||||||
private long keyUpdateInterval;
|
private final long keyUpdateInterval;
|
||||||
private long tokenMaxLifetime;
|
private final long tokenMaxLifetime;
|
||||||
private long tokenRemoverScanInterval;
|
private final long tokenRemoverScanInterval;
|
||||||
private long tokenRenewInterval;
|
private final long tokenRenewInterval;
|
||||||
/**
|
/**
|
||||||
* Whether to store a token's tracking ID in its TokenInformation.
|
* Whether to store a token's tracking ID in its TokenInformation.
|
||||||
* Can be overridden by a subclass.
|
* Can be overridden by a subclass.
|
||||||
@ -140,6 +141,8 @@ private String formatTokenId(TokenIdent id) {
|
|||||||
*/
|
*/
|
||||||
protected Object noInterruptsLock = new Object();
|
protected Object noInterruptsLock = new Object();
|
||||||
|
|
||||||
|
private final ReentrantReadWriteLock apiLock = new ReentrantReadWriteLock(true);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Create a secret manager
|
* Create a secret manager
|
||||||
* @param delegationKeyUpdateInterval the number of milliseconds for rolling
|
* @param delegationKeyUpdateInterval the number of milliseconds for rolling
|
||||||
@ -169,21 +172,29 @@ public AbstractDelegationTokenSecretManager(long delegationKeyUpdateInterval,
|
|||||||
public void startThreads() throws IOException {
|
public void startThreads() throws IOException {
|
||||||
Preconditions.checkState(!running);
|
Preconditions.checkState(!running);
|
||||||
updateCurrentKey();
|
updateCurrentKey();
|
||||||
synchronized (this) {
|
this.apiLock.writeLock().lock();
|
||||||
|
try {
|
||||||
running = true;
|
running = true;
|
||||||
tokenRemoverThread = new Daemon(new ExpiredTokenRemover());
|
tokenRemoverThread = new Daemon(new ExpiredTokenRemover());
|
||||||
tokenRemoverThread.start();
|
tokenRemoverThread.start();
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Reset all data structures and mutable state.
|
* Reset all data structures and mutable state.
|
||||||
*/
|
*/
|
||||||
public synchronized void reset() {
|
public void reset() {
|
||||||
setCurrentKeyId(0);
|
this.apiLock.writeLock().lock();
|
||||||
allKeys.clear();
|
try {
|
||||||
setDelegationTokenSeqNum(0);
|
setCurrentKeyId(0);
|
||||||
currentTokens.clear();
|
allKeys.clear();
|
||||||
|
setDelegationTokenSeqNum(0);
|
||||||
|
currentTokens.clear();
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -210,17 +221,27 @@ protected long getTokenRenewInterval() {
|
|||||||
* @param key delegation key.
|
* @param key delegation key.
|
||||||
* @throws IOException raised on errors performing I/O.
|
* @throws IOException raised on errors performing I/O.
|
||||||
*/
|
*/
|
||||||
public synchronized void addKey(DelegationKey key) throws IOException {
|
public void addKey(DelegationKey key) throws IOException {
|
||||||
if (running) // a safety check
|
if (running) // a safety check
|
||||||
throw new IOException("Can't add delegation key to a running SecretManager.");
|
throw new IOException("Can't add delegation key to a running SecretManager.");
|
||||||
if (key.getKeyId() > getCurrentKeyId()) {
|
this.apiLock.writeLock().lock();
|
||||||
setCurrentKeyId(key.getKeyId());
|
try {
|
||||||
|
if (key.getKeyId() > getCurrentKeyId()) {
|
||||||
|
setCurrentKeyId(key.getKeyId());
|
||||||
|
}
|
||||||
|
allKeys.put(key.getKeyId(), key);
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
allKeys.put(key.getKeyId(), key);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public synchronized DelegationKey[] getAllKeys() {
|
public DelegationKey[] getAllKeys() {
|
||||||
return allKeys.values().toArray(new DelegationKey[0]);
|
this.apiLock.readLock().lock();
|
||||||
|
try {
|
||||||
|
return allKeys.values().toArray(new DelegationKey[0]);
|
||||||
|
} finally {
|
||||||
|
this.apiLock.readLock().unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// HDFS
|
// HDFS
|
||||||
@ -263,8 +284,13 @@ protected void updateStoredToken(TokenIdent ident, long renewDate) throws IOExce
|
|||||||
*
|
*
|
||||||
* @return currentId.
|
* @return currentId.
|
||||||
*/
|
*/
|
||||||
protected synchronized int getCurrentKeyId() {
|
protected int getCurrentKeyId() {
|
||||||
return currentId;
|
this.apiLock.readLock().lock();
|
||||||
|
try {
|
||||||
|
return currentId;
|
||||||
|
} finally {
|
||||||
|
this.apiLock.readLock().unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -273,8 +299,13 @@ protected synchronized int getCurrentKeyId() {
|
|||||||
*
|
*
|
||||||
* @return currentId.
|
* @return currentId.
|
||||||
*/
|
*/
|
||||||
protected synchronized int incrementCurrentKeyId() {
|
protected int incrementCurrentKeyId() {
|
||||||
return ++currentId;
|
this.apiLock.writeLock().lock();
|
||||||
|
try {
|
||||||
|
return ++currentId;
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -283,8 +314,13 @@ protected synchronized int incrementCurrentKeyId() {
|
|||||||
*
|
*
|
||||||
* @param keyId keyId.
|
* @param keyId keyId.
|
||||||
*/
|
*/
|
||||||
protected synchronized void setCurrentKeyId(int keyId) {
|
protected void setCurrentKeyId(int keyId) {
|
||||||
currentId = keyId;
|
this.apiLock.writeLock().lock();
|
||||||
|
try {
|
||||||
|
currentId = keyId;
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -293,8 +329,13 @@ protected synchronized void setCurrentKeyId(int keyId) {
|
|||||||
*
|
*
|
||||||
* @return delegationTokenSequenceNumber.
|
* @return delegationTokenSequenceNumber.
|
||||||
*/
|
*/
|
||||||
protected synchronized int getDelegationTokenSeqNum() {
|
protected int getDelegationTokenSeqNum() {
|
||||||
return delegationTokenSequenceNumber;
|
this.apiLock.readLock().lock();
|
||||||
|
try {
|
||||||
|
return delegationTokenSequenceNumber;
|
||||||
|
} finally {
|
||||||
|
this.apiLock.readLock().unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -303,8 +344,13 @@ protected synchronized int getDelegationTokenSeqNum() {
|
|||||||
*
|
*
|
||||||
* @return delegationTokenSequenceNumber.
|
* @return delegationTokenSequenceNumber.
|
||||||
*/
|
*/
|
||||||
protected synchronized int incrementDelegationTokenSeqNum() {
|
protected int incrementDelegationTokenSeqNum() {
|
||||||
return ++delegationTokenSequenceNumber;
|
this.apiLock.writeLock().lock();
|
||||||
|
try {
|
||||||
|
return ++delegationTokenSequenceNumber;
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -313,8 +359,13 @@ protected synchronized int incrementDelegationTokenSeqNum() {
|
|||||||
*
|
*
|
||||||
* @param seqNum seqNum.
|
* @param seqNum seqNum.
|
||||||
*/
|
*/
|
||||||
protected synchronized void setDelegationTokenSeqNum(int seqNum) {
|
protected void setDelegationTokenSeqNum(int seqNum) {
|
||||||
delegationTokenSequenceNumber = seqNum;
|
this.apiLock.writeLock().lock();
|
||||||
|
try {
|
||||||
|
delegationTokenSequenceNumber = seqNum;
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -401,34 +452,39 @@ protected void updateToken(TokenIdent ident,
|
|||||||
* @param renewDate token renew time
|
* @param renewDate token renew time
|
||||||
* @throws IOException raised on errors performing I/O.
|
* @throws IOException raised on errors performing I/O.
|
||||||
*/
|
*/
|
||||||
public synchronized void addPersistedDelegationToken(
|
public void addPersistedDelegationToken(
|
||||||
TokenIdent identifier, long renewDate) throws IOException {
|
TokenIdent identifier, long renewDate) throws IOException {
|
||||||
if (running) {
|
if (running) {
|
||||||
// a safety check
|
// a safety check
|
||||||
throw new IOException(
|
throw new IOException(
|
||||||
"Can't add persisted delegation token to a running SecretManager.");
|
"Can't add persisted delegation token to a running SecretManager.");
|
||||||
}
|
}
|
||||||
int keyId = identifier.getMasterKeyId();
|
this.apiLock.writeLock().lock();
|
||||||
DelegationKey dKey = allKeys.get(keyId);
|
try {
|
||||||
byte[] password = null;
|
int keyId = identifier.getMasterKeyId();
|
||||||
if (dKey == null) {
|
DelegationKey dKey = allKeys.get(keyId);
|
||||||
LOG.warn("No KEY found for persisted identifier, expiring stored token "
|
byte[] password = null;
|
||||||
+ formatTokenId(identifier));
|
if (dKey == null) {
|
||||||
// make sure the token is expired
|
LOG.warn("No KEY found for persisted identifier, expiring stored token " + formatTokenId(
|
||||||
renewDate = 0L;
|
identifier));
|
||||||
} else {
|
// make sure the token is expired
|
||||||
password = createPassword(identifier.getBytes(), dKey.getKey());
|
renewDate = 0L;
|
||||||
}
|
} else {
|
||||||
if (identifier.getSequenceNumber() > getDelegationTokenSeqNum()) {
|
password = createPassword(identifier.getBytes(), dKey.getKey());
|
||||||
setDelegationTokenSeqNum(identifier.getSequenceNumber());
|
}
|
||||||
}
|
if (identifier.getSequenceNumber() > getDelegationTokenSeqNum()) {
|
||||||
if (getTokenInfo(identifier) == null) {
|
setDelegationTokenSeqNum(identifier.getSequenceNumber());
|
||||||
currentTokens.put(identifier, new DelegationTokenInformation(renewDate,
|
}
|
||||||
password, getTrackingIdIfEnabled(identifier)));
|
if (getTokenInfo(identifier) == null) {
|
||||||
addTokenForOwnerStats(identifier);
|
currentTokens.put(identifier, new DelegationTokenInformation(renewDate, password,
|
||||||
} else {
|
getTrackingIdIfEnabled(identifier)));
|
||||||
throw new IOException("Same delegation token being added twice: "
|
addTokenForOwnerStats(identifier);
|
||||||
+ formatTokenId(identifier));
|
} else {
|
||||||
|
throw new IOException("Same delegation token being added twice: " +
|
||||||
|
formatTokenId(identifier));
|
||||||
|
}
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -441,17 +497,18 @@ private void updateCurrentKey() throws IOException {
|
|||||||
LOG.info("Updating the current master key for generating delegation tokens");
|
LOG.info("Updating the current master key for generating delegation tokens");
|
||||||
/* Create a new currentKey with an estimated expiry date. */
|
/* Create a new currentKey with an estimated expiry date. */
|
||||||
int newCurrentId;
|
int newCurrentId;
|
||||||
synchronized (this) {
|
newCurrentId = incrementCurrentKeyId();
|
||||||
newCurrentId = incrementCurrentKeyId();
|
|
||||||
}
|
|
||||||
DelegationKey newKey = new DelegationKey(newCurrentId, System
|
DelegationKey newKey = new DelegationKey(newCurrentId, System
|
||||||
.currentTimeMillis()
|
.currentTimeMillis()
|
||||||
+ keyUpdateInterval + tokenMaxLifetime, generateSecret());
|
+ keyUpdateInterval + tokenMaxLifetime, generateSecret());
|
||||||
//Log must be invoked outside the lock on 'this'
|
//Log must be invoked outside the lock on 'this'
|
||||||
logUpdateMasterKey(newKey);
|
logUpdateMasterKey(newKey);
|
||||||
synchronized (this) {
|
this.apiLock.writeLock().lock();
|
||||||
|
try {
|
||||||
currentKey = newKey;
|
currentKey = newKey;
|
||||||
storeDelegationKey(currentKey);
|
storeDelegationKey(currentKey);
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -461,7 +518,8 @@ private void updateCurrentKey() throws IOException {
|
|||||||
* @throws IOException raised on errors performing I/O.
|
* @throws IOException raised on errors performing I/O.
|
||||||
*/
|
*/
|
||||||
protected void rollMasterKey() throws IOException {
|
protected void rollMasterKey() throws IOException {
|
||||||
synchronized (this) {
|
this.apiLock.writeLock().lock();
|
||||||
|
try {
|
||||||
removeExpiredKeys();
|
removeExpiredKeys();
|
||||||
/* set final expiry date for retiring currentKey */
|
/* set final expiry date for retiring currentKey */
|
||||||
currentKey.setExpiryDate(Time.now() + tokenMaxLifetime);
|
currentKey.setExpiryDate(Time.now() + tokenMaxLifetime);
|
||||||
@ -471,46 +529,59 @@ protected void rollMasterKey() throws IOException {
|
|||||||
* allKeys just in case.
|
* allKeys just in case.
|
||||||
*/
|
*/
|
||||||
updateDelegationKey(currentKey);
|
updateDelegationKey(currentKey);
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
updateCurrentKey();
|
updateCurrentKey();
|
||||||
}
|
}
|
||||||
|
|
||||||
private synchronized void removeExpiredKeys() {
|
private void removeExpiredKeys() {
|
||||||
long now = Time.now();
|
this.apiLock.writeLock().lock();
|
||||||
for (Iterator<Map.Entry<Integer, DelegationKey>> it = allKeys.entrySet()
|
try {
|
||||||
.iterator(); it.hasNext();) {
|
long now = Time.now();
|
||||||
Map.Entry<Integer, DelegationKey> e = it.next();
|
for (Iterator<Map.Entry<Integer, DelegationKey>> it =
|
||||||
if (e.getValue().getExpiryDate() < now) {
|
allKeys.entrySet().iterator(); it.hasNext();) {
|
||||||
it.remove();
|
Map.Entry<Integer, DelegationKey> e = it.next();
|
||||||
// ensure the tokens generated by this current key can be recovered
|
if (e.getValue().getExpiryDate() < now) {
|
||||||
// with this current key after this current key is rolled
|
it.remove();
|
||||||
if(!e.getValue().equals(currentKey))
|
// ensure the tokens generated by this current key can be recovered
|
||||||
removeStoredMasterKey(e.getValue());
|
// with this current key after this current key is rolled
|
||||||
|
if (!e.getValue().equals(currentKey)) {
|
||||||
|
removeStoredMasterKey(e.getValue());
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected synchronized byte[] createPassword(TokenIdent identifier) {
|
protected byte[] createPassword(TokenIdent identifier) {
|
||||||
int sequenceNum;
|
this.apiLock.writeLock().lock();
|
||||||
long now = Time.now();
|
|
||||||
sequenceNum = incrementDelegationTokenSeqNum();
|
|
||||||
identifier.setIssueDate(now);
|
|
||||||
identifier.setMaxDate(now + tokenMaxLifetime);
|
|
||||||
identifier.setMasterKeyId(currentKey.getKeyId());
|
|
||||||
identifier.setSequenceNumber(sequenceNum);
|
|
||||||
LOG.info("Creating password for identifier: " + formatTokenId(identifier)
|
|
||||||
+ ", currentKey: " + currentKey.getKeyId());
|
|
||||||
byte[] password = createPassword(identifier.getBytes(), currentKey.getKey());
|
|
||||||
DelegationTokenInformation tokenInfo = new DelegationTokenInformation(now
|
|
||||||
+ tokenRenewInterval, password, getTrackingIdIfEnabled(identifier));
|
|
||||||
try {
|
try {
|
||||||
METRICS.trackStoreToken(() -> storeToken(identifier, tokenInfo));
|
int sequenceNum;
|
||||||
} catch (IOException ioe) {
|
long now = Time.now();
|
||||||
LOG.error("Could not store token " + formatTokenId(identifier) + "!!",
|
sequenceNum = incrementDelegationTokenSeqNum();
|
||||||
ioe);
|
identifier.setIssueDate(now);
|
||||||
|
identifier.setMaxDate(now + tokenMaxLifetime);
|
||||||
|
identifier.setMasterKeyId(currentKey.getKeyId());
|
||||||
|
identifier.setSequenceNumber(sequenceNum);
|
||||||
|
LOG.info("Creating password for identifier: " + formatTokenId(identifier) +
|
||||||
|
", currentKey: " + currentKey.getKeyId());
|
||||||
|
byte[] password = createPassword(identifier.getBytes(), currentKey.getKey());
|
||||||
|
DelegationTokenInformation tokenInfo =
|
||||||
|
new DelegationTokenInformation(now + tokenRenewInterval, password,
|
||||||
|
getTrackingIdIfEnabled(identifier));
|
||||||
|
try {
|
||||||
|
METRICS.trackStoreToken(() -> storeToken(identifier, tokenInfo));
|
||||||
|
} catch (IOException ioe) {
|
||||||
|
LOG.error("Could not store token " + formatTokenId(identifier) + "!!", ioe);
|
||||||
|
}
|
||||||
|
return password;
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
return password;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -526,7 +597,6 @@ protected synchronized byte[] createPassword(TokenIdent identifier) {
|
|||||||
*/
|
*/
|
||||||
protected DelegationTokenInformation checkToken(TokenIdent identifier)
|
protected DelegationTokenInformation checkToken(TokenIdent identifier)
|
||||||
throws InvalidToken {
|
throws InvalidToken {
|
||||||
assert Thread.holdsLock(this);
|
|
||||||
DelegationTokenInformation info = getTokenInfo(identifier);
|
DelegationTokenInformation info = getTokenInfo(identifier);
|
||||||
String err;
|
String err;
|
||||||
if (info == null) {
|
if (info == null) {
|
||||||
@ -546,9 +616,14 @@ protected DelegationTokenInformation checkToken(TokenIdent identifier)
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public synchronized byte[] retrievePassword(TokenIdent identifier)
|
public byte[] retrievePassword(TokenIdent identifier)
|
||||||
throws InvalidToken {
|
throws InvalidToken {
|
||||||
return checkToken(identifier).getPassword();
|
this.apiLock.readLock().lock();
|
||||||
|
try {
|
||||||
|
return checkToken(identifier).getPassword();
|
||||||
|
} finally {
|
||||||
|
this.apiLock.readLock().unlock();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
protected String getTrackingIdIfEnabled(TokenIdent ident) {
|
protected String getTrackingIdIfEnabled(TokenIdent ident) {
|
||||||
@ -558,12 +633,17 @@ protected String getTrackingIdIfEnabled(TokenIdent ident) {
|
|||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
public synchronized String getTokenTrackingId(TokenIdent identifier) {
|
public String getTokenTrackingId(TokenIdent identifier) {
|
||||||
DelegationTokenInformation info = getTokenInfo(identifier);
|
this.apiLock.readLock().lock();
|
||||||
if (info == null) {
|
try {
|
||||||
return null;
|
DelegationTokenInformation info = getTokenInfo(identifier);
|
||||||
|
if (info == null) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
return info.getTrackingId();
|
||||||
|
} finally {
|
||||||
|
this.apiLock.readLock().unlock();
|
||||||
}
|
}
|
||||||
return info.getTrackingId();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -572,12 +652,17 @@ public synchronized String getTokenTrackingId(TokenIdent identifier) {
|
|||||||
* @param password Password in the token.
|
* @param password Password in the token.
|
||||||
* @throws InvalidToken InvalidToken.
|
* @throws InvalidToken InvalidToken.
|
||||||
*/
|
*/
|
||||||
public synchronized void verifyToken(TokenIdent identifier, byte[] password)
|
public void verifyToken(TokenIdent identifier, byte[] password)
|
||||||
throws InvalidToken {
|
throws InvalidToken {
|
||||||
byte[] storedPassword = retrievePassword(identifier);
|
this.apiLock.readLock().lock();
|
||||||
if (!MessageDigest.isEqual(password, storedPassword)) {
|
try {
|
||||||
throw new InvalidToken("token " + formatTokenId(identifier)
|
byte[] storedPassword = retrievePassword(identifier);
|
||||||
+ " is invalid, password doesn't match");
|
if (!MessageDigest.isEqual(password, storedPassword)) {
|
||||||
|
throw new InvalidToken("token " + formatTokenId(identifier)
|
||||||
|
+ " is invalid, password doesn't match");
|
||||||
|
}
|
||||||
|
} finally {
|
||||||
|
this.apiLock.readLock().unlock();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -589,57 +674,55 @@ public synchronized void verifyToken(TokenIdent identifier, byte[] password)
|
|||||||
* @throws InvalidToken if the token is invalid
|
* @throws InvalidToken if the token is invalid
|
||||||
* @throws AccessControlException if the user can't renew token
|
* @throws AccessControlException if the user can't renew token
|
||||||
*/
|
*/
|
||||||
public synchronized long renewToken(Token<TokenIdent> token,
|
public long renewToken(Token<TokenIdent> token,
|
||||||
String renewer) throws InvalidToken, IOException {
|
String renewer) throws InvalidToken, IOException {
|
||||||
ByteArrayInputStream buf = new ByteArrayInputStream(token.getIdentifier());
|
this.apiLock.writeLock().lock();
|
||||||
DataInputStream in = new DataInputStream(buf);
|
try {
|
||||||
TokenIdent id = createIdentifier();
|
ByteArrayInputStream buf = new ByteArrayInputStream(token.getIdentifier());
|
||||||
id.readFields(in);
|
DataInputStream in = new DataInputStream(buf);
|
||||||
LOG.info("Token renewal for identifier: " + formatTokenId(id)
|
TokenIdent id = createIdentifier();
|
||||||
+ "; total currentTokens " + currentTokens.size());
|
id.readFields(in);
|
||||||
|
LOG.info("Token renewal for identifier: " + formatTokenId(id) + "; total currentTokens "
|
||||||
|
+ currentTokens.size());
|
||||||
|
|
||||||
long now = Time.now();
|
long now = Time.now();
|
||||||
if (id.getMaxDate() < now) {
|
if (id.getMaxDate() < now) {
|
||||||
throw new InvalidToken(renewer + " tried to renew an expired token "
|
throw new InvalidToken(renewer + " tried to renew an expired token " + formatTokenId(id) +
|
||||||
+ formatTokenId(id) + " max expiration date: "
|
" max expiration date: " + Time.formatTime(id.getMaxDate()) + " currentTime: " +
|
||||||
+ Time.formatTime(id.getMaxDate())
|
Time.formatTime(now));
|
||||||
+ " currentTime: " + Time.formatTime(now));
|
}
|
||||||
}
|
if ((id.getRenewer() == null) || (id.getRenewer().toString().isEmpty())) {
|
||||||
if ((id.getRenewer() == null) || (id.getRenewer().toString().isEmpty())) {
|
throw new AccessControlException(renewer + " tried to renew a token " + formatTokenId(id) +
|
||||||
throw new AccessControlException(renewer +
|
" without a renewer");
|
||||||
" tried to renew a token " + formatTokenId(id)
|
}
|
||||||
+ " without a renewer");
|
if (!id.getRenewer().toString().equals(renewer)) {
|
||||||
}
|
throw new AccessControlException(renewer + " tries to renew a token " + formatTokenId(id) +
|
||||||
if (!id.getRenewer().toString().equals(renewer)) {
|
" with non-matching renewer " + id.getRenewer());
|
||||||
throw new AccessControlException(renewer
|
}
|
||||||
+ " tries to renew a token " + formatTokenId(id)
|
DelegationKey key = getDelegationKey(id.getMasterKeyId());
|
||||||
+ " with non-matching renewer " + id.getRenewer());
|
if (key == null) {
|
||||||
}
|
throw new InvalidToken("Unable to find master key for keyId=" + id.getMasterKeyId() +
|
||||||
DelegationKey key = getDelegationKey(id.getMasterKeyId());
|
" from cache. Failed to renew an unexpired token " + formatTokenId(id) +
|
||||||
if (key == null) {
|
" with sequenceNumber=" + id.getSequenceNumber());
|
||||||
throw new InvalidToken("Unable to find master key for keyId="
|
}
|
||||||
+ id.getMasterKeyId()
|
byte[] password = createPassword(token.getIdentifier(), key.getKey());
|
||||||
+ " from cache. Failed to renew an unexpired token "
|
if (!MessageDigest.isEqual(password, token.getPassword())) {
|
||||||
+ formatTokenId(id) + " with sequenceNumber="
|
throw new AccessControlException(
|
||||||
+ id.getSequenceNumber());
|
renewer + " is trying to renew a token " + formatTokenId(id) + " with wrong password");
|
||||||
}
|
}
|
||||||
byte[] password = createPassword(token.getIdentifier(), key.getKey());
|
long renewTime = Math.min(id.getMaxDate(), now + tokenRenewInterval);
|
||||||
if (!MessageDigest.isEqual(password, token.getPassword())) {
|
String trackingId = getTrackingIdIfEnabled(id);
|
||||||
throw new AccessControlException(renewer
|
DelegationTokenInformation info =
|
||||||
+ " is trying to renew a token "
|
new DelegationTokenInformation(renewTime, password, trackingId);
|
||||||
+ formatTokenId(id) + " with wrong password");
|
|
||||||
}
|
|
||||||
long renewTime = Math.min(id.getMaxDate(), now + tokenRenewInterval);
|
|
||||||
String trackingId = getTrackingIdIfEnabled(id);
|
|
||||||
DelegationTokenInformation info = new DelegationTokenInformation(renewTime,
|
|
||||||
password, trackingId);
|
|
||||||
|
|
||||||
if (getTokenInfo(id) == null) {
|
if (getTokenInfo(id) == null) {
|
||||||
throw new InvalidToken("Renewal request for unknown token "
|
throw new InvalidToken("Renewal request for unknown token " + formatTokenId(id));
|
||||||
+ formatTokenId(id));
|
}
|
||||||
|
METRICS.trackUpdateToken(() -> updateToken(id, info));
|
||||||
|
return renewTime;
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
METRICS.trackUpdateToken(() -> updateToken(id, info));
|
|
||||||
return renewTime;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -651,37 +734,41 @@ public synchronized long renewToken(Token<TokenIdent> token,
|
|||||||
* @throws InvalidToken for invalid token
|
* @throws InvalidToken for invalid token
|
||||||
* @throws AccessControlException if the user isn't allowed to cancel
|
* @throws AccessControlException if the user isn't allowed to cancel
|
||||||
*/
|
*/
|
||||||
public synchronized TokenIdent cancelToken(Token<TokenIdent> token,
|
public TokenIdent cancelToken(Token<TokenIdent> token,
|
||||||
String canceller) throws IOException {
|
String canceller) throws IOException {
|
||||||
ByteArrayInputStream buf = new ByteArrayInputStream(token.getIdentifier());
|
this.apiLock.writeLock().lock();
|
||||||
DataInputStream in = new DataInputStream(buf);
|
try {
|
||||||
TokenIdent id = createIdentifier();
|
ByteArrayInputStream buf = new ByteArrayInputStream(token.getIdentifier());
|
||||||
id.readFields(in);
|
DataInputStream in = new DataInputStream(buf);
|
||||||
LOG.info("Token cancellation requested for identifier: "
|
TokenIdent id = createIdentifier();
|
||||||
+ formatTokenId(id));
|
id.readFields(in);
|
||||||
|
LOG.info("Token cancellation requested for identifier: " + formatTokenId(id));
|
||||||
if (id.getUser() == null) {
|
|
||||||
throw new InvalidToken("Token with no owner " + formatTokenId(id));
|
if (id.getUser() == null) {
|
||||||
|
throw new InvalidToken("Token with no owner " + formatTokenId(id));
|
||||||
|
}
|
||||||
|
String owner = id.getUser().getUserName();
|
||||||
|
Text renewer = id.getRenewer();
|
||||||
|
HadoopKerberosName cancelerKrbName = new HadoopKerberosName(canceller);
|
||||||
|
String cancelerShortName = cancelerKrbName.getShortName();
|
||||||
|
if (!canceller.equals(owner) &&
|
||||||
|
(renewer == null || renewer.toString().isEmpty() ||
|
||||||
|
!cancelerShortName.equals(renewer.toString()))) {
|
||||||
|
throw new AccessControlException(canceller + " is not authorized to cancel the token " +
|
||||||
|
formatTokenId(id));
|
||||||
|
}
|
||||||
|
DelegationTokenInformation info = currentTokens.remove(id);
|
||||||
|
if (info == null) {
|
||||||
|
throw new InvalidToken("Token not found " + formatTokenId(id));
|
||||||
|
}
|
||||||
|
METRICS.trackRemoveToken(() -> {
|
||||||
|
removeTokenForOwnerStats(id);
|
||||||
|
removeStoredToken(id);
|
||||||
|
});
|
||||||
|
return id;
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
String owner = id.getUser().getUserName();
|
|
||||||
Text renewer = id.getRenewer();
|
|
||||||
HadoopKerberosName cancelerKrbName = new HadoopKerberosName(canceller);
|
|
||||||
String cancelerShortName = cancelerKrbName.getShortName();
|
|
||||||
if (!canceller.equals(owner)
|
|
||||||
&& (renewer == null || renewer.toString().isEmpty() || !cancelerShortName
|
|
||||||
.equals(renewer.toString()))) {
|
|
||||||
throw new AccessControlException(canceller
|
|
||||||
+ " is not authorized to cancel the token " + formatTokenId(id));
|
|
||||||
}
|
|
||||||
DelegationTokenInformation info = currentTokens.remove(id);
|
|
||||||
if (info == null) {
|
|
||||||
throw new InvalidToken("Token not found " + formatTokenId(id));
|
|
||||||
}
|
|
||||||
METRICS.trackRemoveToken(() -> {
|
|
||||||
removeTokenForOwnerStats(id);
|
|
||||||
removeStoredToken(id);
|
|
||||||
});
|
|
||||||
return id;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -762,7 +849,8 @@ public void readFields(DataInput in) throws IOException {
|
|||||||
private void removeExpiredToken() throws IOException {
|
private void removeExpiredToken() throws IOException {
|
||||||
long now = Time.now();
|
long now = Time.now();
|
||||||
Set<TokenIdent> expiredTokens = new HashSet<>();
|
Set<TokenIdent> expiredTokens = new HashSet<>();
|
||||||
synchronized (this) {
|
this.apiLock.writeLock().lock();
|
||||||
|
try {
|
||||||
Iterator<Map.Entry<TokenIdent, DelegationTokenInformation>> i =
|
Iterator<Map.Entry<TokenIdent, DelegationTokenInformation>> i =
|
||||||
getCandidateTokensForCleanup().entrySet().iterator();
|
getCandidateTokensForCleanup().entrySet().iterator();
|
||||||
while (i.hasNext()) {
|
while (i.hasNext()) {
|
||||||
@ -774,6 +862,8 @@ private void removeExpiredToken() throws IOException {
|
|||||||
i.remove();
|
i.remove();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
} finally {
|
||||||
|
this.apiLock.writeLock().unlock();
|
||||||
}
|
}
|
||||||
// don't hold lock on 'this' to avoid edit log updates blocking token ops
|
// don't hold lock on 'this' to avoid edit log updates blocking token ops
|
||||||
logExpireTokens(expiredTokens);
|
logExpireTokens(expiredTokens);
|
||||||
@ -818,10 +908,10 @@ public void stopThreads() {
|
|||||||
* is secretMgr running
|
* is secretMgr running
|
||||||
* @return true if secret mgr is running
|
* @return true if secret mgr is running
|
||||||
*/
|
*/
|
||||||
public synchronized boolean isRunning() {
|
public boolean isRunning() {
|
||||||
return running;
|
return running;
|
||||||
}
|
}
|
||||||
|
|
||||||
private class ExpiredTokenRemover extends Thread {
|
private class ExpiredTokenRemover extends Thread {
|
||||||
private long lastMasterKeyUpdate;
|
private long lastMasterKeyUpdate;
|
||||||
private long lastTokenCacheCleanup;
|
private long lastTokenCacheCleanup;
|
||||||
|
Loading…
Reference in New Issue
Block a user