Pass relocations to the elf generator

lib/Language/Elna/Object/Elf.hs

@@ -5,14 +5,18 @@ module Language.Elna.Object.Elf
     , Elf32_Word
     , Elf32_Sword
     , Elf32_Ehdr(..)
+    , Elf32_Rel(..)
+    , Elf32_Rela(..)
     , Elf32_Shdr(..)
     , Elf32_Sym(..)
     , ElfIdentification(..)
     , ElfMachine(..)
     , ElfVersion(..)
-    , ElfType(..)
     , ElfClass(..)
     , ElfData(..)
+    , ElfType(..)
+    , ElfSymbolBinding(..)
+    , ElfSymbolType(..)
     , elf32Addr
     , elf32Half
     , elf32Off
@@ -20,10 +24,15 @@ module Language.Elna.Object.Elf
     , elf32Sword
     , elf32Word
     , elf32Ehdr
+    , elf32Rel
+    , elf32Rela
     , elf32Sym
     , elfIdentification
+    , rInfo
+    , stInfo
     ) where
 
+import Data.Bits (Bits(..))
 import qualified Data.ByteString.Builder as ByteString.Builder
 import Data.Int (Int32)
 import Data.Word (Word8, Word16, Word32)
@@ -118,7 +127,8 @@ data Elf32_Shdr = Elf32_Shdr
     } deriving Eq
 
 data ElfMachine
-    = EM_NONE -- ^ No machine.
+    = ElfMachine Elf32_Half
+    | EM_NONE -- ^ No machine.
     | EM_M32 -- ^ AT&T WE 32100.
     | EM_SPARC -- ^ SPARC.
     | EM_386 -- ^ Intel Architecture.
@@ -140,7 +150,7 @@ instance Enum ElfMachine
     toEnum 7 = EM_860
     toEnum 8 = EM_MIPS
     toEnum 10 = EM_MIPS_RS4_BE
-    toEnum _ = error "Unknown Elf machine"
+    toEnum x = ElfMachine $ fromIntegral x
     fromEnum EM_NONE = 0
     fromEnum EM_M32 = 1
     fromEnum EM_SPARC = 2
@@ -150,9 +160,11 @@ instance Enum ElfMachine
     fromEnum EM_860 = 7
     fromEnum EM_MIPS = 8
     fromEnum EM_MIPS_RS4_BE = 10
+    fromEnum (ElfMachine x) = fromIntegral x
 
 data ElfVersion
-    = EV_NONE -- ^ Invalid versionn.
+    = ElfVersion Elf32_Word
+    | EV_NONE -- ^ Invalid versionn.
     | EV_CURRENT -- ^ Current version.
     deriving Eq
 
@@ -160,12 +172,14 @@ instance Enum ElfVersion
   where
     toEnum 0 = EV_NONE
     toEnum 1 = EV_CURRENT
-    toEnum _ = error "Unknown Elf version"
+    toEnum x = ElfVersion $ fromIntegral x
     fromEnum EV_NONE = 0
     fromEnum EV_CURRENT = 1
+    fromEnum (ElfVersion x) = fromIntegral x
 
 data ElfType
-    = ET_NONE -- ^ No file type.
+    = ElfType Elf32_Half
+    | ET_NONE -- ^ No file type.
     | ET_REL -- ^ Relocatable file.
     | ET_EXEC -- ^ Executable file.
     | ET_DYN -- ^ Shared object file.
@@ -183,7 +197,7 @@ instance Enum ElfType
     toEnum 4 = ET_CORE
     toEnum 0xff00 = ET_LOPROC
     toEnum 0xffff = ET_HIPROC
-    toEnum _ = error "Unknown Elf type"
+    toEnum x = ElfType $ fromIntegral x
     fromEnum ET_NONE = 0
     fromEnum ET_REL = 1
     fromEnum ET_EXEC = 2
@@ -191,6 +205,7 @@ instance Enum ElfType
     fromEnum ET_CORE = 4
     fromEnum ET_LOPROC = 0xff00
     fromEnum ET_HIPROC = 0xffff
+    fromEnum (ElfType x) = fromIntegral x
 
 data Elf32_Sym = Elf32_Sym
     { st_name :: Elf32_Word
@@ -201,7 +216,72 @@ data Elf32_Sym = Elf32_Sym
     , st_shndx :: Elf32_Half 
     } deriving Eq
 
--- * Help types.
+data ElfSymbolBinding
+    = ElfSymbolBinding Word8
+    | STB_LOCAL
+    | STB_GLOBAL
+    | STB_WEAK
+    | STB_LOPROC
+    | STB_HIPROC
+    deriving Eq
+
+instance Enum ElfSymbolBinding
+  where
+    toEnum 0 = STB_LOCAL
+    toEnum 1 = STB_GLOBAL
+    toEnum 2 = STB_WEAK
+    toEnum 13 = STB_LOPROC
+    toEnum 15 = STB_HIPROC
+    toEnum x = ElfSymbolBinding $ fromIntegral x
+    fromEnum STB_LOCAL = 0
+    fromEnum STB_GLOBAL = 1
+    fromEnum STB_WEAK = 2
+    fromEnum STB_LOPROC = 13
+    fromEnum STB_HIPROC = 15
+    fromEnum (ElfSymbolBinding x) = fromIntegral x
+
+data ElfSymbolType
+    = ElfSymbolType Word8
+    | STT_NOTYPE
+    | STT_OBJECT
+    | STT_FUNC
+    | STT_SECTION
+    | STT_FILE
+    | STT_LOPROC
+    | STT_HIPROC
+    deriving Eq
+
+instance Enum ElfSymbolType
+  where
+    toEnum 0 = STT_NOTYPE
+    toEnum 1 = STT_OBJECT
+    toEnum 2 = STT_FUNC
+    toEnum 3 = STT_SECTION
+    toEnum 4 = STT_FILE
+    toEnum 13 = STT_LOPROC
+    toEnum 15 = STT_HIPROC
+    toEnum x = ElfSymbolType $ fromIntegral x
+    fromEnum STT_NOTYPE = 0
+    fromEnum STT_OBJECT = 1
+    fromEnum STT_FUNC = 2
+    fromEnum STT_SECTION = 3
+    fromEnum STT_FILE = 4
+    fromEnum STT_LOPROC = 13
+    fromEnum STT_HIPROC = 15
+    fromEnum (ElfSymbolType x) = fromIntegral x
+
+data Elf32_Rel = Elf32_Rel
+    { r_offset :: Elf32_Addr
+    , r_info :: Elf32_Word
+    } deriving Eq
+
+data Elf32_Rela = Elf32_Rela
+    { r_offset :: Elf32_Addr
+    , r_info :: Elf32_Word
+    , r_addend :: Elf32_Sword
+    } deriving Eq
+
+-- * Help types and functions.
 
 data ByteOrder = LSB | MSB
     deriving Eq
@@ -217,6 +297,9 @@ instance Show ElfEncodingError
     show (ElfUnsupportedClassError class') =
         concat ["Elf class \"", show class', "\" is not supported."]
 
+fromIntegralEnum :: (Enum a, Num b) => a -> b
+fromIntegralEnum = fromIntegral . fromEnum
+
 -- * Encoding functions.
 
 elf32Addr :: ByteOrder -> Elf32_Addr -> ByteString.Builder.Builder
@@ -243,9 +326,9 @@ elfIdentification :: ElfIdentification -> ByteString.Builder.Builder
 elfIdentification (ElfIdentification elfClass elfData)
     = ByteString.Builder.word8 0x7f
     <> ByteString.Builder.string7 "ELF"
-    <> ByteString.Builder.word8 (fromIntegral $ fromEnum elfClass)
-    <> ByteString.Builder.word8 (fromIntegral $ fromEnum elfData)
-    <> ByteString.Builder.word8 (fromIntegral $ fromEnum EV_CURRENT)
+    <> ByteString.Builder.word8 (fromIntegralEnum elfClass)
+    <> ByteString.Builder.word8 (fromIntegralEnum elfData)
+    <> ByteString.Builder.word8 (fromIntegralEnum EV_CURRENT)
     <> ByteString.Builder.byteString (ByteString.replicate 9 0)
 
 elf32Ehdr :: Elf32_Ehdr -> Either ElfEncodingError ByteString.Builder.Builder
@@ -253,9 +336,9 @@ elf32Ehdr Elf32_Ehdr{..} = encode <$> byteOrder
   where
     encode byteOrder'
         = elfIdentification e_ident
-        <> elf32Half byteOrder' (fromIntegral $ fromEnum e_type)
-        <> elf32Half byteOrder' (fromIntegral $ fromEnum e_machine)
-        <> elf32Word byteOrder' (fromIntegral $ fromEnum e_version)
+        <> elf32Half byteOrder' (fromIntegralEnum e_type)
+        <> elf32Half byteOrder' (fromIntegralEnum e_machine)
+        <> elf32Word byteOrder' (fromIntegralEnum e_version)
         <> elf32Addr byteOrder' e_entry
         <> elf32Off byteOrder' e_phoff
         <> elf32Off byteOrder' e_shoff
@@ -294,3 +377,22 @@ elf32Sym byteOrder Elf32_Sym{..}
     <> ByteString.Builder.word8 st_info
     <> ByteString.Builder.word8 st_other
     <> elf32Half byteOrder st_shndx
+
+elf32Rel :: ByteOrder -> Elf32_Rel -> ByteString.Builder.Builder
+elf32Rel byteOrder Elf32_Rel{..}
+    = elf32Addr byteOrder r_offset 
+    <> elf32Word byteOrder r_info
+
+elf32Rela :: ByteOrder -> Elf32_Rela -> ByteString.Builder.Builder
+elf32Rela byteOrder Elf32_Rela{..}
+    = elf32Addr byteOrder r_offset 
+    <> elf32Word byteOrder r_info
+    <> elf32Sword byteOrder r_addend
+
+stInfo :: ElfSymbolBinding -> ElfSymbolType -> Word8
+stInfo binding type' = fromIntegralEnum binding `shiftL` 4
+    .|. (fromIntegralEnum type' .&. 0xf)
+
+rInfo :: Elf32_Word -> Word8 -> Elf32_Word
+rInfo symbol type' = symbol `shiftL` 8
+    .|. fromIntegralEnum type'