X-Git-Url: http://j8takagi.net/cgi-bin/gitweb.cgi?p=YACASL2.git;a=blobdiff_plain;f=src%2Fcasl2.c;h=c7ef4f034c167bb37312cf2bbb508c978a97b654;hp=68cbb24f32087a36b9d08054caa540bccc3378cd;hb=86e559d164166966a797a1e5855871d48e087ddd;hpb=21b0a348f1e1660a3da0ebf0c4e7009a4ac11c71

diff --git a/src/casl2.c b/src/casl2.c
index 68cbb24..c7ef4f0 100644
--- a/src/casl2.c
+++ b/src/casl2.c
@@ -1,5 +1,6 @@
 #include "package.h"
 #include "assemble.h"
+#include "exec.h"
 
 /**
  * @brief CASL IIのエラーをエラーリストに追加
@@ -15,7 +16,7 @@ void addcerrlist_casl2();
  *
  * @param *str ファイル名
  */
-const char *objfile_name(const char *str);
+char *objfile_name(const char *str);
 
 /**
  * @brief casl2コマンドのオプション
@@ -45,6 +46,7 @@ static struct option longopts[] = {
  */
 CERR cerr_casl2[] = {
     { 126, "no source file" },
+    { 127, "invalid option" },
 };
 
 void addcerrlist_casl2()
@@ -52,10 +54,13 @@ void addcerrlist_casl2()
     addcerrlist(ARRAYSIZE(cerr_casl2), cerr_casl2);
 }
 
-const char *objfile_name(const char *str)
+char *objfile_name(const char *name)
 {
     const char *default_name = "a.o";
-    return (str == NULL) ? default_name : str;
+    return strdup_chk(
+        (name == NULL || !name[0]) ? default_name : name,
+        "objfile_name"
+        );
 }
 
 /**
@@ -68,12 +73,24 @@ const char *objfile_name(const char *str)
  */
 int main(int argc, char *argv[])
 {
-    int memsize = DEFAULT_MEMSIZE, clocks = DEFAULT_CLOCKS, opt, i, stat = 0;
-    char *af[argc], *objfile = NULL;
-    const char *version = PACKAGE_VERSION,  *cmdversion = "casl2 of YACASL2 version %s\n";
+    int memsize = DEFAULT_MEMSIZE;
+    int clocks = DEFAULT_CLOCKS;
+    int opt = 0;
+    int stat = 0;
+    int asfilecnt = 0;
+    char **asfile = NULL;
+    char *objfile = NULL;
+    const char *version = PACKAGE_VERSION;
+    const char *cmdversion = "casl2 of YACASL2 version %s\n";
     const char *usage =
         "Usage: %s [-slLaAtTdmvh] [-oO[<OBJECTFILE>]] [-M <MEMORYSIZE>] [-C <CLOCKS>] FILE1[ FILE2  ...]\n";
 
+    /* エラーの定義 */
+    cerr_init();
+    addcerrlist_casl2();
+    addcerrlist_assemble();
+    addcerrlist_exec();
+
     /* オプションの処理 */
     while((opt = getopt_long(argc, argv, "tTdslLmao::O::AM:C:vh", longopts, NULL)) != -1) {
         switch(opt) {
@@ -95,11 +112,11 @@ int main(int argc, char *argv[])
             asmode.onlyassemble = true;
             break;
         case 'o':
-            objfile = strdup_chk(objfile_name(optarg), "objfile");
+            objfile = objfile_name(optarg);
             break;
         case 'O':
             asmode.onlyassemble = true;
-            objfile = strdup_chk(objfile_name(optarg), "objfile");
+            objfile = objfile_name(optarg);
             break;
         case 't':
             execmode.trace = true;
@@ -122,51 +139,51 @@ int main(int argc, char *argv[])
             break;
         case 'v':
             fprintf(stdout, cmdversion, version);
-            return 0;
+            goto casl2fin;
         case 'h':
             fprintf(stdout, usage, argv[0]);
-            return 0;
+            goto casl2fin;
         case '?':
             fprintf(stderr, usage, argv[0]);
-            exit(1);
+            setcerr(212, "");    /* invalid option */
+            goto casl2fin;
         }
     }
 
-    /* エラーの定義 */
-    cerr_init();
-    addcerrlist_casl2();
-    addcerrlist_assemble();
-    addcerrlist_exec();
-
     /* ソースファイルが指定されていない場合は終了 */
     if(argv[optind] == NULL) {
         setcerr(126, "");    /* no source file */
-        fprintf(stderr, "CASL2 error - %d: %s\n", cerr->num, cerr->msg);
-        freecerr();                                    /* エラーの解放 */
-        exit(1);
+        fprintf(stderr, "casl2 error - %d: %s\n", cerr->num, cerr->msg);
+        goto casl2fin;
     }
     create_cmdtable(HASH_CMDTYPE);                 /* 命令の名前とタイプがキーのハッシュ表を作成 */
     reset(memsize, clocks);                        /* 仮想マシンCOMET IIのリセット */
-    for(i = 0; i < argc - optind; i++) {           /* 引数からファイル名配列を取得 */
-        af[i] = argv[optind + i];
+    asfilecnt = argc - optind;
+    asfile = calloc_chk(asfilecnt, sizeof(char *), "asfile");
+    for(int i = 0; i < asfilecnt; i++) {           /* 引数からファイル名配列を取得 */
+        asfile[i] = argv[optind + i];
     }
-    assemble(i, af, 0);                            /* アセンブル */
-    if(asmode.onlylabel == true || cerr->num > 0) {
-        goto casl2fin;
+    /* アセンブル */
+    if(assemble(asfilecnt, asfile, 0) == false || asmode.onlylabel == true) {
+        goto shutdown;
     }
     /* オブジェクトファイル名が指定されている場合は、アセンブル結果をオブジェクトファイルに出力 */
     if(objfile != NULL) {
         outassemble(objfile);
-        FREE(objfile);
     }
     /* onlyassembleモード以外の場合、仮想マシンCOMET IIを実行 */
     if(asmode.onlyassemble == false) {
         exec();                                    /* 仮想マシンCOMET IIの実行 */
     }
+shutdown:
+    shutdown();                                   /* 仮想マシンCOMET IIのシャットダウン */
 casl2fin:
-    shutdown();                                    /* 仮想マシンCOMET IIのシャットダウン */
+    FREE(objfile);
+    FREE(asfile);
     free_cmdtable(HASH_CMDTYPE);
-    stat = (cerr->num == 0) ? 0 : 1;
+    if(cerr->num > 0) {
+        stat = 1;
+    }
     freecerr();                                    /* エラーの解放 */
     return stat;
 }