nativeMaschine: s/unsigned int/ptrdiff_t/g

[mate.git] / Mate / Strings.hs

diff --git a/Mate/Strings.hs b/Mate/Strings.hs
index 2867bad5a9c7dd6500604d02d9dee0ff7029f72e..f41f02265c91d31b4d4bb48e40001126248936ed 100644 (file)
--- a/Mate/Strings.hs
+++ b/Mate/Strings.hs
@@ -1,5 +1,6 @@
+{-# LANGUAGE CPP #-}
 {-# LANGUAGE OverloadedStrings #-}
-{-# LANGUAGE ForeignFunctionInterface #-}
+#include "debug.h"
 module Mate.Strings (
   getUniqueStringAddr
   ) where
@@ -8,34 +9,77 @@ import Data.Word
 import qualified Data.Map as M
 import qualified Data.ByteString.Lazy as B
 import qualified Data.ByteString.Internal as BI
+#ifdef DEBUG
 import Text.Printf
+#endif
 
-import Foreign.Ptr
-import Foreign.ForeignPtr
-import Foreign.Marshal.Array
+import JVM.ClassFile
+
+import Foreign
+import Foreign.C.Types
 
 import Mate.Types
+import Mate.ClassPool
+import Mate.Debug
+import Mate.GarbageAlloc
 
 
 getUniqueStringAddr :: B.ByteString -> IO Word32
 getUniqueStringAddr str = do
-  smap <- get_stringsmap >>= ptr2stringsmap
+  smap <- getStringMap
   case M.lookup str smap of
     Nothing -> do
       addr <- allocateJavaString str
-      let smap' = M.insert str addr smap
-      stringsmap2ptr smap' >>= set_stringsmap
+      setStringMap $ M.insert str addr smap
       return addr
     Just addr -> return addr
 
--- TOOD(bernhard): quite hackish
 allocateJavaString :: B.ByteString -> IO Word32
 allocateJavaString str = do
+  {- we have to build a java object layout here, where String object looks like
+   -
+   -  this -+
+   -        |
+   -        v
+   -  +-------------+-------+-------+----------------+--------+
+   -  | MethodTable | value | count | cachedhashcode | offset |
+   -  +-------------+-------+-------+----------------+--------+
+   -        |           |
+   -        |           +------------+
+   -        v                        v
+   -  java/lang/String           +--------+--------+--------+-----+------------------+
+   -                             | length | str[0] | str[1] | ... | str [length - 1] |
+   -                             +--------+--------+--------+-----+------------------+
+   -  all cells are 32bit wide, except str[i] of course (they're 8bit [but
+   -  should be 16bit, TODO]).
+   -}
+  -- build object layout
+  fsize <- getObjectSize "java/lang/String"
+  printfStr "string: fsize: %d (should be 4 * 5)\n" fsize
+  tblptr <- mallocObject $ fromIntegral fsize
+  let ptr = intPtrToPtr (fromIntegral tblptr) :: Ptr CPtrdiff
+  mtbl <- getMethodTable "java/lang/String"
+  poke ptr $ fromIntegral mtbl
+
+  -- build array layout
   let strlen = fromIntegral $ B.length str
-  let str_unpacked = (map fromIntegral $ B.unpack str) :: [Word8]
-  arr <- newArray str_unpacked
-  newstr <- BI.create strlen (\x -> BI.memcpy x arr (fromIntegral strlen))
-  let (newstrptr, _, _) = BI.toForeignPtr newstr
-  let w32_ptr = fromIntegral $ ptrToIntPtr $ unsafeForeignPtrToPtr newstrptr
-  printf "new str ptr: 0x%08x\n" w32_ptr
-  return w32_ptr
+  -- (+1) for \0, (+4) for length
+  newstr <- mallocString (strlen + 5)
+  BI.memset newstr 0 (fromIntegral $ strlen + 5)
+  arr <- newArray ((map fromIntegral $ B.unpack str) :: [Word8])
+  copyBytes (plusPtr newstr 4) arr strlen
+  printfStr "new str ptr: (%s)@%d\n" (toString str) strlen
+
+  let newstr_length = castPtr newstr :: Ptr CPtrdiff
+  poke newstr_length $ fromIntegral strlen
+
+  -- set value pointer
+  poke (plusPtr ptr 4) (fromIntegral (ptrToIntPtr newstr) :: CPtrdiff)
+  -- set count field
+  poke (plusPtr ptr 8) (fromIntegral strlen :: CPtrdiff)
+  -- set hash code (TODO)
+  poke (plusPtr ptr 12) (0 :: CPtrdiff)
+  -- set offset
+  poke (plusPtr ptr 16) (0 :: CPtrdiff)
+
+  return $ fromIntegral tblptr