バージョン表示機能を追加
[YACASL2.git] / src / casl2.c
index 75902c3..204d307 100644 (file)
@@ -8,6 +8,7 @@
 #include "cerr.h"
 #include "assemble.h"
 #include "exec.h"
+#include "package.h"
 
 /**
  * casl2コマンドのオプション
@@ -26,6 +27,7 @@ static struct option longopts[] = {
     { "dump", no_argument, NULL, 'd' },
     { "memorysize", required_argument, NULL, 'M' },
     { "clocks", required_argument, NULL, 'C' },
+    { "version", no_argument, NULL, 'v' },
     { "help", no_argument, NULL, 'h' },
     { 0, 0, 0, 0 },
 };
@@ -48,32 +50,76 @@ void addcerrlist_casl2()
 /**
  * アセンブル結果を書き込むファイルの名前
  */
-static const char *objfile_name(const char *str)
+const char *objfile_name(const char *str)
 {
     const char *default_name = "a.o";
     return (str == NULL) ? default_name : str;
 }
 
+/**
+ * 指定された1つまたは複数のファイルを2回アセンブル
+ */
+void assemble(int filec, char *filev[])
+{
+    int i;
+    PASS pass;
+    WORD bp[filec];
+
+    create_cmdtype_code();                         /* 命令の名前とタイプがキーのハッシュ表を作成 */
+    asptr = malloc_chk(sizeof(ASPTR), "asptr");    /* アセンブル時のプロパティ用の領域確保 */
+    asptr->prog = malloc_chk(LABELSIZE + 1, "asptr.prog");
+    asptr->ptr = 0;
+    /* アセンブル。ラベル表作成のため、2回行う */
+    for(pass = FIRST; pass <= SECOND; pass++) {
+        for(i = 0; i < filec; i++) {
+            /* データの格納開始位置 */
+            if(pass == FIRST) {
+                bp[i] = asptr->ptr;
+            } else if(pass == SECOND) {
+                asptr->ptr = bp[i];
+            }
+            if(execmode.trace == true || execmode.dump == true ||
+               asmode.src == true || asmode.label == true || asmode.asdetail == true)
+            {
+                fprintf(stdout, "\nAssemble %s (%d)\n", filev[i], pass);
+            }
+            /* ファイルをアセンブル */
+            if(assemblefile(filev[i], pass) == false) {
+                goto asfin;
+            }
+        }
+        if(pass == FIRST && asmode.label == true) {
+            fprintf(stdout, "\nLabel::::\n");
+            printlabel();
+            if(asmode.onlylabel == true) {
+                break;
+            }
+        }
+    }
+asfin:
+    freelabel();                                  /* ラベルハッシュ表を解放 */
+    free_cmdtype_code();                          /* 命令の名前とタイプがキーのハッシュ表を解放 */
+    FREE(asptr->prog);                            /* アセンブル時のプロパティを解放 */
+    FREE(asptr);
+}
+
 /**
  *  casl2コマンドのメイン
  */
 int main(int argc, char *argv[])
 {
-    int memsize = DEFAULT_MEMSIZE, clocks = DEFAULT_CLOCKS;
-    int status = 0, opt, i;
-    PASS pass;
-    bool res = false;
-    WORD beginptr[argc];
-    char *objfile = NULL;
+    int memsize = DEFAULT_MEMSIZE, clocks = DEFAULT_CLOCKS, opt, i, stat;
+    char *af[argc], *objfile = NULL;
+    const char *version = PACKAGE_VERSION,  *cmdversion = "casl2 of YACASL2 version %s\n";
     const char *usage =
-        "Usage: %s [-slLaAtTdh] [-oO[<OBJECTFILE>]] [-M <MEMORYSIZE>] [-C <CLOCKS>] FILE1[ FILE2  ...]\n";
+        "Usage: %s [-slLaAtTdvh] [-oO[<OBJECTFILE>]] [-M <MEMORYSIZE>] [-C <CLOCKS>] FILE1[ FILE2  ...]\n";
 
     cerr_init();
     addcerrlist_casl2();
     addcerrlist_assemble();
     addcerrlist_exec();
     /* オプションの処理 */
-    while((opt = getopt_long(argc, argv, "tTdslLao::O::AM:C:h", longopts, NULL)) != -1) {
+    while((opt = getopt_long(argc, argv, "tTdslLao::O::AM:C:vh", longopts, NULL)) != -1) {
         switch(opt) {
         case 's':
             asmode.src = true;
@@ -115,76 +161,43 @@ int main(int argc, char *argv[])
         case 'C':
             clocks = atoi(optarg);
             break;
+        case 'v':
+            fprintf(stdout, cmdversion, version);
+            return 0;
         case 'h':
             fprintf(stdout, usage, argv[0]);
             return 0;
         case '?':
             fprintf(stderr, usage, argv[0]);
-            exit(-1);
+            exit(1);
         }
     }
     /* ソースファイルが指定されていない場合は終了 */
     if(argv[optind] == NULL) {
-        setcerr(126, NULL);    /* no source file */
+        setcerr(126, "");    /* no source file */
         fprintf(stderr, "CASL2 error - %d: %s\n", cerr->num, cerr->msg);
-        exit(-1);
+        exit(1);
     }
-    /* COMET II仮想マシンのリセット */
-    reset(memsize, clocks);
-    /* アセンブル。ラベル表作成のため、2回行う */
-    for(pass = FIRST; pass <= SECOND; pass++) {
-        if(pass == FIRST) {
-            create_cmdtype_code();        /* 命令の名前とタイプがキーのハッシュ表を作成 */
-            asptr = malloc_chk(sizeof(ASPTR), "asptr"); /* アセンブル時のプロパティ用の領域確保 */
-        }
-        for(i = optind; i < argc; i++) {
-            /* データの格納開始位置 */
-            if(pass == FIRST) {
-                beginptr[i] = asptr->ptr;
-            } else if(pass == SECOND) {
-                asptr->ptr = beginptr[i];
-            }
-            asptr->prog = NULL;
-            if(execmode.trace == true || execmode.dump == true || asmode.src == true ||
-               asmode.label == true || asmode.asdetail == true)
-            {
-                fprintf(stdout, "\nAssemble %s (%d)\n", argv[i], pass);
-            }
-            if((res = assemble(argv[i], pass)) == false) {
-                freecerr();            /* エラーの解放 */
-                exit(-1);
-            }
-        }
-        if(pass == FIRST && asmode.label == true) {
-            fprintf(stdout, "\nLabel::::\n");
-            printlabel();
-            if(asmode.onlylabel == true) {
-                return 0;
-            }
-        }
-        if(pass == SECOND) {
-            freelabel();            /* ラベルハッシュ表を解放 */
-            free_chk(asptr->prog, "asptr.prog"); /* プログラム名を解放 */
-            free_chk(asptr, "asptr");       /* アセンブル時のプロパティを解放 */
-            free_cmdtype_code();    /* 命令の名前とタイプがキーのハッシュ表を解放 */
-        }
+    reset(memsize, clocks);                        /* 仮想マシンCOMET IIのリセット */
+    for(i = 0; i < argc - optind; i++) {           /* 引数からファイル名配列を取得 */
+        af[i] = argv[optind + i];
     }
-    if(res == true) {
-        if(objfile != NULL) {
-            outassemble(objfile);
-            free_chk(objfile, "objfile");
-        }
-        if(asmode.onlyassemble == false) {
-            create_code_type();    /* 命令のコードとタイプがキーのハッシュ表を作成 */
-            res = exec();          /* プログラム実行 */
-            free_code_type();      /* 命令のコードとタイプがキーのハッシュ表を解放 */
-        }
+    assemble(i, af);                               /* アセンブル */
+    if(asmode.onlylabel == true || cerr->num > 0) {
+        goto casl2fin;
+    }
+    /* オブジェクトファイル名が指定されている場合は、アセンブル結果をオブジェクトファイルに出力 */
+    if(objfile != NULL) {
+        outassemble(objfile);
+        FREE(objfile);
     }
-    /* COMET II仮想マシンのシャットダウン */
-    shutdown();
-    if(cerr->num > 0) {
-        status = -1;
+    /* onlyassembleモード以外の場合、仮想マシンCOMET IIを実行 */
+    if(asmode.onlyassemble == false) {
+        exec();                                    /* 仮想マシンCOMET IIの実行 */
     }
-    freecerr();            /* エラーの解放 */
-    return status;
+casl2fin:
+    shutdown();                                    /* 仮想マシンCOMET IIのシャットダウン */
+    stat = (cerr->num == 0) ? 0 : 1;
+    freecerr();                                    /* エラーの解放 */
+    return stat;
 }