Makes sure it is OK for the profiled code to send/receive messages

by:

1. Making sure at the end the message with the actual result of the
   whole block is obtained by using a reference.
2. Forwarding all messages the spawned process might have obtained in
   the meantime to the original process.

Author: Qqwy

lib/exprof/macro.ex

@@ -16,13 +16,14 @@ defmodule ExProf.Macro do
   """
   defmacro profile(do: code) do
     quote do
+      ref = make_ref()
       pid = spawn_link(ExProf.Macro, :execute_profile, [fn -> unquote(code) end])
       ExProf.start(pid)
-      send pid, self()
+      send pid, {ref, self()}
 
       result =
         receive do
-          result -> result
+          {^ref, result} -> result
         end
 
       ExProf.stop
@@ -37,8 +38,17 @@ defmodule ExProf.Macro do
   """
   def execute_profile(func) do
     receive do
-      sender ->
-        send sender, func.()
+      {ref, sender} ->
+        send sender, {ref, func.()}
+        forward_other_messages(sender)
+    end
+  end
+
+  defp forward_other_messages(sender) do
+    receive do
+      message ->
+        send sender, message
+        forward_other_messages(sender)
     end
   end
 end