Valgrindで見つかったメモリー周辺の問題点を修正

[YACASL2.git] / src / token.c

diff --git a/src/token.c b/src/token.c
index c185d8d..c7bfade 100644 (file)
--- a/src/token.c
+++ b/src/token.c
@@ -46,7 +46,7 @@ OPD *opdtok(const char *str)
     if(str == NULL) {
         return opd;
     }
     if(str == NULL) {
         return opd;
     }

-    p = q = r = strdup_chk(str, "opdtopk.p");
+    p = q = r = strdup_chk(str, "opdtok.p");

     do {
         /* オペランド数が多すぎる場合はエラー */
         if(opd->opdc >= OPDSIZE) {
     do {
         /* オペランド数が多すぎる場合はエラー */
         if(opd->opdc >= OPDSIZE) {


@@ -92,7 +92,7 @@ OPD *opdtok(const char *str)
             rcnt = 0;
         }
     } while(sepc == ',');
             rcnt = 0;
         }
     } while(sepc == ',');

-    free_chk(p, "opdtok.p");
+    FREE(p);

     return opd;
 }
 
     return opd;
 }
 


@@ -103,7 +103,7 @@ CMDLINE *linetok(const char *line)
 {
     char *tokens, *p, *sepp;
     bool quoting = false;
 {
     char *tokens, *p, *sepp;
     bool quoting = false;

-    CMDLINE *cmdl = malloc_chk(sizeof(CMDLINE), "cmdl");
+    CMDLINE *cmdl = NULL;

 
     if(line == NULL || strlen(line) == 0) {
         return NULL;
 
     if(line == NULL || strlen(line) == 0) {
         return NULL;


@@ -119,50 +119,51 @@ CMDLINE *linetok(const char *line)
             break;
         }
     }
             break;
         }
     }

-    if(*tokens == '\0') {
-        return NULL;
-    }
-    p = tokens;
-    /* 行の先頭が空白またはタブの場合、ラベルは空 */
-    if((sepp = p + strcspn(p, " \t\n")) == p){
-        cmdl->label = NULL;
-    } else {        /* ラベルを取得 */
-        *sepp = '\0';
-        /* 文字列が長すぎる場合はエラー */
-        if(strlen(p) > LABELSIZE) {
-            setcerr(104, p);    /* label length is too long */
+    if(*tokens != '\0') {
+        p = tokens;
+        cmdl = malloc_chk(sizeof(CMDLINE), "cmdl");
+        /* ラベルの取得。行の先頭が空白またはタブの場合、ラベルは空 */
+        if((sepp = p + strcspn(p, " \t\n")) == p){
+            cmdl->label = NULL;
+        } else {        /* ラベルを取得 */
+            *sepp = '\0';
+            /* 文字列が長すぎる場合はエラー */
+            if(strlen(p) > LABELSIZE) {
+                setcerr(104, p);    /* label length is too long */
+            }
+            cmdl->label = strdup_chk(p, "cmdl.label");
+            p = sepp + 1;

         }
         }

-        cmdl->label = strdup_chk(p, "cmdl.label");
-        p = sepp + 1;
-    }
-    while(*p == ' ' || *p == '\t') {
-        p++;
-    }
-    /* 命令がない場合、終了 */
-    if(*p == '\n' || *p == '\0') {
-        /* ラベルが定義されていて命令がない場合はエラー */
-        if(cmdl->label != NULL) {
-            setcerr(105, NULL);    /* no command in the line */
+        /* ラベルと命令の間の空白をスキップ */
+        while(*p == ' ' || *p == '\t') {
+            p++;
+        }
+        /* 命令とオペランドの取得 */
+        if(*p == '\n' || *p == '\0') {        /* 命令がない場合は、終了 */
+            if(cmdl->label != NULL) {         /* ラベルが定義されていて命令がない場合はエラー */
+                setcerr(105, NULL);    /* no command in the line */
+            }
+        } else {
+            /* 命令の取得 */
+            sepp = p + strcspn(p, " \t\n");
+            *sepp = '\0';
+            cmdl->cmd = strdup_chk(p, "cmdl.cmd");
+            p = sepp + 1;
+            /* 命令とオペランドの間の空白をスキップ */
+            while(*p == ' ' || *p == '\t') {
+                p++;
+            }
+            /* 改行かタブまでの文字列を取得。
+               「'」で囲まれた文字列に含まれる場合があるため、空白は無視 */
+            if((sepp = p + strcspn(p, "\t\n")) > p) {
+                *sepp = '\0';
+                cmdl->opd = opdtok(p);
+            } else {
+                cmdl->opd = malloc_chk(sizeof(OPD *), "cmdl.opd");
+                cmdl->opd->opdc = 0;
+            }

         }
         }

-        return NULL;
-    }
-    /* 命令を取得 */
-    sepp = p + strcspn(p, " \t\n");
-    *sepp = '\0';
-    cmdl->cmd = strdup_chk(p, "cmdl.cmd");
-    p = sepp + 1;
-    while(*p == ' ' || *p == '\t') {
-        p++;
-    }
-    /* オペランドを取得 */
-    cmdl->opd = malloc_chk(sizeof(OPD), "cmdl.opd");
-    /* 改行かタブまでの文字列を取得。
-       「'」で囲まれた文字列に含まれる場合があるため、空白は無視 */
-    if((sepp = p + strcspn(p, "\t\n")) > p) {
-        *sepp = '\0';
-        cmdl->opd = opdtok(p);
-    } else {
-        cmdl->opd->opdc = 0;

     }
     }

+    FREE(tokens);

     return cmdl;
 }
     return cmdl;
 }