メモリリークの修正
[YACASL2.git] / src / casl2.c
1 #include "package.h"
2 #include "assemble.h"
3
4 /**
5  * @brief CASL IIのエラーをエラーリストに追加
6  *
7  * @return なし
8  */
9 void addcerrlist_casl2();
10
11 /**
12  * @brief アセンブル結果を書き込むファイルの名前
13  *
14  * @return ファイル名
15  *
16  * @param *str ファイル名
17  */
18 const char *objfile_name(const char *str);
19
20 /**
21  * @brief casl2コマンドのオプション
22  */
23 static struct option longopts[] = {
24     { "source", no_argument, NULL, 's' },
25     { "label", no_argument, NULL, 'l' },
26     { "labelonly", no_argument, NULL, 'L' },
27     { "assembledetail", no_argument, NULL, 'a' },
28     { "assembledetailonly", no_argument, NULL, 'A' },
29     { "assembleout", optional_argument, NULL, 'o' },
30     { "assembleoutonly", optional_argument, NULL, 'O' },
31     { "trace", no_argument, NULL, 't' },
32     { "tracearithmetic", no_argument, NULL, 't' },
33     { "tracelogical", no_argument, NULL, 'T' },
34     { "dump", no_argument, NULL, 'd' },
35     { "monitor", no_argument, NULL, 'm' },
36     { "memorysize", required_argument, NULL, 'M' },
37     { "clocks", required_argument, NULL, 'C' },
38     { "version", no_argument, NULL, 'v' },
39     { "help", no_argument, NULL, 'h' },
40     { 0, 0, 0, 0 },
41 };
42
43 /**
44  * @brief casl2のエラー定義
45  */
46 CERR cerr_casl2[] = {
47     { 126, "no source file" },
48     { 127, "invalid option" },
49 };
50
51 void addcerrlist_casl2()
52 {
53     addcerrlist(ARRAYSIZE(cerr_casl2), cerr_casl2);
54 }
55
56 const char *objfile_name(const char *str)
57 {
58     const char *default_name = "a.o";
59     return (str == NULL) ? default_name : str;
60 }
61
62 /**
63  * @brief casl2コマンドのメイン
64  *
65  * @return 正常終了時は0、エラー発生時は1
66  *
67  * @param argc コマンドライン引数の数
68  * @param *argv[] コマンドライン引数の配列
69  */
70 int main(int argc, char *argv[])
71 {
72     int memsize = DEFAULT_MEMSIZE, clocks = DEFAULT_CLOCKS, opt, i, stat = 0;
73     char *af[argc], *objfile = NULL;
74     const char *version = PACKAGE_VERSION,  *cmdversion = "casl2 of YACASL2 version %s\n";
75     const char *usage =
76         "Usage: %s [-slLaAtTdmvh] [-oO[<OBJECTFILE>]] [-M <MEMORYSIZE>] [-C <CLOCKS>] FILE1[ FILE2  ...]\n";
77
78     /* エラーの定義 */
79     cerr_init();
80     addcerrlist_casl2();
81     addcerrlist_assemble();
82     addcerrlist_exec();
83
84     /* オプションの処理 */
85     while((opt = getopt_long(argc, argv, "tTdslLmao::O::AM:C:vh", longopts, NULL)) != -1) {
86         switch(opt) {
87         case 's':
88             asmode.src = true;
89             break;
90         case 'l':
91             asmode.label = true;
92             break;
93         case 'L':
94             asmode.label = true;
95             asmode.onlylabel = true;
96             break;
97         case 'a':
98             asmode.asdetail = true;
99             break;
100         case 'A':
101             asmode.asdetail = true;
102             asmode.onlyassemble = true;
103             break;
104         case 'o':
105             objfile = strdup_chk(objfile_name(optarg), "objfile");
106             break;
107         case 'O':
108             asmode.onlyassemble = true;
109             objfile = strdup_chk(objfile_name(optarg), "objfile");
110             break;
111         case 't':
112             execmode.trace = true;
113             break;
114         case 'T':
115             execmode.trace = true;
116             execmode.logical = true;
117             break;
118         case 'd':
119             execmode.dump = true;
120             break;
121         case 'm':
122             execmode.step = true;
123             break;
124         case 'M':
125             memsize = atoi(optarg);
126             break;
127         case 'C':
128             clocks = atoi(optarg);
129             break;
130         case 'v':
131             fprintf(stdout, cmdversion, version);
132             goto casl2fin;
133         case 'h':
134             fprintf(stdout, usage, argv[0]);
135             goto casl2fin;
136         case '?':
137             fprintf(stderr, usage, argv[0]);
138             setcerr(127, "");    /* invalid option */
139             goto casl2fin;
140         }
141     }
142
143     /* ソースファイルが指定されていない場合は終了 */
144     if(argv[optind] == NULL) {
145         setcerr(126, "");    /* no source file */
146         fprintf(stderr, "CASL2 error - %d: %s\n", cerr->num, cerr->msg);
147         goto casl2fin;
148     }
149     create_cmdtable(HASH_CMDTYPE);                 /* 命令の名前とタイプがキーのハッシュ表を作成 */
150     reset(memsize, clocks);                        /* 仮想マシンCOMET IIのリセット */
151     for(i = 0; i < argc - optind; i++) {           /* 引数からファイル名配列を取得 */
152         af[i] = argv[optind + i];
153     }
154     /* アセンブル */
155     if(assemble(i, af, 0) == false || asmode.onlylabel == true) {
156         goto freecmdtable;
157     }
158     /* オブジェクトファイル名が指定されている場合は、アセンブル結果をオブジェクトファイルに出力 */
159     if(objfile != NULL) {
160         outassemble(objfile);
161     }
162     /* onlyassembleモード以外の場合、仮想マシンCOMET IIを実行 */
163     if(asmode.onlyassemble == false) {
164         exec();                                    /* 仮想マシンCOMET IIの実行 */
165     }
166 freecmdtable:
167     shutdown();                                    /* 仮想マシンCOMET IIのシャットダウン */
168     free_cmdtable(HASH_CMDTYPE);
169 casl2fin:
170     FREE(objfile);
171     if(cerr->num > 0) {
172         stat = 1;
173     }
174     freecerr();                                    /* エラーの解放 */
175     return stat;
176 }