PingLoaderImpl should use ContextClient instead of using WeakMember

third_party/WebKit/Source/core/loader/PingLoader.cpp

 /*
  * Copyright (C) 2010 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 13 matching lines @@
  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  *
  */
 
 #include "core/loader/PingLoader.h"
 
+#include "core/dom/ContextLifecycleObserver.h"
 #include "core/dom/DOMArrayBufferView.h"
 #include "core/dom/Document.h"
 #include "core/dom/SecurityContext.h"
 #include "core/fetch/CrossOriginAccessControl.h"
 #include "core/fetch/FetchContext.h"
 #include "core/fetch/FetchInitiatorTypeNames.h"
 #include "core/fetch/FetchUtils.h"
 #include "core/fetch/ResourceFetcher.h"
 #include "core/fetch/UniqueIdentifier.h"
 #include "core/fileapi/File.h"
@@ skipping 136 matching lines @@
 
   const AtomicString getContentType() const { return m_contentType; }
 
  private:
   const Member<FormData> m_data;
   RefPtr<EncodedFormData> m_entityBody;
   AtomicString m_contentType;
 };
 
 class PingLoaderImpl : public GarbageCollectedFinalized<PingLoaderImpl>,
+                       public ContextClient,
                        private WebURLLoaderClient {
+  USING_GARBAGE_COLLECTED_MIXIN(PingLoaderImpl);
   WTF_MAKE_NONCOPYABLE(PingLoaderImpl);
 
  public:
   PingLoaderImpl(LocalFrame*,
                  ResourceRequest&,
                  const AtomicString&,
                  StoredCredentials,
                  bool);
   ~PingLoaderImpl() override;
 
   DECLARE_VIRTUAL_TRACE();
 
  private:
   void dispose();
 
   // WebURLLoaderClient
   bool willFollowRedirect(WebURLRequest&, const WebURLResponse&) override;
   void didReceiveResponse(const WebURLResponse&) final;
   void didReceiveData(const char*, int) final;
   void didFinishLoading(double, int64_t, int64_t) final;
   void didFail(const WebURLError&, int64_t, int64_t) final;
 
   void timeout(TimerBase*);
 
   void didFailLoading(LocalFrame*);
 
-  WeakMember<LocalFrame> m_frame;
   std::unique_ptr<WebURLLoader> m_loader;
   Timer<PingLoaderImpl> m_timeout;
   String m_url;
   unsigned long m_identifier;
   SelfKeepAlive<PingLoaderImpl> m_keepAlive;
 
   bool m_isBeacon;
 
   RefPtr<SecurityOrigin> m_origin;
   CORSEnabled m_corsMode;
 };
 
 PingLoaderImpl::PingLoaderImpl(LocalFrame* frame,
                                ResourceRequest& request,
                                const AtomicString& initiator,
                                StoredCredentials credentialsAllowed,
                                bool isBeacon)
-    : m_frame(frame),
+    : ContextClient(frame),
       m_timeout(this, &PingLoaderImpl::timeout),
       m_url(request.url()),
       m_identifier(createUniqueIdentifier()),
       m_keepAlive(this),
       m_isBeacon(isBeacon),
       m_origin(frame->document()->getSecurityOrigin()),
       m_corsMode(IsCORSEnabled) {
   const AtomicString contentType = request.httpContentType();
   if (!contentType.isNull() &&
       FetchUtils::isSimpleHeader(AtomicString("content-type"), contentType))
@@ skipping 60 matching lines @@
   DCHECK(!newRequest.isNull());
   DCHECK(!redirectResponse.isNull());
 
   String errorDescription;
   ResourceLoaderOptions options;
   // TODO(tyoshino): Save updated data in options.securityOrigin and pass it
   // on the next time.
   if (!CrossOriginAccessControl::handleRedirect(
           m_origin, newRequest, redirectResponse, AllowStoredCredentials,
           options, errorDescription)) {
-    if (m_frame) {
-      if (m_frame->document()) {
-        m_frame->document()->addConsoleMessage(ConsoleMessage::create(
+    if (frame()) {
+      if (frame()->document()) {
+        frame()->document()->addConsoleMessage(ConsoleMessage::create(
             JSMessageSource, ErrorMessageLevel, errorDescription));
       }
     }
     // Cancel the load and self destruct.
     dispose();
 
     return false;
   }
   // FIXME: http://crbug.com/427429 is needed to correctly propagate updates of
   // Origin: following this successful redirect.
 
   return true;
 }
 
 void PingLoaderImpl::didReceiveResponse(const WebURLResponse& response) {
-  if (m_frame) {
+  if (frame()) {
     TRACE_EVENT1("devtools.timeline", "ResourceFinish", "data",
                  InspectorResourceFinishEvent::data(m_identifier, 0, true));
     const ResourceResponse& resourceResponse = response.toResourceResponse();
     InspectorInstrumentation::didReceiveResourceResponse(
-        m_frame, m_identifier, 0, resourceResponse, 0);
-    didFailLoading(m_frame);
+        frame(), m_identifier, 0, resourceResponse, 0);
+    didFailLoading(frame());
   }
   dispose();
 }
 
 void PingLoaderImpl::didReceiveData(const char*, int) {
-  if (m_frame) {
+  if (frame()) {
     TRACE_EVENT1("devtools.timeline", "ResourceFinish", "data",
                  InspectorResourceFinishEvent::data(m_identifier, 0, true));
-    didFailLoading(m_frame);
+    didFailLoading(frame());
   }
   dispose();
 }
 
 void PingLoaderImpl::didFinishLoading(double, int64_t, int64_t) {
-  if (m_frame) {
+  if (frame()) {
     TRACE_EVENT1("devtools.timeline", "ResourceFinish", "data",
                  InspectorResourceFinishEvent::data(m_identifier, 0, true));
-    didFailLoading(m_frame);
+    didFailLoading(frame());
   }
   dispose();
 }
 
 void PingLoaderImpl::didFail(const WebURLError& resourceError,
                              int64_t,
                              int64_t) {
-  if (m_frame) {
+  if (frame()) {
     TRACE_EVENT1("devtools.timeline", "ResourceFinish", "data",
                  InspectorResourceFinishEvent::data(m_identifier, 0, true));
-    didFailLoading(m_frame);
+    didFailLoading(frame());
   }
   dispose();
 }
 
 void PingLoaderImpl::timeout(TimerBase*) {
-  if (m_frame) {
+  if (frame()) {
     TRACE_EVENT1("devtools.timeline", "ResourceFinish", "data",
                  InspectorResourceFinishEvent::data(m_identifier, 0, true));
-    didFailLoading(m_frame);
+    didFailLoading(frame());
   }
   dispose();
 }
 
 void PingLoaderImpl::didFailLoading(LocalFrame* frame) {
   InspectorInstrumentation::didFailLoading(
       frame, m_identifier, ResourceError::cancelledError(m_url));
   frame->console().didFailLoading(m_identifier,
                                   ResourceError::cancelledError(m_url));
 }
 
 DEFINE_TRACE(PingLoaderImpl) {
-  visitor->trace(m_frame);
+  ContextClient::trace(visitor);
 }
 
 void finishPingRequestInitialization(
     ResourceRequest& request,
     LocalFrame* frame,
     WebURLRequest::RequestContext requestContext) {
   request.setRequestContext(requestContext);
   FetchContext& fetchContext = frame->document()->fetcher()->context();
   fetchContext.addAdditionalRequestHeaders(request, FetchSubresource);
   fetchContext.populateRequestData(request);
@@ skipping 151 matching lines @@
 bool PingLoader::sendBeacon(LocalFrame* frame,
                             int allowance,
                             const KURL& beaconURL,
                             Blob* data,
                             int& payloadLength) {
   BeaconBlob beacon(data);
   return sendBeaconCommon(frame, allowance, beaconURL, beacon, payloadLength);
 }
 
 }  // namespace blink