Move securityCheck out of V8WrapperInstantiationScope

third_party/WebKit/Source/bindings/core/v8/V8DOMWrapper.h

Index: third_party/WebKit/Source/bindings/core/v8/V8DOMWrapper.h
diff --git a/third_party/WebKit/Source/bindings/core/v8/V8DOMWrapper.h b/third_party/WebKit/Source/bindings/core/v8/V8DOMWrapper.h
index bc0ca27546ad2755bf250fb64e15a136b3ddf7e9..1dba1545c1d85ffda95591948a826c37eaddcbe0 100644
--- a/third_party/WebKit/Source/bindings/core/v8/V8DOMWrapper.h
+++ b/third_party/WebKit/Source/bindings/core/v8/V8DOMWrapper.h
@@ -31,10 +31,10 @@
 #ifndef V8DOMWrapper_h
 #define V8DOMWrapper_h
 
-#include "bindings/core/v8/BindingSecurity.h"
 #include "bindings/core/v8/DOMDataStore.h"
 #include "bindings/core/v8/ScriptWrappable.h"
 #include "bindings/core/v8/V8Binding.h"
+#include "bindings/core/v8/WrapperCreationSecurityCheck.h"
 #include "core/CoreExport.h"
 #include "v8/include/v8.h"
 #include "wtf/Compiler.h"
@@ -125,11 +125,11 @@ class V8WrapperInstantiationScope {
  public:
   V8WrapperInstantiationScope(v8::Local<v8::Object> creationContext,
                               v8::Isolate* isolate,
-                              bool withSecurityCheck)
+                              const WrapperTypeInfo* type)
       : m_didEnterContext(false),
         m_context(isolate->GetCurrentContext()),
         m_tryCatch(isolate),
-        m_convertExceptions(false) {
+        m_type(type) {
     // creationContext should not be empty. Because if we have an
     // empty creationContext, we will end up creating
     // a new object in the context currently entered. This is wrong.
@@ -141,12 +141,15 @@ class V8WrapperInstantiationScope {
     // context is different from the context that we are about to enter.
     if (contextForWrapper == m_context)
       return;
-    if (withSecurityCheck) {
-      securityCheck(isolate, contextForWrapper);
-    } else {
-      m_convertExceptions = true;
-    }
     m_context = v8::Local<v8::Context>::New(isolate, contextForWrapper);

[Yuki, 2017/04/06 08:26:16]

Why do we need to call v8::Local::New?
I thought that
  m_context = contextForWrapper;
is enough.

[adithyas, 2017/04/06 19:04:20]

This is how it was written when V8WrapperInstantiationScope was first written,
I'm not really sure why. I've updated it to m_context = contextForWrapper.

+
+    if (!WrapperCreationSecurityCheck::verifyContextAccess(isolate, m_context,
+                                                           m_type)) {
+      DCHECK(m_tryCatch.HasCaught());
+      m_tryCatch.ReThrow();
+      return;
+    }
+
     m_didEnterContext = true;
     m_context->Enter();
   }
@@ -157,26 +160,26 @@ class V8WrapperInstantiationScope {
       return;
     }
     m_context->Exit();
-    // Rethrow any cross-context exceptions as security error.
-    if (m_tryCatch.HasCaught()) {
-      if (m_convertExceptions) {
-        m_tryCatch.Reset();
-        convertException();
-      }
-      m_tryCatch.ReThrow();
-    }
+
+    v8::Isolate* isolate = m_context->GetIsolate();

[Yuki, 2017/04/06 08:26:15]

Put this line just before its use, or you may not need this variable.

[adithyas, 2017/04/06 19:04:20]

Fixed.

+
+    if (!m_tryCatch.HasCaught())
+      return;
+
+    v8::Local<v8::Value> caughtException = m_tryCatch.Exception();
+    m_tryCatch.Reset();
+    WrapperCreationSecurityCheck::rethrowCrossContextException(
+        isolate, m_context, m_type, caughtException);
+    m_tryCatch.ReThrow();
   }
 
   v8::Local<v8::Context> context() const { return m_context; }
 
  private:
-  void securityCheck(v8::Isolate*, v8::Local<v8::Context> contextForWrapper);
-  void convertException();
-
   bool m_didEnterContext;
   v8::Local<v8::Context> m_context;
   v8::TryCatch m_tryCatch;
-  bool m_convertExceptions;
+  const WrapperTypeInfo* m_type;
 };
 
 }  // namespace blink