123 lines
3.7 KiB
Java
123 lines
3.7 KiB
Java
package net.limework.networking;
|
|
|
|
import com.jcraft.jsch.JSch;
|
|
import com.jcraft.jsch.JSchException;
|
|
import com.jcraft.jsch.Session;
|
|
import org.bukkit.plugin.java.JavaPlugin;
|
|
import org.bukkit.scheduler.BukkitRunnable;
|
|
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.FileNotFoundException;
|
|
|
|
public class AppleTunneler extends JavaPlugin {
|
|
|
|
private Session sshSession;
|
|
private final JSch jsch = new JSch();
|
|
|
|
private synchronized void createNewConnection() throws JSchException {
|
|
if (sshSession != null) {
|
|
sshSession.disconnect();
|
|
}
|
|
sshSession = jsch.getSession(getUsername(), getHost(), getPort());
|
|
sshSession.connect(2000);
|
|
loadRemoteToLocalPortForwarding();
|
|
}
|
|
|
|
private void connectionFailedLog(JSchException exception) {
|
|
getLogger().severe("==========================");
|
|
getLogger().severe("failed to start a connection");
|
|
exception.printStackTrace();
|
|
getLogger().severe("==========================");
|
|
|
|
}
|
|
|
|
private void loadIds() throws JSchException {
|
|
File file = new File(getDataFolder(), "Identities");
|
|
if (file.mkdir()) {
|
|
getLogger().info("created Identities folder.");
|
|
}
|
|
File[] files = file.listFiles();
|
|
|
|
if (files == null) {
|
|
return;
|
|
}
|
|
if (files.length == 0) {
|
|
getLogger().warning("no Identities files found, please put them in " + file.getPath());
|
|
throw new RuntimeException("no Identities files found");
|
|
}
|
|
for (File id : files) {
|
|
jsch.addIdentity(id.getPath());
|
|
}
|
|
}
|
|
|
|
private void loadRemoteToLocalPortForwarding() throws JSchException {
|
|
for (String line : getConfig().getStringList("ports")) {
|
|
String[] array = line.split(":");
|
|
sshSession.setPortForwardingL(Integer.parseInt(array[0]), array[1], Integer.parseInt(array[2]));
|
|
}
|
|
}
|
|
|
|
private void loadKnownHostFile() throws FileNotFoundException, JSchException {
|
|
File knownHosts = new File(getDataFolder(), ".known_hosts");
|
|
if (knownHosts.isDirectory()) {
|
|
knownHosts.delete();
|
|
}
|
|
if (!knownHosts.exists()) {
|
|
getLogger().severe("FILE at path: " + knownHosts.getAbsolutePath() + " Does not exists");
|
|
throw new RuntimeException("Known host file does not exists in plugin folder");
|
|
}
|
|
this.jsch.setKnownHosts(new FileInputStream(knownHosts));
|
|
|
|
}
|
|
|
|
@Override
|
|
public void onEnable() {
|
|
this.saveDefaultConfig();
|
|
try {
|
|
loadIds();
|
|
loadKnownHostFile();
|
|
} catch (JSchException | FileNotFoundException e) {
|
|
throw new RuntimeException(e);
|
|
}
|
|
try {
|
|
createNewConnection();
|
|
} catch (JSchException e) {
|
|
connectionFailedLog(e);
|
|
}
|
|
new BukkitRunnable() {
|
|
private boolean isConnecting = false;
|
|
|
|
@Override
|
|
public void run() {
|
|
if (isConnecting) {
|
|
return;
|
|
}
|
|
if (!sshSession.isConnected()) {
|
|
isConnecting = true;
|
|
try {
|
|
createNewConnection();
|
|
} catch (JSchException e) {
|
|
connectionFailedLog(e);
|
|
}
|
|
isConnecting = false;
|
|
}
|
|
}
|
|
}.runTaskTimerAsynchronously(this, 0, 20);
|
|
|
|
}
|
|
|
|
private int getPort() {
|
|
return this.getConfig().getInt("port");
|
|
}
|
|
|
|
private String getUsername() {
|
|
return this.getConfig().getString("username");
|
|
}
|
|
|
|
private String getHost() {
|
|
return this.getConfig().getString("host");
|
|
}
|
|
|
|
}
|