YARN-10555. Missing access check before getAppAttempts (#2608)

Co-authored-by: lujie <lujie@foxmail.com>
Signed-off-by: Akira Ajisaka <aajisaka@apache.org>
This commit is contained in:
lujiefsi 2021-05-16 23:44:38 -05:00 committed by GitHub
parent fcd4140e5f
commit d92a25b790
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 5 deletions

View File

@ -142,7 +142,7 @@ protected void generateApplicationTable(Block html,
continue; continue;
} }
AppAttemptInfo attemptInfo = AppAttemptInfo attemptInfo =
new AppAttemptInfo(this.rm, rmAppAttempt, rmApp.getUser(), new AppAttemptInfo(this.rm, rmAppAttempt, true, rmApp.getUser(),
WebAppUtils.getHttpSchemePrefix(conf)); WebAppUtils.getHttpSchemePrefix(conf));
Set<String> nodes = rmAppAttempt.getBlacklistedNodes(); Set<String> nodes = rmAppAttempt.getBlacklistedNodes();
// nodes which are blacklisted by the application // nodes which are blacklisted by the application

View File

@ -1127,7 +1127,7 @@ public AppAttemptsInfo getAppAttempts(@Context HttpServletRequest hsr,
AppAttemptsInfo appAttemptsInfo = new AppAttemptsInfo(); AppAttemptsInfo appAttemptsInfo = new AppAttemptsInfo();
for (RMAppAttempt attempt : app.getAppAttempts().values()) { for (RMAppAttempt attempt : app.getAppAttempts().values()) {
AppAttemptInfo attemptInfo = new AppAttemptInfo(rm, attempt, AppAttemptInfo attemptInfo = new AppAttemptInfo(rm, attempt,
app.getUser(), hsr.getScheme() + "://"); hasAccess(app, hsr), app.getUser(), hsr.getScheme() + "://");
appAttemptsInfo.add(attemptInfo); appAttemptsInfo.add(attemptInfo);
} }

View File

@ -51,8 +51,8 @@ public class AppAttemptInfo {
public AppAttemptInfo() { public AppAttemptInfo() {
} }
public AppAttemptInfo(ResourceManager rm, RMAppAttempt attempt, String user, public AppAttemptInfo(ResourceManager rm, RMAppAttempt attempt,
String schemePrefix) { Boolean hasAccess, String user, String schemePrefix) {
this.startTime = 0; this.startTime = 0;
this.containerId = ""; this.containerId = "";
this.nodeHttpAddress = ""; this.nodeHttpAddress = "";
@ -65,7 +65,7 @@ public AppAttemptInfo(ResourceManager rm, RMAppAttempt attempt, String user,
this.startTime = attempt.getStartTime(); this.startTime = attempt.getStartTime();
this.finishedTime = attempt.getFinishTime(); this.finishedTime = attempt.getFinishTime();
Container masterContainer = attempt.getMasterContainer(); Container masterContainer = attempt.getMasterContainer();
if (masterContainer != null) { if (masterContainer != null && hasAccess) {
this.containerId = masterContainer.getId().toString(); this.containerId = masterContainer.getId().toString();
this.nodeHttpAddress = masterContainer.getNodeHttpAddress(); this.nodeHttpAddress = masterContainer.getNodeHttpAddress();
this.nodeId = masterContainer.getNodeId().toString(); this.nodeId = masterContainer.getNodeId().toString();