X-Git-Url: http://j8takagi.net/cgi-bin/gitweb.cgi?p=YACASL2.git;a=blobdiff_plain;f=src%2Ftoken.c;h=e8b630b0753ca6ac0542017a9568bdb0784ab6ae;hp=b846c69b265477cb93312cf841d4624e76ca77f3;hb=4ee27a568fb9222907a566e59aaefe248f08a8e4;hpb=f1803bd560071fb724b2c7ff2f5f35fa5086d10a

diff --git a/src/token.c b/src/token.c
index b846c69..e8b630b 100644
--- a/src/token.c
+++ b/src/token.c
@@ -4,7 +4,7 @@
 /* 「,」区切りの文字列から、オペランドのトークンを取得 */
 OPD *opdtok(const char *str)
 {
-    OPD *opd = malloc(sizeof(OPD));
+    OPD *opd = malloc_chk(sizeof(OPD), "opd");
     char *p, *q, *sepp;
     int sepc = ',', qcnt = 0;
     bool quoting = false;
@@ -17,7 +17,7 @@ OPD *opdtok(const char *str)
     do {
         /* オペランド数が多すぎる場合はエラー */
         if(opd->opdc >= OPDSIZE) {
-            setcerr(117, str);    /* operand is too many */
+            setcerr(117, NULL);    /* operand is too many */
             break;
         }
         /* 先頭が「=」の場合の処理 */
@@ -65,28 +65,27 @@ OPD *opdtok(const char *str)
 /* 空白またはタブで区切られた1行から、トークンを取得 */
 CMDLINE *linetok(const char *line)
 {
-    char *tokens, *p, *q, *sepp;
-    bool quote = false;
-    CMDLINE *cmdl = malloc(sizeof(CMDLINE));
+    char *tokens, *p, *sepp;
+    bool quoting = false;
+    CMDLINE *cmdl = malloc_chk(sizeof(CMDLINE), "cmdl");
 
     if(line == NULL || strlen(line) == 0) {
         return NULL;
     }
-    tokens = p = strdup(line);
-    /* 空行の場合、NULLを返す */
-    if(*p == '\0') {
-        return NULL;
-    }
-    /* ' 'で囲まれていない;以降の文字列は、コメントとして削除 */
-    while(p != NULL) {
-        if((q = strchr(tokens, '\'')) != NULL) {
-            quote = !quote;
-            p = q;
-        }
-        if(quote == false && (p = strchr(p, ';')) != NULL) {
+    tokens = strdup(line);
+    /* コメントを削除 */
+    for(p = tokens; *p != '\0'; p++) {
+        /* 「'」で囲まれた文字列の処理。「''」は無視 */
+        if(*p == '\'' && *(p+1) != '\'' && !(p > tokens && *(p-1) == '\'')) {
+            quoting = !quoting;
+        } else if(quoting == false && *p == ';') {
             *p = '\0';
+            break;
         }
     }
+    if(*tokens == '\0') {
+        return NULL;
+    }
     p = tokens;
     /* 行の先頭が空白またはタブの場合、ラベルは空 */
     if((sepp = p + strcspn(p, " \t\n")) == p){
@@ -120,7 +119,7 @@ CMDLINE *linetok(const char *line)
         p++;
     }
     /* オペランドを取得 */
-    cmdl->opd = malloc(sizeof(OPD));
+    cmdl->opd = malloc_chk(sizeof(OPD), "cmdl->opd");
     /* 改行かタブまでの文字列を取得。
        「'」で囲まれた文字列に含まれる場合があるため、空白は無視 */
     if((sepp = p + strcspn(p, "\t\n")) > p) {