X-Git-Url: http://j8takagi.net/cgi-bin/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=src%2Ftoken.c;h=b9ff3527edd95bdccb1ace4f16a4165c003bffb0;hb=f9e27534c723ae199ed32404047b9a2c6519a8a9;hp=68ed3a7c5dc8a53dfa3489fb0fe6a82f2f133bef;hpb=bb9d41081432d335dcb0b3ad8c9d7dd9c6790b6f;p=YACASL2.git

diff --git a/src/token.c b/src/token.c
index 68ed3a7..b9ff352 100644
--- a/src/token.c
+++ b/src/token.c
@@ -4,29 +4,36 @@
 /* 「,」区切りの文字列から、オペランドのトークンを取得 */
 OPD *opdtok(const char *str)
 {
-    OPD *opd = malloc(sizeof(OPD));
-    char *p, *q, *r, *sepp;
-    int sepc = ',', len;
+    OPD *opd = malloc_chk(sizeof(OPD), "opd");
+    char *p, *q, *sepp;
+    int sepc = ',', qcnt = 0;
     bool quoting = false;
 
     opd->opdc = 0;
     if(str == NULL) {
         return opd;
     }
-    p = q = strdup(str);
+    p = q = strdup_chk(str, "opdtopk.p");
     do {
         /* オペランド数が多すぎる場合はエラー */
         if(opd->opdc >= OPDSIZE) {
-            setcerr(117, str);    /* operand is too many */
+            setcerr(117, NULL);    /* operand is too many */
             break;
         }
         /* 先頭が「=」の場合の処理 */
         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) {
             /* 閉じ「'」がないまま文字列が終了した場合 */
@@ -43,17 +50,13 @@ OPD *opdtok(const char *str)
                 setcerr(121, NULL);    /* cannot get operand token */
                 break;
             }
-            len = strlen(r = p);
-            while(*r != '\0' && (r = strstr(r, "''")) != NULL) {
-                len--;
-                r += 2;
-            };
-            if(len > OPDSIZE + 2) {    /* OPDSIZE + 「'」2文字分 */
+            if(strlen(p) - qcnt > OPDSIZE) {
                 setcerr(118, NULL);    /* operand length is too long */
                 break;
             }
-            opd->opdv[(++opd->opdc)-1] = strdup(p);
+            opd->opdv[(++opd->opdc)-1] = strdup_chk(p, "opd.opdv[]");
             p = q = sepp + 1;
+            qcnt = 0;
         }
     } while(sepc == ',');
     return opd;
@@ -64,12 +67,12 @@ CMDLINE *linetok(const char *line)
 {
     char *tokens, *p, *sepp;
     bool quoting = false;
-    CMDLINE *cmdl = malloc(sizeof(CMDLINE));
+    CMDLINE *cmdl = malloc_chk(sizeof(CMDLINE), "cmdl");
 
     if(line == NULL || strlen(line) == 0) {
         return NULL;
     }
-    tokens = strdup(line);
+    tokens = strdup_chk(line, "tokens");
     /* コメントを削除 */
     for(p = tokens; *p != '\0'; p++) {
         /* 「'」で囲まれた文字列の処理。「''」は無視 */
@@ -93,7 +96,7 @@ CMDLINE *linetok(const char *line)
         if(strlen(p) > LABELSIZE) {
             setcerr(104, p);    /* label length is too long */
         }
-        cmdl->label = strdup(p);
+        cmdl->label = strdup_chk(p, "cmdl.label");
         p = sepp + 1;
     }
     while(*p == ' ' || *p == '\t') {
@@ -110,13 +113,13 @@ CMDLINE *linetok(const char *line)
     /* 命令を取得 */
     sepp = p + strcspn(p, " \t\n");
     *sepp = '\0';
-    cmdl->cmd = strdup(p);
+    cmdl->cmd = strdup_chk(p, "cmdl.cmd");
     p = sepp + 1;
     while(*p == ' ' || *p == '\t') {
         p++;
     }
     /* オペランドを取得 */
-    cmdl->opd = malloc(sizeof(OPD));
+    cmdl->opd = malloc_chk(sizeof(OPD), "cmdl.opd");
     /* 改行かタブまでの文字列を取得。
        「'」で囲まれた文字列に含まれる場合があるため、空白は無視 */
     if((sepp = p + strcspn(p, "\t\n")) > p) {