StaticPtrTable.hs 4.43 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
-- | Code generation for the Static Pointer Table
--
-- (c) 2014 I/O Tweag
--
-- Each module that uses 'static' keyword declares an initialization function of
-- the form hs_spt_init_<module>() which is emitted into the _stub.c file and
-- annotated with __attribute__((constructor)) so that it gets executed at
-- startup time.
--
-- The function's purpose is to call hs_spt_insert to insert the static
-- pointers of this module in the hashtable of the RTS, and it looks something
-- like this:
--
-- > static void hs_hpc_init_Main(void) __attribute__((constructor));
-- > static void hs_hpc_init_Main(void) {
-- >
-- >   static StgWord64 k0[2] = {16252233372134256ULL,7370534374096082ULL};
-- >   extern StgPtr Main_r2wb_closure;
-- >   hs_spt_insert(k0, &Main_r2wb_closure);
-- >
-- >   static StgWord64 k1[2] = {12545634534567898ULL,5409674567544151ULL};
-- >   extern StgPtr Main_r2wc_closure;
-- >   hs_spt_insert(k1, &Main_r2wc_closure);
-- >
-- > }
--
-- where the constants are fingerprints produced from the static forms.
--
-- The linker must find the definitions matching the @extern StgPtr <name>@
-- declarations. For this to work, the identifiers of static pointers need to be
-- exported. This is done in TidyPgm.chooseExternalIds.
--
-- There is also a finalization function for the time when the module is
-- unloaded.
--
-- > static void hs_hpc_fini_Main(void) __attribute__((destructor));
-- > static void hs_hpc_fini_Main(void) {
-- >
-- >   static StgWord64 k0[2] = {16252233372134256ULL,7370534374096082ULL};
-- >   hs_spt_remove(k0);
-- >
-- >   static StgWord64 k1[2] = {12545634534567898ULL,5409674567544151ULL};
-- >   hs_spt_remove(k1);
-- >
-- > }
--

{-# LANGUAGE ViewPatterns #-}
module StaticPtrTable (sptModuleInitCode) where

import CLabel
import CoreSyn
import DataCon
import Id
import Literal
import Module
import Outputable
import PrelNames

import Data.Maybe
import GHC.Fingerprint

-- | @sptModuleInitCode module binds@ is a C stub to insert the static entries
-- found in @binds@ of @module@ into the static pointer table.
--
-- A bind is considered a static entry if it is an application of the
-- data constructor @StaticPtr@.
--
sptModuleInitCode :: Module -> CoreProgram -> SDoc
sptModuleInitCode this_mod binds =
    sptInitCode $ catMaybes
                $ map (\(b, e) -> ((,) b) <$> staticPtrFp e)
                $ flattenBinds binds
  where
    staticPtrFp :: CoreExpr -> Maybe Fingerprint
    staticPtrFp (collectTyBinders -> (_, e))
      | (Var v, _ : Lit lit0 : Lit lit1 : _) <- collectArgs e
      , Just con <- isDataConId_maybe v
      , dataConName con == staticPtrDataConName
      , Just w0 <- fromPlatformWord64Rep lit0
      , Just w1 <- fromPlatformWord64Rep lit1
      = Just $ Fingerprint (fromInteger w0) (fromInteger w1)
    staticPtrFp _ = Nothing

    fromPlatformWord64Rep (MachWord w)   = Just w
    fromPlatformWord64Rep (MachWord64 w) = Just w
    fromPlatformWord64Rep _              = Nothing

    sptInitCode :: [(Id, Fingerprint)] -> SDoc
    sptInitCode [] = Outputable.empty
    sptInitCode entries = vcat
      [ text "static void hs_spt_init_" <> ppr this_mod
             <> text "(void) __attribute__((constructor));"
      , text "static void hs_spt_init_" <> ppr this_mod <> text "(void)"
      , braces $ vcat $
          [  text "static StgWord64 k" <> int i <> text "[2] = "
             <> pprFingerprint fp <> semi
          $$ text "extern StgPtr "
             <> (ppr $ mkClosureLabel (idName n) (idCafInfo n)) <> semi
          $$ text "hs_spt_insert" <> parens
               (hcat $ punctuate comma
                  [ char 'k' <> int i
                  , char '&' <> ppr (mkClosureLabel (idName n) (idCafInfo n))
                  ]
               )
          <> semi
          |  (i, (n, fp)) <- zip [0..] entries
          ]
      , text "static void hs_spt_fini_" <> ppr this_mod
             <> text "(void) __attribute__((destructor));"
      , text "static void hs_spt_fini_" <> ppr this_mod <> text "(void)"
      , braces $ vcat $
          [  text "StgWord64 k" <> int i <> text "[2] = "
             <> pprFingerprint fp <> semi
          $$ text "hs_spt_remove" <> parens (char 'k' <> int i) <> semi
          | (i, (_, fp)) <- zip [0..] entries
          ]
      ]

    pprFingerprint :: Fingerprint -> SDoc
    pprFingerprint (Fingerprint w1 w2) =
      braces $ hcat $ punctuate comma
                 [ integer (fromIntegral w1) <> text "ULL"
                 , integer (fromIntegral w2) <> text "ULL"
                 ]