com.lambdatest.tunnel.Tunnel Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of lambdatest-tunnel-binary Show documentation
Show all versions of lambdatest-tunnel-binary Show documentation
LambdaTest tunnel binary in Java
package com.lambdatest.tunnel;
import java.io.*;
import java.nio.file.*;
import java.net.HttpURLConnection;
import java.net.ServerSocket;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.*;
import java.util.concurrent.TimeoutException;
import java.util.concurrent.locks.ReentrantLock;
import org.apache.http.HttpResponse;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.methods.HttpDelete;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import java.util.HashMap;
import com.lambdatest.httpserver.HttpServer;
import com.lambdatest.tunnel.TestDaemonThread1;
import com.lambdatest.KillPort;
/**
* Creates and manages a secure tunnel connection to LambdaTest.
*/
public class Tunnel {
private static final List IGNORE_KEYS = Arrays.asList("user", "key", "infoAPIPort", "binarypath",
"load-balanced", "mitm", "pacfile", "mTLSHosts", "clientKey",
"clientCert", "allowHosts", "verbose", "serverDomain", "usePrivateIp", "retry-proxy-error",
"retry-proxy-error-count", "ntlm", "ntlmPassword", "ntlmUsername", "maxSSHConnections");
private boolean tunnelFlag = false;
private int infoAPIPortValue;
private Map parameters;
private String TunnelID;
private Integer tunnelCount = 0;
TunnelBinary tunnelBinary;
private Process process = null;
private ReentrantLock mutex = new ReentrantLock();
TestDaemonThread1 t1 = new TestDaemonThread1();// creating thread
public Tunnel() throws TunnelException {
parameters = new HashMap();
parameters.put("bypassHosts", "--bypassHosts");
parameters.put("callbackURL", "--callbackURL");
parameters.put("config", "--config");
parameters.put("controller", "--controller");
parameters.put("egress-only ", "--egress-only ");
parameters.put("dir", "--dir");
parameters.put("dns", "--dns");
parameters.put("emulateChrome", "--emulateChrome");
parameters.put("env", "--env");
parameters.put("help", "--help");
parameters.put("infoAPIPort", "--infoAPIPort");
parameters.put("ingress-only", "--ingress-only");
parameters.put("key", "--key");
parameters.put("localDomains", "--local-domains");
parameters.put("logFile", "--logFile");
parameters.put("mode", "--mode");
parameters.put("noProxy", "--no-proxy");
parameters.put("pidfile", "--pidfile");
parameters.put("port", "--port");
parameters.put("proxyHost", "--proxy-host");
parameters.put("proxyPass", "--proxy-pass");
parameters.put("proxyPort", "--proxy-port");
parameters.put("proxyUser", "--proxy-user");
parameters.put("sharedTunnel", "--shared-tunnel");
parameters.put("sshConnType", "--sshConnType");
parameters.put("tunnelName", "--tunnelName");
parameters.put("user", "--user");
parameters.put("load-balanced", "--load-balanced");
parameters.put("v", "--v");
parameters.put("version", "--version");
parameters.put("basicAuth", "--basic-auth");
parameters.put("mitm", "--mitm");
parameters.put("skip-upgrade", "--skip-upgrade");
parameters.put("pacfile", "--pacfile");
parameters.put("mTLSHosts", "--mTLSHosts");
parameters.put("clientKey", "--clientKey");
parameters.put("clientCert", "--clientCert");
parameters.put("allowHosts", "--allowHosts");
parameters.put("verbose", "--verbose");
parameters.put("serverDomain", "--server-domain");
parameters.put("usePrivateIp", "--use-private-ip");
parameters.put("retry-proxy-error", "--retry-proxy-error");
parameters.put("retry-proxy-error-count", "--retry-proxy-error-count");
parameters.put("ntlm", "--ntlm");
parameters.put("ntlmUsername", "--ntlm-username");
parameters.put("ntlmPassword", "--ntlm-password");
parameters.put("maxSSHConnections", "--maxSSHConnections");
}
/**
* Starts Tunnel instance with options
*
* @param options Options for the Tunnel instance
*/
public synchronized Boolean start(Map options) {
try {
tunnelBinary = new TunnelBinary(options.get("binary"));
// Get path of downloaded tunnel in project directory
mutex.lock();
if (options.containsKey("infoAPIPort") && options.get("infoAPIPort").matches("^[0-9]+"))
infoAPIPortValue = Integer.parseInt(options.get("infoAPIPort"));
else
infoAPIPortValue = findAvailablePort();
t1.setDaemon(true);// now t1 is daemon thread
t1.start();// starting threads
clearTheFile();
verifyTunnelStarted(options, infoAPIPortValue);
System.out.println("tunnel Verified");
if (options.get("load-balanced") != "" && options.get("load-balanced") != null) {
if (options.get("tunnelName") == "" || options.get("tunnelName") == null) {
options.put("tunnelName", "Maven_Tunnel_LambdaTest_" + options.get("key"));
}
}
try {
for (int i = 0; i < 5; i++) {
Thread.sleep(1000);
}
} catch (Exception e) {
System.out.println(e);
}
// Check if tunnel binary path contains spaces
String tunnelBinaryPath = "";
if (options.get("binary") != null) {
tunnelBinaryPath += options.get("binary");
} else {
String path = tunnelBinary.getBinaryPath();
tunnelBinaryPath += path;
}
boolean isWhiteSpaceInBinaryPath = tunnelBinaryPath.contains(" ");
if (isWhiteSpaceInBinaryPath) {
String[] command = passParametersToTunnelV2(options);
runCommandV2(command);
} else {
String command = passParametersToTunnel(options);
runCommand(command);
}
mutex.unlock();
return true;
} catch (Exception e) {
e.printStackTrace();
return false;
}
}
public void verifyTunnelStarted(Map options, int infoAPIPortValue) throws TunnelException {
if (options.get("user") == null || options.get("user") == "" || options.get("key") == null
|| options.get("key") == "") {
tunnelFlag = false;
throw new TunnelException("Username/AccessKey Cannot Be Empty");
}
}
public synchronized void stop() throws Exception {
// Return the control if the tunnel is not even started
if (!tunnelFlag)
return;
try {
mutex.lock();
stopTunnel();
process.waitFor();
mutex.unlock();
} catch (Exception e) {
throw e;
}
}
private static Integer findAvailablePort() throws IOException {
ServerSocket s = new ServerSocket(0);
s.close();
return s.getLocalPort();
}
public static void clearTheFile() throws IOException {
FileWriter fwOb = new FileWriter("tunnel.log", false);
PrintWriter pwOb = new PrintWriter(fwOb, false);
pwOb.flush();
pwOb.close();
fwOb.close();
}
public void stopTunnel() throws TunnelException {
try {
String deleteEndpoint = "http://127.0.0.1:" + String.valueOf(infoAPIPortValue) + "/api/v1.0/stop";
CloseableHttpClient httpclient = HttpClients.createDefault();
HttpDelete httpDelete = new HttpDelete(deleteEndpoint);
HttpResponse response = httpclient.execute(httpDelete);
BufferedReader br = new BufferedReader(new InputStreamReader((response.getEntity().getContent())));
// Throw runtime exception if status code isn't 200
if (response.getStatusLine().getStatusCode() != 200) {
throw new RuntimeException("Failed : HTTP error code : " + response.getStatusLine().getStatusCode());
}
Path pathOfFile = Paths.get(".lambdatest/tunnelprocs/" +t1.port + ".txt");
boolean result = Files.deleteIfExists(pathOfFile);
if (result)
System.out.println("File is deleted");
else
System.out.println("File does not exists");
KillPort killPort = new KillPort();
killPort.killProcess(t1.port);
System.out.println("Tunnel closed successfully && Port process killed");
} catch (Exception e) {
throw new TunnelException("Tunnel with ID: " + TunnelID + " has been closed!");
}
}
// Give parameters to the tunnel for starting it in runCommand.
public String passParametersToTunnel(Map options) {
String command = "";
if (options.get("binary") != null) {
command += options.get("binary");
} else {
String binaryPath = tunnelBinary.getBinaryPath();
command += binaryPath;
}
command += " --user ";
if (options.get("user") != null)
command += options.get("user");
command += " --key ";
if (options.get("key") != null)
command += options.get("key");
command += " --infoAPIPort ";
command += String.valueOf(infoAPIPortValue);
if (options.get("load-balanced") != "" && options.get("load-balanced") != null) {
command += " --load-balanced ";
}
if (options.get("skip-upgrade") != "" && options.get("skip-upgrade") != null) {
command += " --skip-upgrade ";
}
if (options.get("basicAuth") != "" && options.get("basicAuth") != null) {
command += " --basic-auth ";
command += options.get("basicAuth");
}
if (options.get("mitm") != "" && options.get("mitm") != null) {
command += " --mitm ";
}
if (options.get("pacfile") != "" && options.get("pacfile") != null) {
command += " --pacfile ";
command += options.get("pacfile");
}
if (options.get("mTLSHosts") != "" && options.get("mTLSHosts") != null) {
command += " --mTLSHosts ";
command += options.get("mTLSHosts");
}
if (options.get("clientKey") != "" && options.get("clientKey") != null) {
command += " --clientKey ";
command += options.get("clientKey");
}
if (options.get("clientCert") != "" && options.get("clientCert") != null) {
command += " --clientCert ";
command += options.get("clientCert");
}
if (options.get("allowHosts") != "" && options.get("allowHosts") != null) {
command += " --allowHosts ";
command += options.get("allowHosts");
}
if (options.get("serverDomain") != "" && options.get("serverDomain") != null) {
command += " --server-domain ";
command += options.get("serverDomain");
}
if (options.get("verbose") != "" && options.get("verbose") != null) {
command += " --verbose ";
}
if (options.get("usePrivateIp") != "" && options.get("usePrivateIp") != null) {
command += " --use-private-ip ";
}
if (options.get("retry-proxy-error") != "" && options.get("retry-proxy-error") != null) {
command += " --retry-proxy-error ";
}
if (options.get("retry-proxy-error-count") != "" && options.get("retry-proxy-error-count") != null) {
command += " --retry-proxy-error-count ";
command += options.get("retry-proxy-error-count");
}
if (options.get("ntlm") != "" && options.get("ntlm") != null) {
command += " --ntlm ";
}
if (options.get("ntlmUsername") != "" && options.get("ntlmUsername") != null) {
command += " --ntlm-username ";
command += options.get("ntlmUsername");
}
if (options.get("ntlmPassword") != "" && options.get("ntlmPassword") != null) {
command += " --ntlm-password ";
command += options.get("ntlmPassword");
}
if (options.get("maxSSHConnections") != "" && options.get("maxSSHConnections") != null) {
command += " --maxSSHConnections ";
command += options.get("maxSSHConnections");
}
if (t1.port != null) {
command += " --callbackURL http://127.0.0.1:" + String.valueOf(t1.port);
}
for (Map.Entry opt : options.entrySet()) {
String parameter = opt.getKey().trim();
if (IGNORE_KEYS.contains(parameter)) {
continue;
}
if (parameters.get(parameter) != null) {
command += " " + parameters.get(parameter) + " ";
if (opt.getValue() != null) {
command += opt.getValue().trim();
}
}
}
return command;
}
// Give parameters to the tunnel for starting it in runCommand.
public String[] passParametersToTunnelV2(Map options) {
ArrayList commandArray = new ArrayList();
if (options.get("binary") != null) {
commandArray.add(options.get("binary"));
} else {
commandArray.add(tunnelBinary.getBinaryPath());
}
commandArray.add("--user");
if (options.get("user") != null)
commandArray.add(options.get("user"));
commandArray.add("--key");
if (options.get("key") != null)
commandArray.add(options.get("key"));
commandArray.add("--infoAPIPort");
commandArray.add(String.valueOf(infoAPIPortValue));
if (options.get("load-balanced") != "" && options.get("load-balanced") != null) {
commandArray.add("--load-balanced");
}
if (options.get("skip-upgrade") != "" && options.get("skip-upgrade") != null) {
commandArray.add("--skip-upgrade");
}
if (options.get("basicAuth") != "" && options.get("basicAuth") != null) {
commandArray.add("--basic-auth");
commandArray.add(options.get("basicAuth"));
}
if (options.get("mitm") != "" && options.get("mitm") != null) {
commandArray.add("--mitm");
}
if (options.get("pacfile") != "" && options.get("pacfile") != null) {
commandArray.add("--pacfile");
commandArray.add(options.get("pacfile"));
}
if (options.get("mTLSHosts") != "" && options.get("mTLSHosts") != null) {
commandArray.add("--mTLSHosts");
commandArray.add(options.get("mTLSHosts"));
}
if (options.get("clientKey") != "" && options.get("clientKey") != null) {
commandArray.add("--clientKey");
commandArray.add(options.get("clientKey"));
}
if (options.get("clientCert") != "" && options.get("clientCert") != null) {
commandArray.add("--clientCert");
commandArray.add(options.get("clientCert"));
}
if (options.get("allowHosts") != "" && options.get("allowHosts") != null) {
commandArray.add("--allowHosts");
commandArray.add(options.get("allowHosts"));
}
if (options.get("serverDomain") != "" && options.get("serverDomain") != null) {
commandArray.add("--server-domain");
commandArray.add(options.get("serverDomain"));
}
if (options.get("verbose") != "" && options.get("verbose") != null) {
commandArray.add("--verbose");
}
if (options.get("usePrivateIp") != "" && options.get("usePrivateIp") != null) {
commandArray.add("--use-private-ip");
}
if (options.get("retry-proxy-error") != "" && options.get("retry-proxy-error") != null) {
commandArray.add("--retry-proxy-error");
}
if (options.get("retry-proxy-error-count") != "" && options.get("retry-proxy-error-count") != null) {
commandArray.add("--retry-proxy-error-count");
commandArray.add(options.get("retry-proxy-error-count"));
}
if (options.get("ntlm") != "" && options.get("ntlm") != null) {
commandArray.add("--ntlm");
}
if (options.get("ntlmUsername") != "" && options.get("ntlmUsername") != null) {
commandArray.add("--ntlm-username");
commandArray.add(options.get("ntlmUsername"));
}
if (options.get("ntlmPassword") != "" && options.get("ntlmPassword") != null) {
commandArray.add("--ntlm-password");
commandArray.add(options.get("ntlmPassword"));
}
if (options.get("maxSSHConnections") != "" && options.get("maxSSHConnections") != null) {
commandArray.add("--maxSSHConnections");
commandArray.add(options.get("maxSSHConnections"));
}
if (t1.port != null) {
commandArray.add("--callbackURL");
commandArray.add("http://127.0.0.1:" + String.valueOf(t1.port));
}
for (Map.Entry opt : options.entrySet()) {
String parameter = opt.getKey().trim();
if (IGNORE_KEYS.contains(parameter)) {
continue;
}
if (parameters.get(parameter) != null) {
commandArray.add(parameters.get(parameter));
if (opt.getValue() != null) {
commandArray.add(opt.getValue().trim());
}
}
}
String[] commandV2 = commandArray.toArray(new String[commandArray.size()]);
return commandV2;
}
public void runCommand(String command) throws IOException {
try {
// ProcessBuilder processBuilder = new ProcessBuilder(command);
// System.out.println("Command String: " + command);
Runtime run = Runtime.getRuntime();
process = run.exec(command);
Boolean update = false;
long start = System.currentTimeMillis();
long end = start;
BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
String line = null;
try {
while ((line = reader.readLine()) != null) {
if (line.contains("Downloading update")) {
update = true;
} else {
if (update) {
System.out.println("Tunnel is updated. restarting...");
runCommand(command);
}
try {
if (t1.port != null) {
BufferedReader br = new BufferedReader(
new FileReader(String.valueOf(".lambdatest/tunnelprocs/" + t1.port) + ".txt"));
if (br.readLine() != null) {
tunnelFlag = true;
System.out.println("Tunnel Started Successfully");
break;
}
}
} catch (Exception e) {
// System.out.println("Not found any file");
}
}
}
} catch (Exception e) {
e.printStackTrace();
}
} catch (IOException e) {
e.printStackTrace();
}
return;
}
public void runCommandV2(String[] command) throws IOException {
try {
// ProcessBuilder processBuilder = new ProcessBuilder(command);
Runtime run = Runtime.getRuntime();
process = run.exec(command);
Boolean update = false;
long start = System.currentTimeMillis();
long end = start;
BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
String line = null;
try {
while ((line = reader.readLine()) != null) {
if (line.contains("Downloading update")) {
update = true;
} else {
if (update) {
System.out.println("Tunnel is updated. restarting...");
runCommandV2(command);
}
try {
if (t1.port != null) {
BufferedReader br = new BufferedReader(
new FileReader(String.valueOf(".lambdatest/tunnelprocs/" + t1.port) + ".txt"));
if (br.readLine() != null) {
tunnelFlag = true;
System.out.println("Tunnel Started Successfully");
break;
}
}
} catch (Exception e) {
// System.out.println("Not found any file");
}
}
}
} catch (Exception e) {
e.printStackTrace();
}
} catch (IOException e) {
e.printStackTrace();
}
return;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy