lan/HMCL
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

implement yushijinhun/authlib-injector#78

Browse Source
This commit is contained in:
yushijinhun
2020-08-12 21:49:46 +08:00
committed by Yuhui Huang
parent 7f674f237b
commit 2f10948239
1 changed files with 18 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
HMCLCore/src/main/java/org/jackhuang/hmcl/auth/authlibinjector/AuthlibInjectorServer.java
View File

@@ -27,9 +27,7 @@ import static org.jackhuang.hmcl.util.io.IOUtils.readFullyWithoutClosing;
import java.io.IOException; import java.io.IOException;
import java.lang.reflect.Type; import java.lang.reflect.Type;
import java.net.HttpURLConnection; import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL; import java.net.URL;
import java.net.URLConnection;
import java.util.LinkedHashMap; import java.util.LinkedHashMap;
import java.util.Map; import java.util.Map;
import java.util.Optional; import java.util.Optional;
@@ -56,30 +54,25 @@ import javafx.beans.Observable;
@JsonAdapter(AuthlibInjectorServer.Deserializer.class) @JsonAdapter(AuthlibInjectorServer.Deserializer.class)
public class AuthlibInjectorServer implements Observable { public class AuthlibInjectorServer implements Observable {
private static final int MAX_REDIRECTS = 5;
private static final Gson GSON = new GsonBuilder().create(); private static final Gson GSON = new GsonBuilder().create();
public static AuthlibInjectorServer locateServer(String url) throws IOException { public static AuthlibInjectorServer locateServer(String url) throws IOException {
try { try {
url = parseInputUrl(url); url = addHttpsIfMissing(url);
HttpURLConnection conn; HttpURLConnection conn = (HttpURLConnection) new URL(url).openConnection();
int redirectCount = 0;
for (; ; ) { String ali = conn.getHeaderField("x-authlib-injector-api-location");
conn = (HttpURLConnection) new URL(url).openConnection(); if (ali != null) {
Optional<String> ali = getApiLocationIndication(conn); URL absoluteAli = new URL(conn.getURL(), ali);
if (ali.isPresent()) { if (!urlEqualsIgnoreSlash(url, absoluteAli.toString())) {
conn.disconnect(); conn.disconnect();
url = ali.get(); url = absoluteAli.toString();
if (redirectCount >= MAX_REDIRECTS) { conn = (HttpURLConnection) absoluteAli.openConnection();
throw new IOException("Exceeded maximum number of redirects (" + MAX_REDIRECTS + ")");
}
redirectCount++;
LOG.info("Redirect API root to: " + url);
} else {
break;
} }
} }
if (!url.endsWith("/"))
url += "/";
try { try {
AuthlibInjectorServer server = new AuthlibInjectorServer(url); AuthlibInjectorServer server = new AuthlibInjectorServer(url);
@@ -93,49 +86,20 @@ public class AuthlibInjectorServer implements Observable {
} }
} }
private static Optional<String> getApiLocationIndication(URLConnection conn) { private static String addHttpsIfMissing(String url) {
return Optional.ofNullable(conn.getHeaderFields().get("X-Authlib-Injector-API-Location"))
.flatMap(list -> list.isEmpty() ? Optional.empty() : Optional.of(list.get(0)))
.flatMap(indication -> {
String currentUrl = appendSuffixSlash(conn.getURL().toString());
String newUrl;
try {
newUrl = appendSuffixSlash(new URL(conn.getURL(), indication).toString());
} catch (MalformedURLException e) {
LOG.warning("Failed to resolve absolute ALI, the header is [" + indication + "]. Ignore it.");
return Optional.empty();
}
if (newUrl.equals(currentUrl)) {
return Optional.empty();
} else {
return Optional.of(newUrl);
}
});
}
private static String parseInputUrl(String url) {
String lowercased = url.toLowerCase(); String lowercased = url.toLowerCase();
if (!lowercased.startsWith("http://") && !lowercased.startsWith("https://")) { if (!lowercased.startsWith("http://") && !lowercased.startsWith("https://")) {
url = "https://" + url; url = "https://" + url;
} }
url = appendSuffixSlash(url);
return url; return url;
} }
private static String appendSuffixSlash(String url) { private static boolean urlEqualsIgnoreSlash(String a, String b) {
if (!url.endsWith("/")) { if (!a.endsWith("/"))
return url + "/"; a += "/";
} else { if (!b.endsWith("/"))
return url; b += "/";
} return a.equals(b);
}
public static AuthlibInjectorServer fetchServerInfo(String url) throws IOException {
AuthlibInjectorServer server = new AuthlibInjectorServer(url);
server.refreshMetadata();
return server;
} }
private String url; private String url;