X-Git-Url: http://j8takagi.net/cgi-bin/gitweb.cgi?p=YACASL2.git;a=blobdiff_plain;f=src%2Flabel.c;h=c76c3a7590ee6a5651c186d0573c875a26705d55;hp=a9e4bc3b95c332aa18ea60108c8b9f5416ac3201;hb=650f92bf8dfdd0095db993f71f9e3867e7119acc;hpb=40008ccbbc76e5ef0e2f595d7aa2c400d543834c

diff --git a/src/label.c b/src/label.c
index a9e4bc3..c76c3a7 100644
--- a/src/label.c
+++ b/src/label.c
@@ -9,14 +9,15 @@ 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]->val.s = strdup_chk(prog, "labelhash.key[].val");
     }
-    keys[i] = malloc(sizeof(HKEY));
+    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);
 }
@@ -26,6 +27,7 @@ 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)
@@ -48,17 +50,21 @@ bool addlabel(const char *prog, const char *label, WORD adr)
         setcerr(101, label);    /* label already defined */
         return false;
     }
-    /* プログラム名、ラベル、アドレスを設定。メモリを確保できない場合はエラー発生 */
-    if((np = malloc(sizeof(LABELTAB))) == NULL || (np->label = strdup(label)) == NULL ||
-       (prog != NULL && (np->prog = strdup(prog)) == NULL))
-    {
-        setcerr(102, NULL);    /* label table is full */
-        return false;
+    /* メモリを確保 */
+    np = malloc_chk(sizeof(LABELTAB), "labels.next");
+    /* プログラム名を設定 */
+    if(prog == NULL) {
+        np->prog = NULL;
+    } else {
+        np->prog = strdup_chk(prog, "labels.prog");
     }
+    /* ラベルを設定 */
+    np->label = strdup_chk(label, "labels.label");
+    /* アドレスを設定 */
     np->adr = adr;
-    /* ラベル数の設定 */
+    /* ラベル数を設定 */
     labelcnt++;
-    /* ハッシュ表の設定 */
+    /* ハッシュ表へ追加 */
     hashval = labelhash(prog, label);
     np->next = labels[hashval];
     labels[hashval] = np;
@@ -80,9 +86,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(sizeof(LABELARRAY));
-            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;
         }
     }
@@ -100,6 +110,7 @@ void freelabel()
 {
     int i;
     LABELTAB *np, *nq;
+
     for(i = 0; i < LABELTABSIZE; i++) {
         for(np = labels[i]; np != NULL; np = nq) {
             nq = np->next;