変数名の整理

[YACASL2.git] / src / label.c

diff --git a/src/label.c b/src/label.c
index b58def5..a65e293 100644 (file)
--- a/src/label.c
+++ b/src/label.c
@@ -1,7 +1,8 @@
 #include "casl2.h"
 #include "assemble.h"
 
-LABELTAB *labels[LABELTABSIZE];
+int labelcnt = 0;                /* ラベル数 */
+LABELTAB *labels[LABELTABSIZE];  /* ラベル表 */
 
 /* プログラム名とラベルに対応するハッシュ値を返す */
 unsigned labelhash(const char *prog, const char *label)
@@ -9,11 +10,11 @@ unsigned labelhash(const char *prog, const char *label)
     HKEY *keys[2];
     int i = 0;
     if(prog != NULL) {
-        keys[i] = malloc(sizeof(HKEY));
+        keys[i] = malloc_chk(sizeof(HKEY), "labelhash.key");
         keys[i]->type = CHARS;
         keys[i++]->val.s = strdup(prog);
     }
-    keys[i] = malloc(sizeof(HKEY));
+    keys[i] = malloc_chk(sizeof(HKEY), "labelhash.key");
     keys[i]->type = CHARS;
     keys[i]->val.s = strdup(label);
     /* ハッシュ値を返す */
@@ -23,10 +24,10 @@ unsigned labelhash(const char *prog, const char *label)
 /* ラベル表からアドレスを検索する */
 WORD getlabel(const char *prog, const char *label)
 {
+    assert(label != NULL);
     LABELTAB *np;
     for(np = labels[labelhash(prog, label)]; np != NULL; np = np->next) {
-        if(((prog == NULL && np->prog == NULL) ||
-            (prog != NULL && np->prog != NULL && strcmp(prog, np->prog) == 0)) &&
+        if((prog == NULL || (np->prog != NULL && strcmp(prog, np->prog) == 0)) &&
            strcmp(label, np->label) == 0)
         {
             return np->adr;
@@ -35,49 +36,75 @@ WORD getlabel(const char *prog, const char *label)
     return 0xFFFF;
 }
 
-/* ã\83©ã\83\99ã\83«ã\82\92表に追加する */
+/* ã\83\97ã\83­ã\82°ã\83©ã\83 å\90\8dã\80\81ã\83©ã\83\99ã\83«ã\80\81ã\82¢ã\83\89ã\83¬ã\82¹ã\82\92ã\83©ã\83\99ã\83«表に追加する */
 bool addlabel(const char *prog, const char *label, WORD adr)
 {
+    assert(label != NULL);
     LABELTAB *np;
     unsigned hashval;
-    char *keys[2];
-    int i = 0;
 
+    /* 登録されたラベルを検索。すでに登録されている場合はエラー発生 */
     if(getlabel(prog, label) != 0xFFFF) {
         setcerr(101, label);    /* label already defined */
         return false;
     }
-    np = (LABELTAB *) malloc(sizeof(*np));
-    if(np == NULL || (np->label = strdup(label)) == NULL ||
-       (prog != NULL && (np->prog = strdup(prog)) == NULL))
-    {
-        setcerr(102, NULL);    /* label table is full */
-        return false;
+    /* メモリを確保 */
+    if((np = malloc_chk(sizeof(LABELTAB), "addlabel.np")) == NULL) {
+        goto cerr102;
     }
-    if(prog != NULL) {
-        keys[i++] = strdup(prog);
+    /* プログラム名を設定 */
+    if(prog == NULL) {
+        np->prog = NULL;
+    } else {
+        if((np->prog = strdup(prog)) == NULL) {
+            goto cerr102;
+        }
     }
-    keys[i] = strdup(label);;
+    /* ラベルを設定 */
+    if((np->label = strdup(label)) == NULL) {
+        goto cerr102;
+    }
+    /* アドレスを設定 */
+    np->adr = adr;
+    /* ラベル数を設定 */
+    labelcnt++;
+    /* ハッシュ表へ追加 */
     hashval = labelhash(prog, label);
     np->next = labels[hashval];
     labels[hashval] = np;
-    np->adr = adr;
     return true;
+cerr102:
+    setcerr(102, NULL);    /* label table is full */
+    return false;
+}
+
+int compare_adr(const void *a, const void *b)
+{
+    return (**(LABELARRAY **)a).adr - (**(LABELARRAY **)b).adr;
 }
 
 /* ラベル表を表示する */
 void printlabel()
 {
-    int i;
+    int i, asize = 0;
     LABELTAB *np;
+    LABELARRAY *ar[labelcnt];
+
     for(i = 0; i < LABELTABSIZE; i++) {
         for(np = labels[i]; np != NULL; np = np->next) {
-            if(np->prog == NULL) {
-                fprintf(stdout, "%s ---> #%04X\n", np->label, np->adr);
-            } else {
-                fprintf(stdout, "%s.%s ---> #%04X\n", np->prog, np->label, np->adr);
-            }
+            assert(np->label != NULL);
+            ar[asize] = malloc_chk(sizeof(LABELARRAY), "ar[asize]");
+            ar[asize]->prog = (np->prog == NULL ? NULL : strdup(np->prog));
+            ar[asize]->label = strdup(np->label);
+            ar[asize++]->adr = np->adr;
+        }
+    }
+    qsort(ar, asize, sizeof(*ar), compare_adr);
+    for(i = 0; i < asize; i++) {
+        if(ar[i]->prog != NULL) {
+            fprintf(stdout, "%s.", ar[i]->prog);
         }
+        fprintf(stdout, "%s ---> #%04X\n", ar[i]->label, ar[i]->adr);
     }
 }
 
@@ -87,9 +114,11 @@ void freelabel()
     int i;
     LABELTAB *np, *nq;
     for(i = 0; i < LABELTABSIZE; i++) {
-        for(np = labels[i]; np != NULL; np = nq){
+        for(np = labels[i]; np != NULL; np = nq) {
             nq = np->next;
-            free(np->prog);
+            if(np->prog != NULL) {
+                free(np->prog);
+            }
             free(np->label);
             free(np);
         }