Add pprof symbol generation option to run_vm_tests so that we can use pprof in the vm benchmark fra…

bin/run_vm_tests.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 <stdio.h>
 
+#include "bin/file.h"
+
 #include "vm/benchmark_test.h"
 #include "vm/dart.h"
 #include "vm/unit_test.h"
 
 // TODO(iposva, asiva): This is a placeholder for the real unittest framework.
 namespace dart {
 
 // Only run tests that match the filter string. The default does not match any
 // tests.
 static const char* const kNone = "No Test or Benchmarks";
@@ skipping 33 matching lines @@
     this->Run();
     OS::Print("%s(RunTime): %ld\n", this->name(), this->score());
     run_matches++;
   } else if (run_filter == kList) {
     fprintf(stdout, "%s\n", this->name());
     run_matches++;
   }
 }
 
 
+static void DumpPprofSymbolInfo(const char* pprof_filename) {
+  if (pprof_filename != NULL) {
+    char* err = NULL;
+    Dart_Isolate isolate = Dart_CreateIsolate(NULL, NULL, NULL, NULL, &err);
+    EXPECT(isolate != NULL);
+    Dart_EnterScope();
+    File* pprof_file =
+        File::Open(pprof_filename, File::kWriteTruncate);
+    ASSERT(pprof_file != NULL);
+    void* buffer;
+    int buffer_size;
+    Dart_GetPprofSymbolInfo(&buffer, &buffer_size);
+    if (buffer_size > 0) {
+      ASSERT(buffer != NULL);
+      pprof_file->WriteFully(buffer, buffer_size);
+    }
+    delete pprof_file;  // Closes the file.
+    Dart_ExitScope();
+    Dart_ShutdownIsolate();
+  }
+}
+
+
 static void PrintUsage() {
   fprintf(stderr, "run_vm_tests [--list | --benchmarks | "
                   "--tests | --all | <test name> | <benchmark name>]\n");
   fprintf(stderr, "run_vm_tests  <test name> [vm-flags ...]\n");
   fprintf(stderr, "run_vm_tests  <benchmark name> [flags ...]\n");
 }
 
 
 static int Main(int argc, const char** argv) {
   // Flags being passed to the Dart VM.
   int dart_argc = 0;
   const char** dart_argv = NULL;
+  const char* pprof_filename = NULL;
 
   if (argc < 2) {
     // Bad parameter count.
     PrintUsage();
     return 1;
   } else if (argc == 2) {
     if (strcmp(argv[1], "--list") == 0) {
       run_filter = kList;
       // List all tests and benchmarks and exit without initializing the VM.
       TestCaseBase::RunAll();
       Benchmark::RunAll(argv[0]);
       return 0;
     } else if (strcmp(argv[1], "--all") == 0) {
       run_filter = kAll;
     } else if (strcmp(argv[1], "--tests") == 0) {
       run_filter = kAllTests;
     } else if (strcmp(argv[1], "--benchmarks") == 0) {
       run_filter = kAllBenchmarks;
     } else {
       run_filter = argv[1];
     }
   } else {
     // First argument is the test name, the rest are vm flags.
     run_filter = argv[1];
-    // Remove the first two values from the arguments.
-    dart_argc = argc - 2;
-    dart_argv = &argv[2];
+    const char* pprof_option = "--generate_pprof_symbols=";
+    int length = strlen(pprof_option);
+    if (strncmp(pprof_option, argv[2], length) == 0) {
+      pprof_filename = (argv[2] + length);
+      Dart_InitPprofSupport();
+      // Remove the first three values from the arguments.
+      dart_argc = argc - 3;
+      dart_argv = &argv[3];
+    } else {
+      // Remove the first two values from the arguments.
+      dart_argc = argc - 2;
+      dart_argv = &argv[2];
+    }
   }
   bool set_vm_flags_success = Flags::ProcessCommandLineFlags(dart_argc,
                                                              dart_argv);
   ASSERT(set_vm_flags_success);
   bool init_success = Dart::InitOnce(NULL, NULL);
   ASSERT(init_success);
   // Apply the filter to all registered tests.
   TestCaseBase::RunAll();
   // Apply the filter to all registered benchmarks.
   Benchmark::RunAll(argv[0]);
   // Print a warning message if no tests or benchmarks were matched.
   if (run_matches == 0) {
     fprintf(stderr, "No tests matched: %s\n", run_filter);
     return 1;
   }
+  // Dump symbol information for the profiler.
+  DumpPprofSymbolInfo(pprof_filename);
   return 0;
 }
 
 }  // namespace dart
 
 
 int main(int argc, const char** argv) {
   return dart::Main(argc, argv);
 }