Make ExecutionContext aware about how it can return CoreProbeSink

third_party/WebKit/Source/core/probe/CoreProbes.h

 /*
  * 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 27 matching lines @@
 #include "core/frame/LocalFrame.h"
 #include "core/html/HTMLSlotElement.h"
 #include "core/page/ChromeClient.h"
 #include "platform/probe/PlatformProbes.h"
 
 namespace blink {
 
 class CoreProbeSink;
 class Resource;
 class ThreadDebugger;
-class WorkerGlobalScope;
 
 namespace probe {
 
 class CORE_EXPORT AsyncTask {
   STACK_ALLOCATED();
 
  public:
   AsyncTask(ExecutionContext*,
             void* task,
             const char* step = nullptr,
             bool enabled = true);
   ~AsyncTask();
 
  private:
   ThreadDebugger* debugger_;
   void* task_;
   bool recurring_;
 };
 
 // Called from generated instrumentation code.
-CORE_EXPORT CoreProbeSink* ToCoreProbeSink(WorkerGlobalScope*);
-CORE_EXPORT CoreProbeSink* ToCoreProbeSinkForNonDocumentContext(
-    ExecutionContext*);
-
 inline CoreProbeSink* ToCoreProbeSink(LocalFrame* frame) {
   return frame ? frame->InstrumentingAgents() : nullptr;
 }
 
 inline CoreProbeSink* ToCoreProbeSink(Document& document) {
-  LocalFrame* frame = document.GetFrame();
-  if (!frame && document.TemplateDocumentHost())
-    frame = document.TemplateDocumentHost()->GetFrame();
-  return ToCoreProbeSink(frame);
+  return document.GetProbeSink();
 }
 
 inline CoreProbeSink* ToCoreProbeSink(Document* document) {
   return document ? ToCoreProbeSink(*document) : nullptr;
 }
 
 inline CoreProbeSink* ToCoreProbeSink(ExecutionContext* context) {
-  if (!context)
-    return nullptr;
-  return context->IsDocument() ? ToCoreProbeSink(*ToDocument(context))
-                               : ToCoreProbeSinkForNonDocumentContext(context);
+  return context ? context->GetProbeSink() : nullptr;
 }
 
 inline CoreProbeSink* ToCoreProbeSink(Node* node) {
   return node ? ToCoreProbeSink(node->GetDocument()) : nullptr;
 }
 
 inline CoreProbeSink* ToCoreProbeSink(EventTarget* event_target) {
   return event_target ? ToCoreProbeSink(event_target->GetExecutionContext())
                       : nullptr;
 }
@@ skipping 20 matching lines @@
                                           unsigned long identifier,
                                           const ResourceResponse&,
                                           Resource*);
 
 }  // namespace probe
 }  // namespace blink
 
 #include "core/CoreProbesInl.h"
 
 #endif  // !defined(CoreProbes_h)