You are viewing a plain text version of this content. The canonical link for it is here.
Posted to common-commits@hadoop.apache.org by li...@apache.org on 2020/08/25 07:21:21 UTC
[hadoop] branch branch-3.1 updated: HADOOP-17159 Ability for
forceful relogin in UserGroupInformation class (#2197)
This is an automated email from the ASF dual-hosted git repository.
liuml07 pushed a commit to branch branch-3.1
in repository https://gitbox.apache.org/repos/asf/hadoop.git
The following commit(s) were added to refs/heads/branch-3.1 by this push:
new 12fb9e0 HADOOP-17159 Ability for forceful relogin in UserGroupInformation class (#2197)
12fb9e0 is described below
commit 12fb9e0600f665aca3e7ebe0be9b95ff232d520f
Author: sguggilam <sa...@gmail.com>
AuthorDate: Mon Aug 24 23:39:57 2020 -0700
HADOOP-17159 Ability for forceful relogin in UserGroupInformation class (#2197)
Contributed by Sandeep Guggilam.
Signed-off-by: Mingliang Liu <li...@apache.org>
Signed-off-by: Steve Loughran <st...@apache.org>
---
.../hadoop/security/UserGroupInformation.java | 35 ++++++++++++++++-----
.../hadoop/security/TestUGILoginFromKeytab.java | 36 ++++++++++++++++++++++
2 files changed, 64 insertions(+), 7 deletions(-)
diff --git a/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/security/UserGroupInformation.java b/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/security/UserGroupInformation.java
index 0e4168c..2471e0a 100644
--- a/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/security/UserGroupInformation.java
+++ b/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/security/UserGroupInformation.java
@@ -1115,7 +1115,26 @@ public class UserGroupInformation {
reloginFromKeytab(false);
}
- private void reloginFromKeytab(boolean checkTGT) throws IOException {
+ /**
+ * Force re-Login a user in from a keytab file. Loads a user identity from a
+ * keytab file and logs them in. They become the currently logged-in user.
+ * This method assumes that {@link #loginUserFromKeytab(String, String)} had
+ * happened already. The Subject field of this UserGroupInformation object is
+ * updated to have the new credentials.
+ *
+ * @param ignoreTimeElapsed Force re-login irrespective of the time of last
+ * login
+ * @throws IOException
+ * @throws KerberosAuthException on a failure
+ */
+ @InterfaceAudience.Public
+ @InterfaceStability.Evolving
+ public void reloginFromKeytab(boolean ignoreTimeElapsed) throws IOException {
+ reloginFromKeytab(false, ignoreTimeElapsed);
+ }
+
+ private void reloginFromKeytab(boolean checkTGT, boolean ignoreTimeElapsed)
+ throws IOException {
if (!shouldRelogin() || !isFromKeytab()) {
return;
}
@@ -1130,7 +1149,7 @@ public class UserGroupInformation {
return;
}
}
- relogin(login);
+ relogin(login, ignoreTimeElapsed);
}
/**
@@ -1151,25 +1170,27 @@ public class UserGroupInformation {
if (login == null) {
throw new KerberosAuthException(MUST_FIRST_LOGIN);
}
- relogin(login);
+ relogin(login, false);
}
- private void relogin(HadoopLoginContext login) throws IOException {
+ private void relogin(HadoopLoginContext login, boolean ignoreTimeElapsed)
+ throws IOException {
// ensure the relogin is atomic to avoid leaving credentials in an
// inconsistent state. prevents other ugi instances, SASL, and SPNEGO
// from accessing or altering credentials during the relogin.
synchronized(login.getSubjectLock()) {
// another racing thread may have beat us to the relogin.
if (login == getLogin()) {
- unprotectedRelogin(login);
+ unprotectedRelogin(login, ignoreTimeElapsed);
}
}
}
- private void unprotectedRelogin(HadoopLoginContext login) throws IOException {
+ private void unprotectedRelogin(HadoopLoginContext login,
+ boolean ignoreTimeElapsed) throws IOException {
assert Thread.holdsLock(login.getSubjectLock());
long now = Time.now();
- if (!hasSufficientTimeElapsed(now)) {
+ if (!hasSufficientTimeElapsed(now) && !ignoreTimeElapsed) {
return;
}
// register most recent relogin attempt
diff --git a/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/security/TestUGILoginFromKeytab.java b/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/security/TestUGILoginFromKeytab.java
index bf4a2cc..7e2c250d 100644
--- a/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/security/TestUGILoginFromKeytab.java
+++ b/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/security/TestUGILoginFromKeytab.java
@@ -154,6 +154,42 @@ public class TestUGILoginFromKeytab {
Assert.assertNotSame(login1, login2);
}
+ /**
+ * Force re-login from keytab using the MiniKDC and verify the UGI can
+ * successfully relogin from keytab as well.
+ */
+ @Test
+ public void testUGIForceReLoginFromKeytab() throws Exception {
+ // Set this to false as we are testing force re-login anyways
+ UserGroupInformation.setShouldRenewImmediatelyForTests(false);
+ String principal = "foo";
+ File keytab = new File(workDir, "foo.keytab");
+ kdc.createPrincipal(keytab, principal);
+
+ UserGroupInformation.loginUserFromKeytab(principal, keytab.getPath());
+ UserGroupInformation ugi = UserGroupInformation.getLoginUser();
+ Assert.assertTrue("UGI should be configured to login from keytab",
+ ugi.isFromKeytab());
+
+ // Verify relogin from keytab.
+ User user = getUser(ugi.getSubject());
+ final long firstLogin = user.getLastLogin();
+ final LoginContext login1 = user.getLogin();
+ Assert.assertNotNull(login1);
+
+ // Sleep for 2 secs to have a difference between first and second login
+ Thread.sleep(2000);
+
+ // Force relogin from keytab
+ ugi.reloginFromKeytab(true);
+ final long secondLogin = user.getLastLogin();
+ final LoginContext login2 = user.getLogin();
+ Assert.assertTrue("User should have been able to relogin from keytab",
+ secondLogin > firstLogin);
+ Assert.assertNotNull(login2);
+ Assert.assertNotSame(login1, login2);
+ }
+
@Test
public void testGetUGIFromKnownSubject() throws Exception {
KerberosPrincipal principal = new KerberosPrincipal("user");
---------------------------------------------------------------------
To unsubscribe, e-mail: common-commits-unsubscribe@hadoop.apache.org
For additional commands, e-mail: common-commits-help@hadoop.apache.org