js/src/jsapi-tests/testSourcePolicy.cpp

   1 /* This Source Code Form is subject to the terms of the Mozilla Public
   2  * License, v. 2.0. If a copy of the MPL was not distributed with this
   3  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   4
   5 #include "mozilla/Utf8.h"  // mozilla::Utf8Unit
   6
   7 #include "js/CompilationAndEvaluation.h"  // JS::CompileFunction, JS::Evaluate
   8 #include "js/GlobalObject.h"              // JS_NewGlobalObject
   9 #include "js/MemoryFunctions.h"
  10 #include "js/SourceText.h"  // JS::Source{Ownership,Text}
  11 #include "jsapi-tests/tests.h"
  12 #include "vm/JSScript.h"
  13
  14 BEGIN_TEST(testBug795104) {
  15   JS::RealmOptions options;
  16   options.behaviors().setDiscardSource(true);
  17
  18   JS::RootedObject g(cx, JS_NewGlobalObject(cx, getGlobalClass(), nullptr,
  19                                             JS::FireOnNewGlobalHook, options));
  20   CHECK(g);
  21
  22   JSAutoRealm ar(cx, g);
  23
  24   const size_t strLen = 60002;
  25   char* s = static_cast<char*>(JS_malloc(cx, strLen));
  26   CHECK(s);
  27
  28   s[0] = '"';
  29   memset(s + 1, 'x', strLen - 2);
  30   s[strLen - 1] = '"';
  31
  32   JS::SourceText<mozilla::Utf8Unit> srcBuf;
  33   CHECK(srcBuf.init(cx, s, strLen, JS::SourceOwnership::Borrowed));
  34
  35   JS::CompileOptions opts(cx);
  36
  37   // We don't want an rval for our JS::Evaluate call
  38   opts.setNoScriptRval(true);
  39
  40   JS::RootedValue unused(cx);
  41   CHECK(JS::Evaluate(cx, opts, srcBuf, &unused));
  42
  43   JS::RootedFunction fun(cx);
  44   JS::RootedObjectVector emptyScopeChain(cx);
  45
  46   // But when compiling a function we don't want to use no-rval
  47   // mode, since it's not supported for functions.
  48   opts.setNoScriptRval(false);
  49
  50   fun = JS::CompileFunction(cx, emptyScopeChain, opts, "f", 0, nullptr, srcBuf);
  51   CHECK(fun);
  52
  53   JS_free(cx, s);
  54
  55   return true;
  56 }
  57 END_TEST(testBug795104)