Expose classes, codes, functions, and libraries collections.

runtime/vm/service.cc

 // Copyright (c) 2013, 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/debugger.h"
 #include "vm/heap_histogram.h"
 #include "vm/isolate.h"
 #include "vm/message.h"
 #include "vm/object.h"
+#include "vm/object_id_ring.h"
+#include "vm/object_store.h"
 #include "vm/port.h"
 #include "vm/service.h"
 
 namespace dart {
 
 typedef void (*ServiceMessageHandler)(Isolate* isolate, JSONStream* stream);
 
 struct ServiceMessageHandlerEntry {
   const char* command;
   ServiceMessageHandler handler;
@@ skipping 89 matching lines @@
 
       handler(isolate, &js);
       const String& reply = String::Handle(String::New(buffer.buf()));
       ASSERT(!reply.IsNull());
       PostReply(reply, reply_port);
     }
   }
 }
 
 
+static void PrintArgumentsAndOptions(JSONStream* js) {
+  js->OpenObject("message");
+  js->OpenArray("arguments");
+  for (intptr_t i = 0; i < js->num_arguments(); i++) {
+    js->PrintValue(js->GetArgument(i));
+  }
+  js->CloseArray();
+  js->OpenArray("option_keys");
+  for (intptr_t i = 0; i < js->num_options(); i++) {
+    js->PrintValue(js->GetOptionKey(i));
+  }
+  js->CloseArray();
+  js->OpenArray("option_values");
+  for (intptr_t i = 0; i < js->num_options(); i++) {
+    js->PrintValue(js->GetOptionValue(i));
+  }
+  js->CloseArray();
+  js->CloseObject();
+}
+
+
+static void PrintCollectionErrorResponse(const char* collection_name,
+                                         JSONStream* js) {
+  js->OpenObject();
+  js->PrintProperty("type", "error");
+  js->PrintfProperty("text", "Must specify collection object id: /%s/id",
+                     collection_name);
+  js->CloseObject();
+}
+
+
 static void HandleName(Isolate* isolate, JSONStream* js) {
   js->OpenObject();
   js->PrintProperty("type", "IsolateName");
   js->PrintProperty("id", static_cast<intptr_t>(isolate->main_port()));
   js->PrintProperty("name", isolate->name());
   js->CloseObject();
 }
 
 
 static void HandleStackTrace(Isolate* isolate, JSONStream* js) {
@@ skipping 18 matching lines @@
   }
   js->CloseArray();
   js->CloseObject();
 }
 
 
 static void HandleObjectHistogram(Isolate* isolate, JSONStream* js) {
   ObjectHistogram* histogram = Isolate::Current()->object_histogram();
   if (histogram == NULL) {
     js->OpenObject();
-    js->PrintProperty("type", "ObjectHistogram");
-    js->PrintProperty("error", "Run with --print_object_histogram");
+    js->PrintProperty("type", "error");
+    js->PrintProperty("text", "Run with --print_object_histogram");
     js->CloseObject();
     return;
   }
   histogram->PrintToJSONStream(js);
 }
 
 
-static void PrintArgumentsAndOptions(JSONStream* js) {
-  js->OpenObject("message");
-  js->OpenArray("arguments");
-  for (intptr_t i = 0; i < js->num_arguments(); i++) {
-    js->PrintValue(js->GetArgument(i));
-  }
-  js->CloseArray();
-  js->OpenArray("option_keys");
-  for (intptr_t i = 0; i < js->num_options(); i++) {
-    js->PrintValue(js->GetOptionKey(i));
-  }
-  js->CloseArray();
-  js->OpenArray("option_values");
-  for (intptr_t i = 0; i < js->num_options(); i++) {
-    js->PrintValue(js->GetOptionValue(i));
-  }
-  js->CloseArray();
-  js->CloseObject();
-}
-
-
 static void HandleEcho(Isolate* isolate, JSONStream* js) {
   js->OpenObject();
   js->PrintProperty("type", "message");
   PrintArgumentsAndOptions(js);
   js->CloseObject();
 }
 
 
+static void HandleLibraries(Isolate* isolate, JSONStream* js) {
+  if (js->num_arguments() == 1) {
+    js->PrintValue(Library::Handle(isolate->object_store()->root_library()));
+    return;
+  }
+  ASSERT(js->num_arguments() >= 2);
+  ObjectIdRing* ring = isolate->object_id_ring();
+  ASSERT(ring != NULL);
+  intptr_t id = atoi(js->GetArgument(1));
+  Object& obj = Object::Handle(ring->GetObjectForId(id));
+  if (!obj.IsLibrary()) {
+    // Object is not a library, replace with null.
+    obj = Object::null();
+  }
+  js->PrintValue(obj, false);
+}
+
+
+static void HandleFunctions(Isolate* isolate, JSONStream* js) {
+  if (js->num_arguments() == 1) {
+    PrintCollectionErrorResponse("functions", js);
+    return;
+  }
+  ASSERT(js->num_arguments() >= 2);
+  ObjectIdRing* ring = isolate->object_id_ring();
+  ASSERT(ring != NULL);
+  intptr_t id = atoi(js->GetArgument(1));
+  Object& obj = Object::Handle(ring->GetObjectForId(id));
+  if (!obj.IsFunction()) {
+    // Object is not a function, replace with null.
+    obj = Object::null();
+  }
+  js->PrintValue(obj, false);
+}
+
+
+static void HandleClasses(Isolate* isolate, JSONStream* js) {
+  if (js->num_arguments() == 1) {
+    PrintCollectionErrorResponse("classes", js);
+    return;
+  }
+  ASSERT(js->num_arguments() >= 2);
+  ObjectIdRing* ring = isolate->object_id_ring();
+  ASSERT(ring != NULL);
+  intptr_t id = atoi(js->GetArgument(1));
+  Object& obj = Object::Handle(ring->GetObjectForId(id));
+  if (!obj.IsClass()) {
+    // Object is not a class, replace with null.
+    obj = Object::null();
+  }
+  js->PrintValue(obj, false);
+}
+
+
+static void HandleCodes(Isolate* isolate, JSONStream* js) {
+  if (js->num_arguments() == 1) {
+    PrintCollectionErrorResponse("codes", js);
+    return;
+  }
+  ASSERT(js->num_arguments() >= 2);
+  ObjectIdRing* ring = isolate->object_id_ring();
+  ASSERT(ring != NULL);
+  intptr_t id = atoi(js->GetArgument(1));
+  Object& obj = Object::Handle(ring->GetObjectForId(id));
+  if (!obj.IsCode()) {

[Ivan Posva, 2013/08/07 00:39:11]

This code is duplicated across multiple functions here:
if (js->num_arguments() == 1) {
  PrintCollectionErrorResponse("codes", js);
  return;
}
ASSERT(js->num_arguments() >= 2);
ObjectIdRing* ring = isolate->object_id_ring();
ASSERT(ring != NULL);
intptr_t id = atoi(js->GetArgument(1));
Object& obj = Object::Handle(ring->GetObjectForId(id));

You should abstract that out somehow.

[Cutch, 2013/08/07 14:47:48]

Done.

+    // Object is not a code, replace with null.
+    obj = Object::null();
+  }
+  js->PrintValue(obj, false);
+}
+
+
 static ServiceMessageHandlerEntry __message_handlers[] = {
   { "name", HandleName },
   { "stacktrace", HandleStackTrace },
   { "objecthistogram", HandleObjectHistogram},
+  { "libraries", HandleLibraries },
+  { "functions", HandleFunctions },
+  { "classes", HandleClasses },
+  { "codes", HandleCodes },
   { "_echo", HandleEcho },
 };
 
 
 static void HandleFallthrough(Isolate* isolate, JSONStream* js) {
   js->OpenObject();
   js->PrintProperty("type", "error");
   js->PrintProperty("text", "request not understood.");
   PrintArgumentsAndOptions(js);
   js->CloseObject();
 }
 
 
 static ServiceMessageHandler FindServiceMessageHandler(const char* command) {
   intptr_t num_message_handlers = sizeof(__message_handlers) /
                                   sizeof(__message_handlers[0]);
   for (intptr_t i = 0; i < num_message_handlers; i++) {
     const ServiceMessageHandlerEntry& entry = __message_handlers[i];
     if (!strcmp(command, entry.command)) {
       return entry.handler;
     }
   }
   return HandleFallthrough;
 }
 
 }  // namespace dart