opendev/gerrit
Code Issues Proposed changes

Revert "Revert "SSH: Simplify CachingPublicKeyAuthenticator implementation""

Browse Source 
This reverts commit 48bf33b2eb.

Change-Id: I145d159327796ee524e05cc2019bb355f245a94b
This commit is contained in:
David Pursehouse
2015-04-07 19:23:11 +09:00
parent c41edaa2c1
commit c7f58ccbd8
1 changed files with 3 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
gerrit-sshd/src/main/java/com/google/gerrit/sshd/CachingPublicKeyAuthenticator.java
View File

@@ -17,56 +17,12 @@ package com.google.gerrit.sshd;
import com.google.inject.Inject; import com.google.inject.Inject;
import com.google.inject.Singleton; import com.google.inject.Singleton;
import org.apache.sshd.common.Session;
import org.apache.sshd.common.SessionListener;
import org.apache.sshd.server.PublickeyAuthenticator;
import org.apache.sshd.server.session.ServerSession;
import java.security.PublicKey;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
@Singleton @Singleton
public class CachingPublicKeyAuthenticator implements PublickeyAuthenticator, public class CachingPublicKeyAuthenticator
SessionListener { extends org.apache.sshd.server.auth.CachingPublicKeyAuthenticator {
private final PublickeyAuthenticator authenticator;
private final Map<ServerSession, Map<PublicKey, Boolean>> sessionCache;
@Inject @Inject
public CachingPublicKeyAuthenticator(DatabasePubKeyAuth authenticator) { public CachingPublicKeyAuthenticator(DatabasePubKeyAuth authenticator) {
this.authenticator = authenticator; super(authenticator);
this.sessionCache = new ConcurrentHashMap<>();
}
@Override
public boolean authenticate(String username, PublicKey key,
ServerSession session) {
Map<PublicKey, Boolean> m = sessionCache.get(session);
if (m == null) {
m = new HashMap<>();
sessionCache.put(session, m);
session.addListener(this);
}
if (m.containsKey(key)) {
return m.get(key);
}
boolean r = authenticator.authenticate(username, key, session);
m.put(key, r);
return r;
}
@Override
public void sessionCreated(Session session) {
}
@Override
public void sessionEvent(Session sesssion, Event event) {
}
@Override
public void sessionClosed(Session session) {
sessionCache.remove(session);
} }
} }