Correctly parse the directive prologue in modules

src/parsing/parser.cc

 // Copyright 2012 the V8 project 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 "src/parsing/parser.h"
 
 #include "src/api.h"
 #include "src/ast/ast.h"
 #include "src/ast/ast-expression-rewriter.h"
 #include "src/ast/ast-expression-visitor.h"
@@ skipping 1286 matching lines @@
   // (Ecma 262 6th Edition, 15.2):
   // Module :
   //    ModuleBody?
   //
   // ModuleBody :
   //    ModuleItem*
 
   DCHECK(scope_->is_module_scope());
   RaiseLanguageMode(STRICT);
 
+  bool directive_prologue = true;  // Parsing directive prologue.
+
   while (peek() != Token::EOS) {
+    if (directive_prologue && peek() != Token::STRING) {
+      directive_prologue = false;
+    }
+
+    Scanner::Location token_loc = scanner()->peek_location();
     Statement* stat = ParseModuleItem(CHECK_OK);
-    if (stat && !stat->IsEmpty()) {
-      body->Add(stat, zone());
+    if (stat == NULL || stat->IsEmpty()) {
+      directive_prologue = false;  // End of directive prologue.
+      continue;
     }
+
+    if (directive_prologue) {
+      // A shot at a directive.
+      ExpressionStatement* e_stat;
+      Literal* literal;
+      // Still processing directive prologue?
+      if ((e_stat = stat->AsExpressionStatement()) != NULL &&
+          (literal = e_stat->expression()->AsLiteral()) != NULL &&
+          literal->raw_value()->IsString()) {
+        // Ignore "use strict" directive in a module, check "use asm".
+        if (literal->raw_value()->AsString() ==
+                ast_value_factory()->use_asm_string() &&
+            token_loc.end_pos - token_loc.beg_pos ==
+                ast_value_factory()->use_asm_string()->length() + 2) {
+          // Store the usage count; The actual use counter on the isolate is
+          // incremented after parsing is done.
+          ++use_counts_[v8::Isolate::kUseAsm];
+          scope_->SetAsmModule();
+        }
+      } else {
+        // End of the directive prologue.
+        directive_prologue = false;
+      }
+    }
+
+    body->Add(stat, zone());
   }
 
   // Check that all exports are bound.
   ModuleDescriptor* descriptor = scope_->module();
   for (ModuleDescriptor::Iterator it = descriptor->iterator(); !it.done();
        it.Advance()) {
     if (scope_->LookupLocal(it.local_name()) == NULL) {
       // TODO(adamk): Pass both local_name and export_name once ParserTraits
       // supports multiple arg error messages.
       // Also try to report this at a better location.
@@ skipping 5533 matching lines @@
                         try_block, target);
     final_loop = target;
   }
 
   return final_loop;
 }
 
 
 }  // namespace internal
 }  // namespace v8