Merge "Use UnknownHostException directly without casting it"

1 files changed, 6 insertions, 7 deletions

diff --git a/src/com/android/networkstack/util/DnsUtils.java b/src/com/android/networkstack/util/DnsUtils.java
index e68976a..2ea5ed8 100644
--- a/src/com/android/networkstack/util/DnsUtils.java
+++ b/src/com/android/networkstack/util/DnsUtils.java
@@ -80,7 +80,7 @@ public class DnsUtils {
 
         if (result.size() == 0) {
             logger.log("FAIL: " + errorMsg.toString());
-            throw new UnknownHostException(errorMsg.toString());
+            throw new UnknownHostException(host);
         }
         logger.log("OK: " + host + " " + result.toString());
         return result.toArray(new InetAddress[0]);
@@ -134,20 +134,19 @@ public class DnsUtils {
 
         TrafficStats.setThreadStatsTag(oldTag);
 
+        String errorMsg = null;
         List<InetAddress> result = null;
-        Exception exception = null;
         try {
             result = resultRef.get(timeoutMs, TimeUnit.MILLISECONDS);
         } catch (ExecutionException e) {
-            exception = e;
+            errorMsg = e.getMessage();
         } catch (TimeoutException | InterruptedException e) {
-            exception = new UnknownHostException("Timeout");
+            errorMsg = "Timeout";
         } finally {
-            logDnsResult(result, watch.stop() /* latency */, logger, type,
-                    exception != null ? exception.getMessage() : "" /* errorMsg */);
+            logDnsResult(result, watch.stop() /* latency */, logger, type, errorMsg);
         }
 
-        if (null != exception) throw (UnknownHostException) exception;
+        if (null != errorMsg) throw new UnknownHostException(host);
 
         return result.toArray(new InetAddress[0]);
     }