well-typed · edsko · Jan 12, 2025 · Jan 12, 2025
diff --git a/.github/workflows/haskell-ci.yml b/.github/workflows/haskell-ci.yml
@@ -33,6 +33,11 @@ jobs:
             compilerVersion: 9.12.1
             setup-method: ghcup
             allow-failure: false
+          - compiler: ghc-9.10.1
+            compilerKind: ghc
+            compilerVersion: 9.10.1
+            setup-method: ghcup
+            allow-failure: false
       fail-fast: false
     steps:
       - name: apt-get install

diff --git a/README.md b/README.md
@@ -36,7 +36,8 @@ processing the event log yourself or by using
 
 ## Limitations and future work
 
-* Requires GHC 9.12
+* Requires GHC 9.10 or 9.12
+  (profiling of _pure_ foreign imports requires ghc 9.12).
 * Standard time profiling tools can _NOT_ be used on the eventlog.
 * It is not possible to profile Haskell functions and FFI functions at the
   same time.

diff --git a/example-pkg-A/example-pkg-A.cabal b/example-pkg-A/example-pkg-A.cabal
@@ -12,14 +12,14 @@ category:           Development
 build-type:         Simple
 extra-source-files: cbits/cbits.h
                     cbits/cbits.c
-tested-with:        GHC ==9.12.1
+tested-with:        GHC ==9.10.1
+                    GHC ==9.12.1
 
 common lang
   ghc-options:
       -Wall
   build-depends:
-      -- For now we don't support ghc < 9.12
-      base >= 4.21 && < 4.22
+      base >= 4.20 && < 4.22
   default-language:
       GHC2021
 

diff --git a/example-pkg-B/example-pkg-B.cabal b/example-pkg-B/example-pkg-B.cabal
@@ -12,14 +12,14 @@ author:             Edsko de Vries
 maintainer:         [email protected]
 category:           Development
 build-type:         Simple
-tested-with:        GHC ==9.12.1
+tested-with:        GHC ==9.10.1
+                    GHC ==9.12.1
 
 common lang
   ghc-options:
       -Wall
   build-depends:
-      -- For now we don't support ghc < 9.12
-      base >= 4.21 && < 4.22
+      base >= 4.20 && < 4.22
   default-language:
       GHC2021
 

diff --git a/trace-foreign-calls/src/Plugin/TraceForeignCalls.hs b/trace-foreign-calls/src/Plugin/TraceForeignCalls.hs
@@ -1,5 +1,6 @@
 {-# LANGUAGE OverloadedStrings #-}
-{-# LANGUAGE ViewPatterns #-}
+{-# LANGUAGE ViewPatterns      #-}
+
 {-# OPTIONS_GHC -Wno-incomplete-uni-patterns #-}
 
 module Plugin.TraceForeignCalls (plugin) where
@@ -20,17 +21,20 @@ import GHC.Tc.Utils.Monad qualified as TC
 import GHC.Types.ForeignCall qualified as Foreign
 import GHC.Types.SourceFile (isHsBootOrSig)
 
+import Plugin.TraceForeignCalls.GHC.Shim
+import Plugin.TraceForeignCalls.GHC.Util
 import Plugin.TraceForeignCalls.Instrument
 import Plugin.TraceForeignCalls.Options
-import Plugin.TraceForeignCalls.Util.GHC
 
 {-------------------------------------------------------------------------------
   Top-level
 
   References:
 
-  - https://downloads.haskell.org/ghc/9.12.1/docs/users_guide/extending_ghc.html#compiler-plugins
+  - https://hackage.haskell.org/package/ghc-9.10.1
   - https://hackage.haskell.org/package/ghc-9.12.1
+
+  - https://downloads.haskell.org/ghc/9.12.1/docs/users_guide/extending_ghc.html#compiler-plugins
   - https://downloads.haskell.org/ghc/9.12.1/docs/users_guide/exts/ffi.html
   - https://www.haskell.org/onlinereport/haskell2010/haskellch8.html
 -------------------------------------------------------------------------------}
@@ -69,7 +73,7 @@ processRenamed options tcGblEnv group
 processGroup :: HsGroup GhcRn -> Instrument (HsGroup GhcRn)
 processGroup  group@HsGroup{
                  hs_fords
-               , hs_valds = XValBindsLR (NValBinds bindingGroups sigs)
+               , hs_valds = existingBindings
                } = do
     mTraceCCS <- findName nameTraceCCS
 
@@ -90,13 +94,11 @@ processGroup  group@HsGroup{
           , map reconstructForeignDecl imports
           ]
       , hs_valds =
-          XValBindsLR $
-            NValBinds
-              (map trivialBindingGroup newValues ++ bindingGroups)
-              (                        newSigs   ++ sigs         )
+          extendValBinds
+            (map trivialBindingGroup newValues)
+            newSigs
+            existingBindings
       }
-processGroup HsGroup{hs_valds = ValBinds{}} =
-    error "impossible (ValBinds is only used before renaming)"
 
 {-------------------------------------------------------------------------------
   Foreign declarations
@@ -256,7 +258,6 @@ renameForeignImport (L l n) = do
 mkWrapper :: ReplacedForeignImport -> Instrument (LSig GhcRn, LHsBind GhcRn)
 mkWrapper rfi@ReplacedForeignImport {
               rfiOriginalName
-            , rfiSuffixedName
             , rfiSigType = L _ sigType
             } = do
     (args, body) <- mkWrapperBody rfi
@@ -272,36 +273,7 @@ mkWrapper rfi@ReplacedForeignImport {
                     sig_body = sig_body sigType
                   }
               }
-      , noLocValue $
-          FunBind {
-               fun_ext     = mkNameSet [unLoc rfiSuffixedName]
-             , fun_id      = rfiOriginalName
-             , fun_matches = MG {
-                   mg_ext  = Generated OtherExpansion SkipPmc
-                 , mg_alts = noLocValue . map noLocValue $ [
-                      Match {
-                          m_ext   = noValue
-                        , m_ctxt  = FunRhs {
-                              mc_fun        = rfiOriginalName
-                            , mc_fixity     = Prefix
-                            , mc_strictness = NoSrcStrict
-                            , mc_an         = AnnFunRhs NoEpTok [] []
-                            }
-                        , m_pats  = noLocValue $ map namedVarPat args
-                        , m_grhss = GRHSs {
-                              grhssExt        = emptyComments
-                            , grhssGRHSs      = map noLocValue [
-                                  GRHS
-                                    noValue
-                                    [] -- guards
-                                    body
-                                ]
-                            , grhssLocalBinds = emptyWhereClause
-                            }
-                        }
-                    ]
-                 }
-            }
+      , mkSimpleFunBind rfiOriginalName [] args body
       )
 
 -- | Make the body for the wrapper
@@ -436,12 +408,9 @@ mkEventLogReturn ReplacedForeignImport{rfiOriginalName} =
       ]
 
 {-------------------------------------------------------------------------------
-  Auxiliary
+  Auxiliary: constructions with fresh names
 -------------------------------------------------------------------------------}
 
-trivialBindingGroup :: LHsBind GhcRn -> (RecFlag,  [LHsBind GhcRn])
-trivialBindingGroup binding = (NonRecursive, [binding])
-
 -- | Create unique name for each argument of the function
 uniqArgsFor :: LHsType GhcRn -> Instrument [Name]
 uniqArgsFor = go [] . unLoc
@@ -455,57 +424,6 @@ uniqArgsFor = go [] . unLoc
     go acc _otherTy =
         return $ reverse acc
 
--- | Check if a function signature returns something in the @IO@ monad
-checkIsIO :: LHsSigType GhcRn -> Bool
-checkIsIO = go . unLoc . sig_body . unLoc
-  where
-    go :: HsType GhcRn -> Bool
-    go HsForAllTy{hst_body} = go (unLoc hst_body)
-    go HsQualTy{hst_body}   = go (unLoc hst_body)
-    go (HsFunTy _ _ _ rhs)  = go (unLoc rhs)
-    go ty =
-        case ty of
-          HsAppTy _ (L _ (HsTyVar _ _ (L _ io))) _ | io == Names.ioTyConName ->
-            True
-          _otherwise ->
-            False
-
-emptyWhereClause :: HsLocalBinds GhcRn
-emptyWhereClause = EmptyLocalBinds noValue
-
-ubstringExpr :: String -> LHsExpr GhcRn
-ubstringExpr = noLocValue . HsLit noValue . mkHsStringPrimLit . fsLit
-
-callLNamedFn :: LIdP GhcRn -> [LHsExpr GhcRn] -> LHsExpr GhcRn
-callLNamedFn fn args =
-    mkHsApps (noLocValue $ HsVar noValue fn) $
-      map mkLHsPar args
-
-callNamedFn :: IdP GhcRn -> [LHsExpr GhcRn] -> LHsExpr GhcRn
-callNamedFn = callLNamedFn . noLocValue
-
-namedLVar :: LIdP GhcRn -> LHsExpr GhcRn
-namedLVar = noLocValue . HsVar noValue
-
-namedVar :: IdP GhcRn -> LHsExpr GhcRn
-namedVar = namedLVar . noLocValue
-
-namedVarPat :: Name -> LPat GhcRn
-namedVarPat = noLocValue . VarPat noValue . noLocValue
-
--- | @IO ()@
-ioUnit :: LHsType GhcRn
-ioUnit =
-    nlHsAppTy
-      (nlHsTyVar NotPromoted Names.ioTyConName)
-      (nlHsTyVar NotPromoted (tyConName unitTyCon))
-
-{-------------------------------------------------------------------------------
-  Auxiliary: construct IO calls
--------------------------------------------------------------------------------}
-
-type RealWorld = LHsExpr GhcRn
-
 -- | Bind to value without evaluating it
 --
 -- Given @e@, constructs
@@ -534,11 +452,16 @@ let_ xNameHint e k = do
           }
     return $ noLocValue $
         HsLet noValue (
-          HsValBinds noValue $
-            XValBindsLR $ NValBinds [(NonRecursive, [binding])] []
+            HsValBinds noValue $ mkValBinds [(NonRecursive, [binding])] []
           )
       $ cont
 
+{-------------------------------------------------------------------------------
+  Auxiliary: construct IO calls
+-------------------------------------------------------------------------------}
+
+type RealWorld = LHsExpr GhcRn
+
 -- | Unwrap @IO@ action
 --
 -- Given @io@ and continuation @k@, constructs
@@ -580,7 +503,7 @@ wrapIO f = do
     body <- f (namedVar s)
     return $
         mkHsApp (namedVar Names.ioDataConName)
-      $ mkHsLam (noLocValue [namedVarPat s])
+      $ mkLambda [namedVarPat s]
       $ body
 
 -- | Similar to 'wrapIO', but in a pure context (essentially @unsafePerformIO@)
@@ -599,7 +522,7 @@ runIO f = do
     runRW  <- findName nameRunRW
     noDup  <- findName nameNoDuplicate
     body   <- f $ callNamedFn noDup [namedVar s]
-    let scrut = callNamedFn runRW [mkHsLam (noLocValue [namedVarPat s]) body]
+    let scrut = callNamedFn runRW [mkLambda [namedVarPat s] body]
     return $ noLocValue $
         HsCase CaseAlt scrut
       $ mkMatchGroup (Generated OtherExpansion SkipPmc) . noLocValue . pure
@@ -728,4 +651,3 @@ callNamedIO f args s = do
     fName <- findName f
     return $ callNamedFn fName (args ++ [s])
 
-