2005-01-31 Zoltan Varga <vargaz@freemail.hu>
[mono.git] / mcs / class / System / System.Net / Dns.cs
index e2d40ee4cf600320adc5d56eaa2b24c7515241a9..67dab9e6bbb8ea369faa9fafbd12a8f4003b3716 100644 (file)
 // System.Net.Dns.cs
 //
 // Author: Mads Pultz (mpultz@diku.dk)
+// Author: Lawrence Pit (loz@cable.a2000.nl)
 //
 // (C) Mads Pultz, 2001
 
+//
+// Permission is hereby granted, free of charge, to any person obtaining
+// a copy of this software and associated documentation files (the
+// "Software"), to deal in the Software without restriction, including
+// without limitation the rights to use, copy, modify, merge, publish,
+// distribute, sublicense, and/or sell copies of the Software, and to
+// permit persons to whom the Software is furnished to do so, subject to
+// the following conditions:
+// 
+// The above copyright notice and this permission notice shall be
+// included in all copies or substantial portions of the Software.
+// 
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+//
+
 using System;
 using System.Net.Sockets;
 using System.Text;
 using System.Collections;
 using System.Threading;
 using System.Runtime.CompilerServices;
+using System.Runtime.Remoting.Messaging;
 
 namespace System.Net {
-        public sealed class Dns {
+       public sealed class Dns {
 
                private Dns () {}
-                
-                /// <summary>
-                /// Helper class
-                /// </summary>
-                private sealed class DnsAsyncResult: IAsyncResult {
-                        private object state;
-                        private WaitHandle waitHandle;
-                        private bool completedSync, completed;
-                        private Worker worker;
-                
-                        public DnsAsyncResult(object state) {
-                                this.state = state;
-                                waitHandle = new ManualResetEvent(false);
-                                completedSync = completed = false;
-                        }       
-                        public object AsyncState {
-                                get { return state; }
-                        }
-                        public WaitHandle AsyncWaitHandle {
-                                set { waitHandle = value; }
-                                get { return waitHandle; }
-                        }
-                        public bool CompletedSynchronously {
-                                get { return completedSync; }
-                        }
-                        public bool IsCompleted {
-                                set { completed = value; }
-                                get { return completed; }
-                        }
-                        public Worker Worker {
-                                set { worker = value; }
-                                get { return worker; }
-                        }
-                }
-
-                /// <summary>
-                /// Helper class for asynchronous calls to DNS server
-                /// </summary>
-                private sealed class Worker {
-                        private AsyncCallback reqCallback;
-                        private DnsAsyncResult reqRes;
-                        private string req;
-                        private IPHostEntry result;
-                        
-                        public Worker(string req, AsyncCallback reqCallback, DnsAsyncResult reqRes) {
-                                this.req = req;
-                                this.reqCallback = reqCallback;
-                                this.reqRes = reqRes;
-                        }
-                        private void End() {
-                                reqCallback(reqRes);
-                                ((ManualResetEvent)reqRes.AsyncWaitHandle).Set();
-                                reqRes.IsCompleted = true;
-                        }
-                        public void GetHostByName() {
-                                lock(reqRes) {
-                                        result = Dns.GetHostByName(req);
-                                        End();
-                                }
-                        }
-                        public void Resolve() {
-                                lock(reqRes) {
-                                        result = Dns.Resolve(req);
-                                        End();
-                                }
-                        }
-                        public IPHostEntry Result {
-                                get { return result; }
-                        }
-                }
-                
-                public static IAsyncResult BeginGetHostByName(string hostName,
-                       AsyncCallback requestCallback, object stateObject)
-                       {
-                        DnsAsyncResult requestResult = new DnsAsyncResult(stateObject);
-                        Worker worker = new Worker(hostName, requestCallback, requestResult);
-                        Thread child = new Thread(new ThreadStart(worker.GetHostByName));
-                        child.Start();
-                        return requestResult;
-                }
-
-                public static IAsyncResult BeginResolve(string hostName,
-                       AsyncCallback requestCallback, object stateObject)
-                {
-                        DnsAsyncResult requestResult = new DnsAsyncResult(stateObject);
-                        Worker worker = new Worker(hostName, requestCallback, requestResult);
-                        Thread child = new Thread(new ThreadStart(worker.Resolve));
-                        child.Start();
-                        return requestResult;
-                }
-                
-                public static IPHostEntry EndGetHostByName(IAsyncResult asyncResult) {
-                        return ((DnsAsyncResult)asyncResult).Worker.Result;
-                }
-
-                public static IPHostEntry EndResolve(IAsyncResult asyncResult) {
-                        return ((DnsAsyncResult)asyncResult).Worker.Result;
-                }
-                
-                
-                [MethodImplAttribute(MethodImplOptions.InternalCall)]
-                private extern static bool GetHostByName_internal(string host, out string h_name, out string[] h_aliases, out string[] h_addr_list);
-                [MethodImplAttribute(MethodImplOptions.InternalCall)]
-                private extern static bool GetHostByAddr_internal(string addr, out string h_name, out string[] h_aliases, out string[] h_addr_list);
-                
-                private static IPHostEntry hostent_to_IPHostEntry(string h_name, string[] h_aliases, string[] h_addrlist) {
-                        IPHostEntry he = new IPHostEntry();
-                        IPAddress[] addrlist = new IPAddress[h_addrlist.Length];
-                        
-                        he.HostName=h_name;
-                        he.Aliases=h_aliases;
-                        for(int i=0; i<h_addrlist.Length; i++) {
-                                addrlist[i]=IPAddress.Parse(h_addrlist[i]);
-                        }
-                        he.AddressList=addrlist;
-
-                        return(he);
-                }
-
-                public static IPHostEntry GetHostByAddress(IPAddress address) {
-                        if (address == null)
-                                throw new ArgumentNullException();
-                        return GetHostByAddress(address.ToString());
-                }
-                
-                public static IPHostEntry GetHostByAddress(string address) {
-                        if (address == null) {
-                                throw new ArgumentNullException();
-                        }
-                        
-                        string h_name;
-                        string[] h_aliases, h_addrlist;
-                        
-                        bool ret = GetHostByAddr_internal(address, out h_name,
-                                                          out h_aliases,
-                                                          out h_addrlist);
-                        if (ret == false) {
-                                throw new SocketException(11001);
-                        }
-                        
-                        return(hostent_to_IPHostEntry(h_name, h_aliases,
-                                                      h_addrlist));
-                }
-
-                public static IPHostEntry GetHostByName(string hostName) {
-                        if (hostName == null) {
-                                throw new ArgumentNullException();
-                        }
-                        
-                        string h_name;
-                        string[] h_aliases, h_addrlist;
-                        
-                        bool ret = GetHostByName_internal(hostName, out h_name,
-                                                          out h_aliases,
-                                                          out h_addrlist);
-                        if (ret == false) {
-                                throw new SocketException(11001);
-                        }
-
-                        return(hostent_to_IPHostEntry(h_name, h_aliases,
-                                                      h_addrlist));
-                }
-                
-                /// <summary>
-                /// This method returns the host name associated with the local host.
-                /// </summary>
-                public static string GetHostName() {
-                        IPHostEntry h = GetHostByAddress("127.0.0.1");
-                        return h.HostName;
-                }
-                
-                /// <summary>
-                /// This method resolves a DNS-style host name or IP
-                /// address.
-                /// </summary>
-                /// <param name=hostName>
-                /// A string containing either a DNS-style host name (e.g.
-                /// www.go-mono.com) or IP address (e.g. 129.250.184.233).
-                /// </param>
-                public static IPHostEntry Resolve(string hostName) {
-                        if (hostName == null)
-                                throw new ArgumentNullException();
-
-                       return GetHostByName (hostName);
-                }
-        }
+               static Dns ()
+               {
+                       System.Net.Sockets.Socket.CheckProtocolSupport();
+               }
+
+               private delegate IPHostEntry GetHostByNameCallback (string hostName);
+               private delegate IPHostEntry ResolveCallback (string hostName);
+
+               public static IAsyncResult BeginGetHostByName (string hostName,
+                       AsyncCallback requestCallback, object stateObject)
+               {
+                       if (hostName == null)
+                               throw new ArgumentNullException();
+
+                       GetHostByNameCallback c = new GetHostByNameCallback (GetHostByName);
+                       return c.BeginInvoke (hostName, requestCallback, stateObject);
+               }
+
+               public static IAsyncResult BeginResolve (string hostName,
+                       AsyncCallback requestCallback, object stateObject)
+               {
+                       if (hostName == null)
+                               throw new ArgumentNullException();
+                       ResolveCallback c = new ResolveCallback (Resolve);
+                       return c.BeginInvoke (hostName, requestCallback, stateObject);
+               }
+
+               public static IPHostEntry EndGetHostByName (IAsyncResult asyncResult) \r
+               {
+                       if (asyncResult == null)
+                               throw new ArgumentNullException ("asyncResult");
+                       AsyncResult async = (AsyncResult) asyncResult;
+                       GetHostByNameCallback cb = (GetHostByNameCallback) async.AsyncDelegate;
+                       return cb.EndInvoke(asyncResult);
+               }
+
+               public static IPHostEntry EndResolve (IAsyncResult asyncResult) \r
+               {
+                       if (asyncResult == null)
+                               throw new ArgumentNullException ("asyncResult");
+                       AsyncResult async = (AsyncResult) asyncResult;
+                       ResolveCallback cb = (ResolveCallback) async.AsyncDelegate;
+                       return cb.EndInvoke(asyncResult);
+               }
+
+               [MethodImplAttribute(MethodImplOptions.InternalCall)]
+               private extern static bool GetHostByName_internal(string host, out string h_name, out string[] h_aliases, out string[] h_addr_list);
+
+               [MethodImplAttribute(MethodImplOptions.InternalCall)]
+               private extern static bool GetHostByAddr_internal(string addr, out string h_name, out string[] h_aliases, out string[] h_addr_list);
+
+               [MethodImplAttribute(MethodImplOptions.InternalCall)]
+               private extern static bool GetHostName_internal(out string h_name);
+               
+               private static IPHostEntry hostent_to_IPHostEntry(string h_name, string[] h_aliases, string[] h_addrlist) \r
+               {
+                       IPHostEntry he = new IPHostEntry();
+                       ArrayList addrlist = new ArrayList();
+
+                       he.HostName = h_name;
+                       he.Aliases = h_aliases;
+                       for(int i=0; i<h_addrlist.Length; i++) {
+                               IPAddress newAddress = IPAddress.Parse(h_addrlist[i]);
+
+                               if( (Socket.SupportsIPv6 && newAddress.AddressFamily == AddressFamily.InterNetworkV6) ||
+                                       (Socket.SupportsIPv4 && newAddress.AddressFamily == AddressFamily.InterNetwork) )
+                                       addrlist.Add(newAddress);
+                       }
+
+                       if(addrlist.Count == 0)
+                               throw new SocketException(11001);
+
+                       he.AddressList = addrlist.ToArray(typeof(IPAddress)) as IPAddress[];
+                       return he;
+               }
+
+               public static IPHostEntry GetHostByAddress(IPAddress address) \r
+               {
+                       if (address == null)
+                               throw new ArgumentNullException();
+                       return GetHostByAddress (address.ToString());
+               }
+\r
+               public static IPHostEntry GetHostByAddress(string address) \r
+               {
+                       if (address == null)
+                               throw new ArgumentNullException();
+
+                       // Undocumented MS behavior: when called with IF_ANY,
+                       // this should return the local host
+                       if (address.Equals ("0.0.0.0"))
+                               return GetHostByAddress ("127.0.0.1");
+
+                       /// Must check the IP format, might send an exception if 
+                       /// invalid string.
+                       IPAddress.Parse(address);
+
+                       string h_name;
+                       string[] h_aliases, h_addrlist;
+
+                       bool ret = GetHostByAddr_internal(address, out h_name,
+                               out h_aliases,
+                               out h_addrlist);
+                       if (!ret)
+                               throw new SocketException(11001);
+
+                       return(hostent_to_IPHostEntry(h_name, h_aliases,
+                               h_addrlist));
+               }
+
+               public static IPHostEntry GetHostByName(string hostName) \r
+               {
+                       if (hostName == null)
+                               throw new ArgumentNullException();
+
+                       string h_name;
+                       string[] h_aliases, h_addrlist;
+
+                       bool ret = GetHostByName_internal(hostName, out h_name,
+                               out h_aliases,
+                               out h_addrlist);
+                       if (ret == false)
+                               throw new SocketException(11001);
+
+                       return(hostent_to_IPHostEntry(h_name, h_aliases,
+                               h_addrlist));
+               }
+
+               /// <summary>
+               /// This method returns the host name associated with the local host.
+               /// </summary>
+               public static string GetHostName() \r
+               {
+                       string hostName;
+
+                       bool ret = GetHostName_internal(out hostName);
+
+                       if (ret == false)
+                               throw new SocketException(11001);
+
+                       return hostName;
+               }
+
+               /// <summary>
+               /// This method resolves a DNS-style host name or IP
+               /// address.
+               /// </summary>
+               /// <param name=hostName>
+               /// A string containing either a DNS-style host name (e.g.
+               /// www.go-mono.com) or IP address (e.g. 129.250.184.233).
+               /// </param>
+               public static IPHostEntry Resolve(string hostName) \r
+               {
+                       if (hostName == null)
+                               throw new ArgumentNullException();
+
+                       IPHostEntry ret = null;
+
+                       try {\r
+                               ret =  GetHostByAddress(hostName);\r
+                       }
+                       catch{}
+
+                       if(ret == null)\r
+                               ret =  GetHostByName(hostName);\r
+
+                       return ret;
+               }
+       }
 }