From 5aa4a68be6f48ae0c7a72b1a4368e5f258c2c92f Mon Sep 17 00:00:00 2001 From: FongMi Date: Thu, 15 Feb 2024 17:09:34 +0800 Subject: [PATCH 1/2] Add 302 auth check --- catvod/src/main/java/com/github/catvod/net/OkHttp.java | 2 +- .../github/catvod/net/interceptor/DefaultInterceptor.java | 8 +++++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/catvod/src/main/java/com/github/catvod/net/OkHttp.java b/catvod/src/main/java/com/github/catvod/net/OkHttp.java index 1e5261d32..ff10e20fe 100644 --- a/catvod/src/main/java/com/github/catvod/net/OkHttp.java +++ b/catvod/src/main/java/com/github/catvod/net/OkHttp.java @@ -132,7 +132,7 @@ public class OkHttp { } private static OkHttpClient.Builder getBuilder() { - OkHttpClient.Builder builder = new OkHttpClient.Builder().addInterceptor(new DefaultInterceptor()).connectTimeout(TIMEOUT, TimeUnit.MILLISECONDS).readTimeout(TIMEOUT, TimeUnit.MILLISECONDS).writeTimeout(TIMEOUT, TimeUnit.MILLISECONDS).dns(dns()).hostnameVerifier((hostname, session) -> true).sslSocketFactory(new SSLCompat(), SSLCompat.TM); + OkHttpClient.Builder builder = new OkHttpClient.Builder().addNetworkInterceptor(new DefaultInterceptor()).connectTimeout(TIMEOUT, TimeUnit.MILLISECONDS).readTimeout(TIMEOUT, TimeUnit.MILLISECONDS).writeTimeout(TIMEOUT, TimeUnit.MILLISECONDS).dns(dns()).hostnameVerifier((hostname, session) -> true).sslSocketFactory(new SSLCompat(), SSLCompat.TM); builder.proxySelector(get().proxy ? selector() : defaultSelector); return builder; } diff --git a/catvod/src/main/java/com/github/catvod/net/interceptor/DefaultInterceptor.java b/catvod/src/main/java/com/github/catvod/net/interceptor/DefaultInterceptor.java index 9279ecfac..4aeeb4db6 100644 --- a/catvod/src/main/java/com/github/catvod/net/interceptor/DefaultInterceptor.java +++ b/catvod/src/main/java/com/github/catvod/net/interceptor/DefaultInterceptor.java @@ -27,6 +27,7 @@ public class DefaultInterceptor implements Interceptor { public Response intercept(@NonNull Chain chain) throws IOException { Response response = chain.proceed(getRequest(chain.request())); String encoding = response.header(HttpHeaders.CONTENT_ENCODING); + if (response.isRedirect() && response.header(HttpHeaders.LOCATION) != null) checkAuth(response); if (response.body() == null || encoding == null || !encoding.equals("deflate")) return response; InflaterInputStream is = new InflaterInputStream(response.body().byteStream(), new Inflater(true)); return response.newBuilder().headers(response.headers()).body(new ResponseBody() { @@ -49,7 +50,7 @@ public class DefaultInterceptor implements Interceptor { }).build(); } - private Request getRequest(@NonNull Request request) { + private Request getRequest(Request request) { URI uri = request.url().uri(); String url = request.url().toString(); Request.Builder builder = request.newBuilder(); @@ -59,4 +60,9 @@ public class DefaultInterceptor implements Interceptor { if (uri.getUserInfo() != null) builder.header(HttpHeaders.AUTHORIZATION, Util.basic(uri.getUserInfo())); return builder.build(); } + + private void checkAuth(Response response) { + URI uri = URI.create(response.header(HttpHeaders.LOCATION)); + if (uri.getUserInfo() != null) response.header(HttpHeaders.AUTHORIZATION, Util.basic(uri.getUserInfo())); + } } From a137f81c831b4cb48c4b0e42a2b6097f81b59b7d Mon Sep 17 00:00:00 2001 From: FongMi Date: Thu, 15 Feb 2024 17:43:27 +0800 Subject: [PATCH 2/2] Clean code --- .../net/interceptor/DefaultInterceptor.java | 42 +++++++++++-------- 1 file changed, 24 insertions(+), 18 deletions(-) diff --git a/catvod/src/main/java/com/github/catvod/net/interceptor/DefaultInterceptor.java b/catvod/src/main/java/com/github/catvod/net/interceptor/DefaultInterceptor.java index 4aeeb4db6..0c4d534bd 100644 --- a/catvod/src/main/java/com/github/catvod/net/interceptor/DefaultInterceptor.java +++ b/catvod/src/main/java/com/github/catvod/net/interceptor/DefaultInterceptor.java @@ -26,9 +26,31 @@ public class DefaultInterceptor implements Interceptor { @Override public Response intercept(@NonNull Chain chain) throws IOException { Response response = chain.proceed(getRequest(chain.request())); + String location = response.header(HttpHeaders.LOCATION); String encoding = response.header(HttpHeaders.CONTENT_ENCODING); - if (response.isRedirect() && response.header(HttpHeaders.LOCATION) != null) checkAuth(response); - if (response.body() == null || encoding == null || !encoding.equals("deflate")) return response; + if (response.isRedirect() && location != null) checkAuth(response, location); + if (response.body() != null && "deflate".equals(encoding)) return deflate(response); + return response; + } + + private Request getRequest(Request request) { + URI uri = request.url().uri(); + String url = request.url().toString(); + Request.Builder builder = request.newBuilder(); + boolean local = url.contains(":" + Proxy.getPort() + "/"); + if (url.contains("+") && local) builder.url(url.replace("+", "%2B")); + if (url.contains("gitcode.net")) builder.header(HttpHeaders.USER_AGENT, Util.CHROME); + if (uri.getUserInfo() != null) builder.header(HttpHeaders.AUTHORIZATION, Util.basic(uri.getUserInfo())); + return builder.build(); + } + + private void checkAuth(Response response, String location) { + URI uri = URI.create(location); + if (uri.getUserInfo() == null) return; + response.header(HttpHeaders.AUTHORIZATION, Util.basic(uri.getUserInfo())); + } + + private Response deflate(Response response) { InflaterInputStream is = new InflaterInputStream(response.body().byteStream(), new Inflater(true)); return response.newBuilder().headers(response.headers()).body(new ResponseBody() { @Nullable @@ -49,20 +71,4 @@ public class DefaultInterceptor implements Interceptor { } }).build(); } - - private Request getRequest(Request request) { - URI uri = request.url().uri(); - String url = request.url().toString(); - Request.Builder builder = request.newBuilder(); - boolean local = url.contains(":" + Proxy.getPort() + "/"); - if (url.contains("+") && local) builder.url(url.replace("+", "%2B")); - if (url.contains("gitcode.net")) builder.header(HttpHeaders.USER_AGENT, Util.CHROME); - if (uri.getUserInfo() != null) builder.header(HttpHeaders.AUTHORIZATION, Util.basic(uri.getUserInfo())); - return builder.build(); - } - - private void checkAuth(Response response) { - URI uri = URI.create(response.header(HttpHeaders.LOCATION)); - if (uri.getUserInfo() != null) response.header(HttpHeaders.AUTHORIZATION, Util.basic(uri.getUserInfo())); - } }