40 lines
1.3 KiB
Java
40 lines
1.3 KiB
Java
package com.ryanmichela.sshd;
|
|
|
|
import org.apache.sshd.server.PasswordAuthenticator;
|
|
import org.apache.sshd.server.session.ServerSession;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
/**
|
|
* Copyright 2013 Ryan Michela
|
|
*/
|
|
public class ConfigPasswordAuthenticator implements PasswordAuthenticator {
|
|
private Map<String, Integer> failCounts = new HashMap<String, Integer>();
|
|
|
|
@Override
|
|
public boolean authenticate(String username, String password, ServerSession serverSession) {
|
|
if (SshdPlugin.instance.getConfig().getString("credentials." + username).equals(password)) {
|
|
failCounts.put(username, 0);
|
|
return true;
|
|
}
|
|
SshdPlugin.instance.getLogger().info("Failed login for " + username + " using password authentication.");
|
|
|
|
try {
|
|
Thread.sleep(3000);
|
|
if (failCounts.containsKey(username)) {
|
|
failCounts.put(username, failCounts.get(username) + 1);
|
|
} else {
|
|
failCounts.put(username, 1);
|
|
}
|
|
if (failCounts.get(username) >= 3) {
|
|
failCounts.put(username, 0);
|
|
serverSession.close(true);
|
|
}
|
|
} catch (InterruptedException e) {
|
|
// do nothing
|
|
}
|
|
return false;
|
|
}
|
|
}
|