gecko-dev/netwerk/dns/DNSRequestChild.h

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set sw=2 ts=8 et tw=80 : */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef mozilla_net_DNSRequestChild_h
#define mozilla_net_DNSRequestChild_h

#include "mozilla/net/PDNSRequestChild.h"
#include "nsICancelable.h"
#include "nsIDNSRecord.h"
#include "nsIDNSListener.h"
#include "nsIDNSByTypeRecord.h"
#include "nsIEventTarget.h"

namespace mozilla {
namespace net {

class DNSRequestChild final : public PDNSRequestChild, public nsICancelable {
  friend class PDNSRequestChild;

 public:
  NS_DECL_THREADSAFE_ISUPPORTS
  NS_DECL_NSICANCELABLE

  DNSRequestChild(const nsACString& aHost, const uint16_t& aType,
                  const OriginAttributes& aOriginAttributes,
                  const uint32_t& aFlags, nsIDNSListener* aListener,
                  nsIEventTarget* target);

  void AddIPDLReference() { AddRef(); }
  void ReleaseIPDLReference();

  // Sends IPDL request to parent
  void StartRequest();
  void CallOnLookupComplete();
  void CallOnLookupByTypeComplete();

 protected:
  friend class CancelDNSRequestEvent;
  friend class ChildDNSService;
  virtual ~DNSRequestChild() {}

  mozilla::ipc::IPCResult RecvLookupCompleted(const DNSRequestResponse& reply);
  virtual void ActorDestroy(ActorDestroyReason why) override;

  nsCOMPtr<nsIDNSListener> mListener;
  nsCOMPtr<nsIEventTarget> mTarget;
  nsCOMPtr<nsIDNSRecord> mResultRecord;
  nsCOMPtr<nsIDNSByTypeRecord>
      mResultByTypeRecords;  // the result of a by-type
                             // query (mType must not be
                             // equal to
                             // nsIDNSService::RESOLVE_TYPE_DEFAULT
                             // (this is reserved for
                             // the standard A/AAAA query)).
  nsresult mResultStatus;
  nsCString mHost;
  uint16_t mType;
  const OriginAttributes mOriginAttributes;
  uint16_t mFlags;
  bool mIPCOpen;
};

}  // namespace net
}  // namespace mozilla

#endif  // mozilla_net_DNSRequestChild_h