2005-01-31 Zoltan Varga <vargaz@freemail.hu>
[mono.git] / mcs / class / System / System.Net / Dns.cs
index 8a05e5fd282b5de4706db9aee775d157d1bf5796..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.InteropServices;
+using System.Runtime.CompilerServices;
 using System.Runtime.Remoting.Messaging;
 
 namespace System.Net {
-
        public sealed class 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; }
-                       }
+               private Dns () {}
+               static Dns ()
+               {
+                       System.Net.Sockets.Socket.CheckProtocolSupport();
                }
-               
-               /// <summary>
-               /// This class conforms to the C structure <c>hostent</c> and is used
-               /// by the Dns class when doing native calls.
-               /// </summary>
-               [StructLayout(LayoutKind.Sequential)]
-               private unsafe class Hostent {
-                       public string h_name;       /* official name */
-                       public byte** h_aliases;    /* alias list */
-                       public short h_addrtype;    /* address type */
-                       public short h_length;      /* address length */
-                       public byte** h_addr_list;  /* address list */
-               }
-               
-               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;
+
+               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);
                }
 
-               [MonoTODO]
-               public static IAsyncResult BeginResolve(string hostName,
-                                               AsyncCallback requestCallback,
-                                               object stateObject) {
-                       // TODO
-                       throw new NotImplementedException();
+               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) {
-                       return ((DnsAsyncResult)asyncResult).Worker.Result;
+
+               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);
                }
 
-               [MonoTODO]
-               public static IPHostEntry EndResolve(IAsyncResult asyncResult) {
-                       // TODO
-                       throw new NotImplementedException();
+               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);
                
-               /// <param name=hostName>
-               /// IP address in network byte order (e.g. Big-Endian).
-               /// </param>
-               /// <param name=length>
-               /// Length of IP address.
-               /// </param>
-               /// <param name=type>
-               /// Type (should be 2, equals AF_INET).
-               /// </param>
-               [DllImport("cygwin1", EntryPoint="gethostbyaddr")]
-               private static extern IntPtr _GetHostByAddress(byte[] hostName,
-                                                              short length,
-                                                              short type);
-               
-               /// <param name=address>
-               /// IP address in network byte order (e.g. Big-Endian).
-               /// </param>
-               [MonoTODO]
-               private static IPHostEntry GetHostByAddress(long address) {
-                       short length = 4;
-                       if (address > uint.MaxValue)
-                               length = 8;
-                       byte[] addr = new byte[length];
-                       for(int i = length - 1, j = 0; i >= 0; --i, ++j) {
-                               byte b = (byte)(address >> i * 8);
-//                             Console.WriteLine(b);
-                               addr[j] = b;
+               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);
                        }
-                       IntPtr p = _GetHostByAddress(addr, length, 2);  // TODO: set type
-                       if (p == IntPtr.Zero)
-                               throw new SocketException();  // TODO: set error code
-                       Hostent h = new Hostent();
-                       System.Runtime.InteropServices.Marshal.PtrToStructure(p, h);
-                       return ToIPHostEntry(h);
+
+                       if(addrlist.Count == 0)
+                               throw new SocketException(11001);
+
+                       he.AddressList = addrlist.ToArray(typeof(IPAddress)) as IPAddress[];
+                       return he;
                }
-               
-               public static IPHostEntry GetHostByAddress(IPAddress address) {
+
+               public static IPHostEntry GetHostByAddress(IPAddress address) \r
+               {
                        if (address == null)
                                throw new ArgumentNullException();
-                       return GetHostByAddress (IPAddress.HostToNetworkOrder (address.Address));
+                       return GetHostByAddress (address.ToString());
                }
-               
-               public static IPHostEntry GetHostByAddress(string address) {
+\r
+               public static IPHostEntry GetHostByAddress(string address) \r
+               {
                        if (address == null)
                                throw new ArgumentNullException();
-                       return GetHostByAddress(CreateAddress(address));
+
+                       // 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));
                }
-               
-               [DllImport("cygwin1", EntryPoint="gethostbyname")]
-               private static extern IntPtr _GetHostByName(string hostName);
 
-               [MonoTODO]
-               public static IPHostEntry GetHostByName(string hostName) {
+               public static IPHostEntry GetHostByName(string hostName) \r
+               {
                        if (hostName == null)
                                throw new ArgumentNullException();
-                       IntPtr p = _GetHostByName(hostName);
-                       //        int errNo = _h_errno;
-                       if (p == IntPtr.Zero)
-                               throw new SocketException();  // TODO: set error code
-                       Hostent h = new Hostent();
-                       System.Runtime.InteropServices.Marshal.PtrToStructure(p, h);
-                       return ToIPHostEntry(h);
+
+                       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;
+               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 resovles a DNS-style host name or IP
+               /// 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) {
+               public static IPHostEntry Resolve(string hostName) \r
+               {
                        if (hostName == null)
                                throw new ArgumentNullException();
-                       try {
-                               long addr = CreateAddress(hostName);
-                               if (addr > uint.MaxValue)
-                                       throw new FormatException("Only IP version 4 addresses are supported");
-                               return GetHostByAddress(addr);
-                       } catch (FormatException) {
-                         return GetHostByName(hostName);
-                       }
-               }
-               
-               /// <summary>
-               /// Utility method. This method converts a Hostent instance to a
-               /// IPHostEntry instance.
-               /// </summary>
-               /// <param name=h>
-               /// Object which should be mapped to a IPHostEntry instance.
-               /// </param>
-               private static unsafe IPHostEntry ToIPHostEntry(Hostent h) {
-                       IPHostEntry res = new IPHostEntry();
-                       
-                       // Set host name
-                       res.HostName = h.h_name;
-                       
-                       // Set IP address list
-                       byte** p = h.h_addr_list;
-                       ArrayList tmp = new ArrayList(1);
-                       while (*p != null) {
-                               tmp.Add(CreateIPAddress(*p, h.h_length));
-                               ++p;
-                       }
-                       IPAddress[] addr_list = new IPAddress[tmp.Count];
-                       for(int i = 0; i < tmp.Count; ++i)
-                               addr_list[i] = (IPAddress)tmp[i];
-                       res.AddressList = addr_list;
-                       
-                       // Set IP aliases
-                       p = h.h_aliases;
-                       tmp.Clear();
-                       while (*p != null) {
-                               tmp.Add(new string((sbyte*)*p));
-                               ++p;
-                       }
-                       string[] aliases = new string[tmp.Count];
-                       for(int i = 0; i < tmp.Count; ++i)
-                               aliases[i] = (string)tmp[i];
-                       res.Aliases = aliases;
-                       
-                       return res;
-               }
-               
-               /// <summary>
-               /// Utility method. Convert IP address in dotted notation
-               /// to IP address.
-               /// </summary>
-               private static long CreateAddress(string address) {
-                       string[] tokens = address.Split('.');
-                       if (tokens.Length % 4 != 0)
-                               throw new FormatException("IP address has invalid length");
-                       long addr = 0;
-                       for(int i = 0, j = tokens.Length - 1; i < tokens.Length; ++i, --j) {
-                               try {
-                                       addr = addr | (((long)byte.Parse(tokens[i])) << j * 8);
-                               } catch (OverflowException) {
-                                       throw new FormatException("Invalid IP address format");
-                               }
-                       }
-                       return addr;
-               }
-       
-               /// <summary>
-               /// Utility method. This method creates a IP address.
-               /// </summary>
-               /// <param name=addr>
-               /// IP address in network byte order (e.g. Big-Endian).
-               /// </param>
-               /// <param name=length>
-               /// Length of IP address (4 or 8 bytes).
-               /// </param>
-               private static unsafe IPAddress CreateIPAddress(byte* addr, short length) {
-                       byte* p = addr;
-                       long res = 0;
-                       for(int i = 0, j = length - 1; i < length; ++i, --j) {
-                               res += *p << j * 8;
-                               ++p;
+
+                       IPHostEntry ret = null;
+
+                       try {\r
+                               ret =  GetHostByAddress(hostName);\r
                        }
-                       if (res > uint.MaxValue)
-                               return new IPAddress(IPAddress.NetworkToHostOrder((long)res));
-                       else
-                               return new IPAddress(IPAddress.NetworkToHostOrder((int)res));
+                       catch{}
+
+                       if(ret == null)\r
+                               ret =  GetHostByName(hostName);\r
+
+                       return ret;
                }
        }
 }