1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
/* -*- 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_DNSRequestBase_h
#define mozilla_net_DNSRequestBase_h
#include "mozilla/net/PDNSRequestParent.h"
#include "nsICancelable.h"
#include "nsIDNSRecord.h"
#include "nsIDNSListener.h"
#include "nsIDNSByTypeRecord.h"
#include "nsIEventTarget.h"
namespace mozilla {
namespace net {
class DNSRequestActor;
class DNSRequestChild;
class DNSRequestHandler;
class DNSRequestParent;
class DNSRequestSender;
// A base class for DNSRequestSender and DNSRequestHandler.
// Provide interfaces for processing DNS requests.
class DNSRequestBase : public nsISupports {
public:
explicit DNSRequestBase() = default;
void SetIPCActor(DNSRequestActor* aActor);
virtual void OnRecvCancelDNSRequest(const nsCString& hostName,
const nsCString& trrServer,
const uint16_t& type,
const OriginAttributes& originAttributes,
const uint32_t& flags,
const nsresult& reason) = 0;
virtual bool OnRecvLookupCompleted(const DNSRequestResponse& reply) = 0;
virtual void OnIPCActorDestroy() = 0;
virtual DNSRequestSender* AsDNSRequestSender() = 0;
virtual DNSRequestHandler* AsDNSRequestHandler() = 0;
protected:
virtual ~DNSRequestBase() = default;
RefPtr<DNSRequestActor> mIPCActor;
};
// DNSRequestSender is used to send an IPC request to DNSRequestHandler and
// deliver the result to nsIDNSListener.
// Note this class could be used both in content process and parent process.
class DNSRequestSender final : public DNSRequestBase, public nsICancelable {
public:
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSICANCELABLE
DNSRequestSender(const nsACString& aHost, const nsACString& aTrrServer,
const uint16_t& aType,
const OriginAttributes& aOriginAttributes,
const uint32_t& aFlags, nsIDNSListener* aListener,
nsIEventTarget* target);
void OnRecvCancelDNSRequest(const nsCString& hostName,
const nsCString& trrServer, const uint16_t& type,
const OriginAttributes& originAttributes,
const uint32_t& flags,
const nsresult& reason) override;
bool OnRecvLookupCompleted(const DNSRequestResponse& reply) override;
void OnIPCActorDestroy() override;
// Sends IPDL request to DNSRequestHandler
void StartRequest();
void CallOnLookupComplete();
DNSRequestSender* AsDNSRequestSender() override { return this; }
DNSRequestHandler* AsDNSRequestHandler() override { return nullptr; }
private:
friend class ChildDNSService;
virtual ~DNSRequestSender() = default;
nsCOMPtr<nsIDNSListener> mListener;
nsCOMPtr<nsIEventTarget> mTarget;
nsCOMPtr<nsIDNSRecord> mResultRecord;
nsresult mResultStatus;
nsCString mHost;
nsCString mTrrServer;
uint16_t mType;
const OriginAttributes mOriginAttributes;
uint16_t mFlags;
};
// DNSRequestHandler handles the dns request and sends the result back via IPC.
// Note this class could be used both in parent process and socket process.
class DNSRequestHandler final : public DNSRequestBase, public nsIDNSListener {
public:
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSIDNSLISTENER
DNSRequestHandler();
void DoAsyncResolve(const nsACString& hostname, const nsACString& trrServer,
uint16_t type, const OriginAttributes& originAttributes,
uint32_t flags);
void OnRecvCancelDNSRequest(const nsCString& hostName,
const nsCString& trrServer, const uint16_t& type,
const OriginAttributes& originAttributes,
const uint32_t& flags,
const nsresult& reason) override;
bool OnRecvLookupCompleted(const DNSRequestResponse& reply) override;
void OnIPCActorDestroy() override;
DNSRequestSender* AsDNSRequestSender() override { return nullptr; }
DNSRequestHandler* AsDNSRequestHandler() override { return this; }
private:
virtual ~DNSRequestHandler() = default;
uint32_t mFlags;
};
// Provides some common methods for DNSRequestChild and DNSRequestParent.
class DNSRequestActor {
public:
NS_INLINE_DECL_PURE_VIRTUAL_REFCOUNTING
explicit DNSRequestActor(DNSRequestBase* aRequest) : mDNSRequest(aRequest) {}
virtual bool CanSend() const = 0;
virtual DNSRequestChild* AsDNSRequestChild() = 0;
virtual DNSRequestParent* AsDNSRequestParent() = 0;
DNSRequestBase* GetDNSRequest() { return mDNSRequest.get(); };
protected:
virtual ~DNSRequestActor() = default;
RefPtr<DNSRequestBase> mDNSRequest;
};
} // namespace net
} // namespace mozilla
#endif // mozilla_net_DNSRequestBase_h
|
