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

diff --git a/src/token.c b/src/token.c
index 68ed3a7..fb35c8b 100644
--- a/src/token.c
+++ b/src/token.c
@@ -1,75 +1,114 @@
-#include "casl2.h"
+#include <stdio.h>
+#include <string.h>
+#include "cerr.h"
+#include "cmem.h"
 #include "assemble.h"
 
-/* 「,」区切りの文字列から、オペランドのトークンを取得 */
+
+/**
+ * 行トークン取得のエラー定義
+ */
+CERR cerr_linetok[] = {
+    { 104, "label length is too long" },
+    { 105, "no command in the line" },
+};
+
+/**
+ * オペランドトークン取得のエラー定義
+ */
+static CERR cerr_opdtok[] = {
+    { 117, "operand too many in DC" },
+    { 118, "operand length too long" },
+    { 121, "cannot get operand token" },
+    { 123, "unclosed quote" },
+};
+
+/**
+ * オペランドトークン取得のエラーを追加
+ */
+void addcerrlist_tok()
+{
+    addcerrlist(ARRAYSIZE(cerr_linetok), cerr_linetok);
+    addcerrlist(ARRAYSIZE(cerr_opdtok), cerr_opdtok);
+}
+
+/**
+ * 「,」区切りの文字列から、オペランドのトークンを取得
+ */
 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, *r, *sepp;     /* pは文字列全体の先頭位置、qはトークンの先頭位置、rは文字の位置 */
+    int sepc = ',', rcnt = 0;
     bool quoting = false;
 
     opd->opdc = 0;
     if(str == NULL) {
         return opd;
     }
-    p = q = strdup(str);
+    p = q = r = strdup_chk(str, "opdtok.p");
     do {
         /* オペランド数が多すぎる場合はエラー */
         if(opd->opdc >= OPDSIZE) {
-            setcerr(117, str);    /* operand is too many */
+            setcerr(117, "");    /* operand is too many */
             break;
         }
-        /* 先頭が「=」の場合の処理 */
-        if(*q == '=') {
-            q++;
+        /* 先頭が等号（=）の場合 */
+        if(*r == '=') {
+            r++;
         }
-        /* 「'」で囲まれた文字列の処理。「''」は無視 */
-        if(*q == '\'' && *(q+1) != '\'' && !(p < q && *(q-1) == '\'')) {
-            quoting = !quoting;
+        /* 「'」の場合 */
+        if(*r == '\'') {
+            /* 「''」以外の場合はquote値を反転 */
+            if(*(r+1) != '\'' && !(q < r && *(r-1) == '\'')) {
+                quoting = !quoting;
+            }
+            /* 文字列の長さを数える。「'」の場合は数えない */
+            if(*(r+1) != '\'') {
+                rcnt++;
+            }
         }
         if(quoting == true) {
             /* 閉じ「'」がないまま文字列が終了した場合 */
-            if(*q == '\0') {
+            if(*r == '\0') {
                 setcerr(123, str);    /* unclosed quote */
                 break;
             }
-            q++;
+            r++;
         } else {
-            sepp = q + strcspn(q, ", ");
+            sepp = r + strcspn(r, ", ");
             sepc = *sepp;
             *sepp = '\0';
-            if(*p == '\0') {
-                setcerr(121, NULL);    /* cannot get operand token */
+            if(*q == '\0') {
+                setcerr(121, "");    /* 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文字分 */
-                setcerr(118, NULL);    /* operand length is too long */
+            if(strlen(q) - rcnt > OPDSIZE) {
+                setcerr(118, "");    /* operand length is too long */
                 break;
             }
-            opd->opdv[(++opd->opdc)-1] = strdup(p);
-            p = q = sepp + 1;
+            opd->opdv[(++opd->opdc)-1] = strdup_chk(q, "opd.opdv[]");
+            q = r = sepp + 1;
+            rcnt = 0;
         }
     } while(sepc == ',');
+    FREE(p);
     return opd;
 }
 
-/* 空白またはタブで区切られた1行から、トークンを取得 */
+/**
+ * 空白またはタブで区切られた1行から、トークンを取得する
+ */
 CMDLINE *linetok(const char *line)
 {
     char *tokens, *p, *sepp;
     bool quoting = false;
-    CMDLINE *cmdl = malloc(sizeof(CMDLINE));
+    CMDLINE *cmdl = NULL;
 
-    if(line == NULL || strlen(line) == 0) {
+    if(*line == '\0') {
         return NULL;
     }
-    tokens = strdup(line);
+    tokens = strdup_chk(line, "tokens");
     /* コメントを削除 */
     for(p = tokens; *p != '\0'; p++) {
         /* 「'」で囲まれた文字列の処理。「''」は無視 */
@@ -80,50 +119,52 @@ CMDLINE *linetok(const char *line)
             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 != '\n' && *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(p);
-        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, "");    /* no command in the line */
+            }
+            FREE(cmdl);
+        } 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(p);
-    p = sepp + 1;
-    while(*p == ' ' || *p == '\t') {
-        p++;
-    }
-    /* オペランドを取得 */
-    cmdl->opd = malloc(sizeof(OPD));
-    /* 改行かタブまでの文字列を取得。
-       「'」で囲まれた文字列に含まれる場合があるため、空白は無視 */
-    if((sepp = p + strcspn(p, "\t\n")) > p) {
-        *sepp = '\0';
-        cmdl->opd = opdtok(p);
-    } else {
-        cmdl->opd->opdc = 0;
     }
+    FREE(tokens);
     return cmdl;
 }