X-Git-Url: http://wien.tomnetworks.com/gitweb/?p=mate.git;a=blobdiff_plain;f=Mate.hs;h=2231a5818be418b39fd6499043eb15927ab9207a;hp=b787c260d3a432a72bb9de885e8f5f6b59a5a673;hb=HEAD;hpb=2cb914fa072ac52bd375324a7ef9b8fc073cb944

diff --git a/Mate.hs b/Mate.hs
index b787c26..2231a58 100644
--- a/Mate.hs
+++ b/Mate.hs
@@ -2,31 +2,73 @@
 module Main where
 
 import System.Environment
+import Data.Char
+import Data.List
+import Data.List.Split
+import qualified Data.ByteString.Lazy as B
+import Control.Monad
 
-import Text.Printf
-
-import JVM.Converter
-import JVM.Dump
+import JVM.ClassFile
+import Java.JAR
 
 import Mate.BasicBlocks
-import Mate.X86CodeGen
 import Mate.MethodPool
+import Mate.Types
+import Mate.ClassPool
+import Mate.NativeMachine
+import Mate.Debug
+
+import Mate.GC.Boehm
 
 main ::  IO ()
 main = do
   args <- getArgs
   register_signal
-  initMethodPool
-  case args of
-    [clspath] -> do
-      cls <- parseClassFile clspath
-      dumpClass cls
-      hmap <- parseMethod cls "main"
-      printMapBB hmap
-      case hmap of
-        Just hmap' -> do
-          entry <- compileBB hmap' cls "main"
-          printf "executing `main' now:\n"
-          executeFuncPtr entry
-        Nothing -> error "main not found"
-    _ -> error "Usage: mate <class-file>"
+  parseArgs args False
+
+parseArgs :: [String] -> Bool -> IO ()
+parseArgs ("-jar":jarpath:_) stdcp = do
+  unless stdcp $ addClassPath "./"
+  addClassPathJAR jarpath
+  res <- readMainClass jarpath
+  case res of
+    Nothing -> error "JAR: no MainClass entry found. Try to pass the jar file via -cp instead."
+    Just mc -> do
+      let bclspath = B.pack . map (fromIntegral . ord) $ mc
+      cls <- getClassFile bclspath
+      executeMain bclspath cls
+
+parseArgs ("-cp":cps) cpset = parseArgs ("-classpath":cps) cpset
+parseArgs ("-classpath":cps:xs) False = do
+  mapM_ addStuff $ splitOn ":" cps
+  parseArgs xs True
+    where
+      addStuff :: String -> IO ()
+      addStuff x
+        | ".jar" `isSuffixOf` x = addClassPathJAR x
+        | otherwise = addClassPath $ x ++ "/"
+parseArgs ("-classpath":xs) _ = parseArgs ("-":xs) True -- usage
+parseArgs (('-':_):_) _ = error "Usage: mate [-cp|-classpath <cp1:cp2:..>] [<class-file> | -jar <jar-file>]"
+-- first argument which isn't prefixed by '-' should be a class file
+parseArgs (clspath:_) stdcp = do
+  unless stdcp $ addClassPath "./"
+  let bclspath = B.pack . map (fromIntegral . ord) $ clspath
+  cls <- getClassFile bclspath
+  executeMain bclspath cls
+parseArgs _ _ = parseArgs ["-"] False
+
+
+executeMain :: B.ByteString -> Class Direct -> IO ()
+executeMain bclspath cls = do 
+  initGC --required on some platforms. [todo bernhard: maybe this should be moved somewhere else - maybe at a global place where vm initialization takes place
+
+  let methods = classMethods cls; methods :: [Method Direct]
+  case find (\x -> methodName x == "main") methods of
+    Just m -> do
+      let mi = MethodInfo "main" bclspath $ methodSignature m
+      rawmethod <- parseMethod cls "main" $ methodSignature m
+      entry <- compileBB mi rawmethod mi
+      addMethodRef entry mi [bclspath]
+      printfInfo "executing `main' now:\n"
+      executeFuncPtr $ fst entry
+    Nothing -> error "main not found"