X-Git-Url: http://j8takagi.net/cgi-bin/gitweb.cgi?p=YACASL2.git;a=blobdiff_plain;f=src%2Flabel.c;h=9124b5cc70a58c276eb2da7fc710ae929e045b92;hp=a65e2939e45c484edc33aa0c75b61960d7292f99;hb=7f7bde5dd1305cfc598fe09030c69aad0dc01368;hpb=4ee27a568fb9222907a566e59aaefe248f08a8e4

diff --git a/src/label.c b/src/label.c
index a65e293..9124b5c 100644
--- a/src/label.c
+++ b/src/label.c
@@ -1,31 +1,44 @@
-#include "casl2.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <assert.h>
+
+#include "cerr.h"
+#include "cmem.h"
+#include "hash.h"
 #include "assemble.h"
 
-int labelcnt = 0;                /* ラベル数 */
-LABELTAB *labels[LABELTABSIZE];  /* ラベル表 */
+static int labelcnt = 0;                /* ラベル数 */
+static LABELTAB *labels[LABELTABSIZE];  /* ラベル表 */
 
-/* プログラム名とラベルに対応するハッシュ値を返す */
+/**
+ * プログラム名とラベルに対応するハッシュ値を返す
+ */
 unsigned labelhash(const char *prog, const char *label)
 {
     HKEY *keys[2];
     int i = 0;
+
     if(prog != NULL) {
-        keys[i] = malloc_chk(sizeof(HKEY), "labelhash.key");
+        keys[i] = malloc_chk(sizeof(HKEY), "labelhash.key[]");
         keys[i]->type = CHARS;
-        keys[i++]->val.s = strdup(prog);
+        keys[i]->val.s = strdup_chk(prog, "labelhash.key[].val");
     }
-    keys[i] = malloc_chk(sizeof(HKEY), "labelhash.key");
+    keys[i] = malloc_chk(sizeof(HKEY), "labelhash.key[]");
     keys[i]->type = CHARS;
-    keys[i]->val.s = strdup(label);
+    keys[i]->val.s = strdup_chk(label, "labelhash.key[].val");
     /* ハッシュ値を返す */
     return hash(i+1, keys, LABELTABSIZE);
 }
 
-/* ラベル表からアドレスを検索する */
+/**
+ * プログラム名とラベルに対応するアドレスをラベル表から検索する
+ */
 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 && strcmp(prog, np->prog) == 0)) &&
            strcmp(label, np->label) == 0)
@@ -36,7 +49,9 @@ WORD getlabel(const char *prog, const char *label)
     return 0xFFFF;
 }
 
-/* プログラム名、ラベル、アドレスをラベル表に追加する */
+/**
+ * プログラム名、ラベル、アドレスをラベル表に追加する
+ */
 bool addlabel(const char *prog, const char *label, WORD adr)
 {
     assert(label != NULL);
@@ -49,21 +64,15 @@ bool addlabel(const char *prog, const char *label, WORD adr)
         return false;
     }
     /* メモリを確保 */
-    if((np = malloc_chk(sizeof(LABELTAB), "addlabel.np")) == NULL) {
-        goto cerr102;
-    }
+    np = malloc_chk(sizeof(LABELTAB), "labels.next");
     /* プログラム名を設定 */
     if(prog == NULL) {
         np->prog = NULL;
     } else {
-        if((np->prog = strdup(prog)) == NULL) {
-            goto cerr102;
-        }
+        np->prog = strdup_chk(prog, "labels.prog");
     }
     /* ラベルを設定 */
-    if((np->label = strdup(label)) == NULL) {
-        goto cerr102;
-    }
+    np->label = strdup_chk(label, "labels.label");
     /* アドレスを設定 */
     np->adr = adr;
     /* ラベル数を設定 */
@@ -73,17 +82,19 @@ bool addlabel(const char *prog, const char *label, WORD adr)
     np->next = labels[hashval];
     labels[hashval] = np;
     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, asize = 0;
@@ -93,9 +104,13 @@ void printlabel()
     for(i = 0; i < LABELTABSIZE; i++) {
         for(np = labels[i]; np != NULL; np = np->next) {
             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] = malloc_chk(sizeof(LABELARRAY), "ar[]");
+            if(np->prog == NULL) {
+                ar[asize]->prog = NULL;
+            } else {
+                ar[asize]->prog = strdup_chk(np->prog, "ar[].prog");
+            }
+            ar[asize]->label = strdup_chk(np->label, "ar[].label");
             ar[asize++]->adr = np->adr;
         }
     }
@@ -108,19 +123,22 @@ void printlabel()
     }
 }
 
-/* ラベル表を解放する */
+/**
+ * ラベル表を解放する
+ */
 void freelabel()
 {
     int i;
     LABELTAB *np, *nq;
+
     for(i = 0; i < LABELTABSIZE; i++) {
         for(np = labels[i]; np != NULL; np = nq) {
             nq = np->next;
             if(np->prog != NULL) {
-                free(np->prog);
+                free_chk(np->prog, "np.prog");
             }
-            free(np->label);
-            free(np);
+            free_chk(np->label, "np.label");
+            free_chk(np, "np");
         }
     }
 }