1package eu.siacs.conversations.utils;
2
3import android.content.Context;
4import android.support.annotation.NonNull;
5import android.util.Log;
6
7import java.io.IOException;
8import java.net.InetAddress;
9import java.util.ArrayList;
10import java.util.Collections;
11import java.util.List;
12
13import de.measite.minidns.DNSClient;
14import de.measite.minidns.DNSName;
15import de.measite.minidns.dnssec.DNSSECResultNotAuthenticException;
16import de.measite.minidns.hla.DnssecResolverApi;
17import de.measite.minidns.hla.ResolverApi;
18import de.measite.minidns.hla.ResolverResult;
19import de.measite.minidns.record.A;
20import de.measite.minidns.record.AAAA;
21import de.measite.minidns.record.Data;
22import de.measite.minidns.record.InternetAddressRR;
23import de.measite.minidns.record.SRV;
24import eu.siacs.conversations.Config;
25
26public class Resolver {
27
28 private static final String DIRECT_TLS_SERVICE = "_xmpps-client";
29 private static final String STARTTLS_SERICE = "_xmpp-client";
30
31
32 public static void registerLookupMechanism(Context context) {
33 DNSClient.addDnsServerLookupMechanism(new AndroidUsingLinkProperties(context));
34 }
35
36 public static List<Result> resolve(String domain) {
37 List<Result> results = new ArrayList<>();
38 try {
39 results.addAll(resolveSrv(domain,true));
40 } catch (IOException e) {
41 Log.d(Config.LOGTAG,Resolver.class.getSimpleName()+": "+e.getMessage());
42 }
43 try {
44 results.addAll(resolveSrv(domain,false));
45 } catch (IOException e) {
46 Log.d(Config.LOGTAG,Resolver.class.getSimpleName()+": "+e.getMessage());
47 }
48 if (results.size() == 0) {
49 results.add(Result.createDefault(domain));
50 }
51 Collections.sort(results);
52 Log.d(Config.LOGTAG,Resolver.class.getSimpleName()+": "+results.toString());
53 return results;
54 }
55
56 private static List<Result> resolveSrv(String domain, final boolean directTls) throws IOException {
57 if (Thread.interrupted()) {
58 return Collections.emptyList();
59 }
60 DNSName dnsName = DNSName.from((directTls ? DIRECT_TLS_SERVICE : STARTTLS_SERICE)+"._tcp."+domain);
61 ResolverResult<SRV> result = resolveWithFallback(dnsName,SRV.class);
62 List<Result> results = new ArrayList<>();
63 for(SRV record : result.getAnswersOrEmptySet()) {
64 final boolean addedIPv4 = results.addAll(resolveIp(record,A.class,result.isAuthenticData(),directTls));
65 results.addAll(resolveIp(record,AAAA.class,result.isAuthenticData(),directTls));
66 if (!addedIPv4 && !Thread.interrupted()) {
67 Result resolverResult = Result.fromRecord(record, directTls);
68 resolverResult.authenticated = resolverResult.isAuthenticated();
69 results.add(resolverResult);
70 }
71 }
72 return results;
73 }
74
75 private static <D extends InternetAddressRR> List<Result> resolveIp(SRV srv, Class<D> type, boolean authenticated, boolean directTls) {
76 if (Thread.interrupted()) {
77 return Collections.emptyList();
78 }
79 List<Result> list = new ArrayList<>();
80 try {
81 ResolverResult<D> results = resolveWithFallback(srv.name,type);
82 for (D record : results.getAnswersOrEmptySet()) {
83 Result resolverResult = Result.fromRecord(srv, directTls);
84 resolverResult.authenticated = results.isAuthenticData() && authenticated;
85 resolverResult.ip = record.getInetAddress();
86 list.add(resolverResult);
87 }
88 } catch (Throwable t) {
89 Log.d(Config.LOGTAG,Resolver.class.getSimpleName()+": error resolving "+type.getSimpleName()+" "+t.getMessage());
90 }
91 return list;
92 }
93
94 private static <D extends Data> ResolverResult<D> resolveWithFallback(DNSName dnsName, Class<D> type) throws IOException {
95 try {
96 final ResolverResult<D> r = DnssecResolverApi.INSTANCE.resolveDnssecReliable(dnsName, type);
97 if (r.wasSuccessful()) {
98 if (r.getAnswers().isEmpty() && type.equals(SRV.class)) {
99 Log.d(Config.LOGTAG,Resolver.class.getSimpleName()+": resolving SRV records of "+dnsName.toString()+" with DNSSEC yielded empty result");
100 }
101 return r;
102 }
103 Log.d(Config.LOGTAG,Resolver.class.getSimpleName()+": error resolving "+type.getSimpleName()+" with DNSSEC. Trying DNS instead.",r.getResolutionUnsuccessfulException());
104 } catch (DNSSECResultNotAuthenticException e) {
105 Log.d(Config.LOGTAG,Resolver.class.getSimpleName()+": error resolving "+type.getSimpleName()+" with DNSSEC. Trying DNS instead.",e);
106 } catch (IOException e) {
107 throw e;
108 } catch (Throwable throwable) {
109 Log.d(Config.LOGTAG,Resolver.class.getSimpleName()+": error resolving "+type.getSimpleName()+" with DNSSEC. Trying DNS instead.",throwable);
110 }
111 return ResolverApi.INSTANCE.resolve(dnsName, type);
112 }
113
114 public static class Result implements Comparable<Result> {
115 private InetAddress ip;
116 private DNSName hostname;
117 private int port = 5222;
118 private boolean directTls = false;
119 private boolean authenticated =false;
120 private int priority;
121
122 public InetAddress getIp() {
123 return ip;
124 }
125
126 public int getPort() {
127 return port;
128 }
129
130 public DNSName getHostname() {
131 return hostname;
132 }
133
134 public boolean isDirectTls() {
135 return directTls;
136 }
137
138 public boolean isAuthenticated() {
139 return authenticated;
140 }
141
142 @Override
143 public String toString() {
144 return "Result{" +
145 "ip='" + (ip==null?null:ip.getHostAddress()) + '\'' +
146 ", hostame='" + hostname.toString() + '\'' +
147 ", port=" + port +
148 ", directTls=" + directTls +
149 ", authenticated=" + authenticated +
150 ", priority=" + priority +
151 '}';
152 }
153
154 @Override
155 public int compareTo(@NonNull Result result) {
156 if (result.priority == priority) {
157 if (directTls == result.directTls) {
158 if (ip == null && result.ip == null) {
159 return 0;
160 } else {
161 return ip != null ? -1 : 1;
162 }
163 } else {
164 return directTls ? -1 : 1;
165 }
166 } else {
167 return priority - result.priority;
168 }
169 }
170
171 public static Result fromRecord(SRV srv, boolean directTls) {
172 Result result = new Result();
173 result.port = srv.port;
174 result.hostname = srv.name;
175 result.directTls = directTls;
176 result.priority = srv.priority;
177 return result;
178 }
179
180 public static Result createDefault(String domain) {
181 Result result = new Result();
182 result.port = 5222;
183 result.hostname = DNSName.from(domain);
184 return result;
185 }
186 }
187
188}