Reviewed-by: Masatake Iwasaki <iwasakims@apache.org> (cherry picked from commit2bd810a507
) (cherry picked from commit77315abe47
) Conflicts: hadoop-common-project/hadoop-kms/src/test/java/org/apache/hadoop/crypto/key/kms/server/TestKMS.java Co-authored-by: Akira Ajisaka <aajisaka@apache.org>
This commit is contained in:
parent
9208218704
commit
289c7d91b4
|
@ -17,7 +17,6 @@
|
||||||
*/
|
*/
|
||||||
package org.apache.hadoop.crypto.key.kms.server;
|
package org.apache.hadoop.crypto.key.kms.server;
|
||||||
|
|
||||||
import java.util.function.Supplier;
|
|
||||||
import com.google.common.cache.LoadingCache;
|
import com.google.common.cache.LoadingCache;
|
||||||
import org.apache.curator.test.TestingServer;
|
import org.apache.curator.test.TestingServer;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
@ -91,7 +90,6 @@ import java.util.LinkedList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Properties;
|
import java.util.Properties;
|
||||||
import java.util.Set;
|
|
||||||
import java.util.UUID;
|
import java.util.UUID;
|
||||||
import java.util.concurrent.Callable;
|
import java.util.concurrent.Callable;
|
||||||
import java.util.concurrent.LinkedBlockingQueue;
|
import java.util.concurrent.LinkedBlockingQueue;
|
||||||
|
@ -112,9 +110,6 @@ import static org.mockito.Mockito.when;
|
||||||
public class TestKMS {
|
public class TestKMS {
|
||||||
private static final Logger LOG = LoggerFactory.getLogger(TestKMS.class);
|
private static final Logger LOG = LoggerFactory.getLogger(TestKMS.class);
|
||||||
|
|
||||||
private static final String SSL_RELOADER_THREAD_NAME =
|
|
||||||
"Truststore reloader thread";
|
|
||||||
|
|
||||||
private SSLFactory sslFactory;
|
private SSLFactory sslFactory;
|
||||||
|
|
||||||
// Keep track of all key providers created during a test case, so they can be
|
// Keep track of all key providers created during a test case, so they can be
|
||||||
|
@ -539,34 +534,6 @@ public class TestKMS {
|
||||||
url.getProtocol().equals("https"));
|
url.getProtocol().equals("https"));
|
||||||
final URI uri = createKMSUri(getKMSUrl());
|
final URI uri = createKMSUri(getKMSUrl());
|
||||||
|
|
||||||
if (ssl) {
|
|
||||||
KeyProvider testKp = createProvider(uri, conf);
|
|
||||||
ThreadGroup threadGroup = Thread.currentThread().getThreadGroup();
|
|
||||||
while (threadGroup.getParent() != null) {
|
|
||||||
threadGroup = threadGroup.getParent();
|
|
||||||
}
|
|
||||||
Thread[] threads = new Thread[threadGroup.activeCount()];
|
|
||||||
threadGroup.enumerate(threads);
|
|
||||||
Thread reloaderThread = null;
|
|
||||||
for (Thread thread : threads) {
|
|
||||||
if ((thread.getName() != null)
|
|
||||||
&& (thread.getName().contains(SSL_RELOADER_THREAD_NAME))) {
|
|
||||||
reloaderThread = thread;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Assert.assertTrue("Reloader is not alive", reloaderThread.isAlive());
|
|
||||||
// Explicitly close the provider so we can verify the internal thread
|
|
||||||
// is shutdown
|
|
||||||
testKp.close();
|
|
||||||
boolean reloaderStillAlive = true;
|
|
||||||
for (int i = 0; i < 10; i++) {
|
|
||||||
reloaderStillAlive = reloaderThread.isAlive();
|
|
||||||
if (!reloaderStillAlive) break;
|
|
||||||
Thread.sleep(1000);
|
|
||||||
}
|
|
||||||
Assert.assertFalse("Reloader is still alive", reloaderStillAlive);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (kerberos) {
|
if (kerberos) {
|
||||||
for (String user : new String[]{"client", "client/host"}) {
|
for (String user : new String[]{"client", "client/host"}) {
|
||||||
doAs(user, new PrivilegedExceptionAction<Void>() {
|
doAs(user, new PrivilegedExceptionAction<Void>() {
|
||||||
|
@ -2362,8 +2329,7 @@ public class TestKMS {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
// Close the client provider. We will verify all providers'
|
// Close the client provider.
|
||||||
// Truststore reloader threads are closed later.
|
|
||||||
kp.close();
|
kp.close();
|
||||||
return null;
|
return null;
|
||||||
} finally {
|
} finally {
|
||||||
|
@ -2374,22 +2340,6 @@ public class TestKMS {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
// verify that providers created by KMSTokenRenewer are closed.
|
|
||||||
if (ssl) {
|
|
||||||
GenericTestUtils.waitFor(new Supplier<Boolean>() {
|
|
||||||
@Override
|
|
||||||
public Boolean get() {
|
|
||||||
final Set<Thread> threadSet = Thread.getAllStackTraces().keySet();
|
|
||||||
for (Thread t : threadSet) {
|
|
||||||
if (t.getName().contains(SSL_RELOADER_THREAD_NAME)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}, 1000, 10000);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
|
|
Loading…
Reference in New Issue