Prototype of I/O statistics for Observatory

runtime/bin/main.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 <stdlib.h>
 #include <string.h>
 #include <stdio.h>
 
 #include "include/dart_api.h"
 #include "include/dart_debugger_api.h"
@@ skipping 792 matching lines @@
     result = Dart_StringToCString(result, &script_source);
     if (Dart_IsError(result)) {
       return result;
     }
     Log::Print("%s\n", script_source);
   }
   return Dart_True();
 }
 
 
+static const char* ServiceRequestError(const char* message) {
+  TextBuffer buffer(128);
+  buffer.Printf("{\"type\":\"Error\",\"text\":\"%s\"", message);
+  return buffer.Steal();
+}
+
+
+static const char* ServiceRequestError(Dart_Handle error) {
+  TextBuffer buffer(128);
+  buffer.Printf("{\"type\":\"Error\",\"text\":\"Internal error %s\"",
+                Dart_GetError(error));
+  return buffer.Steal();
+}
+
+
+class DartScope {
+ public:
+  DartScope() { Dart_EnterScope(); }
+  ~DartScope() { Dart_ExitScope(); }
+};
+
+
+static const char* ServiceRequestHandler(
+    const char* name,
+    const char** arguments,
+    intptr_t num_arguments,
+    const char** option_keys,
+    const char** option_values,
+    intptr_t num_options,
+    void* user_data) {
+  DartScope scope;
+  const char* kSockets = "sockets";
+  if (num_arguments == 2 &&
+      strncmp(arguments[1], kSockets, strlen(kSockets)) == 0) {
+    Dart_Handle dart_io_str = Dart_NewStringFromCString("dart:io");
+    if (Dart_IsError(dart_io_str)) return ServiceRequestError(dart_io_str);
+    Dart_Handle io_lib = Dart_LookupLibrary(dart_io_str);
+    if (Dart_IsError(io_lib)) return ServiceRequestError(io_lib);
+    Dart_Handle handler_function_name =
+        Dart_NewStringFromCString("_socketsStats");
+    if (Dart_IsError(handler_function_name)) {
+      return ServiceRequestError(handler_function_name);
+    }
+    Dart_Handle result = Dart_Invoke(io_lib, handler_function_name, 0, NULL);
+    if (Dart_IsError(result)) return ServiceRequestError(result);
+    const char *json;
+    result = Dart_StringToCString(result, &json);
+    if (Dart_IsError(result)) return ServiceRequestError(result);
+    return strdup(json);
+  } else {
+    return ServiceRequestError("Unrecognized path");
+  }
+}
+
+
 void main(int argc, char** argv) {
   char* script_name;
   CommandLineOptions vm_options(argc);
   CommandLineOptions dart_options(argc);
   bool print_flags_seen = false;
   bool verbose_debug_seen = false;
 
   // Perform platform specific initialization.
   if (!Platform::Initialize()) {
     Log::PrintErr("Initialization failed\n");
@@ skipping 64 matching lines @@
   EventHandler::Start();
 
   // Start the VM service isolate, if necessary.
   if (start_vm_service) {
     ASSERT(vm_service_server_port >= 0);
     bool r = VmService::Start(vm_service_server_port);
     if (!r) {
       Log::PrintErr("Could not start VM Service isolate %s\n",
                     VmService::GetErrorMessage());
     }
+    Dart_RegisterIsolateServiceRequestCallback(
+        "io", &ServiceRequestHandler, NULL);
   }
 
   // Call CreateIsolateAndSetup which creates an isolate and loads up
   // the specified application script.
   char* error = NULL;
   bool is_compile_error = false;
   char* isolate_name = BuildIsolateName(script_name, "main");
   IsolateData* isolate_data = new IsolateData(script_name);
   Dart_Isolate isolate = CreateIsolateAndSetupHelper(script_name,
                                                      "main",
@@ skipping 131 matching lines @@
   exit(Process::GlobalExitCode());
 }
 
 }  // namespace bin
 }  // namespace dart
 
 int main(int argc, char** argv) {
   dart::bin::main(argc, argv);
   UNREACHABLE();
 }