Fix #16392: revertCAFs in external interpreter when necessary
authorPhuong Trinh <lolotp@fb.com>
Tue, 5 Mar 2019 15:48:46 +0000 (15:48 +0000)
committerMarge Bot <ben+marge-bot@smart-cactus.org>
Thu, 7 Mar 2019 19:01:42 +0000 (14:01 -0500)
We revert CAFs when loading/adding modules in ghci (presumably to refresh
execution states and to allow for object code to be unloaded from the runtime).
However, with `-fexternal-interpreter` enabled, we are only doing it in the
ghci process instead of the external interpreter process where the cafs are
allocated and computed. This makes sure that revertCAFs is done in the
appropriate process no matter if that flag is present or not.

ghc/GHCi/UI/Monad.hs
libraries/ghci/GHCi/Message.hs
libraries/ghci/GHCi/Run.hs
testsuite/tests/ghci/T16392/A.hs [new file with mode: 0644]
testsuite/tests/ghci/T16392/T16392.script [new file with mode: 0644]
testsuite/tests/ghci/T16392/T16392.stderr [new file with mode: 0644]
testsuite/tests/ghci/T16392/T16392.stdout [new file with mode: 0644]
testsuite/tests/ghci/T16392/all.T [new file with mode: 0644]

index 51f1366..fb88727 100644 (file)
@@ -456,14 +456,13 @@ printStats dflags ActionStats{actionAllocs = mallocs, actionElapsedTime = secs}
 
 revertCAFs :: GhciMonad m => m ()
 revertCAFs = do
-  liftIO rts_revertCAFs
+  hsc_env <- GHC.getSession
+  liftIO $ iservCmd hsc_env RtsRevertCAFs
   s <- getGHCiState
   when (not (ghc_e s)) turnOffBuffering
      -- Have to turn off buffering again, because we just
      -- reverted stdout, stderr & stdin to their defaults.
 
-foreign import ccall "revertCAFs" rts_revertCAFs  :: IO ()
-        -- Make it "safe", just in case
 
 -----------------------------------------------------------------------------
 -- To flush buffers for the *interpreted* computation we need
index 959942e..319eebd 100644 (file)
@@ -61,6 +61,7 @@ import System.IO.Error
 data Message a where
   -- | Exit the iserv process
   Shutdown :: Message ()
+  RtsRevertCAFs :: Message ()
 
   -- RTS Linker -------------------------------------------
 
@@ -485,7 +486,9 @@ getMessage = do
       33 -> Msg <$> (AddSptEntry <$> get <*> get)
       34 -> Msg <$> (RunTH <$> get <*> get <*> get <*> get)
       35 -> Msg <$> (GetClosure <$> get)
-      _  -> Msg <$> (Seq <$> get)
+      36 -> Msg <$> (Seq <$> get)
+      37 -> Msg <$> return RtsRevertCAFs
+      _  -> error $ "Unknown Message code " ++ (show b)
 
 putMessage :: Message a -> Put
 putMessage m = case m of
@@ -526,6 +529,7 @@ putMessage m = case m of
   RunTH st q loc ty           -> putWord8 34 >> put st >> put q >> put loc >> put ty
   GetClosure a                -> putWord8 35 >> put a
   Seq a                       -> putWord8 36 >> put a
+  RtsRevertCAFs               -> putWord8 37
 
 -- -----------------------------------------------------------------------------
 -- Reading/writing messages
index 72099b2..a931e62 100644 (file)
@@ -44,9 +44,13 @@ import Unsafe.Coerce
 -- -----------------------------------------------------------------------------
 -- Implement messages
 
+foreign import ccall "revertCAFs" rts_revertCAFs  :: IO ()
+        -- Make it "safe", just in case
+
 run :: Message a -> IO a
 run m = case m of
   InitLinker -> initObjLinker RetainCAFs
+  RtsRevertCAFs -> rts_revertCAFs
   LookupSymbol str -> fmap toRemotePtr <$> lookupSymbol str
   LookupClosure str -> lookupClosure str
   LoadDLL str -> loadDLL str
diff --git a/testsuite/tests/ghci/T16392/A.hs b/testsuite/tests/ghci/T16392/A.hs
new file mode 100644 (file)
index 0000000..31bfb7f
--- /dev/null
@@ -0,0 +1,11 @@
+module A (caf, c_two) where
+
+import Debug.Trace (trace)
+
+data C = C Int Int
+
+caf :: C
+caf = C 3 (trace "value forced" 4)
+
+c_two :: C -> Int
+c_two (C _ b) = b
diff --git a/testsuite/tests/ghci/T16392/T16392.script b/testsuite/tests/ghci/T16392/T16392.script
new file mode 100644 (file)
index 0000000..5fdcb17
--- /dev/null
@@ -0,0 +1,5 @@
+:set -fobject-code
+:load A.hs
+c_two caf
+:load A.hs
+c_two caf
diff --git a/testsuite/tests/ghci/T16392/T16392.stderr b/testsuite/tests/ghci/T16392/T16392.stderr
new file mode 100644 (file)
index 0000000..3473a38
--- /dev/null
@@ -0,0 +1,2 @@
+value forced
+value forced
diff --git a/testsuite/tests/ghci/T16392/T16392.stdout b/testsuite/tests/ghci/T16392/T16392.stdout
new file mode 100644 (file)
index 0000000..7290ba8
--- /dev/null
@@ -0,0 +1,2 @@
+4
+4
diff --git a/testsuite/tests/ghci/T16392/all.T b/testsuite/tests/ghci/T16392/all.T
new file mode 100644 (file)
index 0000000..a77e0fd
--- /dev/null
@@ -0,0 +1,4 @@
+test('T16392',
+     [extra_files(['A.hs']),
+      extra_ways(['ghci-ext'])],
+     ghci_script, ['T16392.script'])