fuckeverythingggg
This commit is contained in:
@@ -30,11 +30,6 @@ import qualified Data.Vector.Strict as V
|
|||||||
import Data.IntMap.Strict (IntMap)
|
import Data.IntMap.Strict (IntMap)
|
||||||
|
|
||||||
|
|
||||||
type Emit = Writer (Vector Text)
|
|
||||||
|
|
||||||
runEmit :: Eff (Emit : es) a -> Eff es (a, Text)
|
|
||||||
runEmit = (mapped . _2 %~ fold) . runWriter
|
|
||||||
|
|
||||||
data Env = MkEnv { vars :: Vector Name }
|
data Env = MkEnv { vars :: Vector Name }
|
||||||
deriving (Show, Generic)
|
deriving (Show, Generic)
|
||||||
|
|
||||||
@@ -49,35 +44,11 @@ instance Ixed Env where
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
-- countLocals :: Exp -> Int
|
|
||||||
|
|
||||||
-- countLocals (ExpPrim p rs es) = length rs + sumOf (each . to countLocals) es
|
|
||||||
|
|
||||||
-- countLocals (ExpFix bs e) =
|
|
||||||
-- sumOf (each . _2 . _MkKappa . ((_1 . each . to (const 1))
|
|
||||||
-- <> (_2 . to countLocals))) bs
|
|
||||||
-- + countLocals e
|
|
||||||
|
|
||||||
-- countLocals (ExpApply _ _) = 0
|
|
||||||
|
|
||||||
tshow :: Show a => a -> Text
|
tshow :: Show a => a -> Text
|
||||||
tshow = T.pack . show
|
tshow = T.pack . show
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
data Module = MkModule
|
|
||||||
{ funcs :: IntMap Func
|
|
||||||
}
|
|
||||||
deriving (Show, Generic)
|
|
||||||
|
|
||||||
data Func = MkFunc
|
|
||||||
{ code :: Text
|
|
||||||
, numLocals :: Int
|
|
||||||
}
|
|
||||||
deriving (Show, Generic)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
lowerVal :: Env -> Val -> Vector Text
|
lowerVal :: Env -> Val -> Vector Text
|
||||||
|
|
||||||
lowerVal g (ValLit l) =
|
lowerVal g (ValLit l) =
|
||||||
|
|||||||
@@ -0,0 +1,28 @@
|
|||||||
|
{- HLINT ignore "Use newtype instead of data" -}
|
||||||
|
module Gyehoek.Wasm
|
||||||
|
()
|
||||||
|
where
|
||||||
|
|
||||||
|
import Data.List (List)
|
||||||
|
import GHC.Generics (Generic)
|
||||||
|
|
||||||
|
|
||||||
|
data Module = MkModule
|
||||||
|
{ typeSection :: List Type
|
||||||
|
}
|
||||||
|
deriving (Show, Generic)
|
||||||
|
|
||||||
|
data Type
|
||||||
|
deriving (Show, Generic)
|
||||||
|
|
||||||
|
data Function = MkFunction
|
||||||
|
{ params :: List Type
|
||||||
|
, result :: List Type
|
||||||
|
, locals :: List Type
|
||||||
|
, body :: Expr
|
||||||
|
}
|
||||||
|
deriving (Show, Generic)
|
||||||
|
|
||||||
|
type Expr = List Instr
|
||||||
|
|
||||||
|
type Instr = ByteString
|
||||||
@@ -1,6 +0,0 @@
|
|||||||
module Gyehoek.Wasm.Build
|
|
||||||
() where
|
|
||||||
|
|
||||||
import Gyehoek.Wasm.Syntax
|
|
||||||
import Control.Lens
|
|
||||||
|
|
||||||
@@ -1,307 +0,0 @@
|
|||||||
{-# LANGUAGE OverloadedRecordDot #-}
|
|
||||||
module Gyehoek.Wasm.Syntax
|
|
||||||
( Module(..)
|
|
||||||
, Type
|
|
||||||
, Import
|
|
||||||
, Tag
|
|
||||||
, Global
|
|
||||||
, Mem
|
|
||||||
, Table
|
|
||||||
, Func
|
|
||||||
, Data
|
|
||||||
, Elem
|
|
||||||
, Export
|
|
||||||
, Idx
|
|
||||||
, TypeIdx
|
|
||||||
, FuncIdx
|
|
||||||
, GlobalIdx
|
|
||||||
, TableIdx
|
|
||||||
, MemIdx
|
|
||||||
, TagIdx
|
|
||||||
, ElemIdx
|
|
||||||
, DataIdx
|
|
||||||
, LabelIdx
|
|
||||||
, LocalIdx
|
|
||||||
, FieldIdx
|
|
||||||
) where
|
|
||||||
|
|
||||||
import Data.Vector (Vector)
|
|
||||||
import Data.Word (Word32)
|
|
||||||
import Data.Binary hiding (putList)
|
|
||||||
import Data.Bits
|
|
||||||
import Data.Int (Int8, Int32)
|
|
||||||
import Data.Binary hiding (putList)
|
|
||||||
import Data.List (List)
|
|
||||||
import Data.Generics.Labels
|
|
||||||
import Control.Lens
|
|
||||||
import Data.Text (Text)
|
|
||||||
import Control.Applicative (Alternative(..))
|
|
||||||
import Data.Foldable (traverse_)
|
|
||||||
import Control.Monad
|
|
||||||
import Data.Binary.Put (runPut)
|
|
||||||
import GHC.Generics (Generic)
|
|
||||||
import Data.Coerce (coerce)
|
|
||||||
|
|
||||||
|
|
||||||
type Type = RecType
|
|
||||||
data Import
|
|
||||||
data Tag
|
|
||||||
data Global
|
|
||||||
data Mem
|
|
||||||
data Table
|
|
||||||
data Func
|
|
||||||
data Data
|
|
||||||
data Elem
|
|
||||||
data Export
|
|
||||||
data DataCount
|
|
||||||
data Code = MkCode { locals :: Vector Local, expr :: Expr }
|
|
||||||
|
|
||||||
newtype Idx = MkIdx Word32
|
|
||||||
deriving (Show, Generic)
|
|
||||||
|
|
||||||
newtype TypeIdx = MkType Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
deriving newtype (Binary)
|
|
||||||
newtype FuncIdx = MkFunc Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
deriving newtype (Binary)
|
|
||||||
newtype GlobalIdx = MkGlobal Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
newtype TableIdx = MkTable Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
newtype MemIdx = MkMem Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
newtype TagIdx = MkTag Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
newtype ElemIdx = MkElem Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
newtype DataIdx = MkData Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
newtype LabelIdx = MkLabel Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
newtype LocalIdx = MkLocal Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
newtype FieldIdx = MkField Idx
|
|
||||||
deriving (Show, Generic)
|
|
||||||
|
|
||||||
data Module = MkModule
|
|
||||||
{ types :: Vector Type
|
|
||||||
, imports :: Vector Import
|
|
||||||
, funcs :: Vector Func
|
|
||||||
, tables :: Vector Table
|
|
||||||
, mems :: Vector Mem
|
|
||||||
, tags :: Vector Tag
|
|
||||||
, globals :: Vector Global
|
|
||||||
, exports :: Vector Export
|
|
||||||
, start :: Maybe FuncIdx
|
|
||||||
, elems :: Vector Elem
|
|
||||||
, dataCounts :: Vector DataCount
|
|
||||||
, code :: Vector Code
|
|
||||||
, datas :: Vector Data
|
|
||||||
}
|
|
||||||
|
|
||||||
instance Semigroup Module where
|
|
||||||
m1 <> m2 = MkModule
|
|
||||||
{ types = m1.types <> m2.types
|
|
||||||
, imports = m1.imports <> m2.imports
|
|
||||||
, tags = m1.tags <> m2.tags
|
|
||||||
, globals = m1.globals <> m2.globals
|
|
||||||
, mems = m1.mems <> m2.mems
|
|
||||||
, tables = m1.tables <> m2.tables
|
|
||||||
, funcs = m1.funcs <> m2.funcs
|
|
||||||
, datas = m1.datas <> m2.datas
|
|
||||||
, elems = m1.elems <> m2.elems
|
|
||||||
, start = m1.start <|> m2.start
|
|
||||||
, exports = m1.exports <> m2.exports
|
|
||||||
}
|
|
||||||
|
|
||||||
instance Monoid Module where
|
|
||||||
mempty = MkModule
|
|
||||||
{ types = mempty
|
|
||||||
, imports = mempty
|
|
||||||
, funcs = mempty
|
|
||||||
, tables = mempty
|
|
||||||
, mems = mempty
|
|
||||||
, tags = mempty
|
|
||||||
, globals = mempty
|
|
||||||
, exports = mempty
|
|
||||||
, start = Nothing
|
|
||||||
, elems = mempty
|
|
||||||
, datas = mempty
|
|
||||||
}
|
|
||||||
|
|
||||||
newtype SectionId = MkSectionId Word8
|
|
||||||
deriving newtype (Binary)
|
|
||||||
|
|
||||||
pattern SectionCustom = MkSectionId 0
|
|
||||||
pattern SectionType = MkSectionId 1
|
|
||||||
pattern SectionImport = MkSectionId 2
|
|
||||||
pattern SectionFunction = MkSectionId 3
|
|
||||||
pattern SectionTable = MkSectionId 4
|
|
||||||
pattern SectionMemory = MkSectionId 5
|
|
||||||
pattern SectionGlobal = MkSectionId 6
|
|
||||||
pattern SectionExport = MkSectionId 7
|
|
||||||
pattern SectionStart = MkSectionId 8
|
|
||||||
pattern SectionElement = MkSectionId 9
|
|
||||||
pattern SectionCode = MkSectionId 10
|
|
||||||
pattern SectionData = MkSectionId 11
|
|
||||||
pattern SectionDataCount = MkSectionId 12
|
|
||||||
pattern SectionTag = MkSectionId 13
|
|
||||||
|
|
||||||
newtype RecType = MkRecType { subtypes :: List SubType }
|
|
||||||
|
|
||||||
data SubType
|
|
||||||
= MkSubType { final :: Bool, supertypes :: List TypeIdx, ct :: CompType }
|
|
||||||
deriving (Show, Generic)
|
|
||||||
|
|
||||||
data CompType = CompTypeFunc { to :: List ValType, from :: List ValType }
|
|
||||||
deriving (Show, Generic)
|
|
||||||
|
|
||||||
data ValType
|
|
||||||
= ValTypeNum NumType
|
|
||||||
deriving (Show, Generic)
|
|
||||||
|
|
||||||
data NumType
|
|
||||||
= F64 | F32 | I64 | I32
|
|
||||||
deriving (Show, Generic)
|
|
||||||
|
|
||||||
data Local
|
|
||||||
data Expr
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
getULEB128 :: (Integral a, Bits a) => Int -> Get a
|
|
||||||
getULEB128 bitsBudget = do
|
|
||||||
if bitsBudget > 0 then return () else fail "integer representation too long"
|
|
||||||
val <- getWord8
|
|
||||||
if bitsBudget >= 7 || val .&. 0x7F < 1 `shiftL` bitsBudget then return () else fail "integer too large"
|
|
||||||
if not (testBit val 7)
|
|
||||||
then return $ fromIntegral val
|
|
||||||
else do
|
|
||||||
rest <- getULEB128 (bitsBudget - 7)
|
|
||||||
return $ (fromIntegral $ val .&. 0x7F) .|. (rest `shiftL` 7)
|
|
||||||
|
|
||||||
putULEB128 :: (Integral a, Bits a) => a -> Put
|
|
||||||
putULEB128 val =
|
|
||||||
if val < 128
|
|
||||||
then putWord8 $ fromIntegral val
|
|
||||||
else do
|
|
||||||
putWord8 $ 0x80 + (0x7F .&. fromIntegral val)
|
|
||||||
putULEB128 $ val `shiftR` 7
|
|
||||||
|
|
||||||
getSLEB128 :: (Integral a, Bits a) => Int -> Get a
|
|
||||||
getSLEB128 bitsBudget = do
|
|
||||||
if bitsBudget > 0 then return () else fail "integer representation too long"
|
|
||||||
let toInt8 :: Word8 -> Int8
|
|
||||||
toInt8 = fromIntegral
|
|
||||||
a <- getWord8
|
|
||||||
let mask = (0xFF `shiftL` (bitsBudget - 1)) .&. 0x7F
|
|
||||||
if bitsBudget >= 7 || a .&. mask == 0 || a .&. mask == mask then return () else fail "integer too large"
|
|
||||||
if not (testBit a 7)
|
|
||||||
then return . fromIntegral . toInt8 $ (a .&. 0x7f) .|. ((a .&. 0x40) `shiftL` 1)
|
|
||||||
else do
|
|
||||||
b <- getSLEB128 (bitsBudget - 7)
|
|
||||||
return $ (b `shiftL` 7) .|. (fromIntegral (a .&. 0x7f))
|
|
||||||
|
|
||||||
putSLEB128 :: (Integral a, Bits a) => a -> Put
|
|
||||||
putSLEB128 a = go a
|
|
||||||
where
|
|
||||||
ext = if a >= 0 then 0 else complement 0
|
|
||||||
go x = do
|
|
||||||
let
|
|
||||||
r = x `shiftR` 7
|
|
||||||
w = x .&. 0x7f
|
|
||||||
if r /= ext
|
|
||||||
then do
|
|
||||||
putWord8 (fromIntegral w .|. 0x80)
|
|
||||||
go r
|
|
||||||
else
|
|
||||||
if (testBit w 6 && a < 0) || (not (testBit w 6) && a >= 0)
|
|
||||||
then putWord8 (fromIntegral w)
|
|
||||||
else do
|
|
||||||
putWord8 (fromIntegral w .|. 0x80)
|
|
||||||
putWord8 (fromIntegral ext .&. 0x7F)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
putSection :: SectionId -> Put -> Put
|
|
||||||
putSection i contents =
|
|
||||||
unless (len == 0) $ do
|
|
||||||
put i
|
|
||||||
put (lengthOf each contents')
|
|
||||||
put contents'
|
|
||||||
where
|
|
||||||
contents' = runPut contents
|
|
||||||
len = lengthOf each contents'
|
|
||||||
|
|
||||||
putU32 :: Word32 -> Put
|
|
||||||
putU32 = putULEB128
|
|
||||||
|
|
||||||
putI32 :: Int32 -> Put
|
|
||||||
putI32 = putSLEB128
|
|
||||||
|
|
||||||
putList :: Foldable f => (a -> Put) -> f a -> Put
|
|
||||||
putList f xs = do
|
|
||||||
putU32 (fromIntegral $ length xs)
|
|
||||||
traverse_ f xs
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
putTypeSection :: Vector Type -> Put
|
|
||||||
putTypeSection = putSection SectionType . putList put
|
|
||||||
|
|
||||||
putCodeSection :: Vector Type -> Put
|
|
||||||
putCodeSection = putSection SectionType . putList put
|
|
||||||
|
|
||||||
putStartSection :: Maybe FuncIdx -> Put
|
|
||||||
putStartSection (Just x) = put x
|
|
||||||
putStartSection Nothing = pure ()
|
|
||||||
|
|
||||||
instance Binary Module where
|
|
||||||
get = _
|
|
||||||
put m = do
|
|
||||||
-- magic
|
|
||||||
traverse_ putWord8 [0x00, 0x61, 0x73, 0x6d]
|
|
||||||
-- version
|
|
||||||
traverse_ putWord8 [0x01, 0x00, 0x00, 0x00]
|
|
||||||
-- sections
|
|
||||||
putTypeSection m.types
|
|
||||||
putStartSection m.start
|
|
||||||
|
|
||||||
instance Binary RecType where
|
|
||||||
get = _
|
|
||||||
put rt = case rt.subtypes of
|
|
||||||
[st] -> put st
|
|
||||||
sts -> do
|
|
||||||
putWord8 0x4e
|
|
||||||
putList put sts
|
|
||||||
|
|
||||||
instance Binary SubType where
|
|
||||||
get = _
|
|
||||||
put st = do
|
|
||||||
putWord8 $ if st.final then 0x4f else 0x50
|
|
||||||
putList put st.supertypes
|
|
||||||
put st.ct
|
|
||||||
|
|
||||||
instance Binary CompType where
|
|
||||||
get = _
|
|
||||||
put = \case
|
|
||||||
CompTypeFunc s t -> do
|
|
||||||
putWord8 0x60
|
|
||||||
traverseOf_ both (putList put) (s,t)
|
|
||||||
|
|
||||||
instance Binary ValType where
|
|
||||||
put = \case
|
|
||||||
ValTypeNum nt -> put nt
|
|
||||||
|
|
||||||
instance Binary NumType where
|
|
||||||
get = _
|
|
||||||
put = putWord8 . \case
|
|
||||||
F64 -> 0x7c
|
|
||||||
F32 -> 0x7d
|
|
||||||
I64 -> 0x7e
|
|
||||||
I32 -> 0x7f
|
|
||||||
|
|
||||||
instance Binary Idx where
|
|
||||||
get = _
|
|
||||||
put = putU32 . coerce
|
|
||||||
+1
-1
@@ -43,7 +43,7 @@ executable gyehoek
|
|||||||
Gyehoek.Options
|
Gyehoek.Options
|
||||||
Gyehoek.Scheme.Syntax
|
Gyehoek.Scheme.Syntax
|
||||||
Gyehoek.Sexp
|
Gyehoek.Sexp
|
||||||
Gyehoek.Wasm.Syntax
|
Gyehoek.Wasm
|
||||||
|
|
||||||
build-depends:
|
build-depends:
|
||||||
, base ^>=4.21.2.0
|
, base ^>=4.21.2.0
|
||||||
|
|||||||
Reference in New Issue
Block a user