100 lines
3.9 KiB
Java
100 lines
3.9 KiB
Java
package com.rarchives.ripme.utils;
|
|
|
|
import java.net.Authenticator;
|
|
import java.net.PasswordAuthentication;
|
|
import java.util.Map;
|
|
import java.util.HashMap;
|
|
|
|
/**
|
|
* Proxy/Socks setter
|
|
*/
|
|
public class Proxy {
|
|
private Proxy() {
|
|
}
|
|
|
|
/**
|
|
* Parse the proxy server settings from string, using the format
|
|
* [user:password]@host[:port].
|
|
*
|
|
* @param fullproxy the string to parse
|
|
* @return HashMap containing proxy server, port, user and password
|
|
*/
|
|
private static Map<String, String> parseServer(String fullproxy) {
|
|
Map<String, String> proxy = new HashMap<String, String>();
|
|
|
|
if (fullproxy.lastIndexOf("@") != -1) {
|
|
int sservli = fullproxy.lastIndexOf("@");
|
|
String userpw = fullproxy.substring(0, sservli);
|
|
String[] usersplit = userpw.split(":");
|
|
proxy.put("user", usersplit[0]);
|
|
proxy.put("password", usersplit[1]);
|
|
fullproxy = fullproxy.substring(sservli + 1);
|
|
}
|
|
String[] servsplit = fullproxy.split(":");
|
|
if (servsplit.length == 2) {
|
|
proxy.put("port", servsplit[1]);
|
|
}
|
|
proxy.put("server", servsplit[0]);
|
|
return proxy;
|
|
}
|
|
|
|
/**
|
|
* Set a HTTP Proxy.
|
|
* WARNING: Authenticated HTTP Proxy won't work from jdk1.8.111 unless
|
|
* passing the flag -Djdk.http.auth.tunneling.disabledSchemes="" to java
|
|
* see https://stackoverflow.com/q/41505219
|
|
*
|
|
* @param fullproxy the proxy, using format [user:password]@host[:port]
|
|
*/
|
|
public static void setHTTPProxy(String fullproxy) {
|
|
Map<String, String> proxyServer = parseServer(fullproxy);
|
|
|
|
if (proxyServer.get("user") != null && proxyServer.get("password") != null) {
|
|
Authenticator.setDefault(new Authenticator(){
|
|
protected PasswordAuthentication getPasswordAuthentication(){
|
|
PasswordAuthentication p = new PasswordAuthentication(proxyServer.get("user"), proxyServer.get("password").toCharArray());
|
|
return p;
|
|
}
|
|
});
|
|
System.setProperty("http.proxyUser", proxyServer.get("user"));
|
|
System.setProperty("http.proxyPassword", proxyServer.get("password"));
|
|
System.setProperty("https.proxyUser", proxyServer.get("user"));
|
|
System.setProperty("https.proxyPassword", proxyServer.get("password"));
|
|
}
|
|
|
|
if (proxyServer.get("port") != null) {
|
|
System.setProperty("http.proxyPort", proxyServer.get("port"));
|
|
System.setProperty("https.proxyPort", proxyServer.get("port"));
|
|
}
|
|
|
|
System.setProperty("http.proxyHost", proxyServer.get("server"));
|
|
System.setProperty("https.proxyHost", proxyServer.get("server"));
|
|
}
|
|
|
|
/**
|
|
* Set a Socks Proxy Server (globally).
|
|
*
|
|
* @param fullsocks the socks server, using format [user:password]@host[:port]
|
|
*/
|
|
public static void setSocks(String fullsocks) {
|
|
|
|
Map<String, String> socksServer = parseServer(fullsocks);
|
|
if (socksServer.get("user") != null && socksServer.get("password") != null) {
|
|
Authenticator.setDefault(new Authenticator(){
|
|
protected PasswordAuthentication getPasswordAuthentication(){
|
|
PasswordAuthentication p = new PasswordAuthentication(socksServer.get("user"), socksServer.get("password").toCharArray());
|
|
return p;
|
|
}
|
|
});
|
|
System.setProperty("java.net.socks.username", socksServer.get("user"));
|
|
System.setProperty("java.net.socks.password", socksServer.get("password"));
|
|
}
|
|
if (socksServer.get("port") != null) {
|
|
System.setProperty("socksProxyPort", socksServer.get("port"));
|
|
}
|
|
|
|
System.setProperty("socksProxyHost", socksServer.get("server"));
|
|
}
|
|
|
|
}
|