Add ability to run the hot reload unit tests using the '--dfe' option

runtime/vm/unit_test.cc

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #include "vm/unit_test.h"
 
 #include <stdio.h>
 
 #include "bin/builtin.h"
 #include "bin/dartutils.h"
@@ skipping 152 matching lines @@
   Dart_Handle dart_args[kNumArgs];
   dart_args[0] = DartUtils::NewString(uri_chars);
   return Dart_Invoke(DartUtils::BuiltinLib(),
                      DartUtils::NewString("_filePathFromUri"), kNumArgs,
                      dart_args);
 }
 
 
 static ThreadLocalKey script_reload_key = kUnsetThreadLocalKey;
 
+static char* CompileTestScriptWithDFE(const char* url,
+                                      const char* source,
+                                      void** kernel_pgm) {
+  Zone* zone = Thread::Current()->zone();
+  char* filename = OS::SCreate(zone, "file:///%s", url);
+  // clang-format off
+  Dart_SourceFile sourcefiles[] = {
+    {
+      filename, source,
+    },
+    {
+      "file:///.packages", "untitled:/"
+    }};
+  // clang-format on
+  int sourcefiles_count = sizeof(sourcefiles) / sizeof(Dart_SourceFile);
+  Dart_KernelCompilationResult compilation_result =
+      Dart_CompileSourcesToKernel(filename, sourcefiles_count, sourcefiles);
+
+  if (compilation_result.status != Dart_KernelCompilationStatus_Ok) {
+    return OS::SCreate(zone, "Compilation failed %s", compilation_result.error);
+  }
+  const uint8_t* kernel_file = compilation_result.kernel;
+  intptr_t kernel_length = compilation_result.kernel_size;
+  if (kernel_file == NULL) {
+    return OS::SCreate(zone, "front end generated a NULL kernel file");
+  }
+  *kernel_pgm = Dart_ReadKernelBinary(kernel_file, kernel_length);
+  if (*kernel_pgm == NULL) {
+    return OS::SCreate(zone, "Failed to read generated kernel binary");
+  }
+  return NULL;
+}
+
 static Dart_Handle LibraryTagHandler(Dart_LibraryTag tag,
                                      Dart_Handle library,
                                      Dart_Handle url) {
+  if (FLAG_use_dart_frontend) {
+    // Reload request.
+    ASSERT(script_reload_key != kUnsetThreadLocalKey);
+    const char* script_source = reinterpret_cast<const char*>(
+        OSThread::GetThreadLocal(script_reload_key));
+    ASSERT(script_source != NULL);
+    OSThread::SetThreadLocal(script_reload_key, 0);
+    const char* urlstr = NULL;
+    Dart_Handle result = Dart_StringToCString(url, &urlstr);
+    if (Dart_IsError(result)) {
+      return Dart_NewApiError("accessing url characters failed");
+    }
+    void* kernel_pgm;
+    char* error = CompileTestScriptWithDFE(urlstr, script_source, &kernel_pgm);
+    if (error == NULL) {
+      return Dart_LoadKernel(kernel_pgm);
+    } else {
+      return Dart_NewApiError(error);
+    }
+  }
   if (tag == Dart_kCanonicalizeUrl) {
     Dart_Handle library_url = Dart_LibraryUrl(library);
     if (Dart_IsError(library_url)) {
       return library_url;
     }
     return Dart_DefaultCanonicalizeUrl(library_url, url);
   }
   if (tag == Dart_kScriptTag) {
     // Reload request.
     ASSERT(script_reload_key != kUnsetThreadLocalKey);
@@ skipping 86 matching lines @@
   DART_CHECK_VALID(lib);
   result = Dart_SetNativeResolver(lib, resolver, NULL);
   DART_CHECK_VALID(result);
   if (finalize_classes) {
     result = Dart_FinalizeLoading(false);
     DART_CHECK_VALID(result);
   }
   return lib;
 }
 
+static Dart_Handle LoadTestScriptWithDFE(const char* script,
+                                         Dart_NativeEntryResolver resolver,
+                                         const char* lib_url,
+                                         bool finalize_classes) {
+  Dart_Handle result = Dart_SetLibraryTagHandler(LibraryTagHandler);
+  EXPECT_VALID(result);
+  void* kernel_pgm = NULL;
+  char* error = CompileTestScriptWithDFE(lib_url, script, &kernel_pgm);
+  if (error == NULL) {
+    Dart_Handle lib = Dart_LoadKernel(kernel_pgm);
+    DART_CHECK_VALID(lib);
+    result = Dart_SetNativeResolver(lib, resolver, NULL);
+    DART_CHECK_VALID(result);
+    if (finalize_classes) {
+      result = Dart_FinalizeLoading(false);
+      DART_CHECK_VALID(result);
+    }
+    return lib;
+  } else {
+    return Dart_NewApiError(error);
+  }
+}
+
 Dart_Handle TestCase::LoadTestScript(const char* script,
                                      Dart_NativeEntryResolver resolver,
                                      const char* lib_url,
                                      bool finalize_classes) {
   if (!FLAG_use_dart_frontend) {
     return LoadTestScriptWithVMParser(script, resolver, lib_url,
                                       finalize_classes);
+  } else {
+    return LoadTestScriptWithDFE(script, resolver, lib_url, finalize_classes);
   }
-
-  Zone* zone = Thread::Current()->zone();
-  char* filename = OS::SCreate(zone, "file:///%s", lib_url);
-  // clang-format off
-  Dart_SourceFile sourcefiles[] = {
-    {
-      filename, script,
-    },
-    {
-      "file:///.packages", "untitled:/"
-    }};
-  // clang-format on
-
-  int sourcefiles_count = sizeof(sourcefiles) / sizeof(Dart_SourceFile);
-  Dart_KernelCompilationResult compilation_result =
-      Dart_CompileSourcesToKernel(filename, sourcefiles_count, sourcefiles);
-
-  if (compilation_result.status != Dart_KernelCompilationStatus_Ok) {
-    return Dart_NewApiError(OS::SCreate(Thread::Current()->zone(),
-                                        "Compilation failed %s",
-                                        compilation_result.error));
-  }
-  const uint8_t* kernel_file = compilation_result.kernel;
-  intptr_t kernel_length = compilation_result.kernel_size;
-  if (kernel_file == NULL) {
-    return Dart_NewApiError(OS::SCreate(
-        Thread::Current()->zone(), "front end generated a NULL kernel file"));
-  }
-  void* kernel_program = Dart_ReadKernelBinary(kernel_file, kernel_length);
-  if (kernel_program == NULL) {
-    return Dart_NewApiError(OS::SCreate(
-        Thread::Current()->zone(), "Failed to read generated kernel binary"));
-  }
-  return Dart_LoadKernel(kernel_program);
 }
 
 #ifndef PRODUCT
 
 
 void TestCase::SetReloadTestScript(const char* script) {
   if (script_reload_key == kUnsetThreadLocalKey) {
     script_reload_key = OSThread::CreateThreadLocal();
   }
   ASSERT(script_reload_key != kUnsetThreadLocalKey);
@@ skipping 202 matching lines @@
   }
   // Copy the remainder of in to out.
   while (*in != '\0') {
     *out++ = *in++;
   }
   *out = '\0';
 }
 
 
 }  // namespace dart