netwerk/test/TestDNS.cpp

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3  *
   4  * The contents of this file are subject to the Mozilla Public License Version
   5  * 1.1 (the "License"); you may not use this file except in compliance with
   6  * the License. You may obtain a copy of the License at
   7  * http://www.mozilla.org/MPL/
   8  *
   9  * Software distributed under the License is distributed on an "AS IS" basis,
  10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11  * for the specific language governing rights and limitations under the
  12  * License.
  13  *
  14  * The Original Code is Mozilla.
  15  *
  16  * The Initial Developer of the Original Code is IBM Corporation.
  17  * Portions created by IBM Corporation are Copyright (C) 2003
  18  * IBM Corporation. All Rights Reserved.
  19  *
  20  * Contributor(s):
  21  *   IBM Corp.
  22  *
  23  * Alternatively, the contents of this file may be used under the terms of
  24  * either the GNU General Public License Version 2 or later (the "GPL"), or
  25  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  26  * in which case the provisions of the GPL or the LGPL are applicable instead
  27  * of those above. If you wish to allow use of your version of this file only
  28  * under the terms of either the GPL or the LGPL, and not to allow others to
  29  * use your version of this file under the terms of the MPL, indicate your
  30  * decision by deleting the provisions above and replace them with the notice
  31  * and other provisions required by the GPL or the LGPL. If you do not delete
  32  * the provisions above, a recipient may use your version of this file under
  33  * the terms of any one of the MPL, the GPL or the LGPL.
  34  *
  35  * ***** END LICENSE BLOCK ***** */
  36
  37 #include "TestCommon.h"
  38 #include <stdio.h>
  39 #include <stdlib.h>
  40 #include "nsIServiceManager.h"
  41 #include "nsPIDNSService.h"
  42 #include "nsIDNSListener.h"
  43 #include "nsIDNSRecord.h"
  44 #include "nsICancelable.h"
  45 #include "nsCOMPtr.h"
  46 #include "nsStringAPI.h"
  47 #include "nsNetCID.h"
  48 #include "prinrval.h"
  49 #include "prthread.h"
  50 #include "prnetdb.h"
  51 #include "nsXPCOM.h"
  52 #include "nsServiceManagerUtils.h"
  53
  54 class myDNSListener : public nsIDNSListener
  55 {
  56 public:
  57     NS_DECL_ISUPPORTS
  58
  59     myDNSListener(const char *host, PRInt32 index)
  60         : mHost(host)
  61         , mIndex(index) {}
  62     virtual ~myDNSListener() {}
  63
  64     NS_IMETHOD OnLookupComplete(nsICancelable *request,
  65                                 nsIDNSRecord  *rec,
  66                                 nsresult       status)
  67     {
  68         printf("%d: OnLookupComplete called [host=%s status=%x rec=%p]\n",
  69             mIndex, mHost.get(), status, (void*)rec);
  70
  71         if (NS_SUCCEEDED(status)) {
  72             nsCAutoString buf;
  73
  74             rec->GetCanonicalName(buf);
  75             printf("%d: canonname=%s\n", mIndex, buf.get());
  76
  77             PRBool hasMore;
  78             while (NS_SUCCEEDED(rec->HasMore(&hasMore)) && hasMore) {
  79                 rec->GetNextAddrAsString(buf);
  80                 printf("%d: => %s\n", mIndex, buf.get());
  81             }
  82         }
  83
  84         return NS_OK;
  85     }
  86
  87 private:
  88     nsCString mHost;
  89     PRInt32   mIndex;
  90 };
  91
  92 NS_IMPL_THREADSAFE_ISUPPORTS1(myDNSListener, nsIDNSListener)
  93
  94 static PRBool IsAscii(const char *s)
  95 {
  96   for (; *s; ++s) {
  97     if (*s & 0x80)
  98       return PR_FALSE;
  99   }
 100
 101   return PR_TRUE;
 102 }
 103
 104 int main(int argc, char **argv)
 105 {
 106     if (test_common_init(&argc, &argv) != 0)
 107         return -1;
 108
 109     int sleepLen = 10; // default: 10 seconds
 110
 111     if (argc == 1) {
 112         printf("usage: TestDNS [-N] hostname1 [hostname2 ...]\n");
 113         return -1;
 114     }
 115
 116     {
 117         nsCOMPtr<nsIServiceManager> servMan;
 118         NS_InitXPCOM2(getter_AddRefs(servMan), nsnull, nsnull);
 119
 120         nsCOMPtr<nsPIDNSService> dns = do_GetService(NS_DNSSERVICE_CONTRACTID);
 121         if (!dns)
 122             return -1;
 123
 124         if (argv[1][0] == '-') {
 125             sleepLen = atoi(argv[1]+1);
 126             argv++;
 127             argc--;
 128         }
 129
 130         for (int j=0; j<2; ++j) {
 131             for (int i=1; i<argc; ++i) {
 132                 // assume non-ASCII input is given in the native charset
 133                 nsCAutoString hostBuf;
 134                 if (IsAscii(argv[i]))
 135                     hostBuf.Assign(argv[i]);
 136                 else
 137                     hostBuf = NS_ConvertUTF16toUTF8(NS_ConvertASCIItoUTF16(argv[i]));
 138
 139                 nsCOMPtr<nsIDNSListener> listener = new myDNSListener(argv[i], i);
 140
 141                 nsCOMPtr<nsICancelable> req;
 142                 nsresult rv = dns->AsyncResolve(hostBuf,
 143                                                 nsIDNSService::RESOLVE_CANONICAL_NAME,
 144                                                 listener, nsnull, getter_AddRefs(req));
 145                 if (NS_FAILED(rv))
 146                     printf("### AsyncResolve failed [rv=%x]\n", rv);
 147             }
 148
 149             printf("main thread sleeping for %d seconds...\n", sleepLen);
 150             PR_Sleep(PR_SecondsToInterval(sleepLen));
 151         }
 152
 153         printf("shutting down main thread...\n");
 154         dns->Shutdown();
 155     }
 156
 157     NS_ShutdownXPCOM(nsnull);
 158     return 0;
 159 }