トークンの処理を推敲

[YACASL2.git] / src / token.c

diff --git a/src/token.c b/src/token.c
index b6d971c..fa531b5 100644 (file)
--- a/src/token.c
+++ b/src/token.c
@@ -34,13 +34,12 @@ static CERR cerr_opdtok[] = {
 OPD *opdtok(const char *str)
 {
     OPD *opd = malloc_chk(sizeof(OPD), "opd");
-    char *tok, *p;
-    char sepc = ',';
+    char *tok, *p, sepc = ',';
     int i = 0, cnt_quote = 0;
     bool quoting = false;
 
     opd->opdc = 0;
-    if(str == NULL || !str[0]) {
+    if(!str || !str[0]) {
         return opd;
     }
     tok = p = strdup_chk(str, "opdtok.p");
@@ -74,17 +73,16 @@ OPD *opdtok(const char *str)
             i++;
         } else {
             i += strcspn(p + i, ", ");
-            sepc = p[i];
-            p[i] = '\0';
-            if(!p[0]) {
+            if(i == 0) {
                 setcerr(121, "");    /* cannot get operand token */
                 break;
             }
-            if(strlen(p) - cnt_quote > OPDSIZE) {
+            if(i - cnt_quote > OPDSIZE) {
                 setcerr(118, "");    /* operand length too long */
                 break;
             }
-            opd->opdv[(++opd->opdc)-1] = strdup_chk(p, "opd.opdv[]");
+            sepc = p[i];
+            opd->opdv[(opd->opdc)++] = strndup_chk(p, i, "opd->opdv[]");
             p += i + 1;
             i = cnt_quote = 0;
         }
@@ -100,11 +98,28 @@ void addcerrlist_tok()
     addcerrlist(ARRAYSIZE(cerr_opdtok), cerr_opdtok);
 }
 
+char *strip_casl2_comment(char *s)
+{
+    int i;
+    bool quoting = false;
+
+    for(i = 0; s[i]; i++) {
+        /* 「'」で囲まれた文字列の場合。「''」は無視 */
+        if(s[i] == '\'' && s[i+1] != '\'' && (i == 0 || s[i-1] != '\'')) {
+            quoting = !quoting;
+        /* 「'」で囲まれた文字列でない場合、文字列末尾の「;」以降を削除 */
+        } else if(quoting == false && s[i] == ';') {
+            s[i] = '\0';
+            break;
+        }
+    }
+    return s;
+}
+
 CMDLINE *linetok(const char *line)
 {
     char *tok = NULL, *p = NULL, *lbl = NULL;
     int i;
-    bool quoting = false;
     CMDLINE *cmdl = NULL;
 
     if(!line[0] || line[0] == '\n') {
@@ -112,34 +127,22 @@ CMDLINE *linetok(const char *line)
     }
     tok = p = strdup_chk(line, "tok");
     /* コメントを削除 */
-    for(i = 0; p[i]; i++) {
-        /* 「'」で囲まれた文字列の場合。「''」は無視 */
-        if(p[i] == '\'' && p[i+1] != '\'' && (i == 0 || p[i-1] != '\'')) {
-            quoting = !quoting;
-        /* 「'」で囲まれた文字列でない場合、文字列末尾の「;」以降を削除 */
-        } else if(quoting == false && p[i] == ';') {
-            p[i] = '\0';
-            break;
-        }
-    }
+    strip_casl2_comment(p);
     /* 文字列末尾の改行と空白を削除 */
-    i = strlen(p) - 1;
-    while(i > 0 && (p[i] == '\n' || p[i] == ' ' || p[i] == '\t')) {
-        p[i--] = '\0';
-    }
+    strip_end(p);
     /* 空行の場合、終了 */
-    if(!p[0] || p[0] == '\n') {
+    if(!p[0]) {
         goto linetokfin;
     }
     cmdl = malloc_chk(sizeof(CMDLINE), "cmdl");
 
     /* ラベルの取得 */
     /* 行の先頭が空白またはタブの場合、ラベルは空 */
-    if((i = strcspn(p, " \t\n")) == 0) {
+    if((i = strcspn(p, " \t")) == 0) {
         lbl = strdup_chk("", "linetok.lbl");
     } else {
         lbl = strndup_chk(p, i, "linetok.lbl");
-        /* 文字列が長すぎる場合はエラー */
+        /* ラベルの文字列が長すぎる場合はエラー */
         if(i > LABELSIZE) {
             setcerr(104, lbl);    /* label length is too long */
             FREE(lbl);
@@ -148,14 +151,12 @@ CMDLINE *linetok(const char *line)
         /* 文字列先頭をラベルの次の文字に移動 */
         p += i;
     }
+    /* ラベル取得の実行 */
     cmdl->label = lbl;
 
     /* 命令の取得 */
-    /* 文字列先頭の、ラベルと命令の間の空白を削除 */
-    for(i = 0; p[i] == ' ' || p[i] == '\t'; i++) {
-        ;
-    }
-    p += i;
+    /* 文字列先頭を、ラベルと命令の間の空白の後ろに移動 */
+    p += strspn(p, " \t");
     /* 命令がない場合は、終了 */
     if(!p[0]) {
         if(cmdl->label) {      /* ラベルが定義されていて命令がない場合はエラー */
@@ -170,13 +171,10 @@ CMDLINE *linetok(const char *line)
     cmdl->cmd = strndup_chk(p, i, "cmdl.cmd");
 
     /* オペランドの取得 */
-    /* 文字列の先頭を命令の次の文字に移動 */
-    p += i + 1;
-    /* 文字列先頭の、命令とオペランドの間の空白を削除 */
-    for(i = 0; p[i] == ' ' || p[i] == '\t'; i++) {
-        ;
-    }
+    /* 文字列先頭を、命令の次の文字に移動 */
     p += i;
+    /* 文字列先頭を、命令とオペランドの間の空白の後ろに移動 */
+    p += strspn(p, " \t");
     /* オペランド取得の実行 */
     cmdl->opd = opdtok(p);
 linetokfin: