netwerk/dns/DNSRequestParent.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set sw=2 ts=8 et tw=80 : */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "mozilla/net/DNSRequestParent.h"
   8 #include "nsIDNSService.h"
   9 #include "nsNetCID.h"
  10 #include "nsThreadUtils.h"
  11 #include "nsIServiceManager.h"
  12 #include "nsICancelable.h"
  13 #include "nsIDNSRecord.h"
  14 #include "nsHostResolver.h"
  15 #include "mozilla/unused.h"
  16
  17 using namespace mozilla::ipc;
  18
  19 namespace mozilla {
  20 namespace net {
  21
  22 DNSRequestParent::DNSRequestParent()
  23   : mIPCClosed(false)
  24 {
  25
  26 }
  27
  28 DNSRequestParent::~DNSRequestParent()
  29 {
  30
  31 }
  32
  33 void
  34 DNSRequestParent::DoAsyncResolve(const nsACString &hostname, uint32_t flags,
  35                                  const nsACString &networkInterface)
  36 {
  37   nsresult rv;
  38   mFlags = flags;
  39   nsCOMPtr<nsIDNSService> dns = do_GetService(NS_DNSSERVICE_CONTRACTID, &rv);
  40   if (NS_SUCCEEDED(rv)) {
  41     nsCOMPtr<nsIThread> mainThread = do_GetMainThread();
  42     nsCOMPtr<nsICancelable> unused;
  43     rv = dns->AsyncResolveExtended(hostname, flags, networkInterface, this,
  44                                    mainThread, getter_AddRefs(unused));
  45   }
  46
  47   if (NS_FAILED(rv) && !mIPCClosed) {
  48     mIPCClosed = true;
  49     unused << SendLookupCompleted(DNSRequestResponse(rv));
  50   }
  51 }
  52
  53 bool
  54 DNSRequestParent::RecvCancelDNSRequest(const nsCString& hostName,
  55                                        const uint32_t& flags,
  56                                        const nsCString& networkInterface,
  57                                        const nsresult& reason)
  58 {
  59   nsresult rv;
  60   nsCOMPtr<nsIDNSService> dns = do_GetService(NS_DNSSERVICE_CONTRACTID, &rv);
  61   if (NS_SUCCEEDED(rv)) {
  62     rv = dns->CancelAsyncResolveExtended(hostName, flags, networkInterface,
  63                                          this, reason);
  64   }
  65   return true;
  66 }
  67
  68 bool
  69 DNSRequestParent::Recv__delete__()
  70 {
  71   mIPCClosed = true;
  72   return true;
  73 }
  74
  75 void
  76 DNSRequestParent::ActorDestroy(ActorDestroyReason why)
  77 {
  78   // We may still have refcount>0 if DNS hasn't called our OnLookupComplete
  79   // yet, but child process has crashed.  We must not send any more msgs
  80   // to child, or IPDL will kill chrome process, too.
  81   mIPCClosed = true;
  82 }
  83 //-----------------------------------------------------------------------------
  84 // DNSRequestParent::nsISupports
  85 //-----------------------------------------------------------------------------
  86
  87 NS_IMPL_ISUPPORTS(DNSRequestParent,
  88                   nsIDNSListener)
  89
  90 //-----------------------------------------------------------------------------
  91 // nsIDNSListener functions
  92 //-----------------------------------------------------------------------------
  93
  94 NS_IMETHODIMP
  95 DNSRequestParent::OnLookupComplete(nsICancelable *request,
  96                                    nsIDNSRecord  *rec,
  97                                    nsresult       status)
  98 {
  99   if (mIPCClosed) {
 100     // nothing to do: child probably crashed
 101     return NS_OK;
 102   }
 103
 104   if (NS_SUCCEEDED(status)) {
 105     MOZ_ASSERT(rec);
 106
 107     nsAutoCString cname;
 108     if (mFlags & nsHostResolver::RES_CANON_NAME) {
 109       rec->GetCanonicalName(cname);
 110     }
 111
 112     // Get IP addresses for hostname (use port 80 as dummy value for NetAddr)
 113     NetAddrArray array;
 114     NetAddr addr;
 115     while (NS_SUCCEEDED(rec->GetNextAddr(80, &addr))) {
 116       array.AppendElement(addr);
 117     }
 118
 119     unused << SendLookupCompleted(DNSRequestResponse(DNSRecord(cname, array)));
 120   } else {
 121     unused << SendLookupCompleted(DNSRequestResponse(status));
 122   }
 123
 124   mIPCClosed = true;
 125   return NS_OK;
 126 }
 127
 128
 129
 130 }} // mozilla::net