1package eu.siacs.conversations.http;
2
3import android.os.Build;
4import android.util.Log;
5
6import org.apache.http.conn.ssl.StrictHostnameVerifier;
7
8import java.io.IOException;
9import java.io.InputStream;
10import java.net.InetAddress;
11import java.net.InetSocketAddress;
12import java.net.Proxy;
13import java.net.UnknownHostException;
14import java.security.KeyManagementException;
15import java.security.NoSuchAlgorithmException;
16import java.util.ArrayList;
17import java.util.List;
18import java.util.concurrent.Executor;
19import java.util.concurrent.Executors;
20import java.util.concurrent.TimeUnit;
21
22import javax.net.ssl.SSLSocketFactory;
23import javax.net.ssl.X509TrustManager;
24
25import eu.siacs.conversations.BuildConfig;
26import eu.siacs.conversations.Config;
27import eu.siacs.conversations.entities.Account;
28import eu.siacs.conversations.entities.Message;
29import eu.siacs.conversations.services.AbstractConnectionManager;
30import eu.siacs.conversations.services.XmppConnectionService;
31import eu.siacs.conversations.utils.TLSSocketFactory;
32import okhttp3.HttpUrl;
33import okhttp3.OkHttpClient;
34import okhttp3.Request;
35import okhttp3.ResponseBody;
36
37public class HttpConnectionManager extends AbstractConnectionManager {
38
39 private final List<HttpDownloadConnection> downloadConnections = new ArrayList<>();
40 private final List<HttpUploadConnection> uploadConnections = new ArrayList<>();
41
42 public static final Executor EXECUTOR = Executors.newFixedThreadPool(4);
43
44 public static final OkHttpClient OK_HTTP_CLIENT;
45
46 static {
47 OK_HTTP_CLIENT = new OkHttpClient.Builder()
48 .addInterceptor(chain -> {
49 final Request original = chain.request();
50 final Request modified = original.newBuilder()
51 .header("User-Agent", getUserAgent())
52 .build();
53 return chain.proceed(modified);
54 })
55 .build();
56 }
57
58
59 public static String getUserAgent() {
60 return String.format("%s/%s", BuildConfig.APP_NAME, BuildConfig.VERSION_NAME);
61 }
62
63 public HttpConnectionManager(XmppConnectionService service) {
64 super(service);
65 }
66
67 public static Proxy getProxy() {
68 final InetAddress localhost;
69 try {
70 localhost = InetAddress.getByAddress(new byte[]{127, 0, 0, 1});
71 } catch (final UnknownHostException e) {
72 throw new IllegalStateException(e);
73 }
74 if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) {
75 return new Proxy(Proxy.Type.SOCKS, new InetSocketAddress(localhost, 9050));
76 } else {
77 return new Proxy(Proxy.Type.HTTP, new InetSocketAddress(localhost, 8118));
78 }
79 }
80
81 public void createNewDownloadConnection(Message message) {
82 this.createNewDownloadConnection(message, false);
83 }
84
85 public void createNewDownloadConnection(final Message message, boolean interactive) {
86 synchronized (this.downloadConnections) {
87 for (HttpDownloadConnection connection : this.downloadConnections) {
88 if (connection.getMessage() == message) {
89 Log.d(Config.LOGTAG, message.getConversation().getAccount().getJid().asBareJid() + ": download already in progress");
90 return;
91 }
92 }
93 final HttpDownloadConnection connection = new HttpDownloadConnection(message, this);
94 connection.init(interactive);
95 this.downloadConnections.add(connection);
96 }
97 }
98
99 public void createNewUploadConnection(final Message message, boolean delay) {
100 synchronized (this.uploadConnections) {
101 for (HttpUploadConnection connection : this.uploadConnections) {
102 if (connection.getMessage() == message) {
103 Log.d(Config.LOGTAG, message.getConversation().getAccount().getJid().asBareJid() + ": upload already in progress");
104 return;
105 }
106 }
107 HttpUploadConnection connection = new HttpUploadConnection(message, Method.determine(message.getConversation().getAccount()), this);
108 connection.init(delay);
109 this.uploadConnections.add(connection);
110 }
111 }
112
113 void finishConnection(HttpDownloadConnection connection) {
114 synchronized (this.downloadConnections) {
115 this.downloadConnections.remove(connection);
116 }
117 }
118
119 void finishUploadConnection(HttpUploadConnection httpUploadConnection) {
120 synchronized (this.uploadConnections) {
121 this.uploadConnections.remove(httpUploadConnection);
122 }
123 }
124
125 OkHttpClient buildHttpClient(final HttpUrl url, final Account account, boolean interactive) {
126 return buildHttpClient(url, account, 30, interactive);
127 }
128
129 OkHttpClient buildHttpClient(final HttpUrl url, final Account account, int readTimeout, boolean interactive) {
130 final String slotHostname = url.host();
131 final boolean onionSlot = slotHostname.endsWith(".onion");
132 final OkHttpClient.Builder builder = OK_HTTP_CLIENT.newBuilder();
133 builder.writeTimeout(30, TimeUnit.SECONDS);
134 builder.readTimeout(readTimeout, TimeUnit.SECONDS);
135 setupTrustManager(builder, interactive);
136 if (mXmppConnectionService.useTorToConnect() || account.isOnion() || onionSlot) {
137 builder.proxy(HttpConnectionManager.getProxy()).build();
138 }
139 return builder.build();
140 }
141
142 private void setupTrustManager(final OkHttpClient.Builder builder, final boolean interactive) {
143 final X509TrustManager trustManager;
144 if (interactive) {
145 trustManager = mXmppConnectionService.getMemorizingTrustManager().getInteractive();
146 } else {
147 trustManager = mXmppConnectionService.getMemorizingTrustManager().getNonInteractive();
148 }
149 try {
150 final SSLSocketFactory sf = new TLSSocketFactory(new X509TrustManager[]{trustManager}, mXmppConnectionService.getRNG());
151 builder.sslSocketFactory(sf, trustManager);
152 builder.hostnameVerifier(new StrictHostnameVerifier());
153 } catch (final KeyManagementException | NoSuchAlgorithmException ignored) {
154 }
155 }
156
157 public static InputStream open(final String url, final boolean tor) throws IOException {
158 return open(HttpUrl.get(url), tor);
159 }
160
161 public static InputStream open(final HttpUrl httpUrl, final boolean tor) throws IOException {
162 final OkHttpClient.Builder builder = OK_HTTP_CLIENT.newBuilder();
163 if (tor) {
164 builder.proxy(HttpConnectionManager.getProxy()).build();
165 }
166 final OkHttpClient client = builder.build();
167 final Request request = new Request.Builder().get().url(httpUrl).build();
168 final ResponseBody body = client.newCall(request).execute().body();
169 if (body == null) {
170 throw new IOException("No response body found");
171 }
172 return body.byteStream();
173 }
174}