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

diff --git a/src/token.c b/src/token.c
index 35ddbf7..e8b630b 100644
--- a/src/token.c
+++ b/src/token.c
@@ -4,13 +4,13 @@
 /* 「,」区切りの文字列から、オペランドのトークンを取得 */
 OPD *opdtok(const char *str)
 {
-    OPD *opd = malloc(sizeof(OPD));
+    OPD *opd = malloc_chk(sizeof(OPD), "opd");
     char *p, *q, *sepp;
-    int sepc = ',';
+    int sepc = ',', qcnt = 0;
     bool quoting = false;
 
     opd->opdc = 0;
-    if(str == NULL || strlen(str) == 0) {
+    if(str == NULL) {
         return opd;
     }
     p = q = strdup(str);
@@ -24,26 +24,39 @@ OPD *opdtok(const char *str)
         if(*q == '=') {
             q++;
         }
-        /* 「'」で囲まれた文字列の処理。「''」は無視 */
-        if(*q == '\'' && *(q+1) != '\'' && !(p < q && *(q-1) == '\'')) {
-            quoting = !quoting;
+        /* 「'」の場合 */
+        if(*q == '\'') {
+            /* 「''」以外の場合はquote値を反転 */
+            if(*(q+1) != '\'' && !(p < q && *(q-1) == '\'')) {
+                quoting = !quoting;
+            }
+            /* 「'」の分、文字列の長さを小さくする */
+            if(*(q+1) != '\'') {
+                qcnt++;
+            }
         }
         if(quoting == true) {
+            /* 閉じ「'」がないまま文字列が終了した場合 */
+            if(*q == '\0') {
+                setcerr(123, str);    /* unclosed quote */
+                break;
+            }
             q++;
         } else {
             sepp = q + strcspn(q, ", ");
             sepc = *sepp;
             *sepp = '\0';
-            if(strlen(p) == 0) {
+            if(*p == '\0') {
                 setcerr(121, NULL);    /* cannot get operand token */
                 break;
             }
-            if(strlen(p) > OPDSIZE + 2) {    /* OPDSIZE + 「'」2文字分 */
-                setcerr(118, p);    /* operand length is too long */
+            if(strlen(p) - qcnt > OPDSIZE) {
+                setcerr(118, NULL);    /* operand length is too long */
                 break;
             }
             opd->opdv[(++opd->opdc)-1] = strdup(p);
             p = q = sepp + 1;
+            qcnt = 0;
         }
     } while(sepc == ',');
     return opd;
@@ -52,19 +65,28 @@ OPD *opdtok(const char *str)
 /* 空白またはタブで区切られた1行から、トークンを取得 */
 CMDLINE *linetok(const char *line)
 {
-    char *tokens, *p, *q, *sepp;
-    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);
+    tokens = strdup(line);
     /* コメントを削除 */
-    if((q = strchr(tokens, ';')) != NULL) {
-        *q = '\0';
+    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(*p == '\0') {
+    if(*tokens == '\0') {
         return NULL;
     }
+    p = tokens;
     /* 行の先頭が空白またはタブの場合、ラベルは空 */
     if((sepp = p + strcspn(p, " \t\n")) == p){
         cmdl->label = NULL;
@@ -97,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) {