Allow origin trials to be enabled by script

third_party/WebKit/Source/bindings/core/v8/ConditionalFeatures.cpp

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "bindings/core/v8/ConditionalFeatures.h"
 
 #include "bindings/core/v8/ScriptState.h"
 #include "bindings/core/v8/V8Document.h"
 #include "bindings/core/v8/V8HTMLLinkElement.h"
 #include "bindings/core/v8/V8Navigator.h"
 #include "bindings/core/v8/V8Window.h"
 #include "core/dom/ExecutionContext.h"
 #include "core/frame/LocalFrame.h"
 #include "core/origin_trials/OriginTrialContext.h"
+#include "core/origin_trials/OriginTrials.h"
 
 namespace blink {
 
 void installConditionalFeaturesCore(const WrapperTypeInfo* wrapperTypeInfo,
                                     const ScriptState* scriptState,
                                     v8::Local<v8::Object> prototypeObject,
                                     v8::Local<v8::Function> interfaceObject) {
   // TODO(iclelland): Generate all of this logic at compile-time, based on the
   // configuration of origin trial enabled attributes and interfaces in IDL
   // files. (crbug.com/615060)
   ExecutionContext* executionContext = scriptState->getExecutionContext();
   if (!executionContext)
     return;
-  OriginTrialContext* originTrialContext = OriginTrialContext::from(
-      executionContext, OriginTrialContext::DontCreateIfNotExists);
   v8::Isolate* isolate = scriptState->isolate();
   const DOMWrapperWorld& world = scriptState->world();
   if (wrapperTypeInfo == &V8HTMLLinkElement::wrapperTypeInfo) {
-    if (RuntimeEnabledFeatures::linkServiceWorkerEnabled() ||
-        (originTrialContext &&
-         originTrialContext->isTrialEnabled("ForeignFetch"))) {
+    if (OriginTrials::linkServiceWorkerEnabled(executionContext)) {
       V8HTMLLinkElement::installLinkServiceWorker(
           isolate, world, v8::Local<v8::Object>(), prototypeObject,
           interfaceObject);
     }
   } else if (wrapperTypeInfo == &V8Window::wrapperTypeInfo) {
     v8::Local<v8::Object> instanceObject = scriptState->context()->Global();
-    if (RuntimeEnabledFeatures::longTaskObserverEnabled() ||
-        (originTrialContext &&
-         originTrialContext->isTrialEnabled("LongTaskObserver"))) {
+    if (OriginTrials::longTaskObserverEnabled(executionContext)) {
       V8Window::installLongTaskObserver(isolate, world, instanceObject,
                                         prototypeObject, interfaceObject);
     }
   } else if (wrapperTypeInfo == &V8Document::wrapperTypeInfo) {
-    if (RuntimeEnabledFeatures::setRootScrollerEnabled() ||
-        (originTrialContext &&
-         originTrialContext->isTrialEnabled("RootScroller"))) {
+    if (OriginTrials::setRootScrollerEnabled(executionContext)) {
       V8Document::installRootScroller(isolate, world, v8::Local<v8::Object>(),
                                       prototypeObject, interfaceObject);
     }
   }
 }
 
+bool installConditionalFeaturesIfNeededCore(
+    const WrapperTypeInfo* wrapperTypeInfo,
+    const ScriptState* scriptState) {
+  if (wrapperTypeInfo == &V8Window::wrapperTypeInfo) {
+    LOG(INFO) << "Just install, don't look for existing instance for type: "

[iclelland, 2017/01/20 14:31:19]

Is this a debug message to be removed?

[chasej, 2017/01/25 19:58:20]

This method is gone in the new, simplified approach.

+              << wrapperTypeInfo->interfaceName;
+    installConditionalFeatures(wrapperTypeInfo, scriptState,

[iclelland, 2017/01/20 14:31:19]

V8Window is a global object, which is the one case where we typically *do*
generally need the instance variable, in order to expose the constructor
interfaces. It's easy to get, as scriptState->context()->Global(), and it's a
singleton, so we don't need to search for multiple instances; is it actually
unnecessary in this case?

[chasej, 2017/01/25 19:58:20]

This method is gone in the new, simplified approach.

+                               v8::Local<v8::Object>(),
+                               v8::Local<v8::Function>());
+    return true;
+  }
+  return false;
+}
+
+void getAffectedTypesForConditionalFeatureCore(

[iclelland, 2017/01/20 14:31:19]

Rather than modifying its input, could this return a static variable, created
with CR_DEFINE_STATIC_LOCAL? It would save the overhead of creating a set and
inserting into it every time this is called.

[chasej, 2017/01/25 19:58:20]

This method is gone in the new, simplified approach.

+    const String& feature,
+    HashSet<const WrapperTypeInfo*>* affectedTypes) {
+  if (feature == "LongTaskObserver") {
+    affectedTypes->add(&V8Window::wrapperTypeInfo);
+    return;
+  }
+}
+
 namespace {
 
 InstallConditionalFeaturesFunction s_installConditionalFeaturesFunction =
     &installConditionalFeaturesCore;
+
+InstallConditionalFeaturesIfNeededFunction
+    s_installConditionalFeaturesIfNeededFunction =
+        &installConditionalFeaturesIfNeededCore;
+
+GetAffectedTypesForConditionalFeatureFunction
+    s_getAffectedTypesForConditionalFeatureFunction =
+        &getAffectedTypesForConditionalFeatureCore;
 }
 
 InstallConditionalFeaturesFunction setInstallConditionalFeaturesFunction(
     InstallConditionalFeaturesFunction newInstallConditionalFeaturesFunction) {
   InstallConditionalFeaturesFunction originalFunction =
       s_installConditionalFeaturesFunction;
   s_installConditionalFeaturesFunction = newInstallConditionalFeaturesFunction;
   return originalFunction;
 }
 
+InstallConditionalFeaturesIfNeededFunction
+setInstallConditionalFeaturesIfNeededFunction(
+    InstallConditionalFeaturesIfNeededFunction
+        newInstallConditionalFeaturesIfNeededFunction) {
+  InstallConditionalFeaturesIfNeededFunction originalFunction =
+      s_installConditionalFeaturesIfNeededFunction;
+  s_installConditionalFeaturesIfNeededFunction =
+      newInstallConditionalFeaturesIfNeededFunction;
+  return originalFunction;
+}
+
+GetAffectedTypesForConditionalFeatureFunction
+setGetAffectedTypesForConditionalFeatureFunction(
+    GetAffectedTypesForConditionalFeatureFunction
+        newGetAffectedTypesForConditionalFeatureFunction) {
+  GetAffectedTypesForConditionalFeatureFunction originalFunction =
+      s_getAffectedTypesForConditionalFeatureFunction;
+  s_getAffectedTypesForConditionalFeatureFunction =
+      newGetAffectedTypesForConditionalFeatureFunction;
+  return originalFunction;
+}
+
 void installConditionalFeatures(const WrapperTypeInfo* type,
                                 const ScriptState* scriptState,
                                 v8::Local<v8::Object> prototypeObject,
                                 v8::Local<v8::Function> interfaceObject) {
   (*s_installConditionalFeaturesFunction)(type, scriptState, prototypeObject,
                                           interfaceObject);
 }
 
 void installPendingConditionalFeaturesOnWindow(const ScriptState* scriptState) {
   DCHECK(scriptState);
   DCHECK(scriptState->context() == scriptState->isolate()->GetCurrentContext());
   DCHECK(scriptState->perContextData());
   DCHECK(scriptState->world().isMainWorld());
   (*s_installConditionalFeaturesFunction)(&V8Window::wrapperTypeInfo,
                                           scriptState, v8::Local<v8::Object>(),
                                           v8::Local<v8::Function>());
 }
 
+bool installPendingConditionalFeature(const ScriptState* scriptState,
+                                      const String& feature) {
+  DCHECK(scriptState);
+  DCHECK(scriptState->context() == scriptState->isolate()->GetCurrentContext());
+  DCHECK(scriptState->perContextData());
+  DCHECK(scriptState->world().isMainWorld());
+
+  // TODO(chasej): Make the installation of features more granular, rather
+  // than attempting to install all features per type
+  HashSet<const WrapperTypeInfo*> affectedTypes;
+  LOG(INFO) << "Install pending feature: " << feature;

[iclelland, 2017/01/20 14:31:19]

Debug, right?

[chasej, 2017/01/25 19:58:20]

Removed (along with this method being rewritten).

+  (*s_getAffectedTypesForConditionalFeatureFunction)(feature, &affectedTypes);
+
+  bool allInstalled = true;
+  for (auto type : affectedTypes) {
+    LOG(INFO) << "Install on affected type: " << type->interfaceName;
+    if (!((*s_installConditionalFeaturesIfNeededFunction)(type, scriptState)))
+      allInstalled = false;
+  }
+
+  return allInstalled;
+}
+
 bool isFeatureEnabledInFrame(const FeaturePolicy::Feature& feature,
                              const LocalFrame* frame) {
   // If there is no frame, or if feature policy is disabled, use defaults.
   bool enabledByDefault =
       (feature.defaultPolicy == FeaturePolicy::FeatureDefault::EnableForAll ||
        (feature.defaultPolicy == FeaturePolicy::FeatureDefault::EnableForSelf &&
         !frame->isCrossOriginSubframe()));
   if (!RuntimeEnabledFeatures::featurePolicyEnabled() || !frame)
     return enabledByDefault;
   FeaturePolicy* featurePolicy = frame->securityContext()->getFeaturePolicy();
   // The policy should always be initialized before checking it to ensure we
   // properly inherit the parent policy.
   DCHECK(featurePolicy);
 
   // Otherwise, check policy.
   return featurePolicy->isFeatureEnabled(feature);
 }
 
 }  // namespace blink