OOB messages and general message refactor.

runtime/vm/custom_isolate_test.cc

 // Copyright (c) 2011, 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 "include/dart_api.h"
 
 #include "vm/unit_test.h"
 
 // Custom Isolate Test.
 //
@@ skipping 67 matching lines @@
 
 // An entry in our event queue.
 class Event {
  protected:
   Event() : next_(NULL) {}
 
  public:
   virtual ~Event() {}
   virtual void Process() = 0;
 
-  virtual bool IsShutdownEvent(Dart_Isolate isolate) {
-    return false;
-  }
-  virtual bool IsMessageEvent(Dart_Isolate isolate, Dart_Port port) {
+  virtual bool IsMessageEvent(Dart_Isolate isolate) {

[Anton Muhin, 2012/01/12 12:58:16]

maybe now move this logic into StartEvent and add a check that it's never
invoked for StartEvent::isolate_ isolate?  Overall, cannot you provide an access
to isolate_ field and filter just by it?

[turnidge, 2012/01/12 19:01:55]

I have added an isolate_ field to the base Event class and now use that.  Is
this what you meant?

[Anton Muhin, 2012/01/13 14:26:53]

Yes, thank you.

On 2012/01/12 19:01:55, turnidge wrote:

     return false;
   }
 
  private:
   friend class EventQueue;
   Event* next_;
 };
 
 
+// A simple event queue for our test.
+class EventQueue {
+ public:
+  EventQueue() {
+    head_ = NULL;
+  }
+
+  void Add(Event* event) {
+    if (head_ == NULL) {
+      head_ = event;
+      tail_ = event;
+    } else {
+      tail_->next_ = event;
+      tail_ = event;
+    }
+  }
+
+  Event* Get() {
+    if (head_ == NULL) {
+      return NULL;
+    }
+    Event* tmp = head_;
+    head_ = head_->next_;
+    if (head_ == NULL) {
+      tail_ = NULL;

[Anton Muhin, 2012/01/12 12:58:16]

technically, you don't need this assignment, correct? (I see it's moved)

[turnidge, 2012/01/12 19:01:55]

Added a comment.

+    }
+
+    return tmp;
+  }
+
+  void Remove(Dart_Isolate isolate) {

[Anton Muhin, 2012/01/12 12:58:16]

nit: longer name like RemoveEventsForIsolate may make life of reader somewhat
easier.

[turnidge, 2012/01/12 19:01:55]

Done.

+    Event* cur = head_;
+    Event* prev = NULL;
+    while (cur != NULL) {
+      Event* next = cur->next_;
+      if (cur->IsMessageEvent(isolate)) {
+        // Remove matching event.
+        if (prev != NULL) {
+          prev->next_ = next;
+        } else {
+          head_ = next;
+        }
+        delete cur;
+      } else {
+        // Advance.
+        prev = cur;
+      }
+      cur = next;
+    }
+    tail_ = prev;
+  }
+
+ private:
+  Event* head_;
+  Event* tail_;
+};
+Event* current_event;

[Anton Muhin, 2012/01/12 12:58:16]

it looks like you don't need current_event any more, correct?

[turnidge, 2012/01/12 19:01:55]

Thanks, removed.

+EventQueue* event_queue;
+
+
 // Start an isolate.
 class StartEvent : public Event {
  public:
   StartEvent(Dart_Isolate isolate, const char* main)
       : isolate_(isolate), main_(main) {}
 
   virtual void Process();
  private:
   Dart_Isolate isolate_;
   const char* main_;
@@ skipping 32 matching lines @@
                              NULL);
   EXPECT_VALID(result);
   free(const_cast<char*>(main_));
   main_ = NULL;
 
   Dart_ExitScope();
   Dart_ExitIsolate();
 }
 
 
-// Shutdown an isolate.
-class ShutdownEvent : public Event {
+// Notify an isolate of a pending message.
+class MessageEvent : public Event {
  public:
-  explicit ShutdownEvent(Dart_Isolate isolate) : isolate_(isolate) {}
+  explicit MessageEvent(Dart_Isolate isolate) : isolate_(isolate) {}
 
-  virtual bool IsShutdownEvent(Dart_Isolate isolate) {
+  ~MessageEvent() {
+  }
+
+  virtual bool IsMessageEvent(Dart_Isolate isolate) {
     return isolate == isolate_;
   }
 
   virtual void Process();
  private:
   Dart_Isolate isolate_;
 };
 
 
-void ShutdownEvent::Process() {
-  OS::Print("<< ShutdownEvent with isolate(%p)--\n", isolate_);
+void MessageEvent::Process() {
+  OS::Print("$$ MessageEvent with isolate(%p)\n", isolate_);
   Dart_EnterIsolate(isolate_);
-  Dart_ShutdownIsolate();
+  Dart_EnterScope();
+
+  Dart_Handle result = Dart_HandleMessage();
+  EXPECT_VALID(result);
+
+  if (!Dart_HasLivePorts()) {
+    OS::Print("<< Shutting down isolate(%p)\n", isolate_);
+    event_queue->Remove(isolate_);
+    Dart_ShutdownIsolate();
+  } else {
+    Dart_ExitScope();
+    Dart_ExitIsolate();
+  }
+  ASSERT(Dart_CurrentIsolate() == NULL);
 }
 
 
-// Deliver a message to an isolate.
-class MessageEvent : public Event {
- public:
-  MessageEvent(Dart_Isolate isolate, Dart_Port dest, Dart_Port reply,
-               Dart_Message msg)
-      : isolate_(isolate), dest_(dest), reply_(reply), msg_(msg) {}
-
-  ~MessageEvent() {
-    free(msg_);
-    msg_ = NULL;
-  }
-
-  virtual bool IsMessageEvent(Dart_Isolate isolate, Dart_Port port) {
-    return isolate == isolate_ && (port == kCloseAllPorts || port == dest_);
-  }
-
-  virtual void Process();
- private:
-  Dart_Isolate isolate_;
-  Dart_Port dest_;
-  Dart_Port reply_;
-  Dart_Message msg_;
-};
-
-
-void MessageEvent::Process() {
-  OS::Print("$$ MessageEvent with dest port %lld--\n", dest_);
-  Dart_EnterIsolate(isolate_);
-  Dart_EnterScope();
-
-  Dart_Handle result = Dart_HandleMessage(dest_, reply_, msg_);
-  EXPECT_VALID(result);
-
-  Dart_ExitScope();
-  Dart_ExitIsolate();
-}
-
-
-// A simple event queue for our test.
-class EventQueue {
- public:
-  EventQueue() {
-    head_ = NULL;
-  }
-
-  void Add(Event* event) {
-    if (head_ == NULL) {
-      head_ = event;
-      tail_ = event;
-    } else {
-      tail_->next_ = event;
-      tail_ = event;
-    }
-  }
-
-  Event* Get() {
-    if (head_ == NULL) {
-      return NULL;
-    }
-    Event* tmp = head_;
-    head_ = head_->next_;
-    if (head_ == NULL) {
-      tail_ = NULL;
-    }
-
-    return tmp;
-  }
-
-  void ClosePort(Dart_Isolate isolate, Dart_Port port) {
-    Event* cur = head_;
-    Event* prev = NULL;
-    while (cur != NULL) {
-      Event* next = cur->next_;
-      if (cur->IsMessageEvent(isolate, port)) {
-        // Remove matching event.
-        if (prev != NULL) {
-          prev->next_ = next;
-        } else {
-          head_ = next;
-        }
-        delete cur;
-      } else {
-        // Advance.
-        prev = cur;
-      }
-      cur = next;
-    }
-    tail_ = prev;
-  }
-
- private:
-  Event* head_;
-  Event* tail_;
-};
-EventQueue* event_queue;
-Event* current_event;
-
-static bool PostMessage(Dart_Isolate dest_isolate,
-                        Dart_Port dest_port,
-                        Dart_Port reply_port,
-                        Dart_Message message) {
-  OS::Print("-- Posting message dest(%d) reply(%d) --\n",
-            dest_port, reply_port);
+static void NotifyMessage(Dart_Isolate dest_isolate) {
+  OS::Print("-- Notify isolate(%p) of pending message --\n", dest_isolate);
   OS::Print("-- Adding MessageEvent to queue --\n");
-  event_queue->Add(
-      new MessageEvent(dest_isolate, dest_port, reply_port, message));
-  return true;
-}
-
-
-static void ClosePort(Dart_Isolate isolate,
-                      Dart_Port port) {
-  OS::Print("-- Closing port (%lld) for isolate(%p) --\n",
-            port, isolate);
-
-  // Remove any pending events for the isolate/port.
-  event_queue->ClosePort(isolate, port);
-
-  Dart_Isolate current = Dart_CurrentIsolate();
-  if (current) {
-    Dart_ExitIsolate();
-  }
-  Dart_EnterIsolate(isolate);
-  if (!Dart_HasLivePorts() &&
-      (current_event == NULL || !current_event->IsShutdownEvent(isolate))) {
-    OS::Print("-- Adding ShutdownEvent to queue --\n");
-    event_queue->Add(new ShutdownEvent(isolate));
-  }
-  Dart_ExitIsolate();
-  if (current) {
-    Dart_EnterIsolate(current);
-  }
+  event_queue->Add(new MessageEvent(dest_isolate));
 }
 
 
 static Dart_NativeFunction NativeLookup(Dart_Handle name, int argc) {
   const char* name_str = NULL;
   EXPECT(Dart_IsString(name));
   EXPECT_VALID(Dart_StringToCString(name, &name_str));
   if (strcmp(name_str, "native_echo") == 0) {
     return &native_echo;
   } else if (strcmp(name_str, "CustomIsolateImpl_start") == 0) {
@@ skipping 34 matching lines @@
   EXPECT_VALID(Dart_StringToCString(param, &isolate_main));
   isolate_main = strdup(isolate_main);
 
   // Save current isolate.
   Dart_Isolate saved_isolate = Dart_CurrentIsolate();
   Dart_ExitIsolate();
 
   // Create a new Dart_Isolate.
   Dart_Isolate new_isolate = TestCase::CreateTestIsolate();
   EXPECT(new_isolate != NULL);
-  Dart_SetMessageCallbacks(&PostMessage, &ClosePort);
+  Dart_SetMessageNotifyCallback(&NotifyMessage);
   Dart_Port new_port = Dart_GetMainPortId();
 
   OS::Print("-- Adding StartEvent to queue --\n");
   event_queue->Add(new StartEvent(new_isolate, isolate_main));
 
   // Restore the original isolate.
   Dart_ExitIsolate();
   Dart_EnterIsolate(saved_isolate);
   Dart_EnterScope();
 
   Dart_Handle send_port = Dart_NewSendPort(new_port);
   EXPECT_VALID(send_port);
   Dart_SetReturnValue(args, send_port);
 
   OS::Print("-- Exit: CustomIsolateImpl_start --\n");
   Dart_ExitScope();
 }
 
 
 UNIT_TEST_CASE(CustomIsolates) {
   event_queue = new EventQueue();
   current_event = NULL;
 
   Dart_Isolate dart_isolate = TestCase::CreateTestIsolate();
   EXPECT(dart_isolate != NULL);
-  Dart_SetMessageCallbacks(&PostMessage, &ClosePort);
+  Dart_SetMessageNotifyCallback(&NotifyMessage);
   Dart_EnterScope();
   Dart_Handle result;
 
   // Create a test library.
   Dart_Handle lib = TestCase::LoadTestScript(kCustomIsolateScriptChars,
                                              NativeLookup);
   EXPECT_VALID(lib);
 
   // Run main.
   result = Dart_InvokeStatic(lib,
@@ skipping 22 matching lines @@
   OS::Print("-- Finished event loop --\n");
   EXPECT_STREQ("Received: 43", saved_echo);
   free(const_cast<char*>(saved_echo));
 
   delete event_queue;
 }
 
 #endif  // TARGET_ARCH_IA32.
 
 }  // namespace dart