Merge pull request #829 from symform/httpwebreq-async-ssl
[mono.git] / mcs / class / System / System.Net / Dns.cs
index f83909c47757dd2eb87c6a9c9b65bc6b6a78e8b9..fb532cd22c2a8a19929b6cfe1b85a20f08d00f7d 100644 (file)
@@ -1,9 +1,17 @@
 // System.Net.Dns.cs
 //
+// Authors:
+//     Mads Pultz (mpultz@diku.dk)
+//     Lawrence Pit (loz@cable.a2000.nl)
+
 // Author: Mads Pultz (mpultz@diku.dk)
-// Author: Lawrence Pit (loz@cable.a2000.nl)
+//        Lawrence Pit (loz@cable.a2000.nl)
+//        Marek Safar (marek.safar@gmail.com)
+//        Gonzalo Paniagua Javier (gonzalo.mono@gmail.com)
 //
 // (C) Mads Pultz, 2001
+// Copyright (c) 2011 Novell, Inc.
+// Copyright (c) 2011 Xamarin, Inc.
 
 //
 // Permission is hereby granted, free of charge, to any person obtaining
@@ -33,121 +41,224 @@ using System.Collections;
 using System.Threading;
 using System.Runtime.CompilerServices;
 using System.Runtime.Remoting.Messaging;
+#if NET_4_5
+using System.Threading.Tasks;
+#endif
+
+#if !MOBILE
+using Mono.Net.Dns;
+#endif
 
 namespace System.Net {
-#if NET_2_0
        public static class Dns {
-#else
-       public sealed class Dns {
-
-               private Dns () {}
+#if !MOBILE
+               static bool use_mono_dns;
+               static SimpleResolver resolver;
 #endif
 
                static Dns ()
                {
                        System.Net.Sockets.Socket.CheckProtocolSupport();
+
+#if !MOBILE
+                       if (Environment.GetEnvironmentVariable ("MONO_DNS") != null) {
+                               resolver = new SimpleResolver ();
+                               use_mono_dns = true;
+                       }
+#endif
                }
 
-#if !NET_2_1 // global remove of async methods
+#if !MOBILE
+               internal static bool UseMonoDns {
+                       get { return use_mono_dns; }
+               }
+#endif
 
                private delegate IPHostEntry GetHostByNameCallback (string hostName);
                private delegate IPHostEntry ResolveCallback (string hostName);
-#if NET_2_0
                private delegate IPHostEntry GetHostEntryNameCallback (string hostName);
                private delegate IPHostEntry GetHostEntryIPCallback (IPAddress hostAddress);
                private delegate IPAddress [] GetHostAddressesCallback (string hostName);
+
+#if !MOBILE
+               static void OnCompleted (object sender, SimpleResolverEventArgs e)
+               {
+                       DnsAsyncResult ares = (DnsAsyncResult) e.UserToken;
+                       IPHostEntry entry = e.HostEntry;
+                       if (entry == null || e.ResolverError != 0) {
+                               ares.SetCompleted (false, new Exception ("Error: " + e.ResolverError));
+                               return;
+                       }
+                       ares.SetCompleted (false, entry);
+               }
+
+               static IAsyncResult BeginAsyncCallAddresses (string host, AsyncCallback callback, object state)
+               {
+                       SimpleResolverEventArgs e = new SimpleResolverEventArgs ();
+                       e.Completed += OnCompleted;
+                       e.HostName = host;
+                       DnsAsyncResult ares = new DnsAsyncResult (callback, state);
+                       e.UserToken = ares;
+                       if (resolver.GetHostAddressesAsync (e) == false)
+                               ares.SetCompleted (true, e.HostEntry); // Completed synchronously
+                       return ares;
+               }
+
+               static IAsyncResult BeginAsyncCall (string host, AsyncCallback callback, object state)
+               {
+                       SimpleResolverEventArgs e = new SimpleResolverEventArgs ();
+                       e.Completed += OnCompleted;
+                       e.HostName = host;
+                       DnsAsyncResult ares = new DnsAsyncResult (callback, state);
+                       e.UserToken = ares;
+                       if (resolver.GetHostEntryAsync (e) == false)
+                               ares.SetCompleted (true, e.HostEntry); // Completed synchronously
+                       return ares;
+               }
+
+               static IPHostEntry EndAsyncCall (DnsAsyncResult ares)
+               {
+                       if (ares == null)
+                               throw new ArgumentException ("Invalid asyncResult");
+                       if (!ares.IsCompleted)
+                               ares.AsyncWaitHandle.WaitOne ();
+                       if (ares.Exception != null)
+                               throw ares.Exception;
+                       IPHostEntry entry = ares.HostEntry;
+                       if (entry == null || entry.AddressList == null || entry.AddressList.Length == 0)
+                               Error_11001 (entry.HostName);
+                       return entry;
+               }
 #endif
 
-#if NET_2_0
                [Obsolete ("Use BeginGetHostEntry instead")]
-#endif
-               public static IAsyncResult BeginGetHostByName (string hostName,
-                       AsyncCallback requestCallback, object stateObject)
+               public static IAsyncResult BeginGetHostByName (string hostName, AsyncCallback requestCallback, object stateObject)
                {
                        if (hostName == null)
                                throw new ArgumentNullException ("hostName");
 
+#if !MOBILE
+                       if (use_mono_dns)
+                               return BeginAsyncCall (hostName, requestCallback, stateObject);
+#endif
+
                        GetHostByNameCallback c = new GetHostByNameCallback (GetHostByName);
                        return c.BeginInvoke (hostName, requestCallback, stateObject);
                }
 
-#if NET_2_0
                [Obsolete ("Use BeginGetHostEntry instead")]
-#endif
-               public static IAsyncResult BeginResolve (string hostName,
-                       AsyncCallback requestCallback, object stateObject)
+               public static IAsyncResult BeginResolve (string hostName, AsyncCallback requestCallback, object stateObject)
                {
                        if (hostName == null)
                                throw new ArgumentNullException ("hostName");
 
+#if !MOBILE
+                       if (use_mono_dns)
+                               return BeginAsyncCall (hostName, requestCallback, stateObject);
+#endif
+
                        ResolveCallback c = new ResolveCallback (Resolve);
                        return c.BeginInvoke (hostName, requestCallback, stateObject);
                }
 
-#if NET_2_0
-               public static IAsyncResult BeginGetHostAddresses (string hostName,
-                       AsyncCallback requestCallback, object stateObject)
+               public static IAsyncResult BeginGetHostAddresses (string hostNameOrAddress, AsyncCallback requestCallback, object state)
                {
-                       if (hostName == null)
+                       if (hostNameOrAddress == null)
                                throw new ArgumentNullException ("hostName");
+                       if (hostNameOrAddress == "0.0.0.0" || hostNameOrAddress == "::0")
+                               throw new ArgumentException ("Addresses 0.0.0.0 (IPv4) " +
+                                       "and ::0 (IPv6) are unspecified addresses. You " +
+                                       "cannot use them as target address.",
+                                       "hostNameOrAddress");
+
+#if !MOBILE
+                       if (use_mono_dns)
+                               return BeginAsyncCallAddresses (hostNameOrAddress, requestCallback, state);
+#endif
 
                        GetHostAddressesCallback c = new GetHostAddressesCallback (GetHostAddresses);
-                       return c.BeginInvoke (hostName, requestCallback, stateObject);
+                       return c.BeginInvoke (hostNameOrAddress, requestCallback, state);
                }
 
-               public static IAsyncResult BeginGetHostEntry (string hostNameOrAddress,
-                       AsyncCallback requestCallback, object stateObject)
+               public static IAsyncResult BeginGetHostEntry (string hostNameOrAddress, AsyncCallback requestCallback, object stateObject)
                {
                        if (hostNameOrAddress == null)
-                               throw new ArgumentNullException ("hostNameOrAddress");
+                               throw new ArgumentNullException ("hostName");
+                       if (hostNameOrAddress == "0.0.0.0" || hostNameOrAddress == "::0")
+                               throw new ArgumentException ("Addresses 0.0.0.0 (IPv4) " +
+                                       "and ::0 (IPv6) are unspecified addresses. You " +
+                                       "cannot use them as target address.",
+                                       "hostNameOrAddress");
+
+#if !MOBILE
+                       if (use_mono_dns)
+                               return BeginAsyncCall (hostNameOrAddress, requestCallback, stateObject);
+#endif
 
                        GetHostEntryNameCallback c = new GetHostEntryNameCallback (GetHostEntry);
                        return c.BeginInvoke (hostNameOrAddress, requestCallback, stateObject);
                }
 
-               public static IAsyncResult BeginGetHostEntry (IPAddress hostAddress,
-                       AsyncCallback requestCallback, object stateObject)
+               public static IAsyncResult BeginGetHostEntry (IPAddress address, AsyncCallback requestCallback, object stateObject)
                {
-                       if (hostAddress == null)
-                               throw new ArgumentNullException ("hostAddress");
+                       if (address == null)
+                               throw new ArgumentNullException ("address");
+
+#if !MOBILE
+                       if (use_mono_dns)
+                               return BeginAsyncCall (address.ToString (), requestCallback, stateObject);
+#endif
 
                        GetHostEntryIPCallback c = new GetHostEntryIPCallback (GetHostEntry);
-                       return c.BeginInvoke (hostAddress, requestCallback, stateObject);
+                       return c.BeginInvoke (address, requestCallback, stateObject);
                }
-#endif
 
-#if NET_2_0
                [Obsolete ("Use EndGetHostEntry instead")]
-#endif
                public static IPHostEntry EndGetHostByName (IAsyncResult asyncResult) 
                {
                        if (asyncResult == null)
                                throw new ArgumentNullException ("asyncResult");
 
+#if !MOBILE
+                       if (use_mono_dns)
+                               return EndAsyncCall (asyncResult as DnsAsyncResult);
+#endif
+
                        AsyncResult async = (AsyncResult) asyncResult;
                        GetHostByNameCallback cb = (GetHostByNameCallback) async.AsyncDelegate;
                        return cb.EndInvoke(asyncResult);
                }
 
-#if NET_2_0
                [Obsolete ("Use EndGetHostEntry instead")]
-#endif
                public static IPHostEntry EndResolve (IAsyncResult asyncResult) 
                {
                        if (asyncResult == null)
                                throw new ArgumentNullException ("asyncResult");
+
+#if !MOBILE
+                       if (use_mono_dns)
+                               return EndAsyncCall (asyncResult as DnsAsyncResult);
+#endif
+
                        AsyncResult async = (AsyncResult) asyncResult;
                        ResolveCallback cb = (ResolveCallback) async.AsyncDelegate;
                        return cb.EndInvoke(asyncResult);
                }
 
-#if NET_2_0
-
                public static IPAddress [] EndGetHostAddresses (IAsyncResult asyncResult) 
                {
                        if (asyncResult == null)
                                throw new ArgumentNullException ("asyncResult");
 
+#if !MOBILE
+                       if (use_mono_dns) {
+                               IPHostEntry entry = EndAsyncCall (asyncResult as DnsAsyncResult);
+                               if (entry == null)
+                                       return null;
+                               return entry.AddressList;
+                       }
+#endif
+
                        AsyncResult async = (AsyncResult) asyncResult;
                        GetHostAddressesCallback cb = (GetHostAddressesCallback) async.AsyncDelegate;
                        return cb.EndInvoke(asyncResult);
@@ -157,17 +268,19 @@ namespace System.Net {
                {
                        if (asyncResult == null)
                                throw new ArgumentNullException ("asyncResult");
+
+#if !MOBILE
+                       if (use_mono_dns)
+                               return EndAsyncCall (asyncResult as DnsAsyncResult);
+#endif
+
                        AsyncResult async = (AsyncResult) asyncResult;
-#if NET_2_0
                        if (async.AsyncDelegate is GetHostEntryIPCallback)
                                return ((GetHostEntryIPCallback) async.AsyncDelegate).EndInvoke (asyncResult);
-#endif
                        GetHostEntryNameCallback cb = (GetHostEntryNameCallback) async.AsyncDelegate;
                        return cb.EndInvoke(asyncResult);
                }
-#endif
-               
-#endif // !NET_2_1: global remove of async methods
+
 
 #if !TARGET_JVM
                [MethodImplAttribute(MethodImplOptions.InternalCall)]
@@ -180,7 +293,13 @@ namespace System.Net {
                private extern static bool GetHostName_internal(out string h_name);
 #endif 
 
-               private static IPHostEntry hostent_to_IPHostEntry(string h_name, string[] h_aliases, string[] h_addrlist) 
+               static void Error_11001 (string hostName)
+               {
+                       throw new SocketException(11001, string.Format ("Could not resolve host '{0}'", hostName));
+
+               }
+
+               private static IPHostEntry hostent_to_IPHostEntry(string originalHostName, string h_name, string[] h_aliases, string[] h_addrlist) 
                {
                        IPHostEntry he = new IPHostEntry();
                        ArrayList addrlist = new ArrayList();
@@ -204,15 +323,13 @@ namespace System.Net {
                        }
 
                        if(addrlist.Count == 0)
-                               throw new SocketException(11001);
+                               Error_11001 (originalHostName);
 
                        he.AddressList = addrlist.ToArray(typeof(IPAddress)) as IPAddress[];
                        return he;
                }
 
-#if NET_2_0
                [Obsolete ("Use GetHostEntry instead")]
-#endif
                public static IPHostEntry GetHostByAddress(IPAddress address)
                {
                        if (address == null)
@@ -221,9 +338,7 @@ namespace System.Net {
                        return GetHostByAddressFromString (address.ToString (), false);
                }
 
-#if NET_2_0
                [Obsolete ("Use GetHostEntry instead")]
-#endif
                public static IPHostEntry GetHostByAddress(string address)
                {
                        if (address == null)
@@ -268,25 +383,27 @@ namespace System.Net {
 #else
                        bool ret = GetHostByAddr_internal(address, out h_name, out h_aliases, out h_addrlist);
                        if (!ret)
-                               throw new SocketException(11001);
+                               Error_11001 (address);
 #endif
-                       return (hostent_to_IPHostEntry (h_name, h_aliases, h_addrlist));
+                       return (hostent_to_IPHostEntry (address, h_name, h_aliases, h_addrlist));
                        
                }
 
-#if NET_2_0
                public static IPHostEntry GetHostEntry (string hostNameOrAddress)
                {
                        if (hostNameOrAddress == null)
                                throw new ArgumentNullException ("hostNameOrAddress");
+                       if (hostNameOrAddress == "0.0.0.0" || hostNameOrAddress == "::0")
+                               throw new ArgumentException ("Addresses 0.0.0.0 (IPv4) " +
+                                       "and ::0 (IPv6) are unspecified addresses. You " +
+                                       "cannot use them as target address.",
+                                       "hostNameOrAddress");
 
-                       if (hostNameOrAddress == "0.0.0.0" || hostNameOrAddress == "")
-                               hostNameOrAddress = "127.0.0.1";
                        IPAddress addr;
-                       if (IPAddress.TryParse (hostNameOrAddress, out addr))
+                       if (hostNameOrAddress.Length > 0 && IPAddress.TryParse (hostNameOrAddress, out addr))
                                return GetHostEntry (addr);
-                       else
-                               return GetHostByName (hostNameOrAddress);
+
+                       return GetHostByName (hostNameOrAddress);
                }
 
                public static IPHostEntry GetHostEntry (IPAddress address)
@@ -302,29 +419,24 @@ namespace System.Net {
                        if (hostNameOrAddress == null)
                                throw new ArgumentNullException ("hostNameOrAddress");
 
-                       if (hostNameOrAddress == "0.0.0.0" || hostNameOrAddress == "")
-                               hostNameOrAddress = "127.0.0.1";
+                       if (hostNameOrAddress == "0.0.0.0" || hostNameOrAddress == "::0")
+                               throw new ArgumentException ("Addresses 0.0.0.0 (IPv4) " +
+                                       "and ::0 (IPv6) are unspecified addresses. You " +
+                                       "cannot use them as target address.",
+                                       "hostNameOrAddress");
 
                        IPAddress addr;
-                       if (IPAddress.TryParse (hostNameOrAddress, out addr))
-                       {
+                       if (hostNameOrAddress.Length > 0 && IPAddress.TryParse (hostNameOrAddress, out addr))
                                return new IPAddress[1] { addr };
-                       }
-                       else
-                       {
-                         
-                               return GetHostEntry (hostNameOrAddress).AddressList;
-                       }
+
+                       return GetHostEntry (hostNameOrAddress).AddressList;
                }
-#endif
 
-#if NET_2_0
                [Obsolete ("Use GetHostEntry instead")]
-#endif
                public static IPHostEntry GetHostByName (string hostName)
                {
                        if (hostName == null)
-                               throw new ArgumentNullException ();
+                               throw new ArgumentNullException ("hostName");
 #if TARGET_JVM
                        if (hostName.Length == 0)
                                hostName = "localhost";
@@ -348,14 +460,11 @@ namespace System.Net {
                        string h_name;
                        string[] h_aliases, h_addrlist;
 
-                       bool ret = GetHostByName_internal(hostName, out h_name,
-                               out h_aliases,
-                               out h_addrlist);
+                       bool ret = GetHostByName_internal(hostName, out h_name, out h_aliases, out h_addrlist);
                        if (ret == false)
-                               throw new SocketException(11001);
+                               Error_11001 (hostName);
 
-                       return(hostent_to_IPHostEntry(h_name, h_aliases,
-                               h_addrlist));
+                       return(hostent_to_IPHostEntry(hostName, h_name, h_aliases, h_addrlist));
 #endif
                }
 
@@ -369,15 +478,13 @@ namespace System.Net {
                        bool ret = GetHostName_internal(out hostName);
 
                        if (ret == false)
-                               throw new SocketException(11001);
+                               Error_11001 (hostName);
 
                        return hostName;
 #endif
                }
 
-#if NET_2_0
                [Obsolete ("Use GetHostEntry instead")]
-#endif
                public static IPHostEntry Resolve(string hostName) 
                {
                        if (hostName == null)
@@ -395,6 +502,23 @@ namespace System.Net {
 
                        return ret;
                }
+
+#if NET_4_5
+               public static Task<IPAddress[]> GetHostAddressesAsync (string hostNameOrAddress)
+               {
+                       return Task<IPAddress[]>.Factory.FromAsync (BeginGetHostAddresses, EndGetHostAddresses, hostNameOrAddress, null);
+               }
+
+               public static Task<IPHostEntry> GetHostEntryAsync (IPAddress address)
+               {
+                       return Task<IPHostEntry>.Factory.FromAsync (BeginGetHostEntry, EndGetHostEntry, address, null);
+               }
+
+               public static Task<IPHostEntry> GetHostEntryAsync (string hostNameOrAddress)
+               {
+                       return Task<IPHostEntry>.Factory.FromAsync (BeginGetHostEntry, EndGetHostEntry, hostNameOrAddress, null);
+               }
+#endif
        }
 }