AstralTransRocketries/authlib-injector
1
0
Fork 0
mirror of https://github.com/unmojang/authlib-injector.git synced 2025-09-30 06:40:47 -04:00
Code Issues Packages Projects Releases Wiki Activity

Support ALI(Api Location Indication) (#18)

Browse Source
This commit is contained in:
yushijinhun 2018-10-05 21:08:48 +08:00
parent 58d0b0e7a1
commit 36f3ef61eb
No known key found for this signature in database
GPG Key ID: 5BC167F73EA558E4
1 changed files with 63 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
src/main/java/moe/yushi/authlibinjector/AuthlibInjector.java
View File

@ -3,12 +3,16 @@ package moe.yushi.authlibinjector;
import static java.nio.charset.StandardCharsets.UTF_8;
import static java.util.Optional.empty;
import static java.util.Optional.of;
import static moe.yushi.authlibinjector.util.IOUtils.asBytes;
import static moe.yushi.authlibinjector.util.IOUtils.asString;
import static moe.yushi.authlibinjector.util.IOUtils.getURL;
import static moe.yushi.authlibinjector.util.IOUtils.removeNewLines;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.lang.instrument.ClassFileTransformer;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.util.Base64;
import java.util.Optional;
import java.util.concurrent.atomic.AtomicBoolean;
@ -67,8 +71,12 @@ public final class AuthlibInjector {
*/
public static final String PROP_SIDE = "authlibinjector.side";
public static final String PROP_ALI_REDIRECT_LIMIT = "authlibinjector.ali.redirectLimit";
// ====
private static final int REDIRECT_LIMIT = Integer.getInteger(PROP_ALI_REDIRECT_LIMIT, 5);
private AuthlibInjector() {}
private static AtomicBoolean booted = new AtomicBoolean(false);
@ -105,7 +113,6 @@ public final class AuthlibInjector {
String apiRoot = System.getProperty(PROP_API_ROOT);
if (apiRoot == null) return empty();
// for further use
ExecutionEnvironment side = detectSide();
Logging.LAUNCH.fine("Detected side: " + side);
@ -117,8 +124,40 @@ public final class AuthlibInjector {
Optional<String> prefetched = getPrefetchedResponse();
if (!prefetched.isPresent()) {
try {
metadataResponse = asString(getURL(apiRoot));
HttpURLConnection connection;
boolean redirectAllowed = side == ExecutionEnvironment.SERVER;
int redirectCount = 0;
for (;;) {
connection = (HttpURLConnection) new URL(apiRoot).openConnection();
Optional<String> ali = getApiLocationIndication(connection);
if (ali.isPresent()) {
if (!redirectAllowed) {
Logging.CONFIG.warning("Redirect is not allowed, ignoring ALI: " + ali.get());
break;
}
connection.disconnect();
apiRoot = ali.get();
if (redirectCount >= REDIRECT_LIMIT) {
Logging.CONFIG.severe("Exceeded maximum number of redirects (" + REDIRECT_LIMIT + "), refusing to redirect to: " + apiRoot);
throw new InjectorInitializationException();
}
redirectCount++;
Logging.CONFIG.info("Redirect to: " + apiRoot);
warnIfHttp(apiRoot);
} else {
break;
}
}
try {
metadataResponse = asString(asBytes(connection.getInputStream()));
} finally {
connection.disconnect();
}
} catch (IOException e) {
Logging.CONFIG.severe("Failed to fetch metadata: " + e);
throw new InjectorInitializationException(e);
@ -175,6 +214,27 @@ public final class AuthlibInjector {
return url;
}
private static Optional<String> getApiLocationIndication(URLConnection conn) {
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) {
Logging.CONFIG.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 ExecutionEnvironment detectSide() {
String specifiedSide = System.getProperty(PROP_SIDE);
if (specifiedSide != null) {